Changeset 3840 for trunk/LMDZ.COMMON/libf/evolution/deftank/visu_evol_layering.py

Timestamp:

Jul 7, 2025, 6:09:01 PM (11 days ago)

Author:

jbclement

Message:

PEM:

• Correction of a bug in the launching script.
• Update of "visu_evol_layering.py", in particular to show value at cursor for 2D heatmaps.
• Few cleanings.

JBC

File:

• trunk/LMDZ.COMMON/libf/evolution/deftank/visu_evol_layering.py (modified) (6 diffs)

trunk/LMDZ.COMMON/libf/evolution/deftank/visu_evol_layering.py

@@ -379 +379 @@
 
 
+def attach_format_coord(ax, mat, x, y, is_pcolormesh=True):
+    """
+    Attach a format_coord function to the axes to display x, y, and value at cursor.
+    Works for both pcolormesh and imshow style grids.
+    """
+    # Determine dimensions
+    if mat.ndim == 2:
+        ny, nx = mat.shape
+    elif mat.ndim == 3 and mat.shape[2] in (3, 4):
+        ny, nx, nc = mat.shape
+    else:
+        raise ValueError(f"Unsupported mat shape {mat.shape}")
+    # Edges or extents
+    if is_pcolormesh:
+        xedges, yedges = x, y
+    else:
+        x0, x1 = x.min(), x.max()
+        y0, y1 = y.min(), y.max()
+
+    def format_coord(xp, yp):
+        # Map to indices
+        if is_pcolormesh:
+            col = np.searchsorted(xedges, xp) - 1
+            row = np.searchsorted(yedges, yp) - 1
+        else:
+            col = int((xp - x0) / (x1 - x0) * nx)
+            row = int((yp - y0) / (y1 - y0) * ny)
+        # Within bounds?
+        if 0 <= row < ny and 0 <= col < nx:
+            if mat.ndim == 2:
+                v = mat[row, col]
+                return f"x={xp:.3g}, y={yp:.3g}, val={v:.3g}"
+            else:
+                vals = mat[row, col]
+                txt = ", ".join(f"{vv:.3g}" for vv in vals[:3])
+                return f"x={xp:.3g}, y={yp:.3g}, val=({txt})"
+        return f"x={xp:.3g}, y={yp:.3g}"
+
+    ax.format_coord = format_coord
+
+
 def plot_stratification_over_time(
     gridded_data,
@@ -454 +495 @@
                     vmax=vmax
                 )
+                x_edges = np.concatenate([date_time, [date_time[-1] + (date_time[-1]-date_time[-2])]])
+                attach_format_coord(ax, mat, x_edges, np.concatenate([sub_ref_grid, [sub_ref_grid[-1] + (sub_ref_grid[-1]-sub_ref_grid[-2])]]), is_pcolormesh=True)
                 ax.set_title(titles[idx], fontsize=12)
                 ax.set_xlabel("Time (Mars years)")
@@ -558 +601 @@
                 origin='lower'
             )
+            x_centers = np.linspace(date_time[0], date_time[-1], display_rgb.shape[1])
+            y_centers = np.linspace(elev.min(), elev.max(), display_rgb.shape[0])
+            attach_format_coord(ax_main, display_rgb, x_centers, y_centers, is_pcolormesh=False)
             ax_main.set_facecolor('white')
             ax_main.set_title(f"Ternary mix over time (Grid point {ig+1}, Slope {isl+1})", fontweight='bold')
@@ -570 +616 @@
                 interpolation='nearest'
             )
+            attach_format_coord(ax_leg, legend_rgba, np.array([0, 1]), np.array([0, np.sqrt(3)/2]), is_pcolormesh=False)
 
             # Draw triangle border
@@ -663 +710 @@
             # Plot
             fig, ax = plt.subplots(figsize=(8, 6), dpi=150)
+            ax.set_title(f"Dust-to-Ice ratio over time (Grid point {ig+1}, Slope {isl+1})", fontweight='bold')
             im = ax.imshow(
                 ratio_display,
@@ -672 +720 @@
                 norm=LogNorm(vmin=10**vmin, vmax=10**vmax)
             )
+            x_centers = np.linspace(date_time[0], date_time[-1], ratio_display.shape[1])
+            y_centers = np.linspace(elev.min(), elev.max(), ratio_display.shape[0])
+            attach_format_coord(ax, ratio_display, x_centers, y_centers, is_pcolormesh=False)
 
             # Add colorbar with simplified ratio labels