1 // SMESH SMESH : implementaion of SMESH idl descriptions
3 // Copyright (C) 2003 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
24 // File : StdMeshers_Projection_2D.cxx
26 // Created : Fri Oct 20 11:37:07 2006
27 // Author : Edward AGAPOV (eap)
30 #include "StdMeshers_Projection_2D.hxx"
32 #include "StdMeshers_ProjectionSource2D.hxx"
33 #include "StdMeshers_ProjectionUtils.hxx"
35 #include "SMESHDS_Hypothesis.hxx"
36 #include "SMESHDS_SubMesh.hxx"
37 #include "SMESH_Block.hxx"
38 #include "SMESH_Gen.hxx"
39 #include "SMESH_Mesh.hxx"
40 #include "SMESH_MeshEditor.hxx"
41 #include "SMESH_Pattern.hxx"
42 #include "SMESH_subMesh.hxx"
43 #include "SMESH_subMeshEventListener.hxx"
44 #include "SMDS_EdgePosition.hxx"
46 #include "utilities.h"
50 #include <TopTools_ListIteratorOfListOfShape.hxx>
51 #include <BRep_Tool.hxx>
57 #define RETURN_BAD_RESULT(msg) { MESSAGE(msg); return false; }
59 typedef StdMeshers_ProjectionUtils TAssocTool;
61 //=======================================================================
62 //function : StdMeshers_Projection_2D
64 //=======================================================================
66 StdMeshers_Projection_2D::StdMeshers_Projection_2D(int hypId, int studyId, SMESH_Gen* gen)
67 :SMESH_2D_Algo(hypId, studyId, gen)
69 _name = "Projection_2D";
70 _shapeType = (1 << TopAbs_FACE); // 1 bit per shape type
72 _compatibleHypothesis.push_back("ProjectionSource2D");
76 //================================================================================
80 //================================================================================
82 StdMeshers_Projection_2D::~StdMeshers_Projection_2D()
85 //=======================================================================
86 //function : CheckHypothesis
88 //=======================================================================
90 bool StdMeshers_Projection_2D::CheckHypothesis(SMESH_Mesh& theMesh,
91 const TopoDS_Shape& theShape,
92 SMESH_Hypothesis::Hypothesis_Status& theStatus)
94 list <const SMESHDS_Hypothesis * >::const_iterator itl;
96 const list <const SMESHDS_Hypothesis * >&hyps = GetUsedHypothesis(theMesh, theShape);
97 if ( hyps.size() == 0 )
99 theStatus = HYP_MISSING;
100 return false; // can't work with no hypothesis
103 if ( hyps.size() > 1 )
105 theStatus = HYP_ALREADY_EXIST;
109 const SMESHDS_Hypothesis *theHyp = hyps.front();
111 string hypName = theHyp->GetName();
115 if (hypName == "ProjectionSource2D")
117 _sourceHypo = static_cast<const StdMeshers_ProjectionSource2D *>(theHyp);
119 // Check hypo parameters
121 SMESH_Mesh* srcMesh = _sourceHypo->GetSourceMesh();
122 SMESH_Mesh* tgtMesh = & theMesh;
127 if ( _sourceHypo->HasVertexAssociation() )
130 TopoDS_Shape edge = TAssocTool::GetEdgeByVertices
131 ( srcMesh, _sourceHypo->GetSourceVertex(1), _sourceHypo->GetSourceVertex(2) );
132 if ( edge.IsNull() ||
133 !TAssocTool::IsSubShape( edge, srcMesh ) ||
134 !TAssocTool::IsSubShape( edge, _sourceHypo->GetSourceFace() ))
136 SCRUTE((edge.IsNull()));
137 SCRUTE((TAssocTool::IsSubShape( edge, srcMesh )));
138 SCRUTE((TAssocTool::IsSubShape( edge, _sourceHypo->GetSourceFace() )));
139 theStatus = HYP_BAD_PARAMETER;
144 edge = TAssocTool::GetEdgeByVertices
145 ( tgtMesh, _sourceHypo->GetTargetVertex(1), _sourceHypo->GetTargetVertex(2) );
146 if ( edge.IsNull() ||
147 !TAssocTool::IsSubShape( edge, tgtMesh ) ||
148 !TAssocTool::IsSubShape( edge, theShape ))
150 SCRUTE((edge.IsNull()));
151 SCRUTE((TAssocTool::IsSubShape( edge, tgtMesh )));
152 SCRUTE((TAssocTool::IsSubShape( edge, theShape )));
153 theStatus = HYP_BAD_PARAMETER;
157 // check a source face
158 if ( !TAssocTool::IsSubShape( _sourceHypo->GetSourceFace(), srcMesh )) {
159 MESSAGE("Bad source face");
160 theStatus = HYP_BAD_PARAMETER;
165 theStatus = HYP_INCOMPATIBLE;
167 return ( theStatus == HYP_OK );
173 //================================================================================
175 * \brief define if a node is new or old
176 * \param node - node to check
177 * \retval bool - true if the node existed before Compute() is called
179 //================================================================================
181 bool isOldNode( const SMDS_MeshNode* node )
183 // old nodes are shared by edges and new ones are shared
184 // only by faces created by mapper
186 SMDS_ElemIteratorPtr invElem = node->GetInverseElementIterator();
187 while ( !isOld && invElem->more() )
188 isOld = ( invElem->next()->GetType() == SMDSAbs_Edge );
192 //================================================================================
194 * \brief Class to remove mesh built by pattern mapper on edges
195 * and vertices in the case of failure of projection algo.
196 * It does it's job at destruction
198 //================================================================================
203 MeshCleaner( SMESH_subMesh* faceSubMesh ): sm(faceSubMesh) {}
204 ~MeshCleaner() { Clean(sm); }
205 void Release() { sm = 0; } // mesh will not be removed
206 static void Clean( SMESH_subMesh* sm )
209 switch ( sm->GetSubShape().ShapeType() ) {
212 SMDS_NodeIteratorPtr nIt = sm->GetSubMeshDS()->GetNodes();
213 SMESHDS_Mesh* mesh = sm->GetFather()->GetMeshDS();
214 while ( nIt->more() ) {
215 const SMDS_MeshNode* node = nIt->next();
216 if ( !isOldNode( node ) )
217 mesh->RemoveNode( node );
219 // do not break but iterate over DependsOn()
222 const map< int, SMESH_subMesh * >& subSM = sm->DependsOn();
223 map< int, SMESH_subMesh * >::const_iterator i_sm = subSM.begin();
224 for ( ; i_sm != subSM.end(); ++i_sm )
225 Clean( i_sm->second );
230 //================================================================================
232 * \brief find new nodes belonging to one free border of mesh on face
233 * \param sm - submesh on edge or vertex containg nodes to choose from
234 * \param face - the face bound the submesh
235 * \param u2nodes - map to fill with nodes
236 * \param seamNodes - set of found nodes
237 * \retval bool - is a success
239 //================================================================================
241 bool getBoundaryNodes ( SMESH_subMesh* sm,
242 const TopoDS_Face& face,
243 map< double, const SMDS_MeshNode* > & u2nodes,
244 set< const SMDS_MeshNode* > & seamNodes)
248 if ( !sm || !sm->GetSubMeshDS() )
249 RETURN_BAD_RESULT("Null submesh");
251 SMDS_NodeIteratorPtr nIt = sm->GetSubMeshDS()->GetNodes();
252 switch ( sm->GetSubShape().ShapeType() ) {
254 case TopAbs_VERTEX: {
255 while ( nIt->more() ) {
256 const SMDS_MeshNode* node = nIt->next();
257 if ( isOldNode( node ) ) continue;
258 u2nodes.insert( make_pair( 0., node ));
259 seamNodes.insert( node );
266 // Get submeshes of sub-vertices
267 const map< int, SMESH_subMesh * >& subSM = sm->DependsOn();
268 if ( subSM.size() != 2 )
269 RETURN_BAD_RESULT("there must be 2 submeshes of sub-vertices"
270 " but we have " << subSM.size());
271 SMESH_subMesh* smV1 = subSM.begin()->second;
272 SMESH_subMesh* smV2 = subSM.rbegin()->second;
273 if ( !smV1->IsMeshComputed() || !smV2->IsMeshComputed() )
274 RETURN_BAD_RESULT("Empty vertex submeshes");
276 // Look for a new node on V1
277 nIt = smV1->GetSubMeshDS()->GetNodes();
278 const SMDS_MeshNode* nV1 = 0;
279 while ( nIt->more() && !nV1 ) {
280 const SMDS_MeshNode* node = nIt->next();
281 if ( !isOldNode( node ) ) nV1 = node;
284 RETURN_BAD_RESULT("No new node found on V1");
286 // Find a new node connected to nV1 and belonging to edge submesh;
287 const SMDS_MeshNode* nE = 0;
288 SMESHDS_SubMesh* smDS = sm->GetSubMeshDS();
289 SMDS_ElemIteratorPtr vElems = nV1->GetInverseElementIterator();
290 while ( vElems->more() && !nE ) {
291 const SMDS_MeshElement* elem = vElems->next();
292 if ( elem->GetType() != SMDSAbs_Face )
293 continue; // new nodes are shared by faces
294 int nbNodes = elem->NbNodes();
295 if ( elem->IsQuadratic() )
297 int iV1 = elem->GetNodeIndex( nV1 );
298 // try next aftre nV1
299 int iE = SMESH_MesherHelper::WrapIndex( iV1 + 1, nbNodes );
300 if ( smDS->Contains( elem->GetNode( iE ) ))
301 nE = elem->GetNode( iE );
303 // try node before nV1
304 iE = SMESH_MesherHelper::WrapIndex( iV1 - 1, nbNodes );
305 if ( smDS->Contains( elem->GetNode( iE )))
306 nE = elem->GetNode( iE );
308 if ( nE && elem->IsQuadratic() ) { // find medium node between nV1 and nE
309 if ( Abs( iV1 - iE ) == 1 )
310 nE = elem->GetNode( Min ( iV1, iE ) + nbNodes );
312 nE = elem->GetNode( elem->NbNodes() - 1 );
316 RETURN_BAD_RESULT("new node on edge not found");
318 // Get the whole free border of a face
319 list< const SMDS_MeshNode* > bordNodes;
320 list< const SMDS_MeshElement* > bordFaces;
321 if ( !SMESH_MeshEditor::FindFreeBorder (nV1, nE, nV1, bordNodes, bordFaces ))
322 RETURN_BAD_RESULT("free border of a face not found by nodes " <<
323 nV1->GetID() << " " << nE->GetID() );
325 // Insert nodes of the free border to the map until node on V2 encountered
326 SMESHDS_SubMesh* v2smDS = smV2->GetSubMeshDS();
327 list< const SMDS_MeshNode* >::iterator bordIt = bordNodes.begin();
328 bordIt++; // skip nV1
329 for ( ; bordIt != bordNodes.end(); ++bordIt ) {
330 const SMDS_MeshNode* node = *bordIt;
331 if ( v2smDS->Contains( node ))
333 if ( node->GetPosition()->GetTypeOfPosition() != SMDS_TOP_EDGE )
334 RETURN_BAD_RESULT("Bad node position type: node " << node->GetID() <<
335 " pos type " << node->GetPosition()->GetTypeOfPosition());
336 const SMDS_EdgePosition* pos =
337 static_cast<const SMDS_EdgePosition*>(node->GetPosition().get());
338 u2nodes.insert( make_pair( pos->GetUParameter(), node ));
339 seamNodes.insert( node );
341 if ( u2nodes.size() != seamNodes.size() )
342 RETURN_BAD_RESULT("Bad node params on edge " << sm->GetId() <<
343 ", " << u2nodes.size() << " != " << seamNodes.size() );
348 RETURN_BAD_RESULT ("Unexpected submesh type");
350 } // bool getBoundaryNodes()
354 //=======================================================================
357 //=======================================================================
359 bool StdMeshers_Projection_2D::Compute(SMESH_Mesh& theMesh, const TopoDS_Shape& theShape)
364 TopoDS_Face tgtFace = TopoDS::Face( theShape.Oriented(TopAbs_FORWARD));
365 TopoDS_Face srcFace = TopoDS::Face( _sourceHypo->GetSourceFace().Oriented(TopAbs_FORWARD));
367 SMESH_Mesh * srcMesh = _sourceHypo->GetSourceMesh();
368 SMESH_Mesh * tgtMesh = & theMesh;
372 SMESHDS_Mesh * meshDS = theMesh.GetMeshDS();
374 SMESH_MesherHelper helper( theMesh );
375 helper.SetSubShape( tgtFace );
377 // ---------------------------
378 // Make subshapes association
379 // ---------------------------
381 TAssocTool::TShapeShapeMap shape2ShapeMap;
382 TAssocTool::InitVertexAssociation( _sourceHypo, shape2ShapeMap );
383 if ( !TAssocTool::FindSubShapeAssociation( tgtFace, tgtMesh, srcFace, srcMesh,
385 RETURN_BAD_RESULT("FindSubShapeAssociation failed");
387 // ----------------------------------------------
388 // Assure that mesh on a source Face is computed
389 // ----------------------------------------------
391 SMESH_subMesh* srcSubMesh = srcMesh->GetSubMesh( srcFace );
392 SMESH_subMesh* tgtSubMesh = tgtMesh->GetSubMesh( tgtFace );
394 if ( tgtMesh == srcMesh ) {
395 if ( !TAssocTool::MakeComputed( srcSubMesh ))
396 RETURN_BAD_RESULT("Impossible to compute the source mesh");
399 if ( !srcSubMesh->IsMeshComputed() )
400 RETURN_BAD_RESULT("Source mesh is not computed");
403 // --------------------
404 // Prepare to mapping
405 // --------------------
407 // Load pattern from the source face
408 SMESH_Pattern mapper;
409 mapper.Load( srcMesh, srcFace );
410 if ( mapper.GetErrorCode() != SMESH_Pattern::ERR_OK )
411 RETURN_BAD_RESULT("SMESH_Pattern::Load() failed");
413 // Find the first target vertex corresponding to first vertex of the <mapper>
414 // and <theReverse> flag needed to call mapper.Apply()
416 TopoDS_Vertex srcV1 = TopoDS::Vertex( mapper.GetSubShape( 1 ));
417 if ( srcV1.IsNull() )
418 RETURN_BAD_RESULT("Mesh is not bound to the face");
419 if ( !shape2ShapeMap.IsBound( srcV1 ))
420 RETURN_BAD_RESULT("Not associated vertices, srcV1 " << srcV1.TShape().operator->() );
421 TopoDS_Vertex tgtV1 = TopoDS::Vertex( shape2ShapeMap( srcV1 ));
423 if ( !TAssocTool::IsSubShape( srcV1, srcFace ))
424 RETURN_BAD_RESULT("Wrong srcV1 " << srcV1.TShape().operator->());
425 if ( !TAssocTool::IsSubShape( tgtV1, tgtFace ))
426 RETURN_BAD_RESULT("Wrong tgtV1 " << tgtV1.TShape().operator->());
428 // try to find out orientation by order of edges
429 bool reverse = false;
430 list< TopoDS_Edge > tgtEdges, srcEdges;
431 list< int > nbEdgesInWires;
432 SMESH_Block::GetOrderedEdges( tgtFace, tgtV1, tgtEdges, nbEdgesInWires);
433 SMESH_Block::GetOrderedEdges( srcFace, srcV1, srcEdges, nbEdgesInWires);
434 if ( nbEdgesInWires.front() > 1 ) // possible to find out
436 TopoDS_Edge srcE1 = srcEdges.front(), tgtE1 = tgtEdges.front();
437 reverse = ( ! srcE1.IsSame( shape2ShapeMap( tgtE1 )));
438 if ( BRep_Tool::IsClosed( tgtE1, tgtFace )) {
439 reverse = ( srcE1.Orientation() == tgtE1.Orientation() );
440 if ( _sourceHypo->GetSourceFace().Orientation() != theShape.Orientation() )
444 else if ( nbEdgesInWires.front() == 1 )
446 // TODO::Compare orientation of curves in a sole edge
447 //RETURN_BAD_RESULT("Not implemented case");
451 RETURN_BAD_RESULT("Bad result from SMESH_Block::GetOrderedEdges()");
454 // --------------------
455 // Perform 2D mapping
456 // --------------------
458 // Compute mesh on a target face
460 mapper.Apply( tgtFace, tgtV1, reverse );
461 if ( mapper.GetErrorCode() != SMESH_Pattern::ERR_OK )
462 RETURN_BAD_RESULT("SMESH_Pattern::Apply() failed");
466 const bool toCreatePolygons = false, toCreatePolyedrs = false;
467 mapper.MakeMesh( tgtMesh, toCreatePolygons, toCreatePolyedrs );
468 if ( mapper.GetErrorCode() != SMESH_Pattern::ERR_OK )
469 RETURN_BAD_RESULT("SMESH_Pattern::MakeMesh() failed");
471 // it will remove mesh built by pattern mapper on edges and vertices
473 MeshCleaner cleaner( tgtSubMesh );
475 // -------------------------------------------------------------------------
476 // mapper doesn't take care of nodes already existing on edges and vertices,
477 // so we must merge nodes created by it with existing ones
478 // -------------------------------------------------------------------------
480 SMESH_MeshEditor editor( tgtMesh );
481 SMESH_MeshEditor::TListOfListOfNodes groupsOfNodes;
483 // Make groups of nodes to merge
485 // loop on edge and vertex submeshes of a target face
486 const map< int, SMESH_subMesh * >& subSM = tgtSubMesh->DependsOn();
487 map< int, SMESH_subMesh * >::const_iterator i_subSM = subSM.begin();
488 for ( ; i_subSM != subSM.end(); ++i_subSM )
490 SMESH_subMesh* sm = i_subSM->second;
491 SMESHDS_SubMesh* smDS = sm->GetSubMeshDS();
493 // Sort new and old nodes of a submesh separately
495 bool isSeam = helper.IsSeamShape( sm->GetId() );
497 enum { NEW_NODES, OLD_NODES };
498 map< double, const SMDS_MeshNode* > u2nodesMaps[2], u2nodesOnSeam;
499 map< double, const SMDS_MeshNode* >::iterator u_oldNode, u_newNode, u_newOnSeam, newEnd;
500 set< const SMDS_MeshNode* > seamNodes;
502 // mapper puts on a seam edge nodes from 2 edges
503 if ( isSeam && ! getBoundaryNodes ( sm, tgtFace, u2nodesOnSeam, seamNodes ))
504 RETURN_BAD_RESULT("getBoundaryNodes() failed");
506 SMDS_NodeIteratorPtr nIt = smDS->GetNodes();
507 while ( nIt->more() )
509 const SMDS_MeshNode* node = nIt->next();
510 bool isOld = isOldNode( node );
512 if ( !isOld && isSeam ) { // new node on a seam edge
513 if ( seamNodes.find( node ) != seamNodes.end())
514 continue; // node is already in the map
517 // sort nodes on edges by its position
518 map< double, const SMDS_MeshNode* > & pos2nodes = u2nodesMaps[ isOld ];
519 switch ( node->GetPosition()->GetTypeOfPosition() )
521 case SMDS_TOP_VERTEX: {
522 pos2nodes.insert( make_pair( 0, node ));
525 case SMDS_TOP_EDGE: {
526 const SMDS_EdgePosition* pos =
527 static_cast<const SMDS_EdgePosition*>(node->GetPosition().get());
528 pos2nodes.insert( make_pair( pos->GetUParameter(), node ));
532 RETURN_BAD_RESULT("Wrong node position type: "<<
533 node->GetPosition()->GetTypeOfPosition());
536 if ( u2nodesMaps[ OLD_NODES ].size() != u2nodesMaps[ NEW_NODES ].size() )
537 RETURN_BAD_RESULT("Different nb of old and new nodes " <<
538 u2nodesMaps[ OLD_NODES ].size() << " != " <<
539 u2nodesMaps[ NEW_NODES ].size());
540 if ( isSeam && u2nodesMaps[ OLD_NODES ].size() != u2nodesOnSeam.size() )
541 RETURN_BAD_RESULT("Different nb of old and seam nodes " <<
542 u2nodesMaps[ OLD_NODES ].size() << " != " << u2nodesOnSeam.size());
544 // Make groups of nodes to merge
545 u_oldNode = u2nodesMaps[ OLD_NODES ].begin();
546 u_newNode = u2nodesMaps[ NEW_NODES ].begin();
547 newEnd = u2nodesMaps[ NEW_NODES ].end();
548 u_newOnSeam = u2nodesOnSeam.begin();
549 for ( ; u_newNode != newEnd; ++u_newNode, ++u_oldNode ) {
550 groupsOfNodes.push_back( list< const SMDS_MeshNode* >() );
551 groupsOfNodes.back().push_back( u_oldNode->second );
552 groupsOfNodes.back().push_back( u_newNode->second );
554 groupsOfNodes.back().push_back( (u_newOnSeam++)->second );
560 editor.MergeNodes( groupsOfNodes );
562 // ---------------------------
563 // Check elements orientation
564 // ---------------------------
566 TopoDS_Face face = tgtFace;
567 if ( !theMesh.IsMainShape( tgtFace ))
569 // find the main shape
570 TopoDS_Shape mainShape = meshDS->ShapeToMesh();
571 switch ( mainShape.ShapeType() ) {
573 case TopAbs_SOLID: break;
575 TopTools_ListIteratorOfListOfShape ancestIt = theMesh.GetAncestors( face );
576 for ( ; ancestIt.More(); ancestIt.Next() ) {
577 TopAbs_ShapeEnum type = ancestIt.Value().ShapeType();
578 if ( type == TopAbs_SOLID ) {
579 mainShape = ancestIt.Value();
581 } else if ( type == TopAbs_SHELL ) {
582 mainShape = ancestIt.Value();
586 // find tgtFace in the main solid or shell to know it's true orientation.
587 TopExp_Explorer exp( mainShape, TopAbs_FACE );
588 for ( ; exp.More(); exp.Next() ) {
589 if ( tgtFace.IsSame( exp.Current() )) {
590 face = TopoDS::Face( exp.Current() );
596 if ( SMESH_Algo::IsReversedSubMesh( face, meshDS ))
598 SMDS_ElemIteratorPtr eIt = meshDS->MeshElements( face )->GetElements();
599 while ( eIt->more() ) {
600 const SMDS_MeshElement* e = eIt->next();
601 if ( e->GetType() == SMDSAbs_Face && !editor.Reorient( e ))
602 RETURN_BAD_RESULT("Pb of SMESH_MeshEditor::Reorient()");
606 cleaner.Release(); // do not remove mesh
611 //=============================================================================
613 * \brief Sets a default event listener to submesh of the source face
614 * \param subMesh - submesh where algo is set
616 * This method is called when a submesh gets HYP_OK algo_state.
617 * After being set, event listener is notified on each event of a submesh.
618 * Arranges that CLEAN event is translated from source submesh to
621 //=============================================================================
623 void StdMeshers_Projection_2D::SetEventListener(SMESH_subMesh* subMesh)
625 if ( _sourceHypo && ! _sourceHypo->GetSourceFace().IsNull() )
627 SMESH_Mesh* srcMesh = _sourceHypo->GetSourceMesh();
629 srcMesh = subMesh->GetFather();
631 SMESH_subMesh* srcFaceSM =
632 srcMesh->GetSubMesh( _sourceHypo->GetSourceFace() );
634 subMesh->SetEventListener( new SMESH_subMeshEventListener(),
635 SMESH_subMeshEventListenerData::MakeData( subMesh ),