src/HYDROGUI/HYDROGUI_DeleteOp.cxx

   1 // Copyright (C) 2014-2015  EDF-R&D
   2 // This library is free software; you can redistribute it and/or
   3 // modify it under the terms of the GNU Lesser General Public
   4 // License as published by the Free Software Foundation; either
   5 // version 2.1 of the License, or (at your option) any later version.
   6 //
   7 // This library is distributed in the hope that it will be useful,
   8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
   9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  10 // Lesser General Public License for more details.
  11 //
  12 // You should have received a copy of the GNU Lesser General Public
  13 // License along with this library; if not, write to the Free Software
  14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  15 //
  16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  17 //
  18
  19 #include "HYDROGUI_DeleteOp.h"
  20 #include "HYDROGUI_DataModel.h"
  21 #include "HYDROGUI_DeleteDlg.h"
  22 #include "HYDROGUI_Module.h"
  23 #include "HYDROGUI_Tool.h"
  24 #include "HYDROGUI_Tool2.h"
  25 #include "HYDROGUI_UpdateFlags.h"
  26
  27 #include <HYDROData_Object.h>
  28 #include <HYDROData_Document.h>
  29
  30 #include <LightApp_Application.h>
  31
  32 #include <SUIT_Desktop.h>
  33 #include <SUIT_MessageBox.h>
  34
  35 #include <QSet>
  36
  37 HYDROGUI_DeleteOp::HYDROGUI_DeleteOp( HYDROGUI_Module* theModule )
  38 : HYDROGUI_Operation( theModule )
  39 {
  40   setName( tr( "DELETE" ) );
  41 }
  42
  43 HYDROGUI_DeleteOp::~HYDROGUI_DeleteOp()
  44 {
  45 }
  46
  47 void HYDROGUI_DeleteOp::startOperation()
  48 {
  49   HYDROGUI_Operation::startOperation();
  50
  51   HYDROData_SequenceOfObjects aSeq = HYDROGUI_Tool::GetSelectedObjects( module() );
  52   if( aSeq.IsEmpty() )
  53   {
  54     abort();
  55     return;
  56   }
  57
  58   bool anIsCanRemove = true;
  59
  60   QStringList anObjNames;
  61   for( Standard_Integer anIndex = 1, aLength = aSeq.Length(); anIndex <= aLength; anIndex++ )
  62   {
  63     Handle(HYDROData_Entity) anObject = aSeq.Value( anIndex );
  64     if( anObject.IsNull() )
  65       continue;
  66
  67     if ( !anObject->CanRemove() )
  68     {
  69       anIsCanRemove = false;
  70       break;
  71     }
  72
  73     anObjNames.append( anObject->GetName() );
  74   }
  75
  76   if ( !anIsCanRemove )
  77   {
  78     SUIT_MessageBox::critical( module()->getApp()->desktop(),
  79                                tr( "DELETE_OBJECTS" ), tr( "DELETE_OBJECTS_IMPOSIBLE" ) );
  80     abort();
  81     return;
  82   }
  83
  84   anObjNames.removeDuplicates();
  85
  86   // check the back-references
  87   QMap<QString,HYDROData_SequenceOfObjects> aBackObjects =
  88     HYDROGUI_Tool::GetObjectsBackReferences( doc(), anObjNames );
  89
  90   QMap<QString,HYDROData_SequenceOfObjects>::const_iterator anIt = aBackObjects.begin(),
  91                                                             aLast = aBackObjects.end();
  92   QStringList aRefList;
  93   for ( ; anIt != aLast; anIt++ ) {
  94     QStringList aSeqList;
  95     HYDROData_SequenceOfObjects::Iterator anIter( anIt.value() );
  96     for ( ; anIter.More(); anIter.Next() )
  97     {
  98       Handle(HYDROData_Entity) anEntity = anIter.Value();
  99       if ( anEntity.IsNull() )
 100         continue;
 101       aSeqList.append( QString( tr( "DELETE_OBJECT_NAME" ) ).arg( anEntity->GetName() ) );
 102     }
 103     aRefList.append( QString( tr( "DELETE_OBJECT_IS_USED_FOR" ) ).arg( anIt.key() )
 104                                                           .arg( aSeqList.join( ", " ) ) );
 105   }
 106
 107   //QString aNamesList = anObjNames.join( "\n" );
 108   if ( !aBackObjects.isEmpty() )
 109   {
 110     SUIT_MessageBox::critical( module()->getApp()->desktop(),
 111                                tr( "DELETE_OBJECTS" ),
 112                                tr( "DELETED_OBJECTS_HAS_BACKREFERENCES" ).arg( aRefList.join( "\n" ) ) );
 113     abort();
 114     return;
 115   }
 116
 117   QString aWarningMsg;
 118
 119   HYDROGUI_DeleteDlg aDeleteDlg( module()->getApp()->desktop() );
 120   aDeleteDlg.setObjectsToRemove( anObjNames );
 121   bool isLastStricklerRemoved = false;
 122   if ( aDeleteDlg.exec() != HYDROGUI_DeleteDlg::Accepted )
 123   {
 124     abort();
 125     return;
 126   }
 127   else
 128   {
 129     QStringList aTableNames =
 130       HYDROGUI_Tool::FindExistingObjectsNames( doc(), KIND_STRICKLER_TABLE, false );
 131     aTableNames += anObjNames;
 132     if ( aTableNames.toSet().size() == anObjNames.size() )
 133     {
 134       aWarningMsg = tr("DELETE_LAST_TABLE_WRN");
 135       isLastStricklerRemoved = true;
 136     }
 137   }
 138
 139   if ( !aWarningMsg.isEmpty() ) {
 140     int anAnswer = SUIT_MessageBox::warning( module()->getApp()->desktop(),
 141                                              tr("WARNING"), aWarningMsg,
 142                                               QMessageBox::Yes | QMessageBox::No,
 143                                               QMessageBox::No );
 144     if ( anAnswer != QMessageBox::Yes ) {
 145       abort();
 146       return;
 147     }
 148   }
 149
 150   startDocOperation();
 151
 152   for( Standard_Integer anIndex = 1, aLength = aSeq.Length(); anIndex <= aLength; anIndex++ )
 153   {
 154     Handle(HYDROData_Entity) anObject = aSeq.Value( anIndex );
 155
 156     if ( !anObject.IsNull() && !anObject->IsRemoved() ) {
 157       anObject->Remove();
 158       module()->setObjectRemoved( anObject );
 159     }
 160   }
 161
 162   if( isLastStricklerRemoved )
 163   {
 164     HYDROGUI_DataModel* aModel = dynamic_cast<HYDROGUI_DataModel*>( module()->dataModel() );
 165     aModel->createDefaultStricklerTable( doc(), 0 );
 166   }
 167
 168   commitDocOperation();
 169
 170   module()->update( UF_Model | UF_Viewer | UF_OCCViewer | UF_VTKViewer );
 171   commit();
 172
 173   module()->enableLCMActions();
 174 }