pysoarlib/util/parse_wm_printout.py

def parse_wm_printout(text):
    """ Given a printout of soar's working memory, parses it into a dictionary of wmes, 
        Where the keys are identifiers, and the values are lists of wme triples rooted at that id

    :param text: The output of a soar print command for working memory
    :type text: str

    :returns dict{ str, list[ (str, str, str) ] }

    """

    ### First: preprocess the output into a string of tokens
    tokens = []
    quote = None
    for word in text.split():
        # Handle quoted strings (Between | |)
        if word[0] == '|':
            quote = word
        elif quote is not None:
            quote += ' ' + word
        if quote is not None:
            if len(quote) > 1 and quote.endswith('|'):
                tokens.append(quote)
                quote = None
            elif len(quote) > 1 and quote.endswith('|)'):
                tokens.append(quote[:-1])
                quote = None
            continue

        # Ignore operator preferences
        if word in [ '+', '>', '<', '!', '=' ]:
            continue
        # Ignore activation values [+23.000]
        if word.startswith("[+") and (word.endswith("]") or word.endswith("])")):
            continue
        # Ignore singleton lti's  (@12533)
        if word.startswith("(@") and word.endswith(")"):
            continue
        # Strip opening parens but add $ to indicate identifier
        if word.startswith("("):
            word = '$' + word[1:]

        # Don't care about closing parens
        word = word.replace(")", "")
        tokens.append(word)

    wmes = dict()
    cur_id = None
    cur_att = None
    cur_wmes = []
    i = 0

    for token in tokens:
        if len(token) == 0:
            continue
        # Identifier
        if token[0] == '$':
            cur_id = token[1:]
            cur_att = None
            cur_wmes = []
            wmes[cur_id] = cur_wmes
        # Attribute
        elif token[0] == '^':
            cur_att = token[1:]
        # Value
        elif cur_id is None:
            print("ERROR: Value " + token + " encountered with no id")
        elif cur_att is None:
            print("ERROR: Value " + token + " encountered with no attribute")
        else:
            cur_wmes.append( (cur_id, cur_att, token) )
    
    return wmes