ahoneybun/qmk_firmware
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Implement codegen for more data types - codegen for return_execute stubs removed

Browse source
This commit is contained in:
zvecr 2022-04-01 23:37:58 +01:00
parent d968e6c005
commit e7d9d6675c
6 changed files with 223 additions and 67 deletions
Download patch file Download diff file Expand all files Collapse all files

58
lib/python/qmk/xap/gen_firmware/header_generator.py
View file

@ -3,6 +3,7 @@
import re
from fnvhash import fnv1a_32
from qmk.casing import to_snake
from qmk.commands import dump_lines
from qmk.git import git_get_version
from qmk.constants import GPL2_HEADER_C_LIKE, GENERATED_HEADER_C_LIKE
@ -112,7 +113,58 @@ def _append_route_capabilities(lines, container, container_id=None, route_stack=
route_stack.pop()
def _append_types(lines, container):
def _append_route_types(lines, container, container_id=None, route_stack=None):
"""Handles creating
"""
if route_stack is None:
route_stack = [container]
else:
route_stack.append(container)
route_name = to_snake('_'.join([r['define'] for r in route_stack]))
# Inbound
if 'request_struct_members' in container:
request_struct_members = container['request_struct_members']
lines.append(f'typedef struct {{')
for member in request_struct_members:
member_type = _get_c_type(member['type'])
member_name = to_snake(member['name'])
lines.append(f' {member_type} {member_name};')
lines.append(f'}} {route_name}_arg_t;')
elif 'request_type' in container:
request_type = container['request_type']
lines.append(f'typedef {_get_c_type(request_type)} {route_name}_arg_t;')
# Outbound
qualifier = 'const' if 'return_constant' in container else ''
if 'return_struct_members' in container:
return_struct_members = container['return_struct_members']
lines.append(f'typedef struct {{')
for member in return_struct_members:
member_type = _get_c_type(member['type'])
member_name = f'{qualifier} {to_snake(member["name"])}'
lines.append(f' {member_type} {member_name};')
lines.append(f'}} {route_name}_t;')
elif 'return_type' in container:
return_type = container['return_type']
if return_type == 'u8[32]':
lines.append(f'typedef struct {{ uint8_t x[32]; }} {route_name}_t;')
else:
lines.append(f'typedef {_get_c_type(return_type)} {route_name}_t;')
# Recurse
if 'routes' in container:
for route_id in container['routes']:
route = container['routes'][route_id]
_append_route_types(lines, route, route_id, route_stack)
route_stack.pop()
def _append_internal_types(lines, container):
"""Handles creating the various constants, types, defines, etc.
"""
response_flags = container.get('response_flags', {})
@ -167,7 +219,9 @@ def generate_header(output_file, keyboard):
lines.append('')
# Types
_append_types(lines, xap_defs)
_append_internal_types(lines, xap_defs)
lines.append('')
_append_route_types(lines, xap_defs)
lines.append('')
# Append the route and command defines

70
lib/python/qmk/xap/gen_firmware/inline_generator.py
View file

@ -47,7 +47,11 @@ def _get_route_type(container):
if container['return_type'] == 'u8':
return 'XAP_VALUE'
elif 'return_constant' in container:
if container['return_type'] == 'u32':
if container['return_type'] == 'u8':
return 'XAP_CONST_MEM'
elif container['return_type'] == 'u16':
return 'XAP_CONST_MEM'
elif container['return_type'] == 'u32':
return 'XAP_CONST_MEM'
elif container['return_type'] == 'struct':
return 'XAP_CONST_MEM'
@ -69,42 +73,7 @@ def _append_routing_table_declaration(lines, container, container_id, route_stac
elif 'return_execute' in container:
execute = container['return_execute']
request_type = container.get('request_type', None)
return_type = container['return_type']
lines.append(
f'''
bool xap_respond_{execute}(xap_token_t token, const uint8_t *data, size_t data_len) {{
if (data_len != {_get_c_size(request_type)}) {{
xap_respond_failure(token, 0);
return false;
}}
uint8_t ret[{_get_c_size(return_type)}] = {{0}};
'''
)
if not request_type:
lines.append(f'''
bool {execute}(uint8_t *ret, uint8_t ret_len);
if(!{execute}(ret, sizeof(ret))) {{
xap_respond_failure(token, 0);
return false;
}}
''')
else:
lines.append(
f'''
{_get_c_type(request_type)} *argp = ({_get_c_type(request_type)} *)&data[0];
bool {execute}({_get_c_type(request_type)} arg, uint8_t *ret, uint8_t ret_len);
if(!{execute}(*argp, ret, sizeof(ret))) {{
xap_respond_failure(token, 0);
return false;
}}
'''
)
lines.append('''
return xap_respond_data(token, ret, sizeof(ret));
}''')
lines.append(f'bool xap_respond_{execute}(xap_token_t token, const uint8_t *data, size_t data_len);')
# elif 'return_value' in container:
# value = container['return_value']
@ -114,21 +83,24 @@ bool xap_respond_{execute}(xap_token_t token, const uint8_t *data, size_t data_l
elif 'return_constant' in container:
if container['return_type'] == 'u32':
if container['return_type'] == 'u8':
constant = container['return_constant']
lines.append('')
lines.append(f'static const uint8_t {route_name}_data PROGMEM = {constant};')
elif container['return_type'] == 'u16':
constant = container['return_constant']
lines.append('')
lines.append(f'static const uint16_t {route_name}_data PROGMEM = {constant};')
elif container['return_type'] == 'u32':
constant = container['return_constant']
lines.append('')
lines.append(f'static const uint32_t {route_name}_data PROGMEM = {constant};')
elif container['return_type'] == 'struct':
lines.append('')
lines.append(f'static const struct {route_name}_t {{')
for member in container['return_struct_members']:
member_type = _get_c_type(member['type'])
member_name = to_snake(member['name'])
lines.append(f' const {member_type} {member_name};')
lines.append(f'}} {route_name}_data PROGMEM = {{')
lines.append(f'static const {route_name}_t {route_name}_data PROGMEM = {{')
for constant in container['return_constant']:
lines.append(f' {constant},')
@ -212,7 +184,11 @@ def _append_routing_table_entry(lines, container, container_id, route_stack):
elif 'return_value' in container:
_append_routing_table_entry_value(lines, container, container_id, route_stack)
elif 'return_constant' in container:
if container['return_type'] == 'u32':
if container['return_type'] == 'u8':
_append_routing_table_entry_const_data(lines, container, container_id, route_stack)
elif container['return_type'] == 'u16':
_append_routing_table_entry_const_data(lines, container, container_id, route_stack)
elif container['return_type'] == 'u32':
_append_routing_table_entry_const_data(lines, container, container_id, route_stack)
elif container['return_type'] == 'struct':
_append_routing_table_entry_const_data(lines, container, container_id, route_stack)