implementing reset_feature

fastplotlib · clewis7 · Dec 12, 2022 · Dec 12, 2022 · Dec 14, 2022 · Dec 15, 2022
commit bc688fc510b28b115c478d9759059f85a695d1e4
diff --git a/fastplotlib/graphics/_base.py b/fastplotlib/graphics/_base.py
@@ -7,6 +7,7 @@

 from abc import ABC, abstractmethod
 from dataclasses import dataclass
+# from .linecollection import LineCollection

 class Graphic:
    def __init__(
@@ -92,7 +93,7 @@ def _set_feature(self, feature: str, new_data: Any, indices: Any):
        pass

    @abstractmethod
-    def _reset_feature(self):
+    def _reset_feature(self, feature: str, old_data: Any, indices: Any):
        pass

    def link(self, event_type: str, target: Graphic, feature: str, new_data: Any, indices_mapper: callable = None):
@@ -104,20 +105,22 @@ def link(self, event_type: str, target: Graphic, feature: str, new_data: Any, in

        if event_type in self.registered_callbacks.keys():
            self.registered_callbacks[event_type].append(
-                CallbackData(target=target, feature=feature, new_data=new_data))
+                CallbackData(target=target, feature=feature, new_data=new_data, old_data=getattr(target, feature).copy()))
        else:
            self.registered_callbacks[event_type] = list()
            self.registered_callbacks[event_type].append(
-                CallbackData(target=target, feature=feature, new_data=new_data))
+                CallbackData(target=target, feature=feature, new_data=new_data, old_data=getattr(target, feature).copy()))

    def event_handler(self, event):
        if event.type in self.registered_callbacks.keys():
            for target_info in self.registered_callbacks[event.type]:
-                target_info.target._set_feature(feature=target_info.feature, new_data=target_info.new_data)
+                target_info.target._reset_feature(feature=target_info.feature, old_data=target_info.old_data, indices=None)
+                target_info.target._set_feature(feature=target_info.feature, new_data=target_info.new_data, indices=None)

 @dataclass
 class CallbackData:
    """Class for keeping track of the info necessary for interactivity after event occurs."""
    target: Graphic
    feature: str
    new_data: Any
+    old_data: Any
diff --git a/fastplotlib/graphics/line.py b/fastplotlib/graphics/line.py
@@ -69,5 +69,10 @@ def _set_feature(self, feature: str, new_data: Any, indices: Any = None):
        else:
            raise ValueError("name arg is not a valid feature")

-    def _reset_feature():
-        pass
+    def _reset_feature(self, feature: str, old_data: Any, indices: Any = None):
+        if feature in ["colors", "data"]:
+            update_func = getattr(self, f"update_{feature}")
+            update_func(old_data)
+        else:
+            raise ValueError("name arg is not a valid feature")
+