|
| 1 | +from ble.blerepl import uart |
| 2 | +import sys |
| 3 | +from micropython import const |
| 4 | + |
| 5 | +class Gamepad: |
| 6 | + |
| 7 | + _DEFAULT_GAMEPAD_INSTANCE = None |
| 8 | + |
| 9 | + X1 = const(0) |
| 10 | + Y1 = const(1) |
| 11 | + X2 = const(2) |
| 12 | + Y2 = const(3) |
| 13 | + BUTTON_A = const(4) |
| 14 | + BUTTON_B = const(5) |
| 15 | + BUTTON_X = const(6) |
| 16 | + BUTTON_Y = const(7) |
| 17 | + BUMPER_L = const(8) |
| 18 | + BUMPER_R = const(9) |
| 19 | + TRIGGER_L = const(10) |
| 20 | + TRIGGER_R = const(11) |
| 21 | + BACK = const(12) |
| 22 | + START = const(13) |
| 23 | + DPAD_UP = const(14) |
| 24 | + DPAD_DN = const(15) |
| 25 | + DPAD_L = const(16) |
| 26 | + DPAD_R = const(17) |
| 27 | + |
| 28 | + _joyData = [ |
| 29 | + 0.0, |
| 30 | + 0.0, |
| 31 | + 0.0, |
| 32 | + 0.0, |
| 33 | + 0,0,0,0,0,0,0,0,0,0,0,0,0,0] |
| 34 | + |
| 35 | + @classmethod |
| 36 | + def get_default_gamepad(cls): |
| 37 | + """ |
| 38 | + Get the default XRP bluetooth joystick instance. This is a singleton, so only one instance of the gamepad sensor will ever exist. |
| 39 | + """ |
| 40 | + if cls._DEFAULT_GAMEPAD_INSTANCE is None: |
| 41 | + cls._DEFAULT_GAMEPAD_INSTANCE = cls() |
| 42 | + cls._DEFAULT_GAMEPAD_INSTANCE.start() |
| 43 | + return cls._DEFAULT_GAMEPAD_INSTANCE |
| 44 | + |
| 45 | + def __init__(self): |
| 46 | + """ |
| 47 | + Manages communication with gamepad data coming from a remote computer via bluetooth |
| 48 | +
|
| 49 | + """ |
| 50 | + def start(self): |
| 51 | + """ |
| 52 | + Signals the remote computer to begin sending gamepad data packets. |
| 53 | + """ |
| 54 | + for i in range(len(self._joyData)): |
| 55 | + self._joyData[i] = 0.0 |
| 56 | + uart.set_data_callback(self._data_callback) |
| 57 | + sys.stdout.write(chr(27)) |
| 58 | + sys.stdout.write(chr(101)) |
| 59 | + |
| 60 | + |
| 61 | + def stop(self): |
| 62 | + """ |
| 63 | + Signals the remote computer to stop sending gamepad data packets. |
| 64 | + """ |
| 65 | + sys.stdout.write(chr(27)) |
| 66 | + sys.stdout.write(chr(102)) |
| 67 | + |
| 68 | + def get_value(self, index:int) -> float: |
| 69 | + """ |
| 70 | + Get the current value of a joystick axis |
| 71 | +
|
| 72 | + :param index: The joystick axis index |
| 73 | + Gamepad.X1, Gamepad.Y1, Gamepad.X2, Gamepad.Y2 |
| 74 | + :type int |
| 75 | + :returns: The value of the joystick between -1 and 1 |
| 76 | + :rtype: float |
| 77 | + """ |
| 78 | + return -self._joyData[index] #returning the negative to make normal for user |
| 79 | + |
| 80 | + def is_button_pressed(self, index:int) -> bool: |
| 81 | + """ |
| 82 | + Checks if a specific button is currently pressed. |
| 83 | +
|
| 84 | + :param index: The button index |
| 85 | + Gamepad.BUTTON_A, Gamepad.TRIGGER_L, Gamepad.DPAD_UP, etc |
| 86 | + :type int |
| 87 | + :returns: The value of the button 1 or 0 |
| 88 | + :rtype: bool |
| 89 | + """ |
| 90 | + return self._joyData[index] > 0 |
| 91 | + |
| 92 | + def _data_callback(self, data): |
| 93 | + if(data[0] == 0x55 and len(data) == data[1] + 2): |
| 94 | + for i in range(2, data[1] + 2, 2): |
| 95 | + self._joyData[data[i]] = round(data[i + 1]/127.5 - 1, 2) |
| 96 | + |
| 97 | + |
0 commit comments