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
|
import pyglet
from pyglet.shapes import Circle, Rectangle, Arc
window = pyglet.window.Window(720, 480)
batch = pyglet.graphics.Batch()
@window.event
def on_draw():
window.clear()
batch.draw()
class ControllerDisplay:
"""A class to visualize all the Controller inputs."""
def __init__(self, batch):
self.label = pyglet.text.Label("No Controller connected.", x=10, y=window.height - 20,
multiline=True, width=720, batch=batch)
self.left_trigger = Rectangle(70, 310, 40, 10, batch=batch)
self.right_trigger = Rectangle(610, 310, 40, 10, batch=batch)
self.d_pad = Rectangle(280, 185, 10, 10, batch=batch)
self.left_stick = Arc(180, 240, 20, batch=batch)
self.left_stick_label = pyglet.text.Label("(0.00, 0.00)", x=180, y=50, anchor_x='center', batch=batch)
self.left_stick_bar_x = Rectangle(180, 30, 0, 10, batch=batch)
self.left_stick_bar_y = Rectangle(180, 10, 0, 10, batch=batch)
self.right_stick = Arc(540, 240, 20, batch=batch)
self.right_stick_label = pyglet.text.Label("(0.00, 0.00)", x=540, y=50, anchor_x='center', batch=batch)
self.right_stick_bar_x = Rectangle(540, 30, 0, 10, batch=batch)
self.right_stick_bar_y = Rectangle(540, 10, 0, 10, batch=batch)
self.l_outline1 = Arc(180, 240, 75, color=(44, 44, 44), batch=batch)
self.l_outline2 = Arc(285, 190, 35, color=(44, 44, 44), batch=batch)
self.r_outline1 = Arc(540, 240, 75, color=(44, 44, 44), batch=batch)
self.r_outline2 = Arc(435, 190, 35, color=(44, 44, 44), batch=batch)
self.buttons = {'a': Circle(435, 170, 9, color=(124, 178, 232), batch=batch),
'b': Circle(455, 190, 9, color=(255, 102, 102), batch=batch),
'x': Circle(415, 190, 9, color=(255, 105, 248), batch=batch),
'y': Circle(435, 210, 9, color=(64, 226, 160), batch=batch),
'leftshoulder': Rectangle(70, 290, 40, 10, batch=batch),
'rightshoulder': Rectangle(610, 290, 40, 10, batch=batch),
'start': Circle(390, 240, 9, batch=batch),
'guide': Circle(360, 240, 9, color=(255, 255, 100), batch=batch),
'back': Circle(330, 240, 9, batch=batch),
'leftstick': Circle(180, 240, 9, batch=batch),
'rightstick': Circle(540, 240, 9, batch=batch)}
for button in self.buttons.values():
button.visible = False
def on_button_press(self, controller, button_name):
if button := self.buttons.get(button_name, None):
button.visible = True
controller.rumble_play_weak(1.0, 0.1)
def on_button_release(self, controller, button_name):
if button := self.buttons.get(button_name, None):
button.visible = False
def on_dpad_motion(self, controller, dpleft, dpright, dpup, dpdown):
position = [280, 185]
if dpup:
position[1] += 25
if dpdown:
position[1] -= 25
if dpleft:
position[0] -= 25
if dpright:
position[0] += 25
self.d_pad.position = position
def on_stick_motion(self, controller, stick, xvalue, yvalue):
if stick == "leftstick":
self.left_stick.position = 180+xvalue*50, 240+yvalue*50
self.left_stick_label.text = f"({xvalue:.2f}, {yvalue:.2f})"
self.left_stick_bar_x.width = xvalue * 100
self.left_stick_bar_y.width = yvalue * 100
elif stick == "rightstick":
self.right_stick.position = 540+xvalue*50, 240+yvalue*50
self.right_stick_label.text = f"({xvalue:.2f}, {yvalue:.2f})"
self.right_stick_bar_x.width = xvalue * 100
self.right_stick_bar_y.width = yvalue * 100
def on_trigger_motion(self, controller, trigger, value):
if trigger == "lefttrigger":
self.left_trigger.y = 310 + (value*50)
controller.rumble_play_weak(value, duration=5)
elif trigger == "righttrigger":
self.right_trigger.y = 310 + (value*50)
controller.rumble_play_strong(value, duration=5)
controller_display = ControllerDisplay(batch=batch)
def on_connect(controller):
controller.open()
controller.rumble_play_weak(1.0, 0.1)
controller_display.label.text = f"Detected: {controller.name}\nController GUID: {controller.guid}"
controller.push_handlers(controller_display)
def on_disconnect(controller):
controller_display.label.text = "No Controller connected."
controller.remove_handlers(controller_display)
# ControllerManager instance to handle hot-plugging:
controller_manager = pyglet.input.ControllerManager()
controller_manager.on_connect = on_connect
controller_manager.on_disconnect = on_disconnect
# Handle already connected controller:
if controllers := controller_manager.get_controllers():
on_connect(controllers[0])
pyglet.app.run()
|
