debug of DTM object

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

diff --git a/src/HYDROGUI/HYDROGUI_DeleteOp.cxx b/src/HYDROGUI/HYDROGUI_DeleteOp.cxx
index 51c2243f23f36d683279563fa44d6d2dd6d8d455..8425cb1d8e1181f7b6c4e1d0a97a51e5545414c5 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_DeleteOp.cxx
+++ b/src/HYDROGUI/HYDROGUI_DeleteOp.cxx
@@ -1,12 +1,8 @@
-// Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
-//
-// Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
-// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
-//
+// Copyright (C) 2014-2015  EDF-R&D
 // This library is free software; you can redistribute it and/or
 // modify it under the terms of the GNU Lesser General Public
 // License as published by the Free Software Foundation; either
-// version 2.1 of the License.
+// version 2.1 of the License, or (at your option) any later version.
 //
 // This library is distributed in the hope that it will be useful,
 // but WITHOUT ANY WARRANTY; without even the implied warranty of
@@ -21,19 +17,23 @@
 //
 
 #include "HYDROGUI_DeleteOp.h"
-
+#include "HYDROGUI_DataModel.h"
 #include "HYDROGUI_DeleteDlg.h"
 #include "HYDROGUI_Module.h"
 #include "HYDROGUI_Tool.h"
+#include "HYDROGUI_Tool2.h"
 #include "HYDROGUI_UpdateFlags.h"
 
 #include <HYDROData_Object.h>
+#include <HYDROData_Document.h>
 
 #include <LightApp_Application.h>
 
 #include <SUIT_Desktop.h>
 #include <SUIT_MessageBox.h>
 
+#include <QSet>
+
 HYDROGUI_DeleteOp::HYDROGUI_DeleteOp( HYDROGUI_Module* theModule )
 : HYDROGUI_Operation( theModule )
 {
@@ -85,7 +85,7 @@ void HYDROGUI_DeleteOp::startOperation()
 
   // check the back-references
   QMap<QString,HYDROData_SequenceOfObjects> aBackObjects =
-    HYDROGUI_Tool::GetObjectsBackReferences( module(), anObjNames );
+    HYDROGUI_Tool::GetObjectsBackReferences( doc(), anObjNames );
 
   QMap<QString,HYDROData_SequenceOfObjects>::const_iterator anIt = aBackObjects.begin(),
                                                             aLast = aBackObjects.end();
@@ -114,25 +114,61 @@ void HYDROGUI_DeleteOp::startOperation()
     return;
   }
 
+  QString aWarningMsg;
+
   HYDROGUI_DeleteDlg aDeleteDlg( module()->getApp()->desktop() );
   aDeleteDlg.setObjectsToRemove( anObjNames );
+  bool isLastStricklerRemoved = false;
   if ( aDeleteDlg.exec() != HYDROGUI_DeleteDlg::Accepted )
   {
     abort();
     return;
   }
+  else
+  {
+    QStringList aTableNames = 
+      HYDROGUI_Tool::FindExistingObjectsNames( doc(), KIND_STRICKLER_TABLE, false );
+    aTableNames += anObjNames;
+    if ( aTableNames.toSet().size() == anObjNames.size() )
+    {
+      aWarningMsg = tr("DELETE_LAST_TABLE_WRN");
+      isLastStricklerRemoved = true;
+    }
+  }
+
+  if ( !aWarningMsg.isEmpty() ) {
+    int anAnswer = SUIT_MessageBox::warning( module()->getApp()->desktop(),
+                                             tr("WARNING"), aWarningMsg,
+                                              QMessageBox::Yes | QMessageBox::No,
+                                              QMessageBox::No );
+    if ( anAnswer != QMessageBox::Yes ) {
+      abort();
+      return;
+    }
+  }
 
   startDocOperation();
 
   for( Standard_Integer anIndex = 1, aLength = aSeq.Length(); anIndex <= aLength; anIndex++ )
   {
     Handle(HYDROData_Entity) anObject = aSeq.Value( anIndex );
-    if ( !anObject.IsNull() && !anObject->IsRemoved() )
+
+    if ( !anObject.IsNull() && !anObject->IsRemoved() ) {
       anObject->Remove();
+      module()->setObjectRemoved( anObject );
+    }
+  }
+
+  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 );
   commit();
+
+  module()->enableLCMActions();
 }