Пример #1
0
def get_current_controllerstate(current_time):
    inputdataset = np.zeros(4)
    state = X.get_state(0)
    triggers = X.get_trigger_values(state)
    joystick = X.get_thumb_values(state)[0][0]
    inputdataset[0] = current_time
    inputdataset[1] = triggers[0]
    inputdataset[2] = triggers[1]
    inputdataset[3] = joystick
    return inputdataset
Пример #2
0
def watch_controller():
    while True:
        print(X.get_connected())
        inputdataset = np.zeros(4)
        state = X.get_state(0)
        triggers = X.get_trigger_values(state)
        joystick = X.get_thumb_values(state)[0][0]
        inputdataset[0] = triggers[0]
        inputdataset[1] = triggers[1]
        inputdataset[2] = joystick
        print(inputdataset)
Пример #3
0
    def run(self):
        """Main loop to do all the realtime processing."""
        check_monitors_interval = 60
        check_monitors_override = False
        check_gamepad_interval = 60
        check_gamepad_override = False

        mouse_distance = 0.0
        old_mouse_pos = old_monitors = None
        old_monitor_index = monitor_index = None
        connected_gamepads = old_gamepads = [False] * 4
        keys = {item: [0] * 255 for item in ('tick', 'prev', 'count', 'held')}
        gamepads = {item: [(0, 0)] * 4 for item in ('thumb_l', 'thumb_r')}
        gamepads.update({item: [0] * 4 for item in ('trig_l', 'trig_r')})
        gamepads['buttons'] = {
            item: [defaultdict(int)] * 4
            for item in ('tick', 'prev', 'count', 'held')
        }
        ticks = 0
        start = time.time()
        while self.state != ThreadState.Stopped:
            self.process_gui()

            # Receive data from the processing thread
            mouse_speed = 0.0
            while not self.process_receiver.empty():
                event, data = self.process_receiver.get()
                if event == ProcessEvent.Error:
                    return self.stop()
                elif event == ProcessEvent.MouseDistance:
                    mouse_speed = data[0]
                    mouse_distance += mouse_speed
                    self.send_gui_event(ThreadEvent.MouseDistance,
                                        mouse_distance)
            self.send_gui_event(ThreadEvent.MouseSpeed, mouse_speed)

            # Get monitor data
            if not ticks % check_monitors_interval or check_monitors_override:
                check_monitors_override = False
                monitors = get_monitor_locations()
            monitors_changed = monitors != old_monitors
            if monitors_changed:
                self.send_process_event(ProcessCommand.MonitorChanged,
                                        monitors)

            # Handle realtime data
            if self.state == ThreadState.Running:
                self.send_process_event(ProcessCommand.Tick, ticks)

                # Get mouse data
                mouse_pos = cursor_position()
                mouse_moved = mouse_pos != old_mouse_pos

                # Get keyboard/mouse clicks
                # Note this will not see anything faster than 1/60th of a second
                for key, val in enumerate(keys['tick']):
                    previously_pressed = bool(val)
                    currently_pressed = bool(check_key_press(key))

                    # Detect individual key releases
                    if previously_pressed and not currently_pressed:
                        keys['tick'][key] = keys['held'][key] = 0
                        self.send_process_event(ProcessCommand.KeyReleased,
                                                key)

                    # Detect when a key is being held down
                    elif previously_pressed and currently_pressed:
                        keys['held'][key] += 1
                        self.send_process_event(ProcessCommand.KeyHeld, key)

                    # Detect when a new key is pressed
                    elif not previously_pressed and currently_pressed:
                        if ticks - keys['prev'][key] < self.double_click_ticks:
                            keys['count'][key] += 1
                        else:
                            keys['count'][key] = 1
                        keys['tick'][key] = keys['prev'][key] = ticks
                        self.send_process_event(ProcessCommand.KeyPressed, key,
                                                keys['count'][key])

                # Check mouse data against monitors
                if mouse_pos is None:  # Cancel if there is no mouse data
                    refresh_monitor_index = False
                elif monitor_index is None:  # If index hasn't been set yet
                    refresh_monitor_index = True
                elif mouse_moved or monitors_changed:  # If there's been any changes
                    refresh_monitor_index = True
                else:
                    refresh_monitor_index = False
                if refresh_monitor_index:
                    mx, my = mouse_pos
                    for i, (x1, y1, x2, y2) in enumerate(monitors):
                        if x1 <= mouse_pos[0] < x2 and y1 <= mouse_pos[1] < y2:
                            # Keep track of which monitor the mouse is on
                            monitor_index = i

                            # Remap mouse position to be within 0 and 1
                            # This is for the GUI "live" preview only
                            remapped = (
                                (mouse_pos[0] - x1) / (x2 - x1),
                                (mouse_pos[1] - y1) / (y2 - y1),
                            )
                            self.send_gui_event(ThreadEvent.MouseMove,
                                                remapped)
                            break

                    # If this part fails, monitors may have changed since being recorded
                    # Skip this loop and try again, but this time refresh the data
                    else:
                        print(
                            f'Unknown monitor detected (mouse: {mouse_pos}, saved: '
                            f'{monitors}, actual: {get_monitor_locations()}')
                        check_monitors_override = True
                        monitor_index = None
                        continue
                mouse_monitor_changed = monitor_index != old_monitor_index

                if mouse_monitor_changed:
                    vres = monitors[monitor_index][3] - monitors[
                        monitor_index][1]
                    print(f'Mouse moved to {vres}p monitor')
                if mouse_moved:
                    self.send_process_event(ProcessCommand.MouseMove,
                                            mouse_pos)

                # Get gamepad information
                if XInput is not None:
                    # Determine which gamepads are connected
                    if not ticks % check_gamepad_interval or check_gamepad_override:
                        check_gamepad_override = False
                        connected_gamepads = XInput.get_connected()
                    gamepads_changed = connected_gamepads != old_gamepads

                    # Print message saying when a change is detected
                    if gamepads_changed:
                        diff = sum(filter(bool, connected_gamepads))
                        if old_gamepads is not None:
                            diff -= sum(filter(bool, old_gamepads))
                        print('Gamepad ' + ('removed', 'detected')[diff > 0])

                    for gamepad in (
                            i for i, active in enumerate(connected_gamepads)
                            if active):
                        # Get a snapshot of the current gamepad state
                        try:
                            state = XInput.get_state(gamepad)
                        except XInput.XInputNotConnectedError:
                            check_gamepad_override = True
                            continue
                        thumb_l, thumb_r = XInput.get_thumb_values(state)
                        trig_l, trig_r = XInput.get_trigger_values(state)
                        buttons = XInput.get_button_values(state)

                        if thumb_l != gamepads['thumb_l'][gamepad]:
                            self.send_processEvent(
                                ProcessCommand.GamepadThumbL, gamepad, thumb_l)
                        if thumb_r != gamepads['thumb_r'][gamepad]:
                            self.send_processEvent(
                                ProcessCommand.GamepadThumbR, gamepad, thumb_r)
                        if trig_l != gamepads['trig_l'][gamepad]:
                            self.send_processEvent(
                                ProcessCommand.GamepadTriggerL, gamepad,
                                trig_l)
                        if trig_r != gamepads['trig_r'][gamepad]:
                            self.send_processEvent(
                                ProcessCommand.GamepadTriggerR, gamepad,
                                trig_r)

                        gamepads['thumb_l'][gamepad] = thumb_l
                        gamepads['thumb_r'][gamepad] = thumb_r
                        gamepads['trig_l'][gamepad] = trig_l
                        gamepads['trig_r'][gamepad] = trig_r

                        for button, val in buttons.items():
                            previously_pressed = bool(
                                gamepads['buttons']['tick'][gamepad][button])
                            currently_pressed = val

                            # Detect individual button releases
                            if previously_pressed and not currently_pressed:
                                gamepads['buttons']['tick'][gamepad][
                                    button] = gamepads['buttons']['held'][
                                        gamepad][button] = 0
                                self.send_process_event(
                                    ProcessCommand.GamepadButtonReleased,
                                    gamepad, button)

                            # Detect when a button is being held down
                            elif previously_pressed and currently_pressed:
                                gamepads['buttons']['held'][gamepad][
                                    button] += 1
                                self.send_process_event(
                                    ProcessCommand.GamepadButtonHeld, gamepad,
                                    button)

                            # Detect when a new button is pressed
                            elif not previously_pressed and currently_pressed:
                                count = gamepads['buttons']['count'][gamepad]
                                prev = gamepads['buttons']['prev'][gamepad]
                                if ticks - prev[
                                        button] < self.double_click_ticks:
                                    count[button] += 1
                                else:
                                    count[button] = 1
                                gamepads['buttons']['tick'][gamepad][
                                    button] = prev[button] = ticks
                                self.send_process_event(
                                    ProcessCommand.GamepadButtonPressed,
                                    gamepad, button, count[button])

                # Store old values
                old_mouse_pos = mouse_pos
                old_monitors = monitors
                old_monitor_index = monitor_index
                old_gamepads = connected_gamepads

            # Ensure loop is running at the correct UPS
            ticks += 1
            time_expected = ticks / self.ups
            time_diff = time.time() - start
            if time_expected > time_diff:
                time.sleep(time_expected - time_diff)
Пример #4
0
 def get_dump(self):
     state = Xi.get_state(self.__user_index)
     return Xi.get_thumb_values(state)[0][0], Xi.get_trigger_values(
         state)[1]
Пример #5
0
 def right_y(self, deadzone=4000):
     xinput.set_deadzone(xinput.DEADZONE_RIGHT_THUMB, deadzone)
     self.refresh()
     # get_thumb_values(state) -> ((LX, LY), (RX, RY))
     return xinput.get_thumb_values(self.latest_state)[1][1]
Пример #6
0
 def left_x(self, deadzone=4000):
     xinput.set_deadzone(xinput.DEADZONE_LEFT_THUMB, deadzone)
     self.refresh()
     # get_thumb_values(state) -> ((LX, LY), (RX, RY))
     return xinput.get_thumb_values(self.latest_state)[0][0]