src/HYDROGUI/HYDROGUI_UpdateObjectOp.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_UpdateObjectOp.h"
  20
  21 #include "HYDROGUI_Module.h"
  22 #include "HYDROGUI_Tool.h"
  23 #include "HYDROGUI_UpdateFlags.h"
  24
  25 #include <HYDROData_Document.h>
  26 #include <HYDROData_Image.h>
  27
  28 HYDROGUI_UpdateObjectOp::HYDROGUI_UpdateObjectOp( HYDROGUI_Module* theModule,
  29                                                   const bool       theIsForced )
  30 : HYDROGUI_Operation( theModule ),
  31   myIsForced( theIsForced )
  32 {
  33   setName( tr( "UPDATE_IMAGE" ) );
  34 }
  35
  36 HYDROGUI_UpdateObjectOp::~HYDROGUI_UpdateObjectOp()
  37 {
  38 }
  39
  40 void HYDROGUI_UpdateObjectOp::startOperation()
  41 {
  42   HYDROGUI_Operation::startOperation();
  43
  44   startDocOperation();
  45
  46   NCollection_Map<Handle(HYDROData_Entity)> aMapOfTreated;
  47
  48   HYDROData_SequenceOfObjects aSeq = HYDROGUI_Tool::GetSelectedObjects( module() );
  49   for( int anIndex = 1, aLength = aSeq.Length(); anIndex <= aLength; anIndex++ )
  50   {
  51     Handle(HYDROData_Entity) anObject = aSeq.Value( anIndex );
  52     updateObject( anObject, aMapOfTreated );
  53   }
  54
  55   commitDocOperation();
  56
  57   module()->update( UF_Model | UF_Viewer | UF_GV_Forced | UF_OCCViewer | UF_OCC_Forced | UF_VTKViewer );
  58   commit();
  59 }
  60
  61 void HYDROGUI_UpdateObjectOp::updateObject( const Handle(HYDROData_Entity)& theObject,
  62                                            NCollection_Map<Handle(HYDROData_Entity)>& theMapOfTreated ) const
  63 {
  64   if( theObject.IsNull() || theMapOfTreated.Contains( theObject ) )
  65     return;
  66
  67   theMapOfTreated.Add( theObject );
  68
  69   HYDROData_SequenceOfObjects aRefSeq = theObject->GetAllReferenceObjects();
  70   for ( int i = 1, n = aRefSeq.Length(); i <= n; ++i )
  71   {
  72     Handle(HYDROData_Entity) anObject = aRefSeq.Value( i );
  73     updateObject( anObject, theMapOfTreated );
  74   }
  75
  76   if ( !myIsForced && !theObject->IsMustBeUpdated() )
  77     return;
  78
  79   theObject->Update();
  80   module()->setIsToUpdate( theObject );
  81 }
  82