slide-whistle-trombone-champ-controller/code.py at main · hannahilea/slide-whistle-trombone-champ-controller · 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
#!/usr/bin/env python3
"""
BLE HID keyboard controller for Trombone Champ game.
Attach one button with a pullup resistor and one potentiometer to Feather nRF52840
"""
import board
import math
from digitalio import DigitalInOut, Direction
from analogio import AnalogIn

import adafruit_ble
from adafruit_ble.advertising import Advertisement
from adafruit_ble.advertising.standard import ProvideServicesAdvertisement
from adafruit_ble.services.standard.hid import HIDService
from adafruit_ble.services.standard.device_info import DeviceInfoService
from adafruit_hid.mouse import Mouse

# Global constants
MAX_SLIDER_Y_VALUE = 300  # roughly corresponds to pixel height of screen
MOUSE_Y_CHANGE_TOLERANCE = 12    # "pixel" change threshold; above this tolerance triggers mouse movement
NUM_SMOOTHED_SLIDER_INPUTS = 60  # Buffer for running average of slider values

# Initialize the onboard controls
clicky_button = DigitalInOut(board.A2)
clicky_button.direction = Direction.INPUT
slide_potentiometer = AnalogIn(board.A1)
onboard_button = DigitalInOut(board.SWITCH)
onboard_button.direction = Direction.INPUT

# Set up bluetooth connection
hid = HIDService()
device_info = DeviceInfoService(
    software_revision=adafruit_ble.__version__, manufacturer="Adafruit Industries"
)
advertisement = ProvideServicesAdvertisement(hid)
advertisement.appearance = 961
scan_response = Advertisement()
scan_response.complete_name = "CircuitPython HID"

ble = adafruit_ble.BLERadio()
ble.name = "Hannex Trombone Champ"
if not ble.connected:
    print("advertising")
    ble.start_advertising(advertisement, scan_response)
else:
    print("already connected")
    print(ble.connections)

# Set up mouse
mouse = Mouse(hid.devices)

# Helper functions for slider mapping
def get_voltage(pin):
    return (pin.value * 3.3) / 65536

def map_pot_percent(voltage, v_min, v_max):
    f = (voltage - v_min) / (v_max - v_min)
    return math.trunc(f * 100)

def main_event_loop():
    while True:
        while not ble.connected:
            pass
        print("Bluetooth connected!")

        is_clicky_button_pushed = False
        is_onboard_button_pushed = False
        send_commands_enabled = True

        slider_buffer = [0 for _ in range(NUM_SMOOTHED_SLIDER_INPUTS)]
        mouse_y_previous = 0.0
        while ble.connected:

            # Update whether mouse commands should be sent
            _is_onboard_button_pushed = not onboard_button.value
            if is_onboard_button_pushed != _is_onboard_button_pushed:
                is_onboard_button_pushed = _is_onboard_button_pushed
                if is_onboard_button_pushed:
                    print("On-board button pressed!")

                    # Update sending state
                    send_commands_enabled = not send_commands_enabled

                    # If sending is newly disabled, release the clicky button!
                    if not send_commands_enabled and is_clicky_button_pushed:
                        is_clicky_button_pushed = False
                        mouse.release(Mouse.LEFT_BUTTON)

                else:
                    print("On-board button released")
                    pass

            if not send_commands_enabled:
                continue

            # Store current raw slider position from voltage
            slider_voltage = get_voltage(slide_potentiometer)
            slider_value_raw = MAX_SLIDER_Y_VALUE * map_pot_percent(slider_voltage, 1.62, 2.13) / 100
            slider_buffer.append(slider_value_raw)
            slider_buffer.pop(0)

            slider_buffer_avg = sum(slider_buffer) / NUM_SMOOTHED_SLIDER_INPUTS
            mouse_y_diff = math.trunc(mouse_y_previous - slider_buffer_avg)

            # Check/update button state
            _is_clicky_button_pushed = clicky_button.value
            if is_clicky_button_pushed != _is_clicky_button_pushed:
                is_clicky_button_pushed = _is_clicky_button_pushed
                # print("Value changed!")  # debug
                if is_clicky_button_pushed:
                    mouse.press(Mouse.LEFT_BUTTON)
                else:
                    mouse.release(Mouse.LEFT_BUTTON)

            # Update mouse position
            elif abs(mouse_y_diff) >= MOUSE_Y_CHANGE_TOLERANCE:
                # print("UPDATING: ", mouse_y_previous, slider_buffer_avg, mouse_y_diff)
                mouse.move(y=mouse_y_diff)
                mouse_y_previous = slider_buffer_avg

        ble.start_advertising(advertisement)


if __name__ == "__main__":
    main_event_loop()