Salome HOME
DCQ:prepare 2.0.0
[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         // _CS_gbo_100504 Do not forget to define the namespace alias gstd
51         namespace gstd { using ::hash_map; } // inherits globals
52 #else
53         #include <ext/hash_map>
54         namespace gstd = __gnu_cxx;
55 #endif
56 #endif
57
58 class SMESHDS_Mesh:public SMDS_Mesh{
59 public:
60   SMESHDS_Mesh(int MeshID);
61   void ShapeToMesh(const TopoDS_Shape & S);
62   bool AddHypothesis(const TopoDS_Shape & SS, const SMESHDS_Hypothesis * H);
63   bool RemoveHypothesis(const TopoDS_Shape & S, const SMESHDS_Hypothesis * H);
64   
65   virtual SMDS_MeshNode* AddNodeWithID(double x, double y, double z, int ID);
66   virtual SMDS_MeshNode * AddNode(double x, double y, double z);
67   
68   virtual SMDS_MeshEdge* AddEdgeWithID(int n1, int n2, int ID);
69   virtual SMDS_MeshEdge* AddEdgeWithID(const SMDS_MeshNode * n1,
70                                        const SMDS_MeshNode * n2, 
71                                        int ID);
72   virtual SMDS_MeshEdge* AddEdge(const SMDS_MeshNode * n1,
73                                  const SMDS_MeshNode * n2);
74   
75   virtual SMDS_MeshFace* AddFaceWithID(int n1, int n2, int n3, int ID);
76   virtual SMDS_MeshFace* AddFaceWithID(const SMDS_MeshNode * n1,
77                                        const SMDS_MeshNode * n2,
78                                        const SMDS_MeshNode * n3, 
79                                        int ID);
80   virtual SMDS_MeshFace* AddFace(const SMDS_MeshNode * n1,
81                                  const SMDS_MeshNode * n2,
82                                  const SMDS_MeshNode * n3);
83
84   virtual SMDS_MeshFace* AddFaceWithID(int n1, int n2, int n3, int n4, int ID);
85   virtual SMDS_MeshFace* AddFaceWithID(const SMDS_MeshNode * n1,
86                                        const SMDS_MeshNode * n2,
87                                        const SMDS_MeshNode * n3,
88                                        const SMDS_MeshNode * n4, 
89                                        int ID);
90   virtual SMDS_MeshFace* AddFace(const SMDS_MeshNode * n1,
91                                  const SMDS_MeshNode * n2,
92                                  const SMDS_MeshNode * n3,
93                                  const SMDS_MeshNode * n4);
94
95   virtual SMDS_MeshVolume* AddVolumeWithID(int n1, int n2, int n3, int n4, int ID);
96   virtual SMDS_MeshVolume* AddVolumeWithID(const SMDS_MeshNode * n1,
97                                            const SMDS_MeshNode * n2,
98                                            const SMDS_MeshNode * n3,
99                                            const SMDS_MeshNode * n4, 
100                                            int ID);
101   virtual SMDS_MeshVolume* AddVolume(const SMDS_MeshNode * n1,
102                                      const SMDS_MeshNode * n2,
103                                      const SMDS_MeshNode * n3,
104                                      const SMDS_MeshNode * n4);
105
106   virtual SMDS_MeshVolume* AddVolumeWithID(int n1, int n2, int n3, int n4, int n5, int ID);
107   virtual SMDS_MeshVolume* AddVolumeWithID(const SMDS_MeshNode * n1,
108                                            const SMDS_MeshNode * n2,
109                                            const SMDS_MeshNode * n3,
110                                            const SMDS_MeshNode * n4,
111                                            const SMDS_MeshNode * n5, 
112                                            int ID);
113   virtual SMDS_MeshVolume* AddVolume(const SMDS_MeshNode * n1,
114                                      const SMDS_MeshNode * n2,
115                                      const SMDS_MeshNode * n3,
116                                      const SMDS_MeshNode * n4,
117                                      const SMDS_MeshNode * n5);
118
119   virtual SMDS_MeshVolume* AddVolumeWithID(int n1, int n2, int n3, int n4, int n5, int n6, int ID);
120   virtual SMDS_MeshVolume* AddVolumeWithID(const SMDS_MeshNode * n1,
121                                            const SMDS_MeshNode * n2,
122                                            const SMDS_MeshNode * n3,
123                                            const SMDS_MeshNode * n4,
124                                            const SMDS_MeshNode * n5,
125                                            const SMDS_MeshNode * n6, 
126                                            int ID);
127   virtual SMDS_MeshVolume* AddVolume(const SMDS_MeshNode * n1,
128                                      const SMDS_MeshNode * n2,
129                                      const SMDS_MeshNode * n3,
130                                      const SMDS_MeshNode * n4,
131                                      const SMDS_MeshNode * n5,
132                                      const SMDS_MeshNode * n6);
133
134   virtual SMDS_MeshVolume* AddVolumeWithID(int n1, int n2, int n3, int n4, int n5, int n6, int n7, int n8, int ID);
135   virtual SMDS_MeshVolume* AddVolumeWithID(const SMDS_MeshNode * n1,
136                                            const SMDS_MeshNode * n2,
137                                            const SMDS_MeshNode * n3,
138                                            const SMDS_MeshNode * n4,
139                                            const SMDS_MeshNode * n5,
140                                            const SMDS_MeshNode * n6,
141                                            const SMDS_MeshNode * n7,
142                                            const SMDS_MeshNode * n8, 
143                                            int ID);
144   virtual SMDS_MeshVolume* AddVolume(const SMDS_MeshNode * n1,
145                                      const SMDS_MeshNode * n2,
146                                      const SMDS_MeshNode * n3,
147                                      const SMDS_MeshNode * n4,
148                                      const SMDS_MeshNode * n5,
149                                      const SMDS_MeshNode * n6,
150                                      const SMDS_MeshNode * n7,
151                                      const SMDS_MeshNode * n8);
152   
153   void MoveNode(const SMDS_MeshNode *, double x, double y, double z);
154   virtual void RemoveNode(const SMDS_MeshNode *);
155   void RemoveElement(const SMDS_MeshElement *);
156
157   void SetNodeInVolume(SMDS_MeshNode * aNode, const TopoDS_Shell & S);
158   void SetNodeOnFace(SMDS_MeshNode * aNode, const TopoDS_Face & S);
159   void SetNodeOnEdge(SMDS_MeshNode * aNode, const TopoDS_Edge & S);
160   void SetNodeOnVertex(SMDS_MeshNode * aNode, const TopoDS_Vertex & S);
161   void UnSetNodeOnShape(const SMDS_MeshNode * aNode);
162   void SetMeshElementOnShape(const SMDS_MeshElement * anElt,
163                              const TopoDS_Shape & S);
164   void UnSetMeshElementOnShape(const SMDS_MeshElement * anElt,
165                                const TopoDS_Shape & S);
166   TopoDS_Shape ShapeToMesh() const;
167   bool HasMeshElements(const TopoDS_Shape & S);
168   SMESHDS_SubMesh * MeshElements(const TopoDS_Shape & S);
169   SMESHDS_SubMesh * MeshElements(const int Index);
170   list<int> SubMeshIndices();
171   const std::map<int,SMESHDS_SubMesh*>& SubMeshes()
172   { return myShapeIndexToSubMesh; }
173
174   bool HasHypothesis(const TopoDS_Shape & S);
175   const list<const SMESHDS_Hypothesis*>& GetHypothesis(const TopoDS_Shape & S) const;
176   SMESHDS_Script * GetScript();
177   void ClearScript();
178   int ShapeToIndex(const TopoDS_Shape & aShape);
179   TopoDS_Shape IndexToShape(int ShapeIndex);
180
181   void NewSubMesh(int Index);
182   void SetNodeInVolume(const SMDS_MeshNode * aNode, int Index);
183   void SetNodeOnFace(SMDS_MeshNode * aNode, int Index);
184   void SetNodeOnEdge(SMDS_MeshNode * aNode, int Index);
185   void SetNodeOnVertex(SMDS_MeshNode * aNode, int Index);
186   void SetMeshElementOnShape(const SMDS_MeshElement * anElt, int Index);
187
188   ~SMESHDS_Mesh();
189   
190 private:
191   struct HashTopoDS_Shape{
192     size_t operator()(const TopoDS_Shape& S) const {
193       return S.HashCode(2147483647);
194     }
195   };
196   typedef std::list<const SMESHDS_Hypothesis*> THypList;
197   typedef gstd::hash_map<TopoDS_Shape,THypList,HashTopoDS_Shape> ShapeToHypothesis;
198   ShapeToHypothesis myShapeToHypothesis;
199
200   int myMeshID;
201   TopoDS_Shape myShape;
202   TopTools_IndexedMapOfShape myIndexToShape;
203   std::map<int,SMESHDS_SubMesh*> myShapeIndexToSubMesh;
204   SMESHDS_Script* myScript;
205 };
206
207
208 #endif