diff --git a/src/View/Results/CustomExportAdis.py b/src/View/Results/CustomExportAdis.py new file mode 100644 index 00000000..d8b845f3 --- /dev/null +++ b/src/View/Results/CustomExportAdis.py @@ -0,0 +1,129 @@ +# CustomPlotValuesSelectionDialog.py -- Pamhyr +# Copyright (C) 2023-2024 INRAE +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see . + +# -*- coding: utf-8 -*- + +from View.Tools.PamhyrWindow import PamhyrDialog + +from PyQt5.QtWidgets import ( + QRadioButton, QCheckBox, QVBoxLayout, +) + +from View.Results.translate import ResultsTranslate + + +class CustomExportAdisDialog(PamhyrDialog): + _pamhyr_ui = "CustomExportAdisDialog" + _pamhyr_name = "Custom Plot Selection" + + def __init__(self, pollutants, parent=None): + trad = ResultsTranslate() + super(CustomExportAdisDialog, self).__init__( + title=trad[self._pamhyr_name], + options=[], + trad=trad, + parent=parent + ) + + if pollutants is not None: + self.pollutants = pollutants + if "total_sediment" in self.pollutants: + self.pollutants.remove("total_sediment") + else: + self.pollutants = pollutants + + self._available_values_x = self._trad.get_dict("values_x") + self._available_values_y = self._trad.get_dict("values_y_pol") + + self.setup_radio_buttons_x() + self.setup_radio_buttons_pol() + self.setup_check_boxes() + + self.value = None + + def setup_radio_buttons_x(self): + self._radio = [] + layout = self.find(QVBoxLayout, "verticalLayout_x") + + for value in self._available_values_x: + btn = QRadioButton( + self._available_values_x[value], + parent=self + ) + self._radio.append((value, btn)) + layout.addWidget(btn) + + self._radio[0][1].setChecked(True) + layout.addStretch() + + def setup_radio_buttons_pol(self): + self._radio2 = [] + layout = self.find(QVBoxLayout, "verticalLayout_pol") + + for value in self.pollutants: + btn = QRadioButton( + value, + parent=self + ) + self._radio2.append((value, btn)) + layout.addWidget(btn) + + self._radio2[0][1].setChecked(True) + layout.addStretch() + + def setup_check_boxes(self): + self._check = [] + layout = self.find(QVBoxLayout, "verticalLayout_y") + + for value in self._available_values_y: + btn = QCheckBox( + self._available_values_y[value], + parent=self + ) + self._check.append((value, btn)) + layout.addWidget(btn) + + self._check[0][1].setChecked(True) + layout.addStretch() + + def accept(self): + x = next( + filter( + lambda r: r[1].isChecked(), + self._radio + ) + )[0] + + y = list( + map( + lambda b: b[0], + filter( + lambda b: b[1].isChecked(), + self._check + ) + ) + ) + + pol = next( + filter( + lambda r: r[1].isChecked(), + self._radio2 + ) + )[0] + + self.value = x, y, pol + + super().accept() diff --git a/src/View/Results/Window.py b/src/View/Results/Window.py index 0ddc87ef..13f15138 100644 --- a/src/View/Results/Window.py +++ b/src/View/Results/Window.py @@ -614,13 +614,13 @@ class ResultsWindow(PamhyrWindow): if x == "rk": timestamp = self._get_current_timestamp() first_line.append(f"Time: {timestamp}s") - val_dict = self._export_rk(timestamp, y, envelop, filename) + val_dict = self._export_rk(timestamp, y, envelop) elif x == "time": profile_id = self._get_current_profile() profile = reach.profile(profile_id) pname = profile.name if profile.name != "" else profile.rk first_line.append(f"Profile: {pname}") - val_dict = self._export_time(profile_id, y, filename) + val_dict = self._export_time(profile_id, y) with open(filename, 'w', newline='') as csvfile: writer = csv.writer(csvfile, delimiter=',', @@ -675,7 +675,7 @@ class ResultsWindow(PamhyrWindow): self._additional_plot.pop(tab_widget.tabText(index)) tab_widget.removeTab(index) - def _export_rk(self, timestamp, y, envelop, filename): + def _export_rk(self, timestamp, y, envelop): reach = self._results.river.reachs[self._get_current_reach()] dict_x = self._trad.get_dict("values_x") dict_y = self._trad.get_dict("values_y") @@ -807,7 +807,7 @@ class ResultsWindow(PamhyrWindow): return my_dict - def _export_time(self, profile, y, filename): + def _export_time(self, profile, y): reach = self._results.river.reachs[self._get_current_reach()] profile = reach.profile(profile) ts = list(self._results.get("timestamps")) diff --git a/src/View/Results/WindowAdisTS.py b/src/View/Results/WindowAdisTS.py index d11cf8d9..28c77932 100644 --- a/src/View/Results/WindowAdisTS.py +++ b/src/View/Results/WindowAdisTS.py @@ -49,8 +49,8 @@ from View.Tools.Plot.PamhyrToolbar import PamhyrPlotToolbar from View.Results.PlotSedAdis import PlotAdis_dx, PlotAdis_dt from View.Results.CustomPlot.Plot import CustomPlot -from View.Results.CustomPlot.CustomPlotValuesSelectionDialog import ( - CustomPlotValuesSelectionDialog, +from View.Results.CustomExportAdis import ( + CustomExportAdisDialog, ) from View.Results.TableAdisTS import TableModel @@ -100,7 +100,7 @@ class ResultsWindowAdisTS(PamhyrWindow): self._hash_data.append(self._results) self._additional_plot = {} - self._pol_id = [1] + self._current_pol_id = [1] self._reach_id = 0 self._profile_id = 0 @@ -114,22 +114,25 @@ class ResultsWindowAdisTS(PamhyrWindow): self.setup_connections() self.update_table_selection_reach(self._reach_id) self.update_table_selection_profile(self._profile_id) - self.update_table_selection_pol(self._pol_id) + self.update_table_selection_pol(self._current_pol_id) except Exception as e: logger_exception(e) return def set_type_pol(self): self._type_pol = [] + self._pol_id_dict = {} tmp_list = self._results.river.reach(0).profiles for pol_index in range(self._results.nb_pollutants): - if self._results.pollutants_list[pol_index] == "total_sediment": + pol_name = self._results.pollutants_list[pol_index] + if pol_name == "total_sediment": self._type_pol.append(-1) else: self._type_pol.append(len( tmp_list[0].get_ts_key( self._timestamps[0], "pols")[pol_index]) ) + self._pol_id_dict[pol_name] = pol_index def setup_table(self): self._table = {} @@ -187,7 +190,7 @@ class ResultsWindowAdisTS(PamhyrWindow): results=self._results, reach_id=self._reach_id, profile_id=self._profile_id, - pol_id=self._pol_id, + pol_id=self._current_pol_id, key="C", type_pol=self._type_pol, trad=self._trad, @@ -213,7 +216,7 @@ class ResultsWindowAdisTS(PamhyrWindow): results=self._results, reach_id=self._reach_id, profile_id=self._profile_id, - pol_id=self._pol_id, + pol_id=self._current_pol_id, key="C", type_pol=self._type_pol, trad=self._trad, @@ -239,7 +242,7 @@ class ResultsWindowAdisTS(PamhyrWindow): results=self._results, reach_id=self._reach_id, profile_id=self._profile_id, - pol_id=self._pol_id, + pol_id=self._current_pol_id, key="M", type_pol=self._type_pol, trad=self._trad, @@ -266,7 +269,7 @@ class ResultsWindowAdisTS(PamhyrWindow): results=self._results, reach_id=self._reach_id, profile_id=self._profile_id, - pol_id=self._pol_id, + pol_id=self._current_pol_id, key="M", type_pol=self._type_pol, trad=self._trad, @@ -475,6 +478,7 @@ class ResultsWindowAdisTS(PamhyrWindow): def update(self, reach_id=None, profile_id=None, pol_id=None, timestamp=None): if reach_id is not None: + self._reach_id = reach_id self.plot_cdt.set_reach(reach_id) self.plot_cdx.set_reach(reach_id) self.plot_mdx.set_reach(reach_id) @@ -486,6 +490,7 @@ class ResultsWindowAdisTS(PamhyrWindow): self.update_table_selection_profile(0) if profile_id is not None: + self._profile_id = profile_id self.plot_cdt.set_profile(profile_id) self.plot_cdx.set_profile(profile_id) self.plot_mdx.set_profile(profile_id) @@ -496,11 +501,11 @@ class ResultsWindowAdisTS(PamhyrWindow): self.update_table_selection_profile(profile_id) if pol_id is not None: - self._pol_id = [p+1 for p in pol_id] # remove total_sediment - self.plot_cdt.set_pollutant(self._pol_id) - self.plot_cdx.set_pollutant(self._pol_id) - self.plot_mdx.set_pollutant(self._pol_id) - self.plot_mdt.set_pollutant(self._pol_id) + self._current_pol_id = [p+1 for p in pol_id] # remove total_sediment + self.plot_cdt.set_pollutant(self._current_pol_id) + self.plot_cdx.set_pollutant(self._current_pol_id) + self.plot_mdx.set_pollutant(self._current_pol_id) + self.plot_mdt.set_pollutant(self._current_pol_id) if timestamp is not None: self.plot_cdt.set_timestamp(timestamp) @@ -635,53 +640,143 @@ class ResultsWindowAdisTS(PamhyrWindow): self._button_play.setIcon(self._icon_start) def export(self): + + pols = self._results.pollutants_list.copy() + dlg = CustomExportAdisDialog(pollutants=pols, + parent=self) + if dlg.exec(): + x, y, pol = dlg.value + else: + return + self.file_dialog( - select_file=False, - callback=lambda d: self.export_to(d[0]) + select_file="AnyFile", + callback=lambda f: self.export_to(f[0], x, y, pol), + default_suffix=".csv", + file_filter=["CSV (*.csv)"], ) - def export_to(self, directory): - for reach in self._results.river.reachs: - self.export_reach(reach, directory) - - def export_reach(self, reach, directory): - name = reach.name - name = name.replace(" ", "-") - - file_name = os.path.join( - directory, - f"reach_{name}.csv" - ) - - with open(file_name, 'w', newline='') as csvfile: + def export_to(self, filename, x, y, pol): + timestamps = sorted(self._results.get("timestamps")) + reach = self._results.river.reachs[self._reach_id] + first_line = [f"Study: {self._results.study.name}", + f"Reach: {reach.name}"] + if x == "rk": + timestamp = self._get_current_timestamp() + first_line.append(f"Time: {timestamp}s") + val_dict = self._export_rk(timestamp, y, pol) + elif x == "time": + profile = reach.profile(self._profile_id) + pname = profile.name if profile.name != "" else profile.rk + first_line.append(f"Profile: {pname}") + val_dict = self._export_time(self._profile_id, y, pol) + with open(filename, 'w', newline='') as csvfile: writer = csv.writer(csvfile, delimiter=',', quotechar='|', quoting=csv.QUOTE_MINIMAL) - writer.writerow(["name", "rk", "data-file"]) - for profile in reach.profiles: - p_file_name = os.path.join( - directory, - f"cs_{profile.geometry.id}.csv" - ) + dict_x = self._trad.get_dict("values_x") + header = [] + writer.writerow(first_line) + for text in val_dict.keys(): + header.append(text) + writer.writerow(header) + for row in range(len(val_dict[dict_x[x]])): + line = [] + for var in val_dict.keys(): + line.append(val_dict[var][row]) + writer.writerow(line) - writer.writerow([ - profile.name, - profile.rk, - p_file_name - ]) + def _export_rk(self, timestamp, y, pol): + reach = self._results.river.reachs[self._reach_id] + dict_x = self._trad.get_dict("values_x") + dict_y = self._trad.get_dict("values_y_pol") + my_dict = {} + my_dict[dict_x["rk"]] = reach.geometry.get_rk() + val_id = {} + val_id["unit_C"] = 0 + val_id["unit_M"] = 2 + val_id["unit_thickness"] = 2 + for unit in y: + if unit == "unit_thickness": + pol_id = 0 + else: + pol_id = self._pol_id_dict[pol] + if unit == "unit_M" and self._type_pol[pol_id] == 1: + my_dict[dict_y[unit]] = [0.0]*len(reach.profiles) + else: + my_dict[dict_y[unit]] = list(map(lambda p: + p.get_ts_key( + timestamp, "pols" + )[pol_id][val_id[unit]], + reach.profiles)) - self.export_profile(reach, profile, p_file_name) + return my_dict - def export_profile(self, reach, profile, file_name): - with open(file_name, 'w', newline='') as csvfile: - writer = csv.writer(csvfile, delimiter=',', - quotechar='|', quoting=csv.QUOTE_MINIMAL) + def _export_time(self, profile, y, pol): + reach = self._results.river.reachs[self._reach_id] + profile = reach.profile(profile) + ts = list(self._results.get("timestamps")) + ts.sort() + dict_x = self._trad.get_dict("values_x") + dict_y = self._trad.get_dict("values_y_pol") + my_dict = {} + my_dict[dict_x["time"]] = ts + val_id = {} + val_id["unit_C"] = 0 + val_id["unit_M"] = 2 + val_id["unit_thickness"] = 2 + for unit in y: + if unit == "unit_thickness": + pol_id = 0 + else: + pol_id = self._pol_id_dict[pol] + if unit == "unit_M" and self._type_pol[pol_id] == 1: + my_dict[dict_y[unit]] = [0.0]*len(ts) + else: + my_dict[dict_y[unit]] = list(map(lambda data_el: + data_el[pol_id][val_id[unit]], + profile.get_key("pols") + )) - writer.writerow(["timestamp", "z", "q"]) - timestamps = sorted(self._results.get("timestamps")) + return my_dict - for ts in timestamps: - writer.writerow([ - ts, - profile.get_ts_key(ts, "Z"), - profile.get_ts_key(ts, "Q"), - ]) + #def export_reach(self, reach, directory): + #name = reach.name + #name = name.replace(" ", "-") + + #file_name = os.path.join( + #directory, + #f"reach_{name}.csv" + #) + + #with open(file_name, 'w', newline='') as csvfile: + #writer = csv.writer(csvfile, delimiter=',', + #quotechar='|', quoting=csv.QUOTE_MINIMAL) + #writer.writerow(["name", "rk", "data-file"]) + #for profile in reach.profiles: + #p_file_name = os.path.join( + #directory, + #f"cs_{profile.geometry.id}.csv" + #) + + #writer.writerow([ + #profile.name, + #profile.rk, + #p_file_name + #]) + + #self.export_profile(reach, profile, p_file_name) + + #def export_profile(self, reach, profile, file_name): + #with open(file_name, 'w', newline='') as csvfile: + #writer = csv.writer(csvfile, delimiter=',', + #quotechar='|', quoting=csv.QUOTE_MINIMAL) + + #writer.writerow(["timestamp", "z", "q"]) + #timestamps = sorted(self._results.get("timestamps")) + + #for ts in timestamps: + #writer.writerow([ + #ts, + #profile.get_ts_key(ts, "Z"), + #profile.get_ts_key(ts, "Q"), + #]) diff --git a/src/View/ui/CustomExportAdisDialog.ui b/src/View/ui/CustomExportAdisDialog.ui new file mode 100644 index 00000000..b1c0f419 --- /dev/null +++ b/src/View/ui/CustomExportAdisDialog.ui @@ -0,0 +1,104 @@ + + + Dialog + + + + 0 + 0 + 194 + 70 + + + + Dialog + + + + + + Qt::Horizontal + + + QDialogButtonBox::Cancel|QDialogButtonBox::Ok + + + + + + + Qt::Horizontal + + + + + + + X axis: + + + + + + + + + + + Y axis: + + + + + + + + + + + Pollutant: + + + + + + + + + + + + + buttonBox + rejected() + Dialog + reject() + + + 316 + 260 + + + 286 + 274 + + + + + buttonBox + accepted() + Dialog + accept() + + + 248 + 254 + + + 157 + 274 + + + + +