Add a script for the ADC2019 system

nszw_solver.py

+import argparse
+import os
+import subprocess
+
+proc = None
+stop_flag = False
+
+def stop():
+    global proc
+    global stop_flag
+
+    stop_flag = True
+    if not proc is None:
+        proc.terminate()
+        print('stopped')
+
+def solve(params):
+    global proc
+    global stop_flag
+
+    stop_flag = False
+
+    # 必要な変数があるか確認
+    assert('problem' in params)
+
+    problem = params['problem']
+
+    pl = problem.splitlines()
+    for l in pl:
+        if "SIZE" in l:
+            board_size_str = l.strip().split()[1]
+            bx, by = [int(v) for v in board_size_str.split('X')]
+            break
+
+    basedir = os.path.abspath(os.path.dirname(__file__))
+
+    cmds = f'{basedir}/solve'.split()
+
+    proc = subprocess.Popen(
+        cmds,
+        stdin=subprocess.PIPE,
+        stderr=subprocess.PIPE,
+        stdout=subprocess.PIPE
+    )
+
+    try:
+        outs, errs = proc.communicate(problem.encode())
+        status = 'done'
+    except Exception as ex:
+        status = 'failed'
+    finally:
+        returncode = proc.returncode
+        outs = outs.decode()
+        errs = errs.decode()
+        print(outs)
+        print(errs)
+
+    solution = ''
+    if returncode == 0:
+        status = 'done'
+
+        lines = outs.splitlines()
+        state = 0
+        for l in lines:
+
+            if "SIZE" in l:
+                board_size_str = l.strip().split()[1]
+                sx, sy = [int(v) for v in board_size_str.split('X')]
+                if sx > by or sy > by:
+                    status = 'failed'
+                    break
+            elif 'ANSWER' in l:
+                state = 1
+                continue
+            elif state == 1:
+                if l.strip() == '':
+                    break
+                else:
+                    if not l.startswith('BLOCK'):
+                        tmp = [int(v.strip()) for v in l.rstrip().split(',')]
+                        for i, v in enumerate(tmp):
+                            if v == -1:
+                                tmp[i] = 0
+                        l = ','.join([str(v) for v in tmp])
+                    solution += l + '\n'
+    else:
+        status = 'failed'
+
+    return {
+        'status': status,
+        'solution': solution
+    }
+
+def main(args):
+
+    with open(args['problem'], 'r') as fp:
+        problem = fp.read()
+    
+    params = {
+        'problem': problem,
+    }
+
+    ret = solve(params)
+
+    print(ret)
+
+if __name__ == "__main__":
+    
+    parser = argparse.ArgumentParser(description='Wrapper Script for ADC2019 Solver')
+    parser.add_argument('problem', type=str, help='File path to the problem file')
+    parser.add_argument('-t', '--timeout', type=int, default=10, help='Timeout for the process of the solver')
+    args = vars(parser.parse_args())
+
+    main(args)