Commit 653a26fe authored by Pierre-Antoine Rouby's avatar Pierre-Antoine Rouby
Browse files

Geometry: Add hydro. struct. at KPZ plot.

Showing with 35 additions and 2 deletions
+35 -2
......@@ -25,8 +25,11 @@ logger = logging.getLogger()
class PlotKPZ(PamhyrPlot):
def __init__(self, canvas=None, trad=None, data=None, toolbar=None,
def __init__(self, canvas=None, trad=None,
study=None, data=None, toolbar=None,
parent=None):
self._study = study
super(PlotKPZ, self).__init__(
canvas=canvas,
trad=trad,
......@@ -75,6 +78,7 @@ class PlotKPZ(PamhyrPlot):
self.draw_current()
self.draw_gl()
self.draw_bottom()
self.draw_profiles_hs(self._data)
self.idle()
self._init = True
......@@ -184,6 +188,34 @@ class PlotKPZ(PamhyrPlot):
color='lightgrey'
)
def draw_profiles_hs(self, reach):
lhs = filter(
lambda hs: hs._input_reach.reach is reach,
self._study.river.hydraulic_structures.lst
)
for hs in lhs:
x = hs.input_kp
z_min = reach.get_z_min()
z_max = reach.get_z_max()
self.canvas.axes.plot(
[x, x],
[min(z_min), max(z_max)],
linestyle="--",
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,
)
@timer
def update(self):
if not self._init:
......
......@@ -376,6 +376,7 @@ class GeometryWindow(PamhyrWindow):
self._plot_kpc = PlotKPZ(
canvas=self._canvas_kpc,
study=self._study,
data=self._reach,
trad=self._trad,
toolbar=self._toolbar_kpc
......
......@@ -120,7 +120,7 @@ class PlotKPC(PamhyrPlot):
self.canvas.axes.plot(
[x, x],
[min(z_min), max(z_max)],
linestyle="solid",
linestyle="--",
lw=1.,
color=self.color_plot_previous,
)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment