fix merge conflicts

fastplotlib · kushalkolar · Dec 11, 2022 · Oct 11, 2022 · Nov 19, 2022 · Nov 19, 2022
commit 7630c9711e56eb9530f55bbbca65c90c6ce234be
diff --git a/fastplotlib/layouts/_subplot.py b/fastplotlib/layouts/_subplot.py
@@ -6,6 +6,7 @@
 from wgpu.gui.auto import WgpuCanvas
 from warnings import warn
 import numpy as np
+from math import copysign
 
 
 class Subplot:
@@ -113,8 +114,13 @@ def animate(self, canvas_dims: Tuple[int, int] = None):
         for dv in self.docked_viewports.values():
             dv.render()
 
-    def add_animations(self, funcs: List[callable]):
-        self._animate_funcs += funcs
+    def add_animations(self, *funcs: callable):
+        for f in funcs:
+            if not callable(f):
+                raise TypeError(
+                    f"all positional arguments to add_animations() must be callable types, you have passed a: {type(f)}"
+                )
+            self._animate_funcs += funcs
 
     def add_graphic(self, graphic, center: bool = True):
         if graphic.name is not None:  # skip for those that have no name
@@ -182,9 +188,33 @@ def set_grid_visibility(self, visible: bool):
         else:
             self.scene.remove(self._grid)
 
+    def get_graphics(self):
+        return self._graphics
+
     def remove_graphic(self, graphic):
         self.scene.remove(graphic.world_object)
 
+    def __getitem__(self, name: str):
+        for graphic in self._graphics:
+            if graphic.name == name:
+                return graphic
+
+        graphic_names = list()
+        for g in self._graphics:
+            graphic_names.append(g.name)
+        raise IndexError(f"no graphic of given name, the current graphics are:\n {graphic_names}")
+
+    def __repr__(self):
+        newline = "\n  "
+        if self.name is not None:
+            return f"'{self.name}' fastplotlib.{self.__class__.__name__} @ {hex(id(self))}\n" \
+                   f"Graphics: \n  " \
+                   f"{newline.join(graphic.__repr__() for graphic in self.get_graphics())}"
+        else:
+            return f"fastplotlib.{self.__class__.__name__} @ {hex(id(self))} \n" \
+                   f"Graphics: \n  " \
+                   f"{newline.join(graphic.__repr__() for graphic in self.get_graphics())}"
+
 
 class _DockedViewport:
     _valid_positions = [