KLL Compiler
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
This repo is archived. You can view files and clone it, but cannot push or open issues/pull-requests.

kll.py 27KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806
  1. #!/usr/bin/env python3
  2. '''
  3. KLL Compiler
  4. Keyboard Layout Langauge
  5. '''
  6. # Copyright (C) 2014-2016 by Jacob Alexander
  7. #
  8. # This file is free software: you can redistribute it and/or modify
  9. # it under the terms of the GNU General Public License as published by
  10. # the Free Software Foundation, either version 3 of the License, or
  11. # (at your option) any later version.
  12. #
  13. # This file is distributed in the hope that it will be useful,
  14. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  16. # GNU General Public License for more details.
  17. #
  18. # You should have received a copy of the GNU General Public License
  19. # along with this file. If not, see <http://www.gnu.org/licenses/>.
  20. ### Imports ###
  21. import argparse
  22. import importlib
  23. import os
  24. import sys
  25. from re import VERBOSE
  26. from kll_lib.containers import *
  27. from kll_lib.hid_dict import *
  28. from funcparserlib.lexer import make_tokenizer, Token, LexerError
  29. from funcparserlib.parser import (some, a, many, oneplus, finished, maybe, skip, NoParseError)
  30. ### Decorators ###
  31. ## Print Decorator Variables
  32. ERROR = '\033[5;1;31mERROR\033[0m:'
  33. ## Python Text Formatting Fixer...
  34. ## Because the creators of Python are averse to proper capitalization.
  35. textFormatter_lookup = {
  36. "usage: " : "Usage: ",
  37. "optional arguments" : "Optional Arguments",
  38. }
  39. def textFormatter_gettext( s ):
  40. return textFormatter_lookup.get( s, s )
  41. argparse._ = textFormatter_gettext
  42. ### Argument Parsing ###
  43. def checkFileExists( filename ):
  44. if not os.path.isfile( filename ):
  45. print ( "{0} {1} does not exist...".format( ERROR, filename ) )
  46. sys.exit( 1 )
  47. def processCommandLineArgs():
  48. # Setup argument processor
  49. pArgs = argparse.ArgumentParser(
  50. usage="%(prog)s [options] <file1>...",
  51. description="Generates .h file state tables and pointer indices from KLL .kll files.",
  52. epilog="Example: {0} mykeyboard.kll -d colemak.kll -p hhkbpro2.kll -p symbols.kll".format( os.path.basename( sys.argv[0] ) ),
  53. formatter_class=argparse.RawTextHelpFormatter,
  54. add_help=False,
  55. )
  56. # Positional Arguments
  57. pArgs.add_argument( 'files', type=str, nargs='+',
  58. help=argparse.SUPPRESS ) # Suppressed help output, because Python output is verbosely ugly
  59. # Optional Arguments
  60. pArgs.add_argument( '-b', '--backend', type=str, default="kiibohd",
  61. help="Specify target backend for the KLL compiler.\n"
  62. "Default: kiibohd\n"
  63. "Options: kiibohd, json" )
  64. pArgs.add_argument( '-d', '--default', type=str, nargs='+',
  65. help="Specify .kll files to layer on top of the default map to create a combined map." )
  66. pArgs.add_argument( '-p', '--partial', type=str, nargs='+', action='append',
  67. help="Specify .kll files to generate partial map, multiple files per flag.\n"
  68. "Each -p defines another partial map.\n"
  69. "Base .kll files (that define the scan code maps) must be defined for each partial map." )
  70. pArgs.add_argument( '-t', '--templates', type=str, nargs='+',
  71. help="Specify template used to generate the keymap.\n"
  72. "Default: <backend specific>" )
  73. pArgs.add_argument( '-o', '--outputs', type=str, nargs='+',
  74. help="Specify output file. Writes to current working directory by default.\n"
  75. "Default: <backend specific>" )
  76. pArgs.add_argument( '-h', '--help', action="help",
  77. help="This message." )
  78. pArgs.add_argument(
  79. '-v', '--version',
  80. action="version",
  81. version="%(prog)s {0}".format( version ),
  82. help="Show program's version number and exit"
  83. )
  84. # Process Arguments
  85. args = pArgs.parse_args()
  86. # Parameters
  87. baseFiles = args.files
  88. defaultFiles = args.default
  89. partialFileSets = args.partial
  90. if defaultFiles is None:
  91. defaultFiles = []
  92. if partialFileSets is None:
  93. partialFileSets = [[]]
  94. # Check file existance
  95. for filename in baseFiles:
  96. checkFileExists( filename )
  97. for filename in defaultFiles:
  98. checkFileExists( filename )
  99. for partial in partialFileSets:
  100. for filename in partial:
  101. checkFileExists( filename )
  102. return (baseFiles, defaultFiles, partialFileSets, args.backend, args.templates, args.outputs)
  103. ### Tokenizer ###
  104. def tokenize( string ):
  105. """str -> Sequence(Token)"""
  106. # Basic Tokens Spec
  107. specs = [
  108. ( 'Comment', ( r' *#.*', ) ),
  109. ( 'Space', ( r'[ \t\r\n]+', ) ),
  110. ( 'USBCode', ( r'U(("[^"]+")|(0x[0-9a-fA-F]+)|([0-9]+))', ) ),
  111. ( 'USBCodeStart', ( r'U\[', ) ),
  112. ( 'ConsCode', ( r'CONS(("[^"]+")|(0x[0-9a-fA-F]+)|([0-9]+))', ) ),
  113. ( 'ConsCodeStart', ( r'CONS\[', ) ),
  114. ( 'SysCode', ( r'SYS(("[^"]+")|(0x[0-9a-fA-F]+)|([0-9]+))', ) ),
  115. ( 'SysCodeStart', ( r'SYS\[', ) ),
  116. ( 'LedCode', ( r'LED(("[^"]+")|(0x[0-9a-fA-F]+)|([0-9]+))', ) ),
  117. ( 'LedCodeStart', ( r'LED\[', ) ),
  118. ( 'ScanCode', ( r'S((0x[0-9a-fA-F]+)|([0-9]+))', ) ),
  119. ( 'ScanCodeStart', ( r'S\[', ) ),
  120. ( 'CodeEnd', ( r'\]', ) ),
  121. ( 'String', ( r'"[^"]*"', ) ),
  122. ( 'SequenceString', ( r"'[^']*'", ) ),
  123. ( 'Operator', ( r'=>|:\+|:-|::|:|=', ) ),
  124. ( 'Number', ( r'(-[ \t]*)?((0x[0-9a-fA-F]+)|(0|([1-9][0-9]*)))', VERBOSE ) ),
  125. ( 'Comma', ( r',', ) ),
  126. ( 'Dash', ( r'-', ) ),
  127. ( 'Plus', ( r'\+', ) ),
  128. ( 'Parenthesis', ( r'\(|\)', ) ),
  129. ( 'None', ( r'None', ) ),
  130. ( 'Name', ( r'[A-Za-z_][A-Za-z_0-9]*', ) ),
  131. ( 'VariableContents', ( r'''[^"' ;:=>()]+''', ) ),
  132. ( 'EndOfLine', ( r';', ) ),
  133. ]
  134. # Tokens to filter out of the token stream
  135. useless = ['Space', 'Comment']
  136. tokens = make_tokenizer( specs )
  137. return [x for x in tokens( string ) if x.type not in useless]
  138. ### Parsing ###
  139. ## Map Arrays
  140. macros_map = Macros()
  141. variables_dict = Variables()
  142. capabilities_dict = Capabilities()
  143. ## Parsing Functions
  144. def make_scanCode( token ):
  145. scanCode = int( token[1:], 0 )
  146. # Check size, to make sure it's valid
  147. # XXX Add better check that takes symbolic names into account (i.e. U"Latch5")
  148. #if scanCode > 0xFF:
  149. # print ( "{0} ScanCode value {1} is larger than 255".format( ERROR, scanCode ) )
  150. # raise
  151. return scanCode
  152. def make_hidCode( type, token ):
  153. # If first character is a U, strip
  154. if token[0] == "U":
  155. token = token[1:]
  156. # CONS specifier
  157. elif 'CONS' in token:
  158. token = token[4:]
  159. # SYS specifier
  160. elif 'SYS' in token:
  161. token = token[3:]
  162. # If using string representation of USB Code, do lookup, case-insensitive
  163. if '"' in token:
  164. try:
  165. hidCode = kll_hid_lookup_dictionary[ type ][ token[1:-1].upper() ][1]
  166. except LookupError as err:
  167. print ( "{0} {1} is an invalid USB HID Code Lookup...".format( ERROR, err ) )
  168. raise
  169. else:
  170. # Already tokenized
  171. if type == 'USBCode' and token[0] == 'USB' or type == 'SysCode' and token[0] == 'SYS' or type == 'ConsCode' and token[0] == 'CONS':
  172. hidCode = token[1]
  173. # Convert
  174. else:
  175. hidCode = int( token, 0 )
  176. # Check size if a USB Code, to make sure it's valid
  177. # XXX Add better check that takes symbolic names into account (i.e. U"Latch5")
  178. #if type == 'USBCode' and hidCode > 0xFF:
  179. # print ( "{0} USBCode value {1} is larger than 255".format( ERROR, hidCode ) )
  180. # raise
  181. # Return a tuple, identifying which type it is
  182. if type == 'USBCode':
  183. return make_usbCode_number( hidCode )
  184. elif type == 'ConsCode':
  185. return make_consCode_number( hidCode )
  186. elif type == 'SysCode':
  187. return make_sysCode_number( hidCode )
  188. print ( "{0} Unknown HID Specifier '{1}'".format( ERROR, type ) )
  189. raise
  190. def make_usbCode( token ):
  191. return make_hidCode( 'USBCode', token )
  192. def make_consCode( token ):
  193. return make_hidCode( 'ConsCode', token )
  194. def make_sysCode( token ):
  195. return make_hidCode( 'SysCode', token )
  196. def make_hidCode_number( type, token ):
  197. lookup = {
  198. 'ConsCode' : 'CONS',
  199. 'SysCode' : 'SYS',
  200. 'USBCode' : 'USB',
  201. }
  202. return ( lookup[ type ], token )
  203. def make_usbCode_number( token ):
  204. return make_hidCode_number( 'USBCode', token )
  205. def make_consCode_number( token ):
  206. return make_hidCode_number( 'ConsCode', token )
  207. def make_sysCode_number( token ):
  208. return make_hidCode_number( 'SysCode', token )
  209. # Replace key-word with None specifier (which indicates a noneOut capability)
  210. def make_none( token ):
  211. return [[[('NONE', 0)]]]
  212. def make_seqString( token ):
  213. # Shifted Characters, and amount to move by to get non-shifted version
  214. # US ANSI
  215. shiftCharacters = (
  216. ( "ABCDEFGHIJKLMNOPQRSTUVWXYZ", 0x20 ),
  217. ( "+", 0x12 ),
  218. ( "&(", 0x11 ),
  219. ( "!#$%", 0x10 ),
  220. ( "*", 0x0E ),
  221. ( ")", 0x07 ),
  222. ( '"', 0x05 ),
  223. ( ":", 0x01 ),
  224. ( "@", -0x0E ),
  225. ( "<>?", -0x10 ),
  226. ( "~", -0x1E ),
  227. ( "{}|", -0x20 ),
  228. ( "^", -0x28 ),
  229. ( "_", -0x32 ),
  230. )
  231. listOfLists = []
  232. shiftKey = kll_hid_lookup_dictionary['USBCode']["SHIFT"]
  233. # Creates a list of USB codes from the string: sequence (list) of combos (lists)
  234. for char in token[1:-1]:
  235. processedChar = char
  236. # Whether or not to create a combo for this sequence with a shift
  237. shiftCombo = False
  238. # Depending on the ASCII character, convert to single character or Shift + character
  239. for pair in shiftCharacters:
  240. if char in pair[0]:
  241. shiftCombo = True
  242. processedChar = chr( ord( char ) + pair[1] )
  243. break
  244. # Do KLL HID Lookup on non-shifted character
  245. # NOTE: Case-insensitive, which is why the shift must be pre-computed
  246. usbCode = kll_hid_lookup_dictionary['USBCode'][ processedChar.upper() ]
  247. # Create Combo for this character, add shift key if shifted
  248. charCombo = []
  249. if shiftCombo:
  250. charCombo = [ [ shiftKey ] ]
  251. charCombo.append( [ usbCode ] )
  252. # Add to list of lists
  253. listOfLists.append( charCombo )
  254. return listOfLists
  255. def make_string( token ):
  256. return token[1:-1]
  257. def make_unseqString( token ):
  258. return token[1:-1]
  259. def make_number( token ):
  260. return int( token, 0 )
  261. # Range can go from high to low or low to high
  262. def make_scanCode_range( rangeVals ):
  263. start = rangeVals[0]
  264. end = rangeVals[1]
  265. # Swap start, end if start is greater than end
  266. if start > end:
  267. start, end = end, start
  268. # Iterate from start to end, and generate the range
  269. return list( range( start, end + 1 ) )
  270. # Range can go from high to low or low to high
  271. # Warn on 0-9 for USBCodes (as this does not do what one would expect) TODO
  272. # Lookup USB HID tags and convert to a number
  273. def make_hidCode_range( type, rangeVals ):
  274. # Check if already integers
  275. if isinstance( rangeVals[0], int ):
  276. start = rangeVals[0]
  277. else:
  278. start = make_hidCode( type, rangeVals[0] )[1]
  279. if isinstance( rangeVals[1], int ):
  280. end = rangeVals[1]
  281. else:
  282. end = make_hidCode( type, rangeVals[1] )[1]
  283. # Swap start, end if start is greater than end
  284. if start > end:
  285. start, end = end, start
  286. # Iterate from start to end, and generate the range
  287. listRange = list( range( start, end + 1 ) )
  288. # Convert each item in the list to a tuple
  289. for item in range( len( listRange ) ):
  290. listRange[ item ] = make_hidCode_number( type, listRange[ item ] )
  291. return listRange
  292. def make_usbCode_range( rangeVals ):
  293. return make_hidCode_range( 'USBCode', rangeVals )
  294. def make_sysCode_range( rangeVals ):
  295. return make_hidCode_range( 'SysCode', rangeVals )
  296. def make_consCode_range( rangeVals ):
  297. return make_hidCode_range( 'ConsCode', rangeVals )
  298. ## Base Rules
  299. const = lambda x: lambda _: x
  300. unarg = lambda f: lambda x: f(*x)
  301. flatten = lambda list: sum( list, [] )
  302. tokenValue = lambda x: x.value
  303. tokenType = lambda t: some( lambda x: x.type == t ) >> tokenValue
  304. operator = lambda s: a( Token( 'Operator', s ) ) >> tokenValue
  305. parenthesis = lambda s: a( Token( 'Parenthesis', s ) ) >> tokenValue
  306. eol = a( Token( 'EndOfLine', ';' ) )
  307. def listElem( item ):
  308. return [ item ]
  309. def listToTuple( items ):
  310. return tuple( items )
  311. # Flatten only the top layer (list of lists of ...)
  312. def oneLayerFlatten( items ):
  313. mainList = []
  314. for sublist in items:
  315. for item in sublist:
  316. mainList.append( item )
  317. return mainList
  318. def capArgExpander( items ):
  319. '''
  320. Capability arguments may need to be expanded
  321. (e.g. 1 16 bit argument needs to be 2 8 bit arguments for the state machine)
  322. If the number is negative, determine width of the final value, mask to max, subtract,
  323. then convert to multiple bytes
  324. '''
  325. newArgs = []
  326. # For each defined argument in the capability definition
  327. for arg in range( 0, len( capabilities_dict[ items[0] ][1] ) ):
  328. argLen = capabilities_dict[ items[0] ][1][ arg ][1]
  329. num = items[1][ arg ]
  330. # Set last bit if value is negative
  331. if num < 0:
  332. max_val = 2 ** (argLen * 8)
  333. num += max_val
  334. # XXX Yes, little endian from how the uC structs work
  335. byteForm = num.to_bytes( argLen, byteorder='little' )
  336. # For each sub-argument, split into byte-sized chunks
  337. for byte in range( 0, argLen ):
  338. newArgs.append( byteForm[ byte ] )
  339. return tuple( [ items[0], tuple( newArgs ) ] )
  340. # Expand ranges of values in the 3rd dimension of the list, to a list of 2nd lists
  341. # i.e. [ sequence, [ combo, [ range ] ] ] --> [ [ sequence, [ combo ] ], <option 2>, <option 3> ]
  342. def optionExpansion( sequences ):
  343. expandedSequences = []
  344. # Total number of combinations of the sequence of combos that needs to be generated
  345. totalCombinations = 1
  346. # List of leaf lists, with number of leaves
  347. maxLeafList = []
  348. # Traverse to the leaf nodes, and count the items in each leaf list
  349. for sequence in sequences:
  350. for combo in sequence:
  351. rangeLen = len( combo )
  352. totalCombinations *= rangeLen
  353. maxLeafList.append( rangeLen )
  354. # Counter list to keep track of which combination is being generated
  355. curLeafList = [0] * len( maxLeafList )
  356. # Generate a list of permuations of the sequence of combos
  357. for count in range( 0, totalCombinations ):
  358. expandedSequences.append( [] ) # Prepare list for adding the new combination
  359. position = 0
  360. # Traverse sequence of combos to generate permuation
  361. for sequence in sequences:
  362. expandedSequences[ -1 ].append( [] )
  363. for combo in sequence:
  364. expandedSequences[ -1 ][ -1 ].append( combo[ curLeafList[ position ] ] )
  365. position += 1
  366. # Increment combination tracker
  367. for leaf in range( 0, len( curLeafList ) ):
  368. curLeafList[ leaf ] += 1
  369. # Reset this position, increment next position (if it exists), then stop
  370. if curLeafList[ leaf ] >= maxLeafList[ leaf ]:
  371. curLeafList[ leaf ] = 0
  372. if leaf + 1 < len( curLeafList ):
  373. curLeafList[ leaf + 1 ] += 1
  374. return expandedSequences
  375. # Converts USB Codes into Capabilities
  376. # These are tuples (<type>, <integer>)
  377. def hidCodeToCapability( items ):
  378. # Items already converted to variants using optionExpansion
  379. for variant in range( 0, len( items ) ):
  380. # Sequence of Combos
  381. for sequence in range( 0, len( items[ variant ] ) ):
  382. for combo in range( 0, len( items[ variant ][ sequence ] ) ):
  383. if items[ variant ][ sequence ][ combo ][0] in backend.requiredCapabilities.keys():
  384. try:
  385. # Use backend capability name and a single argument
  386. items[ variant ][ sequence ][ combo ] = tuple(
  387. [ backend.capabilityLookup( items[ variant ][ sequence ][ combo ][0] ),
  388. tuple( [ hid_lookup_dictionary[ items[ variant ][ sequence ][ combo ] ] ] ) ]
  389. )
  390. except KeyError:
  391. print ( "{0} {1} is an invalid HID lookup value".format( ERROR, items[ variant ][ sequence ][ combo ] ) )
  392. sys.exit( 1 )
  393. return items
  394. # Convert tuple of tuples to list of lists
  395. def listit( t ):
  396. return list( map( listit, t ) ) if isinstance( t, ( list, tuple ) ) else t
  397. # Convert list of lists to tuple of tuples
  398. def tupleit( t ):
  399. return tuple( map( tupleit, t ) ) if isinstance( t, ( tuple, list ) ) else t
  400. ## Evaluation Rules
  401. def eval_scanCode( triggers, operator, results ):
  402. # Convert to lists of lists of lists to tuples of tuples of tuples
  403. # Tuples are non-mutable, and can be used has index items
  404. triggers = tuple( tuple( tuple( sequence ) for sequence in variant ) for variant in triggers )
  405. results = tuple( tuple( tuple( sequence ) for sequence in variant ) for variant in results )
  406. # Lookup interconnect id (Current file scope)
  407. # Default to 0 if not specified
  408. if 'ConnectId' not in variables_dict.overallVariables.keys():
  409. id_num = 0
  410. else:
  411. id_num = int( variables_dict.overallVariables['ConnectId'] )
  412. # Iterate over all combinations of triggers and results
  413. for sequence in triggers:
  414. # Convert tuple of tuples to list of lists so each element can be modified
  415. trigger = listit( sequence )
  416. # Create ScanCode entries for trigger
  417. for seq_index, combo in enumerate( sequence ):
  418. for com_index, scancode in enumerate( combo ):
  419. trigger[ seq_index ][ com_index ] = macros_map.scanCodeStore.append( ScanCode( scancode, id_num ) )
  420. # Convert back to a tuple of tuples
  421. trigger = tupleit( trigger )
  422. for result in results:
  423. # Append Case
  424. if operator == ":+":
  425. macros_map.appendScanCode( trigger, result )
  426. # Remove Case
  427. elif operator == ":-":
  428. macros_map.removeScanCode( trigger, result )
  429. # Replace Case
  430. # Soft Replace Case is the same for Scan Codes
  431. elif operator == ":" or operator == "::":
  432. macros_map.replaceScanCode( trigger, result )
  433. def eval_usbCode( triggers, operator, results ):
  434. # Convert to lists of lists of lists to tuples of tuples of tuples
  435. # Tuples are non-mutable, and can be used has index items
  436. triggers = tuple( tuple( tuple( sequence ) for sequence in variant ) for variant in triggers )
  437. results = tuple( tuple( tuple( sequence ) for sequence in variant ) for variant in results )
  438. # Iterate over all combinations of triggers and results
  439. for trigger in triggers:
  440. scanCodes = macros_map.lookupUSBCodes( trigger )
  441. for scanCode in scanCodes:
  442. for result in results:
  443. # Soft Replace needs additional checking to see if replacement is necessary
  444. if operator == "::" and not macros_map.softReplaceCheck( scanCode ):
  445. continue
  446. # Cache assignment until file finishes processing
  447. macros_map.cacheAssignment( operator, scanCode, result )
  448. def eval_variable( name, content ):
  449. # Content might be a concatenation of multiple data types, convert everything into a single string
  450. assigned_content = ""
  451. for item in content:
  452. assigned_content += str( item )
  453. variables_dict.assignVariable( name, assigned_content )
  454. def eval_capability( name, function, args ):
  455. capabilities_dict[ name ] = [ function, args ]
  456. def eval_define( name, cdefine_name ):
  457. variables_dict.defines[ name ] = cdefine_name
  458. map_scanCode = unarg( eval_scanCode )
  459. map_usbCode = unarg( eval_usbCode )
  460. set_variable = unarg( eval_variable )
  461. set_capability = unarg( eval_capability )
  462. set_define = unarg( eval_define )
  463. ## Sub Rules
  464. usbCode = tokenType('USBCode') >> make_usbCode
  465. scanCode = tokenType('ScanCode') >> make_scanCode
  466. consCode = tokenType('ConsCode') >> make_consCode
  467. sysCode = tokenType('SysCode') >> make_sysCode
  468. none = tokenType('None') >> make_none
  469. name = tokenType('Name')
  470. number = tokenType('Number') >> make_number
  471. comma = tokenType('Comma')
  472. dash = tokenType('Dash')
  473. plus = tokenType('Plus')
  474. content = tokenType('VariableContents')
  475. string = tokenType('String') >> make_string
  476. unString = tokenType('String') # When the double quotes are still needed for internal processing
  477. seqString = tokenType('SequenceString') >> make_seqString
  478. unseqString = tokenType('SequenceString') >> make_unseqString # For use with variables
  479. # Code variants
  480. code_end = tokenType('CodeEnd')
  481. # Scan Codes
  482. scanCode_start = tokenType('ScanCodeStart')
  483. scanCode_range = number + skip( dash ) + number >> make_scanCode_range
  484. scanCode_listElem = number >> listElem
  485. scanCode_innerList = oneplus( ( scanCode_range | scanCode_listElem ) + skip( maybe( comma ) ) ) >> flatten
  486. scanCode_expanded = skip( scanCode_start ) + scanCode_innerList + skip( code_end )
  487. scanCode_elem = scanCode >> listElem
  488. scanCode_combo = oneplus( ( scanCode_expanded | scanCode_elem ) + skip( maybe( plus ) ) )
  489. scanCode_sequence = oneplus( scanCode_combo + skip( maybe( comma ) ) )
  490. # USB Codes
  491. usbCode_start = tokenType('USBCodeStart')
  492. usbCode_number = number >> make_usbCode_number
  493. usbCode_range = ( usbCode_number | unString ) + skip( dash ) + ( number | unString ) >> make_usbCode_range
  494. usbCode_listElemTag = unString >> make_usbCode
  495. usbCode_listElem = ( usbCode_number | usbCode_listElemTag ) >> listElem
  496. usbCode_innerList = oneplus( ( usbCode_range | usbCode_listElem ) + skip( maybe( comma ) ) ) >> flatten
  497. usbCode_expanded = skip( usbCode_start ) + usbCode_innerList + skip( code_end )
  498. usbCode_elem = usbCode >> listElem
  499. usbCode_combo = oneplus( ( usbCode_expanded | usbCode_elem ) + skip( maybe( plus ) ) ) >> listElem
  500. usbCode_sequence = oneplus( ( usbCode_combo | seqString ) + skip( maybe( comma ) ) ) >> oneLayerFlatten
  501. # Cons Codes
  502. consCode_start = tokenType('ConsCodeStart')
  503. consCode_number = number >> make_consCode_number
  504. consCode_range = ( consCode_number | unString ) + skip( dash ) + ( number | unString ) >> make_consCode_range
  505. consCode_listElemTag = unString >> make_consCode
  506. consCode_listElem = ( consCode_number | consCode_listElemTag ) >> listElem
  507. consCode_innerList = oneplus( ( consCode_range | consCode_listElem ) + skip( maybe( comma ) ) ) >> flatten
  508. consCode_expanded = skip( consCode_start ) + consCode_innerList + skip( code_end )
  509. consCode_elem = consCode >> listElem
  510. # Sys Codes
  511. sysCode_start = tokenType('SysCodeStart')
  512. sysCode_number = number >> make_sysCode_number
  513. sysCode_range = ( sysCode_number | unString ) + skip( dash ) + ( number | unString ) >> make_sysCode_range
  514. sysCode_listElemTag = unString >> make_sysCode
  515. sysCode_listElem = ( sysCode_number | sysCode_listElemTag ) >> listElem
  516. sysCode_innerList = oneplus( ( sysCode_range | sysCode_listElem ) + skip( maybe( comma ) ) ) >> flatten
  517. sysCode_expanded = skip( sysCode_start ) + sysCode_innerList + skip( code_end )
  518. sysCode_elem = sysCode >> listElem
  519. # HID Codes
  520. hidCode_elem = usbCode_expanded | usbCode_elem | sysCode_expanded | sysCode_elem | consCode_expanded | consCode_elem
  521. # Capabilities
  522. capFunc_arguments = many( number + skip( maybe( comma ) ) ) >> listToTuple
  523. capFunc_elem = name + skip( parenthesis('(') ) + capFunc_arguments + skip( parenthesis(')') ) >> capArgExpander >> listElem
  524. capFunc_combo = oneplus( ( hidCode_elem | capFunc_elem ) + skip( maybe( plus ) ) ) >> listElem
  525. capFunc_sequence = oneplus( ( capFunc_combo | seqString ) + skip( maybe( comma ) ) ) >> oneLayerFlatten
  526. # Trigger / Result Codes
  527. triggerCode_outerList = scanCode_sequence >> optionExpansion
  528. triggerUSBCode_outerList = usbCode_sequence >> optionExpansion >> hidCodeToCapability
  529. resultCode_outerList = ( ( capFunc_sequence >> optionExpansion ) | none ) >> hidCodeToCapability
  530. ## Main Rules
  531. #| <variable> = <variable contents>;
  532. variable_contents = name | content | string | number | comma | dash | unseqString
  533. variable_expression = name + skip( operator('=') ) + oneplus( variable_contents ) + skip( eol ) >> set_variable
  534. #| <capability name> => <c function>;
  535. capability_arguments = name + skip( operator(':') ) + number + skip( maybe( comma ) )
  536. capability_expression = name + skip( operator('=>') ) + name + skip( parenthesis('(') ) + many( capability_arguments ) + skip( parenthesis(')') ) + skip( eol ) >> set_capability
  537. #| <define name> => <c define>;
  538. define_expression = name + skip( operator('=>') ) + name + skip( eol ) >> set_define
  539. #| <trigger> : <result>;
  540. operatorTriggerResult = operator(':') | operator(':+') | operator(':-') | operator('::')
  541. scanCode_expression = triggerCode_outerList + operatorTriggerResult + resultCode_outerList + skip( eol ) >> map_scanCode
  542. usbCode_expression = triggerUSBCode_outerList + operatorTriggerResult + resultCode_outerList + skip( eol ) >> map_usbCode
  543. def parse( tokenSequence ):
  544. """Sequence(Token) -> object"""
  545. # Top-level Parser
  546. expression = scanCode_expression | usbCode_expression | variable_expression | capability_expression | define_expression
  547. kll_text = many( expression )
  548. kll_file = maybe( kll_text ) + skip( finished )
  549. return kll_file.parse( tokenSequence )
  550. def processKLLFile( filename ):
  551. with open( filename, encoding='utf-8' ) as file:
  552. data = file.read()
  553. try:
  554. tokenSequence = tokenize( data )
  555. except LexerError as err:
  556. print ( "{0} Tokenization error in '{1}' - {2}".format( ERROR, filename, err ) )
  557. sys.exit( 1 )
  558. #print ( pformat( tokenSequence ) ) # Display tokenization
  559. try:
  560. tree = parse( tokenSequence )
  561. except (NoParseError, KeyError) as err:
  562. print ( "{0} Parsing error in '{1}' - {2}".format( ERROR, filename, err ) )
  563. sys.exit( 1 )
  564. ### Misc Utility Functions ###
  565. def gitRevision( kllPath ):
  566. import subprocess
  567. # Change the path to where kll.py is
  568. origPath = os.getcwd()
  569. os.chdir( kllPath )
  570. # Just in case git can't be found
  571. try:
  572. # Get hash of the latest git commit
  573. revision = subprocess.check_output( ['git', 'rev-parse', 'HEAD'] ).decode()[:-1]
  574. # Get list of files that have changed since the commit
  575. changed = subprocess.check_output( ['git', 'diff-index', '--name-only', 'HEAD', '--'] ).decode().splitlines()
  576. # Get commit date
  577. date = subprocess.check_output( ['git', 'show', '-s', '--format=%ci'] ).decode()[:-1]
  578. except:
  579. revision = "<no git>"
  580. changed = []
  581. date = "<no date>"
  582. # Change back to the old working directory
  583. os.chdir( origPath )
  584. return revision, changed, date
  585. ### Main Entry Point ###
  586. if __name__ == '__main__':
  587. # Look up git information on the compiler
  588. gitRev, gitChanges, gitDate = gitRevision( os.path.dirname( os.path.realpath( __file__ ) ) )
  589. global version
  590. version = "BACKPORT 0.3d.{0} - {1}".format( gitRev, gitDate )
  591. (baseFiles, defaultFiles, partialFileSets, backend_name, templates, outputs) = processCommandLineArgs()
  592. # Load backend module
  593. global backend
  594. backend_import = importlib.import_module( "backends.{0}".format( backend_name ) )
  595. backend = backend_import.Backend( templates )
  596. # Process base layout files
  597. for filename in baseFiles:
  598. variables_dict.setCurrentFile( filename )
  599. processKLLFile( filename )
  600. macros_map.completeBaseLayout() # Indicates to macros_map that the base layout is complete
  601. variables_dict.baseLayoutFinished()
  602. # Default combined layer
  603. for filename in defaultFiles:
  604. variables_dict.setCurrentFile( filename )
  605. processKLLFile( filename )
  606. # Apply assignment cache, see 5.1.2 USB Codes for why this is necessary
  607. macros_map.replayCachedAssignments()
  608. # Iterate through additional layers
  609. for partial in partialFileSets:
  610. # Increment layer for each -p option
  611. macros_map.addLayer()
  612. variables_dict.incrementLayer() # DefaultLayer is layer 0
  613. # Iterate and process each of the file in the layer
  614. for filename in partial:
  615. variables_dict.setCurrentFile( filename )
  616. processKLLFile( filename )
  617. # Apply assignment cache, see 5.1.2 USB Codes for why this is necessary
  618. macros_map.replayCachedAssignments()
  619. # Remove un-marked keys to complete the partial layer
  620. macros_map.removeUnmarked()
  621. # Do macro correlation and transformation
  622. macros_map.generate()
  623. # Process needed templating variables using backend
  624. backend.process(
  625. capabilities_dict,
  626. macros_map,
  627. variables_dict,
  628. gitRev,
  629. gitChanges
  630. )
  631. # Generate output file using template and backend
  632. backend.generate( outputs )
  633. # Successful Execution
  634. sys.exit( 0 )