1 // Copyright (C) 2007-2008 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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.
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.
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
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
22 //=============================================================================
23 // File : NETGENPlugin_NETGEN_3D.cxx
24 // Moved here from SMESH_NETGEN_3D.cxx
25 // Created : lundi 27 Janvier 2003
26 // Author : Nadir BOUHAMOU (CEA)
28 //=============================================================================
30 #include "NETGENPlugin_NETGEN_3D.hxx"
32 #include "NETGENPlugin_Mesher.hxx"
34 #include "SMDS_MeshElement.hxx"
35 #include "SMDS_MeshNode.hxx"
36 #include "SMESHDS_Mesh.hxx"
37 #include "SMESH_Comment.hxx"
38 #include "SMESH_ControlsDef.hxx"
39 #include "SMESH_Gen.hxx"
40 #include "SMESH_Mesh.hxx"
41 #include "SMESH_MesherHelper.hxx"
42 #include "SMESH_MeshEditor.hxx"
43 #include "StdMeshers_QuadToTriaAdaptor.hxx"
45 #include <BRep_Tool.hxx>
47 #include <TopExp_Explorer.hxx>
50 #include <Standard_Failure.hxx>
51 #include <Standard_ErrorHandler.hxx>
53 #include "utilities.h"
66 using namespace nglib;
69 //=============================================================================
73 //=============================================================================
75 NETGENPlugin_NETGEN_3D::NETGENPlugin_NETGEN_3D(int hypId, int studyId,
77 : SMESH_3D_Algo(hypId, studyId, gen)
79 MESSAGE("NETGENPlugin_NETGEN_3D::NETGENPlugin_NETGEN_3D");
81 _shapeType = (1 << TopAbs_SHELL) | (1 << TopAbs_SOLID);// 1 bit /shape type
82 _compatibleHypothesis.push_back("MaxElementVolume");
84 _maxElementVolume = 0.;
86 _hypMaxElementVolume = NULL;
88 _requireShape = false; // can work without shape
91 //=============================================================================
95 //=============================================================================
97 NETGENPlugin_NETGEN_3D::~NETGENPlugin_NETGEN_3D()
99 MESSAGE("NETGENPlugin_NETGEN_3D::~NETGENPlugin_NETGEN_3D");
102 //=============================================================================
106 //=============================================================================
108 bool NETGENPlugin_NETGEN_3D::CheckHypothesis
110 const TopoDS_Shape& aShape,
111 SMESH_Hypothesis::Hypothesis_Status& aStatus)
113 MESSAGE("NETGENPlugin_NETGEN_3D::CheckHypothesis");
115 _hypMaxElementVolume = NULL;
116 _maxElementVolume = DBL_MAX;
118 list<const SMESHDS_Hypothesis*>::const_iterator itl;
119 const SMESHDS_Hypothesis* theHyp;
121 const list<const SMESHDS_Hypothesis*>& hyps = GetUsedHypothesis(aMesh, aShape);
122 int nbHyp = hyps.size();
125 aStatus = SMESH_Hypothesis::HYP_OK;
126 //aStatus = SMESH_Hypothesis::HYP_MISSING;
127 return true; // can work with no hypothesis
131 theHyp = (*itl); // use only the first hypothesis
133 string hypName = theHyp->GetName();
137 if (hypName == "MaxElementVolume")
139 _hypMaxElementVolume = static_cast<const StdMeshers_MaxElementVolume*> (theHyp);
140 ASSERT(_hypMaxElementVolume);
141 _maxElementVolume = _hypMaxElementVolume->GetMaxVolume();
143 aStatus = SMESH_Hypothesis::HYP_OK;
146 aStatus = SMESH_Hypothesis::HYP_INCOMPATIBLE;
151 //=============================================================================
153 *Here we are going to use the NETGEN mesher
155 //=============================================================================
157 bool NETGENPlugin_NETGEN_3D::Compute(SMESH_Mesh& aMesh,
158 const TopoDS_Shape& aShape)
160 MESSAGE("NETGENPlugin_NETGEN_3D::Compute with maxElmentsize = " << _maxElementVolume);
162 SMESHDS_Mesh* meshDS = aMesh.GetMeshDS();
164 const int invalid_ID = -1;
166 SMESH::Controls::Area areaControl;
167 SMESH::Controls::TSequenceOfXYZ nodesCoords;
169 // -------------------------------------------------------------------
170 // get triangles on aShell and make a map of nodes to Netgen node IDs
171 // -------------------------------------------------------------------
173 SMESH_MesherHelper helper(aMesh);
174 SMESH_MesherHelper* myTool = &helper;
175 bool _quadraticMesh = myTool->IsQuadraticSubMesh(aShape);
177 typedef map< const SMDS_MeshNode*, int, TIDCompare > TNodeToIDMap;
178 TNodeToIDMap nodeToNetgenID;
179 list< const SMDS_MeshElement* > triangles;
180 list< bool > isReversed; // orientation of triangles
182 TopAbs_ShapeEnum mainType = aMesh.GetShapeToMesh().ShapeType();
183 bool checkReverse = ( mainType == TopAbs_COMPOUND || mainType == TopAbs_COMPSOLID );
185 // for the degeneraged edge: ignore all but one node on it;
186 // map storing ids of degen edges and vertices and their netgen id:
187 map< int, int* > degenShapeIdToPtrNgId;
188 map< int, int* >::iterator shId_ngId;
189 list< int > degenNgIds;
191 StdMeshers_QuadToTriaAdaptor Adaptor;
192 Adaptor.Compute(aMesh,aShape);
194 for (TopExp_Explorer exp(aShape,TopAbs_FACE);exp.More();exp.Next())
196 const TopoDS_Shape& aShapeFace = exp.Current();
197 const SMESHDS_SubMesh * aSubMeshDSFace = meshDS->MeshElements( aShapeFace );
198 if ( aSubMeshDSFace )
201 if ( checkReverse && helper.NbAncestors(aShapeFace, aMesh, aShape.ShapeType()) > 1 )
202 // IsReversedSubMesh() can work wrong on strongly curved faces,
203 // so we use it as less as possible
204 isRev = SMESH_Algo::IsReversedSubMesh( TopoDS::Face(aShapeFace), meshDS );
206 SMDS_ElemIteratorPtr iteratorElem = aSubMeshDSFace->GetElements();
207 while ( iteratorElem->more() ) // loop on elements on a face
210 const SMDS_MeshElement* elem = iteratorElem->next();
212 return error( COMPERR_BAD_INPUT_MESH, "Null element encounters");
213 bool isTraingle = ( elem->NbNodes()==3 || (_quadraticMesh && elem->NbNodes()==6 ));
215 //return error( COMPERR_BAD_INPUT_MESH,
216 // SMESH_Comment("Not triangle element ")<<elem->GetID());
218 const list<const SMDS_FaceOfNodes*>* faces = Adaptor.GetTriangles(elem);
220 return error( COMPERR_BAD_INPUT_MESH,
221 SMESH_Comment("Not triangles in adaptor for element ")<<elem->GetID());
223 list<const SMDS_FaceOfNodes*>::const_iterator itf = faces->begin();
224 for(; itf!=faces->end(); itf++ ) {
225 triangles.push_back( (*itf) );
226 isReversed.push_back( isRev );
227 // put triange's nodes to nodeToNetgenID map
228 SMDS_ElemIteratorPtr triangleNodesIt = (*itf)->nodesIterator();
229 while ( triangleNodesIt->more() ) {
230 const SMDS_MeshNode * node =
231 static_cast<const SMDS_MeshNode *>(triangleNodesIt->next());
232 if(myTool->IsMedium(node))
234 nodeToNetgenID.insert( make_pair( node, invalid_ID ));
240 triangles.push_back( elem );
241 isReversed.push_back( isRev );
242 // put elem nodes to nodeToNetgenID map
243 SMDS_ElemIteratorPtr triangleNodesIt = elem->nodesIterator();
244 while ( triangleNodesIt->more() ) {
245 const SMDS_MeshNode * node =
246 static_cast<const SMDS_MeshNode *>(triangleNodesIt->next());
247 if(myTool->IsMedium(node))
249 nodeToNetgenID.insert( make_pair( node, invalid_ID ));
253 // check if a trainge is degenerated
254 areaControl.GetPoints( elem, nodesCoords );
255 double area = areaControl.GetValue( nodesCoords );
256 if ( area <= DBL_MIN ) {
257 MESSAGE( "Warning: Degenerated " << elem );
261 // look for degeneraged edges and vetices
262 for (TopExp_Explorer expE(aShapeFace,TopAbs_EDGE);expE.More();expE.Next())
264 TopoDS_Edge aShapeEdge = TopoDS::Edge( expE.Current() );
265 if ( BRep_Tool::Degenerated( aShapeEdge ))
267 degenNgIds.push_back( invalid_ID );
268 int* ptrIdOnEdge = & degenNgIds.back();
270 int edgeID = meshDS->ShapeToIndex( aShapeEdge );
271 degenShapeIdToPtrNgId.insert( make_pair( edgeID, ptrIdOnEdge ));
272 // remember vertex id
273 int vertexID = meshDS->ShapeToIndex( TopExp::FirstVertex( aShapeEdge ));
274 degenShapeIdToPtrNgId.insert( make_pair( vertexID, ptrIdOnEdge ));
279 // ---------------------------------
280 // Feed the Netgen with surface mesh
281 // ---------------------------------
283 int Netgen_NbOfNodes = 0;
284 int Netgen_param2ndOrder = 0;
285 double Netgen_paramFine = 1.;
286 double Netgen_paramSize = pow( 72, 1/6. ) * pow( _maxElementVolume, 1/3. );
288 double Netgen_point[3];
289 int Netgen_triangle[3];
290 int Netgen_tetrahedron[4];
294 Ng_Mesh * Netgen_mesh = Ng_NewMesh();
296 // set nodes and remember thier netgen IDs
297 bool isDegen = false, hasDegen = !degenShapeIdToPtrNgId.empty();
298 TNodeToIDMap::iterator n_id = nodeToNetgenID.begin();
299 for ( ; n_id != nodeToNetgenID.end(); ++n_id )
301 const SMDS_MeshNode* node = n_id->first;
303 // ignore nodes on degenerated edge
305 int shapeId = node->GetPosition()->GetShapeId();
306 shId_ngId = degenShapeIdToPtrNgId.find( shapeId );
307 isDegen = ( shId_ngId != degenShapeIdToPtrNgId.end() );
308 if ( isDegen && *(shId_ngId->second) != invalid_ID ) {
309 n_id->second = *(shId_ngId->second);
313 Netgen_point [ 0 ] = node->X();
314 Netgen_point [ 1 ] = node->Y();
315 Netgen_point [ 2 ] = node->Z();
316 Ng_AddPoint(Netgen_mesh, Netgen_point);
317 n_id->second = ++Netgen_NbOfNodes; // set netgen ID
319 if ( isDegen ) // all nodes on a degen edge get one netgen ID
320 *(shId_ngId->second) = n_id->second;
324 list< const SMDS_MeshElement* >::iterator tria = triangles.begin();
325 list< bool >::iterator reverse = isReversed.begin();
326 for ( ; tria != triangles.end(); ++tria, ++reverse )
329 SMDS_ElemIteratorPtr triangleNodesIt = (*tria)->nodesIterator();
330 while ( triangleNodesIt->more() ) {
331 const SMDS_MeshNode * node =
332 static_cast<const SMDS_MeshNode *>(triangleNodesIt->next());
333 if(myTool->IsMedium(node))
335 Netgen_triangle[ *reverse ? 2 - i : i ] = nodeToNetgenID[ node ];
339 // ignore degenerated triangles, they have 2 or 3 same ids
340 (Netgen_triangle[0] != Netgen_triangle[1] &&
341 Netgen_triangle[0] != Netgen_triangle[2] &&
342 Netgen_triangle[2] != Netgen_triangle[1] ))
344 Ng_AddSurfaceElement(Netgen_mesh, NG_TRIG, Netgen_triangle);
348 // -------------------------
349 // Generate the volume mesh
350 // -------------------------
352 Ng_Meshing_Parameters Netgen_param;
354 Netgen_param.secondorder = Netgen_param2ndOrder;
355 Netgen_param.fineness = Netgen_paramFine;
356 Netgen_param.maxh = Netgen_paramSize;
361 #if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
364 status = Ng_GenerateVolumeMesh(Netgen_mesh, &Netgen_param);
366 catch (Standard_Failure& exc) {
367 error(COMPERR_OCC_EXCEPTION, exc.GetMessageString());
368 status = NG_VOLUME_FAILURE;
371 error("Exception in Ng_GenerateVolumeMesh()");
372 status = NG_VOLUME_FAILURE;
374 if ( GetComputeError()->IsOK() ) {
376 case NG_SURFACE_INPUT_ERROR:error( status, "NG_SURFACE_INPUT_ERROR");
377 case NG_VOLUME_FAILURE: error( status, "NG_VOLUME_FAILURE");
378 case NG_STL_INPUT_ERROR: error( status, "NG_STL_INPUT_ERROR");
379 case NG_SURFACE_FAILURE: error( status, "NG_SURFACE_FAILURE");
380 case NG_FILE_NOT_FOUND: error( status, "NG_FILE_NOT_FOUND");
384 int Netgen_NbOfNodesNew = Ng_GetNP(Netgen_mesh);
386 int Netgen_NbOfTetra = Ng_GetNE(Netgen_mesh);
388 MESSAGE("End of Volume Mesh Generation. status=" << status <<
389 ", nb new nodes: " << Netgen_NbOfNodesNew - Netgen_NbOfNodes <<
390 ", nb tetra: " << Netgen_NbOfTetra);
392 // -------------------------------------------------------------------
393 // Feed back the SMESHDS with the generated Nodes and Volume Elements
394 // -------------------------------------------------------------------
396 bool isOK = ( /*status == NG_OK &&*/ Netgen_NbOfTetra > 0 );// get whatever built
399 // vector of nodes in which node index == netgen ID
400 vector< const SMDS_MeshNode* > nodeVec ( Netgen_NbOfNodesNew + 1 );
401 // insert old nodes into nodeVec
402 for ( n_id = nodeToNetgenID.begin(); n_id != nodeToNetgenID.end(); ++n_id ) {
403 nodeVec.at( n_id->second ) = n_id->first;
405 // create and insert new nodes into nodeVec
406 int nodeIndex = Netgen_NbOfNodes + 1;
407 int shapeID = meshDS->ShapeToIndex( aShape );
408 for ( ; nodeIndex <= Netgen_NbOfNodesNew; ++nodeIndex )
410 Ng_GetPoint( Netgen_mesh, nodeIndex, Netgen_point );
411 SMDS_MeshNode * node = meshDS->AddNode(Netgen_point[0],
414 meshDS->SetNodeInVolume(node, shapeID);
415 nodeVec.at(nodeIndex) = node;
418 // create tetrahedrons
419 for ( int elemIndex = 1; elemIndex <= Netgen_NbOfTetra; ++elemIndex )
421 Ng_GetVolumeElement(Netgen_mesh, elemIndex, Netgen_tetrahedron);
422 SMDS_MeshVolume * elt = myTool->AddVolume (nodeVec.at( Netgen_tetrahedron[0] ),
423 nodeVec.at( Netgen_tetrahedron[1] ),
424 nodeVec.at( Netgen_tetrahedron[2] ),
425 nodeVec.at( Netgen_tetrahedron[3] ));
426 meshDS->SetMeshElementOnShape(elt, shapeID );
430 Ng_DeleteMesh(Netgen_mesh);
433 NETGENPlugin_Mesher::RemoveTmpFiles();
435 return (status == NG_OK);
438 bool NETGENPlugin_NETGEN_3D::Compute(SMESH_Mesh& aMesh,
439 SMESH_MesherHelper* aHelper)
441 MESSAGE("NETGENPlugin_NETGEN_3D::Compute with maxElmentsize = " << _maxElementVolume);
442 const int invalid_ID = -1;
443 bool _quadraticMesh = false;
444 typedef map< const SMDS_MeshNode*, int, TIDCompare > TNodeToIDMap;
445 TNodeToIDMap nodeToNetgenID;
446 list< const SMDS_MeshElement* > triangles;
447 SMESHDS_Mesh* MeshDS = aHelper->GetMeshDS();
449 SMESH_MesherHelper::MType MeshType = aHelper->IsQuadraticMesh();
451 if(MeshType == SMESH_MesherHelper::COMP)
452 return error( COMPERR_BAD_INPUT_MESH,
453 SMESH_Comment("Mesh with linear and quadratic elements given."));
454 else if (MeshType == SMESH_MesherHelper::QUADRATIC)
455 _quadraticMesh = true;
457 StdMeshers_QuadToTriaAdaptor Adaptor;
458 Adaptor.Compute(aMesh);
460 SMDS_FaceIteratorPtr fIt = MeshDS->facesIterator();
461 TIDSortedElemSet sortedFaces; // 0020279: control the "random" use when using mesh algorithms
462 while( fIt->more()) sortedFaces.insert( fIt->next() );
464 TIDSortedElemSet::iterator itFace = sortedFaces.begin(), fEnd = sortedFaces.end();
465 for ( ; itFace != fEnd; ++itFace ) {
467 const SMDS_MeshElement* elem = *itFace;
469 return error( COMPERR_BAD_INPUT_MESH, "Null element encounters");
470 bool isTraingle = ( elem->NbNodes()==3 || (_quadraticMesh && elem->NbNodes()==6 ));
472 //return error( COMPERR_BAD_INPUT_MESH,
473 // SMESH_Comment("Not triangle element ")<<elem->GetID());
475 const list<const SMDS_FaceOfNodes*>* faces = Adaptor.GetTriangles(elem);
477 return error( COMPERR_BAD_INPUT_MESH,
478 SMESH_Comment("Not triangles in adaptor for element ")<<elem->GetID());
480 list<const SMDS_FaceOfNodes*>::const_iterator itf = faces->begin();
481 for(; itf!=faces->end(); itf++ ) {
482 triangles.push_back( (*itf) );
483 // put triange's nodes to nodeToNetgenID map
484 SMDS_ElemIteratorPtr triangleNodesIt = (*itf)->nodesIterator();
485 while ( triangleNodesIt->more() ) {
486 const SMDS_MeshNode * node =
487 static_cast<const SMDS_MeshNode *>(triangleNodesIt->next());
488 if(aHelper->IsMedium(node))
490 nodeToNetgenID.insert( make_pair( node, invalid_ID ));
496 triangles.push_back( elem );
497 // put elem nodes to nodeToNetgenID map
498 SMDS_ElemIteratorPtr triangleNodesIt = elem->nodesIterator();
499 while ( triangleNodesIt->more() ) {
500 const SMDS_MeshNode * node =
501 static_cast<const SMDS_MeshNode *>(triangleNodesIt->next());
502 if(aHelper->IsMedium(node))
504 nodeToNetgenID.insert( make_pair( node, invalid_ID ));
509 // ---------------------------------
510 // Feed the Netgen with surface mesh
511 // ---------------------------------
513 int Netgen_NbOfNodes = 0;
514 int Netgen_param2ndOrder = 0;
515 double Netgen_paramFine = 1.;
516 double Netgen_paramSize = pow( 72, 1/6. ) * pow( _maxElementVolume, 1/3. );
518 double Netgen_point[3];
519 int Netgen_triangle[3];
520 int Netgen_tetrahedron[4];
524 Ng_Mesh * Netgen_mesh = Ng_NewMesh();
526 // set nodes and remember thier netgen IDs
528 TNodeToIDMap::iterator n_id = nodeToNetgenID.begin();
529 for ( ; n_id != nodeToNetgenID.end(); ++n_id )
531 const SMDS_MeshNode* node = n_id->first;
533 Netgen_point [ 0 ] = node->X();
534 Netgen_point [ 1 ] = node->Y();
535 Netgen_point [ 2 ] = node->Z();
536 Ng_AddPoint(Netgen_mesh, Netgen_point);
537 n_id->second = ++Netgen_NbOfNodes; // set netgen ID
542 list< const SMDS_MeshElement* >::iterator tria = triangles.begin();
543 for ( ; tria != triangles.end(); ++tria)
546 SMDS_ElemIteratorPtr triangleNodesIt = (*tria)->nodesIterator();
547 while ( triangleNodesIt->more() ) {
548 const SMDS_MeshNode * node =
549 static_cast<const SMDS_MeshNode *>(triangleNodesIt->next());
550 if(aHelper->IsMedium(node))
552 Netgen_triangle[ i ] = nodeToNetgenID[ node ];
556 Ng_AddSurfaceElement(Netgen_mesh, NG_TRIG, Netgen_triangle);
559 // -------------------------
560 // Generate the volume mesh
561 // -------------------------
563 Ng_Meshing_Parameters Netgen_param;
565 Netgen_param.secondorder = Netgen_param2ndOrder;
566 Netgen_param.fineness = Netgen_paramFine;
567 Netgen_param.maxh = Netgen_paramSize;
572 #if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
575 status = Ng_GenerateVolumeMesh(Netgen_mesh, &Netgen_param);
577 catch (Standard_Failure& exc) {
578 error(COMPERR_OCC_EXCEPTION, exc.GetMessageString());
579 status = NG_VOLUME_FAILURE;
582 error("Bad mesh input!!!");
583 status = NG_VOLUME_FAILURE;
585 if ( GetComputeError()->IsOK() ) {
586 error( status, "Bad mesh input!!!");
589 int Netgen_NbOfNodesNew = Ng_GetNP(Netgen_mesh);
591 int Netgen_NbOfTetra = Ng_GetNE(Netgen_mesh);
593 MESSAGE("End of Volume Mesh Generation. status=" << status <<
594 ", nb new nodes: " << Netgen_NbOfNodesNew - Netgen_NbOfNodes <<
595 ", nb tetra: " << Netgen_NbOfTetra);
597 // -------------------------------------------------------------------
598 // Feed back the SMESHDS with the generated Nodes and Volume Elements
599 // -------------------------------------------------------------------
601 bool isOK = ( Netgen_NbOfTetra > 0 );// get whatever built
604 // vector of nodes in which node index == netgen ID
605 vector< const SMDS_MeshNode* > nodeVec ( Netgen_NbOfNodesNew + 1 );
606 // insert old nodes into nodeVec
607 for ( n_id = nodeToNetgenID.begin(); n_id != nodeToNetgenID.end(); ++n_id ) {
608 nodeVec.at( n_id->second ) = n_id->first;
610 // create and insert new nodes into nodeVec
611 int nodeIndex = Netgen_NbOfNodes + 1;
613 for ( ; nodeIndex <= Netgen_NbOfNodesNew; ++nodeIndex )
615 Ng_GetPoint( Netgen_mesh, nodeIndex, Netgen_point );
616 SMDS_MeshNode * node = aHelper->AddNode(Netgen_point[0],
619 nodeVec.at(nodeIndex) = node;
622 // create tetrahedrons
623 for ( int elemIndex = 1; elemIndex <= Netgen_NbOfTetra; ++elemIndex )
625 Ng_GetVolumeElement(Netgen_mesh, elemIndex, Netgen_tetrahedron);
626 aHelper->AddVolume (nodeVec.at( Netgen_tetrahedron[0] ),
627 nodeVec.at( Netgen_tetrahedron[1] ),
628 nodeVec.at( Netgen_tetrahedron[2] ),
629 nodeVec.at( Netgen_tetrahedron[3] ));
633 Ng_DeleteMesh(Netgen_mesh);
636 NETGENPlugin_Mesher::RemoveTmpFiles();
638 return (status == NG_OK);