From 573a3fca2b93b06180120abcf6e7af87417ffdf1 Mon Sep 17 00:00:00 2001
From: Pierre-Antoine Rouby <pierre-antoine.rouby@inrae.fr>
Date: Wed, 21 Jun 2023 09:21:25 +0200
Subject: [PATCH] Mage: Add export of RUG file.
---
src/Solver/Mage.py | 32 ++++++++++++++++++++++++++++++++
1 file changed, 32 insertions(+)
diff --git a/src/Solver/Mage.py b/src/Solver/Mage.py
index 28142dca..d4792b7f 100644
--- a/src/Solver/Mage.py
+++ b/src/Solver/Mage.py
@@ -122,6 +122,37 @@ class Mage(AbstractSolver):
for bound in lst.get_tab(tab):
self._export_BC(bound, repertory, qlog)
+ @timer
+ def _export_RUG(self, study, repertory, qlog = None):
+ if qlog is not None:
+ qlog.put("Export RUG file")
+
+ # Write header
+ with open(os.path.join(repertory, f"0.RUG"), "w+") as f:
+ f.write("* This file is generate by PAMHYR, please don't modify\n")
+
+ id = 1
+ edges = study.river.edges()
+ for edge in edges:
+ sections = edge.sections
+
+ for section in sections.sections:
+ num = f"{id:>3}"
+ bkp = f"{section.begin_kp:>10.3f}"
+ ekp = f"{section.end_kp:>10.3f}"
+
+ if section.begin_kp != section.end_kp:
+ print("TODO")
+
+ strickler = section.begin_strickler
+ coef_1 = f"{strickler.minor:>10.3f}"
+ coef_2 = f"{strickler.medium:>10.3f}"
+
+ f.write(f"K{num} {bkp}{ekp}{coef_1}{coef_2}\n")
+ id += 1
+
+ f.write("* END OF FILE")
+
@timer
def export(self, study, repertory, qlog = None):
self._export_ST(study, repertory, qlog)
@@ -216,5 +247,6 @@ class Mage8(Mage):
self._export_NET(study, repertory, qlog)
self._export_ST(study, repertory, qlog)
self._export_bound_cond(study, repertory, qlog)
+ self._export_RUG(study, repertory, qlog)
return True
--
GitLab