wip

script/device_gen.py

@@ -1,7 +1,6 @@
 import glob
 import hashlib
-import yaml
-import os
+import re
 
 import modm_devices
 
@@ -106,6 +105,66 @@ def __getitem__(self, item) -> modm_devices.device.Device:
         return value
 
 
+def map_signal(s):
+    if s is None:
+        return None
+
+    if s["driver"] == "tim" and s["name"].startswith("ch"):
+        return {
+            "func": "timer",
+            "af": int(s.get("af", "-1")),
+            "instance": int(s["instance"]),
+            "name": s["name"],
+        }
+    elif s["driver"] == "spi" and (
+        s["name"] == "sck"
+        or s["name"] == "mosi"
+        or s["name"] == "miso"
+        or s["name"] == "tx"
+        or s["name"] == "rx"
+    ):
+        return {
+            "func": "spi",
+            "af": int(s.get("af", "-1")),
+            "instance": int(s["instance"]),
+            "name": {"tx": "mosi", "rx": "miso"}.get(s["name"], s["name"]),
+        }
+    elif (s["driver"] == "uart" or s["driver"] == "usart") and (
+        s["name"] == "rx" or s["name"] == "tx"
+    ):
+        return {
+            "func": "serial",
+            "af": int(s.get("af", "-1")),
+            "instance": int(s["instance"]),
+            "name": s["name"],
+        }
+    elif s["driver"] == "adc" and re.match(r"in\d+", s.get("name", "in0")):
+        return {
+            "func": "adc",
+            "af": -1,
+            "instance": int(s["instance"]),
+            "name": s.get("name", "in0xff")[2:],
+        }
+    else:
+        return None
+
+
+def map_tag(f):
+    if f is None:
+        return None
+
+    if f["func"] == "timer":
+        return f"TIMER_TAG(TIMER{f['instance']}, TIMER_{f['name']})".upper()
+    elif f["func"] == "spi":
+        return f"SPI_TAG(SPI_PORT{f['instance']}, RES_SPI_{f['name']})".upper()
+    elif f["func"] == "serial":
+        return f"SERIAL_TAG(SERIAL_PORT{f['instance']}, RES_SERIAL_{f['name']})".upper()
+    elif f["func"] == "adc":
+        return f"ADC_TAG(ADC_DEVICE{f['instance']}, {f['name']})".upper()
+    else:
+        return None
+
+
 devices = [
     "stm32f405rg",
     "stm32f411re",
@@ -123,100 +182,58 @@ def __getitem__(self, item) -> modm_devices.device.Device:
     pins = {}
     key = next((x for x in caches.keys() if x.startswith(device)), None)
 
-    for driver in caches[key].get_all_drivers("gpio"):
-        for pin in driver["gpio"]:
-            funcs = []
+    with open(f"src/system/{device[:9]}/gpio_pins.in", "w") as file:
+        for driver in caches[key].get_all_drivers("gpio"):
+            for pin in driver["gpio"]:
+                file.write(f"GPIO_PIN({pin['port']}, {pin['pin']})\n".upper())
+                if "signal" not in pin:
+                    continue
 
-            if "signal" in pin:
                 for s in pin["signal"]:
-                    if s["driver"] == "tim" and s["name"].startswith("ch"):
-                        funcs.append(
-                            {
-                                "func": "timer",
-                                "af": int(s["af"]),
-                                "instance": int(s["instance"]),
-                                "name": s["name"],
-                            }
-                        )
-
-                    if s["driver"] == "spi" and (
-                        s["name"] == "sck" or s["name"] == "mosi" or s["name"] == "miso"
-                    ):
-                        funcs.append(
-                            {
-                                "func": "spi",
-                                "af": int(s["af"]),
-                                "instance": int(s["instance"]),
-                                "name": s["name"],
-                            }
-                        )
-
-                    if (s["driver"] == "uart" or s["driver"] == "usart") and (
-                        s["name"] == "rx" or s["name"] == "tx"
-                    ):
-                        funcs.append(
-                            {
-                                "func": "serial",
-                                "af": int(s["af"]),
-                                "instance": int(s["instance"]),
-                                "name": s["name"],
-                            }
-                        )
-
-                    if s["driver"] == "adc" and not (
-                        s["name"].startswith("inp") or s["name"].startswith("inn")
-                    ):
-                        funcs.append(
-                            {
-                                "func": "adc",
-                                "af": -1,
-                                "instance": int(s["instance"]),
-                                "name": s["name"][2:],
-                            }
-                        )
-
-            pins[f"P{pin['port']}{pin['pin']}".upper()] = funcs
-
-    with open(f"src/system/{device[:9]}/gpio_pins.yaml", "w") as file:
-        documents = yaml.dump(pins, file, sort_keys=False)
-
-for filename in glob.glob("src/system/*/gpio_pins.yaml"):
-    dir = os.path.dirname(filename)
-
-    with open(filename, "r") as f:
-        pins = yaml.load(f, Loader=yaml.Loader)
-
-    with open(f"{dir}/gpio_pins.in", "w") as file:
-        for k, funcs in pins.items():
-            port = k[1]
-            pin = k[2:]
-            file.write(f"GPIO_PIN({port}, {pin})\n")
-
-            for f in funcs:
-                line = f"GPIO_AF(PIN_{port}{pin}, {f['af']}, "
-
-                if f["func"] == "timer":
-                    line = (
-                        line
-                        + f"TIMER_TAG(TIMER{f['instance']}, TIMER_{f['name'].upper()}))\n"
-                    )
-
-                if f["func"] == "spi":
-                    line = (
-                        line
-                        + f"SPI_TAG(SPI_PORT{f['instance']}, RES_SPI_{f['name'].upper()}))\n"
-                    )
-
-                if f["func"] == "serial":
-                    line = (
-                        line
-                        + f"SERIAL_TAG(SERIAL_PORT{f['instance']}, RES_SERIAL_{f['name'].upper()}))\n"
-                    )
-
-                if f["func"] == "adc":
-                    line = (
-                        line
-                        + f"ADC_TAG(ADC_DEVICE{f['instance']}, {f['name'].upper()}))\n"
+                    f = map_signal(s)
+                    s = map_tag(f)
+                    if s is None:
+                        continue
+                    file.write(
+                        f"GPIO_AF(PIN_{pin['port']}{pin['pin']}, {f['af']}, {s})\n".upper()
                     )
 
-                file.write(line)
+    with open(f"src/system/{device[:9]}/dma.in", "w") as file:
+        for driver in caches[key].get_all_drivers("dma"):
+            if "streams" in driver:
+                for dma in driver["streams"]:
+                    for stream in dma["stream"]:
+                        for channel in stream["channel"]:
+                            funcs = [
+                                r
+                                for s in channel["signal"]
+                                if (r := map_tag(map_signal(s))) is not None
+                            ]
+                            for func in funcs:
+                                entry = ", ".join(
+                                    [
+                                        f".tag = {func}",
+                                        f".port_index = {dma['instance']}",
+                                        f".stream_index = {stream['position']}",
+                                        f".channel = LL_DMA_CHANNEL_{channel['position']}",
+                                    ]
+                                )
+                                file.write("{ " + entry + " },\n")
+            if "requests" in driver:
+                for dma in driver["requests"]:
+                    for request in dma["request"]:
+                        funcs = [
+                            r
+                            for s in request["signal"]
+                            if (r := map_tag(map_signal(s))) is not None
+                        ]
+                        for func in funcs:
+                            entry = ", ".join(
+                                [
+                                    f".tag = {func}",
+                                    f".port_index = -1",
+                                    f".stream_index = -1",
+                                    f".request = {request['position']}",
+                                ]
+                            )
+                            file.write("{ " + entry + " },\n")

src/config/feature.h

@@ -6,6 +6,7 @@
 #define USE_BLACKBOX
 
 #define USE_MAX7456
+#define USE_RGB_LED
 
 #define USE_MOTOR_DSHOT
 #define USE_MOTOR_PWM

src/core/main.c

@@ -79,18 +79,18 @@ __attribute__((__used__)) int main() {
 
   // setup filters early
   filter_global_init();
-  timer_alloc_init();
   pid_init();
 
   // init some hardware things
   gpio_ports_init();
+  debug_pin_init();
 
   // Turn on LED during boot so that if a delay is used as part of using programming pins for other functions,
   // the FC does not appear inactive while programming times out
   led_init();
   led_on(LEDALL);
 
-  debug_pin_init();
+  usb_init();
   buzzer_init();
   usb_init();
 
@@ -124,7 +124,7 @@ __attribute__((__used__)) int main() {
 
   rx_init();
   vtx_init();
-  rgb_init();
+  rgb_led_init();
 
   blackbox_init();
   imu_init();

src/core/target.h

@@ -186,6 +186,7 @@ typedef struct {
   gpio_pins_t fpv;
   gpio_pins_t vbat;
   gpio_pins_t ibat;
+  gpio_pins_t rgb_led;
 
   target_invert_pin_t sdcard_detect;
   target_invert_pin_t buzzer;
@@ -213,6 +214,7 @@ typedef struct {
   MEMBER(fpv, gpio_pins_t)                                                       \
   MEMBER(vbat, gpio_pins_t)                                                      \
   MEMBER(ibat, gpio_pins_t)                                                      \
+  MEMBER(rgb_led, gpio_pins_t)                                                   \
   MEMBER(sdcard_detect, target_invert_pin_t)                                     \
   MEMBER(buzzer, target_invert_pin_t)                                            \
   ARRAY_MEMBER(motor_pins, MOTOR_PIN_MAX, gpio_pins_t)                           \

src/core/tasks.c

@@ -11,6 +11,7 @@
 #include "io/blackbox.h"
 #include "io/buzzer.h"
 #include "io/led.h"
+#include "io/rgb_led.h"
 #include "io/usb_configurator.h"
 #include "io/vbat.h"
 #include "io/vtx.h"
@@ -22,14 +23,7 @@
 void util_task() {
   // handle led commands
   led_update();
-
-#if (RGB_LED_NUMBER > 0)
-  // RGB led control
-  rgb_led_lvc();
-#ifdef RGB_LED_DMA
-  rgb_dma_start();
-#endif
-#endif
+  rgb_led_update();
 
   buzzer_update();
 }