marc/ddgo-converter
1
0
Fork 0
mirror of https://github.com/marcriera/ddgo-converter.git synced 2025-04-11 07:29:28 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Initial support for classic controllers

Browse source
This commit is contained in:
Marc Riera 2022-11-18 19:35:15 +01:00
parent 6f1a890895
commit bed298ac99
1 changed files with 123 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

127
ddgo-converter/gamepads/physical.py
View file

@ -1,4 +1,4 @@
from enum import IntEnum
from enum import IntFlag, IntEnum, auto
import evdev
from hashlib import sha1
from events.input import InputEvent
@ -8,6 +8,8 @@ def create_gamepad(vid, pid, name):
match vid, pid:
case (0x0f0d, 0x00c1) | (0x33dd, 0x0002):
return SwitchGamepad(vid, pid, name)
case 0x054c, 0x0268:
return ClassicGamepad(vid, pid, name)
return PhysicalGamepad(vid, pid, name)
class PhysicalGamepad:
@ -64,9 +66,6 @@ class SwitchGamepad(PhysicalGamepad):
return
def read_input(self):
# time.sleep(5)
# print("Read from ZKNS-001 correct")
# return InputEvent(InputEvent.EventType.PRESS_BUTTON, InputEvent.Button.BUTTON_A)
select([self.device], [], [], 5)
try:
event = self.device.read_one()
@ -158,3 +157,123 @@ class SwitchGamepad(PhysicalGamepad):
return input_events
except OSError:
return [InputEvent(InputEvent.EventType.ERROR, None)]
class ClassicGamepad(PhysicalGamepad):
class ButtonType(IntEnum):
BUTTON = 0
POWER = 1
BRAKE = 2
class Buttons(IntFlag):
BUTTON_A = auto()
BUTTON_B = auto()
BUTTON_C = auto()
BUTTON_SELECT = auto()
BUTTON_START = auto()
class Power(IntFlag):
POWER1 = auto()
POWER2 = auto()
POWER3 = auto()
class Brake(IntFlag):
BRAKE1 = auto()
BRAKE2 = auto()
BRAKE3 = auto()
BRAKE4 = auto()
class ButtonConfig():
def __init__(self, type, code, button):
self.type = type
self.code = code
self.button = button
def __init__(self, * args):
super().__init__(* args)
self.type = self.GamepadType.CLASSIC
self.config = [ self.ButtonConfig(self.ButtonType.BUTTON, 308, self.Buttons.BUTTON_A), self.ButtonConfig(self.ButtonType.BUTTON, 304, self.Buttons.BUTTON_B), self.ButtonConfig(self.ButtonType.BUTTON, 305, self.Buttons.BUTTON_C), self.ButtonConfig(self.ButtonType.BUTTON, 314, self.Buttons.BUTTON_SELECT), self.ButtonConfig(self.ButtonType.BUTTON, 315, self.Buttons.BUTTON_START),
self.ButtonConfig(self.ButtonType.POWER, 307, self.Power.POWER1), self.ButtonConfig(self.ButtonType.POWER, 546, self.Power.POWER2), self.ButtonConfig(self.ButtonType.POWER, 547, self.Power.POWER3),
self.ButtonConfig(self.ButtonType.BRAKE, 310, self.Brake.BRAKE1), self.ButtonConfig(self.ButtonType.BRAKE, 312, self.Brake.BRAKE2), self.ButtonConfig(self.ButtonType.BRAKE, 311, self.Brake.BRAKE3), self.ButtonConfig(self.ButtonType.BRAKE, 313, self.Brake.BRAKE4) ]
self.device = self._get_device()
self.buttons = IntFlag(0)
self.power = IntFlag(0)
self.brake = IntFlag(0)
def start(self):
try:
self.device.grab()
except:
return
def stop(self):
try:
self.device.ungrab()
except:
return
def read_input(self):
select([self.device], [], [], 5)
try:
event = self.device.read_one()
input_events = []
if event is not None:
for button in self.config:
match button.type:
case self.ButtonType.POWER:
if event.type == evdev.ecodes.EV_KEY and event.code == button.code and event.value == 0:
self.power &= ~button.button
if event.type == evdev.ecodes.EV_KEY and event.code == button.code and event.value == 1:
self.power |= button.button
case self.ButtonType.BRAKE:
if event.type == evdev.ecodes.EV_KEY and event.code == button.code and event.value == 0:
self.brake &= ~button.button
if event.type == evdev.ecodes.EV_KEY and event.code == button.code and event.value == 1:
self.brake |= button.button
case self.ButtonType.BUTTON:
if event.type == evdev.ecodes.EV_KEY and event.code == button.code and event.value == 0:
self.buttons &= ~button.button
if event.type == evdev.ecodes.EV_KEY and event.code == button.code and event.value == 1:
self.buttons |= button.button
input_events.append(InputEvent(InputEvent.EventType(self.Buttons.BUTTON_A in self.buttons), InputEvent.Button.BUTTON_A))
input_events.append(InputEvent(InputEvent.EventType(self.Buttons.BUTTON_B in self.buttons), InputEvent.Button.BUTTON_B))
input_events.append(InputEvent(InputEvent.EventType(self.Buttons.BUTTON_C in self.buttons), InputEvent.Button.BUTTON_C))
input_events.append(InputEvent(InputEvent.EventType(self.Buttons.BUTTON_SELECT in self.buttons), InputEvent.Button.BUTTON_SELECT))
input_events.append(InputEvent(InputEvent.EventType(self.Buttons.BUTTON_START in self.buttons), InputEvent.Button.BUTTON_START))
match self.power:
case 6:
input_events.append(InputEvent(InputEvent.EventType.POWER_NOTCH, 0))
case 5:
input_events.append(InputEvent(InputEvent.EventType.POWER_NOTCH, 1))
case 4:
input_events.append(InputEvent(InputEvent.EventType.POWER_NOTCH, 2))
case 3:
input_events.append(InputEvent(InputEvent.EventType.POWER_NOTCH, 3))
case 2:
input_events.append(InputEvent(InputEvent.EventType.POWER_NOTCH, 4))
case 1:
input_events.append(InputEvent(InputEvent.EventType.POWER_NOTCH, 5))
match self.brake:
case 14:
input_events.append(InputEvent(InputEvent.EventType.BRAKE_NOTCH, 0))
case 13:
input_events.append(InputEvent(InputEvent.EventType.BRAKE_NOTCH, 1))
case 12:
input_events.append(InputEvent(InputEvent.EventType.BRAKE_NOTCH, 2))
case 11:
input_events.append(InputEvent(InputEvent.EventType.BRAKE_NOTCH, 3))
case 10:
input_events.append(InputEvent(InputEvent.EventType.BRAKE_NOTCH, 4))
case 9:
input_events.append(InputEvent(InputEvent.EventType.BRAKE_NOTCH, 5))
case 8:
input_events.append(InputEvent(InputEvent.EventType.BRAKE_NOTCH, 6))
case 7:
input_events.append(InputEvent(InputEvent.EventType.BRAKE_NOTCH, 7))
case 6:
input_events.append(InputEvent(InputEvent.EventType.BRAKE_NOTCH, 8))
case 0:
input_events.append(InputEvent(InputEvent.EventType.BRAKE_NOTCH, 9))
return input_events
except OSError:
return [InputEvent(InputEvent.EventType.ERROR, None)]