Compare commits

..

6 Commits

Author SHA1 Message Date
Dylan Jeannin 6d16d9dff4 Fix PEP8 2026-07-06 15:45:47 +02:00
Dylan Jeannin b60ab881b1 Weather parameters: time series for each parameter, add/del/sort functions working 2026-07-06 15:37:45 +02:00
Dylan Jeannin f2d76a0d67 WeatherParameters: button connections in another function to clear code 2026-07-06 12:31:01 +02:00
Dylan Jeannin 5dcc82d81e WeatherParameters: Correct add and implement del function on tab_spec 2026-07-06 11:47:37 +02:00
Dylan Jeannin a01e47f205 Weather Parameters: Display relative parameters in spec tab 2026-07-06 10:26:49 +02:00
Dylan Jeannin c87563b7ed Cleaning some code 2026-07-06 09:57:05 +02:00
11 changed files with 1038 additions and 65 deletions

View File

@ -421,7 +421,7 @@ class WeatherParameters(SQLSubModel):
self.modified() self.modified()
@property @property
def wptype(self): def type(self):
return self._type return self._type
@property @property
@ -541,6 +541,16 @@ class WeatherParameters(SQLSubModel):
) )
self.modified() self.modified()
def set_as_deleted_i(self, indexes):
for i in indexes:
self._data[i].set_as_deleted()
self.modified()
def set_as_not_deleted_i(self, indexes):
for i in indexes:
self._data[i].set_as_not_deleted()
self.modified()
def delete(self, els): def delete(self, els):
self._data = list( self._data = list(
filter( filter(

View File

@ -76,7 +76,7 @@ class SpecificHumidity(WeatherParameters):
) )
self._type = "SH" self._type = "SH"
self._header = ["time", "discharge"] self._header = ["time", "humidity percentage"]
self._types = [SpecificHumidity.time_convert, float] self._types = [SpecificHumidity.time_convert, float]
@ -89,7 +89,7 @@ class GlobalRadiation(WeatherParameters):
) )
self._type = "GR" self._type = "GR"
self._header = ["time", "discharge"] self._header = ["time", "radiation"]
self._types = [GlobalRadiation.time_convert, float] self._types = [GlobalRadiation.time_convert, float]
@ -102,7 +102,7 @@ class ReferenceWindSpeed(WeatherParameters):
) )
self._type = "RWS" self._type = "RWS"
self._header = ["time", "discharge"] self._header = ["time", "wind speed"]
self._types = [ReferenceWindSpeed.time_convert, float] self._types = [ReferenceWindSpeed.time_convert, float]
@ -115,7 +115,7 @@ class GroundwaterTemperature(WeatherParameters):
) )
self._type = "GT" self._type = "GT"
self._header = ["time", "discharge"] self._header = ["time", "temperature"]
self._types = [GroundwaterTemperature.time_convert, float] self._types = [GroundwaterTemperature.time_convert, float]
@ -141,7 +141,7 @@ class Albedo(WeatherParameters):
) )
self._type = "ALB" self._type = "ALB"
self._header = ["time", "discharge"] self._header = ["time", "albedo"]
self._types = [Albedo.time_convert, float] self._types = [Albedo.time_convert, float]
@ -154,7 +154,7 @@ class ShadingCoefficient(WeatherParameters):
) )
self._type = "SC" self._type = "SC"
self._header = ["time", "discharge"] self._header = ["time", "shading coefficient"]
self._types = [ShadingCoefficient.time_convert, float] self._types = [ShadingCoefficient.time_convert, float]
@ -167,5 +167,5 @@ class CloudCoverFraction(WeatherParameters):
) )
self._type = "CCF" self._type = "CCF"
self._header = ["time", "discharge"] self._header = ["time", "cloud cover fraction"]
self._types = [CloudCoverFraction.time_convert, float] self._types = [CloudCoverFraction.time_convert, float]

View File

