diff options
Diffstat (limited to 'lib/python/qmk/cli')
| -rw-r--r-- | lib/python/qmk/cli/__init__.py | 1 | ||||
| -rwxr-xr-x | lib/python/qmk/cli/generate/keyboard_c.py | 75 | 
2 files changed, 76 insertions, 0 deletions
| diff --git a/lib/python/qmk/cli/__init__.py b/lib/python/qmk/cli/__init__.py index 85baa238a8..d7192631a3 100644 --- a/lib/python/qmk/cli/__init__.py +++ b/lib/python/qmk/cli/__init__.py @@ -52,6 +52,7 @@ subcommands = [      'qmk.cli.generate.dfu_header',      'qmk.cli.generate.docs',      'qmk.cli.generate.info_json', +    'qmk.cli.generate.keyboard_c',      'qmk.cli.generate.keyboard_h',      'qmk.cli.generate.layouts',      'qmk.cli.generate.rgb_breathe_table', diff --git a/lib/python/qmk/cli/generate/keyboard_c.py b/lib/python/qmk/cli/generate/keyboard_c.py new file mode 100755 index 0000000000..a9b742f323 --- /dev/null +++ b/lib/python/qmk/cli/generate/keyboard_c.py @@ -0,0 +1,75 @@ +"""Used by the make system to generate keyboard.c from info.json. +""" +from milc import cli + +from qmk.info import info_json +from qmk.commands import dump_lines +from qmk.keyboard import keyboard_completer, keyboard_folder +from qmk.path import normpath +from qmk.constants import GPL2_HEADER_C_LIKE, GENERATED_HEADER_C_LIKE + + +def _gen_led_config(info_data): +    """Convert info.json content to g_led_config +    """ +    cols = info_data['matrix_size']['cols'] +    rows = info_data['matrix_size']['rows'] + +    config_type = None +    if 'layout' in info_data.get('rgb_matrix', {}): +        config_type = 'rgb_matrix' +    elif 'layout' in info_data.get('led_matrix', {}): +        config_type = 'led_matrix' + +    lines = [] +    if not config_type: +        return lines + +    matrix = [['NO_LED'] * cols for i in range(rows)] +    pos = [] +    flags = [] + +    led_config = info_data[config_type]['layout'] +    for index, item in enumerate(led_config, start=0): +        if 'matrix' in item: +            (x, y) = item['matrix'] +            matrix[x][y] = str(index) +        pos.append(f'{{ {item.get("x", 0)},{item.get("y", 0)} }}') +        flags.append(str(item.get('flags', 0))) + +    if config_type == 'rgb_matrix': +        lines.append('#ifdef RGB_MATRIX_ENABLE') +        lines.append('#include "rgb_matrix.h"') +    elif config_type == 'led_matrix': +        lines.append('#ifdef LED_MATRIX_ENABLE') +        lines.append('#include "led_matrix.h"') + +    lines.append('__attribute__ ((weak)) led_config_t g_led_config = {') +    lines.append('  {') +    for line in matrix: +        lines.append(f'    {{ {",".join(line)} }},') +    lines.append('  },') +    lines.append(f'  {{ {",".join(pos)} }},') +    lines.append(f'  {{ {",".join(flags)} }},') +    lines.append('};') +    lines.append('#endif') + +    return lines + + [email protected]('-o', '--output', arg_only=True, type=normpath, help='File to write to') [email protected]('-q', '--quiet', arg_only=True, action='store_true', help="Quiet mode, only output error messages") [email protected]('-kb', '--keyboard', arg_only=True, type=keyboard_folder, completer=keyboard_completer, required=True, help='Keyboard to generate keyboard.c for.') [email protected]('Used by the make system to generate keyboard.c from info.json', hidden=True) +def generate_keyboard_c(cli): +    """Generates the keyboard.h file. +    """ +    kb_info_json = info_json(cli.args.keyboard) + +    # Build the layouts.h file. +    keyboard_h_lines = [GPL2_HEADER_C_LIKE, GENERATED_HEADER_C_LIKE, '#include QMK_KEYBOARD_H', ''] + +    keyboard_h_lines.extend(_gen_led_config(kb_info_json)) + +    # Show the results +    dump_lines(cli.args.output, keyboard_h_lines, cli.args.quiet) | 
