def latex(text):
amr = AMR(text)
text = str(amr)
for x in re.findall('x[0-9]+ ?/ ?[^()\s]+', text):
text = text.replace(x, '(' + x + ')')
edges = [(e, id) for e, id in zip(amr.edges(), amr.edge_ids())]
elems = []
max_depth = paren_utils.max_depth(text)
prev_depth = 0
depth = 0
i = 0
node_depth = {}
for t in paren_utils.paren_iter(text):
node = amr.NODE_RE.match(t).group()
id = node.split('/')[0].strip()
# clean node
if re.match('x[0-9]+/', node):
node = node.split('/')[1]
node = node.replace('"', '``', 1).replace('"', "''", 1)
prev_depth = depth
depth = paren_utils.depth_at(text, text.index(t))
if depth > prev_depth:
i = 0
node_depth[id] = depth
num_nodes = paren_utils.mark_depth(text).count(f'<{depth}>')
x = AMR_Latex.get_x(i, num_nodes)
y = AMR_Latex.get_y(depth, max_depth)
color = AMR_Latex.get_color(i)
elems.append(f'\t\\node[{color}]({id}) at ({x},{y}) {{{node}}};')
i += 1
for edge, id in edges:
source = id.split('_')[0]
target = id.split('_')[2]
dir1 = 'south'
dir2 = 'north'
if node_depth[source] > node_depth[target]:
dir1 = 'north'
dir2 = 'south'
if node_depth[source] == node_depth[target]:
dir1 = 'north'
dir2 = 'north'
elems.append(
f'\t\draw[->, thick] ({source}.{dir1}) -- ({target}.{dir2}) node[midway, above, sloped] {{{edge}}};'
)
latex = '\n\\begin{tikzpicture}[\n'
latex += 'red/.style={rectangle, draw=red!60, fill=red!5, very thick, minimum size=7mm},\n'
latex += 'blue/.style={rectangle, draw=blue!60, fill=blue!5, very thick, minimum size=7mm},\n'
latex += 'green/.style={rectangle, draw=green!60, fill=green!5, very thick, minimum size=7mm},\n'
latex += 'purple/.style={rectangle, draw=purple!60, fill=purple!5, very thick, minimum size=7mm},\n'
latex += 'orange/.style={rectangle, draw=orange!60, fill=orange!5, very thick, minimum size=7mm},\n'
latex += ']\n'
latex += '\n'.join(elems)
latex += '\n\end{tikzpicture}\n'
return latex
def normalize_entity(root, nodes, edges):
normalize_ids = {
id: i
for i, id in enumerate(sorted(nodes, key=lambda x: nodes[x]))
}
normalized_entity = AMR()
for n in nodes:
normalized_entity.nodes[normalize_ids[n]] = nodes[n]
for s, r, t in edges:
normalized_entity.edges.append((normalize_ids[s], r, normalize_ids[t]))
normalized_entity.edges = sorted(normalized_entity.edges)
normalized_entity.root = normalize_ids[root]
return normalized_entity