@ -0,0 +1,104 @@
# Plot.py -- Pamhyr
# Copyright (C) 2023-2025 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 <https://www.gnu.org/licenses/>.
# -*- coding: utf-8 -*-
import logging
from datetime import datetime
from tools import timer, trace
from View.Tools.PamhyrPlot import PamhyrPlot
from PyQt5.QtCore import (
QCoreApplication
)
_translate = QCoreApplication.translate
logger = logging.getLogger()
class Plot(PamhyrPlot):
def __init__(self, mode="time", data=None,
trad=None, canvas=None, toolbar=None,
parent=None):
super(Plot, self).__init__(
canvas=canvas,
trad=trad,
data=data,
toolbar=toolbar,
parent=parent
)
self._table_headers = self._trad.get_dict("table_headers")
header = self.data.header
self.label_x = self._table_headers[header[0]]
self.label_y = self._table_headers[header[1]]
self._mode = mode
self._isometric_axis = False
self._auto_relim_update = True
self._autoscale_update = False
def custom_ticks(self):
if self.data.header[0] != "time":
return
self.set_ticks_time_formater()
@timer
def draw(self):
self.init_axes()
if len(self.data) == 0:
self._init = False
return
self.draw_data()
self.custom_ticks()
self.idle()
self._init = True
def draw_data(self):
# Plot data
x = list(map(lambda v: v[0], self.data.data))
y = list(map(lambda v: v[1], self.data.data))
self._line, = self.canvas.axes.plot(
x, y,
color=self.color_plot,
**self.plot_default_kargs
)
@timer
def update(self, ind=None):
if not self._init:
self.draw()
return
self.update_data()
self.update_idle()
def update_data(self):
x = list(map(lambda v: v[0], self.data.data))
y = list(map(lambda v: v[1], self.data.data))
self._line.set_data(x, y)

View File

@ -0,0 +1,218 @@
# Table.py -- Pamhyr
# Copyright (C) 2023-2025 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 <https://www.gnu.org/licenses/>.
# -*- coding: utf-8 -*-
import logging
import traceback
from datetime import date, time, datetime, timedelta
from tools import (
trace, timer,
timestamp_to_old_pamhyr_date,
old_pamhyr_date_to_timestamp
)
from View.Tools.PamhyrTable import PamhyrTableModel
from PyQt5.QtCore import (
Qt, QVariant, QAbstractTableModel,
QCoreApplication, QModelIndex, pyqtSlot,
QRect, QTime, QDateTime,
)
from PyQt5.QtWidgets import (
QTableView, QAbstractItemView, QSpinBox,
QTimeEdit, QDateTimeEdit, QItemDelegate,
)
from Model.WeatherParameters.WeatherParametersTypes import (
NotDefined, WeatherParam, AirTemperature, SpecificHumidity,
GlobalRadiation, ReferenceWindSpeed, GroundwaterTemperature,
GroundwaterFlowRate, Albedo, ShadingCoefficient, CloudCoverFraction
)
from View.WeatherParameters.Edit.UndoCommand import (
SetDataCommand, AddCommand, DelCommand,
SortCommand, MoveCommand, PasteCommand,
)
from View.WeatherParameters.Edit.translate import *
_translate = QCoreApplication.translate
logger = logging.getLogger()
class TableModel(PamhyrTableModel):
def data(self, index, role):
if role == Qt.TextAlignmentRole:
return Qt.AlignHCenter | Qt.AlignVCenter
if role != Qt.ItemDataRole.DisplayRole:
return QVariant()
row = index.row()
column = index.column()
value = QVariant()
if 0 <= column < 2:
v = self._data.get_i(row)[column]
if self._data.get_type_column(column) == float:
value = f"{v:.4f}"
elif self._data.header[column] == "time":
if self._opt_data == "time":
value = timestamp_to_old_pamhyr_date(int(v))
else:
value = str(datetime.fromtimestamp(v))
else:
value = f"{v}"
return value
def setData(self, index, value, role=Qt.EditRole):
if not index.isValid() or role != Qt.EditRole:
return False
if self.is_same_data(index, value):
return False
row = index.row()
column = index.column()
try:
self._undo.push(
SetDataCommand(
self._data, row, column, value
)
)
except Exception as e:
logger.info(e)
logger.debug(traceback.format_exc())
self.update()
return True
def add(self, row, parent=QModelIndex()):
self.beginInsertRows(parent, row, row - 1)
self._undo.push(
AddCommand(
self._data, row
)
)
self.endInsertRows()
self.update()
def delete(self, rows, parent=QModelIndex()):
self.beginRemoveRows(parent, rows[0], rows[-1])
self._undo.push(
DelCommand(
self._data, rows
)
)
self.endRemoveRows()
def sort(self, _reverse, parent=QModelIndex()):
self.layoutAboutToBeChanged.emit()
self._undo.push(
SortCommand(
self._data, _reverse
)
)
self.layoutAboutToBeChanged.emit()
self.update()
def move_up(self, row, parent=QModelIndex()):
if row <= 0:
return
target = row + 2
self.beginMoveRows(parent, row - 1, row - 1, parent, target)
self._undo_stack.push(
MoveCommand(
self._data, "up", row
)
)
self.endMoveRows()
self.update()
def move_down(self, index, parent=QModelIndex()):
row = index.row()
if row >= len(self._data):
return
target = row
self.beginMoveRows(parent, row + 1, row + 1, parent, target)
self._undo_stack.push(
MoveCommand(
self._data, "down", row
)
)
self.endMoveRows()
self.update()
def paste(self, row, header, data):
if len(data) == 0:
return
self.layoutAboutToBeChanged.emit()
self._undo.push(
PasteCommand(
self._data, row,
list(
map(
lambda d: self._data.new_from_data(header, d),
data
)
)
)
)
self.layoutAboutToBeChanged.emit()
self.update()
def auto_sort(self):
self.layoutAboutToBeChanged.emit()
self._data.sort(key=lambda x: x[0])
self.layoutAboutToBeChanged.emit()
def update(self):
# self.auto_sort()
self.layoutChanged.emit()
def get(self, row):
if row < 0 or row >= len(self._lst):
return None
return self._lst[row]
def _global_row(self, row):
wp = self.get(row)
if wp is None:
return None
return self._data.index(wp)

View File

@ -0,0 +1,167 @@
# UndoCommand.py -- Pamhyr
# Copyright (C) 2023-2025 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 <https://www.gnu.org/licenses/>.
# -*- coding: utf-8 -*-
from copy import deepcopy
from tools import trace, timer
from PyQt5.QtWidgets import (
QMessageBox, QUndoCommand, QUndoStack,
)
from Model.WeatherParameters.WeatherParameters import WeatherParameters
class SetDataCommand(QUndoCommand):
def __init__(self, data, index, column, new_value):
QUndoCommand.__init__(self)
self._data = data
self._index = index
self._column = column
self._old = self._data.get_i(self._index)[self._column]
_type = self._data.get_type_column(self._column)
self._new = _type(new_value)
def undo(self):
self._data._set_i_c_v(self._index, self._column, self._old)
def redo(self):
self._data._set_i_c_v(self._index, self._column, self._new)
class AddCommand(QUndoCommand):
def __init__(self, data, index):
QUndoCommand.__init__(self)
self._data = data
self._index = index
self._new = None
def undo(self):
self._data.delete_i([self._index])
def redo(self):
if self._new is None:
self._new = self._data.add(self._index)
else:
self._data.insert(self._index, self._new)
class DelCommand(QUndoCommand):
def __init__(self, data, rows):
QUndoCommand.__init__(self)
self._data = data
self._rows = rows
self._wp = []
for row in rows:
self._wp.append((row, self._data.get_i(row)))
self._wp.sort()
def undo(self):
self._data.set_as_not_deleted_i(self._rows)
def redo(self):
self._data.set_as_deleted_i(self._rows)
class SortCommand(QUndoCommand):
def __init__(self, data, _reverse):
QUndoCommand.__init__(self)
self._data = data
self._reverse = _reverse
self._old = self._data.data
self._indexes = None
def undo(self):
ll = self._data.data
self._data.sort(
key=lambda x: self._indexes[ll.index(x)]
)
def redo(self):
self._data.sort(
_reverse=self._reverse,
key=lambda x: x[0]
)
if self._indexes is None:
self._indexes = list(
map(
lambda p: self._old.index(p),
self._data.data
)
)
self._old = None
class MoveCommand(QUndoCommand):
def __init__(self, data, up, i):
QUndoCommand.__init__(self)
self._data = data
self._up = up == "up"
self._i = i
def undo(self):
if self._up:
self._data.move_up(self._i)
else:
self._data.move_down(self._i)
def redo(self):
if self._up:
self._data.move_up(self._i)
else:
self._data.move_down(self._i)
class PasteCommand(QUndoCommand):
def __init__(self, data, row, wps):
QUndoCommand.__init__(self)
self._data = data
self._row = row
self._wps = wps
self._wps.reverse()
def undo(self):
self._data.delete(self._wps)
def redo(self):
for bc in self._wps:
self._data.insert(self._row, bc)
class DuplicateCommand(QUndoCommand):
def __init__(self, data, rows, bc):
QUndoCommand.__init__(self)
self._data = data
self._rows = rows
self._wp = deepcopy(bc)
self._wp.reverse()
def undo(self):
self._data.delete(self._wp)
def redo(self):
for bc in self._wp:
self._data.insert(self._rows[0], bc)

