Undo: select dives on undo of editing commands

Save selected dives when generating an edit-command. Restore the
selection and current dive in undo()/redo().

Signed-off-by: Berthold Stoeger <bstoeger@mail.tuwien.ac.at>

1 files changed, 5 insertions, 1 deletions

diff --git a/desktop-widgets/command_edit.h b/desktop-widgets/command_edit.h
index 4363cb9bd..9b5923af2 100644
--- a/desktop-widgets/command_edit.h
+++ b/desktop-widgets/command_edit.h
@@ -34,7 +34,9 @@ protected:
 	bool workToBeDone() override;
 
 	std::vector<dive *> dives; // Dives to be edited.
-	struct dive *current; // On undo, we set the current dive at the time of the operation.
+	// On undo, we set the selection and current dive at the time of the operation.
+	std::vector<dive *> selectedDives;
+	struct dive *current;
 public:
 	EditBase(T newValue, bool currentDiveOnly);
 
@@ -162,6 +164,8 @@ class EditTagsBase : public Base {
 	// the active dive when the user initialized the action. This dive
 	// will be made the current dive on redo / undo.
 	std::vector<dive *> dives;
+	// On undo, we set the selection and current dive at the time of the operation.
+	std::vector<dive *> selectedDives;
 	struct dive *current;
 	QStringList newList;	// Temporary until initialized
 public: