1 files changed, 10 insertions, 0 deletions
diff --git a/desktop-widgets/command_edit.h b/desktop-widgets/command_edit.h
index 5f5924e67..d91121c7d 100644
--- a/desktop-widgets/command_edit.h
+++ b/desktop-widgets/command_edit.h
@@ -56,6 +56,16 @@ public:
 	DiveField fieldId() const override;
 };
 
+class EditMode : public EditBase<int> {
+	int index;
+public:
+	EditMode(const QVector<dive *> &dives, int indexIn, int newValue, int oldValue);
+	void set(struct dive *d, int i) const override;
+	int data(struct dive *d) const override;
+	QString fieldName() const override;
+	DiveField fieldId() const override;
+};
+
 } // namespace Command
 
 #endif