wip

fastplotlib · clewis7 · Dec 12, 2022 · Dec 12, 2022 · Dec 14, 2022 · Dec 15, 2022
commit 24cf1164a6fac64d36ac6f00b9aac1735476bb0e
diff --git a/fastplotlib/graphics/line.py b/fastplotlib/graphics/line.py
@@ -1,6 +1,7 @@
 from typing import *
 import numpy as np
 import pygfx
+from typing import *

 from ._base import Graphic

@@ -58,4 +59,67 @@ def __init__(
            material=material(thickness=size, vertex_colors=True)
        )

+<<<<<<< HEAD
        self.world_object.position.z = z_position
+=======
+        self.events = {}
+
+    def fix_data(self):
+        # TODO: data should probably be a property of any Graphic?? Or use set_data() and get_data()
+        if self.data.ndim == 1:
+            self.data = np.dstack([np.arange(self.data.size), self.data])[0]
+
+        if self.data.shape[1] != 3:
+            if self.data.shape[1] != 2:
+                raise ValueError("Must pass 1D, 2D or 3D data")
+            # make it 2D with zlevel
+            if self.zlevel is None:
+                self.zlevel = 0
+
+            # zeros
+            zs = np.full(self.data.shape[0], fill_value=self.zlevel, dtype=np.float32)
+
+            self.data = np.dstack([self.data[:, 0], self.data[:, 1], zs])[0]
+
+    def update_data(self, data: np.ndarray):
+        self.data = data.astype(np.float32)
+        self.fix_data()
+
+        self.world_object.geometry.positions.data[:] = self.data
+        self.world_object.geometry.positions.update_range()
+
+    def update_colors(self, colors: np.ndarray):
+        super(LineGraphic, self)._set_colors(colors=colors, colors_length=self.data.shape[0], cmap=None, alpha=None)
+
+        self.world_object.geometry.colors.data[:] = self.colors
+        self.world_object.geometry.colors.update_range()
+
+    @property
+    def indices(self) -> Any:
+        return self.indices
+
+    @property
+    def features(self) -> List[str]:
+        return self.features
+
+    def _set_feature(self, name: str, new_data: Any, indices: Any):
+        if name == "color":
+            self.update_colors(new_data)
+        elif name == "data":
+            self.update_data(new_data)
+        else:
+            raise ValueError("name arg is not a valid feature")
+
+    def link(self, event: str, target: Graphic, feature: str, new_data: Any, indices_mapper: callable = None):
+        valid_events = ["click"]
+        if event in valid_events:
+            self.world_object.add_event_handler(self.event_handler, event)
+            self.events[event] = list()
+            self.events[event].append((target, feature, new_data))
+        else:
+            raise ValueError("event not possible")
+
+    def event_handler(self, event):
+        for event in self.events[event]:
+            event[0]._set_feature(name=event[1], new_data=event[2], indices=None)
+>>>>>>> f19c9c1 (wip)