1 files changed, 13 insertions, 0 deletions
diff --git a/qt-ui/undobuffer.h b/qt-ui/undobuffer.h
index a292d3ba7..beae8e390 100644
--- a/qt-ui/undobuffer.h
+++ b/qt-ui/undobuffer.h
@@ -4,6 +4,19 @@
 #include <QObject>
 #include "dive.h"
 
+class UndoCommand {
+private:
+	dive* stateBefore;
+	dive* stateAfter;
+	QString name;
+
+public:
+	explicit UndoCommand(QString commandName, dive* affectedDive);
+	void setStateAfter(dive* affectedDive);
+	void undo();
+	void redo();
+};
+
 class UndoBuffer : public QObject
 {
 	Q_OBJECT