1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
import openvr
import sys
import time
import typing
EVENT_RISING_EDGE = 0
EVENT_FALLING_EDGE = 1
EVENT_POSE = 2
class InputEvent:
def __init__(self,
opcode: int,
pos: typing.Tuple[float,float,float] = None):
self.opcode = opcode
self.pos = pos
def __str__(self):
if self.opcode == EVENT_RISING_EDGE:
return "EVENT_RISING_EDGE"
elif self.opcode == EVENT_FALLING_EDGE:
return "EVENT_FALLING_EDGE"
elif self.opcode == EVENT_POSE:
return f"EVENT_POSE: {self.pos}"
def pollButtonPress(hand: str = "right", button: str = "b") -> InputEvent:
openvr.init(openvr.VRApplication_Overlay)
system = openvr.VRSystem()
button_mapping = {
'a': k_EButton_Index_Controller_A,
'b': k_EButton_Index_Controller_B,
'thumbstick': k_EButton_SteamVR_Touchpad,
}
print("SteamVR session created. Listening for controller input...")
while True:
# Drain input events.
event = openvr.VREvent_t()
while system.pollNextEvent(event):
# Event processing, e.g. button presses, goes here
if event.eventType == openvr.VREvent_ButtonPress or \
event.eventType == openvr.VREvent_ButtonUnpress:
print(f"event.data.controller.button: {event.data.controller.button}")
continue
print(f"event: {dir(event)}")
print(f"event.data: {dir(event.data)}")
print(f"event.data.controller: {dir(event.data.controller)}")
print(f"event.data.controller.button: {event.data.controller.button}")
print(f"event.data.keyboard: {dir(event.data.keyboard)}")
print(f"event.data.keyboard.cNewInput: {int.from_bytes(event.data.keyboard.cNewInput, byteorder='little')}")
print(f"event.data.keyboard.uUserValue: {event.data.keyboard.uUserValue}")
print(f"event.data.mouse: {dir(event.data.mouse)}")
print(f"event.data.mouse.button: {event.data.mouse.button}")
print(f"event.data.touchPadMove: {dir(event.data.touchPadMove)}")
print(f"event.data.touchPadMove.bFingerDown: {event.data.touchPadMove.bFingerDown}")
is_rising = event.eventType == openvr.VREvent_ButtonPress
# Check if the intended button is pressed
if button == 'thumbstick':
_, controller_state = system.getControllerState(event.trackedDeviceIndex)
mouse_x = controller_state.rAxis[0].x
mouse_y = controller_state.rAxis[0].y
print(f"mouse x/y: {mouse_x}/{mouse_y}")
print(f"mouse rad: {mouse_x**2 + mouse_y**2}")
dead_zone_radius = 0.05
thumbstick_moved = mouse_x**2 + mouse_y**2 > dead_zone_radius**2
if event.data.controller.button == button_mapping['thumbstick'] and not thumbstick_moved:
if is_rising:
yield InputEvent(EVENT_RISING_EDGE)
else:
yield InputEvent(EVENT_FALLING_EDGE)
elif event.data.controller.button == button_mapping[button]:
if is_rising:
yield InputEvent(EVENT_RISING_EDGE)
else:
yield InputEvent(EVENT_FALLING_EDGE)
# Check poses.
# TODO(yum) use this. Thinking about adding gestures: swipe to scale
# up/down, etc.
if False:
poses = (openvr.TrackedDevicePose_t * openvr.k_unMaxTrackedDeviceCount)()
system.getDeviceToAbsoluteTrackingPose(openvr.TrackingUniverseStanding, 0, poses)
pose = None
for i in range(openvr.k_unMaxTrackedDeviceCount):
if system.getControllerRoleForTrackedDeviceIndex(i) == openvr.TrackedControllerRole_RightHand:
pose = poses[i]
if pose and pose.bPoseIsValid:
position = pose.mDeviceToAbsoluteTracking.m[0][3], \
pose.mDeviceToAbsoluteTracking.m[1][3], \
pose.mDeviceToAbsoluteTracking.m[2][3]
yield InputEvent(EVENT_POSE, pos=position)
# Max out a 100 Hz.
time.sleep(0.01)
openvr.shutdown()
if __name__ == "__main__":
if len(sys.argv) != 3:
print("Usage: script_name.py [left|right] [a|b|thumbstick]")
sys.exit(1)
hand = sys.argv[1]
button = sys.argv[2]
gen = pollButtonPress(hand, button)
while True:
print(next(gen))
|
