ControlSocket/input.py at main · NaviVani-dev/ControlSocket · GitHub

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 vgamepad as vg
import sys

xbox_buttons = {
  "a": vg.XUSB_BUTTON.XUSB_GAMEPAD_A,
  "b": vg.XUSB_BUTTON.XUSB_GAMEPAD_B,
  "x": vg.XUSB_BUTTON.XUSB_GAMEPAD_X,
  "y": vg.XUSB_BUTTON.XUSB_GAMEPAD_Y,
  "start": vg.XUSB_BUTTON.XUSB_GAMEPAD_START,
  "select": vg.XUSB_BUTTON.XUSB_GAMEPAD_BACK,
  "home": vg.XUSB_BUTTON.XUSB_GAMEPAD_GUIDE,
  "l1": vg.XUSB_BUTTON.XUSB_GAMEPAD_LEFT_SHOULDER,
  "r1": vg.XUSB_BUTTON.XUSB_GAMEPAD_RIGHT_SHOULDER,
  "lthumb": vg.XUSB_BUTTON.XUSB_GAMEPAD_LEFT_THUMB,
  "rthumb": vg.XUSB_BUTTON.XUSB_GAMEPAD_RIGHT_THUMB,
  "dpad_up": vg.XUSB_BUTTON.XUSB_GAMEPAD_DPAD_UP,
  "dpad_down": vg.XUSB_BUTTON.XUSB_GAMEPAD_DPAD_DOWN,
  "dpad_left": vg.XUSB_BUTTON.XUSB_GAMEPAD_DPAD_LEFT,
  "dpad_right": vg.XUSB_BUTTON.XUSB_GAMEPAD_DPAD_RIGHT
}

ps4_buttons = {
  "a": vg.DS4_BUTTONS.DS4_BUTTON_CROSS,
  "b": vg.DS4_BUTTONS.DS4_BUTTON_CIRCLE,
  "x": vg.DS4_BUTTONS.DS4_BUTTON_SQUARE,
  "y": vg.DS4_BUTTONS.DS4_BUTTON_TRIANGLE,
  "start": vg.DS4_BUTTONS.DS4_BUTTON_SHARE,
  "select": vg.DS4_BUTTONS.DS4_BUTTON_OPTIONS,
  "home": vg.DS4_SPECIAL_BUTTONS.DS4_SPECIAL_BUTTON_PS,
  "l1": vg.DS4_BUTTONS.DS4_BUTTON_SHOULDER_LEFT,
  "r1": vg.DS4_BUTTONS.DS4_BUTTON_SHOULDER_RIGHT,
  "lthumb": vg.DS4_BUTTONS.DS4_BUTTON_TRIGGER_LEFT,
  "rthumb": vg.DS4_BUTTONS.DS4_BUTTON_TRIGGER_RIGHT,
  "dpad_none": vg.DS4_DPAD_DIRECTIONS.DS4_BUTTON_DPAD_NONE,
  "dpad_up": vg.DS4_DPAD_DIRECTIONS.DS4_BUTTON_DPAD_NORTH,
  "dpad_upleft": vg.DS4_DPAD_DIRECTIONS.DS4_BUTTON_DPAD_NORTHWEST,
  "dpad_upright": vg.DS4_DPAD_DIRECTIONS.DS4_BUTTON_DPAD_NORTHEAST,
  "dpad_down": vg.DS4_DPAD_DIRECTIONS.DS4_BUTTON_DPAD_SOUTH,
  "dpad_downleft": vg.DS4_DPAD_DIRECTIONS.DS4_BUTTON_DPAD_SOUTHWEST,
  "dpad_downright": vg.DS4_DPAD_DIRECTIONS.DS4_BUTTON_DPAD_SOUTHEAST,
  "dpad_left": vg.DS4_DPAD_DIRECTIONS.DS4_BUTTON_DPAD_WEST,
  "dpad_right": vg.DS4_DPAD_DIRECTIONS.DS4_BUTTON_DPAD_EAST
}

buttons_index = ["a","b","x","y","start","select","home","l1","r1","lthumb","rthumb"]
dpad_index = ["dpad_none","dpad_up","dpad_upleft","dpad_upright","dpad_down","dpad_downleft","dpad_downright","dpad_left","dpad_right"]
triggers_index = ["l2", "r2"]
joystick_index = ["lstick", "rstick"]

dpad_directions = {
  "dpad_up": 0,
  "dpad_down": 1,
  "dpad_left": 2,
  "dpad_right": 3
}

class Gamepad:
  def __init__(self, type="xbox"):
    if type == "xbox":
      self.controller = "xbox"
      self.gamepad = vg.VX360Gamepad()
      self.button_map = xbox_buttons
    elif type == "ps4":
      self.controller = "ps4"
      self.gamepad = vg.VDS4Gamepad()
      self.button_map = ps4_buttons
      self.dpad_status = [False,False,False,False] # up, down, left, right
    else:
      raise ValueError("invalid gamepad type, must be 'xbox' or 'ps4'")

  def input(self,input,value):
    if not self.gamepad:
      print("gamepad not initialized")
      return

    if self.controller == "ps4" and input == "home":
      if value == 1:
        self.gamepad.press_special_button(self.button_map[input])
      elif value == 0:
        self.gamepad.release_special_button(self.button_map[input])
    elif self.controller == "ps4" and input in dpad_index:
      if value == 1:
        self.dpad_status[dpad_directions[input]] = True
      elif value == 0:
        self.dpad_status[dpad_directions[input]] = False
      if all(not status for status in self.dpad_status):
        self.gamepad.directional_pad(direction=ps4_buttons["dpad_none"])
      elif self.dpad_status[0]:
        if self.dpad_status[2]:
          self.gamepad.directional_pad(direction=ps4_buttons["dpad_upleft"])
        elif self.dpad_status[3]:
          self.gamepad.directional_pad(direction=ps4_buttons["dpad_upright"])
        else:
          self.gamepad.directional_pad(direction=ps4_buttons["dpad_up"])
      elif self.dpad_status[1]:
        if self.dpad_status[2]:
          self.gamepad.directional_pad(direction=ps4_buttons["dpad_downleft"])
        elif self.dpad_status[3]:
          self.gamepad.directional_pad(direction=ps4_buttons["dpad_downright"])
        else:
          self.gamepad.directional_pad(direction=ps4_buttons["dpad_down"])
      elif self.dpad_status[2]:
        self.gamepad.directional_pad(direction=ps4_buttons["dpad_left"])
      elif self.dpad_status[3]:
        self.gamepad.directional_pad(direction=ps4_buttons["dpad_right"])
    elif input in buttons_index or input in dpad_index:
      if value == 1:
        self.gamepad.press_button(self.button_map[input])
      elif value == 0:
        self.gamepad.release_button(self.button_map[input])
    elif input == triggers_index[0]:
      self.gamepad.left_trigger_float(value)
    elif input == triggers_index[1]:
      self.gamepad.right_trigger_float(value)
    elif input == joystick_index[0]:
      if sys.platform == "win32":
        self.gamepad.left_joystick_float(value[0], value[1])
      else:
        self.gamepad.left_joystick_float(value[0], -value[1])
    elif input == joystick_index[1]:
      if sys.platform == "win32":
        self.gamepad.right_joystick_float(value[0], value[1])
      else:
        self.gamepad.right_joystick_float(value[0], -value[1])
    self.gamepad.update()