# Window.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 -*- import os import logging from tools import trace, timer, logger_exception from View.Tools.PamhyrWindow import PamhyrWindow from PyQt5.QtGui import ( QKeySequence, QIcon, ) from PyQt5.QtCore import ( Qt, QVariant, QAbstractTableModel, QCoreApplication, QModelIndex, pyqtSlot, QRect, QItemSelectionModel, ) from PyQt5.QtWidgets import ( QDialogButtonBox, QPushButton, QLineEdit, QFileDialog, QTableView, QAbstractItemView, QUndoStack, QShortcut, QAction, QItemDelegate, QComboBox, QVBoxLayout, QHeaderView, QTabWidget, QVBoxLayout, QToolBar, QAction, QToolButton, ) from Modules import Modules from View.InitialConditionsAdisTS.UndoCommand import ( SetCommand, ) from View.InitialConditionsAdisTS.TableDefault import ( InitialConditionTableDefaultModel, ) from View.InitialConditionsAdisTS.Table import ( InitialConditionTableModel, ComboBoxDelegate, ) from View.InitialConditionsAdisTS.translate import IcAdisTSTranslate from Solver.Mage import Mage8 _translate = QCoreApplication.translate logger = logging.getLogger() class InitialConditionsAdisTSWindow(PamhyrWindow): _pamhyr_ui = "InitialConditionsAdisTS" _pamhyr_name = "Initial condition AdisTS" def __init__(self, data=None, study=None, config=None, parent=None): self._data = [] self._data.append(data) trad = IcAdisTSTranslate() name = ( trad[self._pamhyr_name] + " - " + study.name + " - " + self._data[0].name ) super(InitialConditionsAdisTSWindow, self).__init__( title=name, study=study, config=config, trad=trad, parent=parent ) self._hash_data.append(data) self._ics_adists_lst = study.river.ic_adists self.setup_table() self.ui.setWindowTitle(self._title) def setup_table(self): path_icons = os.path.join(self._get_ui_directory(), f"ressources") table_default = self.find(QTableView, f"tableView") self._table = InitialConditionTableDefaultModel( table_view=table_default, table_headers=self._trad.get_dict("table_headers"), editable_headers=["name", "concentration", "eg", "em", "ed"], delegates={}, data=self._data, undo=self._undo_stack, trad=self._trad ) table_default.setModel(self._table) table_default.setSelectionBehavior(QAbstractItemView.SelectRows) table_default.horizontalHeader().setSectionResizeMode( QHeaderView.Stretch ) table_default.setAlternatingRowColors(True) layout = self.find(QVBoxLayout, f"verticalLayout_1") toolBar = QToolBar() layout.addWidget(toolBar) action_add = QAction(self) action_add.setIcon(QIcon(os.path.join(path_icons, f"add.png"))) action_add.triggered.connect(self.add) action_delete = QAction(self) action_delete.setIcon(QIcon(os.path.join(path_icons, f"del.png"))) action_delete.triggered.connect(self.delete) toolBar.addAction(action_add) toolBar.addAction(action_delete) self.table_spec = QTableView() layout.addWidget(self.table_spec) self._delegate_reach = ComboBoxDelegate( trad=self._trad, data=self._study.river, ic_spec_lst=self._data[0]._data, parent=self, mode="reaches" ) self._delegate_rk = ComboBoxDelegate( trad=self._trad, data=self._study.river, ic_spec_lst=self._data[0]._data, parent=self, mode="rk" ) self._table_spec = InitialConditionTableModel( table_view=self.table_spec, table_headers=self._trad.get_dict("table_headers_spec"), editable_headers=["name", "reach", "start_rk", "end_rk", "concentration", "eg", "em", "ed", "rate"], delegates={ "reach": self._delegate_reach, "start_rk": self._delegate_rk, "end_rk": self._delegate_rk }, data=self._data[0], undo=self._undo_stack, trad=self._trad, river=self._study.river ) self.table_spec.setModel(self._table_spec) self.table_spec.setSelectionBehavior(QAbstractItemView.SelectRows) self.table_spec.horizontalHeader().setSectionResizeMode( QHeaderView.Stretch ) self.table_spec.setAlternatingRowColors(True) selectionModel = self.table_spec.selectionModel() index = self.table_spec.model().index(0, 0) selectionModel.select( index, QItemSelectionModel.Rows | QItemSelectionModel.ClearAndSelect | QItemSelectionModel.Select ) self.table_spec.scrollTo(index) def index_selected_row(self): # table = self.find(QTableView, f"tableView") table = self.table_spec rows = table.selectionModel()\ .selectedRows() if len(rows) == 0: return 0 return rows[0].row() def index_selected_rows(self): # table = self.find(QTableView, f"tableView") table = self.table_spec return list( # Delete duplicate set( map( lambda i: i.row(), table.selectedIndexes() ) ) ) def move_up(self): row = self.index_selected_row() self._table.move_up(row) self._update() def move_down(self): row = self.index_selected_row() self._table.move_down(row) self._update() def _copy(self): rows = list( map( lambda row: row.row(), self.tableView.selectionModel().selectedRows() ) ) table = list( map( lambda eic: list( map( lambda k: eic[1][k], ["rk", "discharge", "elevation"] ) ), filter( lambda eic: eic[0] in rows, enumerate(self._ics.lst()) ) ) ) self.copyTableIntoClipboard(table) def _paste(self): header, data = self.parseClipboardTable() if len(data) + len(header) == 0: return logger.debug( "IC: Paste: " + f"header = {header}, " + f"data = {data}" ) try: row = self.index_selected_row() # self._table.paste(row, header, data) self._table.paste(row, [], data) except Exception as e: logger_exception(e) self._update() def _undo(self): self._table.undo() self._update() def _redo(self): self._table.redo() self._update() def add(self): rows = self.index_selected_rows() if len(self._data[0]._data) == 0 or len(rows) == 0: self._table_spec.add(0) else: self._table_spec.add(rows[0]) def delete(self): rows = self.index_selected_rows() if len(rows) == 0: return self._table_spec.delete(rows)