File size: 754 Bytes
7145fd6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
from .parser import parse_prefix_to_sympy
if __name__ == "__main__":
import argparse
parser = argparse.ArgumentParser("Parse result prefix to equation")
parser.add_argument("--input", required=True, help="Input result file")
args = parser.parse_args()
res_list = []
with open(args.input, 'r') as f:
for line in f:
comps = line.strip().split("\t")
if line[0] == 'H':
num = int(comps[0][2:])
tokens = comps[2].split(" ")
res_list.append((num, tokens))
for n, toks in res_list:
try:
ex = parse_prefix_to_sympy(toks)
print(n, ex)
except Exception as e:
print(n, "could not parse:", str(e))
|