gBoards Common (#8921)
Co-Authored-By: Drashna Jaelre <drashna@live.com>
This commit is contained in:
Jeremy Bernhardt
GitHub
298
keyboards/gboards/dicts/aset/_generator/main.go
Normal file
298
keyboards/gboards/dicts/aset/_generator/main.go
Normal file
@@ -0,0 +1,298 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"io/ioutil"
|
||||
"fmt"
|
||||
"encoding/json"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"sort"
|
||||
"hash/crc64"
|
||||
//"encoding/base64"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// Show Usage
|
||||
if len(os.Args) < 3 {
|
||||
fmt.Println("Usage: ./keymap-gen src-dir out-dir")
|
||||
fmt.Println("Outputs c files in out-dir")
|
||||
fmt.Println("Make sure the dirs exist.")
|
||||
return
|
||||
}
|
||||
|
||||
files, err := filepath.Glob(os.Args[1] + "/*.json")
|
||||
if err != nil {
|
||||
fmt.Printf("Could not open src-dir: %v\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
for _, fname := range(files) {
|
||||
fmt.Println("Processing: ", fname)
|
||||
// Read the source
|
||||
data, err := ioutil.ReadFile(fname)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
// Unbundle Data
|
||||
var FullDict map[string]Entry
|
||||
json.Unmarshal(data, &FullDict)
|
||||
|
||||
// Loop over entries and store
|
||||
var output []string
|
||||
for i,v := range FullDict {
|
||||
if i == "0" {
|
||||
continue
|
||||
}
|
||||
|
||||
// Special handling for numbermap
|
||||
var entry string
|
||||
if strings.Contains(fname, "num") {
|
||||
entry = v.toKeymap("NUM|")
|
||||
} else {
|
||||
entry = v.toKeymap("")
|
||||
}
|
||||
|
||||
if entry != "" {
|
||||
output = append(output, entry)
|
||||
}
|
||||
}
|
||||
|
||||
// Sort by length, then amount of whitespace lol
|
||||
sort.Slice(output, func (i,j int) bool {
|
||||
var maxLen int
|
||||
if len(output[i]) > len(output[j]) {
|
||||
maxLen = len(output[i])
|
||||
} else {
|
||||
maxLen = len(output[j])
|
||||
}
|
||||
|
||||
return maxLen-strings.Count(output[i][:40], " ") < maxLen-strings.Count(output[j][:40], " ")
|
||||
})
|
||||
|
||||
// Whack a disclaimer
|
||||
output = append([]string{"// This file is automatically generated. Do not edit it!\n\n"}, output...)
|
||||
|
||||
// Write all data out
|
||||
outName := filepath.Base(fname)
|
||||
outName = os.Args[2] + outName[:len(outName)-5]+".def"
|
||||
fmt.Println("Saving: ", outName)
|
||||
ioutil.WriteFile(outName, []byte(strings.Join(output, "")), 0755)
|
||||
}
|
||||
}
|
||||
|
||||
func (e Entry) toKeymap(prefix string) (string) {
|
||||
// storage for parts
|
||||
var command, chord, arg string
|
||||
wordInfo := parseWords(e)
|
||||
|
||||
// Handle prefix
|
||||
if prefix != "" {
|
||||
chord = prefix
|
||||
}
|
||||
|
||||
// Format the chord
|
||||
keys := []string{"AA", "AS", "AE", "AT", "AN", "AI", "AO", "AP"}
|
||||
for i, v := range e.Input {
|
||||
chord += keys[v-1]
|
||||
|
||||
if i != len(e.Input)-1 {
|
||||
chord += "|"
|
||||
}
|
||||
}
|
||||
|
||||
// Handle specials/base first
|
||||
var ok bool
|
||||
var v []string
|
||||
|
||||
if e.Special != "" {
|
||||
v, ok = QMKLookup[e.Special]
|
||||
}
|
||||
if !ok && e.Base != "" {
|
||||
v, ok = QMKLookup[e.Base]
|
||||
}
|
||||
|
||||
if ok {
|
||||
// Determine way to send key
|
||||
if len(v) == 1 {
|
||||
command = "PRES("
|
||||
} else {
|
||||
command = "KEYS("
|
||||
}
|
||||
}
|
||||
|
||||
if ok {
|
||||
if len(v) > 1 {
|
||||
arg += "{"
|
||||
}
|
||||
|
||||
// String together args
|
||||
for ii, vv := range(v) {
|
||||
arg += vv
|
||||
if ii == len(v)-1 && len(v) > 1 {
|
||||
arg += ", COMBO_END}"
|
||||
} else if ii != len(v)-1 {
|
||||
arg += ", "
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
hash := crc64.Checksum([]byte(fmt.Sprintf("%v%v", arg, chord)), crc64.MakeTable(crc64.ECMA))
|
||||
hashStr := fmt.Sprintf("cmb_%x", hash)
|
||||
wordSpacer := strings.Repeat(" ", 42-len(arg))
|
||||
if command == "KEYS(" {
|
||||
arg = fmt.Sprintf("%v, %v %v", hashStr, wordSpacer, arg)
|
||||
} else {
|
||||
arg = fmt.Sprintf("%65v", arg)
|
||||
}
|
||||
|
||||
goto Found
|
||||
}
|
||||
|
||||
// Parse out word info
|
||||
if wordInfo.LRank == 0 && wordInfo.RRank == 0 {
|
||||
goto Blank
|
||||
}
|
||||
|
||||
if wordInfo.LRank != 0 || wordInfo.RRank != 0 {
|
||||
if wordInfo.LRank != 0 && wordInfo.RRank != 0 {
|
||||
// Just blank the structure and recall
|
||||
left, right := e, e
|
||||
left.Trw = nil
|
||||
right.Tlw = nil
|
||||
return fmt.Sprintf("%v%v", left.toKeymap("LFT|"), right.toKeymap("RGT|"))
|
||||
}
|
||||
|
||||
var word string
|
||||
if wordInfo.LRank > wordInfo.RRank {
|
||||
word = wordInfo.LWord
|
||||
} else {
|
||||
word = wordInfo.RWord
|
||||
}
|
||||
|
||||
// Add in thumb
|
||||
chord = "AR|" + chord
|
||||
|
||||
|
||||
// generate function name
|
||||
hash := crc64.Checksum([]byte(word), crc64.MakeTable(crc64.ECMA))
|
||||
hashStr := fmt.Sprintf("str_%016X", hash)
|
||||
command = "SUBS("
|
||||
wordSpacer := strings.Repeat(" ", 40-len(word))
|
||||
arg = fmt.Sprintf("%v, %v \"%v \"", hashStr, wordSpacer, word)
|
||||
goto Found
|
||||
}
|
||||
|
||||
panic(e.String())
|
||||
|
||||
Found:
|
||||
chord += ","
|
||||
return fmt.Sprintf("%v%-35v%v)\n", command, chord, arg)
|
||||
|
||||
Blank:
|
||||
return ""
|
||||
}
|
||||
|
||||
type Entry struct {
|
||||
Input []int
|
||||
Base string
|
||||
Tlw []interface{}
|
||||
Trw []interface{}
|
||||
Special string
|
||||
}
|
||||
type Word struct {
|
||||
LWord string
|
||||
LRank float64
|
||||
RWord string
|
||||
RRank float64
|
||||
}
|
||||
|
||||
func parseWords(e Entry) (ret Word) {
|
||||
if len(e.Tlw) > 0 {
|
||||
ret.LWord = e.Tlw[0].(string)
|
||||
ret.LRank= e.Tlw[1].(float64)
|
||||
}
|
||||
if len(e.Trw) > 0 {
|
||||
ret.RWord = e.Trw[0].(string)
|
||||
ret.RRank= e.Trw[1].(float64)
|
||||
}
|
||||
return ret
|
||||
}
|
||||
func (e Entry) String() (ret string) {
|
||||
ret = fmt.Sprintln("Input: ", e.Input)
|
||||
ret += fmt.Sprintln("Base: ", e.Base)
|
||||
ret += fmt.Sprintln("Left: ", e.Tlw)
|
||||
ret += fmt.Sprintln("Right: ", e.Trw)
|
||||
ret += fmt.Sprintln("Special: ", e.Special)
|
||||
|
||||
return ret
|
||||
}
|
||||
|
||||
var QMKLookup = map[string][]string {
|
||||
"!":[]string{"KC_LSFT", "KC_1"},
|
||||
"'":[]string{"KC_QUOT"},
|
||||
"(":[]string{"KC_LSFT", "KC_9"},
|
||||
")":[]string{"KC_LSFT", "KC_0"},
|
||||
",":[]string{"KC_COMM"},
|
||||
"-":[]string{"KC_MINS"},
|
||||
".":[]string{"KC_DOT"},
|
||||
";":[]string{"KC_SCLN"},
|
||||
"?":[]string{"KC_QUOT"},
|
||||
"a":[]string{"KC_A"},
|
||||
"b":[]string{"KC_B"},
|
||||
"c":[]string{"KC_C"},
|
||||
"d":[]string{"KC_D"},
|
||||
"e":[]string{"KC_E"},
|
||||
"f":[]string{"KC_F"},
|
||||
"g":[]string{"KC_G"},
|
||||
"h":[]string{"KC_H"},
|
||||
"i":[]string{"KC_I"},
|
||||
"j":[]string{"KC_J"},
|
||||
"k":[]string{"KC_K"},
|
||||
"l":[]string{"KC_L"},
|
||||
"m":[]string{"KC_M"},
|
||||
"n":[]string{"KC_N"},
|
||||
"o":[]string{"KC_O"},
|
||||
"p":[]string{"KC_P"},
|
||||
"q":[]string{"KC_Q"},
|
||||
"r":[]string{"KC_R"},
|
||||
"s":[]string{"KC_S"},
|
||||
"t":[]string{"KC_T"},
|
||||
"u":[]string{"KC_U"},
|
||||
"v":[]string{"KC_V"},
|
||||
"w":[]string{"KC_W"},
|
||||
"x":[]string{"KC_X"},
|
||||
"y":[]string{"KC_Y"},
|
||||
"z":[]string{"KC_Z"},
|
||||
|
||||
//specials
|
||||
"bksp":[]string{"KC_BSPC"},
|
||||
"enter":[]string{"KC_ENT"},
|
||||
//"numsym":[]string{"NUM)"}, //TODO: Sticky
|
||||
//"LETTERS":[]string{"KC_SPC"},
|
||||
|
||||
//symbols
|
||||
"[":[]string{"KC_LBRC"},
|
||||
"]":[]string{"KC_RBRC"},
|
||||
" ":[]string{"KC_SPC"},
|
||||
"1":[]string{"KC_1"},
|
||||
"2":[]string{"KC_2"},
|
||||
"3":[]string{"KC_3"},
|
||||
"4":[]string{"KC_4"},
|
||||
"5":[]string{"KC_5"},
|
||||
"6":[]string{"KC_6"},
|
||||
"7":[]string{"KC_7"},
|
||||
"8":[]string{"KC_8"},
|
||||
"9":[]string{"KC_9"},
|
||||
"0":[]string{"KC_0"},
|
||||
"=":[]string{"KC_EQL"},
|
||||
"Fn":[]string{"KC_NO"},
|
||||
"SPACE":[]string{"KC_SPC"},
|
||||
"Home":[]string{"KC_HOME"},
|
||||
"End":[]string{"KC_END"},
|
||||
" ":[]string{"KC_TAB"},
|
||||
" ":[]string{"KC_TAB"},
|
||||
"\t":[]string{"KC_TAB"},
|
||||
"`":[]string{"KC_GRV"},
|
||||
}
|
||||
Reference in New Issue
Block a user