Merge remote-tracking branch 'origin/BR_1321_ECW' into BR_DEMO

[modules/hydro.git] / src / HYDROGUI / HYDROGUI_ZLevelsOp.cxx

diff --git a/src/HYDROGUI/HYDROGUI_ZLevelsOp.cxx b/src/HYDROGUI/HYDROGUI_ZLevelsOp.cxx
index 2cdc39cd8ee40bdb94a529d2a13f3265bdd94eac..916044265214780ad2010f27a7a6e3848c4b9170 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_ZLevelsOp.cxx
+++ b/src/HYDROGUI/HYDROGUI_ZLevelsOp.cxx
@@ -21,7 +21,7 @@
 #include "HYDROGUI_ZLevelsDlg.h"
 #include "HYDROGUI_DataModel.h"
 #include "HYDROGUI_Module.h"
-#include "HYDROGUI_Tool.h"
+#include "HYDROGUI_Tool2.h"
 #include "HYDROGUI_UpdateFlags.h"
 #include "HYDROGUI_ListSelector.h"
 
@@ -80,6 +80,7 @@ void HYDROGUI_ZLevelsOp::startOperation()
   // Show the dialog
   if ( !myDlg ) {
     myDlg = new HYDROGUI_ZLevelsDlg( module()->getApp()->desktop(), module() );
+       connect( myDlg, SIGNAL( applyOrderAndClose() ), this, SLOT( onApplyAndClose() ) );
     connect( myDlg, SIGNAL( applyOrder() ), this, SLOT( onApply() ) );
     connect( myDlg, SIGNAL( rejected() ), this, SLOT( onCancel() ) );
   }
@@ -131,4 +132,11 @@ void HYDROGUI_ZLevelsOp::processCancel()
     delete myDlg;
     myDlg = 0;
   }
-}
\ No newline at end of file
+}
+
+void HYDROGUI_ZLevelsOp::onApplyAndClose()
+{
+  HYDROGUI_Operation::onApplyAndClose();
+  if ( myDlg )
+    myDlg->reject();
+}