Salome HOME
correct previous integration (Porting to Python 2.6)
[modules/smesh.git] / src / StdMeshers / StdMeshers_Projection_2D.cxx
1 //  Copyright (C) 2007-2008  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 //  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
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.
10 //
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.
15 //
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
19 //
20 //  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22 //  SMESH SMESH : implementaion of SMESH idl descriptions
23 // File      : StdMeshers_Projection_2D.cxx
24 // Module    : SMESH
25 // Created   : Fri Oct 20 11:37:07 2006
26 // Author    : Edward AGAPOV (eap)
27 //
28 #include "StdMeshers_Projection_2D.hxx"
29
30 #include "StdMeshers_ProjectionSource2D.hxx"
31 #include "StdMeshers_ProjectionUtils.hxx"
32
33 #include "SMESHDS_Hypothesis.hxx"
34 #include "SMESHDS_SubMesh.hxx"
35 #include "SMESH_Block.hxx"
36 #include "SMESH_Gen.hxx"
37 #include "SMESH_Mesh.hxx"
38 #include "SMESH_MesherHelper.hxx"
39 #include "SMESH_Pattern.hxx"
40 #include "SMESH_subMesh.hxx"
41 #include "SMESH_subMeshEventListener.hxx"
42 #include "SMESH_Comment.hxx"
43 #include "SMDS_EdgePosition.hxx"
44
45 #include "utilities.h"
46
47 #include <BRep_Tool.hxx>
48 #include <Bnd_B2d.hxx>
49 #include <TopExp.hxx>
50 #include <TopExp_Explorer.hxx>
51 #include <TopTools_ListIteratorOfListOfShape.hxx>
52 #include <TopoDS.hxx>
53
54
55 using namespace std;
56
57 #define RETURN_BAD_RESULT(msg) { MESSAGE(")-: Error: " << msg); return false; }
58
59 typedef StdMeshers_ProjectionUtils TAssocTool;
60
61 //=======================================================================
62 //function : StdMeshers_Projection_2D
63 //purpose  : 
64 //=======================================================================
65
66 StdMeshers_Projection_2D::StdMeshers_Projection_2D(int hypId, int studyId, SMESH_Gen* gen)
67   :SMESH_2D_Algo(hypId, studyId, gen)
68 {
69   _name = "Projection_2D";
70   _shapeType = (1 << TopAbs_FACE);      // 1 bit per shape type
71
72   _compatibleHypothesis.push_back("ProjectionSource2D");
73   _sourceHypo = 0;
74 }
75
76 //================================================================================
77 /*!
78  * \brief Destructor
79  */
80 //================================================================================
81
82 StdMeshers_Projection_2D::~StdMeshers_Projection_2D()
83 {}
84
85 //=======================================================================
86 //function : CheckHypothesis
87 //purpose  : 
88 //=======================================================================
89
90 bool StdMeshers_Projection_2D::CheckHypothesis(SMESH_Mesh&                          theMesh,
91                                                const TopoDS_Shape&                  theShape,
92                                                SMESH_Hypothesis::Hypothesis_Status& theStatus)
93 {
94   list <const SMESHDS_Hypothesis * >::const_iterator itl;
95
96   const list <const SMESHDS_Hypothesis * >&hyps = GetUsedHypothesis(theMesh, theShape);
97   if ( hyps.size() == 0 )
98   {
99     theStatus = HYP_MISSING;
100     return false;  // can't work with no hypothesis
101   }
102
103   if ( hyps.size() > 1 )
104   {
105     theStatus = HYP_ALREADY_EXIST;
106     return false;
107   }
108
109   const SMESHDS_Hypothesis *theHyp = hyps.front();
110
111   string hypName = theHyp->GetName();
112
113   theStatus = HYP_OK;
114
115   if (hypName == "ProjectionSource2D")
116   {
117     _sourceHypo = static_cast<const StdMeshers_ProjectionSource2D *>(theHyp);
118
119     // Check hypo parameters
120
121     SMESH_Mesh* srcMesh = _sourceHypo->GetSourceMesh();
122     SMESH_Mesh* tgtMesh = & theMesh;
123     if ( !srcMesh )
124       srcMesh = tgtMesh;
125
126     // check vertices
127     if ( _sourceHypo->HasVertexAssociation() )
128     {
129       // source vertices
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() ))
135       {
136         theStatus = HYP_BAD_PARAMETER;
137         SCRUTE((edge.IsNull()));
138         SCRUTE((TAssocTool::IsSubShape( edge, srcMesh )));
139         SCRUTE((TAssocTool::IsSubShape( edge, _sourceHypo->GetSourceFace() )));
140       }
141       else
142       {
143         // target vertices
144         edge = TAssocTool::GetEdgeByVertices
145           ( tgtMesh, _sourceHypo->GetTargetVertex(1), _sourceHypo->GetTargetVertex(2) );
146         if ( edge.IsNull() || !TAssocTool::IsSubShape( edge, tgtMesh ))
147         {
148           theStatus = HYP_BAD_PARAMETER;
149           SCRUTE((edge.IsNull()));
150           SCRUTE((TAssocTool::IsSubShape( edge, tgtMesh )));
151         }
152         // PAL16203
153         else if ( !_sourceHypo->IsCompoundSource() &&
154                   !TAssocTool::IsSubShape( edge, theShape ))
155         {
156           theStatus = HYP_BAD_PARAMETER;
157           SCRUTE((TAssocTool::IsSubShape( edge, theShape )));
158         }
159       }
160     }
161     // check a source face
162     if ( !TAssocTool::IsSubShape( _sourceHypo->GetSourceFace(), srcMesh ) ||
163          ( srcMesh == tgtMesh && theShape == _sourceHypo->GetSourceFace() ))
164     {
165       theStatus = HYP_BAD_PARAMETER;
166       SCRUTE((TAssocTool::IsSubShape( _sourceHypo->GetSourceFace(), srcMesh )));
167       SCRUTE((srcMesh == tgtMesh));
168       SCRUTE(( theShape == _sourceHypo->GetSourceFace() ));
169     }
170   }
171   else
172   {
173     theStatus = HYP_INCOMPATIBLE;
174   }
175   return ( theStatus == HYP_OK );
176 }
177
178 namespace {
179
180
181   //================================================================================
182   /*!
183    * \brief define if a node is new or old
184     * \param node - node to check
185     * \retval bool - true if the node existed before Compute() is called
186    */
187   //================================================================================
188
189   bool isOldNode( const SMDS_MeshNode* node )
190   {
191     // old nodes are shared by edges and new ones are shared
192     // only by faces created by mapper
193     SMDS_ElemIteratorPtr invEdge = node->GetInverseElementIterator(SMDSAbs_Edge);
194     bool isOld = invEdge->more();
195     return isOld;
196   }
197
198   //================================================================================
199   /*!
200    * \brief Class to remove mesh built by pattern mapper on edges
201    * and vertices in the case of failure of projection algo.
202    * It does it's job at destruction
203    */
204   //================================================================================
205
206   class MeshCleaner {
207     SMESH_subMesh* sm;
208   public:
209     MeshCleaner( SMESH_subMesh* faceSubMesh ): sm(faceSubMesh) {}
210     ~MeshCleaner() { Clean(sm); }
211     void Release() { sm = 0; } // mesh will not be removed
212     static void Clean( SMESH_subMesh* sm, bool withSub=true )
213     {
214       if ( !sm ) return;
215       // PAL16567, 18920. Remove face nodes as well
216 //       switch ( sm->GetSubShape().ShapeType() ) {
217 //       case TopAbs_VERTEX:
218 //       case TopAbs_EDGE: {
219         SMDS_NodeIteratorPtr nIt = sm->GetSubMeshDS()->GetNodes();
220         SMESHDS_Mesh* mesh = sm->GetFather()->GetMeshDS();
221         while ( nIt->more() ) {
222           const SMDS_MeshNode* node = nIt->next();
223           if ( !isOldNode( node ) )
224             mesh->RemoveNode( node );
225         }
226         // do not break but iterate over DependsOn()
227 //       }
228 //       default:
229         if ( !withSub ) return;
230         SMESH_subMeshIteratorPtr smIt = sm->getDependsOnIterator(false,false);
231         while ( smIt->more() )
232           Clean( smIt->next(), false );
233 //       }
234     }
235   };
236
237   //================================================================================
238   /*!
239    * \brief find new nodes belonging to one free border of mesh on face
240     * \param sm - submesh on edge or vertex containg nodes to choose from
241     * \param face - the face bound the submesh
242     * \param u2nodes - map to fill with nodes
243     * \param seamNodes - set of found nodes
244     * \retval bool - is a success
245    */
246   //================================================================================
247
248   bool getBoundaryNodes ( SMESH_subMesh*                        sm,
249                           const TopoDS_Face&                    face,
250                           map< double, const SMDS_MeshNode* > & u2nodes,
251                           set< const SMDS_MeshNode* > &         seamNodes)
252   {
253     u2nodes.clear();
254     seamNodes.clear();
255     if ( !sm || !sm->GetSubMeshDS() )
256       RETURN_BAD_RESULT("Null submesh");
257
258     SMDS_NodeIteratorPtr nIt = sm->GetSubMeshDS()->GetNodes();
259     switch ( sm->GetSubShape().ShapeType() ) {
260
261     case TopAbs_VERTEX: {
262       while ( nIt->more() ) {
263         const SMDS_MeshNode* node = nIt->next();
264         if ( isOldNode( node ) ) continue;
265         u2nodes.insert( make_pair( 0., node ));
266         seamNodes.insert( node );
267         return true;
268       }
269       break;
270     }
271     case TopAbs_EDGE: {
272       
273       // Get submeshes of sub-vertices
274       const map< int, SMESH_subMesh * >& subSM = sm->DependsOn();
275       if ( subSM.size() != 2 )
276         RETURN_BAD_RESULT("there must be 2 submeshes of sub-vertices"
277                           " but we have " << subSM.size());
278       SMESH_subMesh* smV1 = subSM.begin()->second;
279       SMESH_subMesh* smV2 = subSM.rbegin()->second;
280       if ( !smV1->IsMeshComputed() || !smV2->IsMeshComputed() )
281         RETURN_BAD_RESULT("Empty vertex submeshes");
282
283       // Look for a new node on V1
284       nIt = smV1->GetSubMeshDS()->GetNodes();
285       const SMDS_MeshNode* nV1 = 0;
286       while ( nIt->more() && !nV1 ) {
287         const SMDS_MeshNode* node = nIt->next();
288         if ( !isOldNode( node ) ) nV1 = node;
289       }
290       if ( !nV1 )
291         RETURN_BAD_RESULT("No new node found on V1");
292
293       // Find a new node connected to nV1 and belonging to edge submesh;
294       const SMDS_MeshNode* nE = 0;
295       SMESHDS_SubMesh* smDS = sm->GetSubMeshDS();
296       SMDS_ElemIteratorPtr vElems = nV1->GetInverseElementIterator(SMDSAbs_Face);
297       while ( vElems->more() && !nE ) {
298         const SMDS_MeshElement* elem = vElems->next();
299         int nbNodes = elem->NbNodes();
300         if ( elem->IsQuadratic() )
301           nbNodes /= 2;
302         int iV1 = elem->GetNodeIndex( nV1 );
303         // try next after nV1
304         int iE = SMESH_MesherHelper::WrapIndex( iV1 + 1, nbNodes );
305         if ( smDS->Contains( elem->GetNode( iE ) ))
306           nE = elem->GetNode( iE );
307         if ( !nE ) {
308           // try node before nV1
309           iE = SMESH_MesherHelper::WrapIndex( iV1 - 1, nbNodes );
310           if ( smDS->Contains( elem->GetNode( iE )))
311             nE = elem->GetNode( iE );
312         }
313         if ( nE && elem->IsQuadratic() ) { // find medium node between nV1 and nE
314           if ( Abs( iV1 - iE ) == 1 )
315             nE = elem->GetNode( Min ( iV1, iE ) + nbNodes );
316           else
317             nE = elem->GetNode( elem->NbNodes() - 1 );
318         }
319       }
320       if ( !nE )
321         RETURN_BAD_RESULT("new node on edge not found");
322
323       // Get the whole free border of a face
324       list< const SMDS_MeshNode* > bordNodes;
325       list< const SMDS_MeshElement* > bordFaces;
326       if ( !SMESH_MeshEditor::FindFreeBorder (nV1, nE, nV1, bordNodes, bordFaces ))
327         RETURN_BAD_RESULT("free border of a face not found by nodes " <<
328                           nV1->GetID() << " " << nE->GetID() );
329
330       // Insert nodes of the free border to the map until node on V2 encountered
331       SMESHDS_SubMesh* v2smDS = smV2->GetSubMeshDS();
332       list< const SMDS_MeshNode* >::iterator bordIt = bordNodes.begin();
333       bordIt++; // skip nV1
334       for ( ; bordIt != bordNodes.end(); ++bordIt ) {
335         const SMDS_MeshNode* node = *bordIt;
336         if ( v2smDS->Contains( node ))
337           break;
338         if ( node->GetPosition()->GetTypeOfPosition() != SMDS_TOP_EDGE )
339           RETURN_BAD_RESULT("Bad node position type: node " << node->GetID() <<
340                             " pos type " << node->GetPosition()->GetTypeOfPosition());
341         const SMDS_EdgePosition* pos =
342           static_cast<const SMDS_EdgePosition*>(node->GetPosition().get());
343         u2nodes.insert( make_pair( pos->GetUParameter(), node ));
344         seamNodes.insert( node );
345       }
346       if ( u2nodes.size() != seamNodes.size() )
347         RETURN_BAD_RESULT("Bad node params on edge " << sm->GetId() <<
348                           ", " << u2nodes.size() << " != " << seamNodes.size() );
349       return true;
350     }
351     default:;
352     }
353     RETURN_BAD_RESULT ("Unexpected submesh type");
354
355   } // bool getBoundaryNodes()
356
357   //================================================================================
358   /*!
359    * \brief Preform projection in case if tgtFace.IsPartner( srcFace )
360    *  \param tgtFace - target face
361    *  \param srcFace - source face
362    *  \param tgtMesh - target mesh
363    *  \param srcMesh - source mesh
364    *  \retval bool - true if succeeded
365    */
366   //================================================================================
367
368   bool projectPartner(const TopoDS_Face&                tgtFace,
369                       const TopoDS_Face&                srcFace,
370                       SMESH_Mesh *                      tgtMesh,
371                       SMESH_Mesh *                      srcMesh,
372                       const TAssocTool::TShapeShapeMap& shape2ShapeMap)
373   {
374     if ( !tgtFace.IsPartner( srcFace ))
375       return false;
376
377     // Fill map of src to tgt nodes with nodes on edges
378
379     map<const SMDS_MeshNode* , const SMDS_MeshNode*> src2tgtNodes;
380     map<const SMDS_MeshNode* , const SMDS_MeshNode*>::iterator srcN_tgtN;
381
382     for ( TopExp_Explorer srcEdge( srcFace, TopAbs_EDGE); srcEdge.More(); srcEdge.Next() )
383     {
384       const TopoDS_Shape& tgtEdge = shape2ShapeMap( srcEdge.Current() );
385       if ( !tgtEdge.IsPartner( srcEdge.Current() ))
386         return false;
387
388       map< double, const SMDS_MeshNode* > srcNodes, tgtNodes;
389       if ( !SMESH_Algo::GetSortedNodesOnEdge( srcMesh->GetMeshDS(),
390                                               TopoDS::Edge( srcEdge.Current() ),
391                                               /*ignoreMediumNodes = */true,
392                                               srcNodes )
393            ||
394            !SMESH_Algo::GetSortedNodesOnEdge( tgtMesh->GetMeshDS(),
395                                               TopoDS::Edge( tgtEdge ),
396                                               /*ignoreMediumNodes = */true,
397                                               tgtNodes )
398            ||
399            srcNodes.size() != tgtNodes.size())
400         return false;
401
402       map< double, const SMDS_MeshNode* >::iterator u_tn = tgtNodes.begin();
403       map< double, const SMDS_MeshNode* >::iterator u_sn = srcNodes.begin();
404       for ( ; u_tn != tgtNodes.end(); ++u_tn, ++u_sn)
405         src2tgtNodes.insert( make_pair( u_sn->second, u_tn->second ));
406     }
407
408     // Make new faces
409
410     // transformation to get location of target nodes from source ones
411     gp_Trsf srcTrsf = srcFace.Location();
412     gp_Trsf tgtTrsf = tgtFace.Location();
413     gp_Trsf trsf = srcTrsf.Inverted() * tgtTrsf;
414
415     // prepare the helper adding quadratic elements if necessary
416     SMESH_MesherHelper helper( *tgtMesh );
417     helper.IsQuadraticSubMesh( tgtFace );
418     helper.SetElementsOnShape( true );
419
420     const SMDS_MeshNode* nullNode = 0;
421
422     SMESHDS_SubMesh* srcSubDS = srcMesh->GetMeshDS()->MeshElements( srcFace );
423     SMDS_ElemIteratorPtr elemIt = srcSubDS->GetElements();
424     while ( elemIt->more() ) // loop on all mesh faces on srcFace
425     {
426       const SMDS_MeshElement* elem = elemIt->next();
427       vector< const SMDS_MeshNode* > tgtFaceNodes;
428       tgtFaceNodes.reserve( elem->NbNodes() );
429       SMDS_ElemIteratorPtr nodeIt = elem->nodesIterator();
430       while ( nodeIt->more() ) // loop on nodes of the source element
431       {
432         const SMDS_MeshNode* srcNode = (const SMDS_MeshNode*) nodeIt->next();
433         srcN_tgtN = src2tgtNodes.insert( make_pair( srcNode, nullNode )).first;
434         if ( srcN_tgtN->second == nullNode )
435         {
436           // create a new node
437           gp_Pnt tgtP = gp_Pnt(srcNode->X(),srcNode->Y(),srcNode->Z()).Transformed( trsf );
438           srcN_tgtN->second = helper.AddNode( tgtP.X(), tgtP.Y(), tgtP.Z() );
439         }
440         tgtFaceNodes.push_back( srcN_tgtN->second );
441       }
442       // create a new face (with reversed orientation)
443       if ( tgtFaceNodes.size() == 3 )
444         helper.AddFace( tgtFaceNodes[0],tgtFaceNodes[2],tgtFaceNodes[1]);
445       else
446         helper.AddFace( tgtFaceNodes[0],tgtFaceNodes[3],tgtFaceNodes[2],tgtFaceNodes[1]);
447     }
448     return true;
449   }
450
451 } // namespace
452
453
454 //=======================================================================
455 //function : Compute
456 //purpose  : 
457 //=======================================================================
458
459 bool StdMeshers_Projection_2D::Compute(SMESH_Mesh& theMesh, const TopoDS_Shape& theShape)
460 {
461   if ( !_sourceHypo )
462     return false;
463
464   SMESH_Mesh * srcMesh = _sourceHypo->GetSourceMesh();
465   SMESH_Mesh * tgtMesh = & theMesh;
466   if ( !srcMesh )
467     srcMesh = tgtMesh;
468
469   SMESHDS_Mesh * meshDS = theMesh.GetMeshDS();
470
471   // ---------------------------
472   // Make subshapes association
473   // ---------------------------
474
475   TopoDS_Face tgtFace = TopoDS::Face( theShape.Oriented(TopAbs_FORWARD));
476   TopoDS_Shape srcShape = _sourceHypo->GetSourceFace().Oriented(TopAbs_FORWARD);
477
478   TAssocTool::TShapeShapeMap shape2ShapeMap;
479   TAssocTool::InitVertexAssociation( _sourceHypo, shape2ShapeMap, tgtFace );
480   if ( !TAssocTool::FindSubShapeAssociation( tgtFace, tgtMesh, srcShape, srcMesh,
481                                              shape2ShapeMap)  ||
482        !shape2ShapeMap.IsBound( tgtFace ))
483     return error(COMPERR_BAD_SHAPE,"Topology of source and target faces seems different" );
484
485   TopoDS_Face srcFace = TopoDS::Face( shape2ShapeMap( tgtFace ).Oriented(TopAbs_FORWARD));
486
487   // ----------------------------------------------
488   // Assure that mesh on a source Face is computed
489   // ----------------------------------------------
490
491   SMESH_subMesh* srcSubMesh = srcMesh->GetSubMesh( srcFace );
492   SMESH_subMesh* tgtSubMesh = tgtMesh->GetSubMesh( tgtFace );
493
494   if ( tgtMesh == srcMesh ) {
495     if ( !TAssocTool::MakeComputed( srcSubMesh ))
496       return error(COMPERR_BAD_INPUT_MESH,"Source mesh not computed");
497   }
498   else {
499     if ( !srcSubMesh->IsMeshComputed() )
500       return error(COMPERR_BAD_INPUT_MESH,"Source mesh not computed");
501   }
502
503   // try to project from same face with different location
504   if ( projectPartner( tgtFace, srcFace, tgtMesh, srcMesh, shape2ShapeMap ))
505     return true;
506
507   // --------------------
508   // Prepare to mapping 
509   // --------------------
510
511   SMESH_MesherHelper helper( theMesh );
512   helper.SetSubShape( tgtFace );
513
514   // Check if node projection to a face is needed
515   Bnd_B2d uvBox;
516   SMDS_ElemIteratorPtr faceIt = srcSubMesh->GetSubMeshDS()->GetElements();
517   int nbFaceNodes = 0;
518   for ( ; nbFaceNodes < 3 && faceIt->more();  ) {
519     const SMDS_MeshElement* face = faceIt->next();
520     SMDS_ElemIteratorPtr nodeIt = face->nodesIterator();
521     while ( nodeIt->more() ) {
522       const SMDS_MeshNode* node = static_cast<const SMDS_MeshNode*>( nodeIt->next() );
523       if ( node->GetPosition()->GetTypeOfPosition() == SMDS_TOP_FACE ) {
524         nbFaceNodes++;
525         uvBox.Add( helper.GetNodeUV( srcFace, node ));
526       }
527     }
528   }
529   const bool toProjectNodes =
530     ( nbFaceNodes > 0 && ( uvBox.IsVoid() || uvBox.SquareExtent() < DBL_MIN ));
531
532   // Load pattern from the source face
533   SMESH_Pattern mapper;
534   mapper.Load( srcMesh, srcFace, toProjectNodes );
535   if ( mapper.GetErrorCode() != SMESH_Pattern::ERR_OK )
536     return error(COMPERR_BAD_INPUT_MESH,"Can't load mesh pattern from the source face");
537
538   // Find the first target vertex corresponding to first vertex of the <mapper>
539   // and <theReverse> flag needed to call mapper.Apply()
540
541   TopoDS_Vertex srcV1 = TopoDS::Vertex( mapper.GetSubShape( 1 ));
542   if ( srcV1.IsNull() )
543     RETURN_BAD_RESULT("Mesh is not bound to the face");
544   if ( !shape2ShapeMap.IsBound( srcV1 ))
545     RETURN_BAD_RESULT("Not associated vertices, srcV1 " << srcV1.TShape().operator->() );
546   TopoDS_Vertex tgtV1 = TopoDS::Vertex( shape2ShapeMap( srcV1 ));
547
548   if ( !TAssocTool::IsSubShape( srcV1, srcFace ))
549     RETURN_BAD_RESULT("Wrong srcV1 " << srcV1.TShape().operator->());
550   if ( !TAssocTool::IsSubShape( tgtV1, tgtFace ))
551     RETURN_BAD_RESULT("Wrong tgtV1 " << tgtV1.TShape().operator->());
552
553   // try to find out orientation by order of edges
554   bool reverse = false;
555   list< TopoDS_Edge > tgtEdges, srcEdges;
556   list< int > nbEdgesInWires;
557   SMESH_Block::GetOrderedEdges( tgtFace, tgtV1, tgtEdges, nbEdgesInWires);
558   SMESH_Block::GetOrderedEdges( srcFace, srcV1, srcEdges, nbEdgesInWires);
559   if ( nbEdgesInWires.front() > 1 ) // possible to find out
560   {
561     TopoDS_Edge srcE1 = srcEdges.front(), tgtE1 = tgtEdges.front();
562     TopoDS_Shape srcE1bis = shape2ShapeMap( tgtE1 );
563     reverse = ( ! srcE1.IsSame( srcE1bis ));
564   }
565   else if ( nbEdgesInWires.front() == 1 )
566   {
567     // TODO::Compare orientation of curves in a sole edge
568     //RETURN_BAD_RESULT("Not implemented case");
569   }
570   else
571   {
572     RETURN_BAD_RESULT("Bad result from SMESH_Block::GetOrderedEdges()");
573   }
574
575   // --------------------
576   // Perform 2D mapping 
577   // --------------------
578
579   // Compute mesh on a target face
580
581   mapper.Apply( tgtFace, tgtV1, reverse );
582   if ( mapper.GetErrorCode() != SMESH_Pattern::ERR_OK )
583     return error("Can't apply source mesh pattern to the face");
584
585   // Create the mesh
586
587   const bool toCreatePolygons = false, toCreatePolyedrs = false;
588   mapper.MakeMesh( tgtMesh, toCreatePolygons, toCreatePolyedrs );
589   if ( mapper.GetErrorCode() != SMESH_Pattern::ERR_OK )
590     return error("Can't make mesh by source mesh pattern");
591
592   // it will remove mesh built by pattern mapper on edges and vertices
593   // in failure case
594   MeshCleaner cleaner( tgtSubMesh );
595
596   // -------------------------------------------------------------------------
597   // mapper doesn't take care of nodes already existing on edges and vertices,
598   // so we must merge nodes created by it with existing ones 
599   // -------------------------------------------------------------------------
600
601   SMESH_MeshEditor editor( tgtMesh );
602   SMESH_MeshEditor::TListOfListOfNodes groupsOfNodes;
603
604   // Make groups of nodes to merge
605
606   // loop on edge and vertex submeshes of a target face
607   SMESH_subMeshIteratorPtr smIt = tgtSubMesh->getDependsOnIterator(false,false);
608   while ( smIt->more() )
609   {
610     SMESH_subMesh*     sm = smIt->next();
611     SMESHDS_SubMesh* smDS = sm->GetSubMeshDS();
612
613     // Sort new and old nodes of a submesh separately
614
615     bool isSeam = helper.IsRealSeam( sm->GetId() );
616
617     enum { NEW_NODES = 0, OLD_NODES };
618     map< double, const SMDS_MeshNode* > u2nodesMaps[2], u2nodesOnSeam;
619     map< double, const SMDS_MeshNode* >::iterator u_oldNode, u_newNode, u_newOnSeam, newEnd;
620     set< const SMDS_MeshNode* > seamNodes;
621
622     // mapper puts on a seam edge nodes from 2 edges
623     if ( isSeam && ! getBoundaryNodes ( sm, tgtFace, u2nodesOnSeam, seamNodes ))
624       RETURN_BAD_RESULT("getBoundaryNodes() failed");
625
626     SMDS_NodeIteratorPtr nIt = smDS->GetNodes();
627     while ( nIt->more() )
628     {
629       const SMDS_MeshNode* node = nIt->next();
630       bool isOld = isOldNode( node );
631
632       if ( !isOld && isSeam ) { // new node on a seam edge
633         if ( seamNodes.find( node ) != seamNodes.end())
634           continue; // node is already in the map
635       }
636
637       // sort nodes on edges by their position
638       map< double, const SMDS_MeshNode* > & pos2nodes = u2nodesMaps[isOld ? OLD_NODES : NEW_NODES];
639       switch ( node->GetPosition()->GetTypeOfPosition() )
640       {
641       case  SMDS_TOP_VERTEX: {
642         pos2nodes.insert( make_pair( 0, node ));
643         break;
644       }
645       case  SMDS_TOP_EDGE:   {
646         const SMDS_EdgePosition* pos =
647           static_cast<const SMDS_EdgePosition*>(node->GetPosition().get());
648         pos2nodes.insert( make_pair( pos->GetUParameter(), node ));
649         break;
650       }
651       default:
652         RETURN_BAD_RESULT("Wrong node position type: "<<
653                           node->GetPosition()->GetTypeOfPosition());
654       }
655     }
656     if ( u2nodesMaps[ NEW_NODES ].size() != u2nodesMaps[ OLD_NODES ].size() )
657     {
658       if ( u2nodesMaps[ NEW_NODES ].size() == 0         &&
659            sm->GetSubShape().ShapeType() == TopAbs_EDGE &&
660            helper.IsDegenShape( sm->GetId() )             )
661         // NPAL15894 (tt88bis.py) - project mesh built by NETGEN_1d_2D that
662         // does not make segments/nodes on degenerated edges
663         continue;
664
665       RETURN_BAD_RESULT("Different nb of old and new nodes on shape #"<< sm->GetId() <<" "<<
666                         u2nodesMaps[ OLD_NODES ].size() << " != " <<
667                         u2nodesMaps[ NEW_NODES ].size());
668     }
669     if ( isSeam && u2nodesMaps[ OLD_NODES ].size() != u2nodesOnSeam.size() ) {
670       RETURN_BAD_RESULT("Different nb of old and seam nodes " <<
671                         u2nodesMaps[ OLD_NODES ].size() << " != " << u2nodesOnSeam.size());
672     }
673     // Make groups of nodes to merge
674     u_oldNode = u2nodesMaps[ OLD_NODES ].begin(); 
675     u_newNode = u2nodesMaps[ NEW_NODES ].begin();
676     newEnd    = u2nodesMaps[ NEW_NODES ].end();
677     u_newOnSeam = u2nodesOnSeam.begin();
678     for ( ; u_newNode != newEnd; ++u_newNode, ++u_oldNode ) {
679       groupsOfNodes.push_back( list< const SMDS_MeshNode* >() );
680       groupsOfNodes.back().push_back( u_oldNode->second );
681       groupsOfNodes.back().push_back( u_newNode->second );
682       if ( isSeam )
683         groupsOfNodes.back().push_back( (u_newOnSeam++)->second );
684     }
685   }
686
687   // Merge
688
689   int nbFaceBeforeMerge = tgtSubMesh->GetSubMeshDS()->NbElements();
690   editor.MergeNodes( groupsOfNodes );
691   int nbFaceAtferMerge = tgtSubMesh->GetSubMeshDS()->NbElements();
692   if ( nbFaceBeforeMerge != nbFaceAtferMerge )
693     return error(COMPERR_BAD_INPUT_MESH, "Probably invalid node parameters on geom faces");
694
695   // ---------------------------
696   // Check elements orientation
697   // ---------------------------
698
699   TopoDS_Face face = tgtFace;
700   if ( !theMesh.IsMainShape( tgtFace ))
701   {
702     // find the main shape
703     TopoDS_Shape mainShape = meshDS->ShapeToMesh();
704     switch ( mainShape.ShapeType() ) {
705     case TopAbs_SHELL:
706     case TopAbs_SOLID: break;
707     default:
708       TopTools_ListIteratorOfListOfShape ancestIt = theMesh.GetAncestors( face );
709       for ( ; ancestIt.More(); ancestIt.Next() ) {
710         TopAbs_ShapeEnum type = ancestIt.Value().ShapeType();
711         if ( type == TopAbs_SOLID ) {
712           mainShape = ancestIt.Value();
713           break;
714         } else if ( type == TopAbs_SHELL ) {
715           mainShape = ancestIt.Value();
716         }
717       }
718     }
719     // find tgtFace in the main solid or shell to know it's true orientation.
720     TopExp_Explorer exp( mainShape, TopAbs_FACE );
721     for ( ; exp.More(); exp.Next() ) {
722       if ( tgtFace.IsSame( exp.Current() )) {
723         face = TopoDS::Face( exp.Current() );
724         break;
725       }
726     }
727   }
728   // Fix orientation
729   if ( SMESH_Algo::IsReversedSubMesh( face, meshDS ))
730   {
731     SMDS_ElemIteratorPtr eIt = meshDS->MeshElements( face )->GetElements();
732     while ( eIt->more() ) {
733       const SMDS_MeshElement* e = eIt->next();
734       if ( e->GetType() == SMDSAbs_Face && !editor.Reorient( e ))
735         RETURN_BAD_RESULT("Pb of SMESH_MeshEditor::Reorient()");
736     }
737   }
738
739   cleaner.Release(); // do not remove mesh
740
741   return true;
742 }
743
744
745 //=======================================================================
746 //function : Evaluate
747 //purpose  : 
748 //=======================================================================
749
750 bool StdMeshers_Projection_2D::Evaluate(SMESH_Mesh& theMesh,
751                                         const TopoDS_Shape& theShape,
752                                         MapShapeNbElems& aResMap)
753 {
754   if ( !_sourceHypo )
755     return false;
756
757   SMESH_Mesh * srcMesh = _sourceHypo->GetSourceMesh();
758   SMESH_Mesh * tgtMesh = & theMesh;
759   if ( !srcMesh )
760     srcMesh = tgtMesh;
761
762   // ---------------------------
763   // Make subshapes association
764   // ---------------------------
765
766   TopoDS_Face tgtFace = TopoDS::Face( theShape.Oriented(TopAbs_FORWARD));
767   TopoDS_Shape srcShape = _sourceHypo->GetSourceFace().Oriented(TopAbs_FORWARD);
768
769   TAssocTool::TShapeShapeMap shape2ShapeMap;
770   TAssocTool::InitVertexAssociation( _sourceHypo, shape2ShapeMap, tgtFace );
771   if ( !TAssocTool::FindSubShapeAssociation( tgtFace, tgtMesh, srcShape, srcMesh,
772                                              shape2ShapeMap)  ||
773        !shape2ShapeMap.IsBound( tgtFace ))
774     return error(COMPERR_BAD_SHAPE,"Topology of source and target faces seems different" );
775
776   TopoDS_Face srcFace = TopoDS::Face( shape2ShapeMap( tgtFace ).Oriented(TopAbs_FORWARD));
777
778   // ----------------------------------------------
779   // Assure that mesh on a source Face is computed
780   // ----------------------------------------------
781
782   SMESH_subMesh* srcSubMesh = srcMesh->GetSubMesh( srcFace );
783
784   if ( !srcSubMesh->IsMeshComputed() )
785     return error(COMPERR_BAD_INPUT_MESH,"Source mesh not computed");
786
787
788   std::vector<int> aVec(SMDSEntity_Last);
789   for(int i=SMDSEntity_Node; i<SMDSEntity_Last; i++) aVec[i] = 0;
790
791   aVec[SMDSEntity_Node] = srcSubMesh->GetSubMeshDS()->NbNodes();
792
793   //bool quadratic = false;
794   SMDS_ElemIteratorPtr elemIt = srcSubMesh->GetSubMeshDS()->GetElements();
795   while ( elemIt->more() ) {
796     const SMDS_MeshElement* E  = elemIt->next();
797     if( E->NbNodes()==3 ) {
798       aVec[SMDSEntity_Triangle]++;
799     }
800     else if( E->NbNodes()==4 ) {
801       aVec[SMDSEntity_Quadrangle]++;
802     }
803     else if( E->NbNodes()==6 && E->IsQuadratic() ) {
804       aVec[SMDSEntity_Quad_Triangle]++;
805     }
806     else if( E->NbNodes()==8 && E->IsQuadratic() ) {
807       aVec[SMDSEntity_Quad_Quadrangle]++;
808     }
809     else {
810       aVec[SMDSEntity_Polygon]++;
811     }
812   }
813
814   SMESH_subMesh * sm = theMesh.GetSubMesh(theShape);
815   aResMap.insert(std::make_pair(sm,aVec));
816
817   return true;
818 }
819
820
821 //=============================================================================
822 /*!
823  * \brief Sets a default event listener to submesh of the source face
824   * \param subMesh - submesh where algo is set
825  *
826  * This method is called when a submesh gets HYP_OK algo_state.
827  * After being set, event listener is notified on each event of a submesh.
828  * Arranges that CLEAN event is translated from source submesh to
829  * the submesh
830  */
831 //=============================================================================
832
833 void StdMeshers_Projection_2D::SetEventListener(SMESH_subMesh* subMesh)
834 {
835   TAssocTool::SetEventListener( subMesh,
836                                 _sourceHypo->GetSourceFace(),
837                                 _sourceHypo->GetSourceMesh() );
838 }