qmk_firmware/keyboards/signum/3_0/elitec/keymaps/default/generate_km.py

#!/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 ("X("+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 writeKeymap(f_template, f_keymap, 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(layout.layers):
                r_counter = rows
                f_keymap.write("/* Layer %d\n" % layer)
                f_keymap.write(" * -------------------------------------------------               -------------------------------------------------\n")
                f_keymap.write(" * |{0}|{1}|{2}|{3}|{4}|{5}|               |{6}|{7}|{8}|{9}|{10}|{11}|\n".format(*map(toLgd, L[:12])))
                f_keymap.write(" * -------------------------------------------------               -------------------------------------------------\n")
                f_keymap.write(" * |{0}|{1}|{2}|{3}|{4}|{5}|               |{6}|{7}|{8}|{9}|{10}|{11}|\n".format(*map(toLgd, L[12:24])))
                f_keymap.write(" * -------------------------------------------------               -------------------------------------------------\n")
                f_keymap.write(" * |{0}|{1}|{2}|{3}|{4}|{5}|               |{6}|{7}|{8}|{9}|{10}|{11}|\n".format(*map(toLgd, L[24:36])))
                f_keymap.write(" * -----------------------------------------------------------------------------------------------------------------\n")
                f_keymap.write(" *  {0} {1} {2}        |{3}|{4}|{5}|{6}|{7}|{8}|        {9} {10} {11}".format(*map(toLgd, L[36:48])).rstrip()+"\n")
                f_keymap.write(" *                                 -------------------------------------------------\n")
                f_keymap.write(" */\n")

                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 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, 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))
                )