mirror of https://gitlab.com/pamhyr/pamhyr2
138 lines
3.2 KiB
Python
138 lines
3.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from copy import deepcopy
|
|
from tools import trace, timer
|
|
|
|
from PyQt5.QtWidgets import (
|
|
QMessageBox, QUndoCommand, QUndoStack,
|
|
)
|
|
|
|
from Model.Network.Node import Node
|
|
from Model.Network.Edge import Edge
|
|
from Model.Network.Graph import Graph
|
|
|
|
class AddNodeCommand(QUndoCommand):
|
|
def __init__(self, graph, node):
|
|
QUndoCommand.__init__(self)
|
|
|
|
self._graph = graph
|
|
self._node = node
|
|
|
|
def undo(self):
|
|
self._graph.remove_node(self._node.name)
|
|
|
|
def redo(self):
|
|
self._graph.insert_node(self._node)
|
|
|
|
class DelNodeCommand(QUndoCommand):
|
|
def __init__(self, graph, node):
|
|
QUndoCommand.__init__(self)
|
|
|
|
self._graph = graph
|
|
self._node = node
|
|
self._edges = []
|
|
|
|
def undo(self):
|
|
for edge in self._edges:
|
|
self._graph.insert_edge(edge)
|
|
|
|
self._graph.insert_node(self._node)
|
|
|
|
def redo(self):
|
|
self._edges = list(
|
|
filter(
|
|
lambda e: (e.node1 == self._node or
|
|
e.node2 == self._node),
|
|
self._graph.edges()
|
|
)
|
|
)
|
|
|
|
for edge in self._edges:
|
|
self._graph.remove_edge(edge.name)
|
|
|
|
self._graph.remove_node(self._node.name)
|
|
|
|
class AddEdgeCommand(QUndoCommand):
|
|
def __init__(self, graph, edge):
|
|
QUndoCommand.__init__(self)
|
|
|
|
self._graph = graph
|
|
self._edge = edge
|
|
|
|
def undo(self):
|
|
self._graph.remove_edge(self._edge.name)
|
|
|
|
def redo(self):
|
|
self._graph.insert_edge(self._edge)
|
|
|
|
class DelEdgeCommand(QUndoCommand):
|
|
def __init__(self, graph, edge):
|
|
QUndoCommand.__init__(self)
|
|
|
|
self._graph = graph
|
|
self._edge = edge
|
|
|
|
def undo(self):
|
|
self._graph.insert_edge(self._edge)
|
|
|
|
def redo(self):
|
|
self._graph.remove_edge(self._edge.name)
|
|
|
|
|
|
class SetCommand(QUndoCommand):
|
|
def __init__(self, element, column, new_value):
|
|
QUndoCommand.__init__(self)
|
|
|
|
self._el = element
|
|
self._column = column
|
|
self._old = self._el[self._column]
|
|
self._new = new_value
|
|
|
|
def undo(self):
|
|
self._el[self._column] = self._old
|
|
|
|
def redo(self):
|
|
self._el[self._column] = self._new
|
|
|
|
|
|
class SetNodeCommand(QUndoCommand):
|
|
def __init__(self, graph, element, column, new_value):
|
|
QUndoCommand.__init__(self)
|
|
|
|
self._el = element
|
|
self._column = column
|
|
self._old = graph.node(self._el[self._column])
|
|
self._new = new_value
|
|
|
|
def undo(self):
|
|
self._el[self._column] = self._old
|
|
|
|
def redo(self):
|
|
self._el[self._column] = self._new
|
|
|
|
class EnableEdgeCommand(QUndoCommand):
|
|
def __init__(self, edge, enable):
|
|
QUndoCommand.__init__(self)
|
|
|
|
self._edge = edge
|
|
self._old = self._edge._enable
|
|
self._enable = enable
|
|
|
|
def undo(self):
|
|
self._edge.enable(enable = self._old)
|
|
|
|
def redo(self):
|
|
self._edge.enable(enable = self._enable)
|
|
|
|
class ReverseEdgeCommand(QUndoCommand):
|
|
def __init__(self, edge):
|
|
QUndoCommand.__init__(self)
|
|
|
|
self._edge = edge
|
|
|
|
def undo(self):
|
|
self._edge.reverse()
|
|
|
|
def redo(self):
|
|
self._edge.reverse()
|