Salome HOME
7d26b22664c44455178b6ee84e7b2bdc450cc685
[modules/smesh.git] / src / SMESHDS / SMESHDS_Mesh.hxx
1 //  SMESH SMESHDS : management of mesh data and SMESH document
2 //
3 //  Copyright (C) 2003  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.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org 
21 //
22 //
23 //
24 //  File   : SMESHDS_Mesh.hxx
25 //  Module : SMESH
26
27 #ifndef _SMESHDS_Mesh_HeaderFile
28 #define _SMESHDS_Mesh_HeaderFile
29
30 #include "SMDS_Mesh.hxx"
31 #include "SMDS_MeshNode.hxx"
32 #include "SMDS_MeshEdge.hxx"
33 #include "SMDS_MeshFace.hxx"
34 #include "SMDS_MeshVolume.hxx"
35 #include "SMESHDS_Hypothesis.hxx"
36 #include "SMESHDS_SubMesh.hxx"
37 #include "SMESHDS_Script.hxx"
38
39 #include <TopTools_IndexedMapOfShape.hxx>
40 #include <TopoDS_Shape.hxx>
41 #include <TopoDS_Shell.hxx>
42 #include <TopoDS_Face.hxx>
43 #include <TopoDS_Vertex.hxx>
44 #include <TopoDS_Edge.hxx>
45
46 //Not portable see http://gcc.gnu.org/onlinedocs/libstdc++/faq/index.html#5_4 to know more.
47 #ifdef __GNUC__
48 #if __GNUC__ < 3
49         #include <hash_map.h>
50 #else
51         #include <ext/hash_map>
52         namespace gstd = __gnu_cxx;
53 #endif
54 #endif
55
56 class SMESHDS_Mesh:public SMDS_Mesh{
57 public:
58   SMESHDS_Mesh(int MeshID);
59   void ShapeToMesh(const TopoDS_Shape & S);
60   bool AddHypothesis(const TopoDS_Shape & SS, const SMESHDS_Hypothesis * H);
61   bool RemoveHypothesis(const TopoDS_Shape & S, const SMESHDS_Hypothesis * H);
62   
63   virtual SMDS_MeshNode* AddNodeWithID(double x, double y, double z, int ID);
64   virtual SMDS_MeshNode * AddNode(double x, double y, double z);
65   
66   virtual SMDS_MeshEdge* AddEdgeWithID(int n1, int n2, int ID);
67   virtual SMDS_MeshEdge* AddEdgeWithID(const SMDS_MeshNode * n1,
68                                        const SMDS_MeshNode * n2, 
69                                        int ID);
70   virtual SMDS_MeshEdge* AddEdge(const SMDS_MeshNode * n1,
71                                  const SMDS_MeshNode * n2);
72   
73   virtual SMDS_MeshFace* AddFaceWithID(int n1, int n2, int n3, int ID);
74   virtual SMDS_MeshFace* AddFaceWithID(const SMDS_MeshNode * n1,
75                                        const SMDS_MeshNode * n2,
76                                        const SMDS_MeshNode * n3, 
77                                        int ID);
78   virtual SMDS_MeshFace* AddFace(const SMDS_MeshNode * n1,
79                                  const SMDS_MeshNode * n2,
80                                  const SMDS_MeshNode * n3);
81
82   virtual SMDS_MeshFace* AddFaceWithID(int n1, int n2, int n3, int n4, int ID);
83   virtual SMDS_MeshFace* AddFaceWithID(const SMDS_MeshNode * n1,
84                                        const SMDS_MeshNode * n2,
85                                        const SMDS_MeshNode * n3,
86                                        const SMDS_MeshNode * n4, 
87                                        int ID);
88   virtual SMDS_MeshFace* AddFace(const SMDS_MeshNode * n1,
89                                  const SMDS_MeshNode * n2,
90                                  const SMDS_MeshNode * n3,
91                                  const SMDS_MeshNode * n4);
92
93   virtual SMDS_MeshVolume* AddVolumeWithID(int n1, int n2, int n3, int n4, int ID);
94   virtual SMDS_MeshVolume* AddVolumeWithID(const SMDS_MeshNode * n1,
95                                            const SMDS_MeshNode * n2,
96                                            const SMDS_MeshNode * n3,
97                                            const SMDS_MeshNode * n4, 
98                                            int ID);
99   virtual SMDS_MeshVolume* AddVolume(const SMDS_MeshNode * n1,
100                                      const SMDS_MeshNode * n2,
101                                      const SMDS_MeshNode * n3,
102                                      const SMDS_MeshNode * n4);
103
104   virtual SMDS_MeshVolume* AddVolumeWithID(int n1, int n2, int n3, int n4, int n5, int ID);
105   virtual SMDS_MeshVolume* AddVolumeWithID(const SMDS_MeshNode * n1,
106                                            const SMDS_MeshNode * n2,
107                                            const SMDS_MeshNode * n3,
108                                            const SMDS_MeshNode * n4,
109                                            const SMDS_MeshNode * n5, 
110                                            int ID);
111   virtual SMDS_MeshVolume* AddVolume(const SMDS_MeshNode * n1,
112                                      const SMDS_MeshNode * n2,
113                                      const SMDS_MeshNode * n3,
114                                      const SMDS_MeshNode * n4,
115                                      const SMDS_MeshNode * n5);
116
117   virtual SMDS_MeshVolume* AddVolumeWithID(int n1, int n2, int n3, int n4, int n5, int n6, int ID);
118   virtual SMDS_MeshVolume* AddVolumeWithID(const SMDS_MeshNode * n1,
119                                            const SMDS_MeshNode * n2,
120                                            const SMDS_MeshNode * n3,
121                                            const SMDS_MeshNode * n4,
122                                            const SMDS_MeshNode * n5,
123                                            const SMDS_MeshNode * n6, 
124                                            int ID);
125   virtual SMDS_MeshVolume* AddVolume(const SMDS_MeshNode * n1,
126                                      const SMDS_MeshNode * n2,
127                                      const SMDS_MeshNode * n3,
128                                      const SMDS_MeshNode * n4,
129                                      const SMDS_MeshNode * n5,
130                                      const SMDS_MeshNode * n6);
131
132   virtual SMDS_MeshVolume* AddVolumeWithID(int n1, int n2, int n3, int n4, int n5, int n6, int n7, int n8, int ID);
133   virtual SMDS_MeshVolume* AddVolumeWithID(const SMDS_MeshNode * n1,
134                                            const SMDS_MeshNode * n2,
135                                            const SMDS_MeshNode * n3,
136                                            const SMDS_MeshNode * n4,
137                                            const SMDS_MeshNode * n5,
138                                            const SMDS_MeshNode * n6,
139                                            const SMDS_MeshNode * n7,
140                                            const SMDS_MeshNode * n8, 
141                                            int ID);
142   virtual SMDS_MeshVolume* AddVolume(const SMDS_MeshNode * n1,
143                                      const SMDS_MeshNode * n2,
144                                      const SMDS_MeshNode * n3,
145                                      const SMDS_MeshNode * n4,
146                                      const SMDS_MeshNode * n5,
147                                      const SMDS_MeshNode * n6,
148                                      const SMDS_MeshNode * n7,
149                                      const SMDS_MeshNode * n8);
150   
151   void MoveNode(const SMDS_MeshNode *, double x, double y, double z);
152   virtual void RemoveNode(const SMDS_MeshNode *);
153   void RemoveElement(const SMDS_MeshElement *);
154
155   void SetNodeInVolume(SMDS_MeshNode * aNode, const TopoDS_Shell & S);
156   void SetNodeOnFace(SMDS_MeshNode * aNode, const TopoDS_Face & S);
157   void SetNodeOnEdge(SMDS_MeshNode * aNode, const TopoDS_Edge & S);
158   void SetNodeOnVertex(SMDS_MeshNode * aNode, const TopoDS_Vertex & S);
159   void UnSetNodeOnShape(const SMDS_MeshNode * aNode);
160   void SetMeshElementOnShape(const SMDS_MeshElement * anElt,
161                              const TopoDS_Shape & S);
162   void UnSetMeshElementOnShape(const SMDS_MeshElement * anElt,
163                                const TopoDS_Shape & S);
164   TopoDS_Shape ShapeToMesh() const;
165   bool HasMeshElements(const TopoDS_Shape & S);
166   SMESHDS_SubMesh * MeshElements(const TopoDS_Shape & S);
167   SMESHDS_SubMesh * MeshElements(const int Index);
168   list<int> SubMeshIndices();
169   const std::map<int,SMESHDS_SubMesh*>& SubMeshes()
170   { return myShapeIndexToSubMesh; }
171
172   bool HasHypothesis(const TopoDS_Shape & S);
173   const list<const SMESHDS_Hypothesis*>& GetHypothesis(const TopoDS_Shape & S) const;
174   SMESHDS_Script * GetScript();
175   void ClearScript();
176   int ShapeToIndex(const TopoDS_Shape & aShape);
177   TopoDS_Shape IndexToShape(int ShapeIndex);
178
179   void NewSubMesh(int Index);
180   void SetNodeInVolume(const SMDS_MeshNode * aNode, int Index);
181   void SetNodeOnFace(SMDS_MeshNode * aNode, int Index);
182   void SetNodeOnEdge(SMDS_MeshNode * aNode, int Index);
183   void SetNodeOnVertex(SMDS_MeshNode * aNode, int Index);
184   void SetMeshElementOnShape(const SMDS_MeshElement * anElt, int Index);
185
186   ~SMESHDS_Mesh();
187   
188 private:
189   struct HashTopoDS_Shape{
190     size_t operator()(const TopoDS_Shape& S) const {
191       return S.HashCode(2147483647);
192     }
193   };
194   typedef std::list<const SMESHDS_Hypothesis*> THypList;
195   typedef gstd::hash_map<TopoDS_Shape,THypList,HashTopoDS_Shape> ShapeToHypothesis;
196   ShapeToHypothesis myShapeToHypothesis;
197
198   int myMeshID;
199   TopoDS_Shape myShape;
200   TopTools_IndexedMapOfShape myIndexToShape;
201   std::map<int,SMESHDS_SubMesh*> myShapeIndexToSubMesh;
202   SMESHDS_Script* myScript;
203 };
204
205
206 #endif