mirror of https://gitlab.com/pamhyr/pamhyr2
Compare commits
No commits in common. "5287f30636a525e9c3be64eb601abf1566aeb449" and "885c8ea5028fa36139c78815345f0e86fc7bc48f" have entirely different histories.
5287f30636
...
885c8ea502
|
|
@ -54,10 +54,8 @@ class Scenario(SQLSubModel):
|
|||
"river_reach", "river_node",
|
||||
"geotiff", "reservoir", "reservoir_data",
|
||||
# Temperature
|
||||
"boundary_condition_temperature",
|
||||
"boundary_condition_data_temperature",
|
||||
"initial_conditions_temperature",
|
||||
"initial_conditions_temperature_spec",
|
||||
"boundary_condition_temperature", "boundary_condition_data_temperature",
|
||||
"initial_conditions_temperature", "initial_conditions_temperature_spec",
|
||||
"weather_parameters", "weather_parameters_data",
|
||||
]
|
||||
|
||||
|
|
|
|||
|
|
@ -117,6 +117,9 @@ class AdisTS(CommandLineSolver):
|
|||
_nodes_names = {}
|
||||
_nodes_views = set()
|
||||
|
||||
def get_reach_name(self, reach):
|
||||
return f"Reach_{reach.pamhyr_id:>3}".replace(" ", "0")
|
||||
|
||||
def get_node_name(self, node):
|
||||
"""Generate a 3 char name for node
|
||||
|
||||
|
|
@ -179,7 +182,7 @@ class AdisTS(CommandLineSolver):
|
|||
# Write header
|
||||
edges = study.river.enable_edges()
|
||||
for edge in edges:
|
||||
name = self.get_reach_name(edge)
|
||||
name = f"Reach_{edge.id + 1:>3}".replace(" ", "0")
|
||||
|
||||
with adists_file_open(
|
||||
os.path.join(repertory, "net", f"{name}.ST"),
|
||||
|
|
@ -274,7 +277,7 @@ class AdisTS(CommandLineSolver):
|
|||
edges = study.river.enable_edges()
|
||||
|
||||
for e in edges:
|
||||
name = self.get_reach_name(e)
|
||||
name = f"Reach_{e.id + 1:>3}".replace(" ", "0")
|
||||
id = name
|
||||
|
||||
n1 = f"{self.get_node_name(e.node1):3}".replace(" ", "x")
|
||||
|
|
@ -450,8 +453,7 @@ class AdisTS(CommandLineSolver):
|
|||
_nodes_views = set()
|
||||
|
||||
def get_reach_name(self, reach):
|
||||
index = self._study.river.get_edge_id(reach) + 1
|
||||
return f"Reach_{index:03d}"
|
||||
return f"Reach_{reach.pamhyr_id:>3}".replace(" ", "0")
|
||||
|
||||
#################################
|
||||
# Adis-TS in weak coupling mode #
|
||||
|
|
|
|||
|
|
@ -113,8 +113,7 @@ class AdisTT(CommandLineSolver):
|
|||
_nodes_views = set()
|
||||
|
||||
def get_reach_name(self, reach):
|
||||
index = self._study.river.get_edge_id(reach) + 1
|
||||
return f"Reach_{index:03d}"
|
||||
return f"Reach_{reach.pamhyr_id:>3}".replace(" ", "0")
|
||||
|
||||
def get_node_name(self, node):
|
||||
"""Generate a 3 char name for node
|
||||
|
|
@ -178,7 +177,7 @@ class AdisTT(CommandLineSolver):
|
|||
# Write header
|
||||
edges = study.river.enable_edges()
|
||||
for edge in edges:
|
||||
name = self.get_reach_name(edge)
|
||||
name = f"Reach_{edge.id + 1:>3}".replace(" ", "0")
|
||||
|
||||
with adistt_file_open(
|
||||
os.path.join(repertory, "net", f"{name}.ST"),
|
||||
|
|
@ -241,7 +240,7 @@ class AdisTT(CommandLineSolver):
|
|||
edges = study.river.enable_edges()
|
||||
|
||||
for e in edges:
|
||||
name = self.get_reach_name(e)
|
||||
name = f"Reach_{e.id + 1:>3}".replace(" ", "0")
|
||||
id = name
|
||||
|
||||
n1 = f"{self.get_node_name(e.node1):3}".replace(" ", "x")
|
||||
|
|
@ -413,6 +412,9 @@ class AdisTT(CommandLineSolver):
|
|||
_nodes_names = {}
|
||||
_nodes_views = set()
|
||||
|
||||
def get_reach_name(self, reach):
|
||||
return f"Reach_{reach.pamhyr_id:>3}".replace(" ", "0")
|
||||
|
||||
#################################
|
||||
# Adis-TT in weak coupling mode #
|
||||
#################################
|
||||
|
|
@ -450,7 +452,7 @@ class AdisTTwc(AdisTT):
|
|||
|
||||
if qlog is not None:
|
||||
qlog.put("Export TEMs files")
|
||||
|
||||
|
||||
name = "TEM"
|
||||
with adistt_file_open(
|
||||
os.path.join(repertory, f"{name}.POL"), "w+"
|
||||
|
|
@ -476,8 +478,7 @@ class AdisTTwc(AdisTT):
|
|||
)
|
||||
|
||||
TEM_BCs = (
|
||||
study.river._BoundaryConditionsTemperature
|
||||
.BCs_Temperature_List
|
||||
study.river._BoundaryConditionsTemperature.BCs_Temperature_List
|
||||
)
|
||||
|
||||
if TEM_BCs:
|
||||
|
|
@ -526,7 +527,8 @@ class AdisTTwc(AdisTT):
|
|||
f.write(f"*État initial pour le polluant {TEM_name}\n")
|
||||
f.write(" ".join(("DEFAULT =",
|
||||
f"{TEM_IC_default.temperature}",
|
||||
"0.0 0.0 0.0\n"))) # fake 0 to match adists
|
||||
"0.0 0.0 0.0\n")))
|
||||
# fake 0 to match adists
|
||||
|
||||
if len(TEM_IC_default._data) != 0:
|
||||
self._export_ICs_AdisTT_Spec(
|
||||
|
|
@ -549,7 +551,8 @@ class AdisTTwc(AdisTT):
|
|||
f"{ic_spec.start_rk}",
|
||||
f"{ic_spec.end_rk}",
|
||||
f"{ic_spec.temperature}"
|
||||
f"0.0 0.0 0.0 0.0\n"))) # fake 0 to match adists
|
||||
f"0.0 0.0 0.0 0.0\n")))
|
||||
# fake 0 to match adists
|
||||
|
||||
return True
|
||||
|
||||
|
|
@ -614,7 +617,7 @@ class AdisTTwc(AdisTT):
|
|||
# ) as f:
|
||||
# files.append(f"{name}.DIF")
|
||||
|
||||
# f.write(f"*Définition des paramètres des fonctions de calcul du\n")
|
||||
# f.write(f"*Définition des paramètres des fonctions de calcul du\n")
|
||||
# f.write(f"*coefficient de diffusion\n")
|
||||
|
||||
# difAdisTT = study.river.dif_adistt.DIF_AdisTT_List
|
||||
|
|
@ -943,6 +946,10 @@ class AdisTTwc(AdisTT):
|
|||
"CCF": ("cld0", "CLD"),
|
||||
}
|
||||
|
||||
_required_weather_extensions = (
|
||||
"TDA", "HSA", "RAY", "VEN", "TNA", "QNA",
|
||||
)
|
||||
|
||||
def _export_temperature_initial_conditions(self, study, repertory):
|
||||
initial_conditions = study.river.ic_temperature.Initial_Conditions_List
|
||||
filename = "TEM.INI"
|
||||
|
|
@ -993,15 +1000,17 @@ class AdisTTwc(AdisTT):
|
|||
|
||||
def _export_weather_file(self, study, repertory, weather_parameter,
|
||||
extension):
|
||||
reach = weather_parameter.reach
|
||||
if reach is None or len(weather_parameter.data) == 0:
|
||||
return None
|
||||
|
||||
filename = f"TEM.{extension}"
|
||||
path = os.path.join(repertory, filename)
|
||||
mode = "a" if os.path.exists(path) else "w+"
|
||||
with adistt_file_open(path, mode) as f:
|
||||
reach_name = self.get_reach_name(reach)
|
||||
reach = weather_parameter.reach
|
||||
if reach is None:
|
||||
return None
|
||||
reach_name = (
|
||||
f"Reach_{study.river.get_edge_id(reach) + 1:>3}"
|
||||
.replace(" ", "0")
|
||||
)
|
||||
f.write(
|
||||
f"${reach_name} {weather_parameter.begin_rk} "
|
||||
f"{weather_parameter.end_rk}\n"
|
||||
|
|
@ -1025,6 +1034,13 @@ class AdisTTwc(AdisTT):
|
|||
if os.path.exists(path):
|
||||
os.remove(path)
|
||||
|
||||
# These files are mandatory for the AdisTT master file. They are
|
||||
# created even when no spatial time series is defined in the study.
|
||||
for extension in self._required_weather_extensions:
|
||||
adistt_file_open(
|
||||
os.path.join(repertory, f"TEM.{extension}"), "w+"
|
||||
).close()
|
||||
|
||||
initial_file = self._export_temperature_initial_conditions(
|
||||
study, repertory
|
||||
)
|
||||
|
|
@ -1047,16 +1063,20 @@ class AdisTTwc(AdisTT):
|
|||
filename = f"{name}.TEM"
|
||||
with adistt_file_open(os.path.join(repertory, filename), "w+") as f:
|
||||
defaults = study.river.weather_parameters
|
||||
for type_, (parameter, extension) in self._weather_files.items():
|
||||
for type_, (parameter, _) in self._weather_files.items():
|
||||
default = defaults.default_for_type(type_)
|
||||
if default is not None:
|
||||
prefix = "*" if extension in weather_files else ""
|
||||
f.write(f"{prefix}{parameter} = {default.value}\n")
|
||||
f.write(f"{parameter} = {default.value}\n")
|
||||
|
||||
f.write(f"file_ini = {initial_file}\n")
|
||||
f.write(f"file_cl = {boundary_file}\n")
|
||||
for extension in self._required_weather_extensions:
|
||||
f.write(
|
||||
f"file_{extension.lower()} = TEM.{extension}\n"
|
||||
)
|
||||
for _, extension in self._weather_files.values():
|
||||
if extension in weather_files:
|
||||
if (extension in weather_files
|
||||
and extension not in self._required_weather_extensions):
|
||||
f.write(
|
||||
f"file_{extension.lower()} = "
|
||||
f"{weather_files[extension]}\n"
|
||||
|
|
@ -1097,20 +1117,6 @@ class AdisTTwc(AdisTT):
|
|||
self._study = study
|
||||
name = study.name.replace(" ", "_")
|
||||
|
||||
mage_path = os.path.join(
|
||||
os.path.abspath(os.path.join(repertory, os.pardir)),
|
||||
mage_rep
|
||||
)
|
||||
source_net = os.path.join(mage_path, "net")
|
||||
destination_net = os.path.join(repertory, "net")
|
||||
|
||||
if qlog is not None:
|
||||
qlog.put("Copy Mage net directory")
|
||||
|
||||
if os.path.exists(destination_net):
|
||||
shutil.rmtree(destination_net)
|
||||
shutil.copytree(source_net, destination_net)
|
||||
|
||||
# Node names must be deterministic and identical in NET and TEM.CDT.
|
||||
self._nodes_cnt = 0
|
||||
self._nodes_names = {}
|
||||
|
|
|
|||
|
|
@ -130,8 +130,7 @@ class Mage(CommandLineSolver):
|
|||
_nodes_views = set()
|
||||
|
||||
def get_reach_name(self, reach):
|
||||
index = self._study.river.get_edge_id(reach) + 1
|
||||
return f"Reach_{index:03d}"
|
||||
return f"Reach_{reach.pamhyr_id:>3}".replace(" ", "0")
|
||||
|
||||
def get_node_name(self, node):
|
||||
"""Generate a 3 char name for node
|
||||
|
|
@ -211,7 +210,7 @@ class Mage(CommandLineSolver):
|
|||
# Write header
|
||||
edges = study.river.enable_edges()
|
||||
for edge in edges:
|
||||
name = self.get_reach_name(edge)
|
||||
name = f"Reach_{edge.pamhyr_id:>3}".replace(" ", "0")
|
||||
|
||||
with mage_file_open(
|
||||
os.path.join(repertory, "net", f"{name}.ST"),
|
||||
|
|
|
|||
|
|
@ -37,7 +37,7 @@ class WPETranslate(WeatherParametersTranslate):
|
|||
"discharge": self._dict["unit_discharge"],
|
||||
"temperature": _translate("WeatherParameters", "Temperature (°C)"),
|
||||
"humidity": _translate("WeatherParameters",
|
||||
"Humidity"),
|
||||
"Humidity"),
|
||||
"radiation": _translate("WeatherParameters", "Radiation (W/m²)"),
|
||||
"wind speed": _translate("WeatherParameters", "Wind Speed (m/s)"),
|
||||
"albedo": _translate("WeatherParameters", "Albedo"),
|
||||
|
|
|
|||
Loading…
Reference in New Issue