diff --git a/.ampy b/.ampy new file mode 100644 index 0000000..8bce788 --- /dev/null +++ b/.ampy @@ -0,0 +1,7 @@ +# These values work only for Feathers on Linux hosts +# right now. If using Docker, requires execution as such: +# +# docker run -t kmk_firmware:latest --device /dev/ttyUSB0 +AMPY_PORT=/dev/ttyUSB0 +AMPY_BAUD=115200 +AMPY_DELAY=1.5 diff --git a/.gitmodules b/.gitmodules new file mode 100644 index 0000000..3134eff --- /dev/null +++ b/.gitmodules @@ -0,0 +1,3 @@ +[submodule "circuitpython"] + path = vendor/circuitpython + url = https://github.com/adafruit/circuitpython.git diff --git a/Makefile b/Makefile new file mode 100644 index 0000000..a716a4d --- /dev/null +++ b/Makefile @@ -0,0 +1,48 @@ +.PHONY: build-feather devdeps lint + +devdeps: Pipfile.lock + pipenv install --dev --ignore-pipfile + +lint: devdeps + pipenv run flake8 + +build-feather-test: lint devdeps + @echo "===> Pulling dependencies, this may take several minutes" + @git submodule update --init --recursive + @echo "===> Building circuitpython/mpy-cross" + @make -C vendor/circuitpython/mpy-cross + @echo "===> Pulling Nordic BLE stack" + @cd vendor/circuitpython/ports/nrf && ./drivers/bluetooth/download_ble_stack.sh 2>/dev/null >/dev/null + @echo "===> Preparing KMK source for bundling into CircuitPython" + @rm -rf vendor/circuitpython/ports/nrf/freeze/kmk + @rm -rf vendor/circuitpython/ports/nrf/freeze/kmk_keyboard_user.py + @cp -av kmk vendor/circuitpython/ports/nrf/freeze/ + @cp -av boards/klardotsh/twotwo_matrix_feather.py vendor/circuitpython/ports/nrf/freeze/kmk_keyboard_user.py + @echo "===> Building and flashing CircuitPython with KMK and your keymap" + @make -C vendor/circuitpython/ports/nrf BOARD=feather_nrf52832 SERIAL=/dev/ttyUSB0 SD=s132 FROZEN_MPY_DIR=freeze clean dfu-gen dfu-flash + @echo "===> Flashing entrypoint if it doesn't already exist" + @sleep 2 + @-timeout -k 5s 10s pipenv run ampy rm main.py 2>/dev/null + @-timeout -k 5s 10s pipenv run ampy put entrypoints/feather_nrf52832.py main.py + @echo "===> Flashed keyboard successfully!" + +# Fully wipe the board with only stock CircuitPython +burn-it-all-with-fire: lint devdeps + @echo "===> Flashing STOCK CircuitPython with no KMK or user keyboard scripts, and wiping entrypoint" + @echo "===> This is the nuclear option. Ctrl-C to cancel, or any key to continue" + @read + @echo "===> Pulling dependencies, this may take several minutes" + @git submodule update --init --recursive + @echo "===> Building circuitpython/mpy-cross" + @make -C vendor/circuitpython/mpy-cross + @echo "===> Pulling Nordic BLE stack" + @cd vendor/circuitpython/ports/nrf && ./drivers/bluetooth/download_ble_stack.sh 2>/dev/null >/dev/null + @echo "===> Preparing KMK source for bundling into CircuitPython" + @rm -rf vendor/circuitpython/ports/nrf/freeze/kmk + @rm -rf vendor/circuitpython/ports/nrf/freeze/kmk_keyboard_user.py + @echo "===> Building CircuitPython WITHOUT KMK or user keyboard script to induce ImportError" + @make -C vendor/circuitpython/ports/nrf BOARD=feather_nrf52832 SERIAL=/dev/ttyUSB0 SD=s132 FROZEN_MPY_DIR=freeze clean dfu-gen dfu-flash + @echo "===> Wiping keyboard config" + @sleep 2 + @-pipenv run ampy rm main.py 2>/dev/null + @echo "===> Wiped! Probably safe to flash keyboard, try Python serial REPL to verify?" diff --git a/Pipfile b/Pipfile new file mode 100644 index 0000000..dfdfabf --- /dev/null +++ b/Pipfile @@ -0,0 +1,14 @@ +[[source]] +url = "https://pypi.org/simple" +verify_ssl = true +name = "pypi" + +[packages] + +[dev-packages] +adafruit-ampy = "*" +"flake8" = "*" +"flake8-comprehensions" = "*" + +[requires] +python_version = "3.7" diff --git a/Pipfile.lock b/Pipfile.lock new file mode 100644 index 0000000..09ac140 --- /dev/null +++ b/Pipfile.lock @@ -0,0 +1,86 @@ +{ + "_meta": { + "hash": { + "sha256": "e246783475c5feb51445494a3f30c397ad6ddaed35c45decc9846970d001ad50" + }, + "pipfile-spec": 6, + "requires": { + "python_version": "3.7" + }, + "sources": [ + { + "name": "pypi", + "url": "https://pypi.org/simple", + "verify_ssl": true + } + ] + }, + "default": {}, + "develop": { + "adafruit-ampy": { + "hashes": [ + "sha256:1055827874010f48c7dbd3cde4b1d7c6f6732661fad193b188a398e88961fc62" + ], + "index": "pypi", + "version": "==1.0.5" + }, + "click": { + "hashes": [ + "sha256:29f99fc6125fbc931b758dc053b3114e55c77a6e4c6c3a2674a2dc986016381d", + "sha256:f15516df478d5a56180fbf80e68f206010e6d160fc39fa508b65e035fd75130b" + ], + "version": "==6.7" + }, + "flake8": { + "hashes": [ + "sha256:7253265f7abd8b313e3892944044a365e3f4ac3fcdcfb4298f55ee9ddf188ba0", + "sha256:c7841163e2b576d435799169b78703ad6ac1bbb0f199994fc05f700b2a90ea37" + ], + "index": "pypi", + "version": "==3.5.0" + }, + "flake8-comprehensions": { + "hashes": [ + "sha256:b83891fec0e680b07aa1fd92e53eb6993be29a0f3673a09badbe8da307c445e0", + "sha256:e4ccf1627f75f192eb7fde640f5edb81c98d04b1390df9d4145ffd7710bb1ef2" + ], + "index": "pypi", + "version": "==1.4.1" + }, + "mccabe": { + "hashes": [ + "sha256:ab8a6258860da4b6677da4bd2fe5dc2c659cff31b3ee4f7f5d64e79735b80d42", + "sha256:dd8d182285a0fe56bace7f45b5e7d1a6ebcbf524e8f3bd87eb0f125271b8831f" + ], + "version": "==0.6.1" + }, + "pycodestyle": { + "hashes": [ + "sha256:682256a5b318149ca0d2a9185d365d8864a768a28db66a84a2ea946bcc426766", + "sha256:6c4245ade1edfad79c3446fadfc96b0de2759662dc29d07d80a6f27ad1ca6ba9" + ], + "version": "==2.3.1" + }, + "pyflakes": { + "hashes": [ + "sha256:08bd6a50edf8cffa9fa09a463063c425ecaaf10d1eb0335a7e8b1401aef89e6f", + "sha256:8d616a382f243dbf19b54743f280b80198be0bca3a5396f1d2e1fca6223e8805" + ], + "version": "==1.6.0" + }, + "pyserial": { + "hashes": [ + "sha256:6e2d401fdee0eab996cf734e67773a0143b932772ca8b42451440cfed942c627", + "sha256:e0770fadba80c31013896c7e6ef703f72e7834965954a78e71a3049488d4d7d8" + ], + "version": "==3.4" + }, + "python-dotenv": { + "hashes": [ + "sha256:122290a38ece9fe4f162dc7c95cae3357b983505830a154d3c98ef7f6c6cea77", + "sha256:4a205787bc829233de2a823aa328e44fd9996fedb954989a21f1fc67c13d7a77" + ], + "version": "==0.9.1" + } + } +} diff --git a/boards/klardotsh/twotwo_matrix_feather.py b/boards/klardotsh/twotwo_matrix_feather.py new file mode 100644 index 0000000..e8b23eb --- /dev/null +++ b/boards/klardotsh/twotwo_matrix_feather.py @@ -0,0 +1,23 @@ +import board + +from kmk.circuitpython.matrix import MatrixScanner +from kmk.common.consts import DiodeOrientation +from kmk.common.keymap import Keymap + + +def main(): + cols = (board.A4, board.A5) + rows = (board.D27, board.A6) + + matrix = MatrixScanner( + cols=cols, rows=rows, + diode_orientation=DiodeOrientation.COLUMNS, + ) + + keymap = Keymap([ + ['A', 'B'], + ['C', 'D'], + ]) + + while True: + keymap.parse(matrix.raw_scan()) diff --git a/entrypoints/feather_nrf52832.py b/entrypoints/feather_nrf52832.py new file mode 100644 index 0000000..1f52211 --- /dev/null +++ b/entrypoints/feather_nrf52832.py @@ -0,0 +1,5 @@ +from kmk_keyboard_user import main + + +if __name__ == '__main__': + main() diff --git a/kmk/__init__.py b/kmk/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/kmk/circuitpython/__init__.py b/kmk/circuitpython/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/kmk/circuitpython/matrix.py b/kmk/circuitpython/matrix.py new file mode 100644 index 0000000..5203cc2 --- /dev/null +++ b/kmk/circuitpython/matrix.py @@ -0,0 +1,61 @@ +import digitalio + +from kmk.common.consts import DiodeOrientation + + +class MatrixScanner: + def __init__(self, cols, rows, diode_orientation=DiodeOrientation.COLUMNS): + # A pin cannot be both a row and column, detect this by combining the + # two tuples into a set and validating that the length did not drop + # + # repr() hackery is because CircuitPython Pin objects are not hashable + unique_pins = {repr(c) for c in cols} | {repr(r) for r in rows} + if len(unique_pins) != len(cols) + len(rows): + raise ValueError('Cannot use a pin as both a column and row') + + self.cols = [digitalio.DigitalInOut(pin) for pin in cols] + self.rows = [digitalio.DigitalInOut(pin) for pin in rows] + self.diode_orientation = diode_orientation + + if self.diode_orientation == DiodeOrientation.COLUMNS: + self.outputs = self.cols + self.inputs = self.rows + elif self.diode_orientation == DiodeOrientation.ROWS: + self.outputs = self.rows + self.inputs = self.cols + else: + raise ValueError('Invalid DiodeOrientation: {}'.format( + self.diode_orientation, + )) + + for pin in self.outputs: + pin.switch_to_output() + + for pin in self.inputs: + pin.switch_to_input(pull=digitalio.Pull.DOWN) + + def _normalize_matrix(self, matrix): + ''' + We always want to internally look at a keyboard as a list of rows, + where a "row" is a list of keycodes (columns). + + This will convert DiodeOrientation.COLUMNS matrix scans into a + ROWS scan, so we never have to think about these things again. + ''' + if self.diode_orientation == DiodeOrientation.ROWS: + return matrix + + return [ + [col[col_entry] for col in matrix] + for col_entry in range(max(len(col) for col in matrix)) + ] + + def raw_scan(self): + matrix = [] + + for opin in self.outputs: + opin.value = True + matrix.append([ipin.value for ipin in self.inputs]) + opin.value = False + + return self._normalize_matrix(matrix) diff --git a/kmk/circuitpython/util.py b/kmk/circuitpython/util.py new file mode 100644 index 0000000..01d63d5 --- /dev/null +++ b/kmk/circuitpython/util.py @@ -0,0 +1,22 @@ +import board +import digitalio +import time +import sys + + +def feather_signal_error_with_led_flash(rate=0.5): + ''' + Flash the red LED for 10 seconds, alternating every $rate + Could be useful as an uncaught exception handler later on, + but is for now unused + ''' + + rled = digitalio.DigitalInOut(board.LED1) + rled.direction = digitalio.Direction.OUTPUT + + # blink for 5 seconds and exit + for cycle in range(10): + rled.value = cycle % 2 + time.sleep(rate) + + sys.exit(1) diff --git a/kmk/common/__init__.py b/kmk/common/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/kmk/common/consts.py b/kmk/common/consts.py new file mode 100644 index 0000000..e631d2d --- /dev/null +++ b/kmk/common/consts.py @@ -0,0 +1,9 @@ +class DiodeOrientation: + ''' + Orientation of diodes on handwired boards. You can think of: + COLUMNS = vertical + ROWS = horizontal + ''' + + COLUMNS = 0 + ROWS = 1 diff --git a/kmk/common/keymap.py b/kmk/common/keymap.py new file mode 100644 index 0000000..cdcfc9c --- /dev/null +++ b/kmk/common/keymap.py @@ -0,0 +1,18 @@ +class Keymap: + def __init__(self, map): + self.map = map + self.state = [ + [False for _ in row] + for row in self.map + ] + + def parse(self, matrix): + for ridx, row in enumerate(matrix): + for cidx, col in enumerate(row): + if col != self.state[ridx][cidx]: + print('{}: {}'.format( + 'KEYDOWN' if col else 'KEYUP', + self.map[ridx][cidx], + )) + + self.state = matrix diff --git a/kmk/contrib/__init__.py b/kmk/contrib/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/kmk/micropython/__init__.py b/kmk/micropython/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/setup.cfg b/setup.cfg new file mode 100644 index 0000000..1d9a970 --- /dev/null +++ b/setup.cfg @@ -0,0 +1,2 @@ +[flake8] +exclude = .git,__pycache__,vendor diff --git a/vendor/circuitpython b/vendor/circuitpython new file mode 160000 index 0000000..9b98ad7 --- /dev/null +++ b/vendor/circuitpython @@ -0,0 +1 @@ +Subproject commit 9b98ad779468676c3d5f1efdc06b454aaed7c407