2022-05-15 23:39:29 +02:00
""" 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
2023-02-10 23:39:13 +01:00
matrix = [ [ ' NO_LED ' ] * cols for _ in range ( rows ) ]
2022-05-15 23:39:29 +02:00
pos = [ ]
flags = [ ]
2023-02-10 23:39:13 +01:00
led_layout = info_data [ config_type ] [ ' layout ' ]
for index , led_data in enumerate ( led_layout ) :
if ' matrix ' in led_data :
row , col = led_data [ ' matrix ' ]
matrix [ row ] [ col ] = str ( index )
pos . append ( f ' {{ { led_data . get ( " x " , 0 ) } , { led_data . get ( " y " , 0 ) } }} ' )
flags . append ( str ( led_data . get ( ' flags ' , 0 ) ) )
2022-05-15 23:39:29 +02:00
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 :
2023-02-10 23:39:13 +01:00
lines . append ( f ' {{ { " , " . join ( line ) } }} , ' )
2022-05-15 23:39:29 +02:00
lines . append ( ' }, ' )
2023-02-10 23:39:13 +01:00
lines . append ( f ' {{ { " , " . join ( pos ) } }} , ' )
lines . append ( f ' {{ { " , " . join ( flags ) } }} , ' )
2022-05-15 23:39:29 +02:00
lines . append ( ' }; ' )
lines . append ( ' #endif ' )
return lines
2023-10-29 02:09:02 +02:00
def _gen_matrix_mask ( info_data ) :
""" Convert info.json content to matrix_mask
"""
cols = info_data [ ' matrix_size ' ] [ ' cols ' ]
rows = info_data [ ' matrix_size ' ] [ ' rows ' ]
# Default mask to everything enabled
mask = [ [ ' 1 ' ] * cols for i in range ( rows ) ]
# Automatically mask out dip_switch.matrix_grid locations
matrix_grid = info_data . get ( ' dip_switch ' , { } ) . get ( ' matrix_grid ' , [ ] )
for row , col in matrix_grid :
mask [ row ] [ col ] = ' 0 '
lines = [ ]
lines . append ( ' #ifdef MATRIX_MASKED ' )
lines . append ( ' __attribute__((weak)) const matrix_row_t matrix_mask[] = { ' )
for i in range ( rows ) :
lines . append ( f ' 0b { " " . join ( reversed ( mask [ i ] ) ) } , ' )
lines . append ( ' }; ' )
lines . append ( ' #endif ' )
return lines
2022-05-15 23:39:29 +02:00
@cli.argument ( ' -o ' , ' --output ' , arg_only = True , type = normpath , help = ' File to write to ' )
@cli.argument ( ' -q ' , ' --quiet ' , arg_only = True , action = ' store_true ' , help = " Quiet mode, only output error messages " )
@cli.argument ( ' -kb ' , ' --keyboard ' , arg_only = True , type = keyboard_folder , completer = keyboard_completer , required = True , help = ' Keyboard to generate keyboard.c for. ' )
@cli.subcommand ( ' 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 ) )
2023-10-29 02:09:02 +02:00
keyboard_h_lines . extend ( _gen_matrix_mask ( kb_info_json ) )
2022-05-15 23:39:29 +02:00
# Show the results
dump_lines ( cli . args . output , keyboard_h_lines , cli . args . quiet )