1 files changed, 28 insertions, 0 deletions
diff --git a/qt-ui/undocommands.h b/qt-ui/undocommands.h
new file mode 100644
index 000000000..addef8138
--- /dev/null
+++ b/qt-ui/undocommands.h
@@ -0,0 +1,28 @@
+#ifndef UNDOCOMMANDS_H
+#define UNDOCOMMANDS_H
+
+#include <QUndoCommand>
+#include "dive.h"
+
+class UndoDeleteDive : public QUndoCommand {
+public:
+	UndoDeleteDive(QList<struct dive*> diveList);
+	virtual void undo();
+	virtual void redo();
+
+private:
+	QList<struct dive*> dives;
+};
+
+class UndoShiftTime : public QUndoCommand {
+public:
+	UndoShiftTime(QList<int> diveList, int amount);
+	virtual void undo();
+	virtual void redo();
+
+private:
+	QList<int> dives;
+	int timeChanged;
+};
+
+#endif // UNDOCOMMANDS_H