-
Notifications
You must be signed in to change notification settings - Fork 4
/
gamepad.py
141 lines (130 loc) · 7.51 KB
/
gamepad.py
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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
#!/usr/bin/python3
import asyncio
from evdev import InputDevice, ecodes, ff, list_devices
class gamepad():
def __init__(self, file = '/dev/input/event3'):
#self.event_value = 0
self.power_on = True
self.device_file = InputDevice(file)
self.joystick_left_y = 0 # values are mapped to [-1 ... 1]
self.joystick_left_x = 0 # values are mapped to [-1 ... 1]
self.joystick_right_x = 0 # values are mapped to [-1 ... 1]
self.joystick_right_y = 0 # values are mapped to [-1 ... 1]
self.trigger_right = 0 # values are mapped to [0 ... 1]
self.trigger_left = 0 # values are mapped to [0 ... 1]
self.button_x = False
self.button_y = False
self.button_b = False
self.button_a = False
self.dpad_up = False
self.dpad_down = False
self.dpad_left = False
self.dpad_right = False
self.bump_left = False
self.bump_right = False
self.rumble_effect = 0
self.effect1_id = 0 # light rumble, played continuously
self.effect2_id = 0 # strong rumble, played once
self.load_effects()
def load_effects(self):
#effect 1, light rumble
rumble = ff.Rumble(strong_magnitude=0x0000, weak_magnitude=0x500)
duration_ms = 300
effect = ff.Effect(ecodes.FF_RUMBLE, -1, 0, ff.Trigger(0, 0), ff.Replay(duration_ms, 0), ff.EffectType(ff_rumble_effect=rumble))
self.effect1_id = self.device_file.upload_effect(effect)
# effect 2, strong rumble
rumble = ff.Rumble(strong_magnitude=0xc000, weak_magnitude=0x0000)
duration_ms = 200
effect = ff.Effect(ecodes.FF_RUMBLE, -1, 0, ff.Trigger(0, 0), ff.Replay(duration_ms, 0), ff.EffectType(ff_rumble_effect=rumble))
self.effect2_id = self.device_file.upload_effect(effect)
async def read_gamepad_input(self): # asyncronus read-out of events
max_abs_joystick_left_x = 0xFFFF/2
uncertainty_joystick_left_x = 2500
max_abs_joystick_left_y = 0xFFFF/2
uncertainty_joystick_left_y = 2500
max_abs_joystick_right_x = 0xFFFF/2
uncertainty_joystick_right_x = 2000
max_abs_joystick_right_y = 0xFFFF/2
uncertainty_joystick_right_y = 2000
max_trigger = 1023
async for event in self.device_file.async_read_loop():
if not(self.power_on): #stop reading device when power_on = false
break
#print(str(event.type) + ' ' + str(event.code) + ' ' + str(event.value))
if event.type == 3: # type is analog trigger or joystick
if event.code == 1: # left joystick y-axis
if -event.value > uncertainty_joystick_left_y:
self.joystick_left_y = (-event.value - uncertainty_joystick_left_y) / (max_abs_joystick_left_y - uncertainty_joystick_left_y + 1)
elif -event.value < -uncertainty_joystick_left_y:
self.joystick_left_y = (-event.value + uncertainty_joystick_left_y) / (max_abs_joystick_left_y - uncertainty_joystick_left_y + 1)
else:
self.joystick_left_y = 0
elif event.code == 0: # left joystick x-axis
if event.value > uncertainty_joystick_left_x:
self.joystick_left_x = (event.value - uncertainty_joystick_left_x) / (max_abs_joystick_left_x - uncertainty_joystick_left_x + 1)
elif event.value < -uncertainty_joystick_left_x:
self.joystick_left_x = (event.value + uncertainty_joystick_left_x) / (max_abs_joystick_left_x - uncertainty_joystick_left_x + 1)
else:
self.joystick_left_x = 0
elif event.code == 3: # right joystick x-axis
if event.value > uncertainty_joystick_right_x:
self.joystick_right_x = (event.value - uncertainty_joystick_right_x) / (max_abs_joystick_right_x - uncertainty_joystick_right_x + 1)
elif event.value < -uncertainty_joystick_right_x:
self.joystick_right_x = (event.value + uncertainty_joystick_right_x) / (max_abs_joystick_right_x - uncertainty_joystick_right_x + 1)
else:
self.joystick_right_x = 0
elif event.code == 4: # right joystick y-axis
if -event.value > uncertainty_joystick_right_y:
self.joystick_right_y = (-event.value - uncertainty_joystick_right_y) / (max_abs_joystick_right_y - uncertainty_joystick_right_y + 1)
elif -event.value < -uncertainty_joystick_left_y:
self.joystick_right_y = (-event.value + uncertainty_joystick_right_y) / (max_abs_joystick_right_y - uncertainty_joystick_right_y + 1)
else:
self.joystick_right_y = 0
elif event.code == 5: # right trigger
self.trigger_right = event.value / max_trigger
elif event.code == 2: # left trigger
self.trigger_left = event.value / max_trigger
elif event.code == 16: # right trigger
if(event.value == -1):
self.dpad_left = True
self.dpad_right = False
elif(event.value == 1):
self.dpad_left = False
self.dpad_right = True
else:
self.dpad_left = False
self.dpad_right = False
elif event.code == 17: # left trigger
if(event.value == -1):
self.dpad_up = True
self.dpad_down = False
elif(event.value == 1):
self.dpad_up = False
self.dpad_down = True
else:
self.dpad_up = False
self.dpad_down = False
if (event.type == 1): # type is button
if event.code == 304: # button "A" pressed ?
self.button_a = True
if event.code == 307: # button "X" pressed ?
self.button_x = True
if event.code == 308: # button "Y" pressed ?
self.button_y = True
if event.code == 305: # button "B" pressed ?
self.button_b = True
if event.code == 311: # bumper "right" pressed ?
self.bump_right = True if event.value == 1 else False
if event.code == 310: # bumper "left" pressed ?
self.bump_left = True if event.value == 1 else False
async def rumble(self): # asyncronus control of force feed back effects
repeat_count = 1
while self.power_on:
if self.rumble_effect == 1:
self.device_file.write(ecodes.EV_FF, self.effect1_id, repeat_count)
elif self.rumble_effect == 2:
self.device_file.write(ecodes.EV_FF, self.effect2_id, repeat_count)
self.rumble_effect = 0 # turn of effect in order to play effect2 only once
await asyncio.sleep(0.2)
def erase_rumble(self):
self.device_file.erase_effect(self.effect1_id)