class Aresta(): def __init__(self, nome, custo, vertice1, vertice2): self.__nome = nome self.__custo = custo self.__vertice1 = vertice1 self.__vertice2 = vertice2 self.__aresta = Edge(str(vertice1.getNome()), str(vertice2.getNome())) self.__aresta.set_label(custo) self.__aresta.set_labelfontcolor("#009933") self.__aresta.set_fontsize(10.0) self.__aresta.set_color("#38516b") def setNome(self, nome): self.__nome = nome def setCusto(self, custo): self.__custo = custo def getNome(self): return self.__nome def getCusto(self): return self.__custo def getVertice1(self): return self.__vertice1 def getVertice2(self): return self.__vertice2 def getVertices(self): return [self.__vertice1, self.__vertice2] def getAresta(self): return self.__aresta
def add_edge_to_parent(node, is_successor=False, parent=None): if parent is None: parent = node.parent g_node = add_node(node, in_successors=is_successor) g_parent_node = add_node(parent) edge = Edge(g_parent_node, g_node, label=node.action_representation(), fontsize=self.font_size) if is_successor: edge.set_color(self.successor_color) edge.set_labelfontcolor(self.successor_color) graph_edges[id(node), id(parent)] = edge