add toggle scale on double click

Author: nvaytet

src/matplotgl/axes.py

@@ -109,13 +109,19 @@ def __init__(self, *, ax: MplAxes, figure=None) -> None:
         # self._margin_with_ticks = 50
         self._thin_margin = 3
 
+        tooltips = {
+            "leftspine": "Double-click to toggle y-scale",
+            "bottomspine": "Double-click to toggle x-scale",
+        }
+
         self._margins = {
             name: ClickableHTML(
                 layout={
                     "grid_area": name,
                     "padding": "0",
                     "margin": "0",
                 },
+                tooltip=tooltips.get(name, ""),
             )
             for name in (
                 "leftspine",
@@ -146,6 +152,9 @@ def __init__(self, *, ax: MplAxes, figure=None) -> None:
         if figure is not None:
             self.set_figure(figure)
 
+        self._margins['leftspine'].on_dblclick(self._toggle_yscale)
+        self._margins['bottomspine'].on_dblclick(self._toggle_xscale)
+
         super().__init__(
             children=[
                 *self._margins.values(),
@@ -487,6 +496,12 @@ def set_yscale(self, scale):
         self.autoscale()
         self._make_yticks()
 
+    def _toggle_xscale(self, _):
+        self.set_xscale("log" if self.get_xscale() == "linear" else "linear")
+
+    def _toggle_yscale(self, _):
+        self.set_yscale("log" if self.get_yscale() == "linear" else "linear")
+
     def zoom(self, box):
         self._zoom_limits = {
             "xmin": box[0],

src/matplotgl/widgets.py

@@ -118,8 +118,8 @@ class ClickableHTML(anywidget.AnyWidget):
     tooltip_text = traitlets.Unicode("").tag(sync=True)
     _dblclick_trigger = traitlets.Int(0).tag(sync=True)
 
-    def __init__(self, value="", tooltip="", on_dblclick=None, **kwargs):
+    def __init__(self, value="", tooltip="", **kwargs):
         super().__init__(value=value, tooltip_text=tooltip, **kwargs)
 
-        if on_dblclick:
-            self.observe(lambda change: on_dblclick(self), "_dblclick_trigger")
+    def on_dblclick(self, on_dblclick):
+        self.observe(lambda change: on_dblclick(self), "_dblclick_trigger")