aboutsummaryrefslogtreecommitdiffstats
path: root/awlsim/gui/fup/undo.py
blob: c291415903c1767539d475390bbaf6d4d9303e70 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
# -*- coding: utf-8 -*-
#
# AWL simulator - FUP - Undo stack
#
# Copyright 2018 Michael Buesch <m@bues.ch>
#
# 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 2 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, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#

from __future__ import division, absolute_import, print_function, unicode_literals
#from awlsim.common.cython_support cimport * #@cy
from awlsim.common.compat import *

from awlsim.gui.util import *
from awlsim.common.blocker import *

import datetime


__all__ = [
	"FupUndoStack",
]


class FupFullSourceUndoCommand(QUndoCommand):
	def __init__(self, stack, prevSource, newSource):
		QUndoCommand.__init__(self)
		self.__stack = stack
		self.__prevSource = prevSource
		self.__newSource = newSource
		self.blocked = Blocker()

		text = datetime.datetime.now().strftime(
			"[%Y-%m-%d %H:%M:%S.%f] FUP diagram change")
		self.setText(text)

	def id(self):
		return 0

	def mergeWith(self, other):
		if self.id() != other.id():
			return False
		if self.__prevSource == other.__prevSource and\
		   self.__newSource == other.__newSource:
			# The undo commands are equal. Just drop 'other'.
			return True
		return False

	def __apply(self, source, actionName):
		if self.blocked:
			return
		fupWidget = self.__stack.fupWidget
		try:
			with fupWidget.undoStackBlocked:
				fupWidget.setSource(source,
						    initUndoStack=False)
				self.__stack.setActiveSource(source)
		except AwlSimError as e:
			MessageBox.handleAwlSimError(fupWidget,
				"Failed to %s a FUP/FBD change" % actionName, e)

	def undo(self):
		"""Undo this change.
		"""
		self.__apply(self.__prevSource, "undo")

	def redo(self):
		"""Redo this change.
		"""
		self.__apply(self.__newSource, "redo")

class FupUndoStack(QUndoStack):
	def __init__(self, fupWidget):
		QUndoStack.__init__(self, fupWidget)
		self.fupWidget = fupWidget
		self.__activeSource = None
		self.appendBlocker = Blocker()

	def setActiveSource(self, source):
		"""Set the source that is currently active.
		"""
		# Duplicate the source
		self.__activeSource = source.dup()
		self.__activeSource.userData.clear()

	def initializeStack(self, source):
		"""Initialize the undo stack.
		"""
		self.setActiveSource(source)
		self.clear()

	def appendSourceChange(self, source):
		"""Append a change to the undo stack.
		"""
		# Duplicate the source
		source = source.dup()
		source.userData.clear()

		# Check if the source changed.
		prevSource = self.__activeSource
		if not prevSource:
			self.__activeSource = source
			return
		if source == prevSource:
			# The source did not change.
			return
		self.__activeSource = source

		# Append a full-source undo command
		cmd = FupFullSourceUndoCommand(self,
					       prevSource,
					       source)
		with cmd.blocked:
			self.push(cmd)
bues.ch cgit interface