src/libengrid/createboundarylayershell.h

   1 // ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
   2 // +                                                                      +
   3 // + This file is part of enGrid.                                         +
   4 // +                                                                      +
   5 // + Copyright 2008-2014 enGits GmbH                                      +
   6 // +                                                                      +
   7 // + enGrid is free software: you can redistribute it and/or modify       +
   8 // + it under the terms of the GNU General Public License as published by +
   9 // + the Free Software Foundation, either version 3 of the License, or    +
  10 // + (at your option) any later version.                                  +
  11 // +                                                                      +
  12 // + enGrid is distributed in the hope that it will be useful,            +
  13 // + but WITHOUT ANY WARRANTY; without even the implied warranty of       +
  14 // + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the        +
  15 // + GNU General Public License for more details.                         +
  16 // +                                                                      +
  17 // + You should have received a copy of the GNU General Public License    +
  18 // + along with enGrid. If not, see <http://www.gnu.org/licenses/>.       +
  19 // +                                                                      +
  20 // ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
  21 #ifndef CREATEBOUNDARYLAYER_H
  22 #define CREATEBOUNDARYLAYER_H
  23
  24 #include "boundarylayeroperation.h"
  25 #include "guimainwindow.h"
  26
  27 class CreateBoundaryLayerShell : public BoundaryLayerOperation
  28 {
  29
  30 private: // attributes
  31
  32   QVector<vtkIdType> layer_cells;
  33
  34   int     m_NumIterations;
  35   bool    m_RemovePoints;
  36   double  m_RelativeHeight;
  37   double  m_AbsoluteHeight;
  38   double  m_Blending;
  39   QString m_VolumeName;
  40   bool    m_Success;
  41
  42   VolumeDefinition                     m_VolDef;
  43   vtkSmartPointer<vtkUnstructuredGrid> m_RestGrid;
  44   vtkSmartPointer<vtkUnstructuredGrid> m_OriginalGrid;
  45   vtkSmartPointer<vtkUnstructuredGrid> m_PrismaticGrid;
  46
  47
  48   /// Boundary codes of the surface we want to remove points on. Normally the one next to the prismatic boundary layer.
  49   QSet<int> m_LayerAdjacentBoundaryCodes;
  50
  51   QVector<vtkIdType> m_ShellNodeMap;
  52   MeshPartition      m_ShellPart;
  53
  54   QMap<int, vec3_t> m_LayerAdjacentOrigins;
  55   QMap<int, vec3_t> m_LayerAdjacentNormals;
  56
  57   QVector<vec3_t> m_OriginalNodeNormals;
  58
  59
  60 private: // data types
  61
  62   class DeleteBadNodes : public RemovePoints
  63   {
  64   protected:
  65
  66     QList<vtkIdType> m_BadNodes;
  67     virtual bool checkEdge(vtkIdType id_node1, vtkIdType id_node2);
  68
  69   public:
  70
  71     DeleteBadNodes(QList<vtkIdType> bad_nodes);
  72
  73   };
  74
  75   friend class DeleteBadNodes;
  76
  77
  78 private: // methods
  79
  80   QList<vtkIdType> findBadNodes(int bc);
  81   QList<vtkIdType> correctAdjacentBC(int bc, int num_levels = 10);
  82   void prepare();
  83   void createLayerNodes(vtkIdType id_node);
  84   void createPrismaticGrid();
  85
  86
  87 protected: // methods
  88
  89   virtual void operate();
  90
  91   void reduceSurface();
  92   void smoothSurface();
  93
  94
  95 public: // methods
  96
  97   CreateBoundaryLayerShell();
  98
  99   void setVolumeName(QString name) { m_VolumeName = name; }
 100   vtkUnstructuredGrid* getPrismaticGrid() { return m_PrismaticGrid; }
 101   QVector<int> getBoundaryLayerCodes() { return m_BoundaryLayerCodes; }
 102   bool success() { return m_Success; }
 103
 104 };
 105
 106 #endif // CREATEBOUNDARYLAYER_H