Compare commits

..

No commits in common. "5287f30636a525e9c3be64eb601abf1566aeb449" and "885c8ea5028fa36139c78815345f0e86fc7bc48f" have entirely different histories.

5 changed files with 50 additions and 45 deletions

View File

@ -54,10 +54,8 @@ class Scenario(SQLSubModel):
"river_reach", "river_node", "river_reach", "river_node",
"geotiff", "reservoir", "reservoir_data", "geotiff", "reservoir", "reservoir_data",
# Temperature # Temperature
"boundary_condition_temperature", "boundary_condition_temperature", "boundary_condition_data_temperature",
"boundary_condition_data_temperature", "initial_conditions_temperature", "initial_conditions_temperature_spec",
"initial_conditions_temperature",
"initial_conditions_temperature_spec",
"weather_parameters", "weather_parameters_data", "weather_parameters", "weather_parameters_data",
] ]

View File

@ -117,6 +117,9 @@ class AdisTS(CommandLineSolver):
_nodes_names = {} _nodes_names = {}
_nodes_views = set() _nodes_views = set()
def get_reach_name(self, reach):
return f"Reach_{reach.pamhyr_id:>3}".replace(" ", "0")
def get_node_name(self, node): def get_node_name(self, node):
"""Generate a 3 char name for node """Generate a 3 char name for node
@ -179,7 +182,7 @@ class AdisTS(CommandLineSolver):
# Write header # Write header
edges = study.river.enable_edges() edges = study.river.enable_edges()
for edge in edges: for edge in edges:
name = self.get_reach_name(edge) name = f"Reach_{edge.id + 1:>3}".replace(" ", "0")
with adists_file_open( with adists_file_open(
os.path.join(repertory, "net", f"{name}.ST"), os.path.join(repertory, "net", f"{name}.ST"),
@ -274,7 +277,7 @@ class AdisTS(CommandLineSolver):
edges = study.river.enable_edges() edges = study.river.enable_edges()
for e in edges: for e in edges:
name = self.get_reach_name(e) name = f"Reach_{e.id + 1:>3}".replace(" ", "0")
id = name id = name
n1 = f"{self.get_node_name(e.node1):3}".replace(" ", "x") n1 = f"{self.get_node_name(e.node1):3}".replace(" ", "x")
@ -450,8 +453,7 @@ class AdisTS(CommandLineSolver):
_nodes_views = set() _nodes_views = set()
def get_reach_name(self, reach): def get_reach_name(self, reach):
index = self._study.river.get_edge_id(reach) + 1 return f"Reach_{reach.pamhyr_id:>3}".replace(" ", "0")
return f"Reach_{index:03d}"
################################# #################################
# Adis-TS in weak coupling mode # # Adis-TS in weak coupling mode #

View File

@ -113,8 +113,7 @@ class AdisTT(CommandLineSolver):
_nodes_views = set() _nodes_views = set()
def get_reach_name(self, reach): def get_reach_name(self, reach):
index = self._study.river.get_edge_id(reach) + 1 return f"Reach_{reach.pamhyr_id:>3}".replace(" ", "0")
return f"Reach_{index:03d}"
def get_node_name(self, node): def get_node_name(self, node):
"""Generate a 3 char name for node """Generate a 3 char name for node
@ -178,7 +177,7 @@ class AdisTT(CommandLineSolver):
# Write header # Write header
edges = study.river.enable_edges() edges = study.river.enable_edges()
for edge in edges: for edge in edges:
name = self.get_reach_name(edge) name = f"Reach_{edge.id + 1:>3}".replace(" ", "0")
with adistt_file_open( with adistt_file_open(
os.path.join(repertory, "net", f"{name}.ST"), os.path.join(repertory, "net", f"{name}.ST"),
@ -241,7 +240,7 @@ class AdisTT(CommandLineSolver):
edges = study.river.enable_edges() edges = study.river.enable_edges()
for e in edges: for e in edges:
name = self.get_reach_name(e) name = f"Reach_{e.id + 1:>3}".replace(" ", "0")
id = name id = name
n1 = f"{self.get_node_name(e.node1):3}".replace(" ", "x") n1 = f"{self.get_node_name(e.node1):3}".replace(" ", "x")
@ -413,6 +412,9 @@ class AdisTT(CommandLineSolver):
_nodes_names = {} _nodes_names = {}
_nodes_views = set() _nodes_views = set()
def get_reach_name(self, reach):
return f"Reach_{reach.pamhyr_id:>3}".replace(" ", "0")
################################# #################################
# Adis-TT in weak coupling mode # # Adis-TT in weak coupling mode #
################################# #################################
@ -450,7 +452,7 @@ class AdisTTwc(AdisTT):
if qlog is not None: if qlog is not None:
qlog.put("Export TEMs files") qlog.put("Export TEMs files")
name = "TEM" name = "TEM"
with adistt_file_open( with adistt_file_open(
os.path.join(repertory, f"{name}.POL"), "w+" os.path.join(repertory, f"{name}.POL"), "w+"
@ -476,8 +478,7 @@ class AdisTTwc(AdisTT):
) )
TEM_BCs = ( TEM_BCs = (
study.river._BoundaryConditionsTemperature study.river._BoundaryConditionsTemperature.BCs_Temperature_List
.BCs_Temperature_List
) )
if TEM_BCs: if TEM_BCs:
@ -526,7 +527,8 @@ class AdisTTwc(AdisTT):
f.write(f"*État initial pour le polluant {TEM_name}\n") f.write(f"*État initial pour le polluant {TEM_name}\n")
f.write(" ".join(("DEFAULT =", f.write(" ".join(("DEFAULT =",
f"{TEM_IC_default.temperature}", 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: if len(TEM_IC_default._data) != 0:
self._export_ICs_AdisTT_Spec( self._export_ICs_AdisTT_Spec(
@ -549,7 +551,8 @@ class AdisTTwc(AdisTT):
f"{ic_spec.start_rk}", f"{ic_spec.start_rk}",
f"{ic_spec.end_rk}", f"{ic_spec.end_rk}",
f"{ic_spec.temperature}" 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 return True
@ -614,7 +617,7 @@ class AdisTTwc(AdisTT):
# ) as f: # ) as f:
# files.append(f"{name}.DIF") # 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") # f.write(f"*coefficient de diffusion\n")
# difAdisTT = study.river.dif_adistt.DIF_AdisTT_List # difAdisTT = study.river.dif_adistt.DIF_AdisTT_List
@ -943,6 +946,10 @@ class AdisTTwc(AdisTT):
"CCF": ("cld0", "CLD"), "CCF": ("cld0", "CLD"),
} }
_required_weather_extensions = (
"TDA", "HSA", "RAY", "VEN", "TNA", "QNA",
)
def _export_temperature_initial_conditions(self, study, repertory): def _export_temperature_initial_conditions(self, study, repertory):
initial_conditions = study.river.ic_temperature.Initial_Conditions_List initial_conditions = study.river.ic_temperature.Initial_Conditions_List
filename = "TEM.INI" filename = "TEM.INI"
@ -993,15 +1000,17 @@ class AdisTTwc(AdisTT):
def _export_weather_file(self, study, repertory, weather_parameter, def _export_weather_file(self, study, repertory, weather_parameter,
extension): extension):
reach = weather_parameter.reach
if reach is None or len(weather_parameter.data) == 0:
return None
filename = f"TEM.{extension}" filename = f"TEM.{extension}"
path = os.path.join(repertory, filename) path = os.path.join(repertory, filename)
mode = "a" if os.path.exists(path) else "w+" mode = "a" if os.path.exists(path) else "w+"
with adistt_file_open(path, mode) as f: 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.write(
f"${reach_name} {weather_parameter.begin_rk} " f"${reach_name} {weather_parameter.begin_rk} "
f"{weather_parameter.end_rk}\n" f"{weather_parameter.end_rk}\n"
@ -1025,6 +1034,13 @@ class AdisTTwc(AdisTT):
if os.path.exists(path): if os.path.exists(path):
os.remove(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( initial_file = self._export_temperature_initial_conditions(
study, repertory study, repertory
) )
@ -1047,16 +1063,20 @@ class AdisTTwc(AdisTT):
filename = f"{name}.TEM" filename = f"{name}.TEM"
with adistt_file_open(os.path.join(repertory, filename), "w+") as f: with adistt_file_open(os.path.join(repertory, filename), "w+") as f:
defaults = study.river.weather_parameters 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_) default = defaults.default_for_type(type_)
if default is not None: if default is not None:
prefix = "*" if extension in weather_files else "" f.write(f"{parameter} = {default.value}\n")
f.write(f"{prefix}{parameter} = {default.value}\n")
f.write(f"file_ini = {initial_file}\n") f.write(f"file_ini = {initial_file}\n")
f.write(f"file_cl = {boundary_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(): 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.write(
f"file_{extension.lower()} = " f"file_{extension.lower()} = "
f"{weather_files[extension]}\n" f"{weather_files[extension]}\n"
@ -1097,20 +1117,6 @@ class AdisTTwc(AdisTT):
self._study = study self._study = study
name = study.name.replace(" ", "_") 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. # Node names must be deterministic and identical in NET and TEM.CDT.
self._nodes_cnt = 0 self._nodes_cnt = 0
self._nodes_names = {} self._nodes_names = {}

View File

@ -130,8 +130,7 @@ class Mage(CommandLineSolver):
_nodes_views = set() _nodes_views = set()
def get_reach_name(self, reach): def get_reach_name(self, reach):
index = self._study.river.get_edge_id(reach) + 1 return f"Reach_{reach.pamhyr_id:>3}".replace(" ", "0")
return f"Reach_{index:03d}"
def get_node_name(self, node): def get_node_name(self, node):
"""Generate a 3 char name for node """Generate a 3 char name for node
@ -211,7 +210,7 @@ class Mage(CommandLineSolver):
# Write header # Write header
edges = study.river.enable_edges() edges = study.river.enable_edges()
for edge in edges: for edge in edges:
name = self.get_reach_name(edge) name = f"Reach_{edge.pamhyr_id:>3}".replace(" ", "0")
with mage_file_open( with mage_file_open(
os.path.join(repertory, "net", f"{name}.ST"), os.path.join(repertory, "net", f"{name}.ST"),

View File

@ -37,7 +37,7 @@ class WPETranslate(WeatherParametersTranslate):
"discharge": self._dict["unit_discharge"], "discharge": self._dict["unit_discharge"],
"temperature": _translate("WeatherParameters", "Temperature (°C)"), "temperature": _translate("WeatherParameters", "Temperature (°C)"),
"humidity": _translate("WeatherParameters", "humidity": _translate("WeatherParameters",
"Humidity"), "Humidity"),
"radiation": _translate("WeatherParameters", "Radiation (W/m²)"), "radiation": _translate("WeatherParameters", "Radiation (W/m²)"),
"wind speed": _translate("WeatherParameters", "Wind Speed (m/s)"), "wind speed": _translate("WeatherParameters", "Wind Speed (m/s)"),
"albedo": _translate("WeatherParameters", "Albedo"), "albedo": _translate("WeatherParameters", "Albedo"),