def _parse_led_config(file, matrix_cols, matrix_rows): """Return any 'raw' led/rgb matrix config """ file_contents = file.read_text(encoding='utf-8') file_contents = comment_remover(file_contents) file_contents = file_contents.replace('\\\n', '') matrix_raw = [] position_raw = [] flags = [] found_led_config = False bracket_count = 0 section = 0 for _type, value in lex(file_contents, CLexer()): # Assume g_led_config..stuff..; if value == 'g_led_config': found_led_config = True elif value == ';': found_led_config = False elif found_led_config: # Assume bracket count hints to section of config we are within if value == '{': bracket_count += 1 if bracket_count == 2: section += 1 elif value == '}': bracket_count -= 1 else: # Assume any non whitespace value here is important enough to stash if _type in [ Token.Literal.Number.Integer, Token.Literal.Number.Float, Token.Literal.Number.Hex, Token.Name ]: if section == 1 and bracket_count == 3: matrix_raw.append(_coerce_led_token(_type, value)) if section == 2 and bracket_count == 3: position_raw.append(_coerce_led_token(_type, value)) if section == 3 and bracket_count == 2: flags.append(_coerce_led_token(_type, value)) # Slightly better intrim format matrix = list(_get_chunks(matrix_raw, matrix_cols)) position = list(_get_chunks(position_raw, 2)) matrix_indexes = list(filter(lambda x: x is not None, matrix_raw)) # If we have not found anything - bail if not section: return None # TODO: Improve crude parsing/validation if len(matrix) != matrix_rows and len(matrix) != (matrix_rows / 2): raise ValueError("Unable to parse g_led_config matrix data") if len(position) != len(flags): raise ValueError("Unable to parse g_led_config position data") if len(matrix_indexes) and (max(matrix_indexes) >= len(flags)): raise ValueError("OOB within g_led_config matrix data") return (matrix, position, flags)
def find_layouts(file): """Returns list of parsed LAYOUT preprocessor macros found in the supplied include file. """ file = Path(file) aliases = {} # Populated with all `#define`s that aren't functions parsed_layouts = {} # Search the file for LAYOUT macros and aliases file_contents = file.read_text() file_contents = comment_remover(file_contents) file_contents = file_contents.replace('\\\n', '') for line in file_contents.split('\n'): if line.startswith('#define') and '(' in line and 'LAYOUT' in line: # We've found a LAYOUT macro macro_name, layout, matrix = _parse_layout_macro(line.strip()) # Reject bad macro names if macro_name.startswith( 'LAYOUT_kc') or not macro_name.startswith('LAYOUT'): continue # Parse the matrix data matrix_locations = _parse_matrix_locations(matrix, file, macro_name) # Parse the layout entries into a basic structure default_key_entry[ 'x'] = -1 # Set to -1 so _default_key(key) will increment it to 0 layout = layout.strip() parsed_layout = [_default_key(key) for key in layout.split(',')] for key in parsed_layout: if key['label'] in matrix_locations: key['matrix'] = matrix_locations[key['label']] parsed_layouts[macro_name] = { 'key_count': len(parsed_layout), 'layout': parsed_layout, 'filename': str(file), } elif '#define' in line: # Attempt to extract a new layout alias try: _, pp_macro_name, pp_macro_text = line.strip().split(' ', 2) aliases[pp_macro_name] = pp_macro_text except ValueError: continue # Populate our aliases for alias, text in aliases.items(): if text in parsed_layouts and 'KEYMAP' not in alias: parsed_layouts[alias] = parsed_layouts[text] return parsed_layouts
def find_layouts(file): """Returns list of parsed LAYOUT preprocessor macros found in the supplied include file. """ file = Path(file) aliases = {} # Populated with all `#define`s that aren't functions parsed_layouts = {} # Search the file for LAYOUT macros and aliases file_contents = file.read_text(encoding='utf-8') file_contents = comment_remover(file_contents) file_contents = file_contents.replace('\\\n', '') for line in file_contents.split('\n'): if layout_macro_define_regex.match( line.lstrip()) and '(' in line and 'LAYOUT' in line: # We've found a LAYOUT macro macro_name, layout, matrix = _parse_layout_macro(line.strip()) # Reject bad macro names if macro_name.startswith( 'LAYOUT_kc') or not macro_name.startswith('LAYOUT'): continue # Parse the matrix data matrix_locations = _parse_matrix_locations(matrix, file, macro_name) # Parse the layout entries into a basic structure default_key_entry[ 'x'] = -1 # Set to -1 so _default_key(key) will increment it to 0 layout = layout.strip() parsed_layout = [_default_key(key) for key in layout.split(',')] for i, key in enumerate(parsed_layout): if 'label' not in key: cli.log.error( 'Invalid LAYOUT macro in %s: Empty parameter name in macro %s at pos %s.', file, macro_name, i) elif key['label'] in matrix_locations: key['matrix'] = matrix_locations[key['label']] parsed_layouts[macro_name] = { 'layout': parsed_layout, 'filename': str(file), } elif '#define' in line: # Attempt to extract a new layout alias try: _, pp_macro_name, pp_macro_text = line.strip().split(' ', 2) aliases[pp_macro_name] = pp_macro_text except ValueError: continue return parsed_layouts, aliases
def _preprocess_c_file(file): """Load file and strip comments """ file_contents = file.read_text(encoding='utf-8') file_contents = comment_remover(file_contents) return file_contents.replace('\\\n', '')