Adists: LC: Rename some 'edge' to 'reach' and fix plotXY error.

scenarios
Pierre-Antoine 2025-08-18 15:56:15 +02:00
parent 7cb3982a1f
commit 2667737490
7 changed files with 42 additions and 35 deletions

View File

@ -277,7 +277,7 @@ class LateralContributionAdisTS(SQLSubModel):
"lateral_contribution_adists(id, " + "lateral_contribution_adists(id, " +
"pollutant, reach, begin_rk, end_rk, scenario) " + "pollutant, reach, begin_rk, end_rk, scenario) " +
"VALUES (" + "VALUES (" +
f"{self.id}, {self._pollutant}, {self.edge}, " + f"{self.id}, {self._pollutant}, {self.reach}, " +
f"{self._begin_rk}, {self._end_rk}" + f"{self._begin_rk}, {self._end_rk}" +
f"{self._status.scenario_id}" + f"{self._status.scenario_id}" +
")" ")"
@ -316,13 +316,13 @@ class LateralContributionAdisTS(SQLSubModel):
return int(data) return int(data)
@property @property
def edge(self): def reach(self):
return self._edge return self._reach
@edge.setter @reach.setter
def edge(self, edge): def reach(self, reach):
self._edge = edge self._reach = reach
self._status.modified() self.modified()
@property @property
def header(self): def header(self):
@ -331,7 +331,7 @@ class LateralContributionAdisTS(SQLSubModel):
@header.setter @header.setter
def header(self, header): def header(self, header):
self._header = header self._header = header
self._status.modified() self.modified()
@property @property
def pollutant(self): def pollutant(self):
@ -348,7 +348,7 @@ class LateralContributionAdisTS(SQLSubModel):
@begin_rk.setter @begin_rk.setter
def begin_rk(self, begin_rk): def begin_rk(self, begin_rk):
self._begin_rk = begin_rk self._begin_rk = begin_rk
self._status.modified() self.modified()
@property @property
def end_rk(self): def end_rk(self):
@ -357,4 +357,4 @@ class LateralContributionAdisTS(SQLSubModel):
@end_rk.setter @end_rk.setter
def end_rk(self, end_rk): def end_rk(self, end_rk):
self._end_rk = end_rk self._end_rk = end_rk
self._status.modified() self.modified()

View File

