aboutsummaryrefslogtreecommitdiffstats
path: root/build/plugins/split_codegen.py
diff options
context:
space:
mode:
authoralexv-smirnov <alex@ydb.tech>2023-03-15 19:59:12 +0300
committeralexv-smirnov <alex@ydb.tech>2023-03-15 19:59:12 +0300
commit056bb284ccf8dd6793ec3a54ffa36c4fb2b9ad11 (patch)
tree4740980126f32e3af7937ba0ca5f83e59baa4ab0 /build/plugins/split_codegen.py
parent269126dcced1cc8b53eb4398b4a33e5142f10290 (diff)
downloadydb-056bb284ccf8dd6793ec3a54ffa36c4fb2b9ad11.tar.gz
add library/cpp/actors, ymake build to ydb oss export
Diffstat (limited to 'build/plugins/split_codegen.py')
-rw-r--r--build/plugins/split_codegen.py43
1 files changed, 43 insertions, 0 deletions
diff --git a/build/plugins/split_codegen.py b/build/plugins/split_codegen.py
new file mode 100644
index 0000000000..f1e60bc142
--- /dev/null
+++ b/build/plugins/split_codegen.py
@@ -0,0 +1,43 @@
+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
+
+
+def onsplit_codegen(unit, *args):
+ '''
+ @usage: SPLIT_CODEGEN(tool prefix opts... [OUT_NUM num] [OUTPUT_INCLUDES output_includes...])
+
+ Generator of a certain number of parts of the .cpp file + one header .h file from .in
+
+ Supports keywords:
+ 1. OUT_NUM <the number of generated Prefix.N.cpp default 25 (N varies from 0 to 24)>
+ 2. OUTPUT_INCLUDES <path to files that will be included in generalnyj of macro files>
+ '''
+ 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])
+
+ tool = flat_args[0]
+ prefix = flat_args[1]
+
+ cmd = [tool, prefix, 'OUT']
+ 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)]
+
+ 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
+
+ unit.on_split_codegen_base(cmd)