cmd_parser.py

import re
import sys

NUMBER_REGEX = r"(?:-?(?:(?:[1-9][0-9]*)|[0-9])(?:\.[0-9]+)?(?:e-?[0-9]+)?)|nan|inf|-inf"
CHAR_REGEX = r"[^'\\]|(?:\\(?:['\\ntr]|x[0-9A-Fa-f]{2}))"
CHAR2_REGEX = r"[^\"\\]|(?:\\(?:[\"\\ntr]|x[0-9A-Fa-f]{2}))"
STRING_REGEX = r"(?:'(?:%s)*')|(?:\"(?:%s)*\")" % (CHAR_REGEX, CHAR2_REGEX)
BOOLEAN_REGEX = r"True|False"
REF_REGEX = r"\*[0-9]+"
NONE_REGEX = r"None"
ARGUMENT_REGEX = re.compile("(%s|%s|%s|%s|%s)[,)]" % (
    NUMBER_REGEX,
    STRING_REGEX,
    BOOLEAN_REGEX,
    REF_REGEX,
    NONE_REGEX
))
LINE_START_REGEX = re.compile(r"([A-Z_]+)\(")

class HeapRef(object):
    def __init__(self, id):
        self.id = id

    def serialize_member(self):
        return '^' + str(self.id)

    def __repr__(self):
        return f"HeapRef({self.id})"
    
    def __eq__(self, other):
        return self.id == other.id

    def __hash__(self):
        return self.id

def parse_value(value):
    if value[0] == "*":
        return HeapRef(int(value[1:]))
    elif value == "nan":
        return float("nan")
    elif value == "inf":
        return float('inf')
    elif value == "-inf":
        return float('-inf')
    else:
        return eval(value)

def parse_line(line):
    args = []
    m = LINE_START_REGEX.match(line)
    if m is None:
        raise Exception("Parse error on line '%s'" % line)
    fun_name = m.group(1)
    start_idx = m.span()[1]
    while True:
        m = ARGUMENT_REGEX.match(line, start_idx)
        if m is None:
            break
        value = parse_value(m.group(1))
        args.append(value)
        if m.span()[1] == len(line):
            break
        if line[m.span()[1]] == "\n":
            break 
        assert line[m.span()[1]] == " "
        start_idx = m.span()[1] + 1
    return (fun_name, args)

def main():
    if len(sys.argv) < 2:
        print("Please provide a .rewind file.")
        exit(1)
    filename = sys.argv[1]
    file = open(filename, "r")
    count = 0
    file_iterator = iter(file)
    try:
        while True:
            line = next(file_iterator)
            if line.startswith("--"):
                continue
            command = parse_line(line)
            count += 1
            # print("\r%d" % count, end="")
    except StopIteration:
        pass
    file.close()

if __name__ == '__main__':
    main()