mirror of
https://github.com/AxioDL/CodeGen.git
synced 2026-03-30 11:45:23 -07:00
Sam Fuller
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
import sys
|
|
import argparse
|
|
from typing import List
|
|
|
|
from codegen import codegen
|
|
|
|
|
|
class ArgumentData(object):
|
|
def __init__(self) -> None:
|
|
self.command = ''
|
|
self.source_file = ''
|
|
self.libclangpath = ''
|
|
self.include_paths: List[str] = []
|
|
self.source_root = ''
|
|
self.output_root = ''
|
|
self.cache_path = None
|
|
|
|
|
|
argparser = argparse.ArgumentParser()
|
|
argparser.add_argument('command', help='What to do')
|
|
argparser.add_argument('source_file', help='Source file to use')
|
|
argparser.add_argument('--libclangpath', help='Path to libclang library file', required=True)
|
|
argparser.add_argument('--include', '-I', help='Define an include path', dest='include_paths', action='append')
|
|
argparser.add_argument('--source-root', help='Root path of all source files', required=True)
|
|
argparser.add_argument('--output-root', help='Root path of all files to be output', required=True)
|
|
argparser.add_argument('--cache-path', help='Path to a directory to store output file caches in')
|
|
|
|
|
|
def main() -> int:
|
|
args = ArgumentData()
|
|
|
|
# noinspection PyTypeChecker
|
|
argparser.parse_args(namespace=args)
|
|
|
|
if args.command.lower() == 'get_output_files':
|
|
return do_get_output_files_command(args)
|
|
if args.command.lower() == 'generate':
|
|
return do_generate_command(args)
|
|
|
|
sys.stderr.write('Invalid Command\n')
|
|
return 1
|
|
|
|
|
|
def do_get_output_files_command(args: ArgumentData) -> int:
|
|
output_files = codegen.get_output_files(
|
|
args.source_file,
|
|
args.include_paths,
|
|
args.libclangpath,
|
|
args.source_root,
|
|
args.output_root,
|
|
args.cache_path
|
|
)
|
|
sys.stdout.write(';'.join(output_files))
|
|
return 0
|
|
|
|
|
|
def do_generate_command(args: ArgumentData) -> int:
|
|
codegen.RunCodegen(args.source_file, args.include_paths, args.libclangpath, args.source_root, args.output_root)
|
|
return 0
|