From a00dd411c46310ca2d489bb282b1c026f4c8a3a5 Mon Sep 17 00:00:00 2001
From: Tomaz Canabrava <tomaz.canabrava@gmail.com>
Date: Thu, 4 Jun 2015 00:30:26 -0300
Subject: [PATCH 21/25] Fix reject changes for ADD/Edit

Reject changes was correct only for Edit, but
not for Add. When in add mode we need to retrieve
the old dive site, and when in edit mode we should
do nothing.

Signed-off-by: Tomaz Canabrava <tomaz.canabrava@gmail.com>
---
 qt-ui/locationinformation.cpp | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/qt-ui/locationinformation.cpp b/qt-ui/locationinformation.cpp
index 17eccf5..9f1262b 100644
--- a/qt-ui/locationinformation.cpp
+++ b/qt-ui/locationinformation.cpp
@@ -113,10 +113,15 @@ void LocationInformationWidget::createDiveSite()
 
 void LocationInformationWidget::rejectChanges()
 {
-	if (currentDs && dive_site_is_empty(currentDs)) {
+	if (current_mode == CREATE_DIVE_SITE) {
+		LocationInformationModel::instance()->removeRow(get_divesite_idx(currentDs));
+		if (displayed_dive.dive_site_uuid) {
+			displayed_dive_site = *get_dive_site_by_uuid(displayed_dive.dive_site_uuid);
+		}
+	} else if ((currentDs && dive_site_is_empty(currentDs))) {
 		LocationInformationModel::instance()->removeRow(get_divesite_idx(currentDs));
-		displayed_dive.dive_site_uuid = 0;
 	}
+
 	resetState();
 	emit stopFilterDiveSite();
 	emit informationManagementEnded();
-- 
2.3.2 (Apple Git-55)

