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

IC: Fix data sorting at the end of generation.

setup.py
Pierre-Antoine Rouby 2023-11-29 16:52:01 +01:00
parent 962bfa945d
commit 2fd44fbfb4
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/Model/InitialConditions/InitialConditions.py
View File

@ -395,6 +395,7 @@ class InitialConditions(SQLSubModel):
self._data.append(new) self._data.append(new)
previous_elevation = elevation previous_elevation = elevation
self._generate_resort_data(profiles)
def generate_discharge(self, discharge: float): def generate_discharge(self, discharge: float):
self._data = [] self._data = []
@ -438,3 +439,15 @@ class InitialConditions(SQLSubModel):
previous_elevation = elevation previous_elevation = elevation
self._data.append(new) self._data.append(new)
self._generate_resort_data(profiles)
def _generate_resort_data(self, profiles):
is_reverse = False
if profiles[0].kp > profiles[-1].kp:
is_reverse = True
self._data.sort(
reverse=not is_reverse,
key=lambda d: d['kp']
)