e69da2db2c
* `qmk info`: Add `--ascii` flag * Fix typo * Force ASCII for Windows/MSYS2 * Make it gooder * Remove redundant windows check * ...And this too * Make pytest work on Windows
152 lines
4.2 KiB
Python
152 lines
4.2 KiB
Python
"""Functions that help us work with keyboards.
|
|
"""
|
|
from array import array
|
|
from math import ceil
|
|
from pathlib import Path
|
|
import os
|
|
from glob import glob
|
|
|
|
from qmk.c_parse import parse_config_h_file
|
|
from qmk.makefile import parse_rules_mk_file
|
|
|
|
BOX_DRAWING_CHARACTERS = {
|
|
"unicode": {
|
|
"tl": "┌",
|
|
"tr": "┐",
|
|
"bl": "└",
|
|
"br": "┘",
|
|
"v": "│",
|
|
"h": "─",
|
|
},
|
|
"ascii": {
|
|
"tl": " ",
|
|
"tr": " ",
|
|
"bl": "|",
|
|
"br": "|",
|
|
"v": "|",
|
|
"h": "_",
|
|
},
|
|
}
|
|
|
|
base_path = os.path.join(os.getcwd(), "keyboards") + os.path.sep
|
|
|
|
|
|
def _find_name(path):
|
|
"""Determine the keyboard name by stripping off the base_path and rules.mk.
|
|
"""
|
|
return path.replace(base_path, "").replace(os.path.sep + "rules.mk", "")
|
|
|
|
|
|
def list_keyboards():
|
|
"""Returns a list of all keyboards.
|
|
"""
|
|
# We avoid pathlib here because this is performance critical code.
|
|
kb_wildcard = os.path.join(base_path, "**", "rules.mk")
|
|
paths = [path for path in glob(kb_wildcard, recursive=True) if 'keymaps' not in path]
|
|
|
|
return sorted(map(_find_name, paths))
|
|
|
|
|
|
def config_h(keyboard):
|
|
"""Parses all the config.h files for a keyboard.
|
|
|
|
Args:
|
|
keyboard: name of the keyboard
|
|
|
|
Returns:
|
|
a dictionary representing the content of the entire config.h tree for a keyboard
|
|
"""
|
|
config = {}
|
|
cur_dir = Path('keyboards')
|
|
rules = rules_mk(keyboard)
|
|
keyboard = Path(rules['DEFAULT_FOLDER'] if 'DEFAULT_FOLDER' in rules else keyboard)
|
|
|
|
for dir in keyboard.parts:
|
|
cur_dir = cur_dir / dir
|
|
config = {**config, **parse_config_h_file(cur_dir / 'config.h')}
|
|
|
|
return config
|
|
|
|
|
|
def rules_mk(keyboard):
|
|
"""Get a rules.mk for a keyboard
|
|
|
|
Args:
|
|
keyboard: name of the keyboard
|
|
|
|
Returns:
|
|
a dictionary representing the content of the entire rules.mk tree for a keyboard
|
|
"""
|
|
keyboard = Path(keyboard)
|
|
cur_dir = Path('keyboards')
|
|
rules = parse_rules_mk_file(cur_dir / keyboard / 'rules.mk')
|
|
|
|
if 'DEFAULT_FOLDER' in rules:
|
|
keyboard = Path(rules['DEFAULT_FOLDER'])
|
|
|
|
for i, dir in enumerate(keyboard.parts):
|
|
cur_dir = cur_dir / dir
|
|
rules = parse_rules_mk_file(cur_dir / 'rules.mk', rules)
|
|
|
|
return rules
|
|
|
|
|
|
def render_layout(layout_data, render_ascii, key_labels=None):
|
|
"""Renders a single layout.
|
|
"""
|
|
textpad = [array('u', ' ' * 200) for x in range(50)]
|
|
style = 'ascii' if render_ascii else 'unicode'
|
|
box_chars = BOX_DRAWING_CHARACTERS[style]
|
|
|
|
for key in layout_data:
|
|
x = ceil(key.get('x', 0) * 4)
|
|
y = ceil(key.get('y', 0) * 3)
|
|
w = ceil(key.get('w', 1) * 4)
|
|
h = ceil(key.get('h', 1) * 3)
|
|
|
|
if key_labels:
|
|
label = key_labels.pop(0)
|
|
if label.startswith('KC_'):
|
|
label = label[3:]
|
|
else:
|
|
label = key.get('label', '')
|
|
|
|
label_len = w - 2
|
|
label_leftover = label_len - len(label)
|
|
|
|
if len(label) > label_len:
|
|
label = label[:label_len]
|
|
|
|
label_blank = ' ' * label_len
|
|
label_border = box_chars['h'] * label_len
|
|
label_middle = label + ' '*label_leftover # noqa: yapf insists there be no whitespace around *
|
|
|
|
top_line = array('u', box_chars['tl'] + label_border + box_chars['tr'])
|
|
lab_line = array('u', box_chars['v'] + label_middle + box_chars['v'])
|
|
mid_line = array('u', box_chars['v'] + label_blank + box_chars['v'])
|
|
bot_line = array('u', box_chars['bl'] + label_border + box_chars['br'])
|
|
|
|
textpad[y][x:x + w] = top_line
|
|
textpad[y + 1][x:x + w] = lab_line
|
|
for i in range(h - 3):
|
|
textpad[y + i + 2][x:x + w] = mid_line
|
|
textpad[y + h - 1][x:x + w] = bot_line
|
|
|
|
lines = []
|
|
for line in textpad:
|
|
if line.tounicode().strip():
|
|
lines.append(line.tounicode().rstrip())
|
|
|
|
return '\n'.join(lines)
|
|
|
|
|
|
def render_layouts(info_json, render_ascii):
|
|
"""Renders all the layouts from an `info_json` structure.
|
|
"""
|
|
layouts = {}
|
|
|
|
for layout in info_json['layouts']:
|
|
layout_data = info_json['layouts'][layout]['layout']
|
|
layouts[layout] = render_layout(layout_data, render_ascii)
|
|
|
|
return layouts
|