ci: pre-commit

Author: jourdain

docs/scripts/fetch_paraview_presets.py

@@ -34,8 +34,7 @@
 from pathlib import Path
 
 SOURCE_URL = (
-    "https://gitlab.kitware.com/paraview/paraview/"
-    "-/raw/master/Remoting/Views/ColorMaps.json"
+    "https://gitlab.kitware.com/paraview/paraview/-/raw/master/Remoting/Views/ColorMaps.json"
 )
 SOURCE_LICENSE = "BSD-3-Clause"
 

examples/wavelet.py

@@ -1,3 +1,14 @@
+#!/usr/bin/env -S uv run --script
+#
+# /// script
+# requires-python = ">=3.11"
+# dependencies = [
+#     "trame>=3.13.2",
+#     "trame-colormaps>=1",
+#     "trame-vtk",
+#     "trame-vuetify",
+# ]
+# ///
 """Example: Wavelet with clipped contour + slice plane.
 
 A fully synthetic VTK pipeline that demonstrates trame-colormaps:
@@ -184,9 +195,7 @@ def _build_ui(self):
                         ),
                         self.left.provide_as("left"),
                     ):
-                        with html.Div(
-                            style="position:absolute;top:0;bottom:0;left:0;right:0;"
-                        ):
+                        with html.Div(style="position:absolute;top:0;bottom:0;left:0;right:0;"):
                             VerticalScalarBar("left", popup_location="right")
 
                     # Middle
@@ -207,9 +216,7 @@ def _build_ui(self):
                         ),
                         self.right.provide_as("right"),
                     ):
-                        with html.Div(
-                            style="position:absolute;top:0;bottom:0;left:0;right:0;"
-                        ):
+                        with html.Div(style="position:absolute;top:0;bottom:0;left:0;right:0;"):
                             VerticalScalarBar("right", popup_location="left")
 
                     # Bottom

src/trame_colormaps/core/presets.py

@@ -45,21 +45,15 @@ def load(filename):
     color_blind = {n for n, p in registry.items() if p.get("ColorBlindSafe", False)}
     defaults_path = _PRESET_DIR / "default_presets.json"
     defaults = (
-        json.loads(defaults_path.read_text())
-        if defaults_path.exists()
-        else sorted(all_names)
+        json.loads(defaults_path.read_text()) if defaults_path.exists() else sorted(all_names)
     )
 
     # Category-based preset sets
-    diverging = {
-        n for n, p in registry.items() if p.get("Category", "").lower() == "diverging"
-    }
+    diverging = {n for n, p in registry.items() if p.get("Category", "").lower() == "diverging"}
     sequential = {
         n for n, p in registry.items() if p.get("Category", "").lower() == "sequential"
     }
-    cyclic = {
-        n for n, p in registry.items() if p.get("Category", "").lower() == "cyclic"
-    }
+    cyclic = {n for n, p in registry.items() if p.get("Category", "").lower() == "cyclic"}
     categorical = {
         n
         for n, p in registry.items()

src/trame_colormaps/core/ticks.py

@@ -51,13 +51,10 @@ def get_nice_ticks(vmin, vmax, n, scale="linear", linthresh=None):
         # Filter out ticks too close to edges
         if len(raw_ticks) > 0:
             raw_ticks = raw_ticks[
-                (raw_ticks > vmin + data_range * 0.01)
-                & (raw_ticks < vmax - data_range * 0.01)
+                (raw_ticks > vmin + data_range * 0.01) & (raw_ticks < vmax - data_range * 0.01)
             ]
         # Priority: always include 0 if range spans negative to positive
-        if vmin < 0 < vmax and not np.any(
-            np.isclose(raw_ticks, 0.0, atol=nice_step * 1e-9)
-        ):
+        if vmin < 0 < vmax and not np.any(np.isclose(raw_ticks, 0.0, atol=nice_step * 1e-9)):
             raw_ticks = np.sort(np.append(raw_ticks, 0.0))
         return raw_ticks
     elif scale == "log":

src/trame_colormaps/core/transforms.py

@@ -222,9 +222,7 @@ def _sl(v):
         new_rgb_points.extend([float(x_min), clamp_r, clamp_g, clamp_b])
         display_rgb_points.extend([float(x_min), clamp_r, clamp_g, clamp_b])
         d_lt = x_min + lt_disp_frac * data_range
-        new_rgb_points.extend(
-            [float(linthresh) - data_range * 1e-9, clamp_r, clamp_g, clamp_b]
-        )
+        new_rgb_points.extend([float(linthresh) - data_range * 1e-9, clamp_r, clamp_g, clamp_b])
         display_rgb_points.extend([d_lt - data_range * 1e-9, clamp_r, clamp_g, clamp_b])
 
     # --- Region 2: [linthresh, x_max] — log-mapped ---
@@ -594,9 +592,7 @@ def symlog(v):
         neg_outward = list(reversed(neg))
         thinned_neg = [neg_outward[i] for i in range(0, len(neg_outward), 2)]
         thinned_pos = [pos[i] for i in range(0, len(pos), 2)]
-        discrete_tick_data = sorted(
-            thinned_neg + zero + thinned_pos, key=lambda t: t["val"]
-        )
+        discrete_tick_data = sorted(thinned_neg + zero + thinned_pos, key=lambda t: t["val"])
     else:
         discrete_tick_data = all_tick_data
 

src/trame_colormaps/dataclasses.py

@@ -481,15 +481,11 @@ def _sl(v):
                         pos = (float(_sl(val)) - sl_min) / sl_range * 100
                     else:
                         pos = (val - vmin) / data_range * 100
-                    ticks.append(
-                        {"position": round(pos, 2), "label": format_log_tick(val)}
-                    )
+                    ticks.append({"position": round(pos, 2), "label": format_log_tick(val)})
 
         # Sample colors from the *linear* CTF so tick contrast matches the
         # displayed colorbar image, not the log/symlog-remapped rendering CTF.
-        rgb_points = (
-            linear_rgb_points if linear_rgb_points else get_rgb_points(self._ctf)
-        )
+        rgb_points = linear_rgb_points if linear_rgb_points else get_rgb_points(self._ctf)
         if len(rgb_points) < 4:
             self.color_ticks = []
             return
@@ -715,9 +711,7 @@ def update_color_preset(
                     self.lut_img_v = result[1]
         elif log_scale == "symlog":
             if discrete_log:
-                result = apply_discrete_symlog(
-                    self._ctf, linthresh, linear_rgb_points, n_sub
-                )
+                result = apply_discrete_symlog(self._ctf, linthresh, linear_rgb_points, n_sub)
                 if result[0] is not None:
                     linear_rgb_points = result[0]
                     self.lut_img_h = result[2]

src/trame_colormaps/widgets.py

@@ -14,21 +14,14 @@
 def buttons(name):
     return [
         {
-            "icon": (
-                f"{name}.color_blind ? 'mdi-shield-check-outline' : 'mdi-palette'"
-            ),
+            "icon": (f"{name}.color_blind ? 'mdi-shield-check-outline' : 'mdi-palette'"),
             "click": f"{name}.color_blind = !{name}.color_blind",
-            "tip": (
-                f"'Toggle to ' + ({name}.color_blind ? "
-                "'All Colors' : 'Colorblind Safe')"
-            ),
+            "tip": (f"'Toggle to ' + ({name}.color_blind ? 'All Colors' : 'Colorblind Safe')"),
         },
         {
             "icon": f"{name}.invert ? 'mdi-invert-colors' : 'mdi-invert-colors-off'",
             "click": f"{name}.invert = !{name}.invert",
-            "tip": (
-                f"'Toggle to ' + ({name}.invert ? 'Normal Preset' : 'Invert Preset')"
-            ),
+            "tip": (f"'Toggle to ' + ({name}.invert ? 'Normal Preset' : 'Invert Preset')"),
         },
         {
             "icon": (
@@ -53,17 +46,11 @@ def buttons(name):
         {
             "icon": f"{name}.diverging ? 'mdi-triangle' : 'mdi-triangle-outline'",
             "click": f"{name}.diverging = !{name}.diverging",
-            "tip": (
-                f"'Toggle to ' + ({name}.diverging ? 'Normal Mode' : 'Difference Mode')"
-            ),
+            "tip": (f"'Toggle to ' + ({name}.diverging ? 'Normal Mode' : 'Difference Mode')"),
         },
         {
-            "icon": (
-                f"{name}.override_range ? 'mdi-arrow-expand-horizontal' : 'mdi-pencil'"
-            ),
-            "click": (
-                f"!{name}.diverging && ({name}.override_range = !{name}.override_range)"
-            ),
+            "icon": (f"{name}.override_range ? 'mdi-arrow-expand-horizontal' : 'mdi-pencil'"),
+            "click": (f"!{name}.diverging && ({name}.override_range = !{name}.override_range)"),
             "tip": (
                 f"{name}.diverging ? 'Range locked in Δ mode'"
                 f" : 'Toggle to ' + ({name}.override_range ? "
@@ -72,8 +59,7 @@ def buttons(name):
         },
         {
             "icon": (
-                f"{name}.discrete_log ? 'mdi-view-sequential' :"
-                " 'mdi-gradient-horizontal'"
+                f"{name}.discrete_log ? 'mdi-view-sequential' : 'mdi-gradient-horizontal'"
             ),
             "click": f"{name}.discrete_log = !{name}.discrete_log",
             "tip": f"'Toggle to ' + ({name}.discrete_log ? 'Continuous' : 'Discrete')",
@@ -224,9 +210,7 @@ def __init__(self, name):
                 f"{name}.n_colors"
                 "]"
             )
-            _v_for = (
-                f"entry in ({name}.invert ? {name}.luts_inverted : {name}.luts_normal)"
-            )
+            _v_for = f"entry in ({name}.invert ? {name}.luts_inverted : {name}.luts_normal)"
             _v_show = (
                 f"({name}.search && {name}.search.length"
                 f" ? entry.name.toLowerCase().includes({name}.search.toLowerCase())"
@@ -308,9 +292,7 @@ def __init__(self, name, popup_location="top", **kwargs):
 class VerticalScalarBar(html.Div):
     def __init__(self, name, popup_location="top", **kwargs):
         super().__init__(
-            classes=(
-                "tcmap-vertical bg-blue-grey-darken-2 d-flex flex-column align-center"
-            ),
+            classes=("tcmap-vertical bg-blue-grey-darken-2 d-flex flex-column align-center"),
         )
         self.server.enable_module(module)
 

tests/test_ticks.py

@@ -103,9 +103,7 @@ def test_log_within_range(self):
     def test_log_majors_first(self):
         """Majors (powers of 10) should appear before any minors."""
         ticks = get_nice_ticks(0, 276, 5, scale="log", linthresh=37.35)
-        powers = [
-            t for t in ticks if t > 0 and np.isclose(np.log10(t) % 1, 0, atol=1e-9)
-        ]
+        powers = [t for t in ticks if t > 0 and np.isclose(np.log10(t) % 1, 0, atol=1e-9)]
         assert len(powers) >= 1
 
     def test_log_nice_minors(self):
@@ -116,9 +114,7 @@ def test_log_nice_minors(self):
                 continue
             mantissa = t / 10.0 ** int(np.floor(np.log10(t)))
             nice = {1, 2, 2.5, 3, 4, 5, 6, 7, 7.5, 8, 9, 10}
-            assert any(np.isclose(mantissa, n, atol=0.01) for n in nice), (
-                f"{t} not nice"
-            )
+            assert any(np.isclose(mantissa, n, atol=0.01) for n in nice), f"{t} not nice"
 
     def test_symlog_has_negative_and_positive(self):
         ticks = get_nice_ticks(-1000, 1000, 10, scale="symlog", linthresh=1.0)
@@ -132,9 +128,5 @@ def test_symlog_includes_zero_when_space(self):
     def test_symlog_majors_before_minors(self):
         """At least one power of 10 should be present."""
         ticks = get_nice_ticks(-276, 276, 9, scale="symlog", linthresh=37.35)
-        powers = [
-            t
-            for t in ticks
-            if t != 0 and np.isclose(np.log10(abs(t)) % 1, 0, atol=1e-9)
-        ]
+        powers = [t for t in ticks if t != 0 and np.isclose(np.log10(abs(t)) % 1, 0, atol=1e-9)]
         assert len(powers) >= 1

tests/test_transforms.py

@@ -204,9 +204,7 @@ def test_returns_tuple(self):
         apply_linear(ctf, PRESET_NAME)
         rescale_ctf(ctf, -1000.0, 1000.0)
         pts = get_rgb_points(ctf)
-        result = apply_discrete_symlog(
-            ctf, linthresh=1.0, linear_rgb_points=pts, n_sub=4
-        )
+        result = apply_discrete_symlog(ctf, linthresh=1.0, linear_rgb_points=pts, n_sub=4)
         assert isinstance(result, tuple)
         assert len(result) == 4