fastplotlib · clewis7 · Dec 12, 2022 · Dec 12, 2022 · Dec 14, 2022 · Dec 15, 2022
diff --git a/examples/linecollection_event.ipynb b/examples/linecollection_event.ipynb
diff --git a/examples/lineplot.ipynb b/examples/lineplot.ipynb
@@ -178,7 +178,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.5"
+   "version": "3.9.2"
   }
  },
  "nbformat": 4,

diff --git a/fastplotlib/graphics/__init__.py b/fastplotlib/graphics/__init__.py
@@ -12,6 +12,6 @@
     "LineGraphic",
     "HistogramGraphic",
     "HeatmapGraphic",
-    "LineCollection",
-    "TextGraphic"
+    "TextGraphic",
+    "LineCollection"
 ]
diff --git a/fastplotlib/graphics/_base.py b/fastplotlib/graphics/_base.py
@@ -5,6 +5,8 @@
 from ..utils import get_colors
 from .features import GraphicFeature, DataFeature, ColorFeature, PresentFeature
 
+from abc import ABC, abstractmethod
+from dataclasses import dataclass
 
 class Graphic:
     def __init__(
@@ -46,6 +48,7 @@ def __init__(
         self.colors = None
 
         self.name = name
+        self.registered_callbacks = dict()
 
         if n_colors is None:
             n_colors = self.data.feature_data.shape[0]
@@ -104,3 +107,74 @@ def __repr__(self):
             return f"'{self.name}' fastplotlib.{self.__class__.__name__} @ {hex(id(self))}"
         else:
             return f"fastplotlib.{self.__class__.__name__} @ {hex(id(self))}"
+
+
+class Interaction(ABC):
+    @property
+    def indices(self) -> Any:
+        return self.indices
+
+    @indices.setter
+    def indices(self, indices: Any):
+        self.indices = indices
+
+    @property
+    @abstractmethod
+    def features(self) -> List[str]:
+        pass
+
+    @abstractmethod
+    def _set_feature(self, feature: str, new_data: Any, indices: Any):
+        pass
+
+    @abstractmethod
+    def _reset_feature(self, feature: str, old_data: Any):
+        pass
+
+    def link(self, event_type: str, target: Any, feature: str, new_data: Any, indices_mapper: callable = None):
+        valid_events = ["click"]
+        if event_type in valid_events:
+            self.world_object.add_event_handler(self.event_handler, event_type)
+        else:
+            raise ValueError("event not possible")
+
+        if isinstance(target.data, List):
+            old_data = list()
+            for line in target.data:
+                old_data.append(getattr(line, feature).copy())
+        else:
+            old_data = getattr(target, feature).copy()
+
+        if event_type in self.registered_callbacks.keys():
+            self.registered_callbacks[event_type].append(
+                CallbackData(target=target, feature=feature, new_data=new_data, old_data=old_data, indices_mapper=indices_mapper))
+        else:
+            self.registered_callbacks[event_type] = list()
+            self.registered_callbacks[event_type].append(
+                CallbackData(target=target, feature=feature, new_data=new_data, old_data=old_data, indices_mapper=indices_mapper))
+
+    def event_handler(self, event):
+        if event.type == "click":
+            # storing click information for each click in self.indices
+            #self.indices(np.array(event.pick_info["index"]))
+            click_info = np.array(event.pick_info["index"])
+        if event.type in self.registered_callbacks.keys():
+            for target_info in self.registered_callbacks[event.type]:
+                # need to map the indices to the target using indices_mapper
+                if target_info.indices_mapper is not None:
+                    indices = target_info.indices_mapper(target=target_info.target, indices=click_info)
+                else:
+                    indices = None
+                # reset feature of target using stored old data
+                target_info.target._reset_feature(feature=target_info.feature, old_data=target_info.old_data)
+                # set feature of target at indice using new data
+                target_info.target._set_feature(feature=target_info.feature, new_data=target_info.new_data[indices], indices=indices)
+
+@dataclass
+class CallbackData:
+    """Class for keeping track of the info necessary for interactivity after event occurs."""
+    target: Any
+    feature: str
+    new_data: Any
+    old_data: Any
+    indices_mapper: callable = None
diff --git a/fastplotlib/graphics/heatmap.py b/fastplotlib/graphics/heatmap.py
@@ -53,24 +53,18 @@ def __init__(
     ):
         """
         Create a Heatmap Graphic
-
         Parameters
         ----------
         data: array-like, must be 2-dimensional
             | array-like, usually numpy.ndarray, must support ``memoryview()``
             | Tensorflow Tensors also work _I think_, but not thoroughly tested
-
         vmin: int, optional
             minimum value for color scaling, calculated from data if not provided
-
         vmax: int, optional
             maximum value for color scaling, calculated from data if not provided
-
         cmap: str, optional
             colormap to use to display the image data, default is ``"plasma"``
-
         selection_options
-
         args:
             additional arguments passed to Graphic
         kwargs:
@@ -140,4 +134,4 @@ def add_highlight(self, event):
         self.world_object.add(self.selection_graphic)
         self._highlights.append(self.selection_graphic)
 
-        return rval
+        return rval
diff --git a/fastplotlib/graphics/image.py b/fastplotlib/graphics/image.py
@@ -3,11 +3,11 @@
 import numpy as np
 import pygfx
 
-from ._base import Graphic
+from ._base import Graphic, Interaction
 from ..utils import quick_min_max, get_cmap_texture
 
 
-class ImageGraphic(Graphic):
+class ImageGraphic(Graphic, Interaction):
     def __init__(
             self,
             data: Any,
@@ -72,6 +72,16 @@ def __init__(
             pygfx.ImageBasicMaterial(clim=(vmin, vmax), map=get_cmap_texture(cmap))
         )
 
+    @property
+    def features(self) -> List[str]:
+        return ["cmap", "data"]
+
+    def _set_feature(self, feature: str, new_data: Any, indices: Any):
+        pass
+
+    def _reset_feature(self, feature: str, old_data: Any):
+        pass
+
     @property
     def clim(self) -> Tuple[float, float]:
         return self.world_object.material.clim

diff --git a/fastplotlib/graphics/line.py b/fastplotlib/graphics/line.py
@@ -1,8 +1,9 @@
 from typing import *
 import numpy as np
 import pygfx
+from typing import *
 
-from ._base import Graphic
+from ._base import Graphic, CallbackData, Interaction
 
 
 class LineGraphic(Graphic):
@@ -57,5 +58,19 @@ def __init__(
             geometry=pygfx.Geometry(positions=self.data.feature_data, colors=self.colors.feature_data),
             material=material(thickness=size, vertex_colors=True)
         )
-
+        
         self.world_object.position.z = z_position
+
+    def _set_feature(self, feature: str, new_data: Any, indices: Any = None):
+        if feature in self.features:
+            update_func = getattr(self, f"update_{feature}")
+            update_func(new_data)
+        else:
+            raise ValueError("name arg is not a valid feature")
+
+    def _reset_feature(self, feature: str, old_data: Any):
+        if feature in self.features:
+            update_func = getattr(self, f"update_{feature}")
+            update_func(old_data)
+        else:
+            raise ValueError("name arg is not a valid feature")
diff --git a/fastplotlib/graphics/linecollection.py b/fastplotlib/graphics/linecollection.py
@@ -1,13 +1,23 @@
 import numpy as np
 import pygfx
-from typing import Union
-from .line import LineGraphic
+from typing import Union, List
+
+from fastplotlib.graphics.line import LineGraphic
 from typing import *
+from fastplotlib.graphics._base import Interaction
+from abc import ABC, abstractmethod
+
 
+class LineCollection:
+    def __init__(self, data: List[np.ndarray],
+                 z_position: Union[List[float], float] = None,
+                 size: Union[float, List[float]] = 2.0,
+                 colors: Union[List[np.ndarray], np.ndarray] = None,
+                 cmap: Union[List[str], str] = None,
+                 *args,
+                 **kwargs):
 
-class LineCollection():
-    def __init__(self, data: List[np.ndarray], z_position: Union[List[float], float] = None, size: Union[float, List[float]] = 2.0, colors: Union[List[np.ndarray], np.ndarray] = None,
-                 cmap: Union[List[str], str] = None, *args, **kwargs):
+        self.name = None
 
         if not isinstance(z_position, float) and z_position is not None:
             if not len(data) == len(z_position):
@@ -22,7 +32,8 @@ def __init__(self, data: List[np.ndarray], z_position: Union[List[float], float]
             if not len(data) == len(cmap):
                 raise ValueError("args must be the same length")
 
-        self.collection = list()
+        self.data = list()
+        self._world_object = pygfx.Group()
 
         for i, d in enumerate(data):
             if isinstance(z_position, list):
@@ -45,10 +56,37 @@ def __init__(self, data: List[np.ndarray], z_position: Union[List[float], float]
             else:
                 _cmap = cmap
 
-            self.collection.append(LineGraphic(d, _z, _size, _colors, _cmap))
+            lg = LineGraphic(d, _z, _size, _colors, _cmap)
+            self.data.append(lg)
+            self._world_object.add(lg.world_object)
+
+    # TODO: make a base class for Collection graphics and put this as a base method
+    @property
+    def world_object(self) -> pygfx.WorldObject:
+        return self._world_object
+
+    @property
+    def features(self) -> List[str]:
+        return ["colors", "data"]
+
+    def _set_feature(self, feature: str, new_data: Any, indices: Any):
+        if feature in self.features:
+            update_func = getattr(self.data[indices], f"update_{feature}")
+            # if indices is a single indices or list of indices
+            self.data[indices].update_colors(new_data)
+        else:
+            raise ValueError("name arg is not a valid feature")
+
+    def _reset_feature(self, feature: str, old_data: Any):
+        if feature in self.features:
+            #update_func = getattr(self, f"update_{feature}")
+            for i, line in enumerate(self.data):
+                line.update_colors(old_data[i])
+        else:
+            raise ValueError("name arg is not a valid feature")
 
     def __getitem__(self, item):
-        return self.collection[item]
+        return self.data[item]