src/libengrid/iterator.h

   1 //
   2 // ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
   3 // +                                                                      +
   4 // + This file is part of enGrid.                                         +
   5 // +                                                                      +
   6 // + Copyright 2008-2013 enGits GmbH                                      +
   7 // +                                                                      +
   8 // + enGrid is free software: you can redistribute it and/or modify       +
   9 // + it under the terms of the GNU General Public License as published by +
  10 // + the Free Software Foundation, either version 3 of the License, or    +
  11 // + (at your option) any later version.                                  +
  12 // +                                                                      +
  13 // + enGrid is distributed in the hope that it will be useful,            +
  14 // + but WITHOUT ANY WARRANTY; without even the implied warranty of       +
  15 // + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the        +
  16 // + GNU General Public License for more details.                         +
  17 // +                                                                      +
  18 // + You should have received a copy of the GNU General Public License    +
  19 // + along with enGrid. If not, see <http://www.gnu.org/licenses/>.       +
  20 // +                                                                      +
  21 // ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
  22 //
  23 #ifndef iterator_H
  24 #define iterator_H
  25
  26 class Iterator;
  27
  28 #include "operation.h"
  29
  30 class Iterator : public Operation
  31 {
  32
  33 private: // attributes
  34
  35 protected: // attributes
  36
  37   struct pair_t
  38   {
  39     vtkIdType item1;
  40     vtkIdType item2;
  41     bool      terminate;
  42   };
  43
  44   QVector<pair_t> pair;
  45
  46   /** array used to mark item for the next iteration loop. */
  47   QVector<bool> mark1;
  48
  49   /** array used to marked "finished" items */
  50   QVector<bool> mark2;
  51
  52   QVector<vtkIdType>  item;
  53
  54   bool volume_iteration;
  55   bool custom_iteration;
  56
  57   QVector<vtkIdType>     cells;
  58   QVector<vtkIdType>     nodes;
  59   QVector<int>           _cells;
  60   QVector<int>           _nodes;
  61   QVector<QSet<int> >    n2c;
  62   QVector<QSet<int> >    n2n;
  63   QVector<QVector<int> > c2c;
  64
  65
  66 protected: // methods
  67
  68   virtual void pass1();
  69   virtual void pass2() {}
  70   void getCells();
  71
  72 public: // methods
  73
  74   Iterator();
  75
  76   void setVolumeIteration()  { volume_iteration = true; }
  77   void setSurfaceIteration() { volume_iteration = false; }
  78
  79 };
  80
  81 #endif
  82