Implement layouts

MarkKoz · 64json · commit 2cf1c9d2de7b · 2019-06-11T15:50:09.000+09:00
diff --git a/algorithm_visualizer/__init__.py b/algorithm_visualizer/__init__.py
@@ -5,11 +5,13 @@
 
 from algorithm_visualizer import randomize as Randomize
 from algorithm_visualizer.commander import Commander
+from algorithm_visualizer.layouts import *
 from algorithm_visualizer.tracers import *
 
 __all__ = (
     "Randomize", "Commander",
-    "Array1DTracer", "Array2DTracer", "ChartTracer", "GraphTracer", "LogTracer", "Tracer"
+    "Array1DTracer", "Array2DTracer", "ChartTracer", "GraphTracer", "LogTracer", "Tracer",
+    "HorizontalLayout", "Layout", "VerticalLayout"
 )
 
 # Types which are serializable by the default JSONEncoder
diff --git a/algorithm_visualizer/layouts.py b/algorithm_visualizer/layouts.py
@@ -0,0 +1,31 @@
+from typing import List
+
+from algorithm_visualizer import Commander
+
+class Layout(Commander):
+    def __init__(self, children: List[Commander]):
+        super().__init__([c.key for c in children])
+
+    @classmethod
+    def setRoot(cls, child: Commander):
+        cls._command(None, "setRoot", child.key)
+
+    def add(self, child: Commander, index: int = None):
+        if index is None:
+            self.command("add", child.key)
+        else:
+            self.command("add", child.key, index)
+
+    def remove(self, child: Commander):
+        self.command("remove", child.key)
+
+    def removeAll(self):
+        self.command("removeAll")
+
+
+class HorizontalLayout(Layout):
+    pass
+
+
+class VerticalLayout(Layout):
+    pass
