src/libengrid/gridsmoother.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
  22 #ifndef gridsmoother_H
  23 #define gridsmoother_H
  24
  25 class GridSmoother;
  26
  27 #include "surfaceoperation.h"
  28
  29 #include <vtkCellLocator.h>
  30 #include <QSettings>
  31
  32 class GridSmoother : public SurfaceOperation
  33 {
  34
  35 private: // types
  36
  37   struct rule_t {
  38     QSet<int> bcs;
  39     double h;
  40   };
  41
  42 private: // attributes
  43
  44   QVector<bool>   m_NodeMarked;
  45   QVector<bool>   m_SurfNode;
  46   int             m_NumMarkedNodes;
  47   QList<rule_t>   m_Rules;
  48   bool            m_FirstCall;
  49   QVector<vec3_t> m_GeoNormal;
  50
  51 protected: // attributes
  52
  53   int m_NumIterations;
  54   int m_NumRelaxations;
  55   int m_NumBoundaryCorrections;
  56   int m_NumSearch;
  57   int m_NumNormalRelaxations;
  58   int m_NumHeightRelaxations;
  59   int m_NumLayers;
  60
  61   double m_Blending;
  62   double m_AbsoluteHeight;
  63   double m_RelativeHeight;
  64   double m_CritAngle;
  65   double m_RadarAngle;
  66   double m_MaxHeightInGaps;
  67   double m_DesiredStretching;
  68   double m_FarRatio;
  69   double m_FaceSizeLowerLimit;
  70   double m_FaceSizeUpperLimit;
  71   double m_FaceAngleLimit;
  72
  73   bool m_StrictPrismChecking;
  74   bool m_CollisionDetected;
  75
  76   QVector<vtkIdType> m_FootToField;
  77
  78   QVector<vtkIdType> m_IdFoot;
  79   QVector<double>    m_Height;
  80   QSet<int>          m_LayerAdjacentBoundaryCodes;
  81
  82 protected: // methods
  83
  84   virtual void operate();
  85
  86   bool setNewPosition(vtkIdType id_node, vec3_t x_new);
  87   void correctDx(int i_nodes, vec3_t &Dx);
  88   bool moveNode(int i_nodes, vec3_t &Dx);
  89   void markNodes();
  90   void computeNormals();
  91   void relaxNormalVectors();
  92   void correctNormalVectors();
  93   void computeDesiredHeights();
  94   void computeHeights();
  95   void computeFeet();
  96   void simpleNodeMovement(int i_nodes);
  97   void getRules();
  98   bool faceFine(vtkIdType id_face, double scale);
  99
 100   void writeDebugFile(QString file_name);
 101
 102 public: // methods
 103
 104   GridSmoother();
 105   void setNumIterations         (int N)    { m_NumIterations  = N; }
 106   void setNumRelaxations        (int N)    { m_NumRelaxations = N; }
 107   void setNumHeightRelaxations  (int N)    { m_NumHeightRelaxations = N; }
 108   void setNumNormalRelaxations  (int N)    { m_NumNormalRelaxations = N; }
 109   void setNumBoundaryCorrections(int N)    { m_NumBoundaryCorrections = N; }
 110   void setRelativeHeight        (double h) { m_RelativeHeight = h; }
 111   void setAbsoluteHeight        (double h) { m_AbsoluteHeight = h; }
 112   void setBlending              (double b) { m_Blending = b; }
 113   void setDesiredStretching     (double s) { m_DesiredStretching = s; }
 114   void setFarRatio              (double r) { m_FarRatio = r; }
 115   void forceNormalCalculation   ()         { m_FirstCall = true; }
 116
 117   void setLayerAdjacentBoundaryCodes(const QSet<int> &abcs) { m_LayerAdjacentBoundaryCodes = abcs; }
 118
 119 };
 120
 121
 122 typedef GridSmoother SmoothVolumeGrid;
 123
 124 #endif