mirrors/monado
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/monado/monado.git synced 2025-01-06 07:06:10 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

ipc: Put generator main in a function, for testing purposes

Browse source
This commit is contained in:
Ryan Pavlik 2020-05-27 17:01:42 -05:00
parent 8bd1ac776e
commit a44a915151
1 changed files with 32 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
src/xrt/ipc/proto.py
View file

@ -140,7 +140,8 @@ header = '''// Copyright 2020, Collabora, Ltd.
*/ */
''' '''
def doH(file):
def doH(file, p):
f = open(file, "w") f = open(file, "w")
f.write(header.format(brief='Generated IPC protocol header', suffix='')) f.write(header.format(brief='Generated IPC protocol header', suffix=''))
f.write(''' f.write('''
@ -210,7 +211,8 @@ ipc_cmd_to_str(ipc_command_t id)
f.write("\n// clang-format on\n") f.write("\n// clang-format on\n")
f.close() f.close()
def doClientC(file):
def doClientC(file, p):
f = open(file, "w") f = open(file, "w")
f.write(header.format(brief='Generated IPC client code', suffix='_client')) f.write(header.format(brief='Generated IPC client code', suffix='_client'))
f.write(''' f.write('''
@ -263,7 +265,8 @@ def doClientC(file):
f.write("\n// clang-format off\n") f.write("\n// clang-format off\n")
f.close() f.close()
def doClientH(file):
def doClientH(file, p):
f = open(file, "w") f = open(file, "w")
f.write(header.format(brief='Generated IPC client code', suffix='_client')) f.write(header.format(brief='Generated IPC client code', suffix='_client'))
f.write(''' f.write('''
@ -283,7 +286,8 @@ def doClientH(file):
f.write("\n// clang-format on\n") f.write("\n// clang-format on\n")
f.close() f.close()
def doServerC(file):
def doServerC(file, p):
f = open(file, "w") f = open(file, "w")
f.write(header.format(brief='Generated IPC server code', suffix='_server')) f.write(header.format(brief='Generated IPC server code', suffix='_server'))
f.write(''' f.write('''
@ -351,7 +355,8 @@ ipc_dispatch(volatile struct ipc_client_state *cs, ipc_command_t *ipc_command)
''') ''')
f.close() f.close()
def doServerH(file):
def doServerH(file, p):
f = open(file, "w") f = open(file, "w")
f.write(header.format(brief='Generated IPC server code', suffix='_server')) f.write(header.format(brief='Generated IPC server code', suffix='_server'))
f.write(''' f.write('''
@ -374,21 +379,27 @@ ipc_dispatch(volatile struct ipc_client_state *cs, ipc_command_t *ipc_command);
f.close() f.close()
parser = argparse.ArgumentParser(description='Protocol generator.') def main():
parser.add_argument('proto', help='Protocol file to use') parser = argparse.ArgumentParser(description='Protocol generator.')
parser.add_argument('output', type=str, nargs='+', help='Output file, uses the ending to figure out what file it should generate') parser.add_argument('proto', help='Protocol file to use')
args = parser.parse_args() parser.add_argument('output', type=str, nargs='+',
help='Output file, uses the ending to figure out what file it should generate')
args = parser.parse_args()
p = Proto.loadAndParse(args.proto) p = Proto.loadAndParse(args.proto)
for output in args.output: for output in args.output:
if output.endswith("ipc_protocol_generated.h"): if output.endswith("ipc_protocol_generated.h"):
doH(output) doH(output, p)
if output.endswith("ipc_client_generated.c"): if output.endswith("ipc_client_generated.c"):
doClientC(output) doClientC(output, p)
if output.endswith("ipc_client_generated.h"): if output.endswith("ipc_client_generated.h"):
doClientH(output) doClientH(output, p)
if output.endswith("ipc_server_generated.c"): if output.endswith("ipc_server_generated.c"):
doServerC(output) doServerC(output, p)
if output.endswith("ipc_server_generated.h"): if output.endswith("ipc_server_generated.h"):
doServerH(output) doServerH(output, p)
if __name__ == "__main__":
main()