update

Ax_Shell/utils/animator.py

@@ -0,0 +1,180 @@
+from typing import cast
+
+import fabric
+from fabric import Property, Service, Signal
+from gi.repository import GLib, Gtk
+
+
+class Animator(Service):
+    @Signal
+    def finished(self) -> None: ...
+
+    @Property(tuple[float, float, float, float], "read-write")
+    def bezier_curve(self) -> tuple[float, float, float, float]:
+        return self._bezier_curve
+
+    @bezier_curve.setter
+    def bezier_curve(self, value: tuple[float, float, float, float]):
+        self._bezier_curve = value
+        return
+
+    @Property(float, "read-write")
+    def value(self):
+        return self._value
+
+    @value.setter
+    def value(self, value: float):
+        self._value = value
+        return
+
+    @Property(float, "read-write")
+    def max_value(self):
+        return self._max_value
+
+    @max_value.setter
+    def max_value(self, value: float):
+        self._max_value = value
+        return
+
+    @Property(float, "read-write")
+    def min_value(self):
+        return self._min_value
+
+    @min_value.setter
+    def min_value(self, value: float):
+        self._min_value = value
+        return
+
+    @Property(bool, "read-write", default_value=False)
+    def playing(self):
+        return self._playing
+
+    @playing.setter
+    def playing(self, value: bool):
+        self._playing = value
+        return
+
+    @Property(bool, "read-write", default_value=False)
+    def repeat(self):
+        return self._repeat
+
+    @repeat.setter
+    def repeat(self, value: bool):
+        self._repeat = value
+        return
+
+    def __init__(
+        self,
+        bezier_curve: tuple[float, float, float, float],
+        duration: float,
+        min_value: float = 0.0,
+        max_value: float = 1.0,
+        repeat: bool = False,
+        tick_widget: Gtk.Widget | None = None,
+        **kwargs,
+    ):
+        super().__init__(**kwargs)
+        self._bezier_curve = (1, 0, 1, 1)
+        self._duration = 5
+        self._value = 0.0
+        self._min_value = 0.0
+        self._max_value = 1.0
+        self._repeat = False
+
+        self.bezier_curve = bezier_curve
+        self.duration = duration
+        self.value = min_value
+        self.min_value = min_value
+        self.max_value = max_value
+        self.repeat = repeat
+
+        self.playing = False
+        self._start_time = None
+        self._tick_handler = None
+        self._timeline_pos = 0
+        self._tick_widget = tick_widget
+
+    def do_get_time_now(self):
+        return GLib.get_monotonic_time() / 1_000_000
+
+    def do_lerp(self, start: float, end: float, time: float) -> float:
+        return start + (end - start) * time
+
+    def do_interpolate_cubic_bezier(self, time: float) -> float:
+        y_points = (0, self.bezier_curve[1], self.bezier_curve[3], 1)
+        return (
+            (1 - time) ** 3 * y_points[0]
+            + 3 * (1 - time) ** 2 * time * y_points[1]
+            + 3 * (1 - time) * time**2 * y_points[2]
+            + time**3 * y_points[3]
+        )
+
+    def do_ease(self, time: float) -> float:
+        return self.do_lerp(
+            self.min_value, self.max_value, self.do_interpolate_cubic_bezier(time)
+        )
+
+    def do_update_value(self, delta_time: float):
+        if not self.playing:
+            return
+
+        elapsed_time = delta_time - cast(float, self._start_time)
+
+        self._timeline_pos = min(1, elapsed_time / self.duration)
+
+        self.value = self.do_ease(self._timeline_pos)
+
+        if not self._timeline_pos >= 1:
+            return
+
+        if not self.repeat:
+            self.value = self.max_value
+            self.finished()
+            self.pause()
+            return
+
+        self._start_time = delta_time
+        self._timeline_pos = 0
+        return
+
+    def do_handle_tick(self, *_):
+        current_time = self.do_get_time_now()
+        self.do_update_value(current_time)
+        return True
+
+    def do_remove_tick_handlers(self):
+        if self._tick_handler:
+            if self._tick_widget:
+                self._tick_widget.remove_tick_callback(self._tick_handler)
+            else:
+                GLib.source_remove(self._tick_handler)
+        self._tick_handler = None
+        return
+
+    def play(self):
+        if self.playing:
+            return
+
+        self._start_time = self.do_get_time_now()
+
+        if not self._tick_handler:
+            if self._tick_widget:
+                self._tick_handler = self._tick_widget.add_tick_callback(
+                    self.do_handle_tick
+                )
+            else:
+                self._tick_handler = GLib.timeout_add(16, self.do_handle_tick)
+
+        self.playing = True
+        return
+
+    def pause(self):
+        self.playing = False
+        return self.do_remove_tick_handlers()
+
+    def stop(self):
+        if not self._tick_handler:
+            self._timeline_pos = 0
+            self.playing = False
+            return
+        return self.do_remove_tick_handlers()