View File

@ -0,0 +1,232 @@
# Window.py -- Pamhyr
# Copyright (C) 2023-2025 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 <https://www.gnu.org/licenses/>.
# -*- coding: utf-8 -*-
from tools import timer, trace
from View.Tools.PamhyrWindow import PamhyrWindow
from View.Tools.PamhyrDelegate import PamhyrExTimeDelegate
from PyQt5.QtGui import (
QKeySequence,
)
from PyQt5.QtCore import (
Qt, QVariant, QAbstractTableModel, QCoreApplication,
)
from PyQt5.QtWidgets import (
QDialogButtonBox, QPushButton, QLineEdit,
QFileDialog, QTableView, QAbstractItemView,
QUndoStack, QShortcut, QAction, QItemDelegate,
QHeaderView,
)
from View.Tools.Plot.PamhyrCanvas import MplCanvas
from View.Tools.Plot.PamhyrToolbar import PamhyrPlotToolbar
from View.WeatherParameters.Edit.translate import WPETranslate
from View.WeatherParameters.Edit.Table import TableModel
from View.WeatherParameters.Edit.Plot import Plot
_translate = QCoreApplication.translate
class EditWeatherParametersWindow(PamhyrWindow):
_pamhyr_ui = "EditWeatherParameters"
_pamhyr_name = "Edit weather parameters"
def __init__(self, data=None,
study=None, config=None,
parent=None):
self._data = data
trad = WPETranslate()
name = trad[self._pamhyr_name]
if self._data is not None:
edge_name = (self._data.reach.name if self._data.reach is not None
else trad['not_associated'])
name += (
f" - {study.name} " +
f" - {self._data.name} ({self._data.id}) " +
# f"({trad.get_dict('long_types')[self._data.type]} - " +
f"{edge_name})"
)
super(EditWeatherParametersWindow, self).__init__(
title=name,
study=study,
config=config,
trad=trad,
parent=parent
)
self._hash_data.append(data)
self.setup_table()
self.setup_plot()
self.setup_connections()
def setup_table(self):
self._delegate_time = PamhyrExTimeDelegate(
data=self._data,
mode=self._study.time_system,
parent=self
)
headers = {}
table_headers = self._trad.get_dict("table_headers")
for h in self._data.header:
headers[h] = table_headers[h]
if self._study.is_editable():
editable_headers = self._data.header
else:
editable_headers = []
table = self.find(QTableView, "tableView")
self._table = TableModel(
table_view=table,
table_headers=headers,
editable_headers=editable_headers,
delegates={
# "time": self._delegate_time,
},
data=self._data,
undo=self._undo_stack,
trad=self._trad,
opt_data=self._study.time_system
)
table.setModel(self._table)
table.setSelectionBehavior(QAbstractItemView.SelectRows)
table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
table.setAlternatingRowColors(True)
def setup_plot(self):
self.canvas = MplCanvas(width=5, height=4, dpi=100)
self.canvas.setObjectName("canvas")
self.toolbar = PamhyrPlotToolbar(
self.canvas, self
)
self.verticalLayout.addWidget(self.toolbar)
self.verticalLayout.addWidget(self.canvas)
self.plot = Plot(
canvas=self.canvas,
data=self._data,
mode=self._study.time_system,
toolbar=self.toolbar,
trad=self._trad,
parent=self
)
self.plot.draw()
def setup_connections(self):
if self._study.is_editable():
self.find(QAction, "action_add").triggered.connect(self.add)
self.find(QAction, "action_del").triggered.connect(self.delete)
self.find(QAction, "action_sort").triggered.connect(self.sort)
self._table.dataChanged.connect(self.update)
self._table.layoutChanged.connect(self.update)
def update(self):
self.plot.update()
def index_selected_row(self):
table = self.find(QTableView, "tableView")
return table.selectionModel()\
.selectedRows()[0]\
.row()
def index_selected_rows(self):
table = self.find(QTableView, "tableView")
return list(
# Delete duplicate
set(
map(
lambda i: i.row(),
table.selectedIndexes()
)
)
)
def add(self):
rows = self.index_selected_rows()
if len(self._data) == 0 or len(rows) == 0:
self._table.add(0)
else:
self._table.add(rows[0])
self.plot.update()
def delete(self):
rows = self.index_selected_rows()
if len(rows) == 0:
return
self._table.delete(rows)
self.plot.update()
def sort(self):
self._table.sort(False)
self.plot.update()
def move_up(self):
row = self.index_selected_row()
self._table.move_up(row)
self.plot.update()
def move_down(self):
row = self.index_selected_row()
self._table.move_down(row)
self.plot.update()
def _copy(self):
rows = self.index_selected_rows()
table = []
table.append(self._data.header)
data = self._data.data
for row in rows:
table.append(list(data[row]))
self.copyTableIntoClipboard(table)
def _paste(self):
header, data = self.parseClipboardTable()
if len(data) == 0:
return
row = 0
rows = self.index_selected_rows()
if len(rows) != 0:
row = rows[0]
self._table.paste(row, header, data)
self.plot.update()
def _undo(self):
self._table.undo()
self.plot.update()
def _redo(self):
self._table.redo()
self.plot.update()

