Salome HOME
Copyright update: 2016
[plugins/netgenplugin.git] / src / NETGENPlugin / NETGENPlugin_NETGEN_2D_ONLY.cxx
1 // Copyright (C) 2007-2016  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, or (at your option) any later version.
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   extern void OCCSetLocalMeshSize(OCCGeometry & geom, Mesh & mesh);
73 }
74
75 using namespace std;
76 using namespace netgen;
77 using namespace nglib;
78
79 //=============================================================================
80 /*!
81  *  
82  */
83 //=============================================================================
84
85 NETGENPlugin_NETGEN_2D_ONLY::NETGENPlugin_NETGEN_2D_ONLY(int        hypId,
86                                                          int        studyId,
87                                                          SMESH_Gen* gen)
88   : SMESH_2D_Algo(hypId, studyId, gen)
89 {
90   _name = "NETGEN_2D_ONLY";
91   
92   _shapeType = (1 << TopAbs_FACE);// 1 bit /shape type
93   _onlyUnaryInput = false; // treat all FACEs at once
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   _hypParameters = 0;
132   _progressByTic = -1;
133
134   const list<const SMESHDS_Hypothesis*>& hyps = GetUsedHypothesis(aMesh, aShape, false);
135
136   if (hyps.empty())
137   {
138     aStatus = HYP_OK; //SMESH_Hypothesis::HYP_MISSING;
139     return true;  // (PAL13464) can work with no hypothesis, LengthFromEdges is default one
140   }
141
142   aStatus = HYP_MISSING;
143
144   bool hasVL = false;
145   list<const SMESHDS_Hypothesis*>::const_iterator ith;
146   for (ith = hyps.begin(); ith != hyps.end(); ++ith )
147   {
148     const SMESHDS_Hypothesis* hyp = (*ith);
149
150     string hypName = hyp->GetName();
151
152     if      ( hypName == "MaxElementArea")
153       _hypMaxElementArea = static_cast<const StdMeshers_MaxElementArea*> (hyp);
154     else if ( hypName == "LengthFromEdges" )
155       _hypLengthFromEdges = static_cast<const StdMeshers_LengthFromEdges*> (hyp);
156     else if ( hypName == "QuadranglePreference" )
157       _hypQuadranglePreference = static_cast<const StdMeshers_QuadranglePreference*>(hyp);
158     else if ( hypName == "NETGEN_Parameters_2D" )
159       _hypParameters = static_cast<const NETGENPlugin_Hypothesis_2D*>(hyp);
160     else if ( hypName == StdMeshers_ViscousLayers2D::GetHypType() )
161       hasVL = true;
162     else {
163       aStatus = HYP_INCOMPATIBLE;
164       return false;
165     }
166   }
167
168   int nbHyps = bool(_hypMaxElementArea) + bool(_hypLengthFromEdges) + bool(_hypParameters );
169   if ( nbHyps > 1 )
170     aStatus = HYP_CONCURENT;
171   else if ( hasVL )
172     error( StdMeshers_ViscousLayers2D::CheckHypothesis( aMesh, aShape, aStatus ));
173   else
174     aStatus = HYP_OK;
175
176   if ( aStatus == HYP_OK && _hypParameters && _hypQuadranglePreference )
177   {
178     aStatus = HYP_INCOMPAT_HYPS;
179     return error(SMESH_Comment("\"") << _hypQuadranglePreference->GetName()
180                  << "\" and \"" << _hypParameters->GetName()
181                  << "\" are incompatible hypotheses");
182   }
183
184   return ( aStatus == HYP_OK );
185 }
186
187 namespace
188 {
189   void limitSize( netgen::Mesh* ngMesh,
190                   const double  maxh )
191   {
192     // get bnd box
193     netgen::Point3d pmin, pmax;
194     ngMesh->GetBox( pmin, pmax, 0 );
195     const double dx = pmax.X() - pmin.X();
196     const double dy = pmax.Y() - pmin.Y();
197     const double dz = pmax.Z() - pmin.Z();
198
199     const int nbX = Max( 2, int( dx / maxh * 3 ));
200     const int nbY = Max( 2, int( dy / maxh * 3 ));
201     const int nbZ = Max( 2, int( dz / maxh * 3 ));
202
203     if ( ! & ngMesh->LocalHFunction() )
204       ngMesh->SetLocalH( pmin, pmax, 0.1 );
205
206     netgen::Point3d p;
207     for ( int i = 0; i <= nbX; ++i )
208     {
209       p.X() = pmin.X() +  i * dx / nbX;
210       for ( int j = 0; j <= nbY; ++j )
211       {
212         p.Y() = pmin.Y() +  j * dy / nbY;
213         for ( int k = 0; k <= nbZ; ++k )
214         {
215           p.Z() = pmin.Z() +  k * dz / nbZ;
216           ngMesh->RestrictLocalH( p, maxh );
217         }
218       }
219     }
220   }
221 }
222
223 //=============================================================================
224 /*!
225  *Here we are going to use the NETGEN mesher
226  */
227 //=============================================================================
228
229 bool NETGENPlugin_NETGEN_2D_ONLY::Compute(SMESH_Mesh&         aMesh,
230                                           const TopoDS_Shape& aShape)
231 {
232   netgen::multithread.terminate = 0;
233   //netgen::multithread.task = "Surface meshing";
234
235   SMESHDS_Mesh* meshDS = aMesh.GetMeshDS();
236   SMESH_MesherHelper helper(aMesh);
237   helper.SetElementsOnShape( true );
238
239   NETGENPlugin_NetgenLibWrapper ngLib;
240   ngLib._isComputeOk = false;
241
242   netgen::Mesh   ngMeshNoLocSize;
243   netgen::Mesh * ngMeshes[2] = { (netgen::Mesh*) ngLib._ngMesh,  & ngMeshNoLocSize };
244   netgen::OCCGeometry occgeoComm;
245
246   // min / max sizes are set as follows:
247   // if ( _hypParameters )
248   //    min and max are defined by the user
249   // else if ( _hypLengthFromEdges )
250   //    min = aMesher.GetDefaultMinSize()
251   //    max = average segment len of a FACE
252   // else if ( _hypMaxElementArea )
253   //    min = aMesher.GetDefaultMinSize()
254   //    max = f( _hypMaxElementArea )
255   // else
256   //    min = aMesher.GetDefaultMinSize()
257   //    max = max segment len of a FACE
258
259   NETGENPlugin_Mesher aMesher( &aMesh, aShape, /*isVolume=*/false);
260   aMesher.SetParameters( _hypParameters ); // _hypParameters -> netgen::mparam
261   const bool toOptimize = _hypParameters ? _hypParameters->GetOptimize() : true;
262   if ( _hypMaxElementArea )
263   {
264     netgen::mparam.maxh = sqrt( 2. * _hypMaxElementArea->GetMaxArea() / sqrt(3.0) );
265   }
266   if ( _hypQuadranglePreference )
267     netgen::mparam.quad = true;
268
269   // local size is common for all FACEs in aShape?
270   const bool isCommonLocalSize = ( !_hypLengthFromEdges && !_hypMaxElementArea && netgen::mparam.uselocalh );
271   const bool isDefaultHyp = ( !_hypLengthFromEdges && !_hypMaxElementArea && !_hypParameters );
272
273   if ( isCommonLocalSize ) // compute common local size in ngMeshes[0]
274   {
275     //list< SMESH_subMesh* > meshedSM[4]; --> all sub-shapes are added to occgeoComm
276     aMesher.PrepareOCCgeometry( occgeoComm, aShape, aMesh );//, meshedSM );
277
278     // local size set at MESHCONST_ANALYSE step depends on
279     // minh, face_maxh, grading and curvaturesafety; find minh if not set by the user
280     if ( !_hypParameters || netgen::mparam.minh < DBL_MIN )
281     {
282       if ( !_hypParameters )
283         netgen::mparam.maxh = occgeoComm.GetBoundingBox().Diam() / 3.;
284       netgen::mparam.minh = aMesher.GetDefaultMinSize( aShape, netgen::mparam.maxh );
285     }
286     // set local size depending on curvature and NOT closeness of EDGEs
287     netgen::occparam.resthcloseedgeenable = false;
288     //netgen::occparam.resthcloseedgefac = 1.0 + netgen::mparam.grading;
289     occgeoComm.face_maxh = netgen::mparam.maxh;
290     netgen::OCCSetLocalMeshSize( occgeoComm, *ngMeshes[0] );
291     occgeoComm.emap.Clear();
292     occgeoComm.vmap.Clear();
293
294     // set local size according to size of existing segments
295     const double factor = netgen::occparam.resthcloseedgefac;
296     TopTools_IndexedMapOfShape edgeMap;
297     TopExp::MapShapes( aMesh.GetShapeToMesh(), TopAbs_EDGE, edgeMap );
298     for ( int iE = 1; iE <= edgeMap.Extent(); ++iE )
299     {
300       const TopoDS_Shape& edge = edgeMap( iE );
301       if ( SMESH_Algo::isDegenerated( TopoDS::Edge( edge ))/* ||
302            helper.IsSubShape( edge, aShape )*/)
303         continue;
304       SMESHDS_SubMesh* smDS = meshDS->MeshElements( edge );
305       if ( !smDS ) continue;
306       SMDS_ElemIteratorPtr segIt = smDS->GetElements();
307       while ( segIt->more() )
308       {
309         const SMDS_MeshElement* seg = segIt->next();
310         SMESH_TNodeXYZ n1 = seg->GetNode(0);
311         SMESH_TNodeXYZ n2 = seg->GetNode(1);
312         gp_XYZ p = 0.5 * ( n1 + n2 );
313         netgen::Point3d pi(p.X(), p.Y(), p.Z());
314         ngMeshes[0]->RestrictLocalH( pi, factor * ( n1 - n2 ).Modulus() );
315       }
316     }
317   }
318   netgen::mparam.uselocalh = toOptimize; // restore as it is used at surface optimization
319
320   // ==================
321   // Loop on all FACEs
322   // ==================
323
324   vector< const SMDS_MeshNode* > nodeVec;
325
326   TopExp_Explorer fExp( aShape, TopAbs_FACE );
327   for ( int iF = 0; fExp.More(); fExp.Next(), ++iF )
328   {
329     TopoDS_Face F = TopoDS::Face( fExp.Current() /*.Oriented( TopAbs_FORWARD )*/);
330     int    faceID = meshDS->ShapeToIndex( F );
331     SMESH_ComputeErrorPtr& faceErr = aMesh.GetSubMesh( F )->GetComputeError();
332
333     _quadraticMesh = helper.IsQuadraticSubMesh( F );
334     const bool ignoreMediumNodes = _quadraticMesh;
335
336     // build viscous layers if required
337     if ( F.Orientation() != TopAbs_FORWARD &&
338          F.Orientation() != TopAbs_REVERSED )
339       F.Orientation( TopAbs_FORWARD ); // avoid pb with TopAbs_INTERNAL
340     SMESH_ProxyMesh::Ptr proxyMesh = StdMeshers_ViscousLayers2D::Compute( aMesh, F );
341     if ( !proxyMesh )
342       continue;
343
344     // ------------------------
345     // get all EDGEs of a FACE
346     // ------------------------
347     TSideVector wires =
348       StdMeshers_FaceSide::GetFaceWires( F, aMesh, ignoreMediumNodes, faceErr, proxyMesh );
349     if ( faceErr && !faceErr->IsOK() )
350       continue;
351     int nbWires = wires.size();
352     if ( nbWires == 0 )
353     {
354       faceErr.reset
355         ( new SMESH_ComputeError
356           ( COMPERR_ALGO_FAILED, "Problem in StdMeshers_FaceSide::GetFaceWires()" ));
357       continue;
358     }
359     if ( wires[0]->NbSegments() < 3 ) // ex: a circle with 2 segments
360     {
361       faceErr.reset
362         ( new SMESH_ComputeError
363           ( COMPERR_BAD_INPUT_MESH, SMESH_Comment("Too few segments: ")<<wires[0]->NbSegments()) );
364       continue;
365     }
366
367     // ----------------------
368     // compute maxh of a FACE
369     // ----------------------
370
371     if ( !_hypParameters )
372     {
373       double edgeLength = 0;
374       if (_hypLengthFromEdges )
375       {
376         // compute edgeLength as an average segment length
377         int nbSegments = 0;
378         for ( int iW = 0; iW < nbWires; ++iW )
379         {
380           edgeLength += wires[ iW ]->Length();
381           nbSegments += wires[ iW ]->NbSegments();
382         }
383         if ( nbSegments )
384           edgeLength /= nbSegments;
385         netgen::mparam.maxh = edgeLength;
386       }
387       else if ( isDefaultHyp )
388       {
389         // set edgeLength by a longest segment
390         double maxSeg2 = 0;
391         for ( int iW = 0; iW < nbWires; ++iW )
392         {
393           const UVPtStructVec& points = wires[ iW ]->GetUVPtStruct();
394           if ( points.empty() )
395             return error( COMPERR_BAD_INPUT_MESH );
396           gp_Pnt pPrev = SMESH_TNodeXYZ( points[0].node );
397           for ( size_t i = 1; i < points.size(); ++i )
398           {
399             gp_Pnt p = SMESH_TNodeXYZ( points[i].node );
400             maxSeg2 = Max( maxSeg2, p.SquareDistance( pPrev ));
401             pPrev = p;
402           }
403         }
404         edgeLength = sqrt( maxSeg2 ) * 1.05;
405         netgen::mparam.maxh = edgeLength;
406       }
407       if ( netgen::mparam.maxh < DBL_MIN )
408         netgen::mparam.maxh = occgeoComm.GetBoundingBox().Diam();
409
410       if ( !isCommonLocalSize )
411       {
412         netgen::mparam.minh = aMesher.GetDefaultMinSize( F, netgen::mparam.maxh );
413       }
414     }
415
416     // prepare occgeom
417     netgen::OCCGeometry occgeom;
418     occgeom.shape = F;
419     occgeom.fmap.Add( F );
420     occgeom.CalcBoundingBox();
421     occgeom.facemeshstatus.SetSize(1);
422     occgeom.facemeshstatus = 0;
423     occgeom.face_maxh_modified.SetSize(1);
424     occgeom.face_maxh_modified = 0;
425     occgeom.face_maxh.SetSize(1);
426     occgeom.face_maxh = netgen::mparam.maxh;
427
428     // -------------------------
429     // Fill netgen mesh
430     // -------------------------
431
432     // MESHCONST_ANALYSE step may lead to a failure, so we make an attempt
433     // w/o MESHCONST_ANALYSE at the second loop
434     int err = 0;
435     enum { LOC_SIZE, NO_LOC_SIZE };
436     int iLoop = isCommonLocalSize ? 0 : 1;
437     for ( ; iLoop < 2; iLoop++ )
438     {
439       //bool isMESHCONST_ANALYSE = false;
440       InitComputeError();
441
442       netgen::Mesh * ngMesh = ngMeshes[ iLoop ];
443       ngMesh->DeleteMesh();
444
445       if ( iLoop == NO_LOC_SIZE )
446       {
447         ngMesh->SetGlobalH ( mparam.maxh );
448         ngMesh->SetMinimalH( mparam.minh );
449         Box<3> bb = occgeom.GetBoundingBox();
450         bb.Increase (bb.Diam()/10);
451         ngMesh->SetLocalH (bb.PMin(), bb.PMax(), mparam.grading);
452       }
453
454       nodeVec.clear();
455       faceErr = aMesher.AddSegmentsToMesh( *ngMesh, occgeom, wires, helper, nodeVec,
456                                            /*overrideMinH=*/!_hypParameters);
457       if ( faceErr && !faceErr->IsOK() )
458         break;
459
460       //if ( !isCommonLocalSize )
461       //limitSize( ngMesh, mparam.maxh * 0.8);
462
463       // -------------------------
464       // Generate surface mesh
465       // -------------------------
466
467       const int startWith = MESHCONST_MESHSURFACE;
468       const int endWith   = toOptimize ? MESHCONST_OPTSURFACE : MESHCONST_MESHSURFACE;
469
470       SMESH_Comment str;
471       try {
472         OCC_CATCH_SIGNALS;
473
474 #ifdef NETGEN_V5
475         err = netgen::OCCGenerateMesh(occgeom, ngMesh, netgen::mparam, startWith, endWith);
476 #else
477         char *optstr = 0;
478         err = netgen::OCCGenerateMesh(occgeom, ngMesh, startWith, endWith, optstr);
479 #endif
480         if ( netgen::multithread.terminate )
481           return false;
482         if ( err )
483           str << "Error in netgen::OCCGenerateMesh() at " << netgen::multithread.task;
484       }
485       catch (Standard_Failure& ex)
486       {
487         err = 1;
488         str << "Exception in  netgen::OCCGenerateMesh()"
489             << " at " << netgen::multithread.task
490             << ": " << ex.DynamicType()->Name();
491         if ( ex.GetMessageString() && strlen( ex.GetMessageString() ))
492           str << ": " << ex.GetMessageString();
493       }
494       catch (...) {
495         err = 1;
496         str << "Exception in  netgen::OCCGenerateMesh()"
497             << " at " << netgen::multithread.task;
498       }
499       if ( err )
500       {
501         if ( aMesher.FixFaceMesh( occgeom, *ngMesh, 1 ))
502           break;
503         if ( iLoop == LOC_SIZE )
504         {
505           netgen::mparam.minh = netgen::mparam.maxh;
506           netgen::mparam.maxh = 0;
507           for ( size_t iW = 0; iW < wires.size(); ++iW )
508           {
509             StdMeshers_FaceSidePtr wire = wires[ iW ];
510             const vector<UVPtStruct>& uvPtVec = wire->GetUVPtStruct();
511             for ( size_t iP = 1; iP < uvPtVec.size(); ++iP )
512             {
513               SMESH_TNodeXYZ   p( uvPtVec[ iP ].node );
514               netgen::Point3d np( p.X(),p.Y(),p.Z());
515               double segLen = p.Distance( uvPtVec[ iP-1 ].node );
516               double   size = ngMesh->GetH( np );
517               netgen::mparam.minh = Min( netgen::mparam.minh, size );
518               netgen::mparam.maxh = Max( netgen::mparam.maxh, segLen );
519             }
520           }
521           //cerr << "min " << netgen::mparam.minh << " max " << netgen::mparam.maxh << endl;
522           netgen::mparam.minh *= 0.9;
523           netgen::mparam.maxh *= 1.1;
524           continue;
525         }
526         else
527         {
528           faceErr.reset( new SMESH_ComputeError( COMPERR_ALGO_FAILED, str ));
529         }
530       }
531
532
533       // ----------------------------------------------------
534       // Fill the SMESHDS with the generated nodes and faces
535       // ----------------------------------------------------
536
537       int nbNodes = ngMesh->GetNP();
538       int nbFaces = ngMesh->GetNSE();
539
540       int nbInputNodes = nodeVec.size()-1;
541       nodeVec.resize( nbNodes+1, 0 );
542
543       // add nodes
544       for ( int ngID = nbInputNodes + 1; ngID <= nbNodes; ++ngID )
545       {
546         const MeshPoint& ngPoint = ngMesh->Point( ngID );
547         SMDS_MeshNode * node = meshDS->AddNode(ngPoint(0), ngPoint(1), ngPoint(2));
548         nodeVec[ ngID ] = node;
549       }
550
551       // create faces
552       int i,j;
553       vector<const SMDS_MeshNode*> nodes;
554       for ( i = 1; i <= nbFaces ; ++i )
555       {
556         const Element2d& elem = ngMesh->SurfaceElement(i);
557         nodes.resize( elem.GetNP() );
558         for (j=1; j <= elem.GetNP(); ++j)
559         {
560           int pind = elem.PNum(j);
561           if ( pind < 1 )
562             break;
563           nodes[ j-1 ] = nodeVec[ pind ];
564           if ( nodes[ j-1 ]->GetPosition()->GetTypeOfPosition() == SMDS_TOP_3DSPACE )
565           {
566             const PointGeomInfo& pgi = elem.GeomInfoPi(j);
567             meshDS->SetNodeOnFace( nodes[ j-1 ], faceID, pgi.u, pgi.v);
568           }
569         }
570         if ( j > elem.GetNP() )
571         {
572           SMDS_MeshFace* face = 0;
573           if ( elem.GetType() == TRIG )
574             face = helper.AddFace(nodes[0],nodes[1],nodes[2]);
575           else
576             face = helper.AddFace(nodes[0],nodes[1],nodes[2],nodes[3]);
577         }
578       }
579
580       break;
581     } // two attempts
582   } // loop on FACEs
583
584   return true;
585 }
586
587 void NETGENPlugin_NETGEN_2D_ONLY::CancelCompute()
588 {
589   SMESH_Algo::CancelCompute();
590   netgen::multithread.terminate = 1;
591 }
592
593 //================================================================================
594 /*!
595  * \brief Return progress of Compute() [0.,1]
596  */
597 //================================================================================
598
599 double NETGENPlugin_NETGEN_2D_ONLY::GetProgress() const
600 {
601   return -1;
602   // const char* task1 = "Surface meshing";
603   // //const char* task2 = "Optimizing surface";
604   // double& progress = const_cast<NETGENPlugin_NETGEN_2D_ONLY*>( this )->_progress;
605   // if ( _progressByTic < 0. &&
606   //      strncmp( netgen::multithread.task, task1, 3 ) == 0 )
607   // {
608   //   progress = Min( 0.25, SMESH_Algo::GetProgressByTic() ); // [0, 0.25]
609   // }
610   // else //if ( strncmp( netgen::multithread.task, task2, 3 ) == 0)
611   // {
612   //   if ( _progressByTic < 0 )
613   //   {
614   //     NETGENPlugin_NETGEN_2D_ONLY* me = (NETGENPlugin_NETGEN_2D_ONLY*) this;
615   //     me->_progressByTic = 0.25 / (_progressTic+1);
616   //   }
617   //   const_cast<NETGENPlugin_NETGEN_2D_ONLY*>( this )->_progressTic++;
618   //   progress = Max( progress, _progressByTic * _progressTic );
619   // }
620   // //cout << netgen::multithread.task << " " << _progressTic << endl;
621   // return Min( progress, 0.99 );
622 }
623
624 //=============================================================================
625 /*!
626  *
627  */
628 //=============================================================================
629
630 bool NETGENPlugin_NETGEN_2D_ONLY::Evaluate(SMESH_Mesh& aMesh,
631                                            const TopoDS_Shape& aShape,
632                                            MapShapeNbElems& aResMap)
633 {
634   TopoDS_Face F = TopoDS::Face(aShape);
635   if(F.IsNull())
636     return false;
637
638   // collect info from edges
639   int nb0d = 0, nb1d = 0;
640   bool IsQuadratic = false;
641   bool IsFirst = true;
642   double fullLen = 0.0;
643   TopTools_MapOfShape tmpMap;
644   for (TopExp_Explorer exp(F, TopAbs_EDGE); exp.More(); exp.Next()) {
645     TopoDS_Edge E = TopoDS::Edge(exp.Current());
646     if( tmpMap.Contains(E) )
647       continue;
648     tmpMap.Add(E);
649     SMESH_subMesh *aSubMesh = aMesh.GetSubMesh(exp.Current());
650     MapShapeNbElemsItr anIt = aResMap.find(aSubMesh);
651     if( anIt==aResMap.end() ) {
652       SMESH_subMesh *sm = aMesh.GetSubMesh(F);
653       SMESH_ComputeErrorPtr& smError = sm->GetComputeError();
654       smError.reset( new SMESH_ComputeError(COMPERR_ALGO_FAILED,"Submesh can not be evaluated",this));
655       return false;
656     }
657     std::vector<int> aVec = (*anIt).second;
658     nb0d += aVec[SMDSEntity_Node];
659     nb1d += Max(aVec[SMDSEntity_Edge],aVec[SMDSEntity_Quad_Edge]);
660     double aLen = SMESH_Algo::EdgeLength(E);
661     fullLen += aLen;
662     if(IsFirst) {
663       IsQuadratic = (aVec[SMDSEntity_Quad_Edge] > aVec[SMDSEntity_Edge]);
664       IsFirst = false;
665     }
666   }
667   tmpMap.Clear();
668
669   // compute edge length
670   double ELen = 0;
671   if (( _hypLengthFromEdges ) || ( !_hypLengthFromEdges && !_hypMaxElementArea )) {
672     if ( nb1d > 0 )
673       ELen = fullLen / nb1d;
674   }
675   if ( _hypMaxElementArea ) {
676     double maxArea = _hypMaxElementArea->GetMaxArea();
677     ELen = sqrt(2. * maxArea/sqrt(3.0));
678   }
679   GProp_GProps G;
680   BRepGProp::SurfaceProperties(F,G);
681   double anArea = G.Mass();
682
683   const int hugeNb = numeric_limits<int>::max()/10;
684   if ( anArea / hugeNb > ELen*ELen )
685   {
686     SMESH_subMesh *sm = aMesh.GetSubMesh(F);
687     SMESH_ComputeErrorPtr& smError = sm->GetComputeError();
688     smError.reset( new SMESH_ComputeError(COMPERR_ALGO_FAILED,"Submesh can not be evaluated.\nToo small element length",this));
689     return false;
690   }
691   int nbFaces = (int) ( anArea / ( ELen*ELen*sqrt(3.) / 4 ) );
692   int nbNodes = (int) ( ( nbFaces*3 - (nb1d-1)*2 ) / 6 + 1 );
693   std::vector<int> aVec(SMDSEntity_Last);
694   for(int i=SMDSEntity_Node; i<SMDSEntity_Last; i++) aVec[i]=0;
695   if( IsQuadratic ) {
696     aVec[SMDSEntity_Node] = nbNodes;
697     aVec[SMDSEntity_Quad_Triangle] = nbFaces;
698   }
699   else {
700     aVec[SMDSEntity_Node] = nbNodes;
701     aVec[SMDSEntity_Triangle] = nbFaces;
702   }
703   SMESH_subMesh *sm = aMesh.GetSubMesh(F);
704   aResMap.insert(std::make_pair(sm,aVec));
705
706   return true;
707 }