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))