src/VTKViewer/VTKViewer_Algorithm.h

   1 //  Copyright (C) 2007-2008  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 //  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 //  This library is free software; you can redistribute it and/or
   7 //  modify it under the terms of the GNU Lesser General Public
   8 //  License as published by the Free Software Foundation; either
   9 //  version 2.1 of the License.
  10 //
  11 //  This library is distributed in the hope that it will be useful,
  12 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 //  Lesser General Public License for more details.
  15 //
  16 //  You should have received a copy of the GNU Lesser General Public
  17 //  License along with this library; if not, write to the Free Software
  18 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 //  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22 //  SALOME VTKViewer : build VTK viewer into Salome desktop
  23 //  File   : VTKViewer_ViewFrame.h
  24 //  Author : Nicolas REJNERI
  25 //  Module : SALOME
  26 //  $Header$
  27 //
  28 #ifndef VTKViewer_Algorithm_H
  29 #define VTKViewer_Algorithm_H
  30
  31 #include <vtkActorCollection.h>
  32
  33 class vtkActor;
  34
  35 namespace VTK
  36 {
  37   /*!For each actor(for ex: someActor) from \a theCollection(that can be dynamic cast to type TActor)\n
  38    * Call method \a theFun(someActor)
  39    */
  40     template<typename TActor, typename TFunction>
  41       TFunction ForEach(vtkActorCollection *theCollection, TFunction theFun)
  42       {
  43         if(theCollection){
  44           theCollection->InitTraversal();
  45           while(vtkActor *anAct = theCollection->GetNextActor())
  46             if(TActor *anActor = dynamic_cast<TActor*>(anAct))
  47               theFun(anActor);
  48         }
  49         return theFun;
  50       }
  51
  52     /*!For each actor(for ex: someActor) from \a theCollection(that can be dynamic cast to type TActor and \n
  53      * method \a thePredicate(someActor) return true) \n
  54      * Call method \a theFun(someActor)
  55      */
  56     template<typename TActor, typename TPredicate, typename TFunction>
  57       TFunction ForEachIf(vtkActorCollection *theCollection,
  58                           TPredicate thePredicate,
  59                           TFunction theFun)
  60       {
  61         if(theCollection){
  62           theCollection->InitTraversal();
  63           while(vtkActor *anAct = theCollection->GetNextActor())
  64             if(TActor *anActor = dynamic_cast<TActor*>(anAct))
  65               if(thePredicate(anActor))
  66                 theFun(anActor);
  67         }
  68         return theFun;
  69       }
  70
  71     /*!Find actor from collection, that can be dynamicaly cast to \a TActor, \n
  72      *and method \a thePredicate(someActor) return true) \n
  73      *\retval someActor
  74      */
  75     template<typename TActor, typename TPredicate>
  76       TActor* Find(vtkActorCollection *theCollection, TPredicate thePredicate)
  77       {
  78         if(theCollection){
  79           theCollection->InitTraversal();
  80           while(vtkActor *anAct = theCollection->GetNextActor())
  81             if(TActor *anActor = dynamic_cast<TActor*>(anAct))
  82               if(thePredicate(anActor))
  83                 return anActor;
  84         }
  85         return NULL;
  86       }
  87
  88 }
  89
  90 #endif