qmk_firmware/keyboards/signum/3_0/keymaps/default/generate_km.py
2023-06-01 17:25:08 +01:00

195 lines
7.4 KiB
Python
Executable File

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import json
import layout
import os
import re
def gen_uc_iter():
length = len(layout.uc_dict)
for key, value in sorted(layout.uc_dict.items()):
length -= 1
if length:
yield (key, value, False)
else:
yield (key, value, True)
def _translate(s):
if re.match("^[0-9]$", s):
return ("KC_{0}".format(s), " {0} ".format(s))
elif re.match("^[a-z]$", s):
return ("KC_{0}".format(s.upper()), " {0} ".format(s))
elif re.match("^[A-Z]$", s):
return ("S(KC_{0})".format(s), " {0} ".format(s))
elif re.match("^F[0-9]{1,2}$", s): # Fn, works from F0 to F99
return ("KC_{0}".format(s), "{0:^7}".format(s))
elif re.match("^DF[0-9]{1,2}$", s): # DFn, works from DF0 to DF99
return ("DF({0})".format(s[2:]), "{0:^7}".format(s))
elif re.match("^MO[0-9]{1,2}$", s): # MOn, works from MO0 to MO99
return ("MO({0})".format(s[2:]), "{0:^7}".format(s))
elif re.match("^OSL[0-9]{1,2}$", s): # OSLn, works from OSL0 to OSL99
return ("OSL({0})".format(s[3:]), "{0:^7}".format(s))
elif re.match("^TG[0-9]{1,2}$", s): # TGn, works from TG0 to TG99
return ("TG({0})".format(s[2:]), "{0:^7}".format(s))
elif re.match("^TO[0-9]{1,2}$", s): # Tn, works from TO0 to TO99
return ("TO({0})".format(s[2:]), "{0:^7}".format(s))
elif re.match("^TT[0-9]{1,2}$", s): # Tn, works from TT0 to TT99
return ("TT({0})".format(s[2:]), "{0:^7}".format(s))
elif s in layout.uc_dict:
return ("UM("+s+")", " {0} ".format(chr(int(layout.uc_dict[s], 0))))
elif s in layout.qmk_dict:
return (layout.qmk_dict[s], "{0:^7}".format(s))
elif s == s.upper() and s.startswith("KC_"):
return (s, "{0:^7}".format(s[2:]))
else:
return ("XXXXXXX", " {0} ".format(chr(128165)))
def toKC(s):
return _translate(s)[0]
def toLgd(s):
return _translate(s)[1]
def quoteC(text):
yield "/*"
for line in text:
yield " * " + line
yield " */\n"
def getKeymapText(id, layer, columns, rows):
keymap = []
keymap.append("Layer %d" % id)
keymap.append("------------------------------------------------- -------------------------------------------------")
keymap.append("|{0}|{1}|{2}|{3}|{4}|{5}| |{6}|{7}|{8}|{9}|{10}|{11}|".format(*map(toLgd, layer[:12])))
keymap.append("------------------------------------------------- -------------------------------------------------")
keymap.append("|{0}|{1}|{2}|{3}|{4}|{5}| |{6}|{7}|{8}|{9}|{10}|{11}|".format(*map(toLgd, layer[12:24])))
keymap.append("------------------------------------------------- -------------------------------------------------")
keymap.append("|{0}|{1}|{2}|{3}|{4}|{5}| |{6}|{7}|{8}|{9}|{10}|{11}|".format(*map(toLgd, layer[24:36])))
keymap.append("-----------------------------------------------------------------------------------------------------------------")
keymap.append(" {0} {1} {2} |{3}|{4}|{5}|{6}|{7}|{8}| {9} {10} {11}".format(*map(toLgd, layer[36:48])).rstrip())
keymap.append(" -------------------------------------------------")
return keymap
def writeKeymap(f_template, f_keymap, layers, columns, rows):
doCopy = False
for line in f_template:
doCopy = True
if line.startswith("//<enum/>"):
doCopy = False
# f_keymap.write(str(layout.uc_dict))
for k, v, isLast in gen_uc_iter():
if isLast:
f_keymap.write(k + "\n")
else:
f_keymap.write(k + ",\n")
elif line.startswith("//<uc_map/>"):
doCopy = False
for k, v, isLast in gen_uc_iter():
if isLast:
f_keymap.write(u"\t[{0}] = {1} // {2}\n".format(k, v, chr(int(v, 0))))
else:
f_keymap.write(u"\t[{0}] = {1}, // {2}\n".format(k, v, chr(int(v, 0))))
elif line.startswith("//<keymaps/>"):
doCopy = False
for layer, L in enumerate(layers):
r_counter = rows
f_keymap.write('\n'.join(quoteC(getKeymapText(layer, L, columns, rows))))
l_code = '\tLAYOUT_ortho_4x12(\n'
for r in range(r_counter):
r_counter -= 1
c_counter = columns
l_code += '\t\t'
for c in range(c_counter):
c_counter -= 1
if c != 0:
l_code += " "
l_code += "%s" % toKC(L[r*columns + columns-c_counter-1])
if r_counter or c_counter:
l_code += ","
l_code += '\n'
if layer + 1 != len(layout.layers):
l_code += "\t),\n\n"
else:
l_code += "\t)\n"
f_keymap.write(l_code)
if doCopy:
f_keymap.write(line)
def getKeymapJSON(keyboard, keymap, layout, layers):
return json.dumps({
'keyboard': keyboard,
'keymap': keymap,
'layout': layout,
'layers': layers
}, sort_keys=True, indent=4)
def getKeymapAsciidoc(title, layers, columns, rows):
yield '= ' + title
yield ''
for id, layer in enumerate(layers):
keymap = getKeymapText(id, layer, columns, rows)
if len(keymap):
yield '.' + keymap[0]
yield '--------------------------'
for line in keymap[1:]:
yield ' ' + line
yield '--------------------------'
yield ''
def layersToKC(layers):
return [list(map(toKC, layer)) for layer in layers]
def pathToKeymap(path):
head, keymap = os.path.split(path)
_, keymapsdir = os.path.split(head)
if keymapsdir == 'keymaps':
return keymap
def pathToKeyboard(path):
head, keymap = os.path.split(path)
head, keymapsdir = os.path.split(head)
if keymapsdir == 'keymaps':
head, dir = os.path.split(head)
while dir not in ('/', 'keyboards'):
yield dir
head, dir = os.path.split(head)
if __name__ == "__main__":
with open("km_template.txt", mode="r") as f_template:
with open("keymap.c", mode="w", encoding='utf-8') as f_keymap:
writeKeymap(f_template, f_keymap, layout.layers, columns=12, rows=4)
abspath = os.path.dirname(os.path.abspath(__file__))
keyboard = list(reversed(list(pathToKeyboard(abspath))))
keymap = pathToKeymap(abspath)
keyboard_layout = 'LAYOUT_ortho_4x12'
with open("%s_%s.json" % ('_'.join(keyboard), keymap), mode="w") as f_keymapjson:
f_keymapjson.write(
getKeymapJSON(
'/'.join(keyboard),
keymap,
keyboard_layout,
layersToKC(layout.layers))
)
with open("keymap.adoc", mode="w") as f_keymapasciidoc:
f_keymapasciidoc.write('\n'.join(getKeymapAsciidoc('Signum 3.0 %s_%s' % ('_'.join(keyboard), keymap), layout.layers, columns=12, rows=4)))
print("Run the following command to generate a PDF from the keymap")
print("a2x -f pdf --xsltproc-opts '--stringparam page.orientation landscape --stringparam body.font.master 12' --fop -v keymap.adoc")