LCM // Import/Export of SHP p.3

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

diff --git a/src/HYDROGUI/HYDROGUI_DeleteOp.cxx b/src/HYDROGUI/HYDROGUI_DeleteOp.cxx
index 83dd008aca4a7b9efbeca9d4c1180072be28c747..78ab4dec9b51b93c0dbd251e066ccc37edaddfce 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_DeleteOp.cxx
+++ b/src/HYDROGUI/HYDROGUI_DeleteOp.cxx
@@ -17,7 +17,7 @@
 //
 
 #include "HYDROGUI_DeleteOp.h"
-
+#include "HYDROGUI_DataModel.h"
 #include "HYDROGUI_DeleteDlg.h"
 #include "HYDROGUI_Module.h"
 #include "HYDROGUI_Tool.h"
@@ -117,16 +117,21 @@ void HYDROGUI_DeleteOp::startOperation()
 
   HYDROGUI_DeleteDlg aDeleteDlg( module()->getApp()->desktop() );
   aDeleteDlg.setObjectsToRemove( anObjNames );
+  bool isLastStricklerRemoved = false;
   if ( aDeleteDlg.exec() != HYDROGUI_DeleteDlg::Accepted )
   {
     abort();
     return;
-  } else {
+  }
+  else
+  {
     QStringList aTableNames = 
       HYDROGUI_Tool::FindExistingObjectsNames( doc(), KIND_STRICKLER_TABLE, false );
     aTableNames += anObjNames;
-    if ( aTableNames.toSet().size() == anObjNames.size() ) {
+    if ( aTableNames.toSet().size() == anObjNames.size() )
+    {
       aWarningMsg = tr("DELETE_LAST_TABLE_WRN");
+      isLastStricklerRemoved = true;
     }
   }
 
@@ -153,6 +158,12 @@ void HYDROGUI_DeleteOp::startOperation()
     }
   }
 
+  if( isLastStricklerRemoved )
+  {
+    HYDROGUI_DataModel* aModel = dynamic_cast<HYDROGUI_DataModel*>( module()->dataModel() );
+    aModel->createDefaultStricklerTable( doc(), 0 );
+  }
+
   commitDocOperation();
 
   module()->update( UF_Model | UF_Viewer | UF_OCCViewer | UF_VTKViewer );