Implement GraphTracer

Author: MarkKoz

algorithm_visualizer/__init__.py

@@ -9,7 +9,7 @@
 
 __all__ = (
     "Randomize", "Commander",
-    "Array1DTracer", "Array2DTracer", "ChartTracer", "LogTracer", "Tracer"
+    "Array1DTracer", "Array2DTracer", "ChartTracer", "GraphTracer", "LogTracer", "Tracer"
 )
 
 # Types which are serializable by the default JSONEncoder

algorithm_visualizer/tracers/__init__.py

@@ -1,7 +1,8 @@
 from .array1d import Array1DTracer
 from .array2d import Array2DTracer
 from .chart import ChartTracer
+from .graph import GraphTracer
 from .log import LogTracer
 from .tracer import Tracer
 
-__all__ = ("Array1DTracer", "Array2DTracer", "ChartTracer", "LogTracer", "Tracer")
+__all__ = ("Array1DTracer", "Array2DTracer", "ChartTracer", "GraphTracer", "LogTracer", "Tracer")

algorithm_visualizer/tracers/graph.py

@@ -0,0 +1,101 @@
+from typing import Any, List, Optional
+
+from algorithm_visualizer import LogTracer, Tracer, _Number, _Serializable
+
+class GraphTracer(Tracer):
+    def set(self, array2d: List[List[_Serializable]] = None):
+        if array2d is None:
+            array2d = []
+        self.command("set", array2d)
+
+    def directed(self, isDirected: bool = True):
+        self.command("directed", isDirected)
+
+    def weighted(self, isWeighted: bool = True):
+        self.command("weighted", isWeighted)
+        return self
+
+    def layoutCircle(self):
+        self.command("layoutCircle")
+        return self
+
+    def layoutTree(self, root: Any = 0, sorted: bool = False):
+        self.command("layoutTree", root, sorted)
+        return self
+
+    def layoutRandom(self):
+        self.command("layoutRandom")
+        return self
+
+    def addNode(
+        self,
+        id: _Serializable,
+        weight: Optional[_Number] = None,
+        x: int = 0,
+        y: int = 0,
+        visitedCount: int = 0,
+        selectedCount: int = 0
+    ):
+        self.command("addNode", id, weight, x, y, visitedCount, selectedCount)
+
+    def updateNode(
+        self,
+        id: _Serializable,
+        weight: Optional[_Number] = None,
+        x: int = 0,
+        y: int = 0,
+        visitedCount: int = 0,
+        selectedCount: int = 0
+    ):
+        self.command("updateNode", id, weight, x, y, visitedCount, selectedCount)
+
+    def removeNode(self, id: _Serializable):
+        self.command("removeNode", id)
+
+    def addEdge(
+        self,
+        source: _Serializable,
+        target: _Serializable,
+        weight: Optional[_Number] = None,
+        visitedCount: int = 0,
+        selectedCount: int = 0
+    ):
+        self.command("addEdge", source, target, weight, visitedCount, selectedCount)
+
+    def updateEdge(
+        self,
+        source: _Serializable,
+        target: _Serializable,
+        weight: Optional[_Number] = None,
+        visitedCount: int = 0,
+        selectedCount: int = 0
+    ):
+        self.command("updateEdge", source, target, weight, visitedCount, selectedCount)
+
+    def removeEdge(self, source: _Serializable, target: _Serializable):
+        self.command("removeEdge", source, target)
+
+    def visit(
+        self,
+        target: _Serializable,
+        source: _Serializable = None,
+        weight: Optional[_Number] = None
+    ):
+        self.command("visit", target, source, weight)
+
+    def leave(
+        self,
+        target: _Serializable,
+        source: _Serializable = None,
+        weight: Optional[_Number] = None
+    ):
+        self.command("leave", target, source, weight)
+
+    def select(self, target: _Serializable, source: _Serializable = None):
+        self.command("select", target, source)
+
+    def deselect(self, target: _Serializable, source: _Serializable = None):
+        self.command("deselect", target, source)
+
+    def log(self, log: LogTracer):
+        self.command("log", log.key)