#!/bin/bash
# Graphics Workload Validation
# Tests Vulkan/OpenGL/video workloads in the guest

set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
OPEN_IOV_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"

echo "=== Graphics Workload Validation ==="
echo "Open-IOV root: $OPEN_IOV_ROOT"

# Check if we're root
if [ "$EUID" -ne 0 ]; then
    echo "ERROR: This script must be run as root"
    exit 1
fi

GUEST_SSH="ssh -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null root@localhost -p 2222"

echo "Testing Vulkan..."
$GUEST_SSH "vulkaninfo | head -50"
$GUEST_SSH "vkcube --duration 5" || echo "vkcube not available or failed"

echo "Testing OpenGL..."
$GUEST_SSH "glxinfo | grep -i nvidia"
$GUEST_SSH "glxgears -info" || echo "glxgears not available"

echo "Testing video encode/decode..."
$GUEST_SSH "ffmpeg -encoders | grep nvenc" || echo "NVENC not available"
$GUEST_SSH "ffmpeg -decoders | grep nvdec" || echo "NVDEC not available"

echo "=== Graphics Workload Validation Complete ==="