mirror of
https://github.com/Ralim/IronOS.git
synced 2025-02-26 07:53:55 +00:00
543 lines
18 KiB
Python
Executable File
543 lines
18 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# coding=utf-8
|
|
from __future__ import print_function
|
|
import json
|
|
import os
|
|
import io
|
|
from datetime import datetime
|
|
import sys
|
|
import fontTables
|
|
import re
|
|
import subprocess
|
|
|
|
TRANSLATION_CPP = "Translation.cpp"
|
|
UNIT_H = "unit.h"
|
|
|
|
try:
|
|
to_unicode = unicode
|
|
except NameError:
|
|
to_unicode = str
|
|
|
|
|
|
# Loading a single JSON file
|
|
def loadJson(fileName, skipFirstLine):
|
|
with io.open(fileName, mode="r", encoding="utf-8") as f:
|
|
if skipFirstLine:
|
|
f.readline()
|
|
|
|
obj = json.loads(f.read())
|
|
|
|
return obj
|
|
|
|
|
|
# Reading all language translations into a dictionary by langCode
|
|
def readTranslations(jsonDir):
|
|
langDict = {}
|
|
UnitDict = {}
|
|
|
|
# Read all translation files from the input dir
|
|
for fileName in os.listdir(jsonDir):
|
|
|
|
fileWithPath = os.path.join(jsonDir, fileName)
|
|
lf = fileName.lower()
|
|
|
|
# Read only translation_XX.json
|
|
if lf.startswith("translation_") and lf.endswith(".json"):
|
|
try:
|
|
lang = loadJson(fileWithPath, False)
|
|
except json.decoder.JSONDecodeError as e:
|
|
print("Failed to decode " + lf)
|
|
print(str(e))
|
|
sys.exit(2)
|
|
|
|
# Extract lang code from file name
|
|
langCode = fileName[12:-5].upper()
|
|
# ...and the one specified in the JSON file...
|
|
try:
|
|
langCodeFromJson = lang['languageCode']
|
|
except KeyError:
|
|
langCodeFromJson = "(missing)"
|
|
|
|
try:
|
|
TempUnitF_FromJson = lang['tempUnitFahrenheit']
|
|
except KeyError:
|
|
TempUnitF_FromJson = True # Default to true.
|
|
|
|
# ...cause they should be the same!
|
|
if langCode != langCodeFromJson:
|
|
raise ValueError("Invalid languageCode " + langCodeFromJson +
|
|
" in file " + fileName)
|
|
|
|
langDict[langCode] = lang
|
|
UnitDict[langCode] = TempUnitF_FromJson
|
|
return langDict, UnitDict
|
|
|
|
|
|
def writeStart(f):
|
|
f.write(
|
|
to_unicode(
|
|
"""// WARNING: THIS FILE WAS AUTO GENERATED BY make_translation.py. PLEASE DO NOT EDIT.
|
|
|
|
#include "Translation.h"
|
|
#ifndef LANG
|
|
#define LANG_EN
|
|
#endif
|
|
"""))
|
|
|
|
|
|
def writeStartUnit(f):
|
|
f.write(
|
|
to_unicode(
|
|
"""// WARNING: THIS FILE WAS AUTO GENERATED BY make_translation.py. PLEASE DO NOT EDIT.
|
|
|
|
/**
|
|
* °F Fahrenheit Support
|
|
* You will find the default Fahrenheit configuration in the translation_xx.json
|
|
* If tempUnitFahrenheit is set to:
|
|
* true - you can switch in menu settings to Fahrenheit or Celsius.
|
|
* false - you see only Celsius. All settings are then is in Celsius only.
|
|
*/
|
|
|
|
#ifndef _UNIT_H
|
|
#define _UNIT_H\n
|
|
"""))
|
|
|
|
def escapeC(s):
|
|
return s.replace("\"", "\\\"")
|
|
|
|
|
|
def getConstants():
|
|
# Extra constants that are used in the firmware that are shared across all languages
|
|
consants = []
|
|
consants.append(('SymbolPlus', '+'))
|
|
consants.append(('SymbolMinus', '-'))
|
|
consants.append(('SymbolSpace', ' '))
|
|
consants.append(('SymbolDot', '.'))
|
|
consants.append(('SymbolDegC', 'C'))
|
|
consants.append(('SymbolDegF', 'F'))
|
|
consants.append(('SymbolMinutes', 'M'))
|
|
consants.append(('SymbolSeconds', 'S'))
|
|
consants.append(('SymbolWatts', 'W'))
|
|
consants.append(('SymbolVolts', 'V'))
|
|
consants.append(('SymbolDC', 'DC'))
|
|
consants.append(('SymbolCellCount', 'S'))
|
|
consants.append(('SymbolVersionNumber', buildVersion))
|
|
return consants
|
|
|
|
|
|
|
|
def getDebugMenu():
|
|
constants = []
|
|
constants.append(datetime.today().strftime('%d-%m-%y'))
|
|
constants.append("HW G ")
|
|
constants.append("HW M ")
|
|
constants.append("HW P ")
|
|
constants.append("Time ")
|
|
constants.append("Move ")
|
|
constants.append("RTip ")
|
|
constants.append("CTip ")
|
|
constants.append("CHan ")
|
|
constants.append("Vin ")
|
|
constants.append("PCB ") # PCB Version AKA IMU version
|
|
return constants
|
|
|
|
|
|
def getLetterCounts(defs, lang):
|
|
textList = []
|
|
# iterate over all strings
|
|
obj = lang['menuOptions']
|
|
for mod in defs['menuOptions']:
|
|
eid = mod['id']
|
|
textList.append(obj[eid]['desc'])
|
|
|
|
obj = lang['messages']
|
|
for mod in defs['messages']:
|
|
eid = mod['id']
|
|
if eid not in obj:
|
|
textList.append(mod['default'])
|
|
else:
|
|
textList.append(obj[eid])
|
|
|
|
obj = lang['characters']
|
|
|
|
for mod in defs['characters']:
|
|
eid = mod['id']
|
|
textList.append(obj[eid])
|
|
|
|
obj = lang['menuOptions']
|
|
for mod in defs['menuOptions']:
|
|
eid = mod['id']
|
|
textList.append(obj[eid]['text2'][0])
|
|
textList.append(obj[eid]['text2'][1])
|
|
|
|
obj = lang['menuGroups']
|
|
for mod in defs['menuGroups']:
|
|
eid = mod['id']
|
|
textList.append(obj[eid]['text2'][0])
|
|
textList.append(obj[eid]['text2'][1])
|
|
|
|
obj = lang['menuGroups']
|
|
for mod in defs['menuGroups']:
|
|
eid = mod['id']
|
|
textList.append(obj[eid]['desc'])
|
|
constants = getConstants()
|
|
for x in constants:
|
|
textList.append(x[1])
|
|
textList.extend(getDebugMenu())
|
|
|
|
# collapse all strings down into the composite letters and store totals for these
|
|
|
|
symbolCounts = {}
|
|
for line in textList:
|
|
line = line.replace('\n', '').replace('\r', '')
|
|
line = line.replace('\\n', '').replace('\\r', '')
|
|
if len(line):
|
|
# print(line)
|
|
for letter in line:
|
|
symbolCounts[letter] = symbolCounts.get(letter, 0) + 1
|
|
symbolCounts = sorted(
|
|
symbolCounts.items(),
|
|
key=lambda kv: (kv[1], kv[0])) # swap to Big -> little sort order
|
|
symbolCounts = list(map(lambda x: x[0], symbolCounts))
|
|
symbolCounts.reverse()
|
|
return symbolCounts
|
|
|
|
|
|
def getFontMapAndTable(textList):
|
|
# the text list is sorted
|
|
# allocate out these in their order as number codes
|
|
symbolMap = {}
|
|
symbolMap['\n'] = '\\x01' # Force insert the newline char
|
|
index = 2 # start at 2, as 0= null terminator,1 = new line
|
|
forcedFirstSymbols = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9']
|
|
# enforce numbers are first
|
|
for sym in forcedFirstSymbols:
|
|
symbolMap[sym] = "\\x%0.2X" % index
|
|
index = index + 1
|
|
if len(textList) > (253 - len(forcedFirstSymbols)):
|
|
print('Error, too many used symbols for this version')
|
|
exit(1)
|
|
print('Generating fonts for {} symbols'.format(len(textList)))
|
|
|
|
for sym in textList:
|
|
if sym not in symbolMap:
|
|
symbolMap[sym] = "\\x%0.2X" % index
|
|
index = index + 1
|
|
# Get the font table
|
|
fontTableStrings = []
|
|
fontSmallTableStrings = []
|
|
fontTable = fontTables.getFontMap()
|
|
fontSmallTable = fontTables.getSmallFontMap()
|
|
for sym in forcedFirstSymbols:
|
|
if sym not in fontTable:
|
|
print('Missing Large font element for {}'.format(sym))
|
|
exit(1)
|
|
fontLine = fontTable[sym]
|
|
fontTableStrings.append(
|
|
fontLine + "//{} -> {}".format(symbolMap[sym], sym))
|
|
if sym not in fontSmallTable:
|
|
print('Missing Small font element for {}'.format(sym))
|
|
exit(1)
|
|
fontLine = fontSmallTable[sym]
|
|
fontSmallTableStrings.append(
|
|
fontLine + "//{} -> {}".format(symbolMap[sym], sym))
|
|
|
|
for sym in textList:
|
|
if sym not in fontTable:
|
|
print('Missing Large font element for {}'.format(sym))
|
|
exit(1)
|
|
if sym not in forcedFirstSymbols:
|
|
fontLine = fontTable[sym]
|
|
fontTableStrings.append(
|
|
fontLine + "//{} -> {}".format(symbolMap[sym], sym))
|
|
if sym not in fontSmallTable:
|
|
print('Missing Small font element for {}'.format(sym))
|
|
exit(1)
|
|
fontLine = fontSmallTable[sym]
|
|
fontSmallTableStrings.append(
|
|
fontLine + "//{} -> {}".format(symbolMap[sym], sym))
|
|
outputTable = "const uint8_t USER_FONT_12[] = {" + to_unicode("\n")
|
|
for line in fontTableStrings:
|
|
# join font table int one large string
|
|
outputTable = outputTable + line + to_unicode("\n")
|
|
outputTable = outputTable + "};" + to_unicode("\n")
|
|
outputTable = outputTable + "const uint8_t USER_FONT_6x8[] = {" + to_unicode(
|
|
"\n")
|
|
for line in fontSmallTableStrings:
|
|
# join font table int one large string
|
|
outputTable = outputTable + line + to_unicode("\n")
|
|
outputTable = outputTable + "};" + to_unicode("\n")
|
|
return (outputTable, symbolMap)
|
|
|
|
|
|
def convStr(symbolConversionTable, text):
|
|
# convert all of the symbols from the string into escapes for their content
|
|
outputString = ""
|
|
for c in text.replace('\\r', '').replace('\\n', '\n'):
|
|
if c not in symbolConversionTable:
|
|
print('Missing font definition for {}'.format(c))
|
|
else:
|
|
outputString = outputString + symbolConversionTable[c]
|
|
return outputString
|
|
|
|
|
|
def writeLanguage(languageCode, defs, f):
|
|
print("Generating block for " + languageCode)
|
|
lang = langDict[languageCode]
|
|
# Iterate over all of the text to build up the symbols & counts
|
|
textList = getLetterCounts(defs, lang)
|
|
# From the letter counts, need to make a symbol translator & write out the font
|
|
(fontTableText, symbolConversionTable) = getFontMapAndTable(textList)
|
|
|
|
f.write(to_unicode("\n#ifdef LANG_" + languageCode + "\n"))
|
|
f.write(fontTableText)
|
|
try:
|
|
langName = lang['languageLocalName']
|
|
except KeyError:
|
|
langName = languageCode
|
|
|
|
f.write(to_unicode("// ---- " + langName + " ----\n\n"))
|
|
|
|
# ----- Writing SettingsDescriptions
|
|
obj = lang['menuOptions']
|
|
f.write(to_unicode("const char* SettingsDescriptions[] = {\n"))
|
|
|
|
maxLen = 25
|
|
index =0
|
|
for mod in defs['menuOptions']:
|
|
eid = mod['id']
|
|
if 'feature' in mod:
|
|
f.write(to_unicode("#ifdef " + mod['feature'] + "\n"))
|
|
f.write(to_unicode(" /* ["+"{:02d}".format(index)+"] " + eid.ljust(maxLen)[:maxLen] + " */ "))
|
|
f.write(
|
|
to_unicode("\"" +
|
|
convStr(symbolConversionTable, (obj[eid]['desc'])) +
|
|
"\"," + "//{} \n".format(obj[eid]['desc'])))
|
|
if 'feature' in mod:
|
|
f.write(to_unicode("#endif\n"))
|
|
index=index+1
|
|
|
|
f.write(to_unicode("};\n\n"))
|
|
|
|
# ----- Writing Message strings
|
|
|
|
obj = lang['messages']
|
|
|
|
for mod in defs['messages']:
|
|
eid = mod['id']
|
|
sourceText = ""
|
|
if 'default' in mod:
|
|
sourceText = (mod['default'])
|
|
if eid in obj:
|
|
sourceText = (obj[eid])
|
|
translatedText = convStr(symbolConversionTable, sourceText)
|
|
f.write(
|
|
to_unicode("const char* " + eid + " = \"" +
|
|
translatedText + "\";" + "//{} \n".format(sourceText.replace('\n', '_'))))
|
|
|
|
f.write(to_unicode("\n"))
|
|
|
|
# ----- Writing Characters
|
|
|
|
obj = lang['characters']
|
|
|
|
for mod in defs['characters']:
|
|
eid = mod['id']
|
|
f.write(
|
|
to_unicode("const char* " + eid + " = \"" +
|
|
convStr(symbolConversionTable, obj[eid]) + "\";" + "//{} \n".format(obj[eid])))
|
|
|
|
f.write(to_unicode("\n"))
|
|
|
|
# Write out firmware constant options
|
|
constants = getConstants()
|
|
for x in constants:
|
|
f.write(
|
|
to_unicode("const char* " + x[0] + " = \"" +
|
|
convStr(symbolConversionTable, x[1]) + "\";" + "//{} \n".format(x[1])))
|
|
|
|
f.write(to_unicode("\n"))
|
|
|
|
# Debug Menu
|
|
f.write(to_unicode("const char* DebugMenu[] = {\n"))
|
|
|
|
for c in getDebugMenu():
|
|
f.write(to_unicode("\t \"" + convStr(symbolConversionTable,
|
|
c) + "\"," + "//{} \n".format(c)))
|
|
f.write(to_unicode("};\n\n"))
|
|
|
|
# ----- Writing SettingsDescriptions
|
|
obj = lang['menuOptions']
|
|
f.write(to_unicode("const char* SettingsShortNames[][2] = {\n"))
|
|
|
|
maxLen = 25
|
|
index = 0
|
|
for mod in defs['menuOptions']:
|
|
eid = mod['id']
|
|
if 'feature' in mod:
|
|
f.write(to_unicode("#ifdef " + mod['feature'] + "\n"))
|
|
f.write(to_unicode(" /* ["+"{:02d}".format(index)+"] " + eid.ljust(maxLen)[:maxLen] + " */ "))
|
|
f.write(
|
|
to_unicode(
|
|
"{ \"" +
|
|
convStr(symbolConversionTable, (obj[eid]['text2'][0])) +
|
|
"\", \"" +
|
|
convStr(symbolConversionTable, (obj[eid]['text2'][1])) +
|
|
"\" }," + "//{} \n".format(obj[eid]['text2'])))
|
|
|
|
if 'feature' in mod:
|
|
f.write(to_unicode("#endif\n"))
|
|
index = index + 1
|
|
|
|
f.write(to_unicode("};\n\n"))
|
|
|
|
# ----- Writing Menu Groups
|
|
obj = lang['menuGroups']
|
|
f.write(
|
|
to_unicode("const char* SettingsMenuEntries[" + str(len(obj)) +
|
|
"] = {\n"))
|
|
|
|
maxLen = 25
|
|
for mod in defs['menuGroups']:
|
|
eid = mod['id']
|
|
f.write(to_unicode(" /* " + eid.ljust(maxLen)[:maxLen] + " */ "))
|
|
f.write(
|
|
to_unicode("\"" +
|
|
convStr(symbolConversionTable, (obj[eid]['text2'][0]) +
|
|
"\\n" + obj[eid]['text2'][1]) + "\"," + "//{} \n".format(obj[eid]['text2'])))
|
|
|
|
f.write(to_unicode("};\n\n"))
|
|
|
|
# ----- Writing Menu Groups Descriptions
|
|
obj = lang['menuGroups']
|
|
f.write(
|
|
to_unicode("const char* SettingsMenuEntriesDescriptions[" +
|
|
str(len(obj)) + "] = {\n"))
|
|
|
|
maxLen = 25
|
|
for mod in defs['menuGroups']:
|
|
eid = mod['id']
|
|
f.write(to_unicode(" /* " + eid.ljust(maxLen)[:maxLen] + " */ "))
|
|
f.write(
|
|
to_unicode("\"" +
|
|
convStr(symbolConversionTable, (obj[eid]['desc'])) +
|
|
"\"," + "//{} \n".format(obj[eid]['desc'])))
|
|
|
|
f.write(to_unicode("};\n\n"))
|
|
|
|
# ----- Block end
|
|
f.write(to_unicode("#endif\n"))
|
|
|
|
|
|
def writeUnit(languageCode, defs, f, UnitCodes):
|
|
print("Generating unit block for " + languageCode)
|
|
lang = langDict[languageCode]
|
|
unit = UnitDict[UnitCodes]
|
|
try:
|
|
langName = lang['languageLocalName']
|
|
except KeyError:
|
|
langName = languageCode
|
|
f.write(to_unicode(" #ifdef LANG_" + languageCode + "\n"))
|
|
if unit:
|
|
f.write(to_unicode(" #define ENABLED_FAHRENHEIT_SUPPORT" + "\n"))
|
|
else: f.write(to_unicode(" //#define ENABLED_FAHRENHEIT_SUPPORT" + "\n"))
|
|
# ----- Block end
|
|
f.write(to_unicode(" #endif /* ---- " + langName + " ---- */\n"))
|
|
|
|
def readVersion():
|
|
with open(os.path.relpath(jsonDir +
|
|
"/../workspace/TS100/version.h"),"r") as version_file:
|
|
try:
|
|
for line in version_file:
|
|
if re.findall(r'^.*(?<=(#define)).*(?<=(BUILD_VERSION))', line):
|
|
line = re.findall(r'\"(.+?)\"',line)
|
|
if line:
|
|
version = line[0]
|
|
try: version += "."+ subprocess.check_output(
|
|
["git","rev-parse", "--short=7", "HEAD"]).strip().decode('ascii').upper()
|
|
# --short=7: the shorted hash with 7 digits. Increase/decrease if needed!
|
|
except OSError: version += " git"
|
|
finally:
|
|
if version_file:
|
|
version_file.close();
|
|
return version
|
|
|
|
def read_opts():
|
|
""" Reading input parameters
|
|
First parameter = json directory
|
|
Second parameter = translation directory
|
|
Third paramter = unit directory
|
|
"""
|
|
if len(sys.argv) > 1:
|
|
jsonDir = sys.argv[1]
|
|
else:
|
|
jsonDir = "."
|
|
|
|
if len(sys.argv) > 2:
|
|
outFileTranslationCPP = sys.argv[2]
|
|
else:
|
|
outDir = os.path.relpath(jsonDir + "/../workspace/TS100/Core/Src")
|
|
outFileTranslationCPP = os.path.join(outDir, TRANSLATION_CPP)
|
|
|
|
if len(sys.argv) > 3:
|
|
outFileUnitH = sys.argv[3]
|
|
else:
|
|
outDir = os.path.relpath(jsonDir + "/../workspace/TS100/Core/Inc")
|
|
outFileUnitH = os.path.join(outDir,UNIT_H)
|
|
|
|
if len(sys.argv) > 4:
|
|
raise Exception("Too many parameters!")
|
|
|
|
return jsonDir, outFileTranslationCPP, outFileUnitH
|
|
|
|
|
|
def orderOutput(langDict):
|
|
# These languages go first
|
|
mandatoryOrder = ['EN']
|
|
|
|
# Then add all others in alphabetical order
|
|
sortedKeys = sorted(langDict.keys())
|
|
|
|
# Add the rest as they come
|
|
for key in sortedKeys:
|
|
if key not in mandatoryOrder:
|
|
mandatoryOrder.append(key)
|
|
|
|
return mandatoryOrder
|
|
|
|
|
|
def writeTarget(outFileTranslationCPP, outFileUnitH, defs, langCodes, UnitCodes):
|
|
# Start writing the file
|
|
with io.open(outFileTranslationCPP, 'w', encoding='utf-8', newline="\n") as f:
|
|
writeStart(f)
|
|
for langCode in langCodes:
|
|
writeLanguage(langCode, defs, f)
|
|
|
|
with io.open(outFileUnitH, 'w', encoding='utf-8', newline="\n") as f:
|
|
writeStartUnit(f)
|
|
for langCode, UnitCode in zip(langCodes, UnitCodes):
|
|
writeUnit(langCode, defs, f, UnitCode)
|
|
f.write(to_unicode("\n#endif /* _UNIT_H */\n"))
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
jsonDir, outFileTranslationCPP, outFileUnitH = read_opts()
|
|
except:
|
|
print("usage: make_translation.py {json dir} {cpp dir}")
|
|
sys.exit(1)
|
|
|
|
try: buildVersion = readVersion()
|
|
except: print("error: could not get/extract build version"); sys.exit(1)
|
|
|
|
print("Build version: " + buildVersion)
|
|
print("Making " + outFileTranslationCPP + " from " + jsonDir)
|
|
print("Making " + outFileUnitH + " from " + jsonDir)
|
|
|
|
langDict, UnitDict = readTranslations(jsonDir)
|
|
defs = loadJson(os.path.join(jsonDir, "translations_def.js"), True)
|
|
langCodes = orderOutput(langDict)
|
|
UnitCodes = orderOutput(UnitDict)
|
|
writeTarget(outFileTranslationCPP, outFileUnitH, defs, langCodes, UnitCodes)
|
|
|
|
print("Done")
|