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
|
import usb.core
import usb.util
# Personal Communication Systems, Inc. SNES Gamepad
CONF_SNES_GAMEPAD = {
"idVendor": 0x0810,
"idProduct": 0xe501,
"iInterface": 0,
}
class Gamepad:
"Simple gamepad handle function"
def __init__(self, conf=CONF_SNES_GAMEPAD):
"Initializes usb subsystem"
self.dev = usb.core.find(idVendor=conf['idVendor'],
idProduct=conf['idProduct'])
if self.dev is None:
raise ValueError('Device not found')
if self.dev.is_kernel_driver_active(conf['iInterface']) is True:
# Detach any kernel driver so it won't interfere with us
self.dev.detach_kernel_driver(conf['iInterface'])
# set the active configuration. With no arguments, the first
# configuration will be the active one
self.dev.set_configuration()
# get an endpoint instance
self.cfg = self.dev.get_active_configuration()
intf = self.cfg[(0, 0)]
self.ep = usb.util.find_descriptor(
intf,
# match the first IN endpoint
custom_match=lambda e:
usb.util.endpoint_direction(e.bEndpointAddress) ==
usb.util.ENDPOINT_IN
)
assert self.ep is not None
def check(self):
"Check the input state"
data = self.dev.read(self.ep.bEndpointAddress,
self.ep.wMaxPacketSize*2, 1000).tolist()
return {
"left": data[3] < 120 or bool(data[5] & 0x80),
"right": data[3] > 140 or bool(data[5] & 0x20),
"up": data[4] < 120 or bool(data[5] & 0x10),
"down": data[4] > 140 or bool(data[5] & 0x40),
"select": bool(data[6] & 0x10),
"start": bool(data[6] & 0x20),
}
|
