Salome HOME
Add support for tetra, pyramid and prism
[modules/smesh.git] / src / SMESH / SMESH_Hexa_3D.cxx
1 //  SMESH SMESH : implementaion of SMESH idl descriptions
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   : SMESH_Hexa_3D.cxx
25 //  Author : Paul RASCLE, EDF
26 //  Module : SMESH
27 //  $Header$
28
29 using namespace std;
30 #include "SMESH_Hexa_3D.hxx"
31 #include "SMESH_Quadrangle_2D.hxx"
32 #include "SMESH_Gen.hxx"
33 #include "SMESH_Mesh.hxx"
34
35 #include "SMDS_MeshElement.hxx"
36 #include "SMDS_MeshNode.hxx"
37 #include "SMDS_FacePosition.hxx"
38
39 #include <TopExp.hxx>
40 #include <TopTools_IndexedDataMapOfShapeListOfShape.hxx>
41 #include <TopTools_ListOfShape.hxx>
42 #include <TopTools_ListIteratorOfListOfShape.hxx>
43 #include <TColStd_ListIteratorOfListOfInteger.hxx>
44
45 #include <BRep_Tool.hxx>
46 #include <Geom_Surface.hxx>
47 #include <Geom_Curve.hxx>
48 #include <Geom2d_Curve.hxx>
49 #include <Handle_Geom2d_Curve.hxx>
50 #include <Handle_Geom_Curve.hxx>
51
52 #include "utilities.h"
53
54 //=============================================================================
55 /*!
56  *  
57  */
58 //=============================================================================
59
60 SMESH_Hexa_3D::SMESH_Hexa_3D(int hypId, int studyId,
61         SMESH_Gen * gen):SMESH_3D_Algo(hypId, studyId, gen)
62 {
63         MESSAGE("SMESH_Hexa_3D::SMESH_Hexa_3D");
64         _name = "Hexa_3D";
65 //   _shapeType = TopAbs_SOLID;
66         _shapeType = (1 << TopAbs_SHELL) | (1 << TopAbs_SOLID); // 1 bit /shape type
67 //   MESSAGE("_shapeType octal " << oct << _shapeType);
68         for (int i = 0; i < 6; i++)
69                 _quads[i] = 0;
70 }
71
72 //=============================================================================
73 /*!
74  *  
75  */
76 //=============================================================================
77
78 SMESH_Hexa_3D::~SMESH_Hexa_3D()
79 {
80         MESSAGE("SMESH_Hexa_3D::~SMESH_Hexa_3D");
81 }
82
83 //=============================================================================
84 /*!
85  *  
86  */
87 //=============================================================================
88
89 bool SMESH_Hexa_3D::CheckHypothesis(SMESH_Mesh & aMesh,
90         const TopoDS_Shape & aShape)
91 {
92         MESSAGE("SMESH_Hexa_3D::CheckHypothesis");
93
94         bool isOk = true;
95
96         // nothing to check
97
98         return isOk;
99 }
100
101 //=============================================================================
102 /*!
103  * Hexahedron mesh on hexaedron like form
104  * -0.  - shape and face mesh verification
105  * -1.  - identify faces and vertices of the "cube"
106  * -2.  - Algorithm from:
107  * "Application de l'interpolation transfinie à la création de maillages
108  *  C0 ou G1 continus sur des triangles, quadrangles, tetraedres, pentaedres
109  *  et hexaedres déformés."
110  * Alain PERONNET - 8 janvier 1999
111  */
112 //=============================================================================
113
114 bool SMESH_Hexa_3D::Compute(SMESH_Mesh & aMesh,
115         const TopoDS_Shape & aShape)throw(SALOME_Exception)
116 {
117         MESSAGE("SMESH_Hexa_3D::Compute");
118
119         bool isOk = false;
120         SMESHDS_Mesh * meshDS = aMesh.GetMeshDS();
121         SMESH_subMesh *theSubMesh = aMesh.GetSubMesh(aShape);
122         //const SMESHDS_SubMesh *& subMeshDS = theSubMesh->GetSubMeshDS();
123
124         // 0.  - shape and face mesh verification
125         // 0.1 - shape must be a solid (or a shell) with 6 faces
126         MESSAGE("---");
127
128         vector < SMESH_subMesh * >meshFaces;
129         for (TopExp_Explorer exp(aShape, TopAbs_FACE); exp.More(); exp.Next())
130         {
131                 SMESH_subMesh *aSubMesh = aMesh.GetSubMeshContaining(exp.Current());
132                 ASSERT(aSubMesh);
133                 meshFaces.push_back(aSubMesh);
134         }
135         if (meshFaces.size() != 6)
136         {
137                 SCRUTE(meshFaces.size());
138                 ASSERT(0);
139                 return false;
140         }
141
142         // 0.2 - is each face meshed with Quadrangle_2D? (so, with a wire of 4 edges)
143         MESSAGE("---");
144
145         for (int i = 0; i < 6; i++)
146         {
147                 TopoDS_Shape aShape = meshFaces[i]->GetSubShape();
148                 SMESH_Algo *algo = _gen->GetAlgo(aMesh, aShape);
149                 string algoName = algo->GetName();
150                 if (algoName != "Quadrangle_2D")
151                 {
152                         // *** delete _quads
153                         SCRUTE(algoName);
154                         ASSERT(0);
155                         return false;
156                 }
157                 SMESH_Quadrangle_2D *quadAlgo =
158                         dynamic_cast < SMESH_Quadrangle_2D * >(algo);
159                 ASSERT(quadAlgo);
160                 try
161                 {
162                         _quads[i] = quadAlgo->CheckAnd2Dcompute(aMesh, aShape);
163                         // *** to delete after usage
164                 }
165                 catch(SALOME_Exception & S_ex)
166                 {
167                         // *** delete _quads
168                         // *** throw exception
169                         ASSERT(0);
170                 }
171         }
172
173         // 1.  - identify faces and vertices of the "cube"
174         // 1.1 - ancestor maps vertex->edges in the cube
175         MESSAGE("---");
176
177         TopTools_IndexedDataMapOfShapeListOfShape MS;
178         TopExp::MapShapesAndAncestors(aShape, TopAbs_VERTEX, TopAbs_EDGE, MS);
179
180         // 1.2 - first face is choosen as face Y=0 of the unit cube
181         MESSAGE("---");
182
183         const TopoDS_Shape & aFace = meshFaces[0]->GetSubShape();
184         const TopoDS_Face & F = TopoDS::Face(aFace);
185
186         // 1.3 - identify the 4 vertices of the face Y=0: V000, V100, V101, V001
187         MESSAGE("---");
188
189         int i = 0;
190         TopoDS_Edge E = _quads[0]->edge[i];     //edge will be Y=0,Z=0 on unit cube
191         double f, l;
192         Handle(Geom2d_Curve) C2d = BRep_Tool::CurveOnSurface(E, F, f, l);
193         TopoDS_Vertex VFirst, VLast;
194         TopExp::Vertices(E, VFirst, VLast);     // corresponds to f and l
195         bool isForward =
196                 (((l - f) * (_quads[0]->last[i] - _quads[0]->first[i])) > 0);
197
198         if (isForward)
199         {
200                 _cube.V000 = VFirst;    // will be (0,0,0) on the unit cube
201                 _cube.V100 = VLast;             // will be (1,0,0) on the unit cube
202         }
203         else
204         {
205                 _cube.V000 = VLast;
206                 _cube.V100 = VFirst;
207         }
208
209         i = 1;
210         E = _quads[0]->edge[i];
211         C2d = BRep_Tool::CurveOnSurface(E, F, f, l);
212         TopExp::Vertices(E, VFirst, VLast);
213         isForward = (((l - f) * (_quads[0]->last[i] - _quads[0]->first[i])) > 0);
214         if (isForward)
215                 _cube.V101 = VLast;             // will be (1,0,1) on the unit cube
216         else
217                 _cube.V101 = VFirst;
218
219         i = 2;
220         E = _quads[0]->edge[i];
221         C2d = BRep_Tool::CurveOnSurface(E, F, f, l);
222         TopExp::Vertices(E, VFirst, VLast);
223         isForward = (((l - f) * (_quads[0]->last[i] - _quads[0]->first[i])) > 0);
224         if (isForward)
225                 _cube.V001 = VLast;             // will be (0,0,1) on the unit cube
226         else
227                 _cube.V001 = VFirst;
228
229         // 1.4 - find edge X=0, Z=0 (ancestor of V000 not in face Y=0)
230         //     - find edge X=1, Z=0 (ancestor of V100 not in face Y=0)
231         //     - find edge X=1, Z=1 (ancestor of V101 not in face Y=0) 
232         //     - find edge X=0, Z=1 (ancestor of V001 not in face Y=0)
233         MESSAGE("---");
234
235         TopoDS_Edge E_0Y0 = EdgeNotInFace(aMesh, aShape, F, _cube.V000, MS);
236         ASSERT(!E_0Y0.IsNull());
237
238         TopoDS_Edge E_1Y0 = EdgeNotInFace(aMesh, aShape, F, _cube.V100, MS);
239         ASSERT(!E_1Y0.IsNull());
240
241         TopoDS_Edge E_1Y1 = EdgeNotInFace(aMesh, aShape, F, _cube.V101, MS);
242         ASSERT(!E_1Y1.IsNull());
243
244         TopoDS_Edge E_0Y1 = EdgeNotInFace(aMesh, aShape, F, _cube.V001, MS);
245         ASSERT(!E_0Y1.IsNull());
246
247         // 1.5 - identify the 4 vertices in face Y=1: V010, V110, V111, V011
248         MESSAGE("---");
249
250         TopExp::Vertices(E_0Y0, VFirst, VLast);
251         if (VFirst.IsSame(_cube.V000))
252                 _cube.V010 = VLast;
253         else
254                 _cube.V010 = VFirst;
255
256         TopExp::Vertices(E_1Y0, VFirst, VLast);
257         if (VFirst.IsSame(_cube.V100))
258                 _cube.V110 = VLast;
259         else
260                 _cube.V110 = VFirst;
261
262         TopExp::Vertices(E_1Y1, VFirst, VLast);
263         if (VFirst.IsSame(_cube.V101))
264                 _cube.V111 = VLast;
265         else
266                 _cube.V111 = VFirst;
267
268         TopExp::Vertices(E_0Y1, VFirst, VLast);
269         if (VFirst.IsSame(_cube.V001))
270                 _cube.V011 = VLast;
271         else
272                 _cube.V011 = VFirst;
273
274         // 1.6 - find remaining faces given 4 vertices
275         MESSAGE("---");
276
277         _indY0 = 0;
278         _cube.quad_Y0 = _quads[_indY0];
279
280         _indY1 = GetFaceIndex(aMesh, aShape, meshFaces,
281                 _cube.V010, _cube.V011, _cube.V110, _cube.V111);
282         _cube.quad_Y1 = _quads[_indY1];
283
284         _indZ0 = GetFaceIndex(aMesh, aShape, meshFaces,
285                 _cube.V000, _cube.V010, _cube.V100, _cube.V110);
286         _cube.quad_Z0 = _quads[_indZ0];
287
288         _indZ1 = GetFaceIndex(aMesh, aShape, meshFaces,
289                 _cube.V001, _cube.V011, _cube.V101, _cube.V111);
290         _cube.quad_Z1 = _quads[_indZ1];
291
292         _indX0 = GetFaceIndex(aMesh, aShape, meshFaces,
293                 _cube.V000, _cube.V001, _cube.V010, _cube.V011);
294         _cube.quad_X0 = _quads[_indX0];
295
296         _indX1 = GetFaceIndex(aMesh, aShape, meshFaces,
297                 _cube.V100, _cube.V101, _cube.V110, _cube.V111);
298         _cube.quad_X1 = _quads[_indX1];
299
300         MESSAGE("---");
301
302         // 1.7 - get convertion coefs from face 2D normalized to 3D normalized
303
304         Conv2DStruct cx0;                       // for face X=0
305         Conv2DStruct cx1;                       // for face X=1
306         Conv2DStruct cy0;
307         Conv2DStruct cy1;
308         Conv2DStruct cz0;
309         Conv2DStruct cz1;
310
311         GetConv2DCoefs(*_cube.quad_X0, meshFaces[_indX0]->GetSubShape(),
312                 _cube.V000, _cube.V010, _cube.V011, _cube.V001, cx0);
313         GetConv2DCoefs(*_cube.quad_X1, meshFaces[_indX1]->GetSubShape(),
314                 _cube.V100, _cube.V110, _cube.V111, _cube.V101, cx1);
315         GetConv2DCoefs(*_cube.quad_Y0, meshFaces[_indY0]->GetSubShape(),
316                 _cube.V000, _cube.V100, _cube.V101, _cube.V001, cy0);
317         GetConv2DCoefs(*_cube.quad_Y1, meshFaces[_indY1]->GetSubShape(),
318                 _cube.V010, _cube.V110, _cube.V111, _cube.V011, cy1);
319         GetConv2DCoefs(*_cube.quad_Z0, meshFaces[_indZ0]->GetSubShape(),
320                 _cube.V000, _cube.V100, _cube.V110, _cube.V010, cz0);
321         GetConv2DCoefs(*_cube.quad_Z1, meshFaces[_indZ1]->GetSubShape(),
322                 _cube.V001, _cube.V101, _cube.V111, _cube.V011, cz1);
323
324         // 1.8 - create a 3D structure for normalized values
325
326         MESSAGE("---");
327         int nbx = _cube.quad_Y0->nbPts[0];
328         int nby = _cube.quad_Y0->nbPts[1];
329         int nbz;
330         if (cx0.a1 != 0)
331                 nbz = _cube.quad_X0->nbPts[1];
332         else
333                 nbz = _cube.quad_X0->nbPts[0];
334         //SCRUTE(nbx);
335         //SCRUTE(nby);
336         //SCRUTE(nbz);
337         int nbxyz = nbx * nby * nbz;
338         Point3DStruct *np = new Point3DStruct[nbxyz];
339
340         // 1.9 - store node indexes of faces
341
342         {
343                 const TopoDS_Face & F = TopoDS::Face(meshFaces[_indX0]->GetSubShape());
344
345                 faceQuadStruct *quad = _cube.quad_X0;
346                 int i = 0;                              // j = x/face , k = y/face
347                 int nbdown = quad->nbPts[0];
348                 int nbright = quad->nbPts[1];
349
350
351                 SMDS_Iterator<const SMDS_MeshNode *> * itf=
352                         aMesh.GetSubMesh(F)->GetSubMeshDS()->GetNodes();
353                         
354                 while(itf->more())
355                 {
356                         const SMDS_MeshNode * node = itf->next();
357                         const SMDS_FacePosition* fpos
358                                 = static_cast<const SMDS_FacePosition*>(node->GetPosition());
359                         double ri = fpos->GetUParameter();
360                         double rj = fpos->GetVParameter();
361                         int i1 = int (ri);
362                         int j1 = int (rj);
363                         int ij1 = j1 * nbdown + i1;
364                         quad->uv_grid[ij1].node = node;
365                 }
366                 delete itf;
367
368                 for (int i1 = 0; i1 < nbdown; i1++)
369                         for (int j1 = 0; j1 < nbright; j1++)
370                         {
371                                 int ij1 = j1 * nbdown + i1;
372                                 int j = cx0.ia * i1 + cx0.ib * j1 + cx0.ic;     // j = x/face
373                                 int k = cx0.ja * i1 + cx0.jb * j1 + cx0.jc;     // k = y/face
374                                 int ijk = k * nbx * nby + j * nbx + i;
375                                 //MESSAGE(" "<<ij1<<" "<<i<<" "<<j<<" "<<ijk);
376                                 np[ijk].node = quad->uv_grid[ij1].node;
377                                 //SCRUTE(np[ijk].nodeId);
378                         }
379         }
380
381         {
382                 const TopoDS_Face & F = TopoDS::Face(meshFaces[_indX1]->GetSubShape());
383
384                 SMDS_Iterator<const SMDS_MeshNode *> * itf=
385                         aMesh.GetSubMesh(F)->GetSubMeshDS()->GetNodes();
386
387                 faceQuadStruct *quad = _cube.quad_X1;
388                 int i = nbx - 1;                // j = x/face , k = y/face
389                 int nbdown = quad->nbPts[0];
390                 int nbright = quad->nbPts[1];
391
392                 while(itf->more())
393                 {
394                         const SMDS_MeshNode * node = itf->next();
395                         const SMDS_FacePosition* fpos
396                                 = static_cast<const SMDS_FacePosition*>(node->GetPosition());
397                         double ri = fpos->GetUParameter();
398                         double rj = fpos->GetVParameter();
399                         int i1 = int (ri);
400                         int j1 = int (rj);
401                         int ij1 = j1 * nbdown + i1;
402                         quad->uv_grid[ij1].node = node;
403                 }
404                 delete itf;
405
406                 for (int i1 = 0; i1 < nbdown; i1++)
407                         for (int j1 = 0; j1 < nbright; j1++)
408                         {
409                                 int ij1 = j1 * nbdown + i1;
410                                 int j = cx1.ia * i1 + cx1.ib * j1 + cx1.ic;     // j = x/face
411                                 int k = cx1.ja * i1 + cx1.jb * j1 + cx1.jc;     // k = y/face
412                                 int ijk = k * nbx * nby + j * nbx + i;
413                                 //MESSAGE(" "<<ij1<<" "<<i<<" "<<j<<" "<<ijk);
414                                 np[ijk].node = quad->uv_grid[ij1].node;
415                                 //SCRUTE(np[ijk].nodeId);
416                         }
417         }
418
419         {
420                 const TopoDS_Face & F = TopoDS::Face(meshFaces[_indY0]->GetSubShape());
421
422                 SMDS_Iterator<const SMDS_MeshNode *> * itf=
423                         aMesh.GetSubMesh(F)->GetSubMeshDS()->GetNodes();
424
425                 faceQuadStruct *quad = _cube.quad_Y0;
426                 int j = 0;                              // i = x/face , k = y/face
427                 int nbdown = quad->nbPts[0];
428                 int nbright = quad->nbPts[1];
429
430                 while(itf->more())
431                 {
432                         const SMDS_MeshNode * node = itf->next();
433                         const SMDS_FacePosition * fpos
434                                 = static_cast<const SMDS_FacePosition*>(node->GetPosition());
435                         double ri = fpos->GetUParameter();
436                         double rj = fpos->GetVParameter();
437                         int i1 = int (ri);
438                         int j1 = int (rj);
439                         int ij1 = j1 * nbdown + i1;
440                         quad->uv_grid[ij1].node = node;
441                 }
442                 delete itf;
443
444                 for (int i1 = 0; i1 < nbdown; i1++)
445                         for (int j1 = 0; j1 < nbright; j1++)
446                         {
447                                 int ij1 = j1 * nbdown + i1;
448                                 int i = cy0.ia * i1 + cy0.ib * j1 + cy0.ic;     // i = x/face
449                                 int k = cy0.ja * i1 + cy0.jb * j1 + cy0.jc;     // k = y/face
450                                 int ijk = k * nbx * nby + j * nbx + i;
451                                 //MESSAGE(" "<<ij1<<" "<<i<<" "<<j<<" "<<ijk);
452                                 np[ijk].node = quad->uv_grid[ij1].node;
453                                 //SCRUTE(np[ijk].nodeId);
454                         }
455         }
456
457         {
458                 const TopoDS_Face & F = TopoDS::Face(meshFaces[_indY1]->GetSubShape());
459
460                 SMDS_Iterator<const SMDS_MeshNode *> * itf=
461                         aMesh.GetSubMesh(F)->GetSubMeshDS()->GetNodes();
462
463                 faceQuadStruct *quad = _cube.quad_Y1;
464                 int j = nby - 1;                // i = x/face , k = y/face
465                 int nbdown = quad->nbPts[0];
466                 int nbright = quad->nbPts[1];
467
468                 while(itf->more())
469                 {
470                         const SMDS_MeshNode * node = itf->next();
471                         const SMDS_FacePosition* fpos =
472                                 static_cast<const SMDS_FacePosition *>(node->GetPosition());
473                         double ri = fpos->GetUParameter();
474                         double rj = fpos->GetVParameter();
475                         int i1 = int (ri);
476                         int j1 = int (rj);
477                         int ij1 = j1 * nbdown + i1;
478                         quad->uv_grid[ij1].node = node;
479                 }
480
481                 for (int i1 = 0; i1 < nbdown; i1++)
482                         for (int j1 = 0; j1 < nbright; j1++)
483                         {
484                                 int ij1 = j1 * nbdown + i1;
485                                 int i = cy1.ia * i1 + cy1.ib * j1 + cy1.ic;     // i = x/face
486                                 int k = cy1.ja * i1 + cy1.jb * j1 + cy1.jc;     // k = y/face
487                                 int ijk = k * nbx * nby + j * nbx + i;
488                                 //MESSAGE(" "<<ij1<<" "<<i<<" "<<j<<" "<<ijk);
489                                 np[ijk].node = quad->uv_grid[ij1].node;
490                                 //SCRUTE(np[ijk].nodeId);
491                         }
492         }
493
494         {
495                 const TopoDS_Face & F = TopoDS::Face(meshFaces[_indZ0]->GetSubShape());
496
497                 SMDS_Iterator<const SMDS_MeshNode *> * itf=
498                         aMesh.GetSubMesh(F)->GetSubMeshDS()->GetNodes();
499
500                 faceQuadStruct *quad = _cube.quad_Z0;
501                 int k = 0;                              // i = x/face , j = y/face
502                 int nbdown = quad->nbPts[0];
503                 int nbright = quad->nbPts[1];
504
505                 while(itf->more())
506                 {
507                         const SMDS_MeshNode * node = itf->next();
508                         const SMDS_FacePosition * fpos
509                                 = static_cast<const SMDS_FacePosition*>(node->GetPosition());
510                         double ri = fpos->GetUParameter();
511                         double rj = fpos->GetVParameter();
512                         int i1 = int (ri);
513                         int j1 = int (rj);
514                         int ij1 = j1 * nbdown + i1;
515                         quad->uv_grid[ij1].node = node;
516                 }
517
518                 for (int i1 = 0; i1 < nbdown; i1++)
519                         for (int j1 = 0; j1 < nbright; j1++)
520                         {
521                                 int ij1 = j1 * nbdown + i1;
522                                 int i = cz0.ia * i1 + cz0.ib * j1 + cz0.ic;     // i = x/face
523                                 int j = cz0.ja * i1 + cz0.jb * j1 + cz0.jc;     // j = y/face
524                                 int ijk = k * nbx * nby + j * nbx + i;
525                                 //MESSAGE(" "<<ij1<<" "<<i<<" "<<j<<" "<<ijk);
526                                 np[ijk].node = quad->uv_grid[ij1].node;
527                                 //SCRUTE(np[ijk].nodeId);
528                         }
529         }
530
531         {
532                 const TopoDS_Face & F = TopoDS::Face(meshFaces[_indZ1]->GetSubShape());
533
534                 SMDS_Iterator<const SMDS_MeshNode *> * itf=
535                         aMesh.GetSubMesh(F)->GetSubMeshDS()->GetNodes();
536
537                 faceQuadStruct *quad = _cube.quad_Z1;
538                 int k = nbz - 1;                // i = x/face , j = y/face
539                 int nbdown = quad->nbPts[0];
540                 int nbright = quad->nbPts[1];
541
542                 while(itf->more())
543                 {
544                         const SMDS_MeshNode * node = itf->next();
545                         const SMDS_FacePosition* fpos
546                                 = static_cast<const SMDS_FacePosition*>(node->GetPosition());
547                         double ri = fpos->GetUParameter();
548                         double rj = fpos->GetVParameter();
549                         int i1 = int (ri);
550                         int j1 = int (rj);
551                         int ij1 = j1 * nbdown + i1;
552                         quad->uv_grid[ij1].node = node;
553                 }
554
555                 for (int i1 = 0; i1 < nbdown; i1++)
556                         for (int j1 = 0; j1 < nbright; j1++)
557                         {
558                                 int ij1 = j1 * nbdown + i1;
559                                 int i = cz1.ia * i1 + cz1.ib * j1 + cz1.ic;     // i = x/face
560                                 int j = cz1.ja * i1 + cz1.jb * j1 + cz1.jc;     // j = y/face
561                                 int ijk = k * nbx * nby + j * nbx + i;
562                                 //MESSAGE(" "<<ij1<<" "<<i<<" "<<j<<" "<<ijk);
563                                 np[ijk].node = quad->uv_grid[ij1].node;
564                                 //SCRUTE(np[ijk].nodeId);
565                         }
566         }
567
568         // 2.0 - for each node of the cube:
569         //       - get the 8 points 3D = 8 vertices of the cube
570         //       - get the 12 points 3D on the 12 edges of the cube
571         //       - get the 6 points 3D on the 6 faces with their ID
572         //       - compute the point 3D
573         //       - store the point 3D in SMESHDS, store its ID in 3D structure
574
575         TopoDS_Shell aShell;
576         TopExp_Explorer exp(aShape, TopAbs_SHELL);
577         if (exp.More())
578         {
579                 aShell = TopoDS::Shell(exp.Current());
580         }
581         else
582         {
583                 MESSAGE("no shell...");
584                 ASSERT(0);
585         }
586
587         Pt3 p000, p001, p010, p011, p100, p101, p110, p111;
588         Pt3 px00, px01, px10, px11;
589         Pt3 p0y0, p0y1, p1y0, p1y1;
590         Pt3 p00z, p01z, p10z, p11z;
591         Pt3 pxy0, pxy1, px0z, px1z, p0yz, p1yz;
592
593         GetPoint(p000, 0, 0, 0, nbx, nby, nbz, np, meshDS);
594         GetPoint(p001, 0, 0, nbz - 1, nbx, nby, nbz, np, meshDS);
595         GetPoint(p010, 0, nby - 1, 0, nbx, nby, nbz, np, meshDS);
596         GetPoint(p011, 0, nby - 1, nbz - 1, nbx, nby, nbz, np, meshDS);
597         GetPoint(p100, nbx - 1, 0, 0, nbx, nby, nbz, np, meshDS);
598         GetPoint(p101, nbx - 1, 0, nbz - 1, nbx, nby, nbz, np, meshDS);
599         GetPoint(p110, nbx - 1, nby - 1, 0, nbx, nby, nbz, np, meshDS);
600         GetPoint(p111, nbx - 1, nby - 1, nbz - 1, nbx, nby, nbz, np, meshDS);
601
602         for (int i = 1; i < nbx - 1; i++)
603         {
604                 for (int j = 1; j < nby - 1; j++)
605                 {
606                         for (int k = 1; k < nbz - 1; k++)
607                         {
608                                 // *** seulement maillage regulier
609                                 // 12 points on edges
610                                 GetPoint(px00, i, 0, 0, nbx, nby, nbz, np, meshDS);
611                                 GetPoint(px01, i, 0, nbz - 1, nbx, nby, nbz, np, meshDS);
612                                 GetPoint(px10, i, nby - 1, 0, nbx, nby, nbz, np, meshDS);
613                                 GetPoint(px11, i, nby - 1, nbz - 1, nbx, nby, nbz, np, meshDS);
614
615                                 GetPoint(p0y0, 0, j, 0, nbx, nby, nbz, np, meshDS);
616                                 GetPoint(p0y1, 0, j, nbz - 1, nbx, nby, nbz, np, meshDS);
617                                 GetPoint(p1y0, nbx - 1, j, 0, nbx, nby, nbz, np, meshDS);
618                                 GetPoint(p1y1, nbx - 1, j, nbz - 1, nbx, nby, nbz, np, meshDS);
619
620                                 GetPoint(p00z, 0, 0, k, nbx, nby, nbz, np, meshDS);
621                                 GetPoint(p01z, 0, nby - 1, k, nbx, nby, nbz, np, meshDS);
622                                 GetPoint(p10z, nbx - 1, 0, k, nbx, nby, nbz, np, meshDS);
623                                 GetPoint(p11z, nbx - 1, nby - 1, k, nbx, nby, nbz, np, meshDS);
624
625                                 // 12 points on faces
626                                 GetPoint(pxy0, i, j, 0, nbx, nby, nbz, np, meshDS);
627                                 GetPoint(pxy1, i, j, nbz - 1, nbx, nby, nbz, np, meshDS);
628                                 GetPoint(px0z, i, 0, k, nbx, nby, nbz, np, meshDS);
629                                 GetPoint(px1z, i, nby - 1, k, nbx, nby, nbz, np, meshDS);
630                                 GetPoint(p0yz, 0, j, k, nbx, nby, nbz, np, meshDS);
631                                 GetPoint(p1yz, nbx - 1, j, k, nbx, nby, nbz, np, meshDS);
632
633                                 int ijk = k * nbx * nby + j * nbx + i;
634                                 double x = double (i) / double (nbx - 1);       // *** seulement
635                                 double y = double (j) / double (nby - 1);       // *** maillage
636                                 double z = double (k) / double (nbz - 1);       // *** regulier
637
638                                 Pt3 X;
639                                 for (int i = 0; i < 3; i++)
640                                 {
641                                         X[i] =
642                                                 (1 - x) * p0yz[i] + x * p1yz[i]
643                                                 + (1 - y) * px0z[i] + y * px1z[i]
644                                                 + (1 - z) * pxy0[i] + z * pxy1[i]
645                                                 - (1 - x) * ((1 - y) * p00z[i] + y * p01z[i])
646                                                 - x * ((1 - y) * p10z[i] + y * p11z[i])
647                                                 - (1 - y) * ((1 - z) * px00[i] + z * px01[i])
648                                                 - y * ((1 - z) * px10[i] + z * px11[i])
649                                                 - (1 - z) * ((1 - x) * p0y0[i] + x * p1y0[i])
650                                                 - z * ((1 - x) * p0y1[i] + x * p1y1[i])
651                                                 + (1 - x) * ((1 - y) * ((1 - z) * p000[i] + z * p001[i])
652                                                 + y * ((1 - z) * p010[i] + z * p011[i]))
653                                                 + x * ((1 - y) * ((1 - z) * p100[i] + z * p101[i])
654                                                 + y * ((1 - z) * p110[i] + z * p111[i]));
655                                 }
656
657                                 SMDS_MeshNode * node = meshDS->AddNode(X[0], X[1], X[2]);
658                                 np[ijk].node = node;
659                                 //meshDS->SetNodeInVolume(node, TopoDS::Solid(aShape));
660                                 meshDS->SetNodeInVolume(node, aShell);
661                         }
662                 }
663         }
664
665         //2.1 - for each node of the cube (less 3 *1 Faces):
666         //      - store hexahedron in SMESHDS
667         MESSAGE("Storing hexahedron into the DS");
668         for (int i = 0; i < nbx - 1; i++)
669                 for (int j = 0; j < nby - 1; j++)
670                         for (int k = 0; k < nbz - 1; k++)
671                         {
672                                 int n1 = k * nbx * nby + j * nbx + i;
673                                 int n2 = k * nbx * nby + j * nbx + i + 1;
674                                 int n3 = k * nbx * nby + (j + 1) * nbx + i + 1;
675                                 int n4 = k * nbx * nby + (j + 1) * nbx + i;
676                                 int n5 = (k + 1) * nbx * nby + j * nbx + i;
677                                 int n6 = (k + 1) * nbx * nby + j * nbx + i + 1;
678                                 int n7 = (k + 1) * nbx * nby + (j + 1) * nbx + i + 1;
679                                 int n8 = (k + 1) * nbx * nby + (j + 1) * nbx + i;
680
681 //    MESSAGE(" "<<n1<<" "<<n2<<" "<<n3<<" "<<n4<<" "<<n5<<" "<<n6<<" "<<n7<<" "<<n8);
682                                 //MESSAGE(" "<<np[n1].nodeId<<" "<<np[n2].nodeId<<" "<<np[n3].nodeId<<" "<<np[n4].nodeId<<" "<<np[n5].nodeId<<" "<<np[n6].nodeId<<" "<<np[n7].nodeId<<" "<<np[n8].nodeId);
683
684                                 SMDS_MeshVolume * elt = meshDS->AddVolume(np[n1].node,
685                                         np[n2].node,
686                                         np[n3].node,
687                                         np[n4].node,
688                                         np[n5].node,
689                                         np[n6].node,
690                                         np[n7].node,
691                                         np[n8].node);
692                                  ;
693                                 meshDS->SetMeshElementOnShape(elt, aShell);
694
695                                 // *** 5 tetrahedres ... verifier orientations,
696                                 //     mettre en coherence &vec quadrangles-> triangles
697                                 //     choisir afficher 1 parmi edges, face et volumes
698 //    int tetra1 = meshDS->AddVolume(np[n1].nodeId,
699 //                   np[n2].nodeId,
700 //                   np[n4].nodeId,
701 //                   np[n5].nodeId);
702 //    int tetra2 = meshDS->AddVolume(np[n2].nodeId,
703 //                   np[n3].nodeId,
704 //                   np[n4].nodeId,
705 //                   np[n7].nodeId);
706 //    int tetra3 = meshDS->AddVolume(np[n5].nodeId,
707 //                   np[n6].nodeId,
708 //                   np[n7].nodeId,
709 //                   np[n2].nodeId);
710 //    int tetra4 = meshDS->AddVolume(np[n5].nodeId,
711 //                   np[n7].nodeId,
712 //                   np[n8].nodeId,
713 //                   np[n4].nodeId);
714 //    int tetra5 = meshDS->AddVolume(np[n5].nodeId,
715 //                   np[n7].nodeId,
716 //                   np[n2].nodeId,
717 //                   np[n4].nodeId);
718
719                         }
720
721         MESSAGE("End of SMESH_Hexa_3D::Compute()");
722         return true;
723 }
724
725 //=============================================================================
726 /*!
727  *  
728  */
729 //=============================================================================
730
731 void SMESH_Hexa_3D::GetPoint(Pt3 p, int i, int j, int k, int nbx, int nby,
732         int nbz, Point3DStruct * np, const SMESHDS_Mesh * meshDS)
733 {
734         int ijk = k * nbx * nby + j * nbx + i;
735         const SMDS_MeshNode * node = np[ijk].node;
736         p[0] = node->X();
737         p[1] = node->Y();
738         p[2] = node->Z();
739         //MESSAGE(" "<<i<<" "<<j<<" "<<k<<" "<<p[0]<<" "<<p[1]<<" "<<p[2]);
740 }
741
742 //=============================================================================
743 /*!
744  *  
745  */
746 //=============================================================================
747
748 int SMESH_Hexa_3D::GetFaceIndex(SMESH_Mesh & aMesh,
749         const TopoDS_Shape & aShape,
750         const vector < SMESH_subMesh * >&meshFaces,
751         const TopoDS_Vertex & V0,
752         const TopoDS_Vertex & V1,
753         const TopoDS_Vertex & V2, const TopoDS_Vertex & V3)
754 {
755         MESSAGE("SMESH_Hexa_3D::GetFaceIndex");
756         int faceIndex = -1;
757         for (int i = 1; i < 6; i++)
758         {
759                 const TopoDS_Shape & aFace = meshFaces[i]->GetSubShape();
760                 //const TopoDS_Face& F = TopoDS::Face(aFace);
761                 TopTools_IndexedMapOfShape M;
762                 TopExp::MapShapes(aFace, TopAbs_VERTEX, M);
763                 bool verticesInShape = false;
764                 if (M.Contains(V0))
765                         if (M.Contains(V1))
766                                 if (M.Contains(V2))
767                                         if (M.Contains(V3))
768                                                 verticesInShape = true;
769                 if (verticesInShape)
770                 {
771                         faceIndex = i;
772                         break;
773                 }
774         }
775         ASSERT(faceIndex > 0);
776         SCRUTE(faceIndex);
777         return faceIndex;
778 }
779
780 //=============================================================================
781 /*!
782  *  
783  */
784 //=============================================================================
785
786 TopoDS_Edge
787         SMESH_Hexa_3D::EdgeNotInFace(SMESH_Mesh & aMesh,
788         const TopoDS_Shape & aShape,
789         const TopoDS_Face & aFace,
790         const TopoDS_Vertex & aVertex,
791         const TopTools_IndexedDataMapOfShapeListOfShape & MS)
792 {
793         MESSAGE("SMESH_Hexa_3D::EdgeNotInFace");
794         TopTools_IndexedDataMapOfShapeListOfShape MF;
795         TopExp::MapShapesAndAncestors(aFace, TopAbs_VERTEX, TopAbs_EDGE, MF);
796         const TopTools_ListOfShape & ancestorsInSolid = MS.FindFromKey(aVertex);
797         const TopTools_ListOfShape & ancestorsInFace = MF.FindFromKey(aVertex);
798         SCRUTE(ancestorsInSolid.Extent());
799         SCRUTE(ancestorsInFace.Extent());
800         ASSERT(ancestorsInSolid.Extent() == 6); // 6 (edges doublees)
801         ASSERT(ancestorsInFace.Extent() == 2);
802
803         TopoDS_Edge E;
804         E.Nullify();
805         TopTools_ListIteratorOfListOfShape its(ancestorsInSolid);
806         for (; its.More(); its.Next())
807         {
808                 TopoDS_Shape ancestor = its.Value();
809                 TopTools_ListIteratorOfListOfShape itf(ancestorsInFace);
810                 bool isInFace = false;
811                 for (; itf.More(); itf.Next())
812                 {
813                         TopoDS_Shape ancestorInFace = itf.Value();
814                         if (ancestorInFace.IsSame(ancestor))
815                         {
816                                 isInFace = true;
817                                 break;
818                         }
819                 }
820                 if (!isInFace)
821                 {
822                         E = TopoDS::Edge(ancestor);
823                         break;
824                 }
825         }
826         return E;
827 }
828
829 //=============================================================================
830 /*!
831  *  
832  */
833 //=============================================================================
834
835 void SMESH_Hexa_3D::GetConv2DCoefs(const faceQuadStruct & quad,
836         const TopoDS_Shape & aShape,
837         const TopoDS_Vertex & V0,
838         const TopoDS_Vertex & V1,
839         const TopoDS_Vertex & V2, const TopoDS_Vertex & V3, Conv2DStruct & conv)
840 {
841         MESSAGE("SMESH_Hexa_3D::GetConv2DCoefs");
842         const TopoDS_Face & F = TopoDS::Face(aShape);
843         TopoDS_Edge E = quad.edge[0];
844         double f, l;
845         Handle(Geom2d_Curve) C2d = BRep_Tool::CurveOnSurface(E, F, f, l);
846         TopoDS_Vertex VFirst, VLast;
847         TopExp::Vertices(E, VFirst, VLast);     // corresponds to f and l
848         bool isForward = (((l - f) * (quad.last[0] - quad.first[0])) > 0);
849         TopoDS_Vertex VA, VB;
850         if (isForward)
851         {
852                 VA = VFirst;
853                 VB = VLast;
854         }
855         else
856         {
857                 VA = VLast;
858                 VB = VFirst;
859         }
860         int a1, b1, c1, a2, b2, c2;
861         if (VA.IsSame(V0))
862                 if (VB.IsSame(V1))
863                 {
864                         a1 = 1;
865                         b1 = 0;
866                         c1 = 0;                         // x
867                         a2 = 0;
868                         b2 = 1;
869                         c2 = 0;                         // y
870                 }
871                 else
872                 {
873                         ASSERT(VB.IsSame(V3));
874                         a1 = 0;
875                         b1 = 1;
876                         c1 = 0;                         // y
877                         a2 = 1;
878                         b2 = 0;
879                         c2 = 0;                         // x
880                 }
881         if (VA.IsSame(V1))
882                 if (VB.IsSame(V2))
883                 {
884                         a1 = 0;
885                         b1 = -1;
886                         c1 = 1;                         // 1-y
887                         a2 = 1;
888                         b2 = 0;
889                         c2 = 0;                         // x
890                 }
891                 else
892                 {
893                         ASSERT(VB.IsSame(V0));
894                         a1 = -1;
895                         b1 = 0;
896                         c1 = 1;                         // 1-x
897                         a2 = 0;
898                         b2 = 1;
899                         c2 = 0;                         // y
900                 }
901         if (VA.IsSame(V2))
902                 if (VB.IsSame(V3))
903                 {
904                         a1 = -1;
905                         b1 = 0;
906                         c1 = 1;                         // 1-x
907                         a2 = 0;
908                         b2 = -1;
909                         c2 = 1;                         // 1-y
910                 }
911                 else
912                 {
913                         ASSERT(VB.IsSame(V1));
914                         a1 = 0;
915                         b1 = -1;
916                         c1 = 1;                         // 1-y
917                         a2 = -1;
918                         b2 = 0;
919                         c2 = 1;                         // 1-x
920                 }
921         if (VA.IsSame(V3))
922                 if (VB.IsSame(V0))
923                 {
924                         a1 = 0;
925                         b1 = 1;
926                         c1 = 0;                         // y
927                         a2 = -1;
928                         b2 = 0;
929                         c2 = 1;                         // 1-x
930                 }
931                 else
932                 {
933                         ASSERT(VB.IsSame(V2));
934                         a1 = 1;
935                         b1 = 0;
936                         c1 = 0;                         // x
937                         a2 = 0;
938                         b2 = -1;
939                         c2 = 1;                         // 1-y
940                 }
941         MESSAGE("X = " << c1 << "+ " << a1 << "*x + " << b1 << "*y");
942         MESSAGE("Y = " << c2 << "+ " << a2 << "*x + " << b2 << "*y");
943         conv.a1 = a1;
944         conv.b1 = b1;
945         conv.c1 = c1;
946         conv.a2 = a2;
947         conv.b2 = b2;
948         conv.c2 = c2;
949
950         int nbdown = quad.nbPts[0];
951         int nbright = quad.nbPts[1];
952         conv.ia = int (a1);
953         conv.ib = int (b1);
954         conv.ic =
955                 int (c1 * a1 * a1) * (nbdown - 1) + int (c1 * b1 * b1) * (nbright - 1);
956         conv.ja = int (a2);
957         conv.jb = int (b2);
958         conv.jc =
959                 int (c2 * a2 * a2) * (nbdown - 1) + int (c2 * b2 * b2) * (nbright - 1);
960         MESSAGE("I " << conv.ia << " " << conv.ib << " " << conv.ic);
961         MESSAGE("J " << conv.ja << " " << conv.jb << " " << conv.jc);
962 }
963
964 //=============================================================================
965 /*!
966  *  
967  */
968 //=============================================================================
969
970 ostream & SMESH_Hexa_3D::SaveTo(ostream & save)
971 {
972         return save << this;
973 }
974
975 //=============================================================================
976 /*!
977  *  
978  */
979 //=============================================================================
980
981 istream & SMESH_Hexa_3D::LoadFrom(istream & load)
982 {
983         return load >> (*this);
984 }
985
986 //=============================================================================
987 /*!
988  *  
989  */
990 //=============================================================================
991
992 ostream & operator <<(ostream & save, SMESH_Hexa_3D & hyp)
993 {
994         return save;
995 }
996
997 //=============================================================================
998 /*!
999  *  
1000  */
1001 //=============================================================================
1002
1003 istream & operator >>(istream & load, SMESH_Hexa_3D & hyp)
1004 {
1005         return load;
1006 }