Adding mx2conf script

1 files changed, 174 insertions, 0 deletions

diff --git a/tools/mx2conf.py b/tools/mx2conf.py
new file mode 100755
index 0000000..59247d5
--- /dev/null
+++ b/tools/mx2conf.py
@@ -0,0 +1,174 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+__author__ = 'Fabien Poussin'
+__version__ = '0.1'
+
+from os.path import expanduser, sep, dirname, abspath
+from argparse import ArgumentParser
+import re
+
+parser = ArgumentParser(description='Generate ChibiOS GPIO header file from STM32CubeMX project files.')
+parser.add_argument('--mx', required=True, type=str, help='STM32CubeMX project file path')
+parser.add_argument('--mcu', default='mcuconf.h', type=str, help='mcuconf.h path')
+parser.add_argument('--hal', default='halconf.h', type=str, help='halconf.h path')
+
+# Always enable
+ALWAYS = ('PAL', 'EXTI')
+
+# In case IPs don't match hal names, or if you want to override (ie: use SERIAL instead of UART driver)
+HAL_TRANSLATE = (
+    ('USB', 'USB_OTG_FS'), 
+    ('USB', 'USB_OTG_HS'), 
+    ('SDC', 'SDMMC'), 
+    ('TRNG', 'RNG'), 
+    ('WSPI', 'QUADSPI'), 
+    ('WDG', 'IWDG'),
+    ('UART', 'USART')
+)
+
+DRIVER_TRANSLATE = (
+    ('SDC', 'SDMMC'),
+    ('SERIAL', r'U(S)?ART'),
+    ('UART', r'U(S)?ART'),
+    ('USB', 'OTG'),
+    ('PWM', 'TIM'),
+    ('ICU', 'TIM'),
+    ('GPT', 'TIM'),
+    ('WDG', 'IWDG'),
+    ('WSPI', 'QUADSPI')
+)
+
+RCC_TRANSLATE = (
+    ('HPRE', 'HCLK'),
+    ('PPRE1', 'APB1CLKDivider'),
+    ('PPRE2', 'APB2CLKDivider'),
+    ('SW', 'SYSCLKSource'),
+    ('SDMMC1SEL', 'SDMMCClockSelection'),
+    ('PLLM_VALUE', 'PLLM'),
+    ('PLLN_VALUE', 'PLLN'),
+    ('PLLQ_VALUE', 'PLLQ'),
+    ('PLLSAIN_VALUE', 'PLLSAIN'),
+    ('PLLSAIR_VALUE', 'PLLSAIR'),
+    ('MCO1SEL', 'RCC_MCO1Source'),
+    ('MCO2SEL', 'RCC_MCO2Source'),
+    ('PLLI2SN_VALUE', 'PLLI2SN'),
+)
+
+def translate_hal(ip):
+    for h in HAL_TRANSLATE:
+        if re.search(h[1], ip, re.M):
+            return h[0]
+    return ip
+
+def translate_driver(ip):
+    for d in DRIVER_TRANSLATE:
+        if re.search(d[0], ip, re.M):
+            return d[1]
+    return ip
+
+def set_boolean_define(line, match, name, value):
+    if name in line and re.search(match, line, re.M):
+        if value == True:
+            line = line.replace('FALSE', 'TRUE')
+        else:
+            line = line.replace('TRUE', 'FALSE')
+        print(line.strip())
+    return line
+
+def get_hal_devices(source):
+    out = []
+    for line in source:
+        if '#define HAL_USE_' in line:
+            l = line.split(' ')
+            dev = ('_').join(l[1].split('_')[2:])
+            if dev not in out:
+                out.append(dev)
+    return out
+
+def get_enabled_drivers(source, hal_devices):
+    out = {}
+    for line in source:
+        if line.startswith('Mcu.IP'):
+            ip_only = re.search(r"^Mcu.IP\d+=((I2C|[A-Z]+_?)+)(\d)?", line) # Extract ip name separated from periph number
+            if ip_only:
+                dev = translate_hal(ip_only.group(1)) # periph name
+                dev_num = ip_only.group(3) # periph number
+                if dev in hal_devices:
+                    if dev not in out.keys():
+                        out[dev] = []
+                    if dev_num:
+                        out[dev].append(dev_num)
+    return out
+
+def get_rcc(source):
+    out = {}
+
+    return out
+
+def update_hal(source, drivers):
+    match = '#define HAL_USE_'
+    for i in range(len(source)):
+        line = source[i]
+        if line.startswith(match):
+            if "TRUE" in line:
+                source[i] = line.replace('TRUE', 'FALSE')
+            
+        for d in drivers:
+            source[i] = set_boolean_define(source[i], match, d, True)
+
+    return source
+
+def update_drivers(source, drivers):
+    for i in range(len(source)):
+        line = source[i]
+        if '_USE_' in line:
+            if 'TRUE' in line:
+                source[i] = line.replace('TRUE', 'FALSE')
+
+        for driver, instances in drivers.items():
+            if instances:
+                for inst in instances:
+                    periph = translate_driver(driver)
+                    match = 'STM32_{0}_USE_{1}{2}'.format(driver, periph, inst)
+                    source[i] = set_boolean_define(source[i], match, driver, True)
+            else:
+                periph = translate_driver(driver)
+                match = 'STM32_{0}_USE_{1}'.format(driver, periph)
+                source[i] = set_boolean_define(source[i], match, driver, True)
+
+    return source
+
+def update_rcc(source, rcc):
+    # TODO
+    return source
+
+if __name__ == '__main__':
+    args = parser.parse_args()
+    cur_path = dirname(abspath(__file__))
+
+    with open(args.mx, 'r') as f:
+        project = f.readlines()
+
+    with open(args.hal, 'r') as f:
+        halconf = f.readlines()
+
+    with open(args.mcu, 'r') as f:
+        mcuconf = f.readlines()
+
+    hal_devices = get_hal_devices(halconf)
+    enabled_drivers = get_enabled_drivers(project, hal_devices)
+    rcc = get_rcc(project)
+
+    for a in ALWAYS:
+        enabled_drivers[a] = []
+
+# Update and save halconf
+    halconf = update_hal(halconf, enabled_drivers)
+    with open(args.hal, 'w') as f:
+        f.write("".join(halconf))
+
+# Update and save mcuconf drivers
+    mcuconf = update_drivers(mcuconf, enabled_drivers)
+    mcuconf = update_rcc(mcuconf, rcc)
+    with open(args.mcu, 'w') as f:
+        f.write("".join(mcuconf))