Salome HOME
refs #1340 part.2
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_DeleteOp.cxx
index 8df9ef26cf8cccff6f0b0d8acc4ddfe2ac75f260..7164ea8ad45a925b9ca6d48bdd2921d1edce5870 100644 (file)
@@ -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
 //
 
 #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 "HYDROGUI_ProfileOp.h"
 
 #include <HYDROData_Object.h>
+#include <HYDROData_Document.h>
 
 #include <LightApp_Application.h>
 
 #include <SUIT_Desktop.h>
 #include <SUIT_MessageBox.h>
+#include <SUIT_Study.h>
+
+#include <QSet>
 
 HYDROGUI_DeleteOp::HYDROGUI_DeleteOp( HYDROGUI_Module* theModule )
 : HYDROGUI_Operation( theModule )
@@ -45,60 +48,143 @@ HYDROGUI_DeleteOp::~HYDROGUI_DeleteOp()
 
 void HYDROGUI_DeleteOp::startOperation()
 {
-  HYDROGUI_Operation::startOperation();
+  QList<SUIT_Operation*> operations = study()->operations();
+  int n = operations.size();
+  if( n>=2 )
+  {
+    SUIT_Operation* anOtherOp = operations[n-2];
+    HYDROGUI_ProfileOp* aProfileOp = dynamic_cast<HYDROGUI_ProfileOp*>( anOtherOp );
+    if( aProfileOp )
+    {
+      aProfileOp->deleteSelected();
+      abort();
+      return;
+    }
+  }
 
-  HYDROData_SequenceOfObjects aFullSeq; // selected objects with their back-references
-  QStringList aFullNames;
+  HYDROGUI_Operation::startOperation();
 
   HYDROData_SequenceOfObjects aSeq = HYDROGUI_Tool::GetSelectedObjects( module() );
-  if( !aSeq.IsEmpty() )
+  if( aSeq.IsEmpty() )
   {
-    for( Standard_Integer anIndex = 1, aLength = aSeq.Length(); anIndex <= aLength; anIndex++ )
+    abort();
+    return;
+  }
+
+  bool anIsCanRemove = true;
+
+  QStringList anObjNames;
+  for( Standard_Integer anIndex = 1, aLength = aSeq.Length(); anIndex <= aLength; anIndex++ )
+  {
+    Handle(HYDROData_Entity) anObject = aSeq.Value( anIndex );
+    if( anObject.IsNull() )
+      continue;
+
+    if ( !anObject->CanRemove() )
     {
-      Handle(HYDROData_Object) anObject = aSeq.Value( anIndex );
-      if( !anObject.IsNull() )
-      {
-        aFullSeq.Append( anObject );
-        aFullNames.append( anObject->GetName() );
-
-        // collect the back-references
-        ObjectKind aKind = anObject->GetKind();
-        if( aKind == KIND_IMAGE || aKind == KIND_POLYLINE )
-        {
-          HYDROData_SequenceOfObjects anObjects;
-          QStringList aNames;
-          HYDROGUI_Tool::GetObjectBackReferences( module(), anObject, anObjects, aNames );
-          aFullSeq.Append( anObjects );
-          aFullNames.append( aNames );
-        }
-      }
+      anIsCanRemove = false;
+      break;
     }
 
-    aFullNames.removeDuplicates();
-    aFullNames.sort();
+    anObjNames.append( anObject->GetName() );
+  }
+
+  if ( !anIsCanRemove )
+  {
+    SUIT_MessageBox::critical( module()->getApp()->desktop(),
+                               tr( "DELETE_OBJECTS" ), tr( "DELETE_OBJECTS_IMPOSIBLE" ) );
+    abort();
+    return;
+  }
+
+  anObjNames.removeDuplicates();
+
+  // check the back-references
+  QMap<QString,HYDROData_SequenceOfObjects> aBackObjects =
+    HYDROGUI_Tool::GetObjectsBackReferences( doc(), anObjNames );
 
-    QString aList = aFullNames.join( "\n" );
-    int anAnswer = SUIT_MessageBox::question( module()->getApp()->desktop(),
-                                              tr( "DELETE_OBJECTS" ),
-                                              tr( "CONFIRM_DELETION" ).arg( aList ),
+  QMap<QString,HYDROData_SequenceOfObjects>::const_iterator anIt = aBackObjects.begin(),
+                                                            aLast = aBackObjects.end();
+  QStringList aRefList;
+  for ( ; anIt != aLast; anIt++ ) {
+    QStringList aSeqList;
+    HYDROData_SequenceOfObjects::Iterator anIter( anIt.value() );
+    for ( ; anIter.More(); anIter.Next() )
+    {
+      Handle(HYDROData_Entity) anEntity = anIter.Value();
+      if ( anEntity.IsNull() )
+        continue;
+      aSeqList.append( QString( tr( "DELETE_OBJECT_NAME" ) ).arg( anEntity->GetName() ) );
+    }
+    aRefList.append( QString( tr( "DELETE_OBJECT_IS_USED_FOR" ) ).arg( anIt.key() )
+                                                          .arg( aSeqList.join( ", " ) ) );
+  }
+
+  //QString aNamesList = anObjNames.join( "\n" );
+  if ( !aBackObjects.isEmpty() )
+  {
+    SUIT_MessageBox::critical( module()->getApp()->desktop(),
+                               tr( "DELETE_OBJECTS" ),
+                               tr( "DELETED_OBJECTS_HAS_BACKREFERENCES" ).arg( aRefList.join( "\n" ) ) );
+    abort();
+    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::No )
-    {
+    if ( anAnswer != QMessageBox::Yes ) {
       abort();
       return;
     }
   }
 
   startDocOperation();
-  for( Standard_Integer anIndex = 1, aLength = aFullSeq.Length(); anIndex <= aLength; anIndex++ )
+
+  for( Standard_Integer anIndex = 1, aLength = aSeq.Length(); anIndex <= aLength; anIndex++ )
   {
-    Handle(HYDROData_Object) anObject = aFullSeq.Value( anIndex );
-    if( !anObject.IsNull() && !anObject->IsRemoved() )
+    Handle(HYDROData_Entity) anObject = aSeq.Value( anIndex );
+
+    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 );
+  module()->update( UF_Model | UF_Viewer | UF_OCCViewer | UF_VTKViewer );
   commit();
+
+  module()->enableLCMActions();
 }