From 2fd44fbfb4e8cea1631f123d859dab22ca583e4f Mon Sep 17 00:00:00 2001
From: Pierre-Antoine Rouby <pierre-antoine.rouby@inrae.fr>
Date: Wed, 29 Nov 2023 16:52:01 +0100
Subject: [PATCH] IC: Fix data sorting at the end of generation.
---
src/Model/InitialConditions/InitialConditions.py | 13 +++++++++++++
1 file changed, 13 insertions(+)
diff --git a/src/Model/InitialConditions/InitialConditions.py b/src/Model/InitialConditions/InitialConditions.py
index c81f940c..439445fa 100644
--- a/src/Model/InitialConditions/InitialConditions.py
+++ b/src/Model/InitialConditions/InitialConditions.py
@@ -395,6 +395,7 @@ class InitialConditions(SQLSubModel):
self._data.append(new)
previous_elevation = elevation
+ self._generate_resort_data(profiles)
def generate_discharge(self, discharge: float):
self._data = []
@@ -438,3 +439,15 @@ class InitialConditions(SQLSubModel):
previous_elevation = elevation
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']
+ )
--
GitLab