@ -121,6 +121,9 @@ class Graph(SQLSubModel):
return edge[0] return edge[0]
def reach(self, reach_name: str):
return self.edge(reach_name)
def _create_node(self, x: float, y: float): def _create_node(self, x: float, y: float):
node = self._node_ctor( node = self._node_ctor(
id=-1, name="", x=x, y=y, id=-1, name="", x=x, y=y,

View File

@ -664,6 +664,9 @@ execution...
Last export at: @date.""" Last export at: @date."""
def reachs(self):
return self.edges()
@property @property
def boundary_condition(self): def boundary_condition(self):
return self._boundary_condition return self._boundary_condition

View File

@ -35,7 +35,7 @@ from PyQt5.QtWidgets import (
) )
from View.LateralContributionsAdisTS.UndoCommand import ( from View.LateralContributionsAdisTS.UndoCommand import (
SetEdgeCommand, SetReachCommand,
SetBeginCommand, SetEndCommand, SetBeginCommand, SetEndCommand,
AddCommand, DelCommand, AddCommand, DelCommand,
) )
@ -48,7 +48,7 @@ _translate = QCoreApplication.translate
class ComboBoxDelegate(QItemDelegate): class ComboBoxDelegate(QItemDelegate):
def __init__(self, data=None, mode="edge", def __init__(self, data=None, mode="reach",
trad=None, parent=None): trad=None, parent=None):
super(ComboBoxDelegate, self).__init__(parent) super(ComboBoxDelegate, self).__init__(parent)
@ -132,12 +132,12 @@ class TableModel(PamhyrTableModel):
row = index.row() row = index.row()
column = index.column() column = index.column()
if self._headers[column] == "edge": if self._headers[column] == "reach":
n = self._lst[row].edge n = self._lst[row].reach
if n is None: if n is None:
return self._trad['not_associated'] return self._trad['not_associated']
return next(filter( return next(filter(
lambda edge: edge.id == n, self._data.edges() lambda r: r.id == n, self._data.reachs()
)).name )).name
elif self._headers[column] == "begin_rk": elif self._headers[column] == "begin_rk":
return str(self._lst[row].begin_rk) return str(self._lst[row].begin_rk)
@ -154,12 +154,12 @@ class TableModel(PamhyrTableModel):
column = index.column() column = index.column()
try: try:
if self._headers[column] == "edge": if self._headers[column] == "reach":
self._undo.push( self._undo.push(
SetEdgeCommand( SetReachCommand(
self._lcs_list, self._lst, self._lcs_list, self._lst,
row, row,
self._data.edge(value).id self._data.reach(value).id
) )
) )
elif self._headers[column] == "begin_rk": elif self._headers[column] == "begin_rk":

View File

@ -63,21 +63,21 @@ class SetEndCommand(QUndoCommand):
self._lcs_lst[self._index].end_rk = float(self._new) self._lcs_lst[self._index].end_rk = float(self._new)
class SetEdgeCommand(QUndoCommand): class SetReachCommand(QUndoCommand):
def __init__(self, lcs, lcs_lst, index, edge): def __init__(self, lcs, lcs_lst, index, reach):
QUndoCommand.__init__(self) QUndoCommand.__init__(self)
self._lcs = lcs self._lcs = lcs
self._lcs_lst = lcs_lst self._lcs_lst = lcs_lst
self._index = index self._index = index
self._old = self._lcs_lst[self._index].edge self._old = self._lcs_lst[self._index].reach
self._new = edge self._new = reach
def undo(self): def undo(self):
self._lcs_lst[self._index].edge = self._old self._lcs_lst[self._index].reach = self._old
def redo(self): def redo(self):
self._lcs_lst[self._index].edge = self._new self._lcs_lst[self._index].reach = self._new
class AddCommand(QUndoCommand): class AddCommand(QUndoCommand):

View File

@ -90,9 +90,9 @@ class LateralContributionAdisTSWindow(PamhyrWindow):
) )
self._delegate_rk.append(delegate_rk) self._delegate_rk.append(delegate_rk)
self._delegate_edge = ComboBoxDelegate( self._delegate_reach = ComboBoxDelegate(
data=self._study.river, data=self._study.river,
mode="edge", mode="reach",
trad=self._trad, trad=self._trad,
parent=self parent=self
) )
@ -103,7 +103,7 @@ class LateralContributionAdisTSWindow(PamhyrWindow):
table_headers=self._trad.get_dict("table_headers"), table_headers=self._trad.get_dict("table_headers"),
editable_headers=self._trad.get_dict("table_headers"), editable_headers=self._trad.get_dict("table_headers"),
delegates={ delegates={
"edge": self._delegate_edge, "reach": self._delegate_reach,
"begin_rk": delegate_rk, "begin_rk": delegate_rk,
"end_rk": delegate_rk, "end_rk": delegate_rk,
}, },
@ -171,25 +171,26 @@ class LateralContributionAdisTSWindow(PamhyrWindow):
tab = "liquid" tab = "liquid"
if len(rows) > 0: if len(rows) > 0:
edge_id = self._study.river\ reach_id = self._study.river\
.lateral_contributions_adists.lst[rows[0]]\ .lateral_contributions_adists.lst[rows[0]]\
.edge .reach
if edge_id: if reach_id:
edge = next(filter(lambda e: e.id == edge_id, reach = next(filter(lambda e: e.id == reach_id,
self._study.river.edges())) self._study.river.reachs()))
data = edge.reach data = reach.reach
lc = self._lcs.lst[rows[0]] lc = self._lcs.lst[rows[0]]
highlight = (lc.begin_rk, lc.end_rk) highlight = (lc.begin_rk, lc.end_rk)
for delegate in self._delegate_rk: for delegate in self._delegate_rk:
delegate.data = edge delegate.data = reach
self.plot = PlotXY( self.plot = PlotXY(
canvas=self.canvas, canvas=self.canvas,
data=data, data=data,
trad=self._trad, trad=self._trad,
toolbar=None, toolbar=None,
parent=self,
) )
self.plot.highlight = highlight self.plot.highlight = highlight
self.plot.update() self.plot.update()

View File

@ -43,7 +43,7 @@ class LCTranslate(MainTranslate):
self._dict["z"] = _translate("Geometry", "Z (m)") self._dict["z"] = _translate("Geometry", "Z (m)")
self._sub_dict["table_headers"] = { self._sub_dict["table_headers"] = {
"edge": self._dict["reach"], "reach": self._dict["reach"],
"begin_rk": _translate("LateralContribution", "Begin rk (m)"), "begin_rk": _translate("LateralContribution", "Begin rk (m)"),
"end_rk": _translate("LateralContribution", "End rk (m)") "end_rk": _translate("LateralContribution", "End rk (m)")
} }