|
@@ -1,11 +1,11 @@
|
|
|
from _common import sort_by_keywords
|
|
|
|
|
|
-# This hard-coded many times in CppParts in various codegens
|
|
|
-_DEFAULT_CPP_PARTS = 20
|
|
|
-# See TCodegenParams::MethodStream usage in factor codegen
|
|
|
-_ADDITIONAL_STREAM_COUNT = 5
|
|
|
-
|
|
|
+# This hard-coded many times in CppParts in various codegens
|
|
|
+_DEFAULT_CPP_PARTS = 20
|
|
|
+# See TCodegenParams::MethodStream usage in factor codegen
|
|
|
+_ADDITIONAL_STREAM_COUNT = 5
|
|
|
|
|
|
+
|
|
|
def onsplit_codegen(unit, *args):
|
|
|
'''
|
|
|
@usage: SPLIT_CODEGEN(tool prefix opts... [OUT_NUM num] [OUTPUT_INCLUDES output_includes...])
|
|
@@ -19,10 +19,10 @@ def onsplit_codegen(unit, *args):
|
|
|
keywords = {"OUT_NUM": 1}
|
|
|
flat_args, spec_args = sort_by_keywords(keywords, args)
|
|
|
|
|
|
- num_outputs = _DEFAULT_CPP_PARTS + _ADDITIONAL_STREAM_COUNT
|
|
|
- if "OUT_NUM" in spec_args:
|
|
|
- num_outputs = int(spec_args["OUT_NUM"][0])
|
|
|
-
|
|
|
+ num_outputs = _DEFAULT_CPP_PARTS + _ADDITIONAL_STREAM_COUNT
|
|
|
+ if "OUT_NUM" in spec_args:
|
|
|
+ num_outputs = int(spec_args["OUT_NUM"][0])
|
|
|
+
|
|
|
tool = flat_args[0]
|
|
|
prefix = flat_args[1]
|
|
|
|
|
@@ -30,14 +30,14 @@ def onsplit_codegen(unit, *args):
|
|
|
for num in range(num_outputs):
|
|
|
cmd.append('{}.{}.cpp'.format(prefix, num))
|
|
|
|
|
|
- cpp_parts = int(num_outputs) - _ADDITIONAL_STREAM_COUNT
|
|
|
- cpp_parts_args = ['--cpp-parts', str(cpp_parts)]
|
|
|
-
|
|
|
+ cpp_parts = int(num_outputs) - _ADDITIONAL_STREAM_COUNT
|
|
|
+ cpp_parts_args = ['--cpp-parts', str(cpp_parts)]
|
|
|
+
|
|
|
if len(flat_args) > 2:
|
|
|
if flat_args[2] != 'OUTPUT_INCLUDES':
|
|
|
cmd.append('OPTS')
|
|
|
- cmd += cpp_parts_args + flat_args[2:]
|
|
|
- else:
|
|
|
- cmd += ['OPTS'] + cpp_parts_args
|
|
|
+ cmd += cpp_parts_args + flat_args[2:]
|
|
|
+ else:
|
|
|
+ cmd += ['OPTS'] + cpp_parts_args
|
|
|
|
|
|
unit.on_split_codegen_base(cmd)
|