From 5327c29641e17fd551681cf6fa14fe5acd0c3d81 Mon Sep 17 00:00:00 2001
From: Pierre-Antoine Rouby <contact@parouby.fr>
Date: Thu, 9 Jul 2026 18:05:10 +0200
Subject: [PATCH] Results: Custom plot: Use 'zfd' np table.

---
 src/View/Results/CustomPlot/Plot.py | 37 +++++++++++++----------------
 1 file changed, 17 insertions(+), 20 deletions(-)

diff --git a/src/View/Results/CustomPlot/Plot.py b/src/View/Results/CustomPlot/Plot.py
index 795902ae..1143cd1a 100644
--- a/src/View/Results/CustomPlot/Plot.py
+++ b/src/View/Results/CustomPlot/Plot.py
@@ -91,12 +91,14 @@ class CustomPlot(PamhyrPlot):
         self.lines = {}
 
     def draw_bottom_with_bedload(self, reach):
+        results = self.data[self._current_res_id]
+        table = results.get("table")["zfd"]
+        ts = results.get_timestamp_id(self._current_timestamp)
+
         rk = reach.geometry.get_rk()
         z = list(
             map(
-                lambda p: p.get_ts_key(
-                    self._current_timestamp, "zfd"
-                ),
+                lambda p: table[ts, p.global_index],
                 reach.profiles
             )
         )
@@ -106,7 +108,10 @@ class CustomPlot(PamhyrPlot):
     def get_ts_zmin(self, profile, res_id):
         results = self.data[res_id]
         reach = results.river.reach(self._current_reach)
-        zfd = reach.profile(profile).get_key("zfd")
+
+        table = results.get("table")["zfd"]
+
+        zfd = table[:, reach.profile(profile).global_index]
         return zfd
 
     def _draw_rk(self):
@@ -170,22 +175,14 @@ class CustomPlot(PamhyrPlot):
                 )
             else:
                 if reach.has_bedload():
-                    z_min1 = list(
-                            map(
-                                lambda p: p.get_ts_key(
-                                    self._current_timestamp, "zfd"
-                                ),
-                                reach1.profiles
-                            )
-                        )
-                    z_min2 = list(
-                            map(
-                                lambda p: p.get_ts_key(
-                                    self._current_timestamp, "zfd"
-                                ),
-                                reach2.profiles
-                            )
-                        )
+                    z_min1 = list(map(
+                        lambda p: table["zfd"][id_ts, p.global_index],
+                        reach1.profiles
+                    ))
+                    z_min2 = list(map(
+                        lambda p: table["zfd"][id_ts, p.global_index],
+                        reach2.profiles
+                    ))
                 else:
                     z_min1 = reach1.geometry.get_z_min()
                     z_min2 = reach2.geometry.get_z_min()