Implemented expand.py

Author: manta1130

expand.py

@@ -0,0 +1,69 @@
+#!/usr/bin/python3
+
+import sys
+import getopt
+
+
+def output_file(filename, output_comment, output_test):
+    global src_path
+
+    res = []
+    with open(src_path+filename+'.rs', 'r') as f:
+        res.append('mod {}{{'.format(filename))
+
+        for line in f:
+            res.append(line.rstrip())
+
+        res.append('}')
+    return res
+
+
+opt_list = ['output-comment', 'output-test', ]
+output_list_all = ('lazysegtree', 'segtree', 'convolution', 'twosat', 'scc',
+                   'fenwicktree', 'math', 'modint', 'maxflow', 'dsu', 'mincostflow', 'string', 'internal_bit', 'internal_math', 'internal_type_traits', 'internal_scc', 'internal_queue')
+dependency_list = {'lazysegtree': ('internal_bit',), 'segtree': ('internal_bit',), 'convolution': ('internal_bit,modint',), 'math': ('internal_math',), 'modint': (
+    'internal_math', 'internal_type_traits'), 'fenwicktree': ('internal_type_traits',), 'twosat': ('internal_scc',), 'scc': ('internal_scc',), 'maxflow': ('internal_queue',)}
+src_path = 'src/'
+
+try:
+    opts, args = getopt.getopt(sys.argv[1:], 'tc', opt_list)
+except getopt.GetoptError as e:
+    print(e)
+    sys.exit(2)
+
+# unimplemented
+output_comment = False
+output_test = False
+
+for o, v in opts:
+    if o == '--output-comment' or o == '-c':
+        output_comment = True
+    if o == '--output-test' or o == '-t':
+        output_test = True
+
+output_list = set()
+
+while len(args) != 0:
+    pop = args.pop()
+    if not pop in output_list_all:
+        print('invalid args:{}'.format(pop))
+        sys.exit(2)
+    output_list.add(pop)
+    if pop in dependency_list:
+        for d in dependency_list[pop]:
+            args.append(d)
+
+output_list = list(output_list)
+output_list.sort()
+
+output_data = []
+for i in output_list:
+    buf = output_file(i, output_comment, output_test)
+    output_data.extend(buf)
+
+for i in output_list:
+    if not i.startswith('internal'):
+        output_data.append('use {}::*;'.format(i))
+
+for i in output_data:
+    print(i)