Add ability to undo deleted dives

Before the dive is deleted, a copy is made and passed to the
undo buffer. When edit->undo is clicked, this dive is restored
to the dive list.

Signed-off-by: Grace Karanja <gracie.karanja89@gmail.com>
Signed-off-by: Dirk Hohndel <dirk@hohndel.org>

1 files changed, 3 insertions, 1 deletions

diff --git a/qt-ui/undobuffer.h b/qt-ui/undobuffer.h
index beae8e390..9fac14710 100644
--- a/qt-ui/undobuffer.h
+++ b/qt-ui/undobuffer.h
@@ -12,7 +12,7 @@ private:
 
 public:
 	explicit UndoCommand(QString commandName, dive* affectedDive);
-	void setStateAfter(dive* affectedDive);
+	void setStateAfter(dive* affectedDive) { stateAfter = affectedDive; }
 	void undo();
 	void redo();
 };
@@ -25,7 +25,9 @@ public:
 	~UndoBuffer();
 	bool canUndo();
 	bool canRedo();
+	UndoCommand *current() const { return list.at(curIdx - 1); }
 private:
+	QList<UndoCommand*> list;
 	int curIdx;
 public slots:
 	void redo();