src/StdMeshers/StdMeshers_Import_1D2D.cxx

   1 // Copyright (C) 2007-2019  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, or (at your option) any later version.
  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
  23 //  SMESH SMESH : implementation of SMESH idl descriptions
  24 //  File   : StdMeshers_Import_1D2D.cxx
  25 //  Module : SMESH
  26 //
  27 #include "StdMeshers_Import_1D2D.hxx"
  28
  29 #include "StdMeshers_Import_1D.hxx"
  30 #include "StdMeshers_ImportSource.hxx"
  31
  32 #include "SMDS_MeshElement.hxx"
  33 #include "SMDS_MeshNode.hxx"
  34 #include "SMESHDS_Group.hxx"
  35 #include "SMESHDS_Mesh.hxx"
  36 #include "SMESH_Comment.hxx"
  37 #include "SMESH_ControlsDef.hxx"
  38 #include "SMESH_Gen.hxx"
  39 #include "SMESH_Group.hxx"
  40 #include "SMESH_Mesh.hxx"
  41 #include "SMESH_MesherHelper.hxx"
  42 #include "SMESH_OctreeNode.hxx"
  43 #include "SMESH_MeshEditor.hxx"
  44 #include "SMESH_subMesh.hxx"
  45
  46 #include "Utils_SALOME_Exception.hxx"
  47 #include "utilities.h"
  48
  49 #include <BRepBndLib.hxx>
  50 #include <BRepClass_FaceClassifier.hxx>
  51 #include <BRepTools.hxx>
  52 #include <BRepTopAdaptor_FClass2d.hxx>
  53 #include <BRep_Builder.hxx>
  54 #include <BRep_Tool.hxx>
  55 #include <Bnd_B2d.hxx>
  56 #include <Bnd_Box.hxx>
  57 #include <GeomAPI_ProjectPointOnSurf.hxx>
  58 #include <GeomAdaptor_Surface.hxx>
  59 #include <Precision.hxx>
  60 #include <TopExp.hxx>
  61 #include <TopExp_Explorer.hxx>
  62 #include <TopoDS.hxx>
  63 #include <TopoDS_Compound.hxx>
  64 #include <TopoDS_Edge.hxx>
  65 #include <TopoDS_Vertex.hxx>
  66
  67 #include <numeric>
  68
  69 using namespace std;
  70
  71 namespace
  72 {
  73   double getMinElemSize2( const SMESHDS_GroupBase* srcGroup )
  74   {
  75     double minSize2 = 1e100;
  76     SMDS_ElemIteratorPtr srcElems = srcGroup->GetElements();
  77     while ( srcElems->more() ) // loop on group contents
  78     {
  79       const SMDS_MeshElement* face = srcElems->next();
  80       int nbN = face->NbCornerNodes();
  81
  82       SMESH_TNodeXYZ prevN( face->GetNode( nbN-1 ));
  83       for ( int i = 0; i < nbN; ++i )
  84       {
  85         SMESH_TNodeXYZ n( face->GetNode( i ) );
  86         double size2 = ( n - prevN ).SquareModulus();
  87         minSize2 = std::min( minSize2, size2 );
  88         prevN = n;
  89       }
  90     }
  91     return minSize2;
  92   }
  93 }
  94
  95 //=============================================================================
  96 /*!
  97  * Creates StdMeshers_Import_1D2D
  98  */
  99 //=============================================================================
 100
 101 StdMeshers_Import_1D2D::StdMeshers_Import_1D2D(int hypId, SMESH_Gen * gen)
 102   :SMESH_2D_Algo(hypId, gen), _sourceHyp(0)
 103 {
 104   _name = "Import_1D2D";
 105   _shapeType = (1 << TopAbs_FACE);
 106
 107   _compatibleHypothesis.push_back("ImportSource2D");
 108   _requireDiscreteBoundary = false;
 109   _supportSubmeshes = true;
 110 }
 111
 112 //=============================================================================
 113 /*!
 114  * Check presence of a hypothesis
 115  */
 116 //=============================================================================
 117
 118 bool StdMeshers_Import_1D2D::CheckHypothesis
 119                          (SMESH_Mesh&                          aMesh,
 120                           const TopoDS_Shape&                  aShape,
 121                           SMESH_Hypothesis::Hypothesis_Status& aStatus)
 122 {
 123   _sourceHyp = 0;
 124
 125   const list <const SMESHDS_Hypothesis * >&hyps = GetUsedHypothesis(aMesh, aShape);
 126   if ( hyps.size() == 0 )
 127   {
 128     aStatus = SMESH_Hypothesis::HYP_MISSING;
 129     return false;  // can't work with no hypothesis
 130   }
 131
 132   if ( hyps.size() > 1 )
 133   {
 134     aStatus = SMESH_Hypothesis::HYP_ALREADY_EXIST;
 135     return false;
 136   }
 137
 138   const SMESHDS_Hypothesis *theHyp = hyps.front();
 139
 140   string hypName = theHyp->GetName();
 141
 142   if (hypName == _compatibleHypothesis.front())
 143   {
 144     _sourceHyp = (StdMeshers_ImportSource1D *)theHyp;
 145     aStatus = SMESH_Hypothesis::HYP_OK;
 146     return true;
 147   }
 148
 149   aStatus = SMESH_Hypothesis::HYP_INCOMPATIBLE;
 150   return true;
 151 }
 152
 153 namespace
 154 {
 155   /*!
 156    * \brief OrientedLink additionally storing a medium node
 157    */
 158   struct TLink : public SMESH_OrientedLink
 159   {
 160     const SMDS_MeshNode* _medium;
 161     TLink( const SMDS_MeshNode* n1,
 162            const SMDS_MeshNode* n2,
 163            const SMDS_MeshNode* medium=0)
 164       : SMESH_OrientedLink( n1,n2 ), _medium( medium ) {}
 165   };
 166 }
 167
 168 //=============================================================================
 169 /*!
 170  * Import elements from the other mesh
 171  */
 172 //=============================================================================
 173
 174 bool StdMeshers_Import_1D2D::Compute(SMESH_Mesh & theMesh, const TopoDS_Shape & theShape)
 175 {
 176   if ( !_sourceHyp ) return false;
 177
 178   const vector<SMESH_Group*>& srcGroups = _sourceHyp->GetGroups(/*loaded=*/true);
 179   if ( srcGroups.empty() )
 180     return error("Invalid source groups");
 181
 182   bool allGroupsEmpty = true;
 183   for ( size_t iG = 0; iG < srcGroups.size() && allGroupsEmpty; ++iG )
 184     allGroupsEmpty = srcGroups[iG]->GetGroupDS()->IsEmpty();
 185   if ( allGroupsEmpty )
 186     return error("No faces in source groups");
 187
 188   SMESH_MesherHelper helper(theMesh);
 189   helper.SetSubShape(theShape);
 190   SMESHDS_Mesh* tgtMesh = theMesh.GetMeshDS();
 191
 192   const TopoDS_Face& geomFace = TopoDS::Face( theShape );
 193   const double  faceTol = helper.MaxTolerance( geomFace );
 194   const int     shapeID = tgtMesh->ShapeToIndex( geomFace );
 195   const bool toCheckOri = (helper.NbAncestors( geomFace, theMesh, TopAbs_SOLID ) == 1 );
 196
 197
 198   Handle(Geom_Surface) surface = BRep_Tool::Surface( geomFace );
 199   const bool reverse =
 200     ( helper.GetSubShapeOri( tgtMesh->ShapeToMesh(), geomFace ) == TopAbs_REVERSED );
 201   gp_Pnt p; gp_Vec du, dv;
 202
 203   // BRepClass_FaceClassifier is most time consuming, so minimize its usage
 204   const double clsfTol = 10 * BRep_Tool::Tolerance( geomFace );
 205   BRepTopAdaptor_FClass2d classifier( geomFace, clsfTol ); //Brimless_FaceClassifier classifier;
 206   Bnd_B2d bndBox2d;
 207   Bnd_Box bndBox3d;
 208   {
 209     Standard_Real umin,umax,vmin,vmax;
 210     BRepTools::UVBounds(geomFace,umin,umax,vmin,vmax);
 211     gp_XY pmin( umin,vmin ), pmax( umax,vmax );
 212     bndBox2d.Add( pmin );
 213     bndBox2d.Add( pmax );
 214     if ( helper.HasSeam() )
 215     {
 216       const int i = helper.GetPeriodicIndex();
 217       pmin.SetCoord( i, helper.GetOtherParam( pmin.Coord( i )));
 218       pmax.SetCoord( i, helper.GetOtherParam( pmax.Coord( i )));
 219       bndBox2d.Add( pmin );
 220       bndBox2d.Add( pmax );
 221     }
 222     bndBox2d.Enlarge( 1e-2 * Sqrt( bndBox2d.SquareExtent() ));
 223
 224     BRepBndLib::Add( geomFace, bndBox3d );
 225     bndBox3d.Enlarge( 1e-2 * sqrt( bndBox3d.SquareExtent() ));
 226   }
 227
 228   set<int> subShapeIDs;
 229   subShapeIDs.insert( shapeID );
 230
 231   // nodes already existing on sub-shapes of the FACE
 232   TIDSortedNodeSet existingNodes;
 233
 234   // get/make nodes on vertices and add them to existingNodes
 235   TopExp_Explorer exp( theShape, TopAbs_VERTEX );
 236   for ( ; exp.More(); exp.Next() )
 237   {
 238     const TopoDS_Vertex& v = TopoDS::Vertex( exp.Current() );
 239     if ( !subShapeIDs.insert( tgtMesh->ShapeToIndex( v )).second )
 240       continue;
 241     const SMDS_MeshNode* n = SMESH_Algo::VertexNode( v, tgtMesh );
 242     if ( !n )
 243     {
 244       _gen->Compute(theMesh,v,/*anUpward=*/true);
 245       n = SMESH_Algo::VertexNode( v, tgtMesh );
 246       if ( !n ) return false; // very strange
 247     }
 248     existingNodes.insert( n );
 249   }
 250
 251   // get EDGEs and their ids and get existing nodes on EDGEs
 252   vector< TopoDS_Edge > edges;
 253   for ( exp.Init( theShape, TopAbs_EDGE ); exp.More(); exp.Next() )
 254   {
 255     const TopoDS_Edge & edge = TopoDS::Edge( exp.Current() );
 256     if ( !SMESH_Algo::isDegenerated( edge ))
 257       if ( subShapeIDs.insert( tgtMesh->ShapeToIndex( edge )).second )
 258       {
 259         edges.push_back( edge );
 260         if ( SMESHDS_SubMesh* eSM = tgtMesh->MeshElements( edge ))
 261         {
 262           typedef SMDS_StdIterator< const SMDS_MeshNode*, SMDS_NodeIteratorPtr > iterator;
 263           existingNodes.insert( iterator( eSM->GetNodes() ), iterator() );
 264         }
 265       }
 266   }
 267   // octree to find existing nodes
 268   SMESH_OctreeNode existingNodeOcTr( existingNodes );
 269   std::map<double, const SMDS_MeshNode*> dist2foundNodes;
 270
 271   // to count now many times a link between nodes encounters
 272   map<TLink, int> linkCount;
 273   map<TLink, int>::iterator link2Nb;
 274   double minGroupTol = Precision::Infinite();
 275
 276   SMESH::Controls::ElementsOnShape onEdgeClassifier;
 277   if ( helper.HasSeam() )
 278   {
 279     TopoDS_Compound edgesCompound;
 280     BRep_Builder    builder;
 281     builder.MakeCompound( edgesCompound );
 282     for ( size_t iE = 0; iE < edges.size(); ++iE )
 283       builder.Add( edgesCompound, edges[ iE ]);
 284     onEdgeClassifier.SetShape( edgesCompound, SMDSAbs_Node );
 285   }
 286
 287   // =========================
 288   // Import faces from groups
 289   // =========================
 290
 291   StdMeshers_Import_1D::TNodeNodeMap* n2n;
 292   StdMeshers_Import_1D::TElemElemMap* e2e;
 293   StdMeshers_Import_1D::TNodeNodeMap::iterator n2nIt;
 294   pair< StdMeshers_Import_1D::TNodeNodeMap::iterator, bool > it_isnew;
 295   vector<TopAbs_State>         nodeState;
 296   vector<const SMDS_MeshNode*> newNodes; // of a face
 297   set   <const SMDS_MeshNode*> bndNodes; // nodes classified ON
 298   vector<bool>                 isNodeIn; // nodes classified IN, by node ID
 299
 300   for ( size_t iG = 0; iG < srcGroups.size(); ++iG )
 301   {
 302     const SMESHDS_GroupBase* srcGroup = srcGroups[iG]->GetGroupDS();
 303
 304     const int meshID = srcGroup->GetMesh()->GetPersistentId();
 305     const SMESH_Mesh* srcMesh = GetMeshByPersistentID( meshID );
 306     if ( !srcMesh ) continue;
 307     StdMeshers_Import_1D::getMaps( srcMesh, &theMesh, n2n, e2e );
 308
 309     const double groupTol = 0.5 * sqrt( getMinElemSize2( srcGroup ));
 310     minGroupTol = std::min( groupTol, minGroupTol );
 311
 312     // clsfTol is 2D tolerance of a probe line
 313     //GeomAdaptor_Surface S( surface );
 314     // const double clsfTol = Min( S.UResolution( 0.1 * groupTol ), -- issue 0023092
 315     //                             S.VResolution( 0.1 * groupTol ));
 316     // another idea: try to use max tol of all edges
 317     //const double clsfTol = 10 * BRep_Tool::Tolerance( geomFace ); // 0.1 * groupTol;
 318
 319     if ( helper.HasSeam() )
 320       onEdgeClassifier.SetMesh( srcMesh->GetMeshDS() );
 321
 322     SMDS_ElemIteratorPtr srcElems = srcGroup->GetElements();
 323     while ( srcElems->more() ) // loop on group contents
 324     {
 325       const SMDS_MeshElement* face = srcElems->next();
 326
 327       SMDS_MeshElement::iterator node = face->begin_nodes();
 328       if ( bndBox3d.IsOut( SMESH_TNodeXYZ( *node )))
 329         continue;
 330
 331       // find or create nodes of a new face
 332       nodeState.resize( face->NbNodes() );
 333       newNodes.resize( nodeState.size() );
 334       newNodes.back() = 0;
 335       int nbCreatedNodes = 0;
 336       bool isOut = false, isIn = false; // if at least one node isIn - do not classify other nodes
 337       for ( size_t i = 0; i < newNodes.size(); ++i, ++node )
 338       {
 339         SMESH_TNodeXYZ nXYZ = *node;
 340         nodeState[ i ] = TopAbs_UNKNOWN;
 341         newNodes [ i ] = 0;
 342
 343         it_isnew = n2n->insert( make_pair( *node, (SMDS_MeshNode*)0 ));
 344         n2nIt    = it_isnew.first;
 345
 346         const SMDS_MeshNode* & newNode = n2nIt->second;
 347         if ( !it_isnew.second && !newNode )
 348           break; // a node is mapped to NULL - it is OUT of the FACE
 349
 350         if ( newNode )
 351         {
 352           if ( !subShapeIDs.count( newNode->getshapeId() ))
 353             break; // node is Imported onto other FACE
 354           if ( newNode->GetID() < (int) isNodeIn.size() &&
 355                isNodeIn[ newNode->GetID() ])
 356             isIn = true;
 357           if ( !isIn && bndNodes.count( *node ))
 358             nodeState[ i ] = TopAbs_ON;
 359         }
 360         else
 361         {
 362           // find a node pre-existing on EDGE or VERTEX
 363           dist2foundNodes.clear();
 364           existingNodeOcTr.NodesAround( nXYZ, dist2foundNodes, groupTol );
 365           if ( !dist2foundNodes.empty() )
 366           {
 367             newNode = dist2foundNodes.begin()->second;
 368             nodeState[ i ] = TopAbs_ON;
 369           }
 370         }
 371
 372         if ( !newNode )
 373         {
 374           // find out if node lies on the surface of theShape
 375           gp_XY uv( Precision::Infinite(), 0 );
 376           isOut = ( !helper.CheckNodeUV( geomFace, *node, uv, groupTol, /*force=*/true ) ||
 377                     bndBox2d.IsOut( uv ));
 378           //int iCoo;
 379           if ( !isOut && !isIn ) // classify
 380           {
 381             nodeState[i] = classifier.Perform( uv ); //classifier.Perform( geomFace, uv, clsfTol );
 382             //nodeState[i] = classifier.State();
 383             isOut = ( nodeState[i] == TopAbs_OUT );
 384             if ( isOut && helper.IsOnSeam( uv ) && onEdgeClassifier.IsSatisfy( (*node)->GetID() ))
 385             {
 386               // uv.SetCoord( iCoo, helper.GetOtherParam( uv.Coord( iCoo )));
 387               // classifier.Perform( geomFace, uv, clsfTol );
 388               // nodeState[i] = classifier.State();
 389               // isOut = ( nodeState[i] == TopAbs_OUT );
 390               nodeState[i] = TopAbs_ON;
 391               isOut = false;
 392             }
 393           }
 394           if ( !isOut ) // create a new node
 395           {
 396             newNode = tgtMesh->AddNode( nXYZ.X(), nXYZ.Y(), nXYZ.Z());
 397             tgtMesh->SetNodeOnFace( newNode, shapeID, uv.X(), uv.Y() );
 398             nbCreatedNodes++;
 399             if ( newNode->GetID() >= (int) isNodeIn.size() )
 400             {
 401               isNodeIn.push_back( false ); // allow allocate more than newNode->GetID()
 402               isNodeIn.resize( newNode->GetID() + 1, false );
 403             }
 404             if ( nodeState[i] == TopAbs_ON )
 405               bndNodes.insert( *node );
 406             else if ( nodeState[i] != TopAbs_UNKNOWN )
 407               isNodeIn[ newNode->GetID() ] = isIn = true;
 408           }
 409         }
 410         if ( !(newNodes[i] = newNode ) || isOut )
 411           break;
 412
 413       } // loop on face nodes
 414
 415       if ( !newNodes.back() )
 416         continue; // not all nodes of the face lie on theShape
 417
 418       if ( !isIn ) // if all nodes are on FACE boundary, a mesh face can be OUT
 419       {
 420         // check state of nodes created for other faces
 421         for ( size_t i = 0; i < nodeState.size() && !isIn; ++i )
 422         {
 423           if ( nodeState[i] != TopAbs_UNKNOWN ) continue;
 424           gp_XY uv = helper.GetNodeUV( geomFace, newNodes[i] );
 425           nodeState[i] = classifier.Perform( uv ); //geomFace, uv, clsfTol );
 426           //nodeState[i] = classifier.State();
 427           isIn = ( nodeState[i] == TopAbs_IN );
 428         }
 429         if ( !isIn ) // classify face center
 430         {
 431           gp_XYZ gc( 0., 0., 0 );
 432           for ( size_t i = 0; i < newNodes.size(); ++i )
 433             gc += SMESH_TNodeXYZ( newNodes[i] );
 434           gc /= newNodes.size();
 435
 436           TopLoc_Location loc;
 437           GeomAPI_ProjectPointOnSurf& proj = helper.GetProjector( geomFace,
 438                                                                   loc,
 439                                                                   helper.MaxTolerance( geomFace ));
 440           if ( !loc.IsIdentity() ) loc.Transformation().Inverted().Transforms( gc );
 441           proj.Perform( gc );
 442           if ( !proj.IsDone() || proj.NbPoints() < 1 )
 443             continue;
 444           Standard_Real U,V;
 445           proj.LowerDistanceParameters(U,V);
 446           gp_XY uv( U,V );
 447           //classifier.Perform( geomFace, uv, clsfTol );
 448           TopAbs_State state = classifier.Perform( uv );
 449           if ( state != TopAbs_IN )
 450             continue;
 451         }
 452       }
 453
 454       // try to find already created face
 455       SMDS_MeshElement * newFace = 0;
 456       if ( nbCreatedNodes == 0 &&
 457            tgtMesh->FindElement(newNodes, SMDSAbs_Face, /*noMedium=*/false))
 458         continue; // repeated face in source groups already created
 459
 460       // check future face orientation
 461       const int nbCorners = face->NbCornerNodes();
 462       const bool isQuad   = ( nbCorners != (int) newNodes.size() );
 463       if ( toCheckOri )
 464       {
 465         int iNode = -1;
 466         gp_Vec geomNorm;
 467         do
 468         {
 469           gp_XY uv = helper.GetNodeUV( geomFace, newNodes[++iNode] );
 470           surface->D1( uv.X(),uv.Y(), p, du,dv );
 471           geomNorm = reverse ? dv^du : du^dv;
 472         }
 473         while ( geomNorm.SquareMagnitude() < 1e-6 && iNode+1 < nbCorners );
 474
 475         int iNext = helper.WrapIndex( iNode+1, nbCorners );
 476         int iPrev = helper.WrapIndex( iNode-1, nbCorners );
 477
 478         SMESH_TNodeXYZ prevNode( newNodes[iPrev] );
 479         SMESH_TNodeXYZ curNode ( newNodes[iNode] );
 480         SMESH_TNodeXYZ nextNode( newNodes[iNext] );
 481         gp_Vec n1n0( prevNode - curNode);
 482         gp_Vec n1n2( nextNode - curNode );
 483         gp_Vec meshNorm = n1n2 ^ n1n0;
 484
 485         if ( geomNorm * meshNorm < 0 )
 486           SMDS_MeshCell::applyInterlace
 487             ( SMDS_MeshCell::reverseSmdsOrder( face->GetEntityType(), newNodes.size() ), newNodes );
 488       }
 489
 490       // make a new face
 491       if ( face->IsPoly() )
 492         newFace = tgtMesh->AddPolygonalFace( newNodes );
 493       else
 494         switch ( newNodes.size() )
 495         {
 496         case 3:
 497           newFace = tgtMesh->AddFace( newNodes[0], newNodes[1], newNodes[2] );
 498           break;
 499         case 4:
 500           newFace = tgtMesh->AddFace( newNodes[0], newNodes[1], newNodes[2], newNodes[3] );
 501           break;
 502         case 6:
 503           newFace = tgtMesh->AddFace( newNodes[0], newNodes[1], newNodes[2],
 504                                       newNodes[3], newNodes[4], newNodes[5]);
 505           break;
 506         case 8:
 507           newFace = tgtMesh->AddFace( newNodes[0], newNodes[1], newNodes[2], newNodes[3],
 508                                       newNodes[4], newNodes[5], newNodes[6], newNodes[7]);
 509           break;
 510         default: continue;
 511         }
 512       tgtMesh->SetMeshElementOnShape( newFace, shapeID );
 513       e2e->insert( make_pair( face, newFace ));
 514
 515       // collect links
 516       const SMDS_MeshNode* medium = 0;
 517       for ( int i = 0; i < nbCorners; ++i )
 518       {
 519         const SMDS_MeshNode* n1 = newNodes[i];
 520         const SMDS_MeshNode* n2 = newNodes[ (i+1)%nbCorners ];
 521         if ( isQuad ) // quadratic face
 522           medium = newNodes[i+nbCorners];
 523         link2Nb = linkCount.insert( make_pair( TLink( n1, n2, medium ), 0)).first;
 524         ++link2Nb->second;
 525       }
 526     } // loop on group contents
 527
 528     // Remove OUT nodes from n2n map
 529     for ( n2nIt = n2n->begin(); n2nIt != n2n->end(); )
 530       if ( !n2nIt->second )
 531         n2n->erase( n2nIt++ );
 532       else
 533         ++n2nIt;
 534
 535   } // loop on src groups
 536
 537   // remove free nodes created on EDGEs
 538   {
 539     set<const SMDS_MeshNode*>::iterator node = bndNodes.begin();
 540     for ( ; node != bndNodes.end(); ++node )
 541     {
 542       n2nIt = n2n->find( *node );
 543       const SMDS_MeshNode* newNode = n2nIt->second;
 544       if ( newNode && newNode->NbInverseElements() == 0 )
 545       {
 546         tgtMesh->RemoveFreeNode( newNode, 0, false );
 547         n2n->erase( n2nIt );
 548       }
 549     }
 550   }
 551
 552   // ==========================================================
 553   // Put nodes on geom edges and create edges on them;
 554   // check if the whole geom face is covered by imported faces
 555   // ==========================================================
 556
 557   // use large tolerance for projection of nodes to edges because of
 558   // BLSURF mesher specifics (issue 0020918, Study2.hdf)
 559   const double projTol = minGroupTol;
 560
 561   bool isFaceMeshed = false;
 562   SMESHDS_SubMesh* tgtFaceSM = tgtMesh->MeshElements( theShape );
 563   if ( tgtFaceSM )
 564   {
 565     // the imported mesh is valid if all external links (encountered once)
 566     // lie on geom edges
 567     subShapeIDs.erase( shapeID ); // to contain edges and vertices only
 568     double u, f, l;
 569     for ( link2Nb = linkCount.begin(); link2Nb != linkCount.end(); ++link2Nb)
 570     {
 571       const TLink& link = (*link2Nb).first;
 572       int nbFaces = link2Nb->second;
 573       if ( nbFaces == 1 )
 574       {
 575         // check if a not shared link lies on face boundary
 576         bool nodesOnBoundary = true;
 577         list< TopoDS_Shape > bndShapes;
 578         for ( int is1stN = 0; is1stN < 2 && nodesOnBoundary; ++is1stN )
 579         {
 580           const SMDS_MeshNode* n = is1stN ? link.node1() : link.node2();
 581           if ( !subShapeIDs.count( n->getshapeId() )) // n is assigned to FACE
 582           {
 583             for ( size_t iE = 0; iE < edges.size(); ++iE )
 584               if ( helper.CheckNodeU( edges[iE], n, u=0, projTol, /*force=*/true ))
 585               {
 586                 BRep_Tool::Range(edges[iE],f,l);
 587                 if ( Abs(u-f) < 2 * faceTol || Abs(u-l) < 2 * faceTol )
 588                   // duplicated node on vertex
 589                   return error("Source elements overlap one another");
 590                 tgtFaceSM->RemoveNode( n );
 591                 tgtMesh->SetNodeOnEdge( n, edges[iE], u );
 592                 break;
 593               }
 594             nodesOnBoundary = subShapeIDs.count( n->getshapeId());
 595           }
 596           if ( nodesOnBoundary )
 597           {
 598             TopoDS_Shape s = helper.GetSubShapeByNode( n, tgtMesh );
 599             if ( s.ShapeType() == TopAbs_VERTEX )
 600               bndShapes.push_front( s ); // vertex first
 601             else
 602               bndShapes.push_back( s ); // edges last
 603           }
 604         }
 605         if ( !nodesOnBoundary )
 606         {
 607           error("free internal link"); // just for an easier debug
 608           break;
 609         }
 610         if ( bndShapes.front().ShapeType() == TopAbs_EDGE && // all link nodes are on EDGEs
 611              bndShapes.front() != bndShapes.back() )
 612           // link nodes on different geom edges
 613           return error(COMPERR_BAD_INPUT_MESH, "Source nodes mismatch target vertices");
 614
 615         // find geom edge the link is on
 616         if ( bndShapes.back().ShapeType() != TopAbs_EDGE ) // all link nodes are on VERTEXes
 617         {
 618           // find geom edge by two vertices
 619           TopoDS_Shape geomEdge = helper.GetCommonAncestor( bndShapes.back(),
 620                                                             bndShapes.front(),
 621                                                             theMesh, TopAbs_EDGE );
 622           if ( geomEdge.IsNull() )
 623           {
 624             error("free internal link");
 625             break; // vertices belong to different edges
 626           }
 627           bndShapes.push_back( geomEdge );
 628         }
 629
 630         // create an edge if not yet exists
 631         newNodes.resize(2);
 632         newNodes[0] = link.node1(), newNodes[1] = link.node2();
 633         const SMDS_MeshElement* edge = tgtMesh->FindElement( newNodes, SMDSAbs_Edge );
 634         if ( edge ) continue;
 635
 636         if ( link._reversed ) std::swap( newNodes[0], newNodes[1] );
 637         if ( link._medium )
 638         {
 639           edge = tgtMesh->AddEdge( newNodes[0], newNodes[1], link._medium );
 640
 641           TopoDS_Edge geomEdge = TopoDS::Edge(bndShapes.back());
 642           helper.CheckNodeU( geomEdge, link._medium, u, projTol, /*force=*/true );
 643           tgtFaceSM->RemoveNode( link._medium );
 644           tgtMesh->SetNodeOnEdge( (SMDS_MeshNode*)link._medium, geomEdge, u );
 645         }
 646         else
 647         {
 648           edge = tgtMesh->AddEdge( newNodes[0], newNodes[1]);
 649         }
 650         if ( !edge )
 651           return false;
 652
 653         tgtMesh->SetMeshElementOnShape( edge, bndShapes.back() );
 654       }
 655       else if ( nbFaces > 2 )
 656       {
 657         return error( COMPERR_BAD_INPUT_MESH, "Non-manifold source mesh");
 658       }
 659     }
 660     isFaceMeshed = ( link2Nb == linkCount.end() && !linkCount.empty());
 661     if ( isFaceMeshed )
 662     {
 663       // check that source faces do not overlap:
 664       // there must be only two edges sharing each vertex and bound to sub-edges of theShape
 665       SMESH_MeshEditor editor( &theMesh );
 666       set<int>::iterator subID = subShapeIDs.begin();
 667       for ( ; subID != subShapeIDs.end(); ++subID )
 668       {
 669         const TopoDS_Shape& s = tgtMesh->IndexToShape( *subID );
 670         if ( s.ShapeType() != TopAbs_VERTEX ) continue;
 671         const SMDS_MeshNode* n = SMESH_Algo::VertexNode( TopoDS::Vertex(s), tgtMesh );
 672         SMDS_ElemIteratorPtr eIt = n->GetInverseElementIterator(SMDSAbs_Edge);
 673         int nbEdges = 0;
 674         while ( eIt->more() )
 675         {
 676           const SMDS_MeshElement* edge = eIt->next();
 677           int sId = editor.FindShape( edge );
 678           nbEdges += subShapeIDs.count( sId );
 679         }
 680         if ( nbEdges < 2 && !helper.IsRealSeam( s ))
 681           return false; // weird
 682         if ( nbEdges > 2 )
 683           return error( COMPERR_BAD_INPUT_MESH, "Source elements overlap one another");
 684       }
 685     }
 686   }
 687   if ( !isFaceMeshed )
 688     return error( COMPERR_BAD_INPUT_MESH,
 689                   "Source elements don't cover totally the geometrical face" );
 690
 691   if ( helper.HasSeam() )
 692   {
 693     // links on seam edges are shared by two faces, so no edges were created on them
 694     // by the previous detection of 2D mesh boundary
 695     for ( size_t iE = 0; iE < edges.size(); ++iE )
 696     {
 697       if ( !helper.IsRealSeam( edges[iE] )) continue;
 698       const TopoDS_Edge& seamEdge = edges[iE];
 699       // to find nodes lying on the seamEdge we check nodes of mesh faces sharing a node on one
 700       // of its vertices; after finding another node on seamEdge we continue the same way
 701       // until finding all nodes.
 702       TopoDS_Vertex      seamVertex = helper.IthVertex( 0, seamEdge );
 703       const SMDS_MeshNode* vertNode = SMESH_Algo::VertexNode( seamVertex, tgtMesh );
 704       set< const SMDS_MeshNode* > checkedNodes; checkedNodes.insert( vertNode );
 705       set< const SMDS_MeshElement* > checkedFaces;
 706       // as a face can have more than one node on the seamEdge, there is a difficulty in selecting
 707       // one of those nodes to treat next; so we simply find all nodes on the seamEdge and
 708       // then sort them by U on edge
 709       typedef list< pair< double, const SMDS_MeshNode* > > TUNodeList;
 710       TUNodeList nodesOnSeam;
 711       double u = helper.GetNodeU( seamEdge, vertNode );
 712       nodesOnSeam.push_back( make_pair( u, vertNode ));
 713       TUNodeList::iterator u2nIt = nodesOnSeam.begin();
 714       for ( ; u2nIt != nodesOnSeam.end(); ++u2nIt )
 715       {
 716         const SMDS_MeshNode* startNode = (*u2nIt).second;
 717         SMDS_ElemIteratorPtr faceIt = startNode->GetInverseElementIterator( SMDSAbs_Face );
 718         while ( faceIt->more() )
 719         {
 720           const SMDS_MeshElement* face = faceIt->next();
 721           if ( !checkedFaces.insert( face ).second ) continue;
 722           for ( int i = 0, nbNodes = face->NbCornerNodes(); i < nbNodes; ++i )
 723           {
 724             const SMDS_MeshNode* n = face->GetNode( i );
 725             if ( n == startNode || !checkedNodes.insert( n ).second ) continue;
 726             if ( helper.CheckNodeU( seamEdge, n, u=0, projTol, /*force=*/true ))
 727               nodesOnSeam.push_back( make_pair( u, n ));
 728           }
 729         }
 730       }
 731       // sort the found nodes by U on the seamEdge; most probably they are in a good order,
 732       // so we can use the hint to spead-up map filling
 733       map< double, const SMDS_MeshNode* > u2nodeMap;
 734       for ( u2nIt = nodesOnSeam.begin(); u2nIt != nodesOnSeam.end(); ++u2nIt )
 735         u2nodeMap.insert( u2nodeMap.end(), *u2nIt );
 736
 737       // create edges
 738       {
 739         SMESH_MesherHelper seamHelper( theMesh );
 740         seamHelper.SetSubShape( edges[ iE ]);
 741         seamHelper.SetElementsOnShape( true );
 742
 743         if ( !checkedFaces.empty() && (*checkedFaces.begin())->IsQuadratic() )
 744           for ( set< const SMDS_MeshElement* >::iterator fIt = checkedFaces.begin();
 745                 fIt != checkedFaces.end(); ++fIt )
 746             seamHelper.AddTLinks( static_cast<const SMDS_MeshFace*>( *fIt ));
 747
 748         map< double, const SMDS_MeshNode* >::iterator n1, n2, u2nEnd = u2nodeMap.end();
 749         for ( n2 = u2nodeMap.begin(), n1 = n2++; n2 != u2nEnd; ++n1, ++n2 )
 750         {
 751           const SMDS_MeshNode* node1 = n1->second;
 752           const SMDS_MeshNode* node2 = n2->second;
 753           seamHelper.AddEdge( node1, node2 );
 754           if ( node2->getshapeId() == helper.GetSubShapeID() )
 755           {
 756             tgtFaceSM->RemoveNode( node2 );
 757             tgtMesh->SetNodeOnEdge( const_cast<SMDS_MeshNode*>( node2 ), seamEdge, n2->first );
 758           }
 759         }
 760       }
 761     } // loop on edges to find seam ones
 762   } // if ( helper.HasSeam() )
 763
 764   // notify sub-meshes of edges on computation
 765   for ( size_t iE = 0; iE < edges.size(); ++iE )
 766   {
 767     SMESH_subMesh * sm = theMesh.GetSubMesh( edges[iE] );
 768     // if ( SMESH_Algo::isDegenerated( edges[iE] ))
 769     //   sm->SetIsAlwaysComputed( true );
 770     sm->ComputeStateEngine(SMESH_subMesh::CHECK_COMPUTE_STATE);
 771     if ( sm->GetComputeState() != SMESH_subMesh::COMPUTE_OK )
 772       return error(SMESH_Comment("Failed to create segments on the edge #") << sm->GetId());
 773   }
 774
 775   // ============
 776   // Copy meshes
 777   // ============
 778
 779   vector<SMESH_Mesh*> srcMeshes = _sourceHyp->GetSourceMeshes();
 780   for ( size_t i = 0; i < srcMeshes.size(); ++i )
 781     StdMeshers_Import_1D::importMesh( srcMeshes[i], theMesh, _sourceHyp, theShape );
 782
 783   return true;
 784 }
 785
 786 //=============================================================================
 787 /*!
 788  * \brief Set needed event listeners and create a submesh for a copied mesh
 789  *
 790  * This method is called only if a submesh has HYP_OK algo_state.
 791  */
 792 //=============================================================================
 793
 794 void StdMeshers_Import_1D2D::SetEventListener(SMESH_subMesh* subMesh)
 795 {
 796   if ( !_sourceHyp )
 797   {
 798     const TopoDS_Shape& tgtShape = subMesh->GetSubShape();
 799     SMESH_Mesh*         tgtMesh  = subMesh->GetFather();
 800     Hypothesis_Status aStatus;
 801     CheckHypothesis( *tgtMesh, tgtShape, aStatus );
 802   }
 803   StdMeshers_Import_1D::setEventListener( subMesh, _sourceHyp );
 804 }
 805 void StdMeshers_Import_1D2D::SubmeshRestored(SMESH_subMesh* subMesh)
 806 {
 807   SetEventListener(subMesh);
 808 }
 809
 810 //=============================================================================
 811 /*!
 812  * Predict nb of mesh entities created by Compute()
 813  */
 814 //=============================================================================
 815
 816 bool StdMeshers_Import_1D2D::Evaluate(SMESH_Mesh &         theMesh,
 817                                       const TopoDS_Shape & theShape,
 818                                       MapShapeNbElems&     aResMap)
 819 {
 820   if ( !_sourceHyp ) return false;
 821
 822   const vector<SMESH_Group*>& srcGroups = _sourceHyp->GetGroups();
 823   if ( srcGroups.empty() )
 824     return error("Invalid source groups");
 825
 826   vector<int> aVec(SMDSEntity_Last,0);
 827
 828   bool toCopyMesh, toCopyGroups;
 829   _sourceHyp->GetCopySourceMesh(toCopyMesh, toCopyGroups);
 830   if ( toCopyMesh ) // the whole mesh is copied
 831   {
 832     vector<SMESH_Mesh*> srcMeshes = _sourceHyp->GetSourceMeshes();
 833     for ( unsigned i = 0; i < srcMeshes.size(); ++i )
 834     {
 835       SMESH_subMesh* sm = StdMeshers_Import_1D::getSubMeshOfCopiedMesh( theMesh, *srcMeshes[i]);
 836       if ( !sm || aResMap.count( sm )) continue; // already counted
 837       const SMDS_MeshInfo& aMeshInfo = srcMeshes[i]->GetMeshDS()->GetMeshInfo();
 838       for (int i = 0; i < SMDSEntity_Last; i++)
 839         aVec[i] = aMeshInfo.NbEntities((SMDSAbs_EntityType)i);
 840     }
 841   }
 842   else
 843   {
 844     // std-like iterator used to get coordinates of nodes of mesh element
 845     typedef SMDS_StdIterator< SMESH_TNodeXYZ, SMDS_ElemIteratorPtr > TXyzIterator;
 846
 847     SMESH_MesherHelper helper(theMesh);
 848     helper.SetSubShape(theShape);
 849
 850     const TopoDS_Face& geomFace = TopoDS::Face( theShape );
 851
 852     // take into account nodes on vertices
 853     TopExp_Explorer exp( theShape, TopAbs_VERTEX );
 854     for ( ; exp.More(); exp.Next() )
 855       theMesh.GetSubMesh( exp.Current())->Evaluate( aResMap );
 856
 857     // to count now many times a link between nodes encounters,
 858     // negative nb additionally means that a link is quadratic
 859     map<SMESH_TLink, int> linkCount;
 860     map<SMESH_TLink, int>::iterator link2Nb;
 861
 862     // count faces and nodes imported from groups
 863     set<const SMDS_MeshNode* > allNodes;
 864     gp_XY uv;
 865     double minGroupTol = 1e100;
 866     for ( size_t iG = 0; iG < srcGroups.size(); ++iG )
 867     {
 868       const SMESHDS_GroupBase* srcGroup = srcGroups[iG]->GetGroupDS();
 869       const double groupTol = 0.5 * sqrt( getMinElemSize2( srcGroup ));
 870       minGroupTol = std::min( groupTol, minGroupTol );
 871       SMDS_ElemIteratorPtr srcElems = srcGroup->GetElements();
 872       SMDS_MeshNode *tmpNode =helper.AddNode(0,0,0);
 873       while ( srcElems->more() ) // loop on group contents
 874       {
 875         const SMDS_MeshElement* face = srcElems->next();
 876         // find out if face is located on geomEdge by projecting
 877         // a gravity center of face to geomFace
 878         gp_XYZ gc(0,0,0);
 879         gc = accumulate( TXyzIterator(face->nodesIterator()), TXyzIterator(), gc)/face->NbNodes();
 880         tmpNode->setXYZ( gc.X(), gc.Y(), gc.Z());
 881         if ( helper.CheckNodeUV( geomFace, tmpNode, uv, groupTol, /*force=*/true ))
 882         {
 883           ++aVec[ face->GetEntityType() ];
 884
 885           // collect links
 886           int nbConers = face->NbCornerNodes();
 887           for ( int i = 0; i < face->NbNodes(); ++i )
 888           {
 889             const SMDS_MeshNode* n1 = face->GetNode(i);
 890             allNodes.insert( n1 );
 891             if ( i < nbConers )
 892             {
 893               const SMDS_MeshNode* n2 = face->GetNode( (i+1)%nbConers );
 894               link2Nb = linkCount.insert( make_pair( SMESH_TLink( n1, n2 ), 0)).first;
 895               if ( (*link2Nb).second )
 896                 link2Nb->second += (link2Nb->second < 0 ) ? -1 : 1;
 897               else
 898                 link2Nb->second += ( face->IsQuadratic() ) ? -1 : 1;
 899             }
 900           }
 901         }
 902       }
 903       helper.GetMeshDS()->RemoveNode(tmpNode);
 904     }
 905
 906     int nbNodes = allNodes.size();
 907     allNodes.clear();
 908
 909     // count nodes and edges on geom edges
 910
 911     double u;
 912     for ( exp.Init(theShape, TopAbs_EDGE); exp.More(); exp.Next() )
 913     {
 914       TopoDS_Edge geomEdge = TopoDS::Edge( exp.Current() );
 915       SMESH_subMesh* sm = theMesh.GetSubMesh( geomEdge );
 916       vector<int>& edgeVec = aResMap[sm];
 917       if ( edgeVec.empty() )
 918       {
 919         edgeVec.resize(SMDSEntity_Last,0);
 920         for ( link2Nb = linkCount.begin(); link2Nb != linkCount.end(); )
 921         {
 922           const SMESH_TLink& link = (*link2Nb).first;
 923           int nbFacesOfLink = Abs( link2Nb->second );
 924           bool eraseLink = ( nbFacesOfLink != 1 );
 925           if ( nbFacesOfLink == 1 )
 926           {
 927             if ( helper.CheckNodeU( geomEdge, link.node1(), u, minGroupTol, /*force=*/true )&&
 928                  helper.CheckNodeU( geomEdge, link.node2(), u, minGroupTol, /*force=*/true ))
 929             {
 930               bool isQuadratic = ( link2Nb->second < 0 );
 931               ++edgeVec[ isQuadratic ? SMDSEntity_Quad_Edge : SMDSEntity_Edge ];
 932               ++edgeVec[ SMDSEntity_Node ];
 933               --nbNodes;
 934               eraseLink = true;
 935             }
 936           }
 937           if ( eraseLink )
 938             linkCount.erase(link2Nb++);
 939           else
 940             link2Nb++;
 941         }
 942         if ( edgeVec[ SMDSEntity_Node] > 0 )
 943           --edgeVec[ SMDSEntity_Node ]; // for one node on vertex
 944       }
 945       else if ( !helper.IsSeamShape( geomEdge ) ||
 946                 geomEdge.Orientation() == TopAbs_FORWARD )
 947       {
 948         nbNodes -= 1+edgeVec[ SMDSEntity_Node ];
 949       }
 950     }
 951
 952     aVec[SMDSEntity_Node] = nbNodes;
 953   }
 954
 955   SMESH_subMesh * sm = theMesh.GetSubMesh(theShape);
 956   aResMap.insert(make_pair(sm,aVec));
 957
 958   return true;
 959 }