import os
import sys
import re
-import argparse
-from mako.template import Template
-from mako.exceptions import RichTraceback
-
-def write_template_to_string(template_filename, **kwargs):
- try:
- template = Template(filename=template_filename)
- # Split + Join fixes line-endings for whatever platform you are using
- return '\n'.join(template.render(**kwargs).splitlines())
- except:
- traceback = RichTraceback()
- for (filename, lineno, function, line) in traceback.traceback:
- print("File %s, line %s, in %s" % (filename, lineno, function))
- print(line, "\n")
- print("%s: %s" % (str(traceback.error.__class__.__name__), traceback.error))
-
-def write_template_to_file(template_filename, output_filename, **kwargs):
- with open(output_filename, "w") as outfile:
- print(write_template_to_string(template_filename, **kwargs), file=outfile)
+from gen_common import *
def parse_event_fields(lines, idx, event_dict):
field_names = []
line = lines[idx].rstrip()
idx += 1
- field = re.match(r"(\s*)(\w+)(\s*)(\w+)", line)
+ field = re.match(r'(\s*)(\w+)(\s*)(\w+)', line)
if field:
field_types.append(field.group(2))
field_names.append(field.group(4))
num_fields += 1
- end_of_event = re.match(r"(\s*)};", line)
+ end_of_event = re.match(r'(\s*)};', line)
event_dict['field_types'] = field_types
event_dict['field_names'] = field_names
line = lines[idx].rstrip()
idx += 1
- preprocessor = re.search(r"#if|#endif", line)
+ preprocessor = re.search(r'#if|#endif', line)
if not preprocessor:
- enum = re.match(r"(\s*)(\w+)(\s*)", line)
+ enum = re.match(r'(\s*)(\w+)(\s*)', line)
if enum:
enum_names.append(line)
- end_of_enum = re.match(r"(\s*)};", line)
+ end_of_enum = re.match(r'(\s*)};', line)
event_dict['names'] = enum_names
return idx
-def parse_protos(filename):
- protos = {}
+def parse_protos(protos, filename):
with open(filename, 'r') as f:
lines=f.readlines()
idx = 0
- protos['events'] = {} # event dictionary containing events with their fields
- protos['event_names'] = [] # needed to keep events in order parsed. dict is not ordered.
- protos['enums'] = {}
- protos['enum_names'] = []
-
eventId = 0
raw_text = []
while idx < len(lines):
idx += 1
# search for event definitions.
- match = re.match(r"(\s*)event(\s*)(\w+)", line)
+ match = re.match(r'(\s*)event(\s*)(\w+)', line)
if match:
eventId += 1
idx = parse_event_fields(lines, idx, protos['events'][event_name])
# search for enums.
- match = re.match(r"(\s*)enum(\s*)(\w+)", line)
+ match = re.match(r'(\s*)enum(\s*)(\w+)', line)
if match:
enum_name = match.group(3)
protos['enums'][enum_name] = {}
idx = parse_enums(lines, idx, protos['enums'][enum_name])
- return protos
-
def main():
# Parse args...
- parser = argparse.ArgumentParser()
- parser.add_argument("--proto", "-p", help="Path to proto file", required=True)
- parser.add_argument("--output", "-o", help="Output filename (i.e. event.h)", required=True)
- parser.add_argument("--gen_event_h", "-geh", help="Generate event header", action="store_true", default=False)
- parser.add_argument("--gen_event_cpp", "-gec", help="Generate event cpp", action="store_true", default=False)
- parser.add_argument("--gen_eventhandler_h", "-gehh", help="Generate eventhandler header", action="store_true", default=False)
- parser.add_argument("--gen_eventhandlerfile_h", "-gehf", help="Generate eventhandler header for writing to files", action="store_true", default=False)
+ parser = ArgumentParser()
+ parser.add_argument('--proto', '-p', dest="protos", nargs='+', help='Path to all proto file(s) to process. Accepts one or more paths (i.e. events.proto and events_private.proto)', required=True)
+ parser.add_argument('--output-dir', help='Output dir (defaults to ./codegen). Will create folder if it does not exist.', required=False, default='codegen')
args = parser.parse_args()
- proto_filename = args.proto
-
- (output_dir, output_filename) = os.path.split(args.output)
-
- if not output_dir:
- output_dir = "."
-
- #print("output_dir = %s" % output_dir, file=sys.stderr)
- #print("output_filename = %s" % output_filename, file=sys.stderr)
+ if not os.path.exists(args.output_dir):
+ MakeDir(args.output_dir)
- if not os.path.exists(proto_filename):
- print("Error: Could not find proto file %s" % proto_filename, file=sys.stderr)
- return 1
+ for f in args.protos:
+ if not os.path.exists(f):
+ print('Error: Could not find proto file %s' % f, file=sys.stderr)
+ return 1
- protos = parse_protos(proto_filename)
-
- # Generate event header
- if args.gen_event_h:
- curdir = os.path.dirname(os.path.abspath(__file__))
- template_file = os.sep.join([curdir, 'templates', 'ar_event_h.template'])
- output_fullpath = os.sep.join([output_dir, output_filename])
-
- write_template_to_file(template_file, output_fullpath,
- filename=output_filename,
- protos=protos)
-
- # Generate event implementation
- if args.gen_event_cpp:
- curdir = os.path.dirname(os.path.abspath(__file__))
- template_file = os.sep.join([curdir, 'templates', 'ar_event_cpp.template'])
- output_fullpath = os.sep.join([output_dir, output_filename])
-
- write_template_to_file(template_file, output_fullpath,
- filename=output_filename,
- protos=protos)
-
- # Generate event handler header
- if args.gen_eventhandler_h:
- curdir = os.path.dirname(os.path.abspath(__file__))
- template_file = os.sep.join([curdir, 'templates', 'ar_eventhandler_h.template'])
- output_fullpath = os.sep.join([output_dir, output_filename])
+ protos = {}
+ protos['events'] = {} # event dictionary containing events with their fields
+ protos['event_names'] = [] # needed to keep events in order parsed. dict is not ordered.
+ protos['enums'] = {}
+ protos['enum_names'] = []
- write_template_to_file(template_file, output_fullpath,
- filename=output_filename,
- event_header="gen_ar_event.h", # todo: fix this!
- protos=protos)
+ # Parse each proto file and add to protos container
+ for f in args.protos:
+ print("Parsing proto file: %s" % os.path.normpath(f))
+ parse_protos(protos, f)
- # Generate event handler header
- if args.gen_eventhandlerfile_h:
- curdir = os.path.dirname(os.path.abspath(__file__))
- template_file = os.sep.join([curdir, 'templates', 'ar_eventhandlerfile_h.template'])
- output_fullpath = os.sep.join([output_dir, output_filename])
+ files = [
+ ["gen_ar_event.hpp", ""],
+ ["gen_ar_event.cpp", ""],
+ ["gen_ar_eventhandler.hpp", "gen_ar_event.hpp"],
+ ["gen_ar_eventhandlerfile.hpp", "gen_ar_eventhandler.hpp"]
+ ]
- write_template_to_file(template_file, output_fullpath,
- filename=output_filename,
- event_header="gen_ar_eventhandler.h", # todo: fix this!
- protos=protos)
+ rval = 0
- return 0
+ try:
+ # Delete existing files
+ for f in files:
+ filename = f[0]
+ output_fullpath = os.path.join(args.output_dir, filename)
+ if os.path.exists(output_fullpath):
+ print("Deleting existing file: %s" % output_fullpath)
+ os.remove(output_fullpath)
+
+ # Generate files from templates
+ for f in files:
+ filename = f[0]
+ event_header = f[1]
+ curdir = os.path.dirname(os.path.abspath(__file__))
+ template_file = os.path.join(curdir, 'templates', filename)
+ output_fullpath = os.path.join(args.output_dir, filename)
+
+ print("Generating: %s" % output_fullpath)
+ MakoTemplateWriter.to_file(template_file, output_fullpath,
+ cmdline=sys.argv,
+ filename=filename,
+ protos=protos,
+ event_header=event_header)
+
+ except Exception as e:
+ print(e)
+ rval = 1
+
+ return rval
if __name__ == '__main__':
sys.exit(main())