def edges(self): """ overrides the edges() function of digraph, to maintain the nodes mapping @rtype list(edge) @return the edges stored in this graph """ return [(self.nodesMap[u], self.nodesMap[v]) for (u, v) in digraph.edges(self)]
def getJson(self): """ @return: json representation of this graph """ # format: (unique id, (isPredicate, (minIndex, maxIndex))) entities = dict([(uid, {'predicate': bool(node.isPredicate), 'feats': self.getFeatsDic(node), 'charIndices': (0,0) if not node.text else self.nodeToCharIndices(node)}) for uid, node in self.nodesMap.items()]) edges = [(src, dest, self.edge_label((src, dest))) for (src, dest) in digraph.edges(self)] return (entities, edges)
def writeToDot(self, filename, writeLabel): """ @deprecated: use getJson instead Outputs a dot file representing this graph @type filename: string @param filename: the file in which to save the dot text """ dot = pygraph.readwrite.dot.pydot.Dot() if writeLabel: label = "\n".join([self.originalSentence]) dot.set_label(label) dot.set_labelloc("bottom") dot.set_labeljust("center") for uid in self.nodesMap: curNode = self.nodesMap[uid] dotNode = pygraph.readwrite.dot.pydot.Node() dotNode.set_shape(curNode.nodeShape) dotNode.set_name(str(uid)) dotNode.set_label("<{0}>".format(curNode)) if curNode.isPredicate: dotNode.set_color(PREDICATE_COLOR) dotNode.set_fontcolor(PREDICATE_COLOR) if curNode.features.get("top", False): dotNode.set_color(TOPNODE_COLOR) dotNode.set_fontcolor(TOPNODE_COLOR) ##### for debug ##### if curNode.features.has_key("Nominal"): dotNode.set_color("blue") dotNode.set_fontcolor("blue") if curNode.features.has_key("VADAS"): dotNode.set_color("green") dotNode.set_fontcolor("green") if curNode.features.has_key("traces"): dotNode.set_color("orange") dotNode.set_fontcolor("orange") if curNode.features.has_key("LV"): dotNode.set_color("purple") dotNode.set_fontcolor("purple") if curNode.features.has_key("heuristics"): dotNode.set_color("teal") dotNode.set_fontcolor("teal") if curNode.features.has_key("debug"): dotNode.set_color("blue") dotNode.set_fontcolor("blue") dot.add_node(dotNode) for (src, dst) in digraph.edges(self): curEdge = pygraph.readwrite.dot.pydot.Edge(src=src, dst=dst) curEdge.set_fontsize("11") label = self.edge_label((src, dst)) if label: if self.is_aux_edge((src, dst)): curEdge.set_style("dashed") curEdge.set_label(label) dot.add_edge(curEdge) if not filename: return dot dot.write(filename)
def writeToDot(self, filename, writeLabel): """ @deprecated: use getJson instead Outputs a dot file representing this graph @type filename: string @param filename: the file in which to save the dot text """ dot = pygraph.readwrite.dot.pydot.Dot() if writeLabel: label = "\n".join([self.originalSentence]) dot.set_label(label) dot.set_labelloc("bottom") dot.set_labeljust("center") for uid in self.nodesMap: curNode = self.nodesMap[uid] dotNode = pygraph.readwrite.dot.pydot.Node() dotNode.set_shape(curNode.nodeShape) dotNode.set_name(str(uid)) dotNode.set_label("<{0}>".format(curNode)) if curNode.isPredicate: dotNode.set_color(PREDICATE_COLOR) dotNode.set_fontcolor(PREDICATE_COLOR) if curNode.features.get("top", False): dotNode.set_color(TOPNODE_COLOR) dotNode.set_fontcolor(TOPNODE_COLOR) ##### for debug ##### if curNode.features.has_key("Nominal"): dotNode.set_color("blue") dotNode.set_fontcolor("blue") if curNode.features.has_key("VADAS"): dotNode.set_color("green") dotNode.set_fontcolor("green") if curNode.features.has_key("traces"): dotNode.set_color("orange") dotNode.set_fontcolor("orange") if curNode.features.has_key("LV"): dotNode.set_color("purple") dotNode.set_fontcolor("purple") if curNode.features.has_key("heuristics"): dotNode.set_color("teal") dotNode.set_fontcolor("teal") if curNode.features.has_key("debug"): dotNode.set_color("blue") dotNode.set_fontcolor("blue") dot.add_node(dotNode) for (src, dst) in digraph.edges(self): curEdge = pygraph.readwrite.dot.pydot.Edge(src=src, dst=dst) curEdge.set_fontsize("11") label = self.edge_label((src, dst)) if label: if self.is_aux_edge((src, dst)): curEdge.set_style("dashed") curEdge.set_label(label) dot.add_edge(curEdge) if not filename: return dot dot.write(filename)