src/NETGENPlugin/NETGENPlugin_NETGEN_2D_ONLY.cxx

   1 // Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 // File      : NETGENPlugin_NETGEN_2D_ONLY.cxx
  21 // Author    : Edward AGAPOV (OCC)
  22 // Project   : SALOME
  23 //
  24 #include "NETGENPlugin_NETGEN_2D_ONLY.hxx"
  25
  26 #include "NETGENPlugin_Mesher.hxx"
  27 #include "NETGENPlugin_Hypothesis_2D.hxx"
  28
  29 #include <SMDS_MeshElement.hxx>
  30 #include <SMDS_MeshNode.hxx>
  31 #include <SMESHDS_Mesh.hxx>
  32 #include <SMESH_Comment.hxx>
  33 #include <SMESH_Gen.hxx>
  34 #include <SMESH_Mesh.hxx>
  35 #include <SMESH_MesherHelper.hxx>
  36 #include <SMESH_subMesh.hxx>
  37 #include <StdMeshers_FaceSide.hxx>
  38 #include <StdMeshers_LengthFromEdges.hxx>
  39 #include <StdMeshers_MaxElementArea.hxx>
  40 #include <StdMeshers_QuadranglePreference.hxx>
  41 #include <StdMeshers_ViscousLayers2D.hxx>
  42
  43 #include <Precision.hxx>
  44 #include <Standard_ErrorHandler.hxx>
  45 #include <Standard_Failure.hxx>
  46
  47 #include <utilities.h>
  48
  49 #include <list>
  50 #include <vector>
  51 #include <limits>
  52
  53 /*
  54   Netgen include files
  55 */
  56 namespace nglib {
  57 #include <nglib.h>
  58 }
  59 #ifndef OCCGEOMETRY
  60 #define OCCGEOMETRY
  61 #endif
  62 #include <occgeom.hpp>
  63 #include <meshing.hpp>
  64 //#include <meshtype.hpp>
  65 namespace netgen {
  66 #ifdef NETGEN_V5
  67   extern int OCCGenerateMesh (OCCGeometry&, Mesh*&, MeshingParameters&, int, int);
  68 #else
  69   extern int OCCGenerateMesh (OCCGeometry&, Mesh*&, int, int, char*);
  70 #endif
  71   extern MeshingParameters mparam;
  72 }
  73
  74 using namespace std;
  75 using namespace netgen;
  76 using namespace nglib;
  77
  78 //#define DUMP_SEGMENTS
  79
  80 //=============================================================================
  81 /*!
  82  *
  83  */
  84 //=============================================================================
  85
  86 NETGENPlugin_NETGEN_2D_ONLY::NETGENPlugin_NETGEN_2D_ONLY(int hypId, int studyId,
  87                                                          SMESH_Gen* gen)
  88   : SMESH_2D_Algo(hypId, studyId, gen)
  89 {
  90   MESSAGE("NETGENPlugin_NETGEN_2D_ONLY::NETGENPlugin_NETGEN_2D_ONLY");
  91   _name = "NETGEN_2D_ONLY";
  92
  93   _shapeType = (1 << TopAbs_FACE);// 1 bit /shape type
  94
  95   _compatibleHypothesis.push_back("MaxElementArea");
  96   _compatibleHypothesis.push_back("LengthFromEdges");
  97   _compatibleHypothesis.push_back("QuadranglePreference");
  98   _compatibleHypothesis.push_back("NETGEN_Parameters_2D");
  99   _compatibleHypothesis.push_back("ViscousLayers2D");
 100
 101   _hypMaxElementArea = 0;
 102   _hypLengthFromEdges = 0;
 103   _hypQuadranglePreference = 0;
 104   _hypParameters = 0;
 105 }
 106
 107 //=============================================================================
 108 /*!
 109  *
 110  */
 111 //=============================================================================
 112
 113 NETGENPlugin_NETGEN_2D_ONLY::~NETGENPlugin_NETGEN_2D_ONLY()
 114 {
 115   MESSAGE("NETGENPlugin_NETGEN_2D_ONLY::~NETGENPlugin_NETGEN_2D_ONLY");
 116 }
 117
 118 //=============================================================================
 119 /*!
 120  *
 121  */
 122 //=============================================================================
 123
 124 bool NETGENPlugin_NETGEN_2D_ONLY::CheckHypothesis (SMESH_Mesh&         aMesh,
 125                                                    const TopoDS_Shape& aShape,
 126                                                    Hypothesis_Status&  aStatus)
 127 {
 128   _hypMaxElementArea = 0;
 129   _hypLengthFromEdges = 0;
 130   _hypQuadranglePreference = 0;
 131   _progressByTic = -1;
 132
 133   const list<const SMESHDS_Hypothesis*>& hyps = GetUsedHypothesis(aMesh, aShape, false);
 134
 135   if (hyps.empty())
 136   {
 137     aStatus = HYP_OK; //SMESH_Hypothesis::HYP_MISSING;
 138     return true;  // (PAL13464) can work with no hypothesis, LengthFromEdges is default one
 139   }
 140
 141   aStatus = HYP_MISSING;
 142
 143   list<const SMESHDS_Hypothesis*>::const_iterator ith;
 144   for (ith = hyps.begin(); ith != hyps.end(); ++ith )
 145   {
 146     const SMESHDS_Hypothesis* hyp = (*ith);
 147
 148     string hypName = hyp->GetName();
 149
 150     if      ( hypName == "MaxElementArea")
 151       _hypMaxElementArea = static_cast<const StdMeshers_MaxElementArea*> (hyp);
 152     else if ( hypName == "LengthFromEdges" )
 153       _hypLengthFromEdges = static_cast<const StdMeshers_LengthFromEdges*> (hyp);
 154     else if ( hypName == "QuadranglePreference" )
 155       _hypQuadranglePreference = static_cast<const StdMeshers_QuadranglePreference*>(hyp);
 156     else if ( hypName == "NETGEN_Parameters_2D" )
 157       _hypParameters = static_cast<const NETGENPlugin_Hypothesis_2D*>(hyp);
 158     else if ( hypName == StdMeshers_ViscousLayers2D::GetHypType() )
 159       continue;
 160     else {
 161       aStatus = HYP_INCOMPATIBLE;
 162       return false;
 163     }
 164   }
 165
 166   int nbHyps = bool(_hypMaxElementArea) + bool(_hypLengthFromEdges) + bool(_hypParameters );
 167   if ( nbHyps > 1 )
 168     aStatus = HYP_CONCURENT;
 169   else
 170     aStatus = HYP_OK;
 171
 172   return ( aStatus == HYP_OK );
 173 }
 174
 175 //=============================================================================
 176 /*!
 177  *Here we are going to use the NETGEN mesher
 178  */
 179 //=============================================================================
 180
 181 bool NETGENPlugin_NETGEN_2D_ONLY::Compute(SMESH_Mesh&         aMesh,
 182                                           const TopoDS_Shape& aShape)
 183 {
 184   netgen::multithread.terminate = 0;
 185   netgen::multithread.task = "Surface meshing";
 186
 187   SMESHDS_Mesh* meshDS = aMesh.GetMeshDS();
 188   int faceID = meshDS->ShapeToIndex( aShape );
 189
 190   SMESH_MesherHelper helper(aMesh);
 191   _quadraticMesh = helper.IsQuadraticSubMesh(aShape);
 192   helper.SetElementsOnShape( true );
 193   const bool ignoreMediumNodes = _quadraticMesh;
 194
 195   // build viscous layers if required
 196   TopoDS_Face F = TopoDS::Face( aShape/*.Oriented( TopAbs_FORWARD )*/);
 197   if ( F.Orientation() != TopAbs_FORWARD &&
 198        F.Orientation() != TopAbs_REVERSED )
 199     F.Orientation( TopAbs_FORWARD ); // avoid pb with TopAbs_INTERNAL
 200   SMESH_ProxyMesh::Ptr proxyMesh = StdMeshers_ViscousLayers2D::Compute( aMesh, F );
 201   if ( !proxyMesh )
 202     return false;
 203
 204   // ------------------------
 205   // get all edges of a face
 206   // ------------------------
 207   TError problem;
 208   TSideVector wires =
 209     StdMeshers_FaceSide::GetFaceWires( F, aMesh, ignoreMediumNodes, problem, proxyMesh );
 210   if ( problem && !problem->IsOK() )
 211     return error( problem );
 212   int nbWires = wires.size();
 213   if ( nbWires == 0 )
 214     return error( "Problem in StdMeshers_FaceSide::GetFaceWires()");
 215   if ( wires[0]->NbSegments() < 3 ) // ex: a circle with 2 segments
 216     return error(COMPERR_BAD_INPUT_MESH,
 217                  SMESH_Comment("Too few segments: ")<<wires[0]->NbSegments());
 218
 219   // --------------------
 220   // compute edge length
 221   // --------------------
 222
 223   NETGENPlugin_Mesher aMesher( &aMesh, aShape, /*isVolume=*/false);
 224   netgen::OCCGeometry occgeo;
 225   aMesher.PrepareOCCgeometry( occgeo, F, aMesh );
 226   occgeo.fmap.Clear(); // face can be reversed, which is wrong in this case (issue 19978)
 227   occgeo.fmap.Add( F );
 228
 229   if ( _hypParameters )
 230   {
 231     aMesher.SetParameters(_hypParameters);
 232   }
 233   else
 234   {
 235     double edgeLength = 0;
 236     if (_hypLengthFromEdges /*|| (!_hypLengthFromEdges && !_hypMaxElementArea)*/)
 237     {
 238       int nbSegments = 0;
 239       for ( int iW = 0; iW < nbWires; ++iW )
 240       {
 241         edgeLength += wires[ iW ]->Length();
 242         nbSegments += wires[ iW ]->NbSegments();
 243       }
 244       if ( nbSegments )
 245         edgeLength /= nbSegments;
 246     }
 247     if ( _hypMaxElementArea )
 248     {
 249       double maxArea = _hypMaxElementArea->GetMaxArea();
 250       edgeLength = sqrt(2. * maxArea/sqrt(3.0));
 251     }
 252     if ( edgeLength < DBL_MIN )
 253       edgeLength = occgeo.GetBoundingBox().Diam();
 254
 255     netgen::mparam.maxh = edgeLength;
 256     netgen::mparam.minh = aMesher.GetDefaultMinSize( aShape, netgen::mparam.maxh );
 257     netgen::mparam.quad = _hypQuadranglePreference ? 1 : 0;
 258     netgen::mparam.grading = 0.5; // coarse mesh by default
 259   }
 260   occgeo.face_maxh = netgen::mparam.maxh;
 261
 262   // -------------------------
 263   // Make input netgen mesh
 264   // -------------------------
 265
 266   NETGENPlugin_NetgenLibWrapper ngLib;
 267   netgen::Mesh * ngMesh = (netgen::Mesh*) ngLib._ngMesh;
 268
 269 #ifndef NETGEN_V5
 270   char *optstr = 0;
 271 #endif
 272   int startWith = MESHCONST_ANALYSE;
 273   int endWith   = MESHCONST_ANALYSE;
 274   int err = 1;
 275
 276   if ( !_hypLengthFromEdges && !_hypMaxElementArea )
 277   {
 278 #ifdef NETGEN_V5
 279     err = netgen::OCCGenerateMesh(occgeo, ngMesh, netgen::mparam, startWith, endWith);
 280 #else
 281     err = netgen::OCCGenerateMesh(occgeo, ngMesh, startWith, endWith, optstr);
 282 #endif
 283     ngLib._ngMesh = 0;
 284     ngLib.setMesh(( nglib::Ng_Mesh*) ngMesh );
 285   }
 286   else
 287   {
 288     Box<3> bb = occgeo.GetBoundingBox();
 289     bb.Increase (bb.Diam()/10);
 290     ngMesh->SetLocalH (bb.PMin(), bb.PMax(), netgen::mparam.grading);
 291     ngMesh->SetGlobalH (netgen::mparam.maxh);
 292   }
 293
 294   vector< const SMDS_MeshNode* > nodeVec;
 295   problem = aMesher.AddSegmentsToMesh( *ngMesh, occgeo, wires, helper, nodeVec );
 296   if ( problem && !problem->IsOK() )
 297     return error( problem );
 298
 299   // limit element size near existing segments
 300   TopTools_IndexedMapOfShape edgeMap, faceMap;
 301   TopExp::MapShapes( aMesh.GetShapeToMesh(), TopAbs_EDGE, edgeMap );
 302   for ( int iE = 1; iE <= edgeMap.Extent(); ++iE )
 303   {
 304     const TopoDS_Shape& edge = edgeMap( iE );
 305     if ( SMESH_Algo::isDegenerated( TopoDS::Edge( edge )) ||
 306          helper.IsSubShape( edge, aShape ))
 307       continue;
 308     SMESHDS_SubMesh* smDS = aMesh.GetMeshDS()->MeshElements( edge );
 309     if ( !smDS ) continue;
 310     SMDS_ElemIteratorPtr segIt = smDS->GetElements();
 311     while ( segIt->more() )
 312     {
 313       const SMDS_MeshElement* seg = segIt->next();
 314       SMESH_TNodeXYZ n1 = seg->GetNode(0);
 315       SMESH_TNodeXYZ n2 = seg->GetNode(1);
 316       gp_XYZ p = 0.5 * ( n1 + n2 );
 317       netgen::Point3d pi(p.X(), p.Y(), p.Z());
 318       ngMesh->RestrictLocalH( pi, Max(( n1 - n2 ).Modulus(), netgen::mparam.minh ));
 319     }
 320   }
 321
 322   // -------------------------
 323   // Generate surface mesh
 324   // -------------------------
 325
 326   startWith = MESHCONST_MESHSURFACE;
 327   endWith   = MESHCONST_OPTSURFACE;
 328
 329   try {
 330     OCC_CATCH_SIGNALS;
 331
 332 #ifdef NETGEN_V5
 333     err = netgen::OCCGenerateMesh(occgeo, ngMesh, netgen::mparam, startWith, endWith);
 334 #else
 335     err = netgen::OCCGenerateMesh(occgeo, ngMesh, startWith, endWith, optstr);
 336 #endif
 337     if(netgen::multithread.terminate)
 338       return false;
 339     if ( err )
 340       error(SMESH_Comment("Error in netgen::OCCGenerateMesh() at ") << netgen::multithread.task);
 341   }
 342   catch (Standard_Failure& ex)
 343   {
 344     SMESH_Comment str("Exception in  netgen::OCCGenerateMesh()");
 345     str << " at " << netgen::multithread.task
 346         << ": " << ex.DynamicType()->Name();
 347     if ( ex.GetMessageString() && strlen( ex.GetMessageString() ))
 348       str << ": " << ex.GetMessageString();
 349     error(str);
 350   }
 351   catch (...) {
 352     SMESH_Comment str("Exception in  netgen::OCCGenerateMesh()");
 353     str << " at " << netgen::multithread.task;
 354     error(str);
 355   }
 356
 357   // ----------------------------------------------------
 358   // Fill the SMESHDS with the generated nodes and faces
 359   // ----------------------------------------------------
 360
 361   int nbNodes = ngMesh->GetNP();
 362   int nbFaces = ngMesh->GetNSE();
 363
 364   int nbInputNodes = nodeVec.size()-1;
 365   nodeVec.resize( nbNodes+1, 0 );
 366
 367   // add nodes
 368   for ( int ngID = nbInputNodes + 1; ngID <= nbNodes; ++ngID )
 369   {
 370     const MeshPoint& ngPoint = ngMesh->Point( ngID );
 371     SMDS_MeshNode * node = meshDS->AddNode(ngPoint(0), ngPoint(1), ngPoint(2));
 372     nodeVec[ ngID ] = node;
 373   }
 374
 375   // create faces
 376   const bool reverse = false; //( aShape.Orientation() == TopAbs_REVERSED );
 377   int i,j;
 378   for ( i = 1; i <= nbFaces ; ++i )
 379   {
 380     const Element2d& elem = ngMesh->SurfaceElement(i);
 381     vector<const SMDS_MeshNode*> nodes( elem.GetNP() );
 382     for (j=1; j <= elem.GetNP(); ++j)
 383     {
 384       int pind = elem.PNum(j);
 385       if ( pind < 1 )
 386         break;
 387       const SMDS_MeshNode* node = nodeVec[ pind ];
 388       if ( reverse )
 389         nodes[ nodes.size()-j ] = node;
 390       else
 391         nodes[ j-1 ] = node;
 392       if ( node->GetPosition()->GetTypeOfPosition() == SMDS_TOP_3DSPACE )
 393       {
 394         const PointGeomInfo& pgi = elem.GeomInfoPi(j);
 395         meshDS->SetNodeOnFace((SMDS_MeshNode*)node, faceID, pgi.u, pgi.v);
 396       }
 397     }
 398     if ( j > elem.GetNP() )
 399     {
 400       SMDS_MeshFace* face = 0;
 401       if ( elem.GetType() == TRIG )
 402         face = helper.AddFace(nodes[0],nodes[1],nodes[2]);
 403       else
 404         face = helper.AddFace(nodes[0],nodes[1],nodes[2],nodes[3]);
 405     }
 406   }
 407
 408   ngLib._isComputeOk = !err;
 409   return !err;
 410 }
 411
 412 void NETGENPlugin_NETGEN_2D_ONLY::CancelCompute()
 413 {
 414   SMESH_Algo::CancelCompute();
 415   netgen::multithread.terminate = 1;
 416 }
 417
 418 //================================================================================
 419 /*!
 420  * \brief Return progress of Compute() [0.,1]
 421  */
 422 //================================================================================
 423
 424 double NETGENPlugin_NETGEN_2D_ONLY::GetProgress() const
 425 {
 426   const char* task1 = "Surface meshing";
 427   //const char* task2 = "Optimizing surface";
 428   double& progress = const_cast<NETGENPlugin_NETGEN_2D_ONLY*>( this )->_progress;
 429   if ( _progressByTic < 0. &&
 430        strncmp( netgen::multithread.task, task1, 3 ) == 0 )
 431   {
 432     progress = Min( 0.25, SMESH_Algo::GetProgressByTic() ); // [0, 0.25]
 433   }
 434   else //if ( strncmp( netgen::multithread.task, task2, 3 ) == 0)
 435   {
 436     if ( _progressByTic < 0 )
 437     {
 438       NETGENPlugin_NETGEN_2D_ONLY* me = (NETGENPlugin_NETGEN_2D_ONLY*) this;
 439       me->_progressByTic = 0.25 / _progressTic;
 440     }
 441     const_cast<NETGENPlugin_NETGEN_2D_ONLY*>( this )->_progressTic++;
 442     progress = Max( progress, _progressByTic * _progressTic );
 443   }
 444   //cout << netgen::multithread.task << " " << _progressTic << endl;
 445   return Min( progress, 0.99 );
 446 }
 447
 448 //=============================================================================
 449 /*!
 450  *
 451  */
 452 //=============================================================================
 453
 454 bool NETGENPlugin_NETGEN_2D_ONLY::Evaluate(SMESH_Mesh& aMesh,
 455                                            const TopoDS_Shape& aShape,
 456                                            MapShapeNbElems& aResMap)
 457 {
 458   TopoDS_Face F = TopoDS::Face(aShape);
 459   if(F.IsNull())
 460     return false;
 461
 462   // collect info from edges
 463   int nb0d = 0, nb1d = 0;
 464   bool IsQuadratic = false;
 465   bool IsFirst = true;
 466   double fullLen = 0.0;
 467   TopTools_MapOfShape tmpMap;
 468   for (TopExp_Explorer exp(F, TopAbs_EDGE); exp.More(); exp.Next()) {
 469     TopoDS_Edge E = TopoDS::Edge(exp.Current());
 470     if( tmpMap.Contains(E) )
 471       continue;
 472     tmpMap.Add(E);
 473     SMESH_subMesh *aSubMesh = aMesh.GetSubMesh(exp.Current());
 474     MapShapeNbElemsItr anIt = aResMap.find(aSubMesh);
 475     if( anIt==aResMap.end() ) {
 476       SMESH_subMesh *sm = aMesh.GetSubMesh(F);
 477       SMESH_ComputeErrorPtr& smError = sm->GetComputeError();
 478       smError.reset( new SMESH_ComputeError(COMPERR_ALGO_FAILED,"Submesh can not be evaluated",this));
 479       return false;
 480     }
 481     std::vector<int> aVec = (*anIt).second;
 482     nb0d += aVec[SMDSEntity_Node];
 483     nb1d += Max(aVec[SMDSEntity_Edge],aVec[SMDSEntity_Quad_Edge]);
 484     double aLen = SMESH_Algo::EdgeLength(E);
 485     fullLen += aLen;
 486     if(IsFirst) {
 487       IsQuadratic = (aVec[SMDSEntity_Quad_Edge] > aVec[SMDSEntity_Edge]);
 488       IsFirst = false;
 489     }
 490   }
 491   tmpMap.Clear();
 492
 493   // compute edge length
 494   double ELen = 0;
 495   if (_hypLengthFromEdges || !_hypLengthFromEdges && !_hypMaxElementArea) {
 496     if ( nb1d > 0 )
 497       ELen = fullLen / nb1d;
 498   }
 499   if ( _hypMaxElementArea ) {
 500     double maxArea = _hypMaxElementArea->GetMaxArea();
 501     ELen = sqrt(2. * maxArea/sqrt(3.0));
 502   }
 503   GProp_GProps G;
 504   BRepGProp::SurfaceProperties(F,G);
 505   double anArea = G.Mass();
 506
 507   const int hugeNb = numeric_limits<int>::max()/10;
 508   if ( anArea / hugeNb > ELen*ELen )
 509   {
 510     SMESH_subMesh *sm = aMesh.GetSubMesh(F);
 511     SMESH_ComputeErrorPtr& smError = sm->GetComputeError();
 512     smError.reset( new SMESH_ComputeError(COMPERR_ALGO_FAILED,"Submesh can not be evaluated.\nToo small element length",this));
 513     return false;
 514   }
 515   int nbFaces = (int) ( anArea / ( ELen*ELen*sqrt(3.) / 4 ) );
 516   int nbNodes = (int) ( ( nbFaces*3 - (nb1d-1)*2 ) / 6 + 1 );
 517   std::vector<int> aVec(SMDSEntity_Last);
 518   for(int i=SMDSEntity_Node; i<SMDSEntity_Last; i++) aVec[i]=0;
 519   if( IsQuadratic ) {
 520     aVec[SMDSEntity_Node] = nbNodes;
 521     aVec[SMDSEntity_Quad_Triangle] = nbFaces;
 522   }
 523   else {
 524     aVec[SMDSEntity_Node] = nbNodes;
 525     aVec[SMDSEntity_Triangle] = nbFaces;
 526   }
 527   SMESH_subMesh *sm = aMesh.GetSubMesh(F);
 528   aResMap.insert(std::make_pair(sm,aVec));
 529
 530   return true;
 531 }