From 629b8567884ed3ba62ef0173c94ce9b480ab29a3 Mon Sep 17 00:00:00 2001
From: Youcef AOUAD <youcef.aouad@inrae.fr>
Date: Mon, 2 Sep 2024 11:42:51 +0200
Subject: [PATCH] data view complete pols

---
 src/Solver/AdisTS.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/src/Solver/AdisTS.py b/src/Solver/AdisTS.py
index 5f7356b4..bea442b1 100644
--- a/src/Solver/AdisTS.py
+++ b/src/Solver/AdisTS.py
@@ -636,6 +636,25 @@ class AdisTSlc(AdisTS):
 
         print(data_tmp["AAA-silt"])
 
+        pollutants_keys = list(data_tmp.keys())
+        timestamps_keys = list(data_tmp[pollutants_keys[0]].keys())
+        phys_data_names = list(data_tmp[pollutants_keys[0]][timestamps_keys[0]].keys())
+
+        print("pol keys: ", pollutants_keys)
+        print("t keys: ", timestamps_keys)
+        print("phys var: ", phys_data_names)
+
+        for i in range(ismax):
+            reach = ip_to_r(i)
+            p_i = ip_to_ri(reach, i)
+
+            for t_data in timestamps_keys:
+                pol_view = []
+                for pol in pollutants_keys:
+                    pol_view.append(set(data_tmp[pol][t_data].values()))
+
+                reach.set(p_i, t_data, "pols", pol_view)
+
     @timer
     def results(self, study, repertory, qlog=None, name=None):
         self._study = study
-- 
GitLab