info.py 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908
  1. """Functions that help us generate and use info.json files.
  2. """
  3. from glob import glob
  4. from pathlib import Path
  5. import jsonschema
  6. from dotty_dict import dotty
  7. from milc import cli
  8. from qmk.constants import CHIBIOS_PROCESSORS, LUFA_PROCESSORS, VUSB_PROCESSORS
  9. from qmk.c_parse import find_layouts, parse_config_h_file, find_led_config
  10. from qmk.json_schema import deep_update, json_load, validate
  11. from qmk.keyboard import config_h, rules_mk
  12. from qmk.keymap import list_keymaps, locate_keymap
  13. from qmk.commands import parse_configurator_json
  14. from qmk.makefile import parse_rules_mk_file
  15. from qmk.math import compute
  16. true_values = ['1', 'on', 'yes']
  17. false_values = ['0', 'off', 'no']
  18. def _valid_community_layout(layout):
  19. """Validate that a declared community list exists
  20. """
  21. return (Path('layouts/default') / layout).exists()
  22. def info_json(keyboard):
  23. """Generate the info.json data for a specific keyboard.
  24. """
  25. cur_dir = Path('keyboards')
  26. root_rules_mk = parse_rules_mk_file(cur_dir / keyboard / 'rules.mk')
  27. if 'DEFAULT_FOLDER' in root_rules_mk:
  28. keyboard = root_rules_mk['DEFAULT_FOLDER']
  29. info_data = {
  30. 'keyboard_name': str(keyboard),
  31. 'keyboard_folder': str(keyboard),
  32. 'keymaps': {},
  33. 'layouts': {},
  34. 'parse_errors': [],
  35. 'parse_warnings': [],
  36. 'maintainer': 'qmk',
  37. }
  38. # Populate the list of JSON keymaps
  39. for keymap in list_keymaps(keyboard, c=False, fullpath=True):
  40. info_data['keymaps'][keymap.name] = {'url': f'https://raw.githubusercontent.com/qmk/qmk_firmware/master/{keymap}/keymap.json'}
  41. # Populate layout data
  42. layouts, aliases = _search_keyboard_h(keyboard)
  43. if aliases:
  44. info_data['layout_aliases'] = aliases
  45. for layout_name, layout_json in layouts.items():
  46. if not layout_name.startswith('LAYOUT_kc'):
  47. layout_json['c_macro'] = True
  48. info_data['layouts'][layout_name] = layout_json
  49. # Merge in the data from info.json, config.h, and rules.mk
  50. info_data = merge_info_jsons(keyboard, info_data)
  51. info_data = _process_defaults(info_data)
  52. info_data = _extract_rules_mk(info_data, rules_mk(str(keyboard)))
  53. info_data = _extract_config_h(info_data, config_h(str(keyboard)))
  54. # Ensure that we have matrix row and column counts
  55. info_data = _matrix_size(info_data)
  56. # Merge in data from <keyboard.c>
  57. info_data = _extract_led_config(info_data, str(keyboard))
  58. # Validate against the jsonschema
  59. try:
  60. validate(info_data, 'qmk.api.keyboard.v1')
  61. except jsonschema.ValidationError as e:
  62. json_path = '.'.join([str(p) for p in e.absolute_path])
  63. cli.log.error('Invalid API data: %s: %s: %s', keyboard, json_path, e.message)
  64. exit(1)
  65. # Make sure we have at least one layout
  66. if not info_data.get('layouts'):
  67. _find_missing_layouts(info_data, keyboard)
  68. if not info_data.get('layouts'):
  69. _log_error(info_data, 'No LAYOUTs defined! Need at least one layout defined in the keyboard.h or info.json.')
  70. # Filter out any non-existing community layouts
  71. for layout in info_data.get('community_layouts', []):
  72. if not _valid_community_layout(layout):
  73. # Ignore layout from future checks
  74. info_data['community_layouts'].remove(layout)
  75. _log_error(info_data, 'Claims to support a community layout that does not exist: %s' % (layout))
  76. # Make sure we supply layout macros for the community layouts we claim to support
  77. for layout in info_data.get('community_layouts', []):
  78. layout_name = 'LAYOUT_' + layout
  79. if layout_name not in info_data.get('layouts', {}) and layout_name not in info_data.get('layout_aliases', {}):
  80. _log_error(info_data, 'Claims to support community layout %s but no %s() macro found' % (layout, layout_name))
  81. # Check that the reported matrix size is consistent with the actual matrix size
  82. _check_matrix(info_data)
  83. return info_data
  84. def _extract_features(info_data, rules):
  85. """Find all the features enabled in rules.mk.
  86. """
  87. # Special handling for bootmagic which also supports a "lite" mode.
  88. if rules.get('BOOTMAGIC_ENABLE') == 'lite':
  89. rules['BOOTMAGIC_LITE_ENABLE'] = 'on'
  90. del rules['BOOTMAGIC_ENABLE']
  91. if rules.get('BOOTMAGIC_ENABLE') == 'full':
  92. rules['BOOTMAGIC_ENABLE'] = 'on'
  93. # Process the rest of the rules as booleans
  94. for key, value in rules.items():
  95. if key.endswith('_ENABLE'):
  96. key = '_'.join(key.split('_')[:-1]).lower()
  97. value = True if value.lower() in true_values else False if value.lower() in false_values else value
  98. if 'config_h_features' not in info_data:
  99. info_data['config_h_features'] = {}
  100. if 'features' not in info_data:
  101. info_data['features'] = {}
  102. if key in info_data['features']:
  103. _log_warning(info_data, 'Feature %s is specified in both info.json and rules.mk, the rules.mk value wins.' % (key,))
  104. info_data['features'][key] = value
  105. info_data['config_h_features'][key] = value
  106. return info_data
  107. def _pin_name(pin):
  108. """Returns the proper representation for a pin.
  109. """
  110. pin = pin.strip()
  111. if not pin:
  112. return None
  113. elif pin.isdigit():
  114. return int(pin)
  115. elif pin == 'NO_PIN':
  116. return None
  117. return pin
  118. def _extract_pins(pins):
  119. """Returns a list of pins from a comma separated string of pins.
  120. """
  121. return [_pin_name(pin) for pin in pins.split(',')]
  122. def _extract_2d_array(raw):
  123. """Return a 2d array of strings
  124. """
  125. out_array = []
  126. while raw[-1] != '}':
  127. raw = raw[:-1]
  128. for row in raw.split('},{'):
  129. if row.startswith('{'):
  130. row = row[1:]
  131. if row.endswith('}'):
  132. row = row[:-1]
  133. out_array.append([])
  134. for val in row.split(','):
  135. out_array[-1].append(val)
  136. return out_array
  137. def _extract_2d_int_array(raw):
  138. """Return a 2d array of ints
  139. """
  140. ret = _extract_2d_array(raw)
  141. return [list(map(int, x)) for x in ret]
  142. def _extract_direct_matrix(direct_pins):
  143. """extract direct_matrix
  144. """
  145. direct_pin_array = _extract_2d_array(direct_pins)
  146. for i in range(len(direct_pin_array)):
  147. for j in range(len(direct_pin_array[i])):
  148. if direct_pin_array[i][j] == 'NO_PIN':
  149. direct_pin_array[i][j] = None
  150. return direct_pin_array
  151. def _extract_audio(info_data, config_c):
  152. """Populate data about the audio configuration
  153. """
  154. audio_pins = []
  155. for pin in 'B5', 'B6', 'B7', 'C4', 'C5', 'C6':
  156. if config_c.get(f'{pin}_AUDIO'):
  157. audio_pins.append(pin)
  158. if audio_pins:
  159. info_data['audio'] = {'pins': audio_pins}
  160. def _extract_encoders_values(config_c, postfix=''):
  161. """Common encoder extraction logic
  162. """
  163. a_pad = config_c.get(f'ENCODERS_PAD_A{postfix}', '').replace(' ', '')[1:-1]
  164. b_pad = config_c.get(f'ENCODERS_PAD_B{postfix}', '').replace(' ', '')[1:-1]
  165. resolutions = config_c.get(f'ENCODER_RESOLUTIONS{postfix}', '').replace(' ', '')[1:-1]
  166. default_resolution = config_c.get('ENCODER_RESOLUTION', '4')
  167. if a_pad and b_pad:
  168. a_pad = list(filter(None, a_pad.split(',')))
  169. b_pad = list(filter(None, b_pad.split(',')))
  170. resolutions = list(filter(None, resolutions.split(',')))
  171. resolutions += [default_resolution] * (len(a_pad) - len(resolutions))
  172. encoders = []
  173. for index in range(len(a_pad)):
  174. encoders.append({'pin_a': a_pad[index], 'pin_b': b_pad[index], "resolution": int(resolutions[index])})
  175. return encoders
  176. def _extract_encoders(info_data, config_c):
  177. """Populate data about encoder pins
  178. """
  179. encoders = _extract_encoders_values(config_c)
  180. if encoders:
  181. if 'encoder' not in info_data:
  182. info_data['encoder'] = {}
  183. if 'rotary' in info_data['encoder']:
  184. _log_warning(info_data, 'Encoder config is specified in both config.h and info.json (encoder.rotary) (Value: %s), the config.h value wins.' % info_data['encoder']['rotary'])
  185. info_data['encoder']['rotary'] = encoders
  186. def _extract_split_encoders(info_data, config_c):
  187. """Populate data about split encoder pins
  188. """
  189. encoders = _extract_encoders_values(config_c, '_RIGHT')
  190. if encoders:
  191. if 'split' not in info_data:
  192. info_data['split'] = {}
  193. if 'encoder' not in info_data['split']:
  194. info_data['split']['encoder'] = {}
  195. if 'right' not in info_data['split']['encoder']:
  196. info_data['split']['encoder']['right'] = {}
  197. if 'rotary' in info_data['split']['encoder']['right']:
  198. _log_warning(info_data, 'Encoder config is specified in both config.h and info.json (encoder.rotary) (Value: %s), the config.h value wins.' % info_data['split']['encoder']['right']['rotary'])
  199. info_data['split']['encoder']['right']['rotary'] = encoders
  200. def _extract_secure_unlock(info_data, config_c):
  201. """Populate data about the secure unlock sequence
  202. """
  203. unlock = config_c.get('SECURE_UNLOCK_SEQUENCE', '').replace(' ', '')[1:-1]
  204. if unlock:
  205. unlock_array = _extract_2d_int_array(unlock)
  206. if 'secure' not in info_data:
  207. info_data['secure'] = {}
  208. if 'unlock_sequence' in info_data['secure']:
  209. _log_warning(info_data, 'Secure unlock sequence is specified in both config.h (SECURE_UNLOCK_SEQUENCE) and info.json (secure.unlock_sequence) (Value: %s), the config.h value wins.' % info_data['secure']['unlock_sequence'])
  210. info_data['secure']['unlock_sequence'] = unlock_array
  211. def _extract_split_main(info_data, config_c):
  212. """Populate data about the split configuration
  213. """
  214. # Figure out how the main half is determined
  215. if config_c.get('SPLIT_HAND_PIN') is True:
  216. if 'split' not in info_data:
  217. info_data['split'] = {}
  218. if 'main' in info_data['split']:
  219. _log_warning(info_data, 'Split main hand is specified in both config.h (SPLIT_HAND_PIN) and info.json (split.main) (Value: %s), the config.h value wins.' % info_data['split']['main'])
  220. info_data['split']['main'] = 'pin'
  221. if config_c.get('SPLIT_HAND_MATRIX_GRID'):
  222. if 'split' not in info_data:
  223. info_data['split'] = {}
  224. if 'main' in info_data['split']:
  225. _log_warning(info_data, 'Split main hand is specified in both config.h (SPLIT_HAND_MATRIX_GRID) and info.json (split.main) (Value: %s), the config.h value wins.' % info_data['split']['main'])
  226. info_data['split']['main'] = 'matrix_grid'
  227. info_data['split']['matrix_grid'] = _extract_pins(config_c['SPLIT_HAND_MATRIX_GRID'])
  228. if config_c.get('EE_HANDS') is True:
  229. if 'split' not in info_data:
  230. info_data['split'] = {}
  231. if 'main' in info_data['split']:
  232. _log_warning(info_data, 'Split main hand is specified in both config.h (EE_HANDS) and info.json (split.main) (Value: %s), the config.h value wins.' % info_data['split']['main'])
  233. info_data['split']['main'] = 'eeprom'
  234. if config_c.get('MASTER_RIGHT') is True:
  235. if 'split' not in info_data:
  236. info_data['split'] = {}
  237. if 'main' in info_data['split']:
  238. _log_warning(info_data, 'Split main hand is specified in both config.h (MASTER_RIGHT) and info.json (split.main) (Value: %s), the config.h value wins.' % info_data['split']['main'])
  239. info_data['split']['main'] = 'right'
  240. if config_c.get('MASTER_LEFT') is True:
  241. if 'split' not in info_data:
  242. info_data['split'] = {}
  243. if 'main' in info_data['split']:
  244. _log_warning(info_data, 'Split main hand is specified in both config.h (MASTER_LEFT) and info.json (split.main) (Value: %s), the config.h value wins.' % info_data['split']['main'])
  245. info_data['split']['main'] = 'left'
  246. def _extract_split_transport(info_data, config_c):
  247. # Figure out the transport method
  248. if config_c.get('USE_I2C') is True:
  249. if 'split' not in info_data:
  250. info_data['split'] = {}
  251. if 'transport' not in info_data['split']:
  252. info_data['split']['transport'] = {}
  253. if 'protocol' in info_data['split']['transport']:
  254. _log_warning(info_data, 'Split transport is specified in both config.h (USE_I2C) and info.json (split.transport.protocol) (Value: %s), the config.h value wins.' % info_data['split']['transport'])
  255. info_data['split']['transport']['protocol'] = 'i2c'
  256. # Ignore transport defaults if "SPLIT_KEYBOARD" is unset
  257. elif 'enabled' in info_data.get('split', {}):
  258. if 'split' not in info_data:
  259. info_data['split'] = {}
  260. if 'transport' not in info_data['split']:
  261. info_data['split']['transport'] = {}
  262. if 'protocol' not in info_data['split']['transport']:
  263. info_data['split']['transport']['protocol'] = 'serial'
  264. def _extract_split_right_pins(info_data, config_c):
  265. # Figure out the right half matrix pins
  266. row_pins = config_c.get('MATRIX_ROW_PINS_RIGHT', '').replace('{', '').replace('}', '').strip()
  267. col_pins = config_c.get('MATRIX_COL_PINS_RIGHT', '').replace('{', '').replace('}', '').strip()
  268. unused_pin_text = config_c.get('UNUSED_PINS_RIGHT')
  269. unused_pins = unused_pin_text.replace('{', '').replace('}', '').strip() if isinstance(unused_pin_text, str) else None
  270. direct_pins = config_c.get('DIRECT_PINS_RIGHT', '').replace(' ', '')[1:-1]
  271. if row_pins or col_pins or direct_pins or unused_pins:
  272. if info_data.get('split', {}).get('matrix_pins', {}).get('right') in info_data:
  273. _log_warning(info_data, 'Right hand matrix data is specified in both info.json and config.h, the config.h values win.')
  274. if 'split' not in info_data:
  275. info_data['split'] = {}
  276. if 'matrix_pins' not in info_data['split']:
  277. info_data['split']['matrix_pins'] = {}
  278. if 'right' not in info_data['split']['matrix_pins']:
  279. info_data['split']['matrix_pins']['right'] = {}
  280. if col_pins:
  281. info_data['split']['matrix_pins']['right']['cols'] = _extract_pins(col_pins)
  282. if row_pins:
  283. info_data['split']['matrix_pins']['right']['rows'] = _extract_pins(row_pins)
  284. if direct_pins:
  285. info_data['split']['matrix_pins']['right']['direct'] = _extract_direct_matrix(direct_pins)
  286. if unused_pins:
  287. info_data['split']['matrix_pins']['right']['unused'] = _extract_pins(unused_pins)
  288. def _extract_matrix_info(info_data, config_c):
  289. """Populate the matrix information.
  290. """
  291. row_pins = config_c.get('MATRIX_ROW_PINS', '').replace('{', '').replace('}', '').strip()
  292. col_pins = config_c.get('MATRIX_COL_PINS', '').replace('{', '').replace('}', '').strip()
  293. unused_pin_text = config_c.get('UNUSED_PINS')
  294. unused_pins = unused_pin_text.replace('{', '').replace('}', '').strip() if isinstance(unused_pin_text, str) else None
  295. direct_pins = config_c.get('DIRECT_PINS', '').replace(' ', '')[1:-1]
  296. info_snippet = {}
  297. if 'MATRIX_ROWS' in config_c and 'MATRIX_COLS' in config_c:
  298. if 'matrix_size' in info_data:
  299. _log_warning(info_data, 'Matrix size is specified in both info.json and config.h, the config.h values win.')
  300. info_data['matrix_size'] = {
  301. 'cols': compute(config_c.get('MATRIX_COLS', '0')),
  302. 'rows': compute(config_c.get('MATRIX_ROWS', '0')),
  303. }
  304. if row_pins and col_pins:
  305. if 'matrix_pins' in info_data and 'cols' in info_data['matrix_pins'] and 'rows' in info_data['matrix_pins']:
  306. _log_warning(info_data, 'Matrix pins are specified in both info.json and config.h, the config.h values win.')
  307. info_snippet['cols'] = _extract_pins(col_pins)
  308. info_snippet['rows'] = _extract_pins(row_pins)
  309. if direct_pins:
  310. if 'matrix_pins' in info_data and 'direct' in info_data['matrix_pins']:
  311. _log_warning(info_data, 'Direct pins are specified in both info.json and config.h, the config.h values win.')
  312. info_snippet['direct'] = _extract_direct_matrix(direct_pins)
  313. if unused_pins:
  314. if 'matrix_pins' not in info_data:
  315. info_data['matrix_pins'] = {}
  316. info_snippet['unused'] = _extract_pins(unused_pins)
  317. if config_c.get('CUSTOM_MATRIX', 'no') != 'no':
  318. if 'matrix_pins' in info_data and 'custom' in info_data['matrix_pins']:
  319. _log_warning(info_data, 'Custom Matrix is specified in both info.json and config.h, the config.h values win.')
  320. info_snippet['custom'] = True
  321. if config_c['CUSTOM_MATRIX'] == 'lite':
  322. info_snippet['custom_lite'] = True
  323. if info_snippet:
  324. info_data['matrix_pins'] = info_snippet
  325. return info_data
  326. # TODO: kill off usb.device_ver in favor of usb.device_version
  327. def _extract_device_version(info_data):
  328. if info_data.get('usb'):
  329. if info_data['usb'].get('device_version') and not info_data['usb'].get('device_ver'):
  330. (major, minor, revision) = info_data['usb']['device_version'].split('.', 3)
  331. info_data['usb']['device_ver'] = f'0x{major.zfill(2)}{minor}{revision}'
  332. if not info_data['usb'].get('device_version') and info_data['usb'].get('device_ver'):
  333. major = int(info_data['usb']['device_ver'][2:4])
  334. minor = int(info_data['usb']['device_ver'][4])
  335. revision = int(info_data['usb']['device_ver'][5])
  336. info_data['usb']['device_version'] = f'{major}.{minor}.{revision}'
  337. def _config_to_json(key_type, config_value):
  338. """Convert config value using spec
  339. """
  340. if key_type.startswith('array'):
  341. if '.' in key_type:
  342. key_type, array_type = key_type.split('.', 1)
  343. else:
  344. array_type = None
  345. config_value = config_value.replace('{', '').replace('}', '').strip()
  346. if array_type == 'int':
  347. return list(map(int, config_value.split(',')))
  348. else:
  349. return config_value.split(',')
  350. elif key_type == 'bool':
  351. return config_value in true_values
  352. elif key_type == 'hex':
  353. return '0x' + config_value[2:].upper()
  354. elif key_type == 'list':
  355. return config_value.split()
  356. elif key_type == 'int':
  357. return int(config_value)
  358. elif key_type == 'str':
  359. return config_value.strip('"')
  360. elif key_type == 'bcd_version':
  361. major = int(config_value[2:4])
  362. minor = int(config_value[4])
  363. revision = int(config_value[5])
  364. return f'{major}.{minor}.{revision}'
  365. return config_value
  366. def _extract_config_h(info_data, config_c):
  367. """Pull some keyboard information from existing config.h files
  368. """
  369. # Pull in data from the json map
  370. dotty_info = dotty(info_data)
  371. info_config_map = json_load(Path('data/mappings/info_config.json'))
  372. for config_key, info_dict in info_config_map.items():
  373. info_key = info_dict['info_key']
  374. key_type = info_dict.get('value_type', 'raw')
  375. try:
  376. if config_key in config_c and info_dict.get('invalid', False):
  377. _log_error(info_data, '%s in config.h is no longer a valid option' % config_key)
  378. elif config_key in config_c and info_dict.get('deprecated', False):
  379. _log_warning(info_data, '%s in config.h is deprecated and will be removed at a later date' % config_key)
  380. if config_key in config_c and info_dict.get('to_json', True):
  381. if dotty_info.get(info_key) and info_dict.get('warn_duplicate', True):
  382. _log_warning(info_data, '%s in config.h is overwriting %s in info.json' % (config_key, info_key))
  383. dotty_info[info_key] = _config_to_json(key_type, config_c[config_key])
  384. except Exception as e:
  385. _log_warning(info_data, f'{config_key}->{info_key}: {e}')
  386. info_data.update(dotty_info)
  387. # Pull data that easily can't be mapped in json
  388. _extract_matrix_info(info_data, config_c)
  389. _extract_audio(info_data, config_c)
  390. _extract_secure_unlock(info_data, config_c)
  391. _extract_split_main(info_data, config_c)
  392. _extract_split_transport(info_data, config_c)
  393. _extract_split_right_pins(info_data, config_c)
  394. _extract_encoders(info_data, config_c)
  395. _extract_split_encoders(info_data, config_c)
  396. _extract_device_version(info_data)
  397. return info_data
  398. def _process_defaults(info_data):
  399. """Process any additional defaults based on currently discovered information
  400. """
  401. defaults_map = json_load(Path('data/mappings/defaults.json'))
  402. for default_type in defaults_map.keys():
  403. thing_map = defaults_map[default_type]
  404. if default_type in info_data:
  405. for key, value in thing_map.get(info_data[default_type], {}).items():
  406. info_data[key] = value
  407. return info_data
  408. def _extract_rules_mk(info_data, rules):
  409. """Pull some keyboard information from existing rules.mk files
  410. """
  411. info_data['processor'] = rules.get('MCU', info_data.get('processor', 'atmega32u4'))
  412. if info_data['processor'] in CHIBIOS_PROCESSORS:
  413. arm_processor_rules(info_data, rules)
  414. elif info_data['processor'] in LUFA_PROCESSORS + VUSB_PROCESSORS:
  415. avr_processor_rules(info_data, rules)
  416. else:
  417. cli.log.warning("%s: Unknown MCU: %s" % (info_data['keyboard_folder'], info_data['processor']))
  418. unknown_processor_rules(info_data, rules)
  419. # Pull in data from the json map
  420. dotty_info = dotty(info_data)
  421. info_rules_map = json_load(Path('data/mappings/info_rules.json'))
  422. for rules_key, info_dict in info_rules_map.items():
  423. info_key = info_dict['info_key']
  424. key_type = info_dict.get('value_type', 'raw')
  425. try:
  426. if rules_key in rules and info_dict.get('invalid', False):
  427. _log_error(info_data, '%s in rules.mk is no longer a valid option' % rules_key)
  428. elif rules_key in rules and info_dict.get('deprecated', False):
  429. _log_warning(info_data, '%s in rules.mk is deprecated and will be removed at a later date' % rules_key)
  430. if rules_key in rules and info_dict.get('to_json', True):
  431. if dotty_info.get(info_key) and info_dict.get('warn_duplicate', True):
  432. _log_warning(info_data, '%s in rules.mk is overwriting %s in info.json' % (rules_key, info_key))
  433. dotty_info[info_key] = _config_to_json(key_type, rules[rules_key])
  434. except Exception as e:
  435. _log_warning(info_data, f'{rules_key}->{info_key}: {e}')
  436. info_data.update(dotty_info)
  437. # Merge in config values that can't be easily mapped
  438. _extract_features(info_data, rules)
  439. return info_data
  440. def find_keyboard_c(keyboard):
  441. """Find all <keyboard>.c files
  442. """
  443. keyboard = Path(keyboard)
  444. current_path = Path('keyboards/')
  445. files = []
  446. for directory in keyboard.parts:
  447. current_path = current_path / directory
  448. keyboard_c_path = current_path / f'{directory}.c'
  449. if keyboard_c_path.exists():
  450. files.append(keyboard_c_path)
  451. return files
  452. def _extract_led_config(info_data, keyboard):
  453. """Scan all <keyboard>.c files for led config
  454. """
  455. cols = info_data['matrix_size']['cols']
  456. rows = info_data['matrix_size']['rows']
  457. # Assume what feature owns g_led_config
  458. feature = "rgb_matrix"
  459. if info_data.get("features", {}).get("led_matrix", False):
  460. feature = "led_matrix"
  461. # Process
  462. for file in find_keyboard_c(keyboard):
  463. try:
  464. ret = find_led_config(file, cols, rows)
  465. if ret:
  466. info_data[feature] = info_data.get(feature, {})
  467. info_data[feature]["layout"] = ret
  468. except Exception as e:
  469. _log_warning(info_data, f'led_config: {file.name}: {e}')
  470. return info_data
  471. def _matrix_size(info_data):
  472. """Add info_data['matrix_size'] if it doesn't exist.
  473. """
  474. if 'matrix_size' not in info_data and 'matrix_pins' in info_data:
  475. info_data['matrix_size'] = {}
  476. if 'direct' in info_data['matrix_pins']:
  477. info_data['matrix_size']['cols'] = len(info_data['matrix_pins']['direct'][0])
  478. info_data['matrix_size']['rows'] = len(info_data['matrix_pins']['direct'])
  479. elif 'cols' in info_data['matrix_pins'] and 'rows' in info_data['matrix_pins']:
  480. info_data['matrix_size']['cols'] = len(info_data['matrix_pins']['cols'])
  481. info_data['matrix_size']['rows'] = len(info_data['matrix_pins']['rows'])
  482. # Assumption of split common
  483. if 'split' in info_data:
  484. if info_data['split'].get('enabled', False):
  485. info_data['matrix_size']['rows'] *= 2
  486. return info_data
  487. def _check_matrix(info_data):
  488. """Check the matrix to ensure that row/column count is consistent.
  489. """
  490. if 'matrix_pins' in info_data and 'matrix_size' in info_data:
  491. actual_col_count = info_data['matrix_size'].get('cols', 0)
  492. actual_row_count = info_data['matrix_size'].get('rows', 0)
  493. col_count = row_count = 0
  494. if 'direct' in info_data['matrix_pins']:
  495. col_count = len(info_data['matrix_pins']['direct'][0])
  496. row_count = len(info_data['matrix_pins']['direct'])
  497. elif 'cols' in info_data['matrix_pins'] and 'rows' in info_data['matrix_pins']:
  498. col_count = len(info_data['matrix_pins']['cols'])
  499. row_count = len(info_data['matrix_pins']['rows'])
  500. if col_count != actual_col_count and col_count != (actual_col_count / 2):
  501. # FIXME: once we can we should detect if split is enabled to do the actual_col_count/2 check.
  502. _log_error(info_data, f'MATRIX_COLS is inconsistent with the size of MATRIX_COL_PINS: {col_count} != {actual_col_count}')
  503. if row_count != actual_row_count and row_count != (actual_row_count / 2):
  504. # FIXME: once we can we should detect if split is enabled to do the actual_row_count/2 check.
  505. _log_error(info_data, f'MATRIX_ROWS is inconsistent with the size of MATRIX_ROW_PINS: {row_count} != {actual_row_count}')
  506. def _search_keyboard_h(keyboard):
  507. keyboard = Path(keyboard)
  508. current_path = Path('keyboards/')
  509. aliases = {}
  510. layouts = {}
  511. for directory in keyboard.parts:
  512. current_path = current_path / directory
  513. keyboard_h = '%s.h' % (directory,)
  514. keyboard_h_path = current_path / keyboard_h
  515. if keyboard_h_path.exists():
  516. new_layouts, new_aliases = find_layouts(keyboard_h_path)
  517. layouts.update(new_layouts)
  518. for alias, alias_text in new_aliases.items():
  519. if alias_text in layouts:
  520. aliases[alias] = alias_text
  521. return layouts, aliases
  522. def _find_missing_layouts(info_data, keyboard):
  523. """Looks for layout macros when they aren't found other places.
  524. If we don't find any layouts from info.json or keyboard.h we widen our search. This is error prone which is why we want to encourage people to follow the standard above.
  525. """
  526. _log_warning(info_data, '%s: Falling back to searching for KEYMAP/LAYOUT macros.' % (keyboard))
  527. for file in glob('keyboards/%s/*.h' % keyboard):
  528. these_layouts, these_aliases = find_layouts(file)
  529. if these_layouts:
  530. for layout_name, layout_json in these_layouts.items():
  531. if not layout_name.startswith('LAYOUT_kc'):
  532. layout_json['c_macro'] = True
  533. info_data['layouts'][layout_name] = layout_json
  534. for alias, alias_text in these_aliases.items():
  535. if alias_text in these_layouts:
  536. if 'layout_aliases' not in info_data:
  537. info_data['layout_aliases'] = {}
  538. info_data['layout_aliases'][alias] = alias_text
  539. def _log_error(info_data, message):
  540. """Send an error message to both JSON and the log.
  541. """
  542. info_data['parse_errors'].append(message)
  543. cli.log.error('%s: %s', info_data.get('keyboard_folder', 'Unknown Keyboard!'), message)
  544. def _log_warning(info_data, message):
  545. """Send a warning message to both JSON and the log.
  546. """
  547. info_data['parse_warnings'].append(message)
  548. cli.log.warning('%s: %s', info_data.get('keyboard_folder', 'Unknown Keyboard!'), message)
  549. def arm_processor_rules(info_data, rules):
  550. """Setup the default info for an ARM board.
  551. """
  552. info_data['processor_type'] = 'arm'
  553. info_data['protocol'] = 'ChibiOS'
  554. if 'bootloader' not in info_data:
  555. info_data['bootloader'] = 'unknown'
  556. if 'STM32' in info_data['processor']:
  557. info_data['platform'] = 'STM32'
  558. elif 'MCU_SERIES' in rules:
  559. info_data['platform'] = rules['MCU_SERIES']
  560. elif 'ARM_ATSAM' in rules:
  561. info_data['platform'] = 'ARM_ATSAM'
  562. return info_data
  563. def avr_processor_rules(info_data, rules):
  564. """Setup the default info for an AVR board.
  565. """
  566. info_data['processor_type'] = 'avr'
  567. info_data['platform'] = rules['ARCH'] if 'ARCH' in rules else 'unknown'
  568. info_data['protocol'] = 'V-USB' if rules.get('MCU') in VUSB_PROCESSORS else 'LUFA'
  569. if 'bootloader' not in info_data:
  570. info_data['bootloader'] = 'atmel-dfu'
  571. # FIXME(fauxpark/anyone): Eventually we should detect the protocol by looking at PROTOCOL inherited from mcu_selection.mk:
  572. # info_data['protocol'] = 'V-USB' if rules.get('PROTOCOL') == 'VUSB' else 'LUFA'
  573. return info_data
  574. def unknown_processor_rules(info_data, rules):
  575. """Setup the default keyboard info for unknown boards.
  576. """
  577. info_data['bootloader'] = 'unknown'
  578. info_data['platform'] = 'unknown'
  579. info_data['processor'] = 'unknown'
  580. info_data['processor_type'] = 'unknown'
  581. info_data['protocol'] = 'unknown'
  582. return info_data
  583. def merge_info_jsons(keyboard, info_data):
  584. """Return a merged copy of all the info.json files for a keyboard.
  585. """
  586. for info_file in find_info_json(keyboard):
  587. # Load and validate the JSON data
  588. new_info_data = json_load(info_file)
  589. if not isinstance(new_info_data, dict):
  590. _log_error(info_data, "Invalid file %s, root object should be a dictionary." % (str(info_file),))
  591. continue
  592. try:
  593. validate(new_info_data, 'qmk.keyboard.v1')
  594. except jsonschema.ValidationError as e:
  595. json_path = '.'.join([str(p) for p in e.absolute_path])
  596. cli.log.error('Not including data from file: %s', info_file)
  597. cli.log.error('\t%s: %s', json_path, e.message)
  598. continue
  599. # Merge layout data in
  600. if 'layout_aliases' in new_info_data:
  601. info_data['layout_aliases'] = {**info_data.get('layout_aliases', {}), **new_info_data['layout_aliases']}
  602. del new_info_data['layout_aliases']
  603. for layout_name, layout in new_info_data.get('layouts', {}).items():
  604. if layout_name in info_data.get('layout_aliases', {}):
  605. _log_warning(info_data, f"info.json uses alias name {layout_name} instead of {info_data['layout_aliases'][layout_name]}")
  606. layout_name = info_data['layout_aliases'][layout_name]
  607. if layout_name in info_data['layouts']:
  608. if len(info_data['layouts'][layout_name]['layout']) != len(layout['layout']):
  609. msg = 'Number of keys for %s does not match! info.json specifies %d keys, C macro specifies %d'
  610. _log_error(info_data, msg % (layout_name, len(layout['layout']), len(info_data['layouts'][layout_name]['layout'])))
  611. else:
  612. for new_key, existing_key in zip(layout['layout'], info_data['layouts'][layout_name]['layout']):
  613. existing_key.update(new_key)
  614. else:
  615. if not all('matrix' in key_data.keys() for key_data in layout['layout']):
  616. _log_error(info_data, f'Layout "{layout_name}" has no "matrix" definition in either "info.json" or "<keyboard>.h"!')
  617. else:
  618. layout['c_macro'] = False
  619. info_data['layouts'][layout_name] = layout
  620. # Update info_data with the new data
  621. if 'layouts' in new_info_data:
  622. del new_info_data['layouts']
  623. deep_update(info_data, new_info_data)
  624. return info_data
  625. def find_info_json(keyboard):
  626. """Finds all the info.json files associated with a keyboard.
  627. """
  628. # Find the most specific first
  629. base_path = Path('keyboards')
  630. keyboard_path = base_path / keyboard
  631. keyboard_parent = keyboard_path.parent
  632. info_jsons = [keyboard_path / 'info.json']
  633. # Add DEFAULT_FOLDER before parents, if present
  634. rules = rules_mk(keyboard)
  635. if 'DEFAULT_FOLDER' in rules:
  636. info_jsons.append(Path(rules['DEFAULT_FOLDER']) / 'info.json')
  637. # Add in parent folders for least specific
  638. for _ in range(5):
  639. if keyboard_parent == base_path:
  640. break
  641. info_jsons.append(keyboard_parent / 'info.json')
  642. keyboard_parent = keyboard_parent.parent
  643. # Return a list of the info.json files that actually exist
  644. return [info_json for info_json in info_jsons if info_json.exists()]
  645. def keymap_json_config(keyboard, keymap):
  646. """Extract keymap level config
  647. """
  648. keymap_folder = locate_keymap(keyboard, keymap).parent
  649. km_info_json = parse_configurator_json(keymap_folder / 'keymap.json')
  650. return km_info_json.get('config', {})
  651. def keymap_json(keyboard, keymap):
  652. """Generate the info.json data for a specific keymap.
  653. """
  654. keymap_folder = locate_keymap(keyboard, keymap).parent
  655. # Files to scan
  656. keymap_config = keymap_folder / 'config.h'
  657. keymap_rules = keymap_folder / 'rules.mk'
  658. keymap_file = keymap_folder / 'keymap.json'
  659. # Build the info.json file
  660. kb_info_json = info_json(keyboard)
  661. # Merge in the data from keymap.json
  662. km_info_json = keymap_json_config(keyboard, keymap) if keymap_file.exists() else {}
  663. deep_update(kb_info_json, km_info_json)
  664. # Merge in the data from config.h, and rules.mk
  665. _extract_rules_mk(kb_info_json, parse_rules_mk_file(keymap_rules))
  666. _extract_config_h(kb_info_json, parse_config_h_file(keymap_config))
  667. return kb_info_json