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

Mage: Minor code style refactoring.

setup.py
Pierre-Antoine Rouby 2023-12-13 16:38:09 +01:00
parent 58300b8c1a
commit ffbd2e2413
1 changed files with 25 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
src/Solver/Mage.py
View File

@ -442,14 +442,9 @@ class Mage(CommandLineSolver):
sin_dict = { sin_dict = {
"ND": "*", "ND": "*",
"S1": "D", "S1": "D", "S2": "T", "S3": "T",
"S2": "T", "OR": "O", "OC": "B", "OV": "F",
"S3": "T", "V1": "V", "V2": "W",
"OR": "O",
"OC": "B",
"OV": "F",
"V1": "V",
"V2": "W",
"BO": "A", "BO": "A",
"UD": "X", "UD": "X",
"PO": "P", "PO": "P",
@ -466,32 +461,34 @@ class Mage(CommandLineSolver):
files.append(f"{name}.SIN") files.append(f"{name}.SIN")
for hs in hydraulic_structures: for hs in hydraulic_structures:
if hs.input_reach.is_enable: if not hs.input_reach.is_enable():
f.write( continue
'* ouvrage au pk ' +
f"{hs.input_kp:>12.1f}" + ' ' + f.write(
hs.name + '\n' '* ouvrage au pk ' +
f"{hs.input_kp:>12.1f}" + ' ' +
hs.name + '\n'
)
for bhs in hs.basic_structures:
reach_id = study.river.get_edge_id(hs.input_reach) + 1
param_str = ' '.join(
[
f'{p:>10.3f}'
for p in self._export_SIN_parameters(bhs)
]
) )
for bhs in hs.basic_structures: f.write(
reach_id = study.river.get_edge_id(hs.input_reach) + 1 f"{sin_dict[bhs._type]} " +
param_str = ' '.join( f"{reach_id} {hs.input_kp:>12.3f} {param_str} " +
[ f"{bhs.name}\n"
f'{p:>10.3f}' )
for p in self._export_SIN_parameters(bhs)
]
)
f.write(
f"{sin_dict[bhs._type]} " +
f"{reach_id} {hs.input_kp:>12.3f} {param_str} " +
f"{bhs.name}\n"
)
return files return files
def _export_SIN_parameters(self, bhs): def _export_SIN_parameters(self, bhs):
res = [9999.999]*5 res = [9999.999] * 5
if len(bhs) == 5: if len(bhs) == 5:
res = self._export_SIN_parameters_5(bhs) res = self._export_SIN_parameters_5(bhs)