src/eliminatesmallbranches.h

   1 //
   2 // ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
   3 // +                                                                      +
   4 // + This file is part of enGrid.                                         +
   5 // +                                                                      +
   6 // + Copyright 2008,2009 Oliver Gloth                                     +
   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
  24 #ifndef ELIMINATESMALLBRANCHES_H
  25 #define ELIMINATESMALLBRANCHES_H
  26
  27 #include "operation.h"
  28
  29 class EliminateSmallBranches : public Operation
  30 {
  31
  32 private: // attributes
  33
  34   int              m_NumLayers;
  35   int              m_NumFillLayers;
  36   QVector<bool>    m_DeleteCell;
  37   QVector<bool>    m_IsSurfaceNode;
  38   QVector<bool>    m_MainVolumeCell;
  39   QList<vtkIdType> m_FillCells;
  40
  41
  42 protected: // methods
  43
  44   bool needsToBeMarked(vtkIdType id_node, int layer = 0);
  45   void unmarkNode(vtkIdType id_node, int layer = 0);
  46   void fill(vtkIdType id_cell);
  47   void fillFromLargestVolume();
  48   void fillLayers();
  49   void fillCraters();
  50   void fixNonManifold();
  51
  52   virtual void operate();
  53
  54
  55 public: // methods
  56
  57   EliminateSmallBranches();
  58
  59 };
  60
  61 #endif // ELIMINATESMALLBRANCHES_H