From e33c9cb2f0149a811f9ea415f6388f6a54fc670c Mon Sep 17 00:00:00 2001
From: Pierre-Antoine Rouby <pierre-antoine.rouby@inrae.fr>
Date: Wed, 15 May 2024 14:37:31 +0200
Subject: [PATCH] Results: Add hydro. struct. on kpc plot.
---
src/View/Results/PlotKPC.py | 29 +++++++++++++++++++++++++++++
1 file changed, 29 insertions(+)
diff --git a/src/View/Results/PlotKPC.py b/src/View/Results/PlotKPC.py
index 01bf7a6a..4d7bc756 100644
--- a/src/View/Results/PlotKPC.py
+++ b/src/View/Results/PlotKPC.py
@@ -79,6 +79,7 @@ class PlotKPC(PamhyrPlot):
self.draw_water_elevation_max(reach)
self.draw_water_elevation_overflow(reach)
self.draw_current(reach)
+ self.draw_profiles_hs(reach)
# self.enable_legend()
@@ -105,6 +106,34 @@ class PlotKPC(PamhyrPlot):
self._river_bottom = z
+ def draw_profiles_hs(self, reach):
+ lhs = filter(
+ lambda hs: hs._input_reach.reach is reach.geometry,
+ self.results.study.river.hydraulic_structures.lst
+ )
+
+ for hs in lhs:
+ x = hs.input_kp
+ z_min = reach.geometry.get_z_min()
+ z_max = reach.geometry.get_z_max()
+
+ self.canvas.axes.plot(
+ [x, x],
+ [min(z_min), max(z_max)],
+ linestyle="solid",
+ lw=1.,
+ color=self.color_plot_previous,
+ )
+
+ self.canvas.axes.annotate(
+ " > " + hs.name,
+ (x, max(z_max)),
+ horizontalalignment='left',
+ verticalalignment='top',
+ annotation_clip=True,
+ fontsize=9, color=self.color_plot_previous,
+ )
+
def sl_compute_bedrock(self, reach):
z_min = reach.geometry.get_z_min()
sl = self.sl_compute_initial(reach)
--
GitLab