compare_results
Theophile Terraz 2024-11-05 11:16:17 +01:00
parent 2896c604a5
commit 6852b08cfc
3 changed files with 19 additions and 43 deletions

View File

@ -360,26 +360,11 @@ class InitialConditions(SQLSubModel):
def get_discharge(self):
return self._data_get("discharge")
def _sort_by_z_and_rk(self, profiles):
profiles.sort(
reverse=False,
key=lambda p: p.rk
)
first_z = profiles[0].z()
last_z = profiles[-1].z()
if first_z > last_z:
profiles.sort(
reverse=True,
key=lambda p: p.rk
)
def generate_growing_constant_depth(self, height: float,
compute_discharge: bool):
profiles = self._reach.reach.profiles.copy()
self._sort_by_z_and_rk(profiles)
profiles.reverse()
previous_elevation = -99999.99
@ -433,12 +418,12 @@ class InitialConditions(SQLSubModel):
previous_elevation = elevation
self._data.append(new)
self._generate_resort_data(profiles)
self._data.reverse()
def generate_discharge(self, discharge: float, compute_height: bool):
profiles = self._reach.reach.profiles.copy()
self._sort_by_z_and_rk(profiles)
profiles.reverse()
previous_elevation = -99999.99
@ -491,7 +476,7 @@ class InitialConditions(SQLSubModel):
previous_elevation = elevation
self._data.append(new)
self._generate_resort_data(profiles)
self._data.reverse()
def generate_height(self,
elevation1: float,
@ -525,13 +510,3 @@ class InitialConditions(SQLSubModel):
new["discharge"] = d
new["elevation"] = elevation
self._data.append(new)
def _generate_resort_data(self, profiles):
is_reverse = False
if profiles[0].rk > profiles[-1].rk:
is_reverse = True
self._data.sort(
reverse=not is_reverse,
key=lambda d: d['rk']
)

View File

@ -77,31 +77,32 @@ class PlotDRK(PamhyrPlot):
rk = self.data.get_rk()
elevation = self.data.get_elevation()
sorted_rk, sorted_elevation = zip(
*sorted(zip(rk, elevation))
)
self.line_rk_elevation = self.canvas.axes.plot(
rk, elevation,
sorted_rk, sorted_elevation,
color=self.color_plot_river_water,
**self.plot_default_kargs
)
z_min = self.data.reach.reach.get_z_min()
geometry_rk = self.data.reach.reach.get_rk()
filtred_elevation = list(
map(
lambda x: elevation[x[0]],
filter(
lambda x: x[1] in geometry_rk,
enumerate(rk)
)
)
sorted_geometry_rk, sorted_z_min = zip(
*sorted(zip(geometry_rk, z_min), reverse=True)
)
self.collection = self.canvas.axes.fill_between(
geometry_rk, z_min, filtred_elevation,
poly_x = sorted_rk + sorted_geometry_rk
poly_y = sorted_elevation + sorted_z_min
self.collection = self.canvas.axes.fill(
poly_x, poly_y,
color=self.color_plot_river_water_zone,
alpha=0.7, interpolate=True
alpha=0.7,
)
@timer
def update(self, ind=None):
self.draw()

View File

@ -782,7 +782,7 @@ class CustomPlot(PamhyrPlot):
@timer
def update(self):
#if not self._init:
# if not self._init:
self.draw_update()
self.draw_current()
return