Salome HOME
22361: EDF SMESH: Quadrangle (mapping) algorithm: faces with more than 4 edges
[modules/smesh.git] / src / StdMeshers / StdMeshers_Quadrangle_2D.hxx
1 // Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License.
10 //
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 // Lesser General Public License for more details.
15 //
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
19 //
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22 //  File   : StdMeshers_Quadrangle_2D.hxx
23 //           Moved here from SMESH_Quadrangle_2D.hxx
24 //  Author : Paul RASCLE, EDF
25 //  Module : SMESH
26
27 #ifndef _SMESH_QUADRANGLE_2D_HXX_
28 #define _SMESH_QUADRANGLE_2D_HXX_
29
30 #include "SMESH_Algo.hxx"
31 #include "SMESH_ProxyMesh.hxx"
32 #include "SMESH_StdMeshers.hxx"
33 #include "StdMeshers_QuadrangleParams.hxx"
34
35 #include <TopoDS_Face.hxx>
36
37 class SMDS_MeshNode;
38 class SMESH_Mesh;
39 class SMESH_MesherHelper;
40 class SMESH_ProxyMesh;
41 class StdMeshers_FaceSide;
42 struct uvPtStruct;
43
44
45 enum TSideID { QUAD_BOTTOM_SIDE=0, QUAD_RIGHT_SIDE, QUAD_TOP_SIDE, QUAD_LEFT_SIDE, NB_QUAD_SIDES };
46
47 typedef uvPtStruct UVPtStruct;
48 typedef struct faceQuadStruct
49 {
50   std::vector< StdMeshers_FaceSide*> side;
51   bool isEdgeOut[4]; // true, if an EDGE has more nodes, than an opposite one
52   UVPtStruct* uv_grid;
53   TopoDS_Face face;
54   ~faceQuadStruct();
55   void shift( size_t nb, bool keepUnitOri );
56   typedef boost::shared_ptr<faceQuadStruct> Ptr;
57 } FaceQuadStruct;
58
59 class STDMESHERS_EXPORT StdMeshers_Quadrangle_2D: public SMESH_2D_Algo
60 {
61 public:
62   StdMeshers_Quadrangle_2D(int hypId, int studyId, SMESH_Gen* gen);
63   virtual ~StdMeshers_Quadrangle_2D();
64
65   virtual bool CheckHypothesis(SMESH_Mesh&         aMesh,
66                                const TopoDS_Shape& aShape,
67                                Hypothesis_Status&  aStatus);
68
69   virtual bool Compute(SMESH_Mesh&         aMesh,
70                        const TopoDS_Shape& aShape);
71
72   virtual bool Evaluate(SMESH_Mesh &         aMesh,
73                         const TopoDS_Shape & aShape,
74                         MapShapeNbElems&     aResMap);
75
76   FaceQuadStruct::Ptr CheckAnd2Dcompute(SMESH_Mesh&         aMesh,
77                                         const TopoDS_Shape& aShape,
78                                         const bool          CreateQuadratic);
79
80   FaceQuadStruct::Ptr CheckNbEdges(SMESH_Mesh&         aMesh,
81                                    const TopoDS_Shape& aShape);
82
83 protected:
84
85   bool CheckNbEdgesForEvaluate(SMESH_Mesh& aMesh,
86                                const TopoDS_Shape & aShape,
87                                MapShapeNbElems& aResMap,
88                                std::vector<int>& aNbNodes,
89                                bool& IsQuadratic);
90
91   bool SetNormalizedGrid(SMESH_Mesh&          aMesh,
92                          const TopoDS_Shape&  aShape,
93                          FaceQuadStruct::Ptr& quad);
94   
95   void SplitQuad(SMESHDS_Mesh *theMeshDS,
96                  const int theFaceID,
97                  const SMDS_MeshNode* theNode1,
98                  const SMDS_MeshNode* theNode2,
99                  const SMDS_MeshNode* theNode3,
100                  const SMDS_MeshNode* theNode4);
101
102   bool ComputeQuadPref(SMESH_Mesh&         aMesh,
103                        const TopoDS_Shape& aShape,
104                        FaceQuadStruct::Ptr quad);
105
106   bool EvaluateQuadPref(SMESH_Mesh&         aMesh,
107                         const TopoDS_Shape& aShape,
108                         std::vector<int>&   aNbNodes,
109                         MapShapeNbElems&    aResMap,
110                         bool                isQuadratic);
111
112   bool ComputeReduced (SMESH_Mesh&         aMesh,
113                        const TopoDS_Shape& aShape,
114                        FaceQuadStruct::Ptr quad);
115
116   void UpdateDegenUV(FaceQuadStruct::Ptr quad);
117
118   void Smooth (FaceQuadStruct::Ptr quad);
119
120   int GetCorners(const TopoDS_Face&          theFace,
121                  SMESH_Mesh &                theMesh,
122                  std::list<TopoDS_Edge>&     theWire,
123                  std::vector<TopoDS_Vertex>& theVertices,
124                  int &                       theNbDegenEdges);
125
126
127   // true if QuadranglePreference hypothesis is assigned that forces
128   // construction of quadrangles if the number of nodes on opposite edges
129   // is not the same in the case where the global number of nodes on edges
130   // is even
131   bool myQuadranglePreference;
132   bool myTrianglePreference;
133   int  myTriaVertexID;
134   bool myNeedSmooth;
135
136   StdMeshers_QuadType  myQuadType;
137   SMESH_MesherHelper*  myHelper; // tool for working with quadratic elements
138   SMESH_ProxyMesh::Ptr myProxyMesh;
139   FaceQuadStruct::Ptr  myQuadStruct;
140 };
141
142 #endif