diff --git a/bridges/edge.py b/bridges/edge.py index dbf90b9..595f8f8 100644 --- a/bridges/edge.py +++ b/bridges/edge.py @@ -14,7 +14,7 @@ class Edge(): - def __init__(self, v1, v2, data=None) -> None: + def __init__(self, v1, v2, data=None, label: str = None, color: Color = None, thickness: float = None) -> None: """ Constructor for a edge Args: @@ -29,6 +29,15 @@ def __init__(self, v1, v2, data=None) -> None: self._edge_data = data self._lvis = LinkVisualizer() + if label: + self.label = label + + if color: + self.color = color + + if thickness: + self.thickness = thickness + @property def tov(self): """ diff --git a/bridges/graph_adj_list.py b/bridges/graph_adj_list.py index e3c6e6e..1808648 100644 --- a/bridges/graph_adj_list.py +++ b/bridges/graph_adj_list.py @@ -115,8 +115,8 @@ def add_edge(self, " does not exist! Add the vertex before creating the edge.") except Exception as e: traceback.print_tb(e.__traceback__) - self.adj_list[source_id] = SLelement(e=Edge(source_id, dest_id, data), next=self.adj_list.get(src), label=label, - color=color, thickness=thickness, opacity=opacity) + self.adj_list[source_id] = SLelement(e=Edge(source_id, dest_id, data, label=label, color=color, + thickness=thickness), next=self.adj_list.get(src)) return self.adj_list[source_id].value