forked from michaelliao/learn-python3
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathjoystick.py
More file actions
61 lines (53 loc) · 1.98 KB
/
joystick.py
File metadata and controls
61 lines (53 loc) · 1.98 KB
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
# joystick control:
import struct
# define button code:
BUTTON_A = 305
BUTTON_B = 304
BUTTON_X = 307
BUTTON_Y = 306
BUTTON_PLUS = 313
BUTTON_MINUS = 312
BUTTON_START = 317
BUTTON_HOME = 316
class JoyStick():
def __init__(self, eventFile):
self.eventFile = eventFile
self.buttonHandler = None
self.joyLeftHandler = None
self.joyRightHandler = None
def setButtonHandler(self, buttonHandler):
self.buttonHandler = buttonHandler
def setJoyLeftHandler(self, joyLeftHandler):
self.joyLeftHandler = joyLeftHandler
def setJoyRightHandler(self, joyRightHandler):
self.joyRightHandler = joyRightHandler
def startLoop(self):
FORMAT = 'llHHI'
EVENT_SIZE = struct.calcsize(FORMAT)
with open(self.eventFile, 'rb') as infile:
lx, ly, rx, ry = 0, 0, 0, 0
while True:
event = infile.read(EVENT_SIZE)
_, _, t, c, v = struct.unpack(FORMAT, event)
if t == 1 and v == 1:
# button pressed:
if self.buttonHandler:
if not self.buttonHandler(c):
return
if t == 3:
if c == 0 and self.joyLeftHandler:
# left stick & horizontal:
lx = v - 32768
self.joyLeftHandler(lx, ly)
elif c == 1 and self.joyLeftHandler:
# left stick & vertical:
ly = v - 32768
self.joyLeftHandler(lx, ly)
elif c == 3 and self.joyRightHandler:
# right stick & horizontal:
rx = v - 32768
self.joyRightHandler(rx, ry)
elif c == 4 and self.joyRightHandler:
# right stick & vertical:
ry = v - 32768
self.joyRightHandler(rx, ry)