You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
711 lines
21 KiB
Python
711 lines
21 KiB
Python
9 years ago
|
#!/usr/bin/env python
|
||
|
# -*- coding: utf-8 -*-
|
||
|
"""Compiler for keymap.c files
|
||
|
|
||
|
This scrip will generate a keymap.c file from a simple
|
||
|
markdown file with a specific layout.
|
||
|
|
||
|
Usage:
|
||
|
python compile_keymap.py INPUT_PATH [OUTPUT_PATH]
|
||
|
"""
|
||
|
from __future__ import division
|
||
|
from __future__ import print_function
|
||
|
from __future__ import absolute_import
|
||
|
from __future__ import unicode_literals
|
||
|
|
||
|
import os
|
||
|
import io
|
||
|
import re
|
||
|
import sys
|
||
|
import json
|
||
|
import unicodedata
|
||
|
import collections
|
||
|
import itertools as it
|
||
|
|
||
|
PY2 = sys.version_info.major == 2
|
||
|
|
||
|
if PY2:
|
||
|
chr = unichr
|
||
|
|
||
|
|
||
|
KEYBOARD_LAYOUTS = {
|
||
|
# These map positions in the parsed layout to
|
||
|
# positions in the KEYMAP MATRIX
|
||
|
'ergodox_ez': [
|
||
|
[ 0, 1, 2, 3, 4, 5, 6], [38, 39, 40, 41, 42, 43, 44],
|
||
|
[ 7, 8, 9, 10, 11, 12, 13], [45, 46, 47, 48, 49, 50, 51],
|
||
|
[14, 15, 16, 17, 18, 19 ], [ 52, 53, 54, 55, 56, 57],
|
||
|
[20, 21, 22, 23, 24, 25, 26], [58, 59, 60, 61, 62, 63, 64],
|
||
|
[27, 28, 29, 30, 31 ], [ 65, 66, 67, 68, 69],
|
||
|
[ 32, 33], [70, 71 ],
|
||
|
[ 34], [72 ],
|
||
|
[ 35, 36, 37], [73, 74, 75 ],
|
||
|
]
|
||
|
}
|
||
|
|
||
|
ROW_INDENTS = {
|
||
|
'ergodox_ez': [0, 0, 0, 0, 0, 1, 0, 0, 0, 2, 5, 0, 6, 0, 4, 0]
|
||
|
}
|
||
|
|
||
|
BLANK_LAYOUTS = [
|
||
|
# Compact Layout
|
||
|
"""
|
||
|
.------------------------------------.------------------------------------.
|
||
|
| | | | | | | | | | | | | | |
|
||
|
!-----+----+----+----+----+----------!-----+----+----+----+----+----+-----!
|
||
|
| | | | | | | | | | | | | | |
|
||
|
!-----+----+----+----x----x----! ! !----x----x----+----+----+-----!
|
||
|
| | | | | | |-----!-----! | | | | | |
|
||
|
!-----+----+----+----x----x----! ! !----x----x----+----+----+-----!
|
||
|
| | | | | | | | | | | | | | |
|
||
|
'-----+----+----+----+----+----------'----------+----+----+----+----+-----'
|
||
|
| | | | | | ! | | | | |
|
||
|
'------------------------' '------------------------'
|
||
|
.-----------. .-----------.
|
||
|
| | | ! | |
|
||
|
.-----+-----+-----! !-----+-----+-----.
|
||
|
! ! | | ! | ! !
|
||
|
! ! !-----! !-----! ! !
|
||
|
| | | | ! | | |
|
||
|
'-----------------' '-----------------'
|
||
|
""",
|
||
|
|
||
|
# Wide Layout
|
||
|
"""
|
||
|
.---------------------------------------------. .---------------------------------------------.
|
||
|
| | | | | | | | ! | | | | | | |
|
||
|
!-------+-----+-----+-----+-----+-------------! !-------+-----+-----+-----+-----+-----+-------!
|
||
|
| | | | | | | | ! | | | | | | |
|
||
|
!-------+-----+-----+-----x-----x-----! ! ! !-----x-----x-----+-----+-----+-------!
|
||
|
| | | | | | |-------! !-------! | | | | | |
|
||
|
!-------+-----+-----+-----x-----x-----! ! ! !-----x-----x-----+-----+-----+-------!
|
||
|
| | | | | | | | ! | | | | | | |
|
||
|
'-------+-----+-----+-----+-----+-------------' '-------------+-----+-----+-----+-----+-------'
|
||
|
| | | | | | ! | | | | |
|
||
|
'------------------------------' '------------------------------'
|
||
|
.---------------. .---------------.
|
||
|
| | | ! | |
|
||
|
.-------+-------+-------! !-------+-------+-------.
|
||
|
! ! | | ! | ! !
|
||
|
! ! !-------! !-------! ! !
|
||
|
| | | | ! | | |
|
||
|
'-----------------------' '-----------------------'
|
||
|
""",
|
||
|
]
|
||
|
|
||
|
|
||
|
DEFAULT_CONFIG = {
|
||
|
"keymaps_includes": [
|
||
|
"keymap_common.h",
|
||
|
],
|
||
|
'filler': "-+.'!:x",
|
||
|
'separator': "|",
|
||
|
'default_key_prefix': ["KC_"],
|
||
|
}
|
||
|
|
||
|
|
||
|
SECTIONS = [
|
||
|
'layout_config',
|
||
|
'layers',
|
||
|
]
|
||
|
|
||
|
|
||
|
# Markdown Parsing
|
||
|
|
||
|
ONELINE_COMMENT_RE = re.compile(r"""
|
||
|
^ # comment must be at the start of the line
|
||
|
\s* # arbitrary whitespace
|
||
|
// # start of the comment
|
||
|
(.*) # the comment
|
||
|
$ # until the end of line
|
||
|
""", re.MULTILINE | re.VERBOSE
|
||
|
)
|
||
|
|
||
|
INLINE_COMMENT_RE = re.compile(r"""
|
||
|
([\,\"\[\]\{\}\d]) # anythig that might end a expression
|
||
|
\s+ # comment must be preceded by whitespace
|
||
|
// # start of the comment
|
||
|
\s # and succeded by whitespace
|
||
|
(?:[^\"\]\}\{\[]*) # the comment (except things which might be json)
|
||
|
$ # until the end of line
|
||
|
""", re.MULTILINE | re.VERBOSE)
|
||
|
|
||
|
TRAILING_COMMA_RE = re.compile(r"""
|
||
|
, # the comma
|
||
|
(?:\s*) # arbitrary whitespace
|
||
|
$ # only works if the trailing comma is followed by newline
|
||
|
(\s*) # arbitrary whitespace
|
||
|
([\]\}]) # end of an array or object
|
||
|
""", re.MULTILINE | re.VERBOSE)
|
||
|
|
||
|
|
||
|
def loads(raw_data):
|
||
|
if isinstance(raw_data, bytes):
|
||
|
raw_data = raw_data.decode('utf-8')
|
||
|
|
||
|
raw_data = ONELINE_COMMENT_RE.sub(r"", raw_data)
|
||
|
raw_data = INLINE_COMMENT_RE.sub(r"\1", raw_data)
|
||
|
raw_data = TRAILING_COMMA_RE.sub(r"\1\2", raw_data)
|
||
|
return json.loads(raw_data)
|
||
|
|
||
|
|
||
|
def parse_config(path):
|
||
|
def reset_section():
|
||
|
section.update({
|
||
|
'name': section.get('name', ""),
|
||
|
'sub_name': "",
|
||
|
'start_line': -1,
|
||
|
'end_line': -1,
|
||
|
'code_lines': [],
|
||
|
})
|
||
|
|
||
|
def start_section(line_index, line):
|
||
|
end_section()
|
||
|
if line.startswith("# "):
|
||
|
name = line[2:]
|
||
|
elif line.startswith("## "):
|
||
|
name = line[3:]
|
||
|
else:
|
||
|
name = ""
|
||
|
|
||
|
name = name.strip().replace(" ", "_").lower()
|
||
|
if name in SECTIONS:
|
||
|
section['name'] = name
|
||
|
else:
|
||
|
section['sub_name'] = name
|
||
|
section['start_line'] = line_index
|
||
|
|
||
|
def end_section():
|
||
|
if section['start_line'] >= 0:
|
||
|
if section['name'] == 'layout_config':
|
||
|
config.update(loads("\n".join(
|
||
|
section['code_lines']
|
||
|
)))
|
||
|
elif section['sub_name'].startswith('layer'):
|
||
|
layer_name = section['sub_name']
|
||
|
config['layer_lines'][layer_name] = section['code_lines']
|
||
|
|
||
|
reset_section()
|
||
|
|
||
|
def amend_section(line_index, line):
|
||
|
section['end_line'] = line_index
|
||
|
section['code_lines'].append(line)
|
||
|
|
||
|
config = DEFAULT_CONFIG.copy()
|
||
|
config.update({
|
||
|
'layer_lines': collections.OrderedDict(),
|
||
|
'macro_ids': {'UM'},
|
||
|
'unicode_macros': {},
|
||
|
})
|
||
|
|
||
|
section = {}
|
||
|
reset_section()
|
||
|
|
||
|
with io.open(path, encoding="utf-8") as fh:
|
||
|
for i, line in enumerate(fh):
|
||
|
if line.startswith("#"):
|
||
|
start_section(i, line)
|
||
|
elif line.startswith(" "):
|
||
|
amend_section(i, line[4:])
|
||
|
else:
|
||
|
# TODO: maybe parse description
|
||
|
pass
|
||
|
|
||
|
end_section()
|
||
|
assert 'layout' in config
|
||
|
return config
|
||
|
|
||
|
# header file parsing
|
||
|
|
||
|
IF0_RE = re.compile(r"""
|
||
|
^
|
||
|
#if 0
|
||
|
$.*?
|
||
|
#endif
|
||
|
""", re.MULTILINE | re.DOTALL | re.VERBOSE)
|
||
|
|
||
|
|
||
|
COMMENT_RE = re.compile(r"""
|
||
|
/\*
|
||
|
.*?
|
||
|
\*/"
|
||
|
""", re.MULTILINE | re.DOTALL | re.VERBOSE)
|
||
|
|
||
|
|
||
|
def read_header_file(path):
|
||
|
with io.open(path, encoding="utf-8") as fh:
|
||
|
data = fh.read()
|
||
|
data, _ = COMMENT_RE.subn("", data)
|
||
|
data, _ = IF0_RE.subn("", data)
|
||
|
return data
|
||
|
|
||
|
|
||
|
def regex_partial(re_str_fmt, flags):
|
||
|
def partial(*args, **kwargs):
|
||
|
re_str = re_str_fmt.format(*args, **kwargs)
|
||
|
return re.compile(re_str, flags)
|
||
|
return partial
|
||
|
|
||
|
|
||
|
KEYDEF_REP = regex_partial(r"""
|
||
|
#define
|
||
|
\s
|
||
|
(
|
||
|
(?:{}) # the prefixes
|
||
|
(?:\w+) # the key name
|
||
|
) # capture group end
|
||
|
""", re.MULTILINE | re.DOTALL | re.VERBOSE)
|
||
|
|
||
|
|
||
|
ENUM_RE = re.compile(r"""
|
||
|
(
|
||
|
enum
|
||
|
\s\w+\s
|
||
|
\{
|
||
|
.*? # the enum content
|
||
|
\}
|
||
|
;
|
||
|
) # capture group end
|
||
|
""", re.MULTILINE | re.DOTALL | re.VERBOSE)
|
||
|
|
||
|
|
||
|
ENUM_KEY_REP = regex_partial(r"""
|
||
|
(
|
||
|
{} # the prefixes
|
||
|
\w+ # the key name
|
||
|
) # capture group end
|
||
|
""", re.MULTILINE | re.DOTALL | re.VERBOSE)
|
||
|
|
||
|
|
||
|
def parse_keydefs(config, data):
|
||
|
prefix_options = "|".join(config['key_prefixes'])
|
||
|
keydef_re = KEYDEF_REP(prefix_options)
|
||
|
enum_key_re = ENUM_KEY_REP(prefix_options)
|
||
|
for match in keydef_re.finditer(data):
|
||
|
yield match.groups()[0]
|
||
|
|
||
|
for enum_match in ENUM_RE.finditer(data):
|
||
|
enum = enum_match.groups()[0]
|
||
|
for key_match in enum_key_re.finditer(enum):
|
||
|
yield key_match.groups()[0]
|
||
|
|
||
|
|
||
|
def parse_valid_keys(config, out_path):
|
||
|
basepath = os.path.abspath(os.path.join(os.path.dirname(out_path)))
|
||
|
dirpaths = []
|
||
|
subpaths = []
|
||
|
while len(subpaths) < 6:
|
||
|
path = os.path.join(basepath, *subpaths)
|
||
|
dirpaths.append(path)
|
||
|
dirpaths.append(os.path.join(path, "tmk_core", "common"))
|
||
|
dirpaths.append(os.path.join(path, "quantum"))
|
||
|
subpaths.append('..')
|
||
|
|
||
|
includes = set(config['keymaps_includes'])
|
||
|
includes.add("keycode.h")
|
||
|
|
||
|
valid_keycodes = set()
|
||
|
for dirpath, include in it.product(dirpaths, includes):
|
||
|
include_path = os.path.join(dirpath, include)
|
||
|
if os.path.exists(include_path):
|
||
|
header_data = read_header_file(include_path)
|
||
|
valid_keycodes.update(
|
||
|
parse_keydefs(config, header_data)
|
||
|
)
|
||
|
return valid_keycodes
|
||
|
|
||
|
|
||
|
# Keymap Parsing
|
||
|
|
||
|
def iter_raw_codes(layer_lines, filler, separator):
|
||
|
filler_re = re.compile("[" + filler + " ]")
|
||
|
for line in layer_lines:
|
||
|
line, _ = filler_re.subn("", line.strip())
|
||
|
if not line:
|
||
|
continue
|
||
|
codes = line.split(separator)
|
||
|
for code in codes[1:-1]:
|
||
|
yield code
|
||
|
|
||
|
|
||
|
def iter_indexed_codes(raw_codes, key_indexes):
|
||
|
key_rows = {}
|
||
|
key_indexes_flat = []
|
||
|
|
||
|
for row_index, key_indexes in enumerate(key_indexes):
|
||
|
for key_index in key_indexes:
|
||
|
key_rows[key_index] = row_index
|
||
|
key_indexes_flat.extend(key_indexes)
|
||
|
assert len(raw_codes) == len(key_indexes_flat)
|
||
|
for raw_code, key_index in zip(raw_codes, key_indexes_flat):
|
||
|
# we keep track of the row mostly for layout purposes
|
||
|
yield raw_code, key_index, key_rows[key_index]
|
||
|
|
||
|
|
||
|
LAYER_CHANGE_RE = re.compile(r"""
|
||
|
(DF|TG|MO)\(\d+\)
|
||
|
""", re.VERBOSE)
|
||
|
|
||
|
|
||
|
MACRO_RE = re.compile(r"""
|
||
|
M\(\w+\)
|
||
|
""", re.VERBOSE)
|
||
|
|
||
|
|
||
|
UNICODE_RE = re.compile(r"""
|
||
|
U[0-9A-F]{4}
|
||
|
""", re.VERBOSE)
|
||
|
|
||
|
|
||
|
NON_CODE = re.compile(r"""
|
||
|
^[^A-Z0-9_]$
|
||
|
""", re.VERBOSE)
|
||
|
|
||
|
|
||
|
def parse_uni_code(raw_code):
|
||
|
macro_id = "UC_" + (
|
||
|
unicodedata.name(raw_code)
|
||
|
.replace(" ", "_")
|
||
|
.replace("-", "_")
|
||
|
)
|
||
|
code = "M({})".format(macro_id)
|
||
|
uc_hex = "{:04X}".format(ord(raw_code))
|
||
|
return code, macro_id, uc_hex
|
||
|
|
||
|
|
||
|
def parse_key_code(raw_code, key_prefixes, valid_keycodes):
|
||
|
if raw_code in valid_keycodes:
|
||
|
return raw_code
|
||
|
|
||
|
for prefix in key_prefixes:
|
||
|
code = prefix + raw_code
|
||
|
if code in valid_keycodes:
|
||
|
return code
|
||
|
|
||
|
|
||
|
def parse_code(raw_code, key_prefixes, valid_keycodes):
|
||
|
if not raw_code:
|
||
|
return 'KC_TRNS', None, None
|
||
|
|
||
|
if LAYER_CHANGE_RE.match(raw_code):
|
||
|
return raw_code, None, None
|
||
|
|
||
|
if MACRO_RE.match(raw_code):
|
||
|
macro_id = raw_code[2:-1]
|
||
|
return raw_code, macro_id, None
|
||
|
|
||
|
if UNICODE_RE.match(raw_code):
|
||
|
hex_code = raw_code[1:]
|
||
|
return parse_uni_code(chr(int(hex_code, 16)))
|
||
|
|
||
|
if NON_CODE.match(raw_code):
|
||
|
return parse_uni_code(raw_code)
|
||
|
|
||
|
code = parse_key_code(raw_code, key_prefixes, valid_keycodes)
|
||
|
return code, None, None
|
||
|
|
||
|
|
||
|
def parse_keymap(config, key_indexes, layer_lines, valid_keycodes):
|
||
|
keymap = {}
|
||
|
raw_codes = list(iter_raw_codes(
|
||
|
layer_lines, config['filler'], config['separator']
|
||
|
))
|
||
|
indexed_codes = iter_indexed_codes(raw_codes, key_indexes)
|
||
|
key_prefixes = config['key_prefixes']
|
||
|
for raw_code, key_index, row_index in indexed_codes:
|
||
|
code, macro_id, uc_hex = parse_code(
|
||
|
raw_code, key_prefixes, valid_keycodes
|
||
|
)
|
||
|
# TODO: line numbers for invalid codes
|
||
|
err_msg = "Could not parse key '{}' on row {}".format(
|
||
|
raw_code, row_index
|
||
|
)
|
||
|
assert code is not None, err_msg
|
||
|
# print(repr(raw_code), repr(code), macro_id, uc_hex)
|
||
|
if macro_id:
|
||
|
config['macro_ids'].add(macro_id)
|
||
|
if uc_hex:
|
||
|
config['unicode_macros'][macro_id] = uc_hex
|
||
|
keymap[key_index] = (code, row_index)
|
||
|
return keymap
|
||
|
|
||
|
|
||
|
def parse_keymaps(config, valid_keycodes):
|
||
|
keymaps = collections.OrderedDict()
|
||
|
key_indexes = config.get(
|
||
|
'key_indexes', KEYBOARD_LAYOUTS[config['layout']]
|
||
|
)
|
||
|
# TODO: maybe validate key_indexes
|
||
|
|
||
|
for layer_name, layer_lines, in config['layer_lines'].items():
|
||
|
keymaps[layer_name] = parse_keymap(
|
||
|
config, key_indexes, layer_lines, valid_keycodes
|
||
|
)
|
||
|
return keymaps
|
||
|
|
||
|
# keymap.c output
|
||
|
|
||
|
USERCODE = """
|
||
|
// Runs just one time when the keyboard initializes.
|
||
|
void matrix_init_user(void) {
|
||
|
|
||
|
};
|
||
|
|
||
|
// Runs constantly in the background, in a loop.
|
||
|
void matrix_scan_user(void) {
|
||
|
uint8_t layer = biton32(layer_state);
|
||
|
|
||
|
ergodox_board_led_off();
|
||
|
ergodox_right_led_1_off();
|
||
|
ergodox_right_led_2_off();
|
||
|
ergodox_right_led_3_off();
|
||
|
switch (layer) {
|
||
|
case L1:
|
||
|
ergodox_right_led_1_on();
|
||
|
break;
|
||
|
case L2:
|
||
|
ergodox_right_led_2_on();
|
||
|
break;
|
||
|
case L3:
|
||
|
ergodox_right_led_3_on();
|
||
|
break;
|
||
|
case L4:
|
||
|
ergodox_right_led_1_on();
|
||
|
ergodox_right_led_2_on();
|
||
|
break;
|
||
|
case L5:
|
||
|
ergodox_right_led_1_on();
|
||
|
ergodox_right_led_3_on();
|
||
|
break;
|
||
|
// case L6:
|
||
|
// ergodox_right_led_2_on();
|
||
|
// ergodox_right_led_3_on();
|
||
|
// break;
|
||
|
// case L7:
|
||
|
// ergodox_right_led_1_on();
|
||
|
// ergodox_right_led_2_on();
|
||
|
// ergodox_right_led_3_on();
|
||
|
// break;
|
||
|
default:
|
||
|
ergodox_board_led_off();
|
||
|
break;
|
||
|
}
|
||
|
};
|
||
|
"""
|
||
|
|
||
|
MACROCODE = """
|
||
|
#define UC_MODE_WIN 0
|
||
|
#define UC_MODE_LINUX 1
|
||
|
#define UC_MODE_OSX 2
|
||
|
|
||
|
// TODO: allow default mode to be configured
|
||
|
static uint16_t unicode_mode = UC_MODE_WIN;
|
||
|
|
||
|
uint16_t hextokeycode(uint8_t hex) {{
|
||
|
if (hex == 0x0) {{
|
||
|
return KC_P0;
|
||
|
}}
|
||
|
if (hex < 0xA) {{
|
||
|
return KC_P1 + (hex - 0x1);
|
||
|
}}
|
||
|
return KC_A + (hex - 0xA);
|
||
|
}}
|
||
|
|
||
|
void unicode_action_function(uint16_t hi, uint16_t lo) {{
|
||
|
switch (unicode_mode) {{
|
||
|
case UC_MODE_WIN:
|
||
|
register_code(KC_LALT);
|
||
|
|
||
|
register_code(KC_PPLS);
|
||
|
unregister_code(KC_PPLS);
|
||
|
|
||
|
register_code(hextokeycode((hi & 0xF0) >> 4));
|
||
|
unregister_code(hextokeycode((hi & 0xF0) >> 4));
|
||
|
register_code(hextokeycode((hi & 0x0F)));
|
||
|
unregister_code(hextokeycode((hi & 0x0F)));
|
||
|
register_code(hextokeycode((lo & 0xF0) >> 4));
|
||
|
unregister_code(hextokeycode((lo & 0xF0) >> 4));
|
||
|
register_code(hextokeycode((lo & 0x0F)));
|
||
|
unregister_code(hextokeycode((lo & 0x0F)));
|
||
|
|
||
|
unregister_code(KC_LALT);
|
||
|
break;
|
||
|
case UC_MODE_LINUX:
|
||
|
register_code(KC_LCTL);
|
||
|
register_code(KC_LSFT);
|
||
|
|
||
|
register_code(KC_U);
|
||
|
unregister_code(KC_U);
|
||
|
|
||
|
register_code(hextokeycode((hi & 0xF0) >> 4));
|
||
|
unregister_code(hextokeycode((hi & 0xF0) >> 4));
|
||
|
register_code(hextokeycode((hi & 0x0F)));
|
||
|
unregister_code(hextokeycode((hi & 0x0F)));
|
||
|
register_code(hextokeycode((lo & 0xF0) >> 4));
|
||
|
unregister_code(hextokeycode((lo & 0xF0) >> 4));
|
||
|
register_code(hextokeycode((lo & 0x0F)));
|
||
|
unregister_code(hextokeycode((lo & 0x0F)));
|
||
|
|
||
|
unregister_code(KC_LCTL);
|
||
|
unregister_code(KC_LSFT);
|
||
|
break;
|
||
|
case UC_MODE_OSX:
|
||
|
break;
|
||
|
}}
|
||
|
}}
|
||
|
|
||
|
const macro_t *action_get_macro(keyrecord_t *record, uint8_t id, uint8_t opt) {{
|
||
|
if (!record->event.pressed) {{
|
||
|
return MACRO_NONE;
|
||
|
}}
|
||
|
// MACRODOWN only works in this function
|
||
|
switch(id) {{
|
||
|
case UM:
|
||
|
unicode_mode = (unicode_mode + 1) % 2;
|
||
|
break;
|
||
|
{macro_cases}
|
||
|
{unicode_macro_cases}
|
||
|
default:
|
||
|
break;
|
||
|
}}
|
||
|
return MACRO_NONE;
|
||
|
}};
|
||
|
"""
|
||
|
|
||
|
|
||
|
UNICODE_MACRO_TEMPLATE = """
|
||
|
case {macro_id}:
|
||
|
unicode_action_function(0x{hi:02x}, 0x{lo:02x});
|
||
|
break;
|
||
|
""".strip()
|
||
|
|
||
|
|
||
|
def unicode_macro_cases(config):
|
||
|
for macro_id, uc_hex in config['unicode_macros'].items():
|
||
|
hi = int(uc_hex, 16) >> 8
|
||
|
lo = int(uc_hex, 16) & 0xFF
|
||
|
unimacro_keys = ", ".join(
|
||
|
"T({})".format(
|
||
|
"KP_" + digit if digit.isdigit() else digit
|
||
|
) for digit in uc_hex
|
||
|
)
|
||
|
yield UNICODE_MACRO_TEMPLATE.format(
|
||
|
macro_id=macro_id, hi=hi, lo=lo
|
||
|
)
|
||
|
|
||
|
|
||
|
def iter_keymap_lines(keymap, row_indents=None):
|
||
|
col_widths = {}
|
||
|
col = 0
|
||
|
# first pass, figure out the column widths
|
||
|
prev_row_index = None
|
||
|
for code, row_index in keymap.values():
|
||
|
if row_index != prev_row_index:
|
||
|
col = 0
|
||
|
if row_indents:
|
||
|
col = row_indents[row_index]
|
||
|
col_widths[col] = max(len(code), col_widths.get(col, 0))
|
||
|
prev_row_index = row_index
|
||
|
col += 1
|
||
|
|
||
|
# second pass, yield the cell values
|
||
|
col = 0
|
||
|
prev_row_index = None
|
||
|
for key_index in sorted(keymap):
|
||
|
code, row_index = keymap[key_index]
|
||
|
if row_index != prev_row_index:
|
||
|
col = 0
|
||
|
yield "\n"
|
||
|
if row_indents:
|
||
|
for indent_col in range(row_indents[row_index]):
|
||
|
pad = " " * (col_widths[indent_col] - 4)
|
||
|
yield (" /*-*/" + pad)
|
||
|
col = row_indents[row_index]
|
||
|
else:
|
||
|
yield pad
|
||
|
yield " {}".format(code)
|
||
|
if key_index < len(keymap) - 1:
|
||
|
yield ","
|
||
|
# This will be yielded on the next iteration when
|
||
|
# we know that we're not at the end of a line.
|
||
|
pad = " " * (col_widths[col] - len(code))
|
||
|
prev_row_index = row_index
|
||
|
col += 1
|
||
|
|
||
|
|
||
|
def iter_keymap_parts(config, keymaps):
|
||
|
# includes
|
||
|
for include_path in config['keymaps_includes']:
|
||
|
yield '#include "{}"\n'.format(include_path)
|
||
|
|
||
|
yield "\n"
|
||
|
|
||
|
# definitions
|
||
|
for i, macro_id in enumerate(sorted(config['macro_ids'])):
|
||
|
yield "#define {} {}\n".format(macro_id, i)
|
||
|
|
||
|
yield "\n"
|
||
|
|
||
|
for i, layer_name in enumerate(config['layer_lines']):
|
||
|
yield '#define L{0:<3} {0:<5} // {1}\n'.format(i, layer_name)
|
||
|
|
||
|
yield "\n"
|
||
|
|
||
|
# keymaps
|
||
|
yield "const uint16_t PROGMEM keymaps[][MATRIX_ROWS][MATRIX_COLS] = {\n"
|
||
|
|
||
|
for i, layer_name in enumerate(config['layer_lines']):
|
||
|
# comment
|
||
|
layer_lines = config['layer_lines'][layer_name]
|
||
|
prefixed_lines = " * " + " * ".join(layer_lines)
|
||
|
yield "/*\n{} */\n".format(prefixed_lines)
|
||
|
|
||
|
# keymap codes
|
||
|
keymap = keymaps[layer_name]
|
||
|
row_indents = ROW_INDENTS.get(config['layout'])
|
||
|
keymap_lines = "".join(iter_keymap_lines(keymap, row_indents))
|
||
|
yield "[L{0}] = KEYMAP({1}\n),\n".format(i, keymap_lines)
|
||
|
|
||
|
yield "};\n\n"
|
||
|
|
||
|
# no idea what this is for
|
||
|
yield "const uint16_t PROGMEM fn_actions[] = {};\n"
|
||
|
|
||
|
# macros
|
||
|
yield MACROCODE.format(
|
||
|
macro_cases="",
|
||
|
unicode_macro_cases="\n".join(unicode_macro_cases(config)),
|
||
|
)
|
||
|
|
||
|
# TODO: dynamically create blinking lights
|
||
|
yield USERCODE
|
||
|
|
||
|
|
||
|
def main(argv=sys.argv[1:]):
|
||
|
if not argv or '-h' in argv or '--help' in argv:
|
||
|
print(__doc__)
|
||
|
return 0
|
||
|
|
||
|
in_path = os.path.abspath(argv[0])
|
||
|
if not os.path.exists(in_path):
|
||
|
print("No such file '{}'".format(in_path))
|
||
|
return 1
|
||
|
|
||
|
if len(argv) > 1:
|
||
|
out_path = os.path.abspath(argv[1])
|
||
|
else:
|
||
|
dirname = os.path.dirname(in_path)
|
||
|
out_path = os.path.join(dirname, "keymap.c")
|
||
|
|
||
|
config = parse_config(in_path)
|
||
|
valid_keys = parse_valid_keys(config, out_path)
|
||
|
keymaps = parse_keymaps(config, valid_keys)
|
||
|
|
||
|
with io.open(out_path, mode="w", encoding="utf-8") as fh:
|
||
|
for part in iter_keymap_parts(config, keymaps):
|
||
|
fh.write(part)
|
||
|
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
sys.exit(main())
|