468 lines
15 KiB
Python
468 lines
15 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
VR功能测试脚本
|
||
测试VR系统的各种功能，包括模拟模式和真实VR模式
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
from main import MyWorld
|
||
|
||
def test_basic_vr_functionality():
|
||
"""测试基本VR功能"""
|
||
print("=== VR基本功能测试 ===")
|
||
|
||
# 创建世界实例
|
||
world = MyWorld()
|
||
|
||
# 检查VR管理器是否正确初始化
|
||
print("✓ VR管理器已创建")
|
||
|
||
# 检查OpenVR库是否可用
|
||
try:
|
||
import openvr
|
||
print("✓ OpenVR库已安装")
|
||
except ImportError:
|
||
print("⚠ OpenVR库未安装，将使用模拟模式")
|
||
|
||
# 测试VR初始化（会自动回退到模拟模式）
|
||
print("\n正在测试VR初始化...")
|
||
vr_success = world.initializeVR()
|
||
|
||
if vr_success:
|
||
print("✓ VR系统初始化成功")
|
||
|
||
# 获取VR信息
|
||
vr_info = world.getVRInfo()
|
||
print(f"\n📊 VR系统信息:")
|
||
print(f" - 模式: {vr_info['mode']}")
|
||
print(f" - 启用状态: {vr_info['enabled']}")
|
||
print(f" - 模拟模式: {vr_info['simulation_mode']}")
|
||
print(f" - 渲染尺寸: {vr_info['render_size']}")
|
||
print(f" - 控制器数量: {vr_info.get('controllers', 0)}")
|
||
|
||
# 测试控制器输入
|
||
print("\n🎮 测试控制器输入:")
|
||
for i in range(2):
|
||
controller_input = world.vr_manager.get_controller_input(i)
|
||
if controller_input and controller_input.get('connected'):
|
||
print(f" 控制器 {i}: 已连接")
|
||
print(f" - 扳机: {controller_input['trigger']:.2f}")
|
||
print(f" - 握把: {controller_input['grip']:.2f}")
|
||
print(f" - 触摸板: ({controller_input['touchpad']['x']:.2f}, {controller_input['touchpad']['y']:.2f})")
|
||
else:
|
||
print(f" 控制器 {i}: 未连接")
|
||
|
||
# 测试VR状态
|
||
print("\n📊 VR系统状态:")
|
||
status = world.getVRStatus()
|
||
print(f" - VR启用: {status['vr_enabled']}")
|
||
print(f" - 控制器总数: {len(status['controllers'])}")
|
||
|
||
else:
|
||
print("✗ VR系统初始化失败")
|
||
return False
|
||
|
||
# 关闭VR系统
|
||
world.shutdownVR()
|
||
print("\n✓ VR系统已关闭")
|
||
|
||
return True
|
||
|
||
def test_simulation_mode():
|
||
"""测试VR模拟模式"""
|
||
print("=== VR模拟模式测试 ===")
|
||
|
||
# 创建世界实例
|
||
world = MyWorld()
|
||
|
||
# 强制启用模拟模式
|
||
print("强制启用VR模拟模式...")
|
||
vr_success = world.vr_manager.enable_simulation_mode()
|
||
|
||
if vr_success:
|
||
print("✓ VR模拟模式启用成功")
|
||
|
||
# 获取模拟数据
|
||
sim_data = world.vr_manager.get_simulation_data()
|
||
if sim_data:
|
||
print(f"\n🎮 模拟数据:")
|
||
print(f" - 头部位置: {sim_data['head_pose']['position']}")
|
||
print(f" - 控制器 0 位置: {sim_data['controller_poses'][0]['position']}")
|
||
print(f" - 控制器 1 位置: {sim_data['controller_poses'][1]['position']}")
|
||
print(f" - 渲染尺寸: {sim_data['render_size']}")
|
||
|
||
# 测试更新模拟数据
|
||
print(f"\n🔧 测试模拟数据更新:")
|
||
new_head_pos = [0.1, 0.1, 1.7]
|
||
success = world.vr_manager.update_simulation_data('head_pose', {
|
||
'position': new_head_pos,
|
||
'rotation': [0, 0, 0, 1]
|
||
})
|
||
if success:
|
||
print(f" ✓ 头部位置更新为: {new_head_pos}")
|
||
|
||
# 测试控制器输入（模拟）
|
||
print(f"\n🎮 模拟控制器输入测试:")
|
||
for i in range(2):
|
||
controller_input = world.vr_manager.get_controller_input(i)
|
||
if controller_input:
|
||
print(f" 控制器 {i}:")
|
||
print(f" - 连接状态: {controller_input['connected']}")
|
||
print(f" - 扳机: {controller_input['trigger']}")
|
||
print(f" - 握把: {controller_input['grip']}")
|
||
print(f" - 菜单键: {controller_input['menu']}")
|
||
|
||
else:
|
||
print("✗ VR模拟模式启用失败")
|
||
return False
|
||
|
||
# 关闭VR系统
|
||
world.vr_manager.shutdown_vr()
|
||
print("\n✓ VR模拟模式已关闭")
|
||
|
||
return True
|
||
|
||
def test_alvr_streaming():
|
||
"""测试ALVR串流功能"""
|
||
print("=== ALVR串流测试 ===")
|
||
|
||
# 创建世界实例
|
||
world = MyWorld()
|
||
|
||
# 初始化VR系统
|
||
print("初始化VR系统...")
|
||
if not world.initializeVR():
|
||
print("✗ VR系统初始化失败")
|
||
return False
|
||
|
||
# 测试ALVR初始化
|
||
print("\n正在测试ALVR初始化...")
|
||
alvr_success = world.initializeALVR()
|
||
|
||
if alvr_success:
|
||
print("✓ ALVR初始化成功")
|
||
|
||
# 测试串流状态
|
||
print("\n📊 ALVR串流状态:")
|
||
status = world.getALVRStreamingStatus()
|
||
print(f" - 连接状态: {world.isALVRConnected()}")
|
||
print(f" - 串流状态: {world.isALVRStreaming()}")
|
||
|
||
# 测试串流质量设置
|
||
print("\n🎥 测试串流质量设置:")
|
||
quality_success = world.setALVRStreamQuality(1920, 1080, 60, 100)
|
||
if quality_success:
|
||
print(" ✓ 串流质量设置成功: 1920x1080@60fps, 100Mbps")
|
||
else:
|
||
print(" ⚠ 串流质量设置失败")
|
||
|
||
# 测试触觉反馈
|
||
print("\n🔔 测试触觉反馈:")
|
||
for controller_id in range(2):
|
||
feedback_success = world.sendHapticFeedback(controller_id, 0.1, 0.5)
|
||
if feedback_success:
|
||
print(f" ✓ 控制器 {controller_id} 触觉反馈发送成功")
|
||
else:
|
||
print(f" ⚠ 控制器 {controller_id} 触觉反馈发送失败")
|
||
|
||
# 测试串流控制
|
||
print("\n🎮 测试串流控制:")
|
||
start_success = world.startALVRStreaming()
|
||
if start_success:
|
||
print(" ✓ 串流开始成功")
|
||
else:
|
||
print(" ⚠ 串流开始失败")
|
||
|
||
# 等待一下然后停止
|
||
import time
|
||
time.sleep(1)
|
||
|
||
stop_success = world.stopALVRStreaming()
|
||
if stop_success:
|
||
print(" ✓ 串流停止成功")
|
||
else:
|
||
print(" ⚠ 串流停止失败")
|
||
|
||
else:
|
||
print("⚠ ALVR初始化失败（这在没有ALVR服务器时是正常的）")
|
||
print(" 提示: 要使用ALVR功能，请:")
|
||
print(" 1. 下载并安装ALVR服务器")
|
||
print(" 2. 启动ALVR服务器")
|
||
print(" 3. 在Quest头盔上安装并启动ALVR客户端")
|
||
|
||
# 关闭系统
|
||
world.shutdownALVR()
|
||
world.shutdownVR()
|
||
print("\n✓ ALVR和VR系统已关闭")
|
||
|
||
return True
|
||
|
||
def test_vr_gui_control_panel():
|
||
"""测试VR GUI控制面板"""
|
||
print("=== VR GUI控制面板测试 ===")
|
||
|
||
try:
|
||
from ui.vr_control_panel import VRControlPanel
|
||
from PyQt5.QtWidgets import QApplication
|
||
|
||
# 创建Qt应用程序
|
||
app = QApplication.instance()
|
||
if app is None:
|
||
app = QApplication(sys.argv)
|
||
|
||
# 创建世界实例
|
||
world = MyWorld()
|
||
|
||
# 创建VR控制面板
|
||
print("创建VR控制面板...")
|
||
control_panel = VRControlPanel(world)
|
||
|
||
# 测试面板功能
|
||
print("✓ VR控制面板创建成功")
|
||
print(" - 面板标题:", control_panel.windowTitle())
|
||
print(" - 面板大小:", control_panel.size().width(), "x", control_panel.size().height())
|
||
|
||
# 显示面板（非阻塞）
|
||
control_panel.show()
|
||
print("✓ VR控制面板已显示")
|
||
|
||
print("\n💡 GUI控制面板功能:")
|
||
print(" - VR系统开关控制")
|
||
print(" - ALVR串流控制")
|
||
print(" - 实时状态监控")
|
||
print(" - 质量设置调整")
|
||
print(" - 控制器状态显示")
|
||
print(" - 性能监控")
|
||
|
||
# 短暂显示然后关闭
|
||
import time
|
||
time.sleep(2)
|
||
control_panel.close()
|
||
|
||
print("\n✓ VR控制面板测试完成")
|
||
|
||
except Exception as e:
|
||
print(f"✗ VR控制面板测试失败: {str(e)}")
|
||
return False
|
||
|
||
return True
|
||
|
||
def test_vr_interaction():
|
||
"""测试VR交互功能"""
|
||
print("=== VR交互功能测试 ===")
|
||
|
||
# 创建世界实例
|
||
world = MyWorld()
|
||
|
||
# 初始化VR系统
|
||
print("初始化VR系统...")
|
||
if not world.initializeVR():
|
||
print("✗ VR系统初始化失败")
|
||
return False
|
||
|
||
# 测试VR输入处理
|
||
print("\n🎮 测试VR输入处理:")
|
||
input_success = world.startVRInput()
|
||
if input_success:
|
||
print(" ✓ VR输入处理启动成功")
|
||
|
||
# 测试控制器射线显示
|
||
print("\n🌟 测试控制器射线:")
|
||
ray_success = world.showControllerRays(True)
|
||
if ray_success:
|
||
print(" ✓ 控制器射线显示启用")
|
||
else:
|
||
print(" ⚠ 控制器射线显示失败")
|
||
|
||
# 测试手势识别
|
||
print("\n✋ 测试手势识别:")
|
||
gesture_success = world.setVRGestureEnabled(True)
|
||
if gesture_success:
|
||
print(" ✓ VR手势识别启用")
|
||
else:
|
||
print(" ⚠ VR手势识别启用失败")
|
||
|
||
# 测试VR交互
|
||
print("\n🤏 测试VR交互:")
|
||
interaction_success = world.setVRInteractionEnabled(True)
|
||
if interaction_success:
|
||
print(" ✓ VR交互功能启用")
|
||
else:
|
||
print(" ⚠ VR交互功能启用失败")
|
||
|
||
# 获取控制器状态
|
||
print("\n🎮 控制器状态:")
|
||
controllers = world.getAllControllers()
|
||
for i, controller in enumerate(controllers):
|
||
if controller:
|
||
print(f" 控制器 {i}: 可用")
|
||
state = world.getControllerState(i)
|
||
if state:
|
||
print(f" - 位置: {state.get('position', 'N/A')}")
|
||
print(f" - 旋转: {state.get('rotation', 'N/A')}")
|
||
print(f" - 按钮: {state.get('buttons', 'N/A')}")
|
||
else:
|
||
print(f" 控制器 {i}: 不可用")
|
||
|
||
# 停止VR输入
|
||
world.stopVRInput()
|
||
print("\n✓ VR输入处理已停止")
|
||
|
||
else:
|
||
print(" ⚠ VR输入处理启动失败")
|
||
|
||
# 关闭VR系统
|
||
world.shutdownVR()
|
||
print("\n✓ VR系统已关闭")
|
||
|
||
return True
|
||
|
||
def run_all_tests():
|
||
"""运行所有测试"""
|
||
print("=== 运行所有VR测试 ===")
|
||
|
||
tests = [
|
||
("基本VR功能", test_basic_vr_functionality),
|
||
("VR模拟模式", test_simulation_mode),
|
||
("ALVR串流", test_alvr_streaming),
|
||
("VR GUI控制面板", test_vr_gui_control_panel),
|
||
("VR交互功能", test_vr_interaction)
|
||
]
|
||
|
||
results = []
|
||
for test_name, test_func in tests:
|
||
print(f"\n{'='*50}")
|
||
print(f"正在运行: {test_name}")
|
||
print('='*50)
|
||
|
||
try:
|
||
success = test_func()
|
||
results.append((test_name, success))
|
||
if success:
|
||
print(f"✓ {test_name} 测试通过")
|
||
else:
|
||
print(f"✗ {test_name} 测试失败")
|
||
except Exception as e:
|
||
print(f"✗ {test_name} 测试出错: {str(e)}")
|
||
results.append((test_name, False))
|
||
|
||
# 打印总结
|
||
print(f"\n{'='*50}")
|
||
print("测试总结")
|
||
print('='*50)
|
||
|
||
passed = sum(1 for _, success in results if success)
|
||
total = len(results)
|
||
|
||
for test_name, success in results:
|
||
status = "✓ 通过" if success else "✗ 失败"
|
||
print(f"{test_name}: {status}")
|
||
|
||
print(f"\n总计: {passed}/{total} 个测试通过")
|
||
|
||
if passed == total:
|
||
print("🎉 所有测试都通过了！")
|
||
else:
|
||
print("⚠ 部分测试失败，请检查上述输出")
|
||
|
||
return passed == total
|
||
|
||
def print_vr_requirements():
|
||
"""打印VR系统要求"""
|
||
print("📋 VR系统要求说明")
|
||
print("="*50)
|
||
print("🔧 软件要求:")
|
||
print(" - Python 3.8+")
|
||
print(" - Panda3D")
|
||
print(" - PyQt5")
|
||
print(" - OpenVR库 (pip install openvr)")
|
||
print(" - 其他依赖见 requirements/vr-requirements.txt")
|
||
|
||
print("\n🎮 硬件要求（完整VR模式）:")
|
||
print(" - 支持VR的显卡 (GTX 1060/RX 580 或更好)")
|
||
print(" - VR头盔 (Quest 2/3, Valve Index, HTC Vive等)")
|
||
print(" - 足够的USB端口或无线连接")
|
||
|
||
print("\n🌐 连接方式:")
|
||
print(" 有线连接:")
|
||
print(" - Valve Index: DisplayPort + USB 3.0")
|
||
print(" - HTC Vive: HDMI + USB 3.0")
|
||
print(" - Quest 2/3: USB-C (Quest Link)")
|
||
print(" 无线连接:")
|
||
print(" - Quest 2/3: 通过ALVR串流")
|
||
print(" - 需要5GHz Wi-Fi网络")
|
||
print(" - 需要ALVR服务器运行")
|
||
|
||
print("\n🔄 启动顺序（完整VR模式）:")
|
||
print(" 1. 确保VR头盔已连接并识别")
|
||
print(" 2. 启动SteamVR")
|
||
print(" 3. (可选) 启动ALVR服务器（Quest无线）")
|
||
print(" 4. 运行VR测试脚本")
|
||
|
||
print("\n🎮 模拟模式说明:")
|
||
print(" - 无需VR硬件")
|
||
print(" - 模拟头盔和控制器追踪")
|
||
print(" - 立体渲染到窗口")
|
||
print(" - 适用于开发和调试")
|
||
|
||
print("\n💡 故障排除:")
|
||
print(" - 如果OpenVR初始化失败，系统会自动切换到模拟模式")
|
||
print(" - 检查SteamVR是否正在运行")
|
||
print(" - 确认VR头盔被系统识别")
|
||
print(" - 检查USB/DisplayPort连接")
|
||
print(" - 重启SteamVR和头盔")
|
||
|
||
def main():
|
||
"""主函数"""
|
||
print("🎮 VR功能测试脚本")
|
||
print("="*50)
|
||
|
||
# 显示要求说明
|
||
print_vr_requirements()
|
||
|
||
print("\n请选择测试类型:")
|
||
print("1. 基本VR功能测试")
|
||
print("2. VR模拟模式测试")
|
||
print("3. ALVR串流测试")
|
||
print("4. VR GUI控制面板")
|
||
print("5. VR交互功能测试")
|
||
print("6. 运行所有测试")
|
||
print("7. 查看VR系统要求")
|
||
|
||
try:
|
||
choice = input("请输入选择 (1-7): ")
|
||
|
||
if choice == "1":
|
||
success = test_basic_vr_functionality()
|
||
elif choice == "2":
|
||
success = test_simulation_mode()
|
||
elif choice == "3":
|
||
success = test_alvr_streaming()
|
||
elif choice == "4":
|
||
success = test_vr_gui_control_panel()
|
||
elif choice == "5":
|
||
success = test_vr_interaction()
|
||
elif choice == "6":
|
||
success = run_all_tests()
|
||
elif choice == "7":
|
||
print_vr_requirements()
|
||
return
|
||
else:
|
||
print("无效的选择")
|
||
return
|
||
|
||
print("\n" + "="*50)
|
||
if success:
|
||
print("✓ 测试成功")
|
||
else:
|
||
print("✗ 测试失败")
|
||
|
||
except KeyboardInterrupt:
|
||
print("\n用户中断测试")
|
||
except Exception as e:
|
||
print(f"测试过程中发生错误: {str(e)}")
|
||
|
||
if __name__ == "__main__":
|
||
main() |