1 // Copyright (C) 2007-2021 CEA/DEN, EDF R&D
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.
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.
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
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 // File : BLSURFPlugin_EnforcedMesh1D.cxx
20 // Author : Edward AGAPOV (OCC)
22 #include "BLSURFPlugin_EnforcedMesh1D.hxx"
24 #include "BLSURFPlugin_BLSURF.hxx"
26 #include <SMDS_IteratorOnIterators.hxx>
27 #include <SMDS_MeshEdge.hxx>
28 #include <SMDS_MeshGroup.hxx>
29 #include <SMESHDS_Group.hxx>
30 #include <SMESHDS_Mesh.hxx>
31 #include <SMESH_Group.hxx>
32 #include <SMESH_Mesh.hxx>
33 #include <SMESH_MeshAlgos.hxx>
34 #include <SMESH_MeshEditor.hxx>
35 #include <SMESH_MesherHelper.hxx>
36 #include <SMESH_TypeDefs.hxx>
38 #include <BRepAdaptor_Curve.hxx>
39 #include <BRepBuilderAPI_MakeVertex.hxx>
40 #include <BRep_Tool.hxx>
41 #include <Extrema_ExtCC.hxx>
42 #include <Extrema_ExtElC.hxx>
43 #include <Geom_Line.hxx>
44 #include <ShapeAnalysis_Curve.hxx>
45 #include <Geom2d_Line.hxx>
47 // allow range iteration on NCollection_IndexedMap
48 template < class IMAP >
49 typename IMAP::const_iterator begin( const IMAP & m ) { return m.cbegin(); }
50 template < class IMAP >
51 typename IMAP::const_iterator end( const IMAP & m ) { return m.cend(); }
56 //================================================================================
58 * \brief Look for a node coincident with a given nodes among end nodes of a segment
59 * and among its internal nodes
60 * \param [in] p - the epoint
61 * \param [in] tol - 3D tolarace
62 * \param [in] segment - the segment
63 * \param [in] segInternalNodes - map of segment internal nodes
64 * \param [out] index - return index of found internal node; -1 if an end node is found
65 * \return const SMDS_MeshNode* - found node
67 //================================================================================
69 const SMDS_MeshNode* findNode( const gp_Pnt& p,
71 const SMDS_MeshElement* segment,
72 BLSURFPlugin_EnforcedMesh1D::TNodesOnSeg & segInternalNodes,
76 SMESH_NodeXYZ node0 = segment->GetNode(0);
77 if ( p.IsEqual( node0, tol ))
78 return index = -1, node0.Node();
79 SMESH_NodeXYZ node1 = segment->GetNode(1);
80 if ( p.IsEqual( node1, tol ))
81 return index = -1, node1.Node();
83 auto seg2nodes = segInternalNodes.find( segment );
84 if ( seg2nodes != segInternalNodes.end() )
86 const std::vector< const SMDS_MeshNode* >& nodes = seg2nodes->second;
87 for ( size_t i = 0; i < nodes.size(); ++i )
88 if ( p.IsEqual( SMESH_NodeXYZ( nodes[i] ), tol ))
97 //================================================================================
99 * \brief Orient segments to correspond to order of nodes in a branch
100 * \param [in] braSegs - segments of the branch
101 * \param [in] braNodes - nodes of the branch
102 * \param [in] nodeIndex - index of a node of the branch
103 * \param [inout] mesh - mesh holding the nodes and segments
107 //================================================================================
109 void orientSegments( const std::vector< const SMDS_MeshElement* > & braSegs,
110 const std::vector< const SMDS_MeshNode* > & braNodes,
111 const size_t nodeIndex,
114 const SMDS_MeshElement* toReverse[2] = { nullptr, nullptr };
116 if ( nodeIndex > 0 &&
117 braSegs[ nodeIndex - 1 ]->GetNode(1) != braNodes[ nodeIndex ])
118 toReverse[ 0 ] = braSegs[ nodeIndex - 1 ];
120 if ( nodeIndex < braSegs.size() &&
121 braSegs[ nodeIndex ]->GetNode(0) != braNodes[ nodeIndex ])
122 toReverse[ bool( toReverse[0]) ] = braSegs[ nodeIndex ];
127 SMESH_MeshEditor editor( mesh );
128 for ( int i = 0; i < 2; ++i )
130 editor.Reorient( toReverse[ i ]);
135 //================================================================================
137 * \brief Create enforced mesh segments in a mesh
138 * \param [in] helper - contains the mesh and the shape to compute
139 * \param [in] hyp - hypothesis containing data of enforced meshes
141 //================================================================================
143 BLSURFPlugin_EnforcedMesh1D::BLSURFPlugin_EnforcedMesh1D( SMESH_MesherHelper& helper,
144 const BLSURFPlugin_Hypothesis* hyp )
145 : _mesh ( helper.GetMesh() ),
146 _shape( helper.GetSubShape() ),
148 _isQuadratic( helper.GetIsQuadratic() )
150 if ( !hyp || !_mesh || hyp->GetEnforcedMeshes().empty() )
153 _tol = 2 * BRep_Tool::MaxTolerance( _shape, TopAbs_VERTEX );
154 _segTag = 1000 + helper.Count( _shape, TopAbs_EDGE, /*ignoreSame=*/false );
156 _nodeTag0 = 1000 + helper.Count( _shape, TopAbs_VERTEX, /*ignoreSame=*/false );
158 for ( const BLSURFPlugin_Hypothesis::EnforcedMesh& enfMesh : hyp->GetEnforcedMeshes() )
160 copyEnforcedMesh( enfMesh, hyp, _shape );
163 for ( const SMDS_MeshElement* segment : _segmentsOnSeveralShapes )
165 splitSegmentOnSeveralShapes( segment );
168 for ( const TopoDS_Shape & face : _faces )
170 splitSelfIntersectingSegments( face );
173 // for ( TEdge2Nodes::Iterator e2nn( _nodesOnEdge ); e2nn.More(); e2nn.Next() )
175 // splitEdgeByNodes( e2nn.Key(), e2nn.Value() );
179 //================================================================================
181 * \brief Convert enforced segments to quadratic
183 //================================================================================
185 BLSURFPlugin_EnforcedMesh1D::~BLSURFPlugin_EnforcedMesh1D()
190 _helper.SetIsQuadratic( true );
192 SMESHDS_Mesh* meshDS = _mesh->GetMeshDS();
193 std::vector<const SMDS_MeshNode *> nodes(2);
194 std::vector<const SMDS_MeshElement *> trias;
196 SMDS_MeshGroup* group = nullptr;
197 if ( !_name2Group.empty() )
198 group = _name2Group.begin()->second;
200 for ( const TopoDS_Shape& f : _faces )
201 if ( HasSegmentsOnFace( TopoDS::Face( f )))
202 while ( _segIterator->more() )
204 const SMDS_MeshElement* segment = _segIterator->next();
205 if ( segment->GetType() != SMDSAbs_Edge )
208 nodes[0] = segment->GetNode(0);
209 nodes[1] = segment->GetNode(1);
210 if ( meshDS->GetElementsByNodes( nodes, trias, SMDSAbs_Face ))
212 if ( !group || !group->Contains( segment ))
214 SMDS_MeshGroup* otherGroup = nullptr;
215 if ( _name2Group.size() > bool( group ))
216 for ( auto & n2g : _name2Group )
217 if ( n2g.second != group && n2g.second->Contains( segment ))
219 otherGroup = n2g.second;
225 _helper.AddTLinks( static_cast< const SMDS_MeshFace* >( trias[0] ));
227 meshDS->RemoveFreeElement( segment, meshDS->MeshElements( f ), /*fromGroup=*/false );
229 SMDS_MeshElement* quadSegment = _helper.AddEdge( nodes[0], nodes[1],
230 /*id=*/0, /*force3d=*/false );
231 if ( group && segment != quadSegment )
233 group->Remove( segment );
234 group->Add( quadSegment );
241 //================================================================================
243 * \brief Add a vertex on EDGE
245 //================================================================================
247 void BLSURFPlugin_EnforcedMesh1D::AddVertexOnEdge( const double* theXYZ )
249 // setup predicates to find the supporting EDGE
250 setupPredicates( _shape );
252 SMESHDS_Mesh* meshDS = _mesh->GetMeshDS();
253 const SMDS_MeshNode* nodeOnE = meshDS->AddNode( theXYZ[0], theXYZ[1], theXYZ[2] );
255 // check if enfNode is on VERTEX
256 bool toRemove = true;
257 TopoDS_Vertex vertex;
259 if ( _onVertexPredicate->IsSatisfy( nodeOnE, &vertex ))
261 toRemove = SMESH_Algo::VertexNode( vertex, meshDS );
263 meshDS->SetNodeOnVertex( nodeOnE, vertex );
265 // find the EDGE supporting theXYZ
266 else if ( _onEdgePredicate->IsSatisfy( nodeOnE, &edge ))
268 gp_Pnt pnt( theXYZ[0], theXYZ[1], theXYZ[2] );
269 toRemove = findNodeOnEdge( pnt, edge );
271 addNodeOnEdge( nodeOnE, edge );
275 meshDS->RemoveFreeNode( nodeOnE, /*submesh=*/nullptr, /*fromGroup=*/false );
278 //================================================================================
280 * \brief Return EDGEs resulted from division of FACE boundary by enforced segments
281 * and enforced vertices
282 * \param [in] edge - possibly divided EDGE
283 * \param [out] splits - split EDGEs
284 * \return bool - true if the EDGE is split
286 //================================================================================
288 bool BLSURFPlugin_EnforcedMesh1D::GetSplitsOfEdge( const TopoDS_Edge& edge,
289 std::vector< TopoDS_Edge > & splits,
290 TopTools_IndexedMapOfShape & edgeTags )
292 if ( _nodesOnEdge.IsBound( edge )) // divide the EDGE
294 splitEdgeByNodes( edge, _nodesOnEdge( edge ));
296 _nodesOnEdge.UnBind( edge );
301 std::vector< TopoDS_Edge > * splitsInMap = _edgeSplitsOfEdge.ChangeSeek( edge );
305 splits.insert( splits.end(), splitsInMap->begin(), splitsInMap->end() );
307 int eTag = edgeTags.Add( edge );
309 size_t index = splits.size() - 1;
310 for ( size_t i = 0; i < splitsInMap->size(); ++i, --index )
312 int splitTag = edgeTags.Add( splits[ index ]);
313 _splitTags2EdgeTag[ splitTag ] = eTag;
315 if ( edge.Orientation() == TopAbs_REVERSED )
316 splits[ index ].Reverse();
323 //================================================================================
325 * \brief Add 1D elements to the mesh
327 //================================================================================
329 void BLSURFPlugin_EnforcedMesh1D::
331 copyEnforcedMesh( const BLSURFPlugin_Hypothesis::EnforcedMesh& theEnfMesh,
332 const BLSURFPlugin_Hypothesis* theHyp,
333 const TopoDS_Shape& theShape)
336 SMDS_ElemIteratorPtr segIt = theHyp->GetEnforcedSegments( theEnfMesh, mesh1D );
337 if ( !segIt->more() )
340 // setup predicates to detect nodes on FACE boundary
341 setupPredicates( theShape );
343 SMDS_MeshGroup* group = getGroup( theEnfMesh._groupName );
344 SMESHDS_Mesh* meshDS = _helper.GetMeshDS();
346 // get ordered nodes and segments of theEnfMesh
347 SMESH_MeshAlgos::TElemGroupVector edgeBranches;
348 SMESH_MeshAlgos::TNodeGroupVector nodeBranches;
349 SMESH_MeshAlgos::Get1DBranches( segIt, edgeBranches, nodeBranches );
352 // Copy nodes and segments from an enforced mesh to my mesh
354 TopoDS_Shape vertex, edge;
356 // first treat ends of branches that can be shared by branches
357 for ( size_t iB = 0; iB < nodeBranches.size(); ++iB )
359 std::vector< const SMDS_MeshNode* > & braNodes = nodeBranches[ iB ];
360 std::vector< const SMDS_MeshElement* > & braSegs = edgeBranches[ iB ];
362 for ( int isLast = 0; isLast < 2; ++isLast )
364 const SMDS_MeshNode* enfNode = isLast ? braNodes.back() : braNodes[0];
365 if ( meshDS->Contains( enfNode ))
366 continue; // already in my mesh
368 const SMDS_MeshNode* newNode = copyEnforcedNode( enfNode );
370 orientSegments( braSegs, braNodes, isLast ? 0 : braNodes.size() - 1, mesh1D );
372 // replace enfNode at branch ends by newNode
373 SMESH_NodeXYZ enfPnt( newNode ? newNode : enfNode );
374 for ( std::vector< const SMDS_MeshNode* > & braNodes : nodeBranches )
376 for ( int isLast = 0; isLast < 2; ++isLast )
378 const SMDS_MeshNode* & endNode = isLast ? braNodes.back() : braNodes[0];
379 if ( endNode == enfNode || enfPnt.SquareDistance( endNode ) < _tol*_tol )
384 } // loop on branch ends
385 } // loop on nodeBranches
387 // copy nodes and segments
389 for ( size_t iB = 0; iB < nodeBranches.size(); ++iB )
391 std::vector< const SMDS_MeshNode* > & braNodes = nodeBranches[ iB ];
392 std::vector< const SMDS_MeshElement* > & braSegs = edgeBranches[ iB ];
394 // copy nodes of the branch
395 for ( size_t i = 0; i < braNodes.size(); ++i )
397 const SMDS_MeshNode* & enfNode = braNodes[ i ];
398 const SMDS_MeshNode* newNode = copyEnforcedNode( enfNode );
400 if ( !newNode ) // orient segments to be able to get enforced not projected node
401 orientSegments( braSegs, braNodes, i, mesh1D );
406 // copy segments of the branch
407 for ( size_t i = 0; i < braSegs.size(); ++i )
409 //braSegs[ i ] = nullptr;
411 const SMDS_MeshNode* node0 = braNodes[ i ];
412 const SMDS_MeshNode* node1 = braNodes[ i + 1 ];
413 if ( !node0 && !node1 )
416 TopoDS_Shape shape0 = _helper.GetSubShapeByNode( node0, meshDS );
417 TopoDS_Shape shape1 = _helper.GetSubShapeByNode( node1, meshDS );
418 if ( shape0.IsNull() && shape1.IsNull() )
421 if ( !node0 && shape1.ShapeType() != TopAbs_FACE )
423 if ( !node1 && shape0.ShapeType() != TopAbs_FACE )
426 if ( !node0 || !node1 ) // create missing node at location of enforced node projected nowhere
428 SMESH_NodeXYZ pn = braSegs[i]->GetNode( !node1 );
429 ( node0 ? node1 : node0 ) = _helper.AddNode( pn->X(), pn->Y(), pn->Z() );
432 SMDS_MeshEdge* newSeg = _helper.AddEdge( node0, node1 );
434 group->Add( newSeg );
435 braSegs[ i ] = newSeg;
437 // check if the both nodes are on the same FACE
438 TopoDS_Shape face = shape0;
439 if ( !shape0.IsSame( shape1 ) && !shape0.IsNull() && !shape1.IsNull() )
441 if ( shape0.ShapeType() == TopAbs_FACE &&
442 _helper.IsSubShape( shape1, shape0 ))
446 else if ( shape1.ShapeType() == TopAbs_FACE &&
447 _helper.IsSubShape( shape0, shape1 ))
451 else // try to find a FACE by projecting a segment middle point
454 gp_Pnt middlePnt = 0.5 * ( SMESH_NodeXYZ( node0 ) + SMESH_NodeXYZ( node1 ));
455 //BLSURFPlugin_BLSURF::projectionPoint projPnt =
456 BLSURFPlugin_BLSURF::getProjectionPoint( TopoDS::Face( face ), middlePnt );
458 if ( !face.IsNull() &&
459 ( !_helper.IsSubShape( shape0, face ) ||
460 !_helper.IsSubShape( shape1, face ) ))
464 if ( !face.IsNull() && face.ShapeType() == TopAbs_FACE )
466 meshDS->SetMeshElementOnShape( newSeg, face );
470 if ( face.IsNull() || shape0.IsNull() || shape1.IsNull() )
472 _segmentsOnSeveralShapes.push_back( newSeg );
475 } // loop on branch segments
477 } // loop on branches
482 //================================================================================
484 * \brief Create a copy of a node of enforced mesh in my mesh
485 * \param [in] enfNode - enforced node
486 * \return const SMDS_MeshNode* - a node in my mesh
488 //================================================================================
490 const SMDS_MeshNode* BLSURFPlugin_EnforcedMesh1D::copyEnforcedNode( const SMDS_MeshNode* enfNode )
492 SMESHDS_Mesh * meshDS = _helper.GetMeshDS();
494 if ( !enfNode || meshDS->Contains( enfNode ))
495 return enfNode; // already in my mesh
497 SMESH_NodeXYZ enfPnt = enfNode;
499 const SMDS_MeshNode* newNode = nullptr;
501 // check if enfNode is on VERTEX
502 TopoDS_Vertex vertex;
503 if ( _onVertexPredicate->IsSatisfy( enfNode, &vertex ))
505 _mesh->GetSubMesh( vertex )->ComputeStateEngine( SMESH_subMesh::COMPUTE );
506 newNode = SMESH_Algo::VertexNode( vertex, meshDS );
509 // check if enfNode is on EDGE
510 bool setNodeOnEdge = false;
514 setNodeOnEdge = _onEdgePredicate->IsSatisfy( enfNode, &edge );
517 newNode = findNodeOnEdge( enfPnt, edge );
518 setNodeOnEdge = !newNode;
522 // create a new node and set it on FACE
526 BLSURFPlugin_BLSURF::projectionPoint projPnt =
527 BLSURFPlugin_BLSURF::getProjectionPoint( face, enfPnt, /*allowStateON=*/true );
528 if ( face.IsNull() ) return newNode;
530 if ( projPnt.state == TopAbs_ON )
532 SMDS_MeshNode* projNode = const_cast< SMDS_MeshNode* >( enfNode );
533 projNode->setXYZ( projPnt.xyz.X(), projPnt.xyz.Y(), projPnt.xyz.Z() );
536 if ( _onVertexPredicate->IsSatisfy( projNode, &vertex ))
538 _mesh->GetSubMesh( vertex )->ComputeStateEngine( SMESH_subMesh::COMPUTE );
539 newNode = SMESH_Algo::VertexNode( vertex, meshDS );
541 else if (( setNodeOnEdge = _onEdgePredicate->IsSatisfy( projNode, &edge )))
543 newNode = findNodeOnEdge( projPnt.xyz, edge );
544 setNodeOnEdge = !newNode;
546 projNode->setXYZ( enfPnt.X(), enfPnt.Y(), enfPnt.Z() );
550 newNode = meshDS->AddNode( projPnt.xyz.X(), projPnt.xyz.Y(), projPnt.xyz.Z() );
552 if ( vertex.IsNull() && edge.IsNull() )
553 meshDS->SetNodeOnFace( newNode, face, projPnt.uv.X(), projPnt.uv.Y() );
556 // set the new node on EDGE
557 if ( newNode && setNodeOnEdge )
559 addNodeOnEdge( newNode, edge );
565 //================================================================================
567 * \brief Split a segment whose nodes are on different FACEs into smaller parts
568 * lying each on one FACE
570 //================================================================================
572 void BLSURFPlugin_EnforcedMesh1D::splitSegmentOnSeveralShapes( const SMDS_MeshElement* segment )
574 const SMDS_MeshNode* node0 = segment->GetNode(0);
575 const SMDS_MeshNode* node1 = segment->GetNode(1);
576 SMESHDS_Mesh * meshDS = _helper.GetMeshDS();
578 TopoDS_Shape shape0 = _helper.GetSubShapeByNode( node0, meshDS );
579 TopoDS_Shape shape1 = _helper.GetSubShapeByNode( node1, meshDS );
581 if ( shape0.IsNull() )
583 std::swap( node0, node1 );
588 gp_XYZ xyz0 = SMESH_NodeXYZ( node0 );
589 gp_XYZ xyz1 = SMESH_NodeXYZ( node1 );
590 gp_XYZ segDir = ( xyz1 - xyz0 ).Normalized();
592 //std::map< double, const SMDS_MeshNode* > mediumNodes; // nodes splitting the segment
594 while ( !shape0.IsSame( shape1 )) // move along the segment till shape1
596 if ( shape0.ShapeType() == TopAbs_FACE ) // make a node on an EDGE of the FACE
597 { // ----------------------------------
600 gp_Pnt edgeIntPnt = getEdgeIntersection( shape0, xyz0, xyz1, edge, paramOnE );
604 // check if edgeIntPnt in on VERTEX
605 TopoDS_Vertex vertex;
606 for ( int iV = 0; iV < 2 && vertex.IsNull(); ++iV )
608 TopoDS_Vertex v = _helper.IthVertex( iV, edge );
609 if ( edgeIntPnt.SquareDistance( BRep_Tool::Pnt( v )) < _tol *_tol )
613 // make a node on the EDGE
614 const SMDS_MeshNode* nodeOnEdge = nullptr;
615 if ( vertex.IsNull() )
617 nodeOnEdge = findNodeOnEdge( edgeIntPnt, edge );
620 if ( shape1.IsNull() && node1 )
624 meshDS->MoveNode( nodeOnEdge, edgeIntPnt.X(), edgeIntPnt.Y(), edgeIntPnt.Z() );
628 nodeOnEdge = _helper.AddNode( edgeIntPnt.X(), edgeIntPnt.Y(), edgeIntPnt.Z() );
630 addNodeOnEdge( nodeOnEdge, edge, paramOnE );
635 _mesh->GetSubMesh( vertex )->ComputeStateEngine( SMESH_subMesh::COMPUTE );
636 nodeOnEdge = SMESH_Algo::VertexNode( vertex, meshDS );
639 // create a sub-segment
640 SMDS_MeshElement* newSeg = _helper.AddEdge( node0, nodeOnEdge );
641 meshDS->SetMeshElementOnShape( newSeg, shape0 );
643 SMESH_MeshEditor::AddToSameGroups( newSeg, segment, meshDS );
646 xyz0 = SMESH_NodeXYZ( node0 );
647 if ( vertex.IsNull() )
653 else // shape0 is EDGE or VERTEX; look for the next FACE
654 { // ------------------------------------------------
656 if ( !shape1.IsNull() &&
657 shape1.ShapeType() == TopAbs_FACE &&
658 _helper.IsSubShape( shape0, shape1 )) // shape0 belongs to FACE shape1
660 SMDS_MeshElement* newSeg = _helper.AddEdge( node0, node1 );
661 SMESH_MeshEditor::AddToSameGroups( newSeg, segment, meshDS );
662 meshDS->SetMeshElementOnShape( newSeg, shape1 );
667 double shift = 10 * _tol;
668 for ( int nbAttemp = 0; face.IsNull() && nbAttemp < 10; ++nbAttemp )
670 xyz0 += segDir * shift;
672 BLSURFPlugin_BLSURF::getProjectionPoint( face, xyz0 );
674 if ( !face.IsNull() )
676 if ( _helper.IsSubShape( shape0, face ))
688 } //while ( !shape0.IsSame( shape1 ))
690 meshDS->RemoveFreeElement( segment, /*submesh=*/nullptr );
693 //================================================================================
695 * \brief Find intersection of FACE EDGEs and a segment
696 * \param [in] theFace - the FACE
697 * \param [in] thePnt0 - first end of the segment
698 * \param [in] thePnt1 - last end of the segment
699 * \param [out] theFounfEdge - return the intersected EDGE
700 * \param [out] theParamOnEdge - return parameter of intersection point on EDGE
701 * \return gp_XYZ - point on an EDGE closest to the segment
703 //================================================================================
705 gp_Pnt BLSURFPlugin_EnforcedMesh1D::getEdgeIntersection( const TopoDS_Shape& theFaceOrEdge,
706 const gp_XYZ& thePnt0,
707 const gp_XYZ& thePnt1,
708 TopoDS_Edge & theFounfEdge,
709 double & theParamOnEdge)
711 const double segLen = ( thePnt1 - thePnt0 ).Modulus();
712 const double maxSegDist2 = segLen * segLen * 0.5 * 0.5;
714 Handle(Geom_Line) segLine = new Geom_Line( thePnt0, thePnt1 - thePnt0 );
715 GeomAdaptor_Curve segLineAdpt( segLine, 0, segLen );
717 TopTools_MapOfShape edges;
718 double minParamOnSeg = segLen;
720 for ( TopExp_Explorer edgeExp( theFaceOrEdge, TopAbs_EDGE ); edgeExp.More(); edgeExp.Next() )
722 const TopoDS_Edge& edge = TopoDS::Edge( edgeExp.Current() );
723 if ( !edges.Add( edge ))
726 Extrema_ExtCC extrema( segLineAdpt, BRepAdaptor_Curve( edge ), _tol, _tol );
728 if ( extrema.IsDone() && !extrema.IsParallel() )
729 for ( int i = 1, nb = extrema.NbExt(); i <= nb; ++i )
730 if ( extrema.SquareDistance( i ) < maxSegDist2 )
732 Extrema_POnCurv pOnSeg, pOnEdge;
733 extrema.Points( i, pOnSeg, pOnEdge );
734 double paramOnSeg = pOnSeg.Parameter();
735 if ( 0 < paramOnSeg && paramOnSeg < minParamOnSeg )
737 minParamOnSeg = paramOnSeg;
738 foundPnt = pOnEdge.Value();
740 theParamOnEdge = pOnEdge.Parameter();
747 //================================================================================
749 * \brief Split self-intersecting segments on a given FACE
751 //================================================================================
753 void BLSURFPlugin_EnforcedMesh1D::splitSelfIntersectingSegments( const TopoDS_Shape & theFace )
755 const TopoDS_Face& face = TopoDS::Face( theFace );
757 SMESHDS_Mesh* meshDS = _helper.GetMeshDS();
758 SMESHDS_SubMesh* sm = meshDS->MeshElements( face );
759 if ( !sm || sm->NbElements() <= 1 )
762 // get ordered nodes and segments on the face
763 SMESH_MeshAlgos::TElemGroupVector edgeBranches;
764 SMESH_MeshAlgos::TNodeGroupVector nodeBranches;
765 SMESH_MeshAlgos::Get1DBranches( sm->GetElements(), edgeBranches, nodeBranches );
767 // create element searcher
768 SMESH_ElementSearcher* elemSearcher;
769 SMESHUtils::Deleter< SMESH_ElementSearcher > elSearchdeleter;
770 std::vector< const SMDS_MeshElement* > foundElems;
772 std::vector< SMDS_ElemIteratorPtr > elemItVec;
773 for ( std::vector< const SMDS_MeshElement* > & braSegs : edgeBranches )
775 SMDS_ElemIteratorPtr segIt =
776 boost::make_shared< SMDS_ElementVectorIterator >( braSegs.begin(), braSegs.end() );
777 elemItVec.push_back( segIt );
779 typedef SMDS_IteratorOnIterators< const SMDS_MeshElement*,
780 std::vector< SMDS_ElemIteratorPtr > > TVecIterator;
781 SMDS_ElemIteratorPtr segIt = boost::make_shared< TVecIterator >( elemItVec );
783 elemSearcher = SMESH_MeshAlgos::GetElementSearcher( *meshDS, segIt, _tol );
784 elSearchdeleter._obj = elemSearcher;
786 // force usage of iterators before they die
787 elemSearcher->FindElementsByPoint( gp_Pnt( 0,0,1e+20), SMDSAbs_Edge, foundElems );
791 // Find intersecting segments
793 std::map< const SMDS_MeshElement* , std::vector< const SMDS_MeshNode* > > segInternalNodes;
794 SMESH_MeshEditor::TListOfListOfNodes nodeGroupsToMerge;
796 for ( std::vector< const SMDS_MeshElement* > & braSegs : edgeBranches )
798 braSegs.push_back( braSegs.back() );
799 const SMDS_MeshElement* prevSeg = nullptr;
801 for ( size_t i = 0, nb = braSegs.size() - 1; i < nb; ++i )
803 const SMDS_MeshElement* seg = braSegs[ i ];
804 const SMDS_MeshElement* nextSeg = braSegs[ i + 1 ];
805 const SMDS_MeshNode* node0 = seg->GetNode(0);
806 const SMDS_MeshNode* node1 = seg->GetNode(1);
808 gp_XYZ xyz0 = SMESH_NodeXYZ( node0 );
809 gp_XYZ xyz1 = SMESH_NodeXYZ( node1 );
810 gp_XYZ middlePnt = 0.5 * ( SMESH_NodeXYZ( node0 ) + SMESH_NodeXYZ( node1 ));
811 double segLen = ( xyz0 - xyz1 ).Modulus();
814 elemSearcher->GetElementsInSphere( middlePnt, 0.5 * segLen + _tol, SMDSAbs_Edge, foundElems );
816 for ( const SMDS_MeshElement* closeSeg : foundElems )
818 if ( closeSeg == prevSeg ||
820 closeSeg == nextSeg )
825 if ( intersectSegments( seg, closeSeg, face, intPnt, uv ))
828 const SMDS_MeshNode* intNode0 = findNode( intPnt, _tol, seg, segInternalNodes, i0 );
829 const SMDS_MeshNode* intNode1 = findNode( intPnt, _tol, closeSeg, segInternalNodes, i1 );
830 if ( !intNode0 && intNode1 )
832 segInternalNodes[ seg ].push_back( intNode1 );
834 else if ( !intNode1 && intNode0 )
836 segInternalNodes[ closeSeg ].push_back( intNode0 );
838 else if ( intNode1 && intNode0 )
840 if ( intNode1 == intNode0 )
842 if ( i0 < 0 && i1 < 0 )
844 nodeGroupsToMerge.push_back // merge end nodes
845 ( std::list< const SMDS_MeshNode* >({ intNode0, intNode1 }));
849 segInternalNodes[ closeSeg ][ i1 ] = intNode0;
853 segInternalNodes[ seg ][ i0 ] = intNode1;
855 else // two internal nodes coincide
857 segInternalNodes[ seg ][ i0 ] = intNode1;
858 nodeGroupsToMerge.push_back
859 ( std::list< const SMDS_MeshNode* >({ intNode1, intNode0 }));
862 else // ( !intNode1 && !intNode0 )
864 intNode0 = _helper.AddNode( intPnt.X(), intPnt.Y(), intPnt.Z() );
865 meshDS->SetNodeOnFace( intNode0, face, uv.X(), uv.Y() );
866 segInternalNodes[ seg ].push_back( intNode0 );
867 segInternalNodes[ closeSeg ].push_back( intNode0 );
874 } // loop on segments of a branch
875 } // loop on branches
878 findIntersectionWithSeamEdge( face, segInternalNodes ); // on periodic FACE
883 for ( auto& seg2nodes : segInternalNodes )
885 const SMDS_MeshElement* seg = seg2nodes.first;
886 std::vector< const SMDS_MeshNode* > & nodes = seg2nodes.second;
887 if ( nodes.empty() ) continue;
889 const SMDS_MeshNode* n0 = seg->GetNode( 0 );
890 const SMDS_MeshNode* n1 = seg->GetNode( 1 );
891 nodes.push_back( n1 );
893 // sort nodes on the segment
894 gp_Pnt p0 = SMESH_NodeXYZ( n0 );
895 std::map< double, const SMDS_MeshNode* > sortedNodes;
896 for ( SMESH_NodeXYZ pn : nodes )
897 sortedNodes.insert({ p0.SquareDistance( pn ), pn.Node() });
900 for ( auto & d2n : sortedNodes )
903 SMDS_MeshElement* newSeg = _helper.AddEdge( n0, n1 );
905 meshDS->SetMeshElementOnShape( newSeg, face );
906 SMESH_MeshEditor::AddToSameGroups( newSeg, seg, meshDS );
908 meshDS->RemoveFreeElement( seg, /*submesh=*/nullptr );
913 SMESH_MeshEditor( _mesh ).MergeNodes( nodeGroupsToMerge );
916 //================================================================================
918 * \brief Find intersections of segments with a seam EDGE on a periodic FACE
920 //================================================================================
922 void BLSURFPlugin_EnforcedMesh1D::findIntersectionWithSeamEdge( const TopoDS_Face & face,
923 TNodesOnSeg & segInternalNodes )
925 _helper.SetSubShape( face );
926 if ( !_helper.HasSeam() )
929 SMESHDS_Mesh* meshDS = _helper.GetMeshDS();
930 SMESHDS_SubMesh* sm = meshDS->MeshElements( face );
931 if ( !sm || sm->NbElements() == 0 )
934 TopTools_MapOfShape treatedEdges;
935 for ( TopExp_Explorer edgeExp( face, TopAbs_EDGE ); edgeExp.More(); edgeExp.Next() )
937 const TopoDS_Edge& edge = TopoDS::Edge( edgeExp.Current() );
938 if ( !_helper.IsSeamShape( edge ) || !treatedEdges.Add( edge ))
941 for ( SMDS_ElemIteratorPtr setIt = sm->GetElements(); setIt->more(); )
943 const SMDS_MeshElement* segment = setIt->next();
944 const SMESH_NodeXYZ pn0 = segment->GetNode( 0 );
945 const SMESH_NodeXYZ pn1 = segment->GetNode( 1 );
947 gp_XY uv0 = _helper.GetNodeUV( face, pn0.Node() );
948 gp_XY uv1 = _helper.GetNodeUV( face, pn1.Node() );
950 for ( int iCoo = 1; iCoo <= 2; ++iCoo )
951 if ( iCoo & _helper.GetPeriodicIndex() )
953 double distParam = Abs( uv0.Coord( iCoo ) - uv1.Coord( iCoo ));
954 if ( distParam > 0.5 * _helper.GetPeriod( iCoo ))
956 double paramOnE; TopoDS_Edge edge2;
957 gp_Pnt edgeIntPnt = getEdgeIntersection( edge, pn0, pn1, edge2, paramOnE );
958 if ( edge2.IsNull() )
961 const SMDS_MeshNode* nodeOnSeam = findNodeOnEdge( edgeIntPnt, edge );
962 bool isOnEdge = nodeOnSeam;
964 int indexOnSegment = 3;
966 nodeOnSeam = findNode( edgeIntPnt, _tol, segment, segInternalNodes, indexOnSegment );
970 nodeOnSeam = _helper.AddNode( edgeIntPnt.X(), edgeIntPnt.Y(), edgeIntPnt.Z() );
975 addNodeOnEdge( nodeOnSeam, edge, paramOnE );
977 if ( indexOnSegment == 3 )
978 segInternalNodes[ segment ].push_back( nodeOnSeam );
985 //================================================================================
987 * \brief Intersect two segments
988 * \param [in] seg1 - segment 1
989 * \param [in] seg2 - segment 2
990 * \param [in] face - the FACE on which segments lie
991 * \param [out] intPnt - intersection point
992 * \param [out] intUV - UV of intersection point on the FACE
993 * \return bool - true if intersection found
995 //================================================================================
997 bool BLSURFPlugin_EnforcedMesh1D::intersectSegments( const SMDS_MeshElement* seg1,
998 const SMDS_MeshElement* seg2,
999 const TopoDS_Face& face,
1001 gp_Pnt2d & intUV ) const
1003 SMESH_NodeXYZ n10 = seg1->GetNode(0);
1004 SMESH_NodeXYZ n11 = seg1->GetNode(1);
1005 SMESH_NodeXYZ n20 = seg2->GetNode(0);
1006 SMESH_NodeXYZ n21 = seg2->GetNode(1);
1007 if ( n10 == n20 || n10 == n21 || n11 == n20 || n10 == n21 )
1010 gp_Lin lin1( n10, n11 - n10 );
1011 gp_Lin lin2( n20, n21 - n20 );
1013 Extrema_ExtElC extrema( lin1, lin2, Precision::Angular() );
1015 if ( !extrema.IsDone() || extrema.IsParallel() )
1018 Extrema_POnCurv poc1, poc2;
1019 extrema.Points( 1, poc1, poc2 );
1021 double len1 = lin1.Direction().XYZ() * ( n11 - n10 );
1022 double len2 = lin2.Direction().XYZ() * ( n21 - n20 );
1023 double u1 = poc1.Parameter();
1024 double u2 = poc2.Parameter();
1026 if ( u1 < -_tol || u1 > len1 + _tol )
1028 if ( u2 < -_tol || u2 > len2 + _tol )
1031 intPnt = 0.5 * ( poc1.Value().XYZ() + poc2.Value().XYZ() );
1033 // compute approximate UV
1038 gp_XY uv10 = _helper.GetNodeUV( face, n10.Node(), n11.Node() );
1039 gp_XY uv11 = _helper.GetNodeUV( face, n11.Node(), n10.Node() );
1040 gp_XY uv1 = uv10 * ( 1 - u1 ) + uv11 * u1;
1042 gp_XY uv20 = _helper.GetNodeUV( face, n20.Node(), n21.Node() );
1043 gp_XY uv21 = _helper.GetNodeUV( face, n21.Node(), n20.Node() );
1044 gp_XY uv2 = uv20 * ( 1 - u2 ) + uv21 * u2;
1046 intUV = 0.5 * ( uv1 + uv2 );
1048 // compute precise UV and XYZ by projecting intPnt to the FACE
1050 Handle(ShapeAnalysis_Surface) surface = _helper.GetSurface( face );
1051 intUV = surface->NextValueOfUV( intUV, intPnt, _tol );
1052 if ( surface->Gap() > Min( len1, len2 ))
1053 intUV = surface->ValueOfUV( intPnt, _tol );
1055 intPnt = surface->Value( intUV );
1060 //================================================================================
1062 * \brief Setup predicates to detect nodes on FACE boundary
1063 * \param [in] shape - shape containing FACEs to mesh
1065 //================================================================================
1067 void BLSURFPlugin_EnforcedMesh1D::setupPredicates( const TopoDS_Shape& theShape )
1069 if ( _onVertexPredicate )
1072 _onVertexPredicate.reset( new TPredicate() );
1073 _onVertexPredicate->SetTolerance( _tol );
1074 _onVertexPredicate->SetMesh( _helper.GetMeshDS() );
1076 TopTools_IndexedMapOfShape vertices;
1077 TopExp::MapShapes( theShape, TopAbs_VERTEX, vertices );
1078 TopoDS_Compound vCompound;
1079 BRep_Builder builder;
1080 builder.MakeCompound( vCompound );
1081 for ( const TopoDS_Shape& v : vertices )
1082 builder.Add( vCompound, v );
1084 _onVertexPredicate->SetShape( vCompound, SMDSAbs_Node );
1088 _onEdgePredicate.reset( new TPredicate() );
1089 _onEdgePredicate->SetTolerance( _tol );
1090 _onEdgePredicate->SetMesh( _helper.GetMeshDS() );
1092 TopTools_IndexedMapOfShape edges;
1093 TopExp::MapShapes( theShape, TopAbs_EDGE, edges );
1094 TopoDS_Compound eCompound;
1095 BRep_Builder builder;
1096 builder.MakeCompound( eCompound );
1097 for ( const TopoDS_Shape& e : edges )
1098 builder.Add( eCompound, e );
1100 _onEdgePredicate->SetShape( eCompound, SMDSAbs_Node );
1105 //================================================================================
1107 * \brief Find or create a group of edges with given name
1109 //================================================================================
1111 SMDS_MeshGroup* BLSURFPlugin_EnforcedMesh1D::getGroup( const std::string& groupName )
1113 SMDS_MeshGroup* group = nullptr;
1114 if ( !groupName.empty() )
1116 if ( _name2Group.count( groupName ))
1117 return _name2Group[ groupName ];
1119 // find existing group
1120 for ( SMESH_Mesh::GroupIteratorPtr grIt = _mesh->GetGroups(); grIt->more(); )
1122 SMESH_Group* grp = grIt->next();
1123 SMESHDS_Group* grpDS = dynamic_cast< SMESHDS_Group* >( grp->GetGroupDS() );
1125 grpDS->GetType() == SMDSAbs_Edge &&
1126 groupName == grp->GetName() )
1128 _name2Group[ groupName ] = & grpDS->SMDSGroup();
1129 return & grpDS->SMDSGroup();
1133 // create a new group
1134 SMESH_Group* grp = _mesh->AddGroup( SMDSAbs_Edge, groupName.c_str() );
1135 SMESHDS_Group* grpDS = static_cast< SMESHDS_Group* >( grp->GetGroupDS() );
1137 group = & grpDS->SMDSGroup();
1138 _name2Group[ groupName ] = group;
1143 //================================================================================
1145 * \brief Look for a node dividing a given EDGE
1147 //================================================================================
1149 const SMDS_MeshNode* BLSURFPlugin_EnforcedMesh1D::findNodeOnEdge( const gp_Pnt& p,
1150 const TopoDS_Edge& edge )
1152 // look for an equal node on the EDGE
1153 if ( std::vector< const SMDS_MeshNode* >* nodesOnE = _nodesOnEdge.ChangeSeek( edge ))
1154 for ( const SMDS_MeshNode* n : *nodesOnE )
1155 if ( p.SquareDistance( SMESH_NodeXYZ( n )) < _tol * _tol )
1163 //================================================================================
1165 * \brief Add a node to an EDGE
1167 //================================================================================
1169 void BLSURFPlugin_EnforcedMesh1D::addNodeOnEdge( const SMDS_MeshNode* node,
1170 const TopoDS_Edge& edge,
1173 _mesh->GetMeshDS()->SetNodeOnEdge( node, edge, u );
1175 std::vector< const SMDS_MeshNode* > * nodesOnE = _nodesOnEdge.ChangeSeek( edge );
1177 nodesOnE = _nodesOnEdge.Bound( edge, std::vector< const SMDS_MeshNode* >() );
1179 nodesOnE->push_back( node );
1182 //================================================================================
1184 * \brief Create EDGEs by dividing a given EDGE by nodes on it
1185 * \param [in] edge - the EDGE to divide
1186 * \param [in] nodes - the nodes to divide by
1188 //================================================================================
1190 void BLSURFPlugin_EnforcedMesh1D::
1192 splitEdgeByNodes( TopoDS_Edge edge, const std::vector< const SMDS_MeshNode* >& nodes )
1194 if ( nodes.empty() )
1197 edge.Orientation( TopAbs_FORWARD );
1199 TopoDS_Vertex v0 = _helper.IthVertex( 0, edge );
1200 TopoDS_Vertex v1 = _helper.IthVertex( 1, edge );
1202 // create VERTEXes and sort them along the EDGE
1204 std::map< double, TopoDS_Vertex > sortedVertices;
1206 BRepAdaptor_Curve curve( edge );
1207 for ( SMESH_NodeXYZ pn : nodes )
1211 ShapeAnalysis_Curve().Project( curve, pn, _tol, projPnt, u, false );
1212 projPnt = curve.Value( u );
1214 TopoDS_Vertex v = BRepBuilderAPI_MakeVertex( projPnt );
1216 sortedVertices.insert({ u, v });
1218 _nOnE2Vertex[ pn.Node() ] = v;
1220 sortedVertices.insert({ BRep_Tool::Parameter( v1, edge ), v1 });
1225 BRep_Builder builder;
1226 std::vector< TopoDS_Edge >& newEdges = *_edgeSplitsOfEdge.Bound( edge, TEdge2Edges::value_type());
1228 double u0 = BRep_Tool::Parameter( v0, edge );
1229 for ( auto& u2v : sortedVertices )
1231 double u1 = u2v.first;
1234 TopoDS_Shape newShape = edge.EmptyCopied();
1235 TopoDS_Edge newEdge = TopoDS::Edge( newShape );
1236 builder.Add( newEdge, v0 );
1237 builder.Add( newEdge, v1 );
1238 builder.Range( newEdge, u0, u1 );
1239 newEdges.push_back( newEdge );
1248 //================================================================================
1250 * \brief Return true if there are enforced segments on a FACE. Start iteration on segments
1252 //================================================================================
1254 bool BLSURFPlugin_EnforcedMesh1D::HasSegmentsOnFace( const TopoDS_Face& face )
1256 _segIterator.reset();
1258 if ( _faces.Contains( face ))
1260 _currentFace = face;
1261 _segIterator = _helper.GetMeshDS()->MeshElements( face )->GetElements();
1263 _helper.SetSubShape( face );
1265 return _segIterator->more();
1271 //================================================================================
1273 * \brief Return next segment on the FACE
1275 //================================================================================
1277 bool BLSURFPlugin_EnforcedMesh1D::NextSegment( Segmemnt & seg,
1278 TopTools_IndexedMapOfShape & vertexTags )
1280 if ( _segIterator && _segIterator->more() )
1282 const SMDS_MeshElement* segment = _segIterator->next();
1284 while ( segment->GetType() != SMDSAbs_Edge && _segIterator->more() )
1285 segment = _segIterator->next();
1286 if ( segment->GetType() != SMDSAbs_Edge )
1289 const SMDS_MeshNode * node[2] = { segment->GetNode(0),
1290 segment->GetNode(1) };
1292 seg._tag = _segTag++;
1294 seg._xyz[0] = SMESH_NodeXYZ( node[0]);
1295 seg._xyz[1] = SMESH_NodeXYZ( node[1]);
1297 seg._uv[0] = _helper.GetNodeUV( _currentFace, node[0], node[1] );
1298 seg._uv[1] = _helper.GetNodeUV( _currentFace, node[1], node[0] );
1300 seg._pcurve = new Geom2d_Line( seg._uv[0], ( seg._uv[1] - seg._uv[0] ));
1303 seg._u[1] = ( seg._uv[1] - seg._uv[0] ).Modulus();
1305 for ( int i = 0; i < 2; ++i )
1307 auto n2v = _nOnE2Vertex.find( node[i] ); // find VERTEX by node
1309 if ( n2v != _nOnE2Vertex.end() )
1310 seg._vTag[i] = vertexTags.Add( n2v->second );
1312 seg._vTag[i] = _nodeTag0 + _nodeTags.Add( node[i] );
1315 if ( !_isQuadratic )
1316 _mesh->GetMeshDS()->UnSetMeshElementOnShape( segment, _currentFace );
1323 //================================================================================
1325 * \brief Return enforced node by the tag that was returned by Segmemnt::_vTag[i]
1327 //================================================================================
1329 const SMDS_MeshNode*
1330 BLSURFPlugin_EnforcedMesh1D::GetNodeByTag( int tag,
1331 const TopTools_IndexedMapOfShape & vertexTags )
1333 const SMDS_MeshNode* node = nullptr;
1335 if ( tag <= vertexTags.Size() && !_nOnE2Vertex.empty() )
1337 const TopoDS_Shape& vertex = vertexTags( tag );
1339 for ( auto n2v = _nOnE2Vertex.begin(); n2v != _nOnE2Vertex.end(); ++n2v )
1340 if ( vertex.IsSame( n2v->second ))
1343 _nOnE2Vertex.erase( n2v );
1350 bool isValid = ( 0 < tag && tag <= _nodeTags.Size() );
1352 node = _nodeTags( tag );
1357 //================================================================================
1359 * \brief Return true if a tag corresponds to the tag of enforced segment
1360 * that was returned by Segment::_tag
1362 //================================================================================
1364 bool BLSURFPlugin_EnforcedMesh1D::IsSegmentTag( int tag ) const
1366 return ( _segTag0 <= tag && tag <= _segTag );
1369 //================================================================================
1371 * \brief Return tag of EDGE by tags of its splits
1373 //================================================================================
1375 int BLSURFPlugin_EnforcedMesh1D::GetTagOfSplitEdge( int splitTag ) const
1377 auto sTag2eTag = _splitTags2EdgeTag.find( splitTag );
1378 return ( sTag2eTag == _splitTags2EdgeTag.end() ) ? splitTag : sTag2eTag->second;