mirror of https://gitlab.com/pamhyr/pamhyr2
debug CI
parent
2896c604a5
commit
6852b08cfc
|
|
@ -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']
|
||||
)
|
||||
|
|
|
|||
|
|
@ -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()
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in New Issue