parouby
/
Pamhyr2
mirror of https://gitlab.com/pamhyr/pamhyr2
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

BC: Make node association unique (#25).

setup.py
Pierre-Antoine Rouby 2023-11-27 08:55:14 +01:00
parent 01fd7dbec8
commit c69cbdb34e
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/Model/BoundaryCondition/BoundaryConditionList.py
View File

@ -89,3 +89,11 @@ class BoundaryConditionList(PamhyrModelListWithTab):
def copy(self):
return copy(self)
def get_assoc_to_node(self, tab, node):
return next(
filter(
lambda i: i.node is node,
self._tabs[tab]
)
)

7
src/View/BoundaryCondition/UndoCommand.py
View File

@ -53,12 +53,19 @@ class SetNodeCommand(QUndoCommand):
self._index = index
self._old = self._bcs.get(self._tab, self._index).node
self._new = node
self._prev_assoc_to_node = self._bcs.get_assoc_to_node(tab, node)
def _previous_assoc_node(self, node):
if self._prev_assoc_to_node is not None:
self._prev_assoc_to_node.node = node
def undo(self):
self._bcs.get(self._tab, self._index).node = self._old
self._previous_assoc_node(self._new)
def redo(self):
self._bcs.get(self._tab, self._index).node = self._new
self._previous_assoc_node(None)
class SetTypeCommand(QUndoCommand):