Implement code expander

.github/workflows/ci.yml

@@ -95,3 +95,21 @@ jobs:
           args: --workspace --no-fail-fast
         env:
           RUST_BACKTRACE: full
+
+  expander_test:
+    strategy:
+      fail-fast: false
+      matrix:
+        toolchain:
+          - 1.42.0-x86_64-unknown-linux-gnu
+          - stable-x86_64-unknown-linux-gnu
+
+    name: Expand_test (${{ matrix.toolchain }})
+    runs-on: ubuntu-18.04
+
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v2
+
+      - name: expand.py tests
+        run: bash ./.github/workflows/test-expand.sh

.github/workflows/test-expand.sh

@@ -0,0 +1,25 @@
+#!/bin/bash
+
+TEST_MODULES=(convolution dsu fenwicktree lazysegtree math maxflow mincostflow modint scc segtree string twosat --all)
+TMP_PATH=$(mktemp -d)
+SCRIPT_DIR=$(cd $(dirname $0); pwd)
+TEST_FILE="test.rs"
+FILE_HEAD="fn main() {}"
+
+if [ -v CI ];then
+    rustup component add rustfmt 
+fi
+
+for MODULE in ${TEST_MODULES[@]};do
+     echo Test module $MODULE 
+     python3 $SCRIPT_DIR/../../expand.py $MODULE > $TMP_PATH/$TEST_FILE
+     echo "Output" $(wc -c < $TMP_PATH/$TEST_FILE) "Bytes"
+     echo $FILE_HEAD >> $TMP_PATH/$TEST_FILE
+     rustc -A warnings $TMP_PATH/$TEST_FILE 
+     if [ $? -ne 0 ];then
+        echo "Error compiling for $MODULE"
+        exit 1
+    else
+        echo "Test passed"
+     fi
+done

expand.py

@@ -0,0 +1,107 @@
+#!/usr/bin/python3
+
+import sys
+import getopt
+import tempfile
+import subprocess
+
+usage = '''Usage:expand.py [options] <output modules>
+Output Modules:
+    convolution
+    dsu
+    fenwicktree
+    lazysegtree
+    math
+    maxflow
+    mincostflow
+    modint
+    scc
+    segtree
+    string
+    twosat
+
+You can select multiple modules for <output modules>
+    e.g.)expand.py math segtree
+
+Options:
+    -a  --all       import all modules   
+    -h  --help      print help
+'''
+output_header = '//https://github.com/rust-lang-ja/ac-library-rs\n'
+opt_list = ['help', 'all']
+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', 'internal_type_traits',), 'mincostflow': ('internal_type_traits',)}
+src_path = 'src/'
+
+
+def output_file(filename):
+    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
+
+
+try:
+    opts, args = getopt.getopt(sys.argv[1:], 'ah', opt_list)
+except getopt.GetoptError as e:
+    print(e)
+    print(usage)
+    sys.exit(2)
+
+if len(opts) == 0 and len(args) == 0:
+    print(usage)
+    sys.exit(0)
+
+for o, v in opts:
+    if o == '--help' or o == '-h':
+        print(usage)
+        sys.exit(0)
+    elif o == '--all' or o == '-a':
+        args = list(output_list_all)
+
+output_list = set()
+
+while len(args) != 0:
+    pop = args.pop()
+    if not pop in output_list_all:
+        print('invalid args:{}'.format(pop))
+        print(usage)
+        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_data.extend(buf)
+
+for i in output_list:
+    # Modules that begin with 'internal' are for internal use, so they are not declared.
+    if not i.startswith('internal'):
+        output_data.append('use {}::*;'.format(i))
+
+# rustfmt
+with tempfile.TemporaryDirectory() as temp_dir:
+    temp_file = temp_dir + '/output.rs'
+    with open(temp_file, 'w') as f:
+        print(output_header, file=f)
+        for i in output_data:
+            print(i, file=f)
+    output_data = subprocess.run(["rustfmt", temp_file])
+    with open(temp_file, 'r') as f:
+        for line in f:
+            print(line, end="")