parouby
/
Pamhyr2
mirror of https://gitlab.com/pamhyr/pamhyr2
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

cosmetic

terraz_dev
Theophile Terraz 2025-02-04 11:50:41 +01:00
parent fa8f07b97e
commit d3729cfe77
1 changed files with 35 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
src/View/Results/PlotH.py
View File

@ -88,20 +88,21 @@ class PlotH(PamhyrPlot):
if len(self._current_res_id) < 1:
return
if len(self._current_res_id) == 1:
self.draw_max(self._current_res_id[0])
for res_id in self._current_res_id:
if self.results[res_id] is None:
continue
results = self.results[res_id]
reach = results.river.reach(self._current_reach_id)
profile = reach.profile(self._current_profile_id)
if reach.geometry.number_profiles == 0:
self._init = False
return
self.draw_max(reach)
self.draw_data(reach, profile)
self.draw_data(res_id)
self.canvas.axes.set_xlabel(
self._trad["unit_time_s"],
@ -122,15 +123,28 @@ class PlotH(PamhyrPlot):
self.draw_current()
self._init = True
def draw_data(self, reach, profile):
def draw_data(self, res_id):
results = self.results[res_id]
reach = results.river.reach(self._current_reach_id)
profile = reach.profile(self._current_profile_id)
x = self._timestamps
y = profile.get_key("Q")
if res_id == 2:
label = "Δ " + self.label_discharge
else:
label = self.label_discharge
if len(self._current_res_id) > 1:
if res_id != 2:
label += f" ({results._solver.name})"
line, = self.canvas.axes.plot(
x, y,
label=self.label_discharge,
color=self.color_plot,
label=label,
color=self.colors[res_id + 1],
**self.plot_default_kargs
)
self._line.append(line)
@ -145,10 +159,23 @@ class PlotH(PamhyrPlot):
lw=1.,
)
def draw_max(self, reach):
def draw_max(self, res_id):
results = self.results[res_id]
reach = results.river.reach(self._current_reach_id)
profile = reach.profile(self._current_profile_id)
if reach.geometry.number_profiles == 0:
self._init = False
return
x = self._timestamps
y = []
if res_id == 2:
label = "Δ " + self.label_discharge_max
else:
label = self.label_discharge_max
for ts in x:
ts_y = -9999
for profile in reach.profiles:
@ -158,7 +185,7 @@ class PlotH(PamhyrPlot):
m, = self.canvas.axes.plot(
x, y,
label=self.label_discharge_max,
label=label,
color=self.color_plot_highlight,
linestyle='dotted',
**self.plot_default_kargs