aboutsummaryrefslogtreecommitdiffstats
path: root/hex33board/testing.py
blob: 86e08649e1e592c1c55ad684c9a6e42d131a6a4c (plain)
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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
from __future__ import annotations

import displayio
from adafruit_ticks import ticks_ms
from adafruit_midi.note_on import NoteOn
from adafruit_midi.note_off import NoteOff
from adafruit_display_text import label
from .util import FONT_10, led_map
from .i2c import i2c_encode

led_map = led_map[:]
led_map.extend([1, 0, None, None, 2, 3])

try:
    import audiomixer
    import synthio
except:
    synthio = None


class Keyboard:
    matrix: Matrix | None
    pixels: NeoPixel | None
    display: Display | None

    midi_usb: MIDI | None
    midi_trs: MIDI | None

    audio_out: PWMAudioOut | None

    def __init__(self, board):
        try:
            self.matrix = board.create_matrix(self)
        except Exception as e:
            print("Error bringing up matrix:", e)
            self.matrix = None

        try:
            self.pixels = board.create_pixels(self, auto_write=False)
        except Exception as e:
            print("Error bringing up pixels:", e)
            self.pixels = None

        try:
            displayio.release_displays()
            self.display = board.create_display(
                self, width=128, height=32, auto_refresh=False
            )

            self.group = displayio.Group()
            self.label = label.Label(
                FONT_10,
                text="Hello World",
                color=0xFFFFFF,
                anchor_point=(0, 0),
                anchored_position=(0, 0),
            )
            self.group.append(self.label)
            self.display.root_group = self.group
            self.display.refresh()
        except Exception as e:
            print("Error bringing up display:", e)
            self.display = None

        try:
            self.midi_usb = board.create_midi_usb(self)
        except Exception as e:
            print("Error bringing up midi_usb:", e)
            self.midi_usb = None

        try:
            self.midi_trs = board.create_midi_trs(self)
            self.midi_trs.out_channel = 0
        except Exception as e:
            print("Error bringing up midi_trs:", e)
            self.midi_trs = None

        try:
            self.i2c = board.create_i2c(board, frequency=1000000, timeout=1000)
            self.i2c.try_lock()
        except Exception as e:
            print("Error bringing up i2c:", e)
            self.i2c = None

        try:
            self.audio_out = board.create_audio_out(self)
            self.synth = synthio.Synthesizer(sample_rate=41100)
            self.mixer = audiomixer.Mixer(
                sample_rate=41100, buffer_size=5120, channel_count=1
            )
            self.filter = self.synth.low_pass_filter(2000, 0.8)
            self.audio_out.play(self.mixer)
            self.mixer.voice[0].play(self.synth)
            self.mixer.voice[0].level = (30 * 30) / 10000
        except Exception as e:
            print("Error bringing up synth:", e)
            self.synth = None

        self.cycle = 0
        self.pressed = {}
        for i, l in enumerate(led_map):
            self.pressed[i] = 0

        self.notes = {}

    def broadcast(self, msg: MIDIMessage):
        if self.midi_usb:
            self.midi_usb.send(msg)
        if self.midi_trs:
            self.midi_trs.send(msg)

    def tick(self):
        self.cycle = (self.cycle + 1) % 1000

        if self.cycle < 300:
            base = 0x660000
        elif self.cycle < 600:
            base = 0x006600
        elif self.cycle < 900:
            base = 0x000066
        else:
            base = 0x666666

        for i, pressed in self.matrix.scan_for_changes():
            pitch = 36 + i
            if pressed:
                self.broadcast(NoteOn(pitch, 127))
                self.pressed[i] |= 0xFF0000
            else:
                self.broadcast(NoteOff(pitch, 127))
                self.pressed[i] &= ~0xFF0000

            if self.i2c:
                msg = i2c_encode(32 + i, pressed)
                try:
                    self.i2c.writeto(0x33, bytes([msg]))
                except OSError:
                    pass

            if self.synth:
                if pressed:
                    note = synthio.Note(synthio.midi_to_hz(pitch), filter=self.filter)
                    self.notes[i] = note
                    self.synth.press(note)
                elif note := self.notes.pop(i, None):
                    self.synth.release(note)

        if self.pixels:
            for i, l in enumerate(led_map):
                if l is None:
                    continue
                self.pixels[l] = self.pressed[i] | base
            self.pixels.show()

        if self.display:
            self.display.refresh()

    def run(self):
        while True:
            self.tick()