board_pins: Add a new "extras" module to support per-board pin aliases

Signed-off-by: Kevin O'Connor <kevin@koconnor.net>

1 files changed, 24 insertions, 0 deletions

diff --git a/klippy/extras/board_pins.py b/klippy/extras/board_pins.py
new file mode 100644
index 00000000..d8084b02
--- /dev/null
+++ b/klippy/extras/board_pins.py
@@ -0,0 +1,24 @@
+# Support for custom board pin aliases
+#
+# Copyright (C) 2019  Kevin O'Connor <kevin@koconnor.net>
+#
+# This file may be distributed under the terms of the GNU GPLv3 license.
+
+class PrinterBoardAliases:
+    def __init__(self, config, chip_name):
+        ppins = config.get_printer().lookup_object('pins')
+        pin_resolver = ppins.get_pin_resolver(chip_name)
+        aliases = config.get("aliases")
+        parts = [a.split('=', 1) for a in aliases.split(',')]
+        for name, value in parts:
+            name, value = name.strip(), value.strip()
+            if value.startswith('<') and value.endswith('>'):
+                pin_resolver.reserve_pin(name, value)
+            else:
+                pin_resolver.alias_pin(name, value)
+
+def load_config(config):
+    return PrinterBoardAliases(config, "mcu")
+
+def load_config_prefix(config):
+    return PrinterBoardAliases(config, config.get_name().split()[1])