View File

@ -0,0 +1,48 @@
# translate.py -- Pamhyr
# Copyright (C) 2023-2025 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 <https://www.gnu.org/licenses/>.
# -*- coding: utf-8 -*-
from PyQt5.QtCore import QCoreApplication
from View.Translate import MainTranslate
from View.WeatherParameters.translate import WeatherParametersTranslate
_translate = QCoreApplication.translate
class WPETranslate(WeatherParametersTranslate):
def __init__(self):
super(WPETranslate, self).__init__()
self._dict["Edit weather parameters"] = _translate(
"WeatherParameters", "Edit weather parameters"
)
self._sub_dict["table_headers"] = {
"time": self._dict["time"],
"date": self._dict["date"],
"discharge": self._dict["unit_discharge"],
"temperature": _translate("WeatherParameters", "Temperature (°C)"),
"humidity percentage": _translate("WeatherParameters",
"Humidity Percentage"),
"radiation": _translate("WeatherParameters", "Radiation (W/m²)"),
"wind speed": _translate("WeatherParameters", "Wind Speed (m/s)"),
"albedo": _translate("WeatherParameters", "Albedo"),
"shading coefficient": _translate("WeatherParameters",
"Shading Coefficient"),
"cloud cover fraction": _translate("WeatherParameters",
"Cloud Cover Fraction"),
}

View File

@ -68,8 +68,8 @@ class ComboBoxDelegate(QItemDelegate):
self.editor = QComboBox(parent) self.editor = QComboBox(parent)
if self._mode == "rk": if self._mode == "rk":
wp = None wp = index.model().get(index.row())
if self._weather_param_lst is not None: if wp is None and self._weather_param_lst is not None:
wp = self._weather_param_lst.get(index.row()) wp = self._weather_param_lst.get(index.row())
if wp is None or wp.reach is None: if wp is None or wp.reach is None:
self.editor.addItems( self.editor.addItems(
@ -102,8 +102,8 @@ class ComboBoxDelegate(QItemDelegate):
if self._mode == "rk": if self._mode == "rk":
value = None value = None
wp = None wp = index.model().get(index.row())
if self._weather_param_lst is not None: if wp is None and self._weather_param_lst is not None:
wp = self._weather_param_lst.get(index.row()) wp = self._weather_param_lst.get(index.row())
if wp is not None and wp.reach is not None: if wp is not None and wp.reach is not None:
profiles = list( profiles = list(
@ -135,21 +135,39 @@ class ComboBoxDelegate(QItemDelegate):
class WeatherParametersTableModel(PamhyrTableModel): class WeatherParametersTableModel(PamhyrTableModel):
def __init__(self, river=None, data=None, **kwargs): def __init__(self, river=None, data=None, **kwargs):
self._river = river self._river = river
super(WeatherParametersTableModel, self).__init__(data=data, **kwargs)
self._data = data self._data = data
self._type = "ND" self._type = "ND"
super(WeatherParametersTableModel, self).__init__(data=data, **kwargs)
def _setup_lst(self): def _setup_lst(self):
if self._type == "ND" or self._data is None:
self._lst = []
return
self._lst = self._data.lst self._lst = self._data.lst
self._lst = list(filter(lambda wp: wp.type == self._type, self._lst))
def update_tab_spec(self, type="ND"): def update_tab_spec(self, type="ND"):
# self._data = data
self._type = type self._type = type
self._setup_lst() self._setup_lst()
self.layoutChanged.emit() self.layoutChanged.emit()
def get(self, row):
if row < 0 or row >= len(self._lst):
return None
return self._lst[row]
def _global_row(self, row):
wp = self.get(row)
if wp is None:
return None
return self._data.index(wp)
def _global_insert_row(self, row):
if row < len(self._lst):
return self._global_row(row)
return len(self._data)
def rowCount(self, parent): def rowCount(self, parent):
return len(self._lst) return len(self._lst)
@ -189,32 +207,36 @@ class WeatherParametersTableModel(PamhyrTableModel):
column = index.column() column = index.column()
try: try:
global_row = self._global_row(row)
if global_row is None:
return False
if self._headers[column] == "start_rk": if self._headers[column] == "start_rk":
_edge = self._data.get(row).reach _edge = self._data.get(global_row).reach
_begin_rk = next( _begin_rk = next(
p for p in _edge.reach.profiles p for p in _edge.reach.profiles
if p.pamhyr_id == value if p.pamhyr_id == value
) )
self._undo.push( self._undo.push(
SetBeginCommand( SetBeginCommand(
self._data, row, _begin_rk self._data, global_row, _begin_rk
) )
) )
elif self._headers[column] == "end_rk": elif self._headers[column] == "end_rk":
_edge = self._data.get(row).reach _edge = self._data.get(global_row).reach
_end_rk = next( _end_rk = next(
p for p in _edge.reach.profiles p for p in _edge.reach.profiles
if p.pamhyr_id == value if p.pamhyr_id == value
) )
self._undo.push( self._undo.push(
SetEndCommand( SetEndCommand(
self._data, row, _end_rk self._data, global_row, _end_rk
) )
) )
elif self._headers[column] == "reach": elif self._headers[column] == "reach":
self._undo.push( self._undo.push(
SetEdgeCommand( SetEdgeCommand(
self._data, row, self._river.edge(value) self._data, global_row, self._river.edge(value)
) )
) )
except Exception as e: except Exception as e:
@ -225,11 +247,12 @@ class WeatherParametersTableModel(PamhyrTableModel):
return True return True
def add(self, row, parent=QModelIndex()): def add(self, row, parent=QModelIndex()):
self.beginInsertRows(parent, row, row - 1) self.beginInsertRows(parent, row, row)
self._undo.push( self._undo.push(
AddCommand( AddCommand(
self._data, self._type, self._lst, row self._data, self._type, self._lst,
self._global_insert_row(row)
) )
) )
@ -239,15 +262,17 @@ class WeatherParametersTableModel(PamhyrTableModel):
def delete(self, rows, parent=QModelIndex()): def delete(self, rows, parent=QModelIndex()):
self.beginRemoveRows(parent, rows[0], rows[-1]) self.beginRemoveRows(parent, rows[0], rows[-1])
data_rows = { data_rows = {
id(ica): i for i, ica in enumerate(self._data._data) id(wp): i for i, wp in enumerate(self._data.lst)
} }
self._undo.push( self._undo.push(
DelCommand( DelCommand(
self._data, wps=self._data,
self._data._data, rows=[
[data_rows[id(self._lst[row])] for row in rows] data_rows[id(self._lst[row])]
for row in rows
if 0 <= row < len(self._lst)
]
) )
) )

View File

@ -150,21 +150,20 @@ class AddCommand(QUndoCommand):
class DelCommand(QUndoCommand): class DelCommand(QUndoCommand):
def __init__(self, data, wps_spec, rows): def __init__(self, wps, rows):
QUndoCommand.__init__(self) QUndoCommand.__init__(self)
self._data = data self._wps = wps
self._wps_spec = wps_spec
self._rows = rows self._rows = rows
self._ic = [] self._wp = []
for row in rows: for row in rows:
self._ic.append((row, self._wps_spec[row])) self._wp.append(self._wps.get(row))
self._ic.sort()
def undo(self): def undo(self):
for row, el in self._ic: for el in self._wp:
self._data.insert(row, el) el.set_as_not_deleted()
def redo(self): def redo(self):
self._data.delete_i(self._rows) for el in self._wp:
el.set_as_deleted()

View File

@ -57,6 +57,8 @@ from View.WeatherParameters.Table import (
from View.WeatherParameters.translate import WeatherParametersTranslate from View.WeatherParameters.translate import WeatherParametersTranslate
from View.WeatherParameters.Edit.Window import EditWeatherParametersWindow
from Solver.Mage import Mage8 from Solver.Mage import Mage8
_translate = QCoreApplication.translate _translate = QCoreApplication.translate
@ -70,7 +72,6 @@ class WeatherParametersWindow(PamhyrWindow):
def __init__(self, data=None, study=None, def __init__(self, data=None, study=None,
config=None, parent=None): config=None, parent=None):
# self._data = []
self._data = data self._data = data
trad = WeatherParametersTranslate() trad = WeatherParametersTranslate()
@ -95,10 +96,38 @@ class WeatherParametersWindow(PamhyrWindow):
"AT", "SH", "GR", "RWS", "GT", "GFR", "ALB", "SC", "CCF" "AT", "SH", "GR", "RWS", "GT", "GFR", "ALB", "SC", "CCF"
] ]
self.setup_table() self.setup_table()
self.setup_connections()
def setup_table(self): def setup_connections(self):
path_icons = os.path.join(self._get_ui_directory(), f"ressources") path_icons = os.path.join(self._get_ui_directory(), f"ressources")
layout = self.find(QVBoxLayout, f"verticalLayout_1")
toolBar = QToolBar()
layout.addWidget(toolBar)
# Add buttons to the toolbar
action_add = QAction(self)
action_add.setIcon(QIcon(os.path.join(path_icons, f"add.png")))
if self._study.is_editable():
action_add.triggered.connect(self.add)
action_delete = QAction(self)
action_delete.setIcon(QIcon(os.path.join(path_icons, f"del.png")))
if self._study.is_editable():
action_delete.triggered.connect(self.delete)
action_edit = QAction(self)
action_edit.setIcon(QIcon(os.path.join(path_icons, f"edit.png")))
action_edit.triggered.connect(self.edit)
toolBar.addAction(action_add)
toolBar.addAction(action_delete)
toolBar.addAction(action_edit)
layout.addWidget(self.table_spec)
def setup_table(self):
self.table_default = self.find(QTableView, f"tableView") self.table_default = self.find(QTableView, f"tableView")
if self._study.is_editable(): if self._study.is_editable():
@ -126,26 +155,7 @@ class WeatherParametersWindow(PamhyrWindow):
self.on_table_default_selection_changed self.on_table_default_selection_changed
) )
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")))
if self._study.is_editable():
action_add.triggered.connect(self.add)
action_delete = QAction(self)
action_delete.setIcon(QIcon(os.path.join(path_icons, f"del.png")))
if self._study.is_editable():
action_delete.triggered.connect(self.delete)
toolBar.addAction(action_add)
toolBar.addAction(action_delete)
self.table_spec = QTableView() self.table_spec = QTableView()
layout.addWidget(self.table_spec)
self._delegate_reach = ComboBoxDelegate( self._delegate_reach = ComboBoxDelegate(
trad=self._trad, trad=self._trad,
@ -213,14 +223,10 @@ class WeatherParametersWindow(PamhyrWindow):
self._selected_default_row_index self._selected_default_row_index
] ]
) )
# self._data = self._study.river.weather_parameters
# self._delegate_rk.weather_param_lst = (
# self._study.river.weather_parameters)
return self._selected_default_row_index return self._selected_default_row_index
def index_selected_row(self): def index_selected_row(self):
# table = self.find(QTableView, f"tableView")
table = self.table_spec table = self.table_spec
rows = table.selectionModel()\ rows = table.selectionModel()\
.selectedRows() .selectedRows()
@ -231,7 +237,6 @@ class WeatherParametersWindow(PamhyrWindow):
return rows[0].row() return rows[0].row()
def index_selected_rows(self): def index_selected_rows(self):
# table = self.find(QTableView, f"tableView")
table = self.table_default table = self.table_default
return list( return list(
# Delete duplicate # Delete duplicate
@ -243,6 +248,18 @@ class WeatherParametersWindow(PamhyrWindow):
) )
) )
def index_selected_rows_spec(self):
table = self.table_spec
return list(
# Delete duplicate
set(
map(
lambda i: i.row(),
table.selectedIndexes()
)
)
)
def move_up(self): def move_up(self):
row = self.index_selected_row() row = self.index_selected_row()
self._table.move_up(row) self._table.move_up(row)
@ -292,7 +309,6 @@ class WeatherParametersWindow(PamhyrWindow):
try: try:
row = self.index_selected_row() row = self.index_selected_row()
# self._table.paste(row, header, data)
self._table.paste(row, [], data) self._table.paste(row, [], data)
except Exception as e: except Exception as e:
logger_exception(e) logger_exception(e)
@ -332,7 +348,31 @@ class WeatherParametersWindow(PamhyrWindow):
self._table_spec.add(rows[0]) self._table_spec.add(rows[0])
def delete(self): def delete(self):
rows = self.index_selected_rows() rows = self.index_selected_rows_spec()
if len(rows) == 0: if len(rows) == 0:
return return
self._table_spec.delete(rows) self._table_spec.delete(rows)
def edit(self):
rows = self.index_selected_rows_spec()
data_rows = {
id(wp): i for i, wp in enumerate(self._data.lst)
}
rows = [data_rows[id(self._table_spec._lst[r])]
for r in rows
if 0 <= r < len(self._table_spec._lst)]
for row in rows:
data = self._data.get(row)
if self.sub_window_exists(
EditWeatherParametersWindow,
data=[self._study, None, data]
):
continue
win = EditWeatherParametersWindow(
data=data,
study=self._study,
parent=self
)
win.show()

View File

@ -0,0 +1,130 @@
<?xml version="1.0" encoding="UTF-8"?>
<ui version="4.0">
<class>MainWindow</class>
<widget class="QMainWindow" name="MainWindow">
<property name="geometry">
<rect>
<x>0</x>
<y>0</y>
<width>800</width>
<height>450</height>
</rect>
</property>
<property name="sizePolicy">
<sizepolicy hsizetype="Expanding" vsizetype="Preferred">
<horstretch>0</horstretch>
<verstretch>0</verstretch>
</sizepolicy>
</property>
<property name="windowTitle">
<string>MainWindow</string>
</property>
<property name="locale">
<locale language="English" country="Europe"/>
</property>
<widget class="QWidget" name="centralwidget">
<layout class="QGridLayout" name="gridLayout">
<item row="0" column="0">
<widget class="QSplitter" name="splitter">
<property name="sizePolicy">
<sizepolicy hsizetype="MinimumExpanding" vsizetype="Preferred">
<horstretch>0</horstretch>
<verstretch>0</verstretch>
</sizepolicy>
</property>
<property name="orientation">
<enum>Qt::Horizontal</enum>
</property>
<widget class="QTableView" name="tableView"/>
<widget class="QWidget" name="verticalLayoutWidget">
<layout class="QVBoxLayout" name="verticalLayout"/>
</widget>
</widget>
</item>
</layout>
</widget>
<widget class="QMenuBar" name="menubar">
<property name="geometry">
<rect>
<x>0</x>
<y>0</y>
<width>800</width>
<height>22</height>
</rect>
</property>
</widget>
<widget class="QToolBar" name="toolBar">
<property name="windowTitle">
<string>toolBar</string>
</property>
<attribute name="toolBarArea">
<enum>TopToolBarArea</enum>
</attribute>
<attribute name="toolBarBreak">
<bool>false</bool>
</attribute>
<addaction name="action_add"/>
<addaction name="action_del"/>
<addaction name="action_sort"/>
</widget>
<action name="action_add">
<property name="checkable">
<bool>false</bool>
</property>
<property name="icon">
<iconset>
<normaloff>ressources/add.png</normaloff>ressources/add.png</iconset>
</property>
<property name="text">
<string>Add</string>
</property>
<property name="toolTip">
<string>Add a new point in boundary condition or lateral source</string>
</property>
<property name="shortcut">
<string>Ctrl+N</string>
</property>
</action>
<action name="action_del">
<property name="icon">
<iconset>
<normaloff>ressources/del.png</normaloff>ressources/del.png</iconset>
</property>
<property name="text">
<string>Delete</string>
</property>
<property name="toolTip">
<string>Delete current selected rows</string>
</property>
<property name="shortcut">
<string>Ctrl+D</string>
</property>
</action>
<action name="action_sort">
<property name="icon">
<iconset>
<normaloff>ressources/sort_A-Z.png</normaloff>ressources/sort_A-Z.png</iconset>
</property>
<property name="text">
<string>Sort</string>
</property>
<property name="toolTip">
<string>Sort points</string>
</property>
</action>
<action name="action_import">
<property name="icon">
<iconset>
<normaloff>ressources/import.png</normaloff>ressources/import.png</iconset>
</property>
<property name="text">
<string>Import</string>
</property>
<property name="toolTip">
<string>Import from file</string>
</property>
</action>
</widget>
<resources/>
<connections/>
</ui>