src/SMDS/SMDS_VtkCellIterator.hxx

   1 // Copyright (C) 2010-2020  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 #ifndef _SMDS_VTKCELLITERATOR_HXX_
  21 #define _SMDS_VTKCELLITERATOR_HXX_
  22
  23 #include "SMDS_ElemIterator.hxx"
  24 #include "SMDS_Mesh.hxx"
  25 #include "SMDSAbs_ElementType.hxx"
  26
  27 #include <vtkCell.h>
  28
  29 typedef std::vector< vtkIdType > TVtkIdList;
  30
  31 //--------------------------------------------------------------------------------
  32 /*!
  33  * \brief Retrieve nodes of a cell
  34  */
  35 struct _GetVtkNodes
  36 {
  37   _GetVtkNodes( TVtkIdList& nodeIds, SMDS_Mesh* mesh, int vtkCellId, SMDSAbs_EntityType type);
  38 };
  39 struct _GetVtkNodesToUNV
  40 {
  41   _GetVtkNodesToUNV( TVtkIdList& nodeIds, SMDS_Mesh* mesh, int vtkCellId, SMDSAbs_EntityType type);
  42 };
  43 struct _GetVtkNodesPolyh
  44 {
  45   _GetVtkNodesPolyh( TVtkIdList& nodeIds, SMDS_Mesh* mesh, int vtkCellId, SMDSAbs_EntityType type);
  46 };
  47
  48 //--------------------------------------------------------------------------------
  49 /*!
  50  * \brief Iterator on nodes of a cell
  51  */
  52 template< class SMDS_ITERATOR = SMDS_ElemIterator, class GET_VTK_NODES = _GetVtkNodes >
  53 class SMDS_VtkCellIterator: public SMDS_ITERATOR
  54 {
  55 public:
  56   typedef typename SMDS_ITERATOR::value_type result_type;
  57
  58   SMDS_VtkCellIterator(SMDS_Mesh* mesh, int vtkCellId, SMDSAbs_EntityType aType)
  59     : _mesh(mesh), _index(0)
  60   {
  61     GET_VTK_NODES getNodes( _vtkIdList, mesh, vtkCellId, aType );
  62   }
  63   virtual ~SMDS_VtkCellIterator() {}
  64   virtual bool        more()      {  return ( _index < _vtkIdList.size() ); }
  65   virtual result_type next()      {
  66     vtkIdType id = _vtkIdList[ _index++ ];
  67     return static_cast<result_type>( _mesh->FindNodeVtk( id ));
  68   }
  69 protected:
  70   SMDS_Mesh* _mesh;
  71   size_t     _index;
  72   TVtkIdList _vtkIdList;
  73 };
  74
  75 //--------------------------------------------------------------------------------
  76 template< class SMDS_ITERATOR = SMDS_ElemIterator >
  77 class SMDS_VtkCellIteratorToUNV: public SMDS_VtkCellIterator< SMDS_ITERATOR, _GetVtkNodesToUNV >
  78 {
  79   typedef SMDS_VtkCellIterator< SMDS_ITERATOR, _GetVtkNodesToUNV > parent_t;
  80 public:
  81   SMDS_VtkCellIteratorToUNV(SMDS_Mesh* mesh, int vtkCellId, SMDSAbs_EntityType type):
  82     parent_t( mesh, vtkCellId, type ) {}
  83 };
  84
  85 //--------------------------------------------------------------------------------
  86 template< class SMDS_ITERATOR = SMDS_ElemIterator >
  87 class SMDS_VtkCellIteratorPolyH: public SMDS_VtkCellIterator< SMDS_ITERATOR, _GetVtkNodesPolyh >
  88 {
  89   typedef SMDS_VtkCellIterator< SMDS_ITERATOR, _GetVtkNodesPolyh > parent_t;
  90 public:
  91   SMDS_VtkCellIteratorPolyH(SMDS_Mesh* mesh, int vtkCellId, SMDSAbs_EntityType type):
  92     parent_t( mesh, vtkCellId, type ) {}
  93 };
  94
  95 #endif