diff --git a/nszw_solver.py b/nszw_solver.py new file mode 100644 index 0000000000000000000000000000000000000000..2607cc922306db9e1ba02ff12e0a21283743b14d --- /dev/null +++ b/nszw_solver.py @@ -0,0 +1,114 @@ +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)