diff --git a/src/View/Results/PlotRKCEnsemble.py b/src/View/Results/PlotRKCEnsemble.py index f5e8e190..3b38cf7c 100644 --- a/src/View/Results/PlotRKCEnsemble.py +++ b/src/View/Results/PlotRKCEnsemble.py @@ -160,7 +160,8 @@ class PlotRKCEnsemble(PamhyrPlot): table = result.get("table")[t] ts = result.get_timestamp_id(self._current_timestamp) - water_z = table[ts, :] + gi_0, gi_1 = reach.global_index + water_z = table[ts, gi_0 : gi_1 + 1] self.water = self.canvas.axes.plot( rk, water_z, @@ -264,7 +265,8 @@ class PlotRKCEnsemble(PamhyrPlot): table = result.get("table")["Z"] ts = result.get_timestamp_id(self._current_timestamp) - water_z = table[ts, :] + gi_0, gi_1 = reach.global_index + water_z = table[ts, gi_0 : gi_1 + 1] self.water[0].set_data( rk, water_z diff --git a/src/View/RunSolver/Window.py b/src/View/RunSolver/Window.py index b78b70df..9faec37f 100644 --- a/src/View/RunSolver/Window.py +++ b/src/View/RunSolver/Window.py @@ -566,6 +566,17 @@ class SolverLogEnsWindow(SolverLogWindow): results.set("table", table) + # Compute global index for reaches and profiles + gi = 0 + for reach in results.river.reachs: + gi_start = gi + + for profile in reach.profiles: + gi += 1 + + reach.set_global_index(range(gi_start, gi)) + + # Compute timestamps index table ts = tables["timestamps"] ts_index = {} for i, t in enumerate(sorted(ts)):