File size: 2,299 Bytes
2130399 |
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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
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
|