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
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
