1 // Copyright (C) 2007-2014 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 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, or (at your option) any later version.
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
23 // File : StdMeshers_Prism_3D.cxx
25 // Created : Fri Oct 20 11:37:07 2006
26 // Author : Edward AGAPOV (eap)
28 #include "StdMeshers_Prism_3D.hxx"
30 #include "SMDS_EdgePosition.hxx"
31 #include "SMDS_VolumeOfNodes.hxx"
32 #include "SMDS_VolumeTool.hxx"
33 #include "SMESH_Comment.hxx"
34 #include "SMESH_Gen.hxx"
35 #include "SMESH_HypoFilter.hxx"
36 #include "SMESH_MesherHelper.hxx"
37 #include "StdMeshers_FaceSide.hxx"
38 #include "StdMeshers_ProjectionSource1D.hxx"
39 #include "StdMeshers_ProjectionSource2D.hxx"
40 #include "StdMeshers_ProjectionUtils.hxx"
41 #include "StdMeshers_Projection_1D.hxx"
42 #include "StdMeshers_Projection_1D2D.hxx"
43 #include "StdMeshers_Quadrangle_2D.hxx"
45 #include "utilities.h"
47 #include <BRepAdaptor_CompCurve.hxx>
48 #include <BRep_Tool.hxx>
49 #include <Bnd_B3d.hxx>
50 #include <Geom2dAdaptor_Curve.hxx>
51 #include <Geom2d_Line.hxx>
52 #include <Geom_Curve.hxx>
54 #include <TopExp_Explorer.hxx>
55 #include <TopTools_ListIteratorOfListOfShape.hxx>
56 #include <TopTools_ListOfShape.hxx>
57 #include <TopTools_MapOfShape.hxx>
58 #include <TopTools_SequenceOfShape.hxx>
67 #define RETURN_BAD_RESULT(msg) { MESSAGE(")-: Error: " << msg); return false; }
68 #define gpXYZ(n) SMESH_TNodeXYZ(n)
71 #define DBGOUT(msg) //cout << msg << endl;
72 #define SHOWYXZ(msg, xyz) \
73 // { gp_Pnt p (xyz); \
74 // cout << msg << " ("<< p.X() << "; " <<p.Y() << "; " <<p.Z() << ") " <<endl; }
77 #define SHOWYXZ(msg, xyz)
80 namespace TAssocTool = StdMeshers_ProjectionUtils;
82 typedef SMESH_Comment TCom;
84 enum { ID_BOT_FACE = SMESH_Block::ID_Fxy0,
85 ID_TOP_FACE = SMESH_Block::ID_Fxy1,
86 BOTTOM_EDGE = 0, TOP_EDGE, V0_EDGE, V1_EDGE, // edge IDs in face
87 NB_WALL_FACES = 4 }; //
91 //=======================================================================
93 * \brief Quadrangle algorithm
95 struct TQuadrangleAlgo : public StdMeshers_Quadrangle_2D
97 TQuadrangleAlgo(int studyId, SMESH_Gen* gen)
98 : StdMeshers_Quadrangle_2D( gen->GetANewId(), studyId, gen)
101 static StdMeshers_Quadrangle_2D* instance( SMESH_Algo* fatherAlgo,
102 SMESH_MesherHelper* helper=0)
104 static TQuadrangleAlgo* algo = new TQuadrangleAlgo( fatherAlgo->GetStudyId(),
105 fatherAlgo->GetGen() );
108 algo->myProxyMesh->GetMesh() != helper->GetMesh() )
109 algo->myProxyMesh.reset( new SMESH_ProxyMesh( *helper->GetMesh() ));
111 algo->myQuadList.clear();
114 algo->_quadraticMesh = helper->GetIsQuadratic();
119 //=======================================================================
121 * \brief Algorithm projecting 1D mesh
123 struct TProjction1dAlgo : public StdMeshers_Projection_1D
125 StdMeshers_ProjectionSource1D myHyp;
127 TProjction1dAlgo(int studyId, SMESH_Gen* gen)
128 : StdMeshers_Projection_1D( gen->GetANewId(), studyId, gen),
129 myHyp( gen->GetANewId(), studyId, gen)
131 StdMeshers_Projection_1D::_sourceHypo = & myHyp;
133 static TProjction1dAlgo* instance( SMESH_Algo* fatherAlgo )
135 static TProjction1dAlgo* algo = new TProjction1dAlgo( fatherAlgo->GetStudyId(),
136 fatherAlgo->GetGen() );
140 //=======================================================================
142 * \brief Algorithm projecting 2D mesh
144 struct TProjction2dAlgo : public StdMeshers_Projection_1D2D
146 StdMeshers_ProjectionSource2D myHyp;
148 TProjction2dAlgo(int studyId, SMESH_Gen* gen)
149 : StdMeshers_Projection_1D2D( gen->GetANewId(), studyId, gen),
150 myHyp( gen->GetANewId(), studyId, gen)
152 StdMeshers_Projection_2D::_sourceHypo = & myHyp;
154 static TProjction2dAlgo* instance( SMESH_Algo* fatherAlgo )
156 static TProjction2dAlgo* algo = new TProjction2dAlgo( fatherAlgo->GetStudyId(),
157 fatherAlgo->GetGen() );
161 //=======================================================================
163 * \brief Returns already computed EDGEs
165 void getPrecomputedEdges( SMESH_MesherHelper& theHelper,
166 const TopoDS_Shape& theShape,
167 vector< TopoDS_Edge >& theEdges)
171 SMESHDS_Mesh* meshDS = theHelper.GetMeshDS();
174 TopTools_IndexedMapOfShape edges;
175 TopExp::MapShapes( theShape, TopAbs_EDGE, edges );
176 for ( int iE = 1; iE <= edges.Extent(); ++iE )
178 const TopoDS_Shape edge = edges( iE );
179 if (( ! ( sm = meshDS->MeshElements( edge ))) ||
180 ( sm->NbElements() == 0 ))
183 // there must not be FACEs meshed with triangles and sharing a computed EDGE
184 // as the precomputed EDGEs are used for propagation other to 'vertical' EDGEs
185 bool faceFound = false;
186 PShapeIteratorPtr faceIt =
187 theHelper.GetAncestors( edge, *theHelper.GetMesh(), TopAbs_FACE );
188 while ( const TopoDS_Shape* face = faceIt->next() )
190 if (( sm = meshDS->MeshElements( *face )) &&
191 ( sm->NbElements() > 0 ) &&
192 ( !theHelper.IsSameElemGeometry( sm, SMDSGeom_QUADRANGLE ) ))
198 theEdges.push_back( TopoDS::Edge( edge ));
202 //================================================================================
204 * \brief Make \a botE be the BOTTOM_SIDE of \a quad.
205 * Return false if the BOTTOM_SIDE is composite
207 //================================================================================
209 bool setBottomEdge( const TopoDS_Edge& botE,
210 FaceQuadStruct::Ptr& quad,
211 const TopoDS_Shape& face)
213 quad->side[ QUAD_TOP_SIDE ].grid->Reverse();
214 quad->side[ QUAD_LEFT_SIDE ].grid->Reverse();
216 bool isComposite = false;
217 for ( size_t i = 0; i < quad->side.size(); ++i )
219 StdMeshers_FaceSidePtr quadSide = quad->side[i];
220 for ( int iE = 0; iE < quadSide->NbEdges(); ++iE )
221 if ( botE.IsSame( quadSide->Edge( iE )))
223 if ( quadSide->NbEdges() > 1 )
224 isComposite = true; //return false;
226 i = quad->side.size(); // to quit from the outer loop
230 if ( edgeIndex != QUAD_BOTTOM_SIDE )
231 quad->shift( quad->side.size() - edgeIndex, /*keepUnitOri=*/false );
233 quad->face = TopoDS::Face( face );
238 //================================================================================
240 * \brief Return iterator pointing to node column for the given parameter
241 * \param columnsMap - node column map
242 * \param parameter - parameter
243 * \retval TParam2ColumnMap::iterator - result
245 * it returns closest left column
247 //================================================================================
249 TParam2ColumnIt getColumn( const TParam2ColumnMap* columnsMap,
250 const double parameter )
252 TParam2ColumnIt u_col = columnsMap->upper_bound( parameter );
253 if ( u_col != columnsMap->begin() )
255 return u_col; // return left column
258 //================================================================================
260 * \brief Return nodes around given parameter and a ratio
261 * \param column - node column
262 * \param param - parameter
263 * \param node1 - lower node
264 * \param node2 - upper node
265 * \retval double - ratio
267 //================================================================================
269 double getRAndNodes( const TNodeColumn* column,
271 const SMDS_MeshNode* & node1,
272 const SMDS_MeshNode* & node2)
274 if ( param >= 1.0 || column->size() == 1) {
275 node1 = node2 = column->back();
279 int i = int( param * ( column->size() - 1 ));
280 double u0 = double( i )/ double( column->size() - 1 );
281 double r = ( param - u0 ) * ( column->size() - 1 );
283 node1 = (*column)[ i ];
284 node2 = (*column)[ i + 1];
288 //================================================================================
290 * \brief Compute boundary parameters of face parts
291 * \param nbParts - nb of parts to split columns into
292 * \param columnsMap - node columns of the face to split
293 * \param params - computed parameters
295 //================================================================================
297 void splitParams( const int nbParts,
298 const TParam2ColumnMap* columnsMap,
299 vector< double > & params)
302 params.reserve( nbParts + 1 );
303 TParam2ColumnIt last_par_col = --columnsMap->end();
304 double par = columnsMap->begin()->first; // 0.
305 double parLast = last_par_col->first;
306 params.push_back( par );
307 for ( int i = 0; i < nbParts - 1; ++ i )
309 double partSize = ( parLast - par ) / double ( nbParts - i );
310 TParam2ColumnIt par_col = getColumn( columnsMap, par + partSize );
311 if ( par_col->first == par ) {
313 if ( par_col == last_par_col ) {
314 while ( i < nbParts - 1 )
315 params.push_back( par + partSize * i++ );
319 par = par_col->first;
320 params.push_back( par );
322 params.push_back( parLast ); // 1.
325 //================================================================================
327 * \brief Return coordinate system for z-th layer of nodes
329 //================================================================================
331 gp_Ax2 getLayerCoordSys(const int z,
332 const vector< const TNodeColumn* >& columns,
335 // gravity center of a layer
338 for ( int i = 0; i < columns.size(); ++i )
340 O += gpXYZ( (*columns[ i ])[ z ]);
341 if ( vertexCol < 0 &&
342 columns[ i ]->front()->GetPosition()->GetTypeOfPosition() == SMDS_TOP_VERTEX )
349 int iPrev = columns.size()-1;
350 for ( int i = 0; i < columns.size(); ++i )
352 gp_Vec v1( O, gpXYZ( (*columns[ iPrev ])[ z ]));
353 gp_Vec v2( O, gpXYZ( (*columns[ i ] )[ z ]));
358 if ( vertexCol >= 0 )
360 O = gpXYZ( (*columns[ vertexCol ])[ z ]);
362 if ( xColumn < 0 || xColumn >= columns.size() )
364 // select a column for X dir
366 for ( int i = 0; i < columns.size(); ++i )
368 double dist = ( O - gpXYZ((*columns[ i ])[ z ])).SquareModulus();
369 if ( dist > maxDist )
378 gp_Vec X( O, gpXYZ( (*columns[ xColumn ])[ z ]));
380 return gp_Ax2( O, Z, X);
383 //================================================================================
385 * \brief Removes submeshes that are or can be meshed with regular grid from given list
386 * \retval int - nb of removed submeshes
388 //================================================================================
390 int removeQuasiQuads(list< SMESH_subMesh* >& notQuadSubMesh,
391 SMESH_MesherHelper* helper,
392 StdMeshers_Quadrangle_2D* quadAlgo)
395 //SMESHDS_Mesh* mesh = notQuadSubMesh.front()->GetFather()->GetMeshDS();
396 list< SMESH_subMesh* >::iterator smIt = notQuadSubMesh.begin();
397 while ( smIt != notQuadSubMesh.end() )
399 SMESH_subMesh* faceSm = *smIt;
400 SMESHDS_SubMesh* faceSmDS = faceSm->GetSubMeshDS();
401 int nbQuads = faceSmDS ? faceSmDS->NbElements() : 0;
404 toRemove = helper->IsStructured( faceSm );
406 toRemove = quadAlgo->CheckNbEdges( *helper->GetMesh(),
407 faceSm->GetSubShape() );
408 nbRemoved += toRemove;
410 smIt = notQuadSubMesh.erase( smIt );
418 //================================================================================
420 * \brief Return and angle between two EDGEs
421 * \return double - the angle normalized so that
428 //================================================================================
430 double normAngle(const TopoDS_Edge & E1, const TopoDS_Edge & E2, const TopoDS_Face & F)
432 return SMESH_MesherHelper::GetAngle( E1, E2, F ) / ( 0.5 * M_PI );
435 //================================================================================
437 * Consider continuous straight EDGES as one side - mark them to unite
439 //================================================================================
441 int countNbSides( const Prism_3D::TPrismTopo & thePrism,
442 vector<int> & nbUnitePerEdge,
443 vector< double > & edgeLength)
445 int nbEdges = thePrism.myNbEdgesInWires.front(); // nb outer edges
446 int nbSides = nbEdges;
449 list< TopoDS_Edge >::const_iterator edgeIt = thePrism.myBottomEdges.begin();
450 std::advance( edgeIt, nbEdges-1 );
451 TopoDS_Edge prevE = *edgeIt;
452 // bool isPrevStraight = SMESH_Algo::IsStraight( prevE );
453 int iPrev = nbEdges - 1;
455 int iUnite = -1; // the first of united EDGEs
457 // analyse angles between EDGEs
459 vector< bool > isCorner( nbEdges );
460 edgeIt = thePrism.myBottomEdges.begin();
461 for ( int iE = 0; iE < nbEdges; ++iE, ++edgeIt )
463 const TopoDS_Edge& curE = *edgeIt;
464 edgeLength[ iE ] = SMESH_Algo::EdgeLength( curE );
466 // double normAngle = normAngle( prevE, curE, thePrism.myBottom );
467 // isCorner[ iE ] = false;
468 // if ( normAngle < 2.0 )
470 // if ( normAngle < 0.001 ) // straight or obtuse angle
472 // // unite EDGEs in order not to put a corner of the unit quadrangle at this VERTEX
475 // nbUnitePerEdge[ iUnite ]++;
476 // nbUnitePerEdge[ iE ] = -1;
481 // isCorner[ iE ] = true;
491 // define which of corners to put on a side of the unit quadrangle
493 // edgeIt = thePrism.myBottomEdges.begin();
494 // for ( int iE = 0; iE < nbEdges; ++iE, ++edgeIt )
496 // const TopoDS_Edge& curE = *edgeIt;
497 // edgeLength[ iE ] = SMESH_Algo::EdgeLength( curE );
499 // const bool isCurStraight = SMESH_Algo::IsStraight( curE );
500 // if ( isPrevStraight && isCurStraight && SMESH_Algo::IsContinuous( prevE, curE ))
504 // nbUnitePerEdge[ iUnite ]++;
505 // nbUnitePerEdge[ iE ] = -1;
513 // isPrevStraight = isCurStraight;
520 void pointsToPython(const std::vector<gp_XYZ>& p)
523 for ( int i = SMESH_Block::ID_V000; i < p.size(); ++i )
525 cout << "mesh.AddNode( " << p[i].X() << ", "<< p[i].Y() << ", "<< p[i].Z() << ") # " << i <<" " ;
526 SMESH_Block::DumpShapeID( i, cout ) << endl;
532 //=======================================================================
533 //function : StdMeshers_Prism_3D
535 //=======================================================================
537 StdMeshers_Prism_3D::StdMeshers_Prism_3D(int hypId, int studyId, SMESH_Gen* gen)
538 :SMESH_3D_Algo(hypId, studyId, gen)
541 _shapeType = (1 << TopAbs_SOLID); // 1 bit per shape type
542 _onlyUnaryInput = false; // accept all SOLIDs at once
543 _requireDiscreteBoundary = false; // mesh FACEs and EDGEs by myself
544 _supportSubmeshes = true; // "source" FACE must be meshed by other algo
545 _neededLowerHyps[ 1 ] = true; // suppress warning on hiding a global 1D algo
546 _neededLowerHyps[ 2 ] = true; // suppress warning on hiding a global 2D algo
548 //myProjectTriangles = false;
549 mySetErrorToSM = true; // to pass an error to a sub-mesh of a current solid or not
552 //================================================================================
556 //================================================================================
558 StdMeshers_Prism_3D::~StdMeshers_Prism_3D()
561 //=======================================================================
562 //function : CheckHypothesis
564 //=======================================================================
566 bool StdMeshers_Prism_3D::CheckHypothesis(SMESH_Mesh& aMesh,
567 const TopoDS_Shape& aShape,
568 SMESH_Hypothesis::Hypothesis_Status& aStatus)
570 // Check shape geometry
572 aStatus = SMESH_Hypothesis::HYP_BAD_GEOMETRY;
574 // find not quadrangle faces
575 list< TopoDS_Shape > notQuadFaces;
576 int nbEdge, nbWire, nbFace = 0;
577 TopExp_Explorer exp( aShape, TopAbs_FACE );
578 for ( ; exp.More(); exp.Next() ) {
580 const TopoDS_Shape& face = exp.Current();
581 nbEdge = TAssocTool::Count( face, TopAbs_EDGE, 0 );
582 nbWire = TAssocTool::Count( face, TopAbs_WIRE, 0 );
583 if ( nbEdge!= 4 || nbWire!= 1 ) {
584 if ( !notQuadFaces.empty() ) {
585 if ( TAssocTool::Count( notQuadFaces.back(), TopAbs_EDGE, 0 ) != nbEdge ||
586 TAssocTool::Count( notQuadFaces.back(), TopAbs_WIRE, 0 ) != nbWire )
587 RETURN_BAD_RESULT("Different not quad faces");
589 notQuadFaces.push_back( face );
592 if ( !notQuadFaces.empty() )
594 if ( notQuadFaces.size() != 2 )
595 RETURN_BAD_RESULT("Bad nb not quad faces: " << notQuadFaces.size());
597 // check total nb faces
598 nbEdge = TAssocTool::Count( notQuadFaces.back(), TopAbs_EDGE, 0 );
599 if ( nbFace != nbEdge + 2 )
600 RETURN_BAD_RESULT("Bad nb of faces: " << nbFace << " but must be " << nbEdge + 2);
604 aStatus = SMESH_Hypothesis::HYP_OK;
608 //=======================================================================
610 //purpose : Compute mesh on a COMPOUND of SOLIDs
611 //=======================================================================
613 bool StdMeshers_Prism_3D::Compute(SMESH_Mesh& theMesh, const TopoDS_Shape& theShape)
615 SMESH_MesherHelper helper( theMesh );
618 int nbSolids = helper.Count( theShape, TopAbs_SOLID, /*skipSame=*/false );
622 TopTools_IndexedDataMapOfShapeListOfShape faceToSolids;
623 TopExp::MapShapesAndAncestors( theShape, TopAbs_FACE, TopAbs_SOLID, faceToSolids );
625 // look for meshed FACEs ("source" FACEs) that must be prism bottoms
626 list< TopoDS_Face > meshedFaces, notQuadMeshedFaces, notQuadFaces;
627 const bool meshHasQuads = ( theMesh.NbQuadrangles() > 0 );
628 //StdMeshers_Quadrangle_2D* quadAlgo = TQuadrangleAlgo::instance( this );
629 for ( int iF = 1; iF <= faceToSolids.Extent(); ++iF )
631 const TopoDS_Face& face = TopoDS::Face( faceToSolids.FindKey( iF ));
632 SMESH_subMesh* faceSM = theMesh.GetSubMesh( face );
633 if ( !faceSM->IsEmpty() )
635 if ( !meshHasQuads ||
636 !helper.IsSameElemGeometry( faceSM->GetSubMeshDS(), SMDSGeom_QUADRANGLE ) ||
637 !helper.IsStructured( faceSM )
639 notQuadMeshedFaces.push_front( face );
640 else if ( myHelper->Count( face, TopAbs_EDGE, /*ignoreSame=*/false ) != 4 )
641 meshedFaces.push_front( face );
643 meshedFaces.push_back( face );
645 // not add not quadrilateral FACE as we can't compute it
646 // else if ( !quadAlgo->CheckNbEdges( theMesh, face ))
647 // // not add not quadrilateral FACE as it can be a prism side
648 // // else if ( myHelper->Count( face, TopAbs_EDGE, /*ignoreSame=*/false ) != 4 )
650 // notQuadFaces.push_back( face );
653 // notQuadFaces are of medium priority, put them before ordinary meshed faces
654 meshedFaces.splice( meshedFaces.begin(), notQuadFaces );
655 // notQuadMeshedFaces are of highest priority, put them before notQuadFaces
656 meshedFaces.splice( meshedFaces.begin(), notQuadMeshedFaces );
658 Prism_3D::TPrismTopo prism;
663 if ( !meshedFaces.empty() )
664 prism.myBottom = meshedFaces.front();
665 return ( initPrism( prism, TopExp_Explorer( theShape, TopAbs_SOLID ).Current() ) &&
669 // find propagation chains from already computed EDGEs
670 vector< TopoDS_Edge > computedEdges;
671 getPrecomputedEdges( helper, theShape, computedEdges );
672 myPropagChains = new TopTools_IndexedMapOfShape[ computedEdges.size() + 1 ];
673 SMESHUtils::ArrayDeleter< TopTools_IndexedMapOfShape > pcDel( myPropagChains );
674 for ( size_t i = 0, nb = 0; i < computedEdges.size(); ++i )
676 StdMeshers_ProjectionUtils::GetPropagationEdge( &theMesh, TopoDS_Edge(),
677 computedEdges[i], myPropagChains + nb );
678 if ( myPropagChains[ nb ].Extent() < 2 ) // an empty map is a termination sign
679 myPropagChains[ nb ].Clear();
684 TopTools_MapOfShape meshedSolids;
685 list< Prism_3D::TPrismTopo > meshedPrism;
686 TopTools_ListIteratorOfListOfShape solidIt;
688 while ( meshedSolids.Extent() < nbSolids )
690 if ( _computeCanceled )
691 return toSM( error( SMESH_ComputeError::New(COMPERR_CANCELED)));
693 // compute prisms having avident computed source FACE
694 while ( !meshedFaces.empty() )
696 TopoDS_Face face = meshedFaces.front();
697 meshedFaces.pop_front();
698 TopTools_ListOfShape& solidList = faceToSolids.ChangeFromKey( face );
699 while ( !solidList.IsEmpty() )
701 TopoDS_Shape solid = solidList.First();
702 solidList.RemoveFirst();
703 if ( meshedSolids.Add( solid ))
706 prism.myBottom = face;
707 if ( !initPrism( prism, solid ) ||
711 SMESHDS_SubMesh* smDS = theMesh.GetMeshDS()->MeshElements( prism.myTop );
712 if ( !myHelper->IsSameElemGeometry( smDS, SMDSGeom_QUADRANGLE ))
714 meshedFaces.push_front( prism.myTop );
716 meshedPrism.push_back( prism );
720 if ( meshedSolids.Extent() == nbSolids )
723 // below in the loop we try to find source FACEs somehow
725 // project mesh from source FACEs of computed prisms to
726 // prisms sharing wall FACEs
727 list< Prism_3D::TPrismTopo >::iterator prismIt = meshedPrism.begin();
728 for ( ; prismIt != meshedPrism.end(); ++prismIt )
730 for ( size_t iW = 0; iW < prismIt->myWallQuads.size(); ++iW )
732 Prism_3D::TQuadList::iterator wQuad = prismIt->myWallQuads[iW].begin();
733 for ( ; wQuad != prismIt->myWallQuads[iW].end(); ++ wQuad )
735 const TopoDS_Face& wFace = (*wQuad)->face;
736 TopTools_ListOfShape& solidList = faceToSolids.ChangeFromKey( wFace );
737 solidIt.Initialize( solidList );
738 while ( solidIt.More() )
740 const TopoDS_Shape& solid = solidIt.Value();
741 if ( meshedSolids.Contains( solid )) {
742 solidList.Remove( solidIt );
743 continue; // already computed prism
745 // find a source FACE of the SOLID: it's a FACE sharing a bottom EDGE with wFace
746 const TopoDS_Edge& wEdge = (*wQuad)->side[ QUAD_TOP_SIDE ].grid->Edge(0);
747 PShapeIteratorPtr faceIt = myHelper->GetAncestors( wEdge, *myHelper->GetMesh(),
749 while ( const TopoDS_Shape* f = faceIt->next() )
751 const TopoDS_Face& candidateF = TopoDS::Face( *f );
753 prism.myBottom = candidateF;
754 mySetErrorToSM = false;
755 if ( !myHelper->IsSubShape( candidateF, prismIt->myShape3D ) &&
756 myHelper->IsSubShape( candidateF, solid ) &&
757 !myHelper->GetMesh()->GetSubMesh( candidateF )->IsMeshComputed() &&
758 initPrism( prism, solid ) &&
759 project2dMesh( prismIt->myBottom, candidateF))
761 mySetErrorToSM = true;
762 if ( !compute( prism ))
764 SMESHDS_SubMesh* smDS = theMesh.GetMeshDS()->MeshElements( prism.myTop );
765 if ( !myHelper->IsSameElemGeometry( smDS, SMDSGeom_QUADRANGLE ))
767 meshedFaces.push_front( prism.myTop );
768 meshedFaces.push_front( prism.myBottom );
770 meshedPrism.push_back( prism );
771 meshedSolids.Add( solid );
775 mySetErrorToSM = true;
777 if ( meshedSolids.Contains( solid ))
778 solidList.Remove( solidIt );
784 if ( !meshedFaces.empty() )
785 break; // to compute prisms with avident sources
788 // find FACEs with local 1D hyps, which has to be computed by now,
789 // or at least any computed FACEs
790 for ( int iF = 1; ( meshedFaces.empty() && iF < faceToSolids.Extent() ); ++iF )
792 const TopoDS_Face& face = TopoDS::Face( faceToSolids.FindKey( iF ));
793 const TopTools_ListOfShape& solidList = faceToSolids.FindFromKey( face );
794 if ( solidList.IsEmpty() ) continue;
795 SMESH_subMesh* faceSM = theMesh.GetSubMesh( face );
796 if ( !faceSM->IsEmpty() )
798 meshedFaces.push_back( face ); // lower priority
802 bool allSubMeComputed = true;
803 SMESH_subMeshIteratorPtr smIt = faceSM->getDependsOnIterator(false,true);
804 while ( smIt->more() && allSubMeComputed )
805 allSubMeComputed = smIt->next()->IsMeshComputed();
806 if ( allSubMeComputed )
808 faceSM->ComputeStateEngine( SMESH_subMesh::COMPUTE );
809 if ( !faceSM->IsEmpty() )
810 meshedFaces.push_front( face ); // higher priority
812 faceSM->ComputeStateEngine( SMESH_subMesh::CHECK_COMPUTE_STATE );
818 // TODO. there are other ways to find out the source FACE:
819 // propagation, topological similarity, ect.
821 // simply try to mesh all not meshed SOLIDs
822 if ( meshedFaces.empty() )
824 for ( TopExp_Explorer solid( theShape, TopAbs_SOLID ); solid.More(); solid.Next() )
826 mySetErrorToSM = false;
828 if ( !meshedSolids.Contains( solid.Current() ) &&
829 initPrism( prism, solid.Current() ))
831 mySetErrorToSM = true;
832 if ( !compute( prism ))
834 meshedFaces.push_front( prism.myTop );
835 meshedFaces.push_front( prism.myBottom );
836 meshedPrism.push_back( prism );
837 meshedSolids.Add( solid.Current() );
839 mySetErrorToSM = true;
843 if ( meshedFaces.empty() ) // set same error to 10 not-computed solids
845 SMESH_ComputeErrorPtr err = SMESH_ComputeError::New
846 ( COMPERR_BAD_INPUT_MESH, "No meshed source face found", this );
848 const int maxNbErrors = 10; // limit nb errors not to overload the Compute dialog
849 TopExp_Explorer solid( theShape, TopAbs_SOLID );
850 for ( int i = 0; ( i < maxNbErrors && solid.More() ); ++i, solid.Next() )
851 if ( !meshedSolids.Contains( solid.Current() ))
853 SMESH_subMesh* sm = theMesh.GetSubMesh( solid.Current() );
854 sm->GetComputeError() = err;
862 //================================================================================
864 * \brief Find wall faces by bottom edges
866 //================================================================================
868 bool StdMeshers_Prism_3D::getWallFaces( Prism_3D::TPrismTopo & thePrism,
869 const int totalNbFaces)
871 thePrism.myWallQuads.clear();
873 SMESH_Mesh* mesh = myHelper->GetMesh();
875 StdMeshers_Quadrangle_2D* quadAlgo = TQuadrangleAlgo::instance( this, myHelper );
877 TopTools_MapOfShape faceMap;
878 TopTools_IndexedDataMapOfShapeListOfShape edgeToFaces;
879 TopExp::MapShapesAndAncestors( thePrism.myShape3D,
880 TopAbs_EDGE, TopAbs_FACE, edgeToFaces );
882 // ------------------------------
883 // Get the 1st row of wall FACEs
884 // ------------------------------
886 list< TopoDS_Edge >::iterator edge = thePrism.myBottomEdges.begin();
887 std::list< int >::iterator nbE = thePrism.myNbEdgesInWires.begin();
890 while ( edge != thePrism.myBottomEdges.end() )
893 if ( BRep_Tool::Curve( *edge, f,l ).IsNull() )
895 edge = thePrism.myBottomEdges.erase( edge );
901 TopTools_ListIteratorOfListOfShape faceIt( edgeToFaces.FindFromKey( *edge ));
902 for ( ; faceIt.More(); faceIt.Next() )
904 const TopoDS_Face& face = TopoDS::Face( faceIt.Value() );
905 if ( !thePrism.myBottom.IsSame( face ))
907 Prism_3D::TQuadList quadList( 1, quadAlgo->CheckNbEdges( *mesh, face ));
908 if ( !quadList.back() )
909 return toSM( error(TCom("Side face #") << shapeID( face )
910 << " not meshable with quadrangles"));
911 bool isCompositeBase = ! setBottomEdge( *edge, quadList.back(), face );
912 if ( isCompositeBase )
914 // it's OK if all EDGEs of the bottom side belongs to the bottom FACE
915 StdMeshers_FaceSidePtr botSide = quadList.back()->side[ QUAD_BOTTOM_SIDE ];
916 for ( int iE = 0; iE < botSide->NbEdges(); ++iE )
917 if ( !myHelper->IsSubShape( botSide->Edge(iE), thePrism.myBottom ))
918 return toSM( error(TCom("Composite 'horizontal' edges are not supported")));
920 if ( faceMap.Add( face ))
921 thePrism.myWallQuads.push_back( quadList );
934 // -------------------------
935 // Find the rest wall FACEs
936 // -------------------------
938 // Compose a vector of indixes of right neighbour FACE for each wall FACE
939 // that is not so evident in case of several WIREs in the bottom FACE
940 thePrism.myRightQuadIndex.clear();
941 for ( size_t i = 0; i < thePrism.myWallQuads.size(); ++i )
942 thePrism.myRightQuadIndex.push_back( i+1 );
943 list< int >::iterator nbEinW = thePrism.myNbEdgesInWires.begin();
944 for ( int iLeft = 0; nbEinW != thePrism.myNbEdgesInWires.end(); ++nbEinW )
946 thePrism.myRightQuadIndex[ iLeft + *nbEinW - 1 ] = iLeft; // 1st EDGE index of a current WIRE
950 while ( totalNbFaces - faceMap.Extent() > 2 )
952 // find wall FACEs adjacent to each of wallQuads by the right side EDGE
955 nbKnownFaces = faceMap.Extent();
956 StdMeshers_FaceSidePtr rightSide, topSide; // sides of the quad
957 for ( size_t i = 0; i < thePrism.myWallQuads.size(); ++i )
959 rightSide = thePrism.myWallQuads[i].back()->side[ QUAD_RIGHT_SIDE ];
960 for ( int iE = 0; iE < rightSide->NbEdges(); ++iE ) // rightSide can be composite
962 const TopoDS_Edge & rightE = rightSide->Edge( iE );
963 TopTools_ListIteratorOfListOfShape face( edgeToFaces.FindFromKey( rightE ));
964 for ( ; face.More(); face.Next() )
965 if ( faceMap.Add( face.Value() ))
967 // a new wall FACE encountered, store it in thePrism.myWallQuads
968 const int iRight = thePrism.myRightQuadIndex[i];
969 topSide = thePrism.myWallQuads[ iRight ].back()->side[ QUAD_TOP_SIDE ];
970 const TopoDS_Edge& newBotE = topSide->Edge(0);
971 const TopoDS_Shape& newWallF = face.Value();
972 thePrism.myWallQuads[ iRight ].push_back( quadAlgo->CheckNbEdges( *mesh, newWallF ));
973 if ( !thePrism.myWallQuads[ iRight ].back() )
974 return toSM( error(TCom("Side face #") << shapeID( newWallF ) <<
975 " not meshable with quadrangles"));
976 if ( ! setBottomEdge( newBotE, thePrism.myWallQuads[ iRight ].back(), newWallF ))
977 return toSM( error(TCom("Composite 'horizontal' edges are not supported")));
981 } while ( nbKnownFaces != faceMap.Extent() );
983 // find wall FACEs adjacent to each of thePrism.myWallQuads by the top side EDGE
984 if ( totalNbFaces - faceMap.Extent() > 2 )
986 for ( size_t i = 0; i < thePrism.myWallQuads.size(); ++i )
988 StdMeshers_FaceSidePtr topSide = thePrism.myWallQuads[i].back()->side[ QUAD_TOP_SIDE ];
989 const TopoDS_Edge & topE = topSide->Edge( 0 );
990 if ( topSide->NbEdges() > 1 )
991 return toSM( error(COMPERR_BAD_SHAPE, TCom("Side face #") <<
992 shapeID( thePrism.myWallQuads[i].back()->face )
993 << " has a composite top edge"));
994 TopTools_ListIteratorOfListOfShape faceIt( edgeToFaces.FindFromKey( topE ));
995 for ( ; faceIt.More(); faceIt.Next() )
996 if ( faceMap.Add( faceIt.Value() ))
998 // a new wall FACE encountered, store it in wallQuads
999 thePrism.myWallQuads[ i ].push_back( quadAlgo->CheckNbEdges( *mesh, faceIt.Value() ));
1000 if ( !thePrism.myWallQuads[ i ].back() )
1001 return toSM( error(TCom("Side face #") << shapeID( faceIt.Value() ) <<
1002 " not meshable with quadrangles"));
1003 if ( ! setBottomEdge( topE, thePrism.myWallQuads[ i ].back(), faceIt.Value() ))
1004 return toSM( error(TCom("Composite 'horizontal' edges are not supported")));
1005 if ( totalNbFaces - faceMap.Extent() == 2 )
1007 i = thePrism.myWallQuads.size(); // to quit from the outer loop
1013 } // while ( totalNbFaces - faceMap.Extent() > 2 )
1015 // ------------------
1016 // Find the top FACE
1017 // ------------------
1019 if ( thePrism.myTop.IsNull() )
1021 // now only top and bottom FACEs are not in the faceMap
1022 faceMap.Add( thePrism.myBottom );
1023 for ( TopExp_Explorer f( thePrism.myShape3D, TopAbs_FACE );f.More(); f.Next() )
1024 if ( !faceMap.Contains( f.Current() )) {
1025 thePrism.myTop = TopoDS::Face( f.Current() );
1028 if ( thePrism.myTop.IsNull() )
1029 return toSM( error("Top face not found"));
1032 // Check that the top FACE shares all the top EDGEs
1033 for ( size_t i = 0; i < thePrism.myWallQuads.size(); ++i )
1035 StdMeshers_FaceSidePtr topSide = thePrism.myWallQuads[i].back()->side[ QUAD_TOP_SIDE ];
1036 const TopoDS_Edge & topE = topSide->Edge( 0 );
1037 if ( !myHelper->IsSubShape( topE, thePrism.myTop ))
1038 return toSM( error( TCom("Wrong source face (#") << shapeID( thePrism.myBottom )));
1044 //=======================================================================
1045 //function : compute
1046 //purpose : Compute mesh on a SOLID
1047 //=======================================================================
1049 bool StdMeshers_Prism_3D::compute(const Prism_3D::TPrismTopo& thePrism)
1051 myHelper->IsQuadraticSubMesh( thePrism.myShape3D );
1052 if ( _computeCanceled )
1053 return toSM( error( SMESH_ComputeError::New(COMPERR_CANCELED)));
1055 // Make all side FACEs of thePrism meshed with quads
1056 if ( !computeWalls( thePrism ))
1059 // Analyse mesh and geometry to find all block sub-shapes and submeshes
1060 if ( !myBlock.Init( myHelper, thePrism ))
1061 return toSM( error( myBlock.GetError()));
1063 SMESHDS_Mesh* meshDS = myHelper->GetMeshDS();
1065 int volumeID = meshDS->ShapeToIndex( thePrism.myShape3D );
1067 // Try to get gp_Trsf to get all nodes from bottom ones
1068 vector<gp_Trsf> trsf;
1069 gp_Trsf bottomToTopTrsf;
1070 if ( !myBlock.GetLayersTransformation( trsf, thePrism ))
1072 else if ( !trsf.empty() )
1073 bottomToTopTrsf = trsf.back();
1075 // To compute coordinates of a node inside a block, it is necessary to know
1076 // 1. normalized parameters of the node by which
1077 // 2. coordinates of node projections on all block sub-shapes are computed
1079 // So we fill projections on vertices at once as they are same for all nodes
1080 myShapeXYZ.resize( myBlock.NbSubShapes() );
1081 for ( int iV = SMESH_Block::ID_FirstV; iV < SMESH_Block::ID_FirstE; ++iV ) {
1082 myBlock.VertexPoint( iV, myShapeXYZ[ iV ]);
1083 SHOWYXZ("V point " <<iV << " ", myShapeXYZ[ iV ]);
1086 // Projections on the top and bottom faces are taken from nodes existing
1087 // on these faces; find correspondence between bottom and top nodes
1088 myBotToColumnMap.clear();
1089 if ( !assocOrProjBottom2Top( bottomToTopTrsf, thePrism ) ) // it also fills myBotToColumnMap
1093 // Create nodes inside the block
1095 // try to use transformation (issue 0020680)
1096 if ( !trsf.empty() )
1098 // loop on nodes inside the bottom face
1099 TNode2ColumnMap::iterator bot_column = myBotToColumnMap.begin();
1100 for ( ; bot_column != myBotToColumnMap.end(); ++bot_column )
1102 const Prism_3D::TNode& tBotNode = bot_column->first; // bottom TNode
1103 if ( tBotNode.GetPositionType() != SMDS_TOP_FACE )
1104 continue; // node is not inside face
1106 // column nodes; middle part of the column are zero pointers
1107 TNodeColumn& column = bot_column->second;
1108 TNodeColumn::iterator columnNodes = column.begin();
1109 for ( int z = 0; columnNodes != column.end(); ++columnNodes, ++z)
1111 const SMDS_MeshNode* & node = *columnNodes;
1112 if ( node ) continue; // skip bottom or top node
1114 gp_XYZ coords = tBotNode.GetCoords();
1115 trsf[z-1].Transforms( coords );
1116 node = meshDS->AddNode( coords.X(), coords.Y(), coords.Z() );
1117 meshDS->SetNodeInVolume( node, volumeID );
1119 } // loop on bottom nodes
1121 else // use block approach
1123 // loop on nodes inside the bottom face
1124 Prism_3D::TNode prevBNode;
1125 TNode2ColumnMap::iterator bot_column = myBotToColumnMap.begin();
1126 for ( ; bot_column != myBotToColumnMap.end(); ++bot_column )
1128 const Prism_3D::TNode& tBotNode = bot_column->first; // bottom TNode
1129 if ( tBotNode.GetPositionType() != SMDS_TOP_FACE )
1130 continue; // node is not inside the FACE
1132 // column nodes; middle part of the column are zero pointers
1133 TNodeColumn& column = bot_column->second;
1135 gp_XYZ botParams, topParams;
1136 if ( !tBotNode.HasParams() )
1138 // compute bottom node parameters
1139 gp_XYZ paramHint(-1,-1,-1);
1140 if ( prevBNode.IsNeighbor( tBotNode ))
1141 paramHint = prevBNode.GetParams();
1142 if ( !myBlock.ComputeParameters( tBotNode.GetCoords(), tBotNode.ChangeParams(),
1143 ID_BOT_FACE, paramHint ))
1144 return toSM( error(TCom("Can't compute normalized parameters for node ")
1145 << tBotNode.myNode->GetID() << " on the face #"
1146 << myBlock.SubMesh( ID_BOT_FACE )->GetId() ));
1147 prevBNode = tBotNode;
1149 botParams = topParams = tBotNode.GetParams();
1150 topParams.SetZ( 1 );
1152 // compute top node parameters
1153 if ( column.size() > 2 ) {
1154 gp_Pnt topCoords = gpXYZ( column.back() );
1155 if ( !myBlock.ComputeParameters( topCoords, topParams, ID_TOP_FACE, topParams ))
1156 return toSM( error(TCom("Can't compute normalized parameters ")
1157 << "for node " << column.back()->GetID()
1158 << " on the face #"<< column.back()->getshapeId() ));
1161 else // top nodes are created by projection using parameters
1163 botParams = topParams = tBotNode.GetParams();
1164 topParams.SetZ( 1 );
1167 myShapeXYZ[ ID_BOT_FACE ] = tBotNode.GetCoords();
1168 myShapeXYZ[ ID_TOP_FACE ] = gpXYZ( column.back() );
1171 TNodeColumn::iterator columnNodes = column.begin();
1172 for ( int z = 0; columnNodes != column.end(); ++columnNodes, ++z)
1174 const SMDS_MeshNode* & node = *columnNodes;
1175 if ( node ) continue; // skip bottom or top node
1177 // params of a node to create
1178 double rz = (double) z / (double) ( column.size() - 1 );
1179 gp_XYZ params = botParams * ( 1 - rz ) + topParams * rz;
1181 // set coords on all faces and nodes
1182 const int nbSideFaces = 4;
1183 int sideFaceIDs[nbSideFaces] = { SMESH_Block::ID_Fx0z,
1184 SMESH_Block::ID_Fx1z,
1185 SMESH_Block::ID_F0yz,
1186 SMESH_Block::ID_F1yz };
1187 for ( int iF = 0; iF < nbSideFaces; ++iF )
1188 if ( !setFaceAndEdgesXYZ( sideFaceIDs[ iF ], params, z ))
1191 // compute coords for a new node
1193 if ( !SMESH_Block::ShellPoint( params, myShapeXYZ, coords ))
1194 return toSM( error("Can't compute coordinates by normalized parameters"));
1196 // if ( !meshDS->MeshElements( volumeID ) ||
1197 // meshDS->MeshElements( volumeID )->NbNodes() == 0 )
1198 // pointsToPython(myShapeXYZ);
1199 SHOWYXZ("TOPFacePoint ",myShapeXYZ[ ID_TOP_FACE]);
1200 SHOWYXZ("BOT Node "<< tBotNode.myNode->GetID(),gpXYZ(tBotNode.myNode));
1201 SHOWYXZ("ShellPoint ",coords);
1204 node = meshDS->AddNode( coords.X(), coords.Y(), coords.Z() );
1205 meshDS->SetNodeInVolume( node, volumeID );
1207 } // loop on bottom nodes
1212 SMESHDS_SubMesh* smDS = myBlock.SubMeshDS( ID_BOT_FACE );
1213 if ( !smDS ) return toSM( error(COMPERR_BAD_INPUT_MESH, "Null submesh"));
1215 // loop on bottom mesh faces
1216 SMDS_ElemIteratorPtr faceIt = smDS->GetElements();
1217 while ( faceIt->more() )
1219 const SMDS_MeshElement* face = faceIt->next();
1220 if ( !face || face->GetType() != SMDSAbs_Face )
1223 // find node columns for each node
1224 int nbNodes = face->NbCornerNodes();
1225 vector< const TNodeColumn* > columns( nbNodes );
1226 for ( int i = 0; i < nbNodes; ++i )
1228 const SMDS_MeshNode* n = face->GetNode( i );
1229 if ( n->GetPosition()->GetTypeOfPosition() == SMDS_TOP_FACE ) {
1230 TNode2ColumnMap::iterator bot_column = myBotToColumnMap.find( n );
1231 if ( bot_column == myBotToColumnMap.end() )
1232 return toSM( error(TCom("No nodes found above node ") << n->GetID() ));
1233 columns[ i ] = & bot_column->second;
1236 columns[ i ] = myBlock.GetNodeColumn( n );
1237 if ( !columns[ i ] )
1238 return toSM( error(TCom("No side nodes found above node ") << n->GetID() ));
1242 AddPrisms( columns, myHelper );
1244 } // loop on bottom mesh faces
1247 myBotToColumnMap.clear();
1253 //=======================================================================
1254 //function : computeWalls
1255 //purpose : Compute 2D mesh on walls FACEs of a prism
1256 //=======================================================================
1258 bool StdMeshers_Prism_3D::computeWalls(const Prism_3D::TPrismTopo& thePrism)
1260 SMESH_Mesh* mesh = myHelper->GetMesh();
1261 SMESHDS_Mesh* meshDS = myHelper->GetMeshDS();
1262 DBGOUT( endl << "COMPUTE Prism " << meshDS->ShapeToIndex( thePrism.myShape3D ));
1264 TProjction1dAlgo* projector1D = TProjction1dAlgo::instance( this );
1265 StdMeshers_Quadrangle_2D* quadAlgo = TQuadrangleAlgo::instance( this, myHelper );
1267 // SMESH_HypoFilter hyp1dFilter( SMESH_HypoFilter::IsAlgo(),/*not=*/true);
1268 // hyp1dFilter.And( SMESH_HypoFilter::HasDim( 1 ));
1269 // hyp1dFilter.And( SMESH_HypoFilter::IsMoreLocalThan( thePrism.myShape3D, *mesh ));
1271 // Discretize equally 'vertical' EDGEs
1272 // -----------------------------------
1273 // find source FACE sides for projection: either already computed ones or
1274 // the 'most composite' ones
1275 const size_t nbWalls = thePrism.myWallQuads.size();
1276 vector< int > wgt( nbWalls, 0 ); // "weight" of a wall
1277 for ( size_t iW = 0; iW != nbWalls; ++iW )
1279 Prism_3D::TQuadList::const_iterator quad = thePrism.myWallQuads[iW].begin();
1280 for ( ; quad != thePrism.myWallQuads[iW].end(); ++quad )
1282 StdMeshers_FaceSidePtr lftSide = (*quad)->side[ QUAD_LEFT_SIDE ];
1283 for ( int i = 0; i < lftSide->NbEdges(); ++i )
1286 const TopoDS_Edge& E = lftSide->Edge(i);
1287 if ( mesh->GetSubMesh( E )->IsMeshComputed() )
1290 wgt[ myHelper->WrapIndex( iW+1, nbWalls)] += 10;
1291 wgt[ myHelper->WrapIndex( iW-1, nbWalls)] += 10;
1293 // else if ( mesh->GetHypothesis( E, hyp1dFilter, true )) // local hypothesis!
1297 // in quadratic mesh, pass ignoreMediumNodes to quad sides
1298 if ( myHelper->GetIsQuadratic() )
1300 quad = thePrism.myWallQuads[iW].begin();
1301 for ( ; quad != thePrism.myWallQuads[iW].end(); ++quad )
1302 for ( int i = 0; i < NB_QUAD_SIDES; ++i )
1303 (*quad)->side[ i ].grid->SetIgnoreMediumNodes( true );
1306 multimap< int, int > wgt2quad;
1307 for ( size_t iW = 0; iW != nbWalls; ++iW )
1308 wgt2quad.insert( make_pair( wgt[ iW ], iW ));
1310 // Project 'vertical' EDGEs, from left to right
1311 multimap< int, int >::reverse_iterator w2q = wgt2quad.rbegin();
1312 for ( ; w2q != wgt2quad.rend(); ++w2q )
1314 const int iW = w2q->second;
1315 const Prism_3D::TQuadList& quads = thePrism.myWallQuads[ iW ];
1316 Prism_3D::TQuadList::const_iterator quad = quads.begin();
1317 for ( ; quad != quads.end(); ++quad )
1319 StdMeshers_FaceSidePtr rgtSide = (*quad)->side[ QUAD_RIGHT_SIDE ]; // tgt
1320 StdMeshers_FaceSidePtr lftSide = (*quad)->side[ QUAD_LEFT_SIDE ]; // src
1321 bool swapLeftRight = ( lftSide->NbSegments( /*update=*/true ) == 0 &&
1322 rgtSide->NbSegments( /*update=*/true ) > 0 );
1323 if ( swapLeftRight )
1324 std::swap( lftSide, rgtSide );
1326 // assure that all the source (left) EDGEs are meshed
1327 int nbSrcSegments = 0;
1328 for ( int i = 0; i < lftSide->NbEdges(); ++i )
1330 const TopoDS_Edge& srcE = lftSide->Edge(i);
1331 SMESH_subMesh* srcSM = mesh->GetSubMesh( srcE );
1332 if ( !srcSM->IsMeshComputed() ) {
1333 DBGOUT( "COMPUTE V edge " << srcSM->GetId() );
1334 TopoDS_Edge prpgSrcE = findPropagationSource( srcE );
1335 if ( !prpgSrcE.IsNull() ) {
1336 srcSM->ComputeSubMeshStateEngine( SMESH_subMesh::COMPUTE );
1337 projector1D->myHyp.SetSourceEdge( prpgSrcE );
1338 projector1D->Compute( *mesh, srcE );
1339 srcSM->ComputeStateEngine( SMESH_subMesh::CHECK_COMPUTE_STATE );
1342 srcSM->ComputeSubMeshStateEngine( SMESH_subMesh::COMPUTE );
1343 srcSM->ComputeStateEngine ( SMESH_subMesh::COMPUTE );
1345 if ( !srcSM->IsMeshComputed() )
1346 return toSM( error( "Can't compute 1D mesh" ));
1348 nbSrcSegments += srcSM->GetSubMeshDS()->NbElements();
1350 // check target EDGEs
1351 int nbTgtMeshed = 0, nbTgtSegments = 0;
1352 vector< bool > isTgtEdgeComputed( rgtSide->NbEdges() );
1353 for ( int i = 0; i < rgtSide->NbEdges(); ++i )
1355 const TopoDS_Edge& tgtE = rgtSide->Edge(i);
1356 SMESH_subMesh* tgtSM = mesh->GetSubMesh( tgtE );
1357 if ( !( isTgtEdgeComputed[ i ] = tgtSM->IsMeshComputed() )) {
1358 tgtSM->ComputeSubMeshStateEngine( SMESH_subMesh::COMPUTE );
1359 tgtSM->ComputeStateEngine ( SMESH_subMesh::COMPUTE );
1361 if ( tgtSM->IsMeshComputed() ) {
1363 nbTgtSegments += tgtSM->GetSubMeshDS()->NbElements();
1366 if ( rgtSide->NbEdges() == nbTgtMeshed ) // all tgt EDGEs meshed
1368 if ( nbTgtSegments != nbSrcSegments )
1370 bool badMeshRemoved = false;
1371 // remove just computed segments
1372 for ( int i = 0; i < rgtSide->NbEdges(); ++i )
1373 if ( !isTgtEdgeComputed[ i ])
1375 const TopoDS_Edge& tgtE = rgtSide->Edge(i);
1376 SMESH_subMesh* tgtSM = mesh->GetSubMesh( tgtE );
1377 tgtSM->ComputeStateEngine( SMESH_subMesh::CLEAN );
1378 badMeshRemoved = true;
1381 if ( !badMeshRemoved )
1383 for ( int i = 0; i < lftSide->NbEdges(); ++i )
1384 addBadInputElements( meshDS->MeshElements( lftSide->Edge( i )));
1385 for ( int i = 0; i < rgtSide->NbEdges(); ++i )
1386 addBadInputElements( meshDS->MeshElements( rgtSide->Edge( i )));
1387 return toSM( error( TCom("Different nb of segment on logically vertical edges #")
1388 << shapeID( lftSide->Edge(0) ) << " and #"
1389 << shapeID( rgtSide->Edge(0) ) << ": "
1390 << nbSrcSegments << " != " << nbTgtSegments ));
1393 else // if ( nbTgtSegments == nbSrcSegments )
1398 // Compute 'vertical projection'
1399 if ( nbTgtMeshed == 0 )
1401 // compute nodes on target VERTEXes
1402 const UVPtStructVec& srcNodeStr = lftSide->GetUVPtStruct();
1403 if ( srcNodeStr.size() == 0 )
1404 return toSM( error( TCom("Invalid node positions on edge #") <<
1405 shapeID( lftSide->Edge(0) )));
1406 vector< SMDS_MeshNode* > newNodes( srcNodeStr.size() );
1407 for ( int is2ndV = 0; is2ndV < 2; ++is2ndV )
1409 const TopoDS_Edge& E = rgtSide->Edge( is2ndV ? rgtSide->NbEdges()-1 : 0 );
1410 TopoDS_Vertex v = myHelper->IthVertex( is2ndV, E );
1411 mesh->GetSubMesh( v )->ComputeStateEngine( SMESH_subMesh::COMPUTE );
1412 const SMDS_MeshNode* n = SMESH_Algo::VertexNode( v, meshDS );
1413 newNodes[ is2ndV ? 0 : newNodes.size()-1 ] = (SMDS_MeshNode*) n;
1416 // compute nodes on target EDGEs
1417 DBGOUT( "COMPUTE V edge (proj) " << shapeID( lftSide->Edge(0)));
1418 rgtSide->Reverse(); // direct it same as the lftSide
1419 myHelper->SetElementsOnShape( false ); // myHelper holds the prism shape
1420 TopoDS_Edge tgtEdge;
1421 for ( size_t iN = 1; iN < srcNodeStr.size()-1; ++iN ) // add nodes
1423 gp_Pnt p = rgtSide->Value3d ( srcNodeStr[ iN ].normParam );
1424 double u = rgtSide->Parameter( srcNodeStr[ iN ].normParam, tgtEdge );
1425 newNodes[ iN ] = meshDS->AddNode( p.X(), p.Y(), p.Z() );
1426 meshDS->SetNodeOnEdge( newNodes[ iN ], tgtEdge, u );
1428 for ( size_t iN = 1; iN < srcNodeStr.size(); ++iN ) // add segments
1430 // find an EDGE to set a new segment
1431 std::pair<int, TopAbs_ShapeEnum> id2type =
1432 myHelper->GetMediumPos( newNodes[ iN-1 ], newNodes[ iN ] );
1433 if ( id2type.second != TopAbs_EDGE )
1435 // new nodes are on different EDGEs; put one of them on VERTEX
1436 const int edgeIndex = rgtSide->EdgeIndex( srcNodeStr[ iN-1 ].normParam );
1437 const double vertexParam = rgtSide->LastParameter( edgeIndex );
1438 TopoDS_Vertex vertex = rgtSide->LastVertex( edgeIndex );
1439 const SMDS_MeshNode* vn = SMESH_Algo::VertexNode( vertex, meshDS );
1440 const gp_Pnt p = BRep_Tool::Pnt( vertex );
1441 const int isPrev = ( Abs( srcNodeStr[ iN-1 ].normParam - vertexParam ) <
1442 Abs( srcNodeStr[ iN ].normParam - vertexParam ));
1443 meshDS->UnSetNodeOnShape( newNodes[ iN-isPrev ] );
1444 meshDS->SetNodeOnVertex ( newNodes[ iN-isPrev ], vertex );
1445 meshDS->MoveNode ( newNodes[ iN-isPrev ], p.X(), p.Y(), p.Z() );
1446 id2type.first = newNodes[ iN-(1-isPrev) ]->getshapeId();
1449 SMESH_MeshEditor::TListOfListOfNodes lln( 1, list< const SMDS_MeshNode* >() );
1450 lln.back().push_back ( vn );
1451 lln.back().push_front( newNodes[ iN-isPrev ] ); // to keep
1452 SMESH_MeshEditor( mesh ).MergeNodes( lln );
1455 SMDS_MeshElement* newEdge = myHelper->AddEdge( newNodes[ iN-1 ], newNodes[ iN ] );
1456 meshDS->SetMeshElementOnShape( newEdge, id2type.first );
1458 myHelper->SetElementsOnShape( true );
1459 for ( int i = 0; i < rgtSide->NbEdges(); ++i ) // update state of sub-meshes
1461 const TopoDS_Edge& E = rgtSide->Edge( i );
1462 SMESH_subMesh* tgtSM = mesh->GetSubMesh( E );
1463 tgtSM->ComputeStateEngine( SMESH_subMesh::CHECK_COMPUTE_STATE );
1466 // to continue projection from the just computed side as a source
1467 if ( !swapLeftRight && rgtSide->NbEdges() > 1 && w2q->second == iW )
1469 std::pair<int,int> wgt2quadKeyVal( w2q->first + 1, thePrism.myRightQuadIndex[ iW ]);
1470 wgt2quad.insert( wgt2quadKeyVal ); // it will be skipped by ++w2q
1471 wgt2quad.insert( wgt2quadKeyVal );
1472 w2q = wgt2quad.rbegin();
1477 // HOPE assigned hypotheses are OK, so that equal nb of segments will be generated
1478 //return toSM( error("Partial projection not implemented"));
1480 } // loop on quads of a composite wall side
1481 } // loop on the ordered wall sides
1485 for ( size_t iW = 0; iW != thePrism.myWallQuads.size(); ++iW )
1487 Prism_3D::TQuadList::const_iterator quad = thePrism.myWallQuads[iW].begin();
1488 for ( ; quad != thePrism.myWallQuads[iW].end(); ++quad )
1490 const TopoDS_Face& face = (*quad)->face;
1491 SMESH_subMesh* fSM = mesh->GetSubMesh( face );
1492 if ( ! fSM->IsMeshComputed() )
1494 // Top EDGEs must be projections from the bottom ones
1495 // to compute stuctured quad mesh on wall FACEs
1496 // ---------------------------------------------------
1497 const TopoDS_Edge& botE = (*quad)->side[ QUAD_BOTTOM_SIDE ].grid->Edge(0);
1498 const TopoDS_Edge& topE = (*quad)->side[ QUAD_TOP_SIDE ].grid->Edge(0);
1499 SMESH_subMesh* botSM = mesh->GetSubMesh( botE );
1500 SMESH_subMesh* topSM = mesh->GetSubMesh( topE );
1501 SMESH_subMesh* srcSM = botSM;
1502 SMESH_subMesh* tgtSM = topSM;
1503 if ( !srcSM->IsMeshComputed() && tgtSM->IsMeshComputed() )
1504 std::swap( srcSM, tgtSM );
1506 if ( !srcSM->IsMeshComputed() )
1508 DBGOUT( "COMPUTE H edge " << srcSM->GetId());
1509 srcSM->ComputeSubMeshStateEngine( SMESH_subMesh::COMPUTE ); // nodes on VERTEXes
1510 srcSM->ComputeStateEngine( SMESH_subMesh::COMPUTE ); // segments on the EDGE
1512 srcSM->ComputeStateEngine( SMESH_subMesh::CHECK_COMPUTE_STATE );
1514 if ( tgtSM->IsMeshComputed() &&
1515 tgtSM->GetSubMeshDS()->NbNodes() != srcSM->GetSubMeshDS()->NbNodes() )
1517 // the top EDGE is computed differently than the bottom one,
1518 // try to clear a wrong mesh
1519 bool isAdjFaceMeshed = false;
1520 PShapeIteratorPtr fIt = myHelper->GetAncestors( tgtSM->GetSubShape(),
1521 *mesh, TopAbs_FACE );
1522 while ( const TopoDS_Shape* f = fIt->next() )
1523 if (( isAdjFaceMeshed = mesh->GetSubMesh( *f )->IsMeshComputed() ))
1525 if ( isAdjFaceMeshed )
1526 return toSM( error( TCom("Different nb of segment on logically horizontal edges #")
1527 << shapeID( botE ) << " and #"
1528 << shapeID( topE ) << ": "
1529 << tgtSM->GetSubMeshDS()->NbElements() << " != "
1530 << srcSM->GetSubMeshDS()->NbElements() ));
1531 tgtSM->ComputeStateEngine( SMESH_subMesh::CLEAN );
1533 if ( !tgtSM->IsMeshComputed() )
1535 // compute nodes on VERTEXes
1536 SMESH_subMeshIteratorPtr smIt = tgtSM->getDependsOnIterator(/*includeSelf=*/false);
1537 while ( smIt->more() )
1538 smIt->next()->ComputeStateEngine( SMESH_subMesh::COMPUTE );
1540 DBGOUT( "COMPUTE H edge (proj) " << tgtSM->GetId());
1541 projector1D->myHyp.SetSourceEdge( TopoDS::Edge( srcSM->GetSubShape() ));
1542 projector1D->InitComputeError();
1543 bool ok = projector1D->Compute( *mesh, tgtSM->GetSubShape() );
1546 SMESH_ComputeErrorPtr err = projector1D->GetComputeError();
1547 if ( err->IsOK() ) err->myName = COMPERR_ALGO_FAILED;
1548 tgtSM->GetComputeError() = err;
1552 tgtSM->ComputeStateEngine( SMESH_subMesh::CHECK_COMPUTE_STATE );
1555 // Compute quad mesh on wall FACEs
1556 // -------------------------------
1558 // make all EDGES meshed
1559 fSM->ComputeSubMeshStateEngine( SMESH_subMesh::COMPUTE );
1560 if ( !fSM->SubMeshesComputed() )
1561 return toSM( error( COMPERR_BAD_INPUT_MESH,
1562 "Not all edges have valid algorithm and hypothesis"));
1564 quadAlgo->InitComputeError();
1565 DBGOUT( "COMPUTE Quad face " << fSM->GetId());
1566 bool ok = quadAlgo->Compute( *mesh, face );
1567 fSM->GetComputeError() = quadAlgo->GetComputeError();
1570 fSM->ComputeStateEngine( SMESH_subMesh::CHECK_COMPUTE_STATE );
1572 if ( myHelper->GetIsQuadratic() )
1574 // fill myHelper with medium nodes built by quadAlgo
1575 SMDS_ElemIteratorPtr fIt = fSM->GetSubMeshDS()->GetElements();
1576 while ( fIt->more() )
1577 myHelper->AddTLinks( dynamic_cast<const SMDS_MeshFace*>( fIt->next() ));
1585 //=======================================================================
1587 * \brief Returns a source EDGE of propagation to a given EDGE
1589 //=======================================================================
1591 TopoDS_Edge StdMeshers_Prism_3D::findPropagationSource( const TopoDS_Edge& E )
1593 if ( myPropagChains )
1594 for ( size_t i = 0; !myPropagChains[i].IsEmpty(); ++i )
1595 if ( myPropagChains[i].Contains( E ))
1596 return TopoDS::Edge( myPropagChains[i].FindKey( 1 ));
1598 return TopoDS_Edge();
1601 //=======================================================================
1602 //function : Evaluate
1604 //=======================================================================
1606 bool StdMeshers_Prism_3D::Evaluate(SMESH_Mesh& theMesh,
1607 const TopoDS_Shape& theShape,
1608 MapShapeNbElems& aResMap)
1610 if ( theShape.ShapeType() == TopAbs_COMPOUND )
1613 for ( TopoDS_Iterator it( theShape ); it.More(); it.Next() )
1614 ok &= Evaluate( theMesh, it.Value(), aResMap );
1617 SMESH_MesherHelper helper( theMesh );
1619 myHelper->SetSubShape( theShape );
1621 // find face contains only triangles
1622 vector < SMESH_subMesh * >meshFaces;
1623 TopTools_SequenceOfShape aFaces;
1624 int NumBase = 0, i = 0, NbQFs = 0;
1625 for (TopExp_Explorer exp(theShape, TopAbs_FACE); exp.More(); exp.Next()) {
1627 aFaces.Append(exp.Current());
1628 SMESH_subMesh *aSubMesh = theMesh.GetSubMesh(exp.Current());
1629 meshFaces.push_back(aSubMesh);
1630 MapShapeNbElemsItr anIt = aResMap.find(meshFaces[i-1]);
1631 if( anIt==aResMap.end() )
1632 return toSM( error( "Submesh can not be evaluated"));
1634 std::vector<int> aVec = (*anIt).second;
1635 int nbtri = Max(aVec[SMDSEntity_Triangle],aVec[SMDSEntity_Quad_Triangle]);
1636 int nbqua = Max(aVec[SMDSEntity_Quadrangle],aVec[SMDSEntity_Quad_Quadrangle]);
1637 if( nbtri==0 && nbqua>0 ) {
1646 std::vector<int> aResVec(SMDSEntity_Last);
1647 for(int i=SMDSEntity_Node; i<SMDSEntity_Last; i++) aResVec[i] = 0;
1648 SMESH_subMesh * sm = theMesh.GetSubMesh(theShape);
1649 aResMap.insert(std::make_pair(sm,aResVec));
1650 return toSM( error( "Submesh can not be evaluated" ));
1653 if(NumBase==0) NumBase = 1; // only quads => set 1 faces as base
1655 // find number of 1d elems for base face
1657 TopTools_MapOfShape Edges1;
1658 for (TopExp_Explorer exp(aFaces.Value(NumBase), TopAbs_EDGE); exp.More(); exp.Next()) {
1659 Edges1.Add(exp.Current());
1660 SMESH_subMesh *sm = theMesh.GetSubMesh(exp.Current());
1662 MapShapeNbElemsItr anIt = aResMap.find(sm);
1663 if( anIt == aResMap.end() ) continue;
1664 std::vector<int> aVec = (*anIt).second;
1665 nb1d += Max(aVec[SMDSEntity_Edge],aVec[SMDSEntity_Quad_Edge]);
1668 // find face opposite to base face
1670 for(i=1; i<=6; i++) {
1671 if(i==NumBase) continue;
1672 bool IsOpposite = true;
1673 for(TopExp_Explorer exp(aFaces.Value(i), TopAbs_EDGE); exp.More(); exp.Next()) {
1674 if( Edges1.Contains(exp.Current()) ) {
1684 // find number of 2d elems on side faces
1686 for(i=1; i<=6; i++) {
1687 if( i==OppNum || i==NumBase ) continue;
1688 MapShapeNbElemsItr anIt = aResMap.find( meshFaces[i-1] );
1689 if( anIt == aResMap.end() ) continue;
1690 std::vector<int> aVec = (*anIt).second;
1691 nb2d += Max(aVec[SMDSEntity_Quadrangle],aVec[SMDSEntity_Quad_Quadrangle]);
1694 MapShapeNbElemsItr anIt = aResMap.find( meshFaces[NumBase-1] );
1695 std::vector<int> aVec = (*anIt).second;
1696 bool IsQuadratic = (aVec[SMDSEntity_Quad_Triangle]>aVec[SMDSEntity_Triangle]) ||
1697 (aVec[SMDSEntity_Quad_Quadrangle]>aVec[SMDSEntity_Quadrangle]);
1698 int nb2d_face0_3 = Max(aVec[SMDSEntity_Triangle],aVec[SMDSEntity_Quad_Triangle]);
1699 int nb2d_face0_4 = Max(aVec[SMDSEntity_Quadrangle],aVec[SMDSEntity_Quad_Quadrangle]);
1700 int nb0d_face0 = aVec[SMDSEntity_Node];
1701 int nb1d_face0_int = ( nb2d_face0_3*3 + nb2d_face0_4*4 - nb1d ) / 2;
1703 std::vector<int> aResVec(SMDSEntity_Last);
1704 for(int i=SMDSEntity_Node; i<SMDSEntity_Last; i++) aResVec[i] = 0;
1706 aResVec[SMDSEntity_Quad_Penta] = nb2d_face0_3 * ( nb2d/nb1d );
1707 aResVec[SMDSEntity_Quad_Hexa] = nb2d_face0_4 * ( nb2d/nb1d );
1708 aResVec[SMDSEntity_Node] = nb0d_face0 * ( 2*nb2d/nb1d - 1 ) - nb1d_face0_int * nb2d/nb1d;
1711 aResVec[SMDSEntity_Node] = nb0d_face0 * ( nb2d/nb1d - 1 );
1712 aResVec[SMDSEntity_Penta] = nb2d_face0_3 * ( nb2d/nb1d );
1713 aResVec[SMDSEntity_Hexa] = nb2d_face0_4 * ( nb2d/nb1d );
1715 SMESH_subMesh * sm = theMesh.GetSubMesh(theShape);
1716 aResMap.insert(std::make_pair(sm,aResVec));
1721 //================================================================================
1723 * \brief Create prisms
1724 * \param columns - columns of nodes generated from nodes of a mesh face
1725 * \param helper - helper initialized by mesh and shape to add prisms to
1727 //================================================================================
1729 void StdMeshers_Prism_3D::AddPrisms( vector<const TNodeColumn*> & columns,
1730 SMESH_MesherHelper* helper)
1732 int nbNodes = columns.size();
1733 int nbZ = columns[0]->size();
1734 if ( nbZ < 2 ) return;
1736 // find out orientation
1737 bool isForward = true;
1738 SMDS_VolumeTool vTool;
1740 switch ( nbNodes ) {
1742 SMDS_VolumeOfNodes tmpPenta ( (*columns[0])[z-1], // bottom
1745 (*columns[0])[z], // top
1748 vTool.Set( &tmpPenta );
1749 isForward = vTool.IsForward();
1753 SMDS_VolumeOfNodes tmpHex( (*columns[0])[z-1], (*columns[1])[z-1], // bottom
1754 (*columns[2])[z-1], (*columns[3])[z-1],
1755 (*columns[0])[z], (*columns[1])[z], // top
1756 (*columns[2])[z], (*columns[3])[z] );
1757 vTool.Set( &tmpHex );
1758 isForward = vTool.IsForward();
1762 const int di = (nbNodes+1) / 3;
1763 SMDS_VolumeOfNodes tmpVol ( (*columns[0] )[z-1],
1764 (*columns[di] )[z-1],
1765 (*columns[2*di])[z-1],
1768 (*columns[2*di])[z] );
1769 vTool.Set( &tmpVol );
1770 isForward = vTool.IsForward();
1773 // vertical loop on columns
1775 helper->SetElementsOnShape( true );
1777 switch ( nbNodes ) {
1779 case 3: { // ---------- pentahedra
1780 const int i1 = isForward ? 1 : 2;
1781 const int i2 = isForward ? 2 : 1;
1782 for ( z = 1; z < nbZ; ++z )
1783 helper->AddVolume( (*columns[0 ])[z-1], // bottom
1784 (*columns[i1])[z-1],
1785 (*columns[i2])[z-1],
1786 (*columns[0 ])[z], // top
1788 (*columns[i2])[z] );
1791 case 4: { // ---------- hexahedra
1792 const int i1 = isForward ? 1 : 3;
1793 const int i3 = isForward ? 3 : 1;
1794 for ( z = 1; z < nbZ; ++z )
1795 helper->AddVolume( (*columns[0])[z-1], (*columns[i1])[z-1], // bottom
1796 (*columns[2])[z-1], (*columns[i3])[z-1],
1797 (*columns[0])[z], (*columns[i1])[z], // top
1798 (*columns[2])[z], (*columns[i3])[z] );
1801 case 6: { // ---------- octahedra
1802 const int iBase1 = isForward ? -1 : 0;
1803 const int iBase2 = isForward ? 0 :-1;
1804 for ( z = 1; z < nbZ; ++z )
1805 helper->AddVolume( (*columns[0])[z+iBase1], (*columns[1])[z+iBase1], // bottom or top
1806 (*columns[2])[z+iBase1], (*columns[3])[z+iBase1],
1807 (*columns[4])[z+iBase1], (*columns[5])[z+iBase1],
1808 (*columns[0])[z+iBase2], (*columns[1])[z+iBase2], // top or bottom
1809 (*columns[2])[z+iBase2], (*columns[3])[z+iBase2],
1810 (*columns[4])[z+iBase2], (*columns[5])[z+iBase2] );
1813 default: // ---------- polyhedra
1814 vector<int> quantities( 2 + nbNodes, 4 );
1815 quantities[0] = quantities[1] = nbNodes;
1816 columns.resize( nbNodes + 1 );
1817 columns[ nbNodes ] = columns[ 0 ];
1818 const int i1 = isForward ? 1 : 3;
1819 const int i3 = isForward ? 3 : 1;
1820 const int iBase1 = isForward ? -1 : 0;
1821 const int iBase2 = isForward ? 0 :-1;
1822 vector<const SMDS_MeshNode*> nodes( 2*nbNodes + 4*nbNodes);
1823 for ( z = 1; z < nbZ; ++z )
1825 for ( int i = 0; i < nbNodes; ++i ) {
1826 nodes[ i ] = (*columns[ i ])[z+iBase1]; // bottom or top
1827 nodes[ 2*nbNodes-i-1 ] = (*columns[ i ])[z+iBase2]; // top or bottom
1829 int di = 2*nbNodes + 4*i;
1830 nodes[ di+0 ] = (*columns[i ])[z ];
1831 nodes[ di+i1] = (*columns[i+1])[z ];
1832 nodes[ di+2 ] = (*columns[i+1])[z-1];
1833 nodes[ di+i3] = (*columns[i ])[z-1];
1835 helper->AddPolyhedralVolume( nodes, quantities );
1838 } // switch ( nbNodes )
1841 //================================================================================
1843 * \brief Find correspondence between bottom and top nodes
1844 * If elements on the bottom and top faces are topologically different,
1845 * and projection is possible and allowed, perform the projection
1846 * \retval bool - is a success or not
1848 //================================================================================
1850 bool StdMeshers_Prism_3D::assocOrProjBottom2Top( const gp_Trsf & bottomToTopTrsf,
1851 const Prism_3D::TPrismTopo& thePrism)
1853 SMESH_subMesh * botSM = myBlock.SubMesh( ID_BOT_FACE );
1854 SMESH_subMesh * topSM = myBlock.SubMesh( ID_TOP_FACE );
1856 SMESHDS_SubMesh * botSMDS = botSM->GetSubMeshDS();
1857 SMESHDS_SubMesh * topSMDS = topSM->GetSubMeshDS();
1859 if ( !botSMDS || botSMDS->NbElements() == 0 )
1861 _gen->Compute( *myHelper->GetMesh(), botSM->GetSubShape(), /*aShapeOnly=*/true );
1862 botSMDS = botSM->GetSubMeshDS();
1863 if ( !botSMDS || botSMDS->NbElements() == 0 )
1864 return toSM( error(TCom("No elements on face #") << botSM->GetId() ));
1867 bool needProject = !topSM->IsMeshComputed();
1868 if ( !needProject &&
1869 (botSMDS->NbElements() != topSMDS->NbElements() ||
1870 botSMDS->NbNodes() != topSMDS->NbNodes()))
1872 MESSAGE("nb elem bot " << botSMDS->NbElements() <<
1873 " top " << ( topSMDS ? topSMDS->NbElements() : 0 ));
1874 MESSAGE("nb node bot " << botSMDS->NbNodes() <<
1875 " top " << ( topSMDS ? topSMDS->NbNodes() : 0 ));
1876 return toSM( error(TCom("Mesh on faces #") << botSM->GetId()
1877 <<" and #"<< topSM->GetId() << " seems different" ));
1880 if ( 0/*needProject && !myProjectTriangles*/ )
1881 return toSM( error(TCom("Mesh on faces #") << botSM->GetId()
1882 <<" and #"<< topSM->GetId() << " seems different" ));
1883 ///RETURN_BAD_RESULT("Need to project but not allowed");
1887 return projectBottomToTop( bottomToTopTrsf );
1890 TopoDS_Face botFace = TopoDS::Face( myBlock.Shape( ID_BOT_FACE ));
1891 TopoDS_Face topFace = TopoDS::Face( myBlock.Shape( ID_TOP_FACE ));
1892 // associate top and bottom faces
1893 TAssocTool::TShapeShapeMap shape2ShapeMap;
1894 const bool sameTopo =
1895 TAssocTool::FindSubShapeAssociation( botFace, myBlock.Mesh(),
1896 topFace, myBlock.Mesh(),
1899 for ( size_t iQ = 0; iQ < thePrism.myWallQuads.size(); ++iQ )
1901 const Prism_3D::TQuadList& quadList = thePrism.myWallQuads[iQ];
1902 StdMeshers_FaceSidePtr botSide = quadList.front()->side[ QUAD_BOTTOM_SIDE ];
1903 StdMeshers_FaceSidePtr topSide = quadList.back ()->side[ QUAD_TOP_SIDE ];
1904 if ( botSide->NbEdges() == topSide->NbEdges() )
1906 for ( int iE = 0; iE < botSide->NbEdges(); ++iE )
1908 TAssocTool::InsertAssociation( botSide->Edge( iE ),
1909 topSide->Edge( iE ), shape2ShapeMap );
1910 TAssocTool::InsertAssociation( myHelper->IthVertex( 0, botSide->Edge( iE )),
1911 myHelper->IthVertex( 0, topSide->Edge( iE )),
1917 TopoDS_Vertex vb, vt;
1918 StdMeshers_FaceSidePtr sideB, sideT;
1919 vb = myHelper->IthVertex( 0, botSide->Edge( 0 ));
1920 vt = myHelper->IthVertex( 0, topSide->Edge( 0 ));
1921 sideB = quadList.front()->side[ QUAD_LEFT_SIDE ];
1922 sideT = quadList.back ()->side[ QUAD_LEFT_SIDE ];
1923 if ( vb.IsSame( sideB->FirstVertex() ) &&
1924 vt.IsSame( sideT->LastVertex() ))
1926 TAssocTool::InsertAssociation( botSide->Edge( 0 ),
1927 topSide->Edge( 0 ), shape2ShapeMap );
1928 TAssocTool::InsertAssociation( vb, vt, shape2ShapeMap );
1930 vb = myHelper->IthVertex( 1, botSide->Edge( botSide->NbEdges()-1 ));
1931 vt = myHelper->IthVertex( 1, topSide->Edge( topSide->NbEdges()-1 ));
1932 sideB = quadList.front()->side[ QUAD_RIGHT_SIDE ];
1933 sideT = quadList.back ()->side[ QUAD_RIGHT_SIDE ];
1934 if ( vb.IsSame( sideB->FirstVertex() ) &&
1935 vt.IsSame( sideT->LastVertex() ))
1937 TAssocTool::InsertAssociation( botSide->Edge( botSide->NbEdges()-1 ),
1938 topSide->Edge( topSide->NbEdges()-1 ),
1940 TAssocTool::InsertAssociation( vb, vt, shape2ShapeMap );
1945 // Find matching nodes of top and bottom faces
1946 TNodeNodeMap n2nMap;
1947 if ( ! TAssocTool::FindMatchingNodesOnFaces( botFace, myBlock.Mesh(),
1948 topFace, myBlock.Mesh(),
1949 shape2ShapeMap, n2nMap ))
1952 return toSM( error(TCom("Mesh on faces #") << botSM->GetId()
1953 <<" and #"<< topSM->GetId() << " seems different" ));
1955 return toSM( error(TCom("Topology of faces #") << botSM->GetId()
1956 <<" and #"<< topSM->GetId() << " seems different" ));
1959 // Fill myBotToColumnMap
1961 int zSize = myBlock.VerticalSize();
1963 TNodeNodeMap::iterator bN_tN = n2nMap.begin();
1964 for ( ; bN_tN != n2nMap.end(); ++bN_tN )
1966 const SMDS_MeshNode* botNode = bN_tN->first;
1967 const SMDS_MeshNode* topNode = bN_tN->second;
1968 if ( botNode->GetPosition()->GetTypeOfPosition() != SMDS_TOP_FACE )
1969 continue; // wall columns are contained in myBlock
1970 // create node column
1971 Prism_3D::TNode bN( botNode );
1972 TNode2ColumnMap::iterator bN_col =
1973 myBotToColumnMap.insert( make_pair ( bN, TNodeColumn() )).first;
1974 TNodeColumn & column = bN_col->second;
1975 column.resize( zSize );
1976 column.front() = botNode;
1977 column.back() = topNode;
1982 //================================================================================
1984 * \brief Remove quadrangles from the top face and
1985 * create triangles there by projection from the bottom
1986 * \retval bool - a success or not
1988 //================================================================================
1990 bool StdMeshers_Prism_3D::projectBottomToTop( const gp_Trsf & bottomToTopTrsf )
1992 SMESHDS_Mesh* meshDS = myBlock.MeshDS();
1993 SMESH_subMesh * botSM = myBlock.SubMesh( ID_BOT_FACE );
1994 SMESH_subMesh * topSM = myBlock.SubMesh( ID_TOP_FACE );
1996 SMESHDS_SubMesh * botSMDS = botSM->GetSubMeshDS();
1997 SMESHDS_SubMesh * topSMDS = topSM->GetSubMeshDS();
1999 if ( topSMDS && topSMDS->NbElements() > 0 )
2000 topSM->ComputeStateEngine( SMESH_subMesh::CLEAN );
2002 const TopoDS_Face& botFace = TopoDS::Face( myBlock.Shape( ID_BOT_FACE )); // oriented within
2003 const TopoDS_Face& topFace = TopoDS::Face( myBlock.Shape( ID_TOP_FACE )); // the 3D SHAPE
2004 int topFaceID = meshDS->ShapeToIndex( topFace );
2006 SMESH_MesherHelper botHelper( *myHelper->GetMesh() );
2007 botHelper.SetSubShape( botFace );
2008 botHelper.ToFixNodeParameters( true );
2010 SMESH_MesherHelper topHelper( *myHelper->GetMesh() );
2011 topHelper.SetSubShape( topFace );
2012 topHelper.ToFixNodeParameters( true );
2013 double distXYZ[4], fixTol = 10 * topHelper.MaxTolerance( topFace );
2015 // Fill myBotToColumnMap
2017 int zSize = myBlock.VerticalSize();
2018 Prism_3D::TNode prevTNode;
2019 SMDS_NodeIteratorPtr nIt = botSMDS->GetNodes();
2020 while ( nIt->more() )
2022 const SMDS_MeshNode* botNode = nIt->next();
2023 const SMDS_MeshNode* topNode = 0;
2024 if ( botNode->GetPosition()->GetTypeOfPosition() != SMDS_TOP_FACE )
2025 continue; // strange
2027 Prism_3D::TNode bN( botNode );
2028 if ( bottomToTopTrsf.Form() == gp_Identity )
2030 // compute bottom node params
2031 gp_XYZ paramHint(-1,-1,-1);
2032 if ( prevTNode.IsNeighbor( bN ))
2034 paramHint = prevTNode.GetParams();
2035 // double tol = 1e-2 * ( prevTNode.GetCoords() - bN.GetCoords() ).Modulus();
2036 // myBlock.SetTolerance( Min( myBlock.GetTolerance(), tol ));
2038 if ( !myBlock.ComputeParameters( bN.GetCoords(), bN.ChangeParams(),
2039 ID_BOT_FACE, paramHint ))
2040 return toSM( error(TCom("Can't compute normalized parameters for node ")
2041 << botNode->GetID() << " on the face #"<< botSM->GetId() ));
2043 // compute top node coords
2044 gp_XYZ topXYZ; gp_XY topUV;
2045 if ( !myBlock.FacePoint( ID_TOP_FACE, bN.GetParams(), topXYZ ) ||
2046 !myBlock.FaceUV ( ID_TOP_FACE, bN.GetParams(), topUV ))
2047 return toSM( error(TCom("Can't compute coordinates "
2048 "by normalized parameters on the face #")<< topSM->GetId() ));
2049 topNode = meshDS->AddNode( topXYZ.X(),topXYZ.Y(),topXYZ.Z() );
2050 meshDS->SetNodeOnFace( topNode, topFaceID, topUV.X(), topUV.Y() );
2052 else // use bottomToTopTrsf
2054 gp_XYZ coords = bN.GetCoords();
2055 bottomToTopTrsf.Transforms( coords );
2056 topNode = meshDS->AddNode( coords.X(), coords.Y(), coords.Z() );
2057 gp_XY topUV = botHelper.GetNodeUV( botFace, botNode, 0, &checkUV );
2058 meshDS->SetNodeOnFace( topNode, topFaceID, topUV.X(), topUV.Y() );
2060 if ( topHelper.CheckNodeUV( topFace, topNode, topUV, fixTol, /*force=*/false, distXYZ ) &&
2061 distXYZ[0] > fixTol && distXYZ[0] < fixTol * 1e+3 )
2062 meshDS->MoveNode( topNode, distXYZ[1], distXYZ[2], distXYZ[3] ); // transform can be inaccurate
2064 // create node column
2065 TNode2ColumnMap::iterator bN_col =
2066 myBotToColumnMap.insert( make_pair ( bN, TNodeColumn() )).first;
2067 TNodeColumn & column = bN_col->second;
2068 column.resize( zSize );
2069 column.front() = botNode;
2070 column.back() = topNode;
2075 const bool oldSetElemsOnShape = myHelper->SetElementsOnShape( false );
2077 // care of orientation;
2078 // if the bottom faces is orienetd OK then top faces must be reversed
2079 bool reverseTop = true;
2080 if ( myHelper->NbAncestors( botFace, *myBlock.Mesh(), TopAbs_SOLID ) > 1 )
2081 reverseTop = ! myHelper->IsReversedSubMesh( botFace );
2082 int iFrw, iRev, *iPtr = &( reverseTop ? iRev : iFrw );
2084 // loop on bottom mesh faces
2085 SMDS_ElemIteratorPtr faceIt = botSMDS->GetElements();
2086 vector< const SMDS_MeshNode* > nodes;
2087 while ( faceIt->more() )
2089 const SMDS_MeshElement* face = faceIt->next();
2090 if ( !face || face->GetType() != SMDSAbs_Face )
2093 // find top node in columns for each bottom node
2094 int nbNodes = face->NbCornerNodes();
2095 nodes.resize( nbNodes );
2096 for ( iFrw = 0, iRev = nbNodes-1; iFrw < nbNodes; ++iFrw, --iRev )
2098 const SMDS_MeshNode* n = face->GetNode( *iPtr );
2099 if ( n->GetPosition()->GetTypeOfPosition() == SMDS_TOP_FACE ) {
2100 TNode2ColumnMap::iterator bot_column = myBotToColumnMap.find( n );
2101 if ( bot_column == myBotToColumnMap.end() )
2102 return toSM( error(TCom("No nodes found above node ") << n->GetID() ));
2103 nodes[ iFrw ] = bot_column->second.back();
2106 const TNodeColumn* column = myBlock.GetNodeColumn( n );
2108 return toSM( error(TCom("No side nodes found above node ") << n->GetID() ));
2109 nodes[ iFrw ] = column->back();
2112 SMDS_MeshElement* newFace = 0;
2113 switch ( nbNodes ) {
2116 newFace = myHelper->AddFace(nodes[0], nodes[1], nodes[2]);
2120 newFace = myHelper->AddFace( nodes[0], nodes[1], nodes[2], nodes[3] );
2124 newFace = meshDS->AddPolygonalFace( nodes );
2127 meshDS->SetMeshElementOnShape( newFace, topFaceID );
2130 myHelper->SetElementsOnShape( oldSetElemsOnShape );
2135 //=======================================================================
2136 //function : project2dMesh
2137 //purpose : Project mesh faces from a source FACE of one prism (theSrcFace)
2138 // to a source FACE of another prism (theTgtFace)
2139 //=======================================================================
2141 bool StdMeshers_Prism_3D::project2dMesh(const TopoDS_Face& theSrcFace,
2142 const TopoDS_Face& theTgtFace)
2144 TProjction2dAlgo* projector2D = TProjction2dAlgo::instance( this );
2145 projector2D->myHyp.SetSourceFace( theSrcFace );
2146 bool ok = projector2D->Compute( *myHelper->GetMesh(), theTgtFace );
2148 SMESH_subMesh* tgtSM = myHelper->GetMesh()->GetSubMesh( theTgtFace );
2149 tgtSM->ComputeStateEngine ( SMESH_subMesh::CHECK_COMPUTE_STATE );
2150 tgtSM->ComputeSubMeshStateEngine( SMESH_subMesh::CHECK_COMPUTE_STATE );
2155 //================================================================================
2157 * \brief Set projection coordinates of a node to a face and it's sub-shapes
2158 * \param faceID - the face given by in-block ID
2159 * \param params - node normalized parameters
2160 * \retval bool - is a success
2162 //================================================================================
2164 bool StdMeshers_Prism_3D::setFaceAndEdgesXYZ( const int faceID, const gp_XYZ& params, int z )
2166 // find base and top edges of the face
2167 enum { BASE = 0, TOP, LEFT, RIGHT };
2168 vector< int > edgeVec; // 0-base, 1-top
2169 SMESH_Block::GetFaceEdgesIDs( faceID, edgeVec );
2171 myBlock.EdgePoint( edgeVec[ BASE ], params, myShapeXYZ[ edgeVec[ BASE ]]);
2172 myBlock.EdgePoint( edgeVec[ TOP ], params, myShapeXYZ[ edgeVec[ TOP ]]);
2174 SHOWYXZ("\nparams ", params);
2175 SHOWYXZ("TOP is " <<edgeVec[ TOP ], myShapeXYZ[ edgeVec[ TOP]]);
2176 SHOWYXZ("BASE is "<<edgeVec[ BASE], myShapeXYZ[ edgeVec[ BASE]]);
2178 if ( faceID == SMESH_Block::ID_Fx0z || faceID == SMESH_Block::ID_Fx1z )
2180 myBlock.EdgePoint( edgeVec[ LEFT ], params, myShapeXYZ[ edgeVec[ LEFT ]]);
2181 myBlock.EdgePoint( edgeVec[ RIGHT ], params, myShapeXYZ[ edgeVec[ RIGHT ]]);
2183 SHOWYXZ("VER "<<edgeVec[ LEFT], myShapeXYZ[ edgeVec[ LEFT]]);
2184 SHOWYXZ("VER "<<edgeVec[ RIGHT], myShapeXYZ[ edgeVec[ RIGHT]]);
2186 myBlock.FacePoint( faceID, params, myShapeXYZ[ faceID ]);
2187 SHOWYXZ("FacePoint "<<faceID, myShapeXYZ[ faceID]);
2192 //=======================================================================
2194 //purpose : If (!isOK), sets the error to a sub-mesh of a current SOLID
2195 //=======================================================================
2197 bool StdMeshers_Prism_3D::toSM( bool isOK )
2199 if ( mySetErrorToSM &&
2202 !myHelper->GetSubShape().IsNull() &&
2203 myHelper->GetSubShape().ShapeType() == TopAbs_SOLID)
2205 SMESH_subMesh* sm = myHelper->GetMesh()->GetSubMesh( myHelper->GetSubShape() );
2206 sm->GetComputeError() = this->GetComputeError();
2207 // clear error in order not to return it twice
2208 _error = COMPERR_OK;
2214 //=======================================================================
2215 //function : shapeID
2216 //purpose : Return index of a shape
2217 //=======================================================================
2219 int StdMeshers_Prism_3D::shapeID( const TopoDS_Shape& S )
2221 if ( S.IsNull() ) return 0;
2222 if ( !myHelper ) return -3;
2223 return myHelper->GetMeshDS()->ShapeToIndex( S );
2226 namespace // utils used by StdMeshers_Prism_3D::IsApplicable()
2228 struct EdgeWithNeighbors
2232 EdgeWithNeighbors(const TopoDS_Edge& E, int iE, int nbE, int shift = 0 ):
2234 _iL( SMESH_MesherHelper::WrapIndex( iE-1, nbE ) + shift ),
2235 _iR( SMESH_MesherHelper::WrapIndex( iE+1, nbE ) + shift )
2238 EdgeWithNeighbors() {}
2243 TopTools_IndexedMapOfShape *_faces; // pointer because its copy constructor is private
2244 TopoDS_Edge _topEdge;
2245 vector< EdgeWithNeighbors >*_edges;
2247 vector< bool > _isCheckedEdge;
2248 int _nbCheckedEdges; // nb of EDGEs whose location is defined
2249 PrismSide *_leftSide;
2250 PrismSide *_rightSide;
2251 const TopoDS_Edge& Edge( int i ) const
2253 return (*_edges)[ i ]._edge;
2255 int FindEdge( const TopoDS_Edge& E ) const
2257 for ( size_t i = 0; i < _edges->size(); ++i )
2258 if ( E.IsSame( Edge( i ))) return i;
2262 //--------------------------------------------------------------------------------
2264 * \brief Return ordered edges of a face
2266 bool getEdges( const TopoDS_Face& face,
2267 vector< EdgeWithNeighbors > & edges,
2268 const bool noHolesAllowed)
2270 list< TopoDS_Edge > ee;
2271 list< int > nbEdgesInWires;
2272 int nbW = SMESH_Block::GetOrderedEdges( face, ee, nbEdgesInWires );
2273 if ( nbW > 1 && noHolesAllowed )
2277 list< TopoDS_Edge >::iterator e = ee.begin();
2278 list< int >::iterator nbE = nbEdgesInWires.begin();
2279 for ( ; nbE != nbEdgesInWires.end(); ++nbE )
2280 for ( iE = 0; iE < *nbE; ++e, ++iE )
2281 if ( SMESH_Algo::isDegenerated( *e ))
2289 e->Orientation( TopAbs_FORWARD ); // for operator==() to work
2294 for ( nbE = nbEdgesInWires.begin(); nbE != nbEdgesInWires.end(); ++nbE )
2296 for ( iE = 0; iE < *nbE; ++e, ++iE )
2297 edges.push_back( EdgeWithNeighbors( *e, iE, *nbE, nbTot ));
2300 return edges.size();
2302 //--------------------------------------------------------------------------------
2304 * \brief Return another faces sharing an edge
2306 const TopoDS_Shape & getAnotherFace( const TopoDS_Face& face,
2307 const TopoDS_Edge& edge,
2308 TopTools_IndexedDataMapOfShapeListOfShape& facesOfEdge)
2310 TopTools_ListIteratorOfListOfShape faceIt( facesOfEdge.FindFromKey( edge ));
2311 for ( ; faceIt.More(); faceIt.Next() )
2312 if ( !face.IsSame( faceIt.Value() ))
2313 return faceIt.Value();
2318 //================================================================================
2320 * \brief Return true if the algorithm can mesh this shape
2321 * \param [in] aShape - shape to check
2322 * \param [in] toCheckAll - if true, this check returns OK if all shapes are OK,
2323 * else, returns OK if at least one shape is OK
2325 //================================================================================
2327 bool StdMeshers_Prism_3D::IsApplicable(const TopoDS_Shape & shape, bool toCheckAll)
2329 TopExp_Explorer sExp( shape, TopAbs_SOLID );
2333 for ( ; sExp.More(); sExp.Next() )
2337 TopExp_Explorer shExp( sExp.Current(), TopAbs_SHELL );
2338 if ( shExp.More() ) {
2339 shell = shExp.Current();
2344 if ( shell.IsNull() ) {
2345 if ( toCheckAll ) return false;
2349 TopTools_IndexedMapOfShape allFaces;
2350 TopExp::MapShapes( shell, TopAbs_FACE, allFaces );
2351 if ( allFaces.Extent() < 3 ) {
2352 if ( toCheckAll ) return false;
2356 if ( allFaces.Extent() == 6 )
2358 TopTools_IndexedMapOfOrientedShape map;
2359 bool isBox = SMESH_Block::FindBlockShapes( TopoDS::Shell( shell ),
2360 TopoDS_Vertex(), TopoDS_Vertex(), map );
2362 if ( !toCheckAll ) return true;
2367 TopTools_IndexedMapOfShape allShapes;
2368 TopExp::MapShapes( shape, allShapes );
2371 TopTools_IndexedDataMapOfShapeListOfShape facesOfEdge;
2372 TopTools_ListIteratorOfListOfShape faceIt;
2373 TopExp::MapShapesAndAncestors( sExp.Current(), TopAbs_EDGE, TopAbs_FACE , facesOfEdge );
2374 if ( facesOfEdge.IsEmpty() ) {
2375 if ( toCheckAll ) return false;
2379 typedef vector< EdgeWithNeighbors > TEdgeWithNeighborsVec;
2380 vector< TEdgeWithNeighborsVec > faceEdgesVec( allFaces.Extent() + 1 );
2381 TopTools_IndexedMapOfShape* facesOfSide = new TopTools_IndexedMapOfShape[ faceEdgesVec.size() ];
2382 SMESHUtils::ArrayDeleter<TopTools_IndexedMapOfShape> delFacesOfSide( facesOfSide );
2384 // try to use each face as a bottom one
2385 bool prismDetected = false;
2386 for ( int iF = 1; iF < allFaces.Extent() && !prismDetected; ++iF )
2388 const TopoDS_Face& botF = TopoDS::Face( allFaces( iF ));
2390 TEdgeWithNeighborsVec& botEdges = faceEdgesVec[ iF ];
2391 if ( botEdges.empty() )
2393 if ( !getEdges( botF, botEdges, /*noHoles=*/false ))
2395 if ( allFaces.Extent()-1 <= (int) botEdges.size() )
2396 continue; // all faces are adjacent to botF - no top FACE
2398 // init data of side FACEs
2399 vector< PrismSide > sides( botEdges.size() );
2400 for ( int iS = 0; iS < botEdges.size(); ++iS )
2402 sides[ iS ]._topEdge = botEdges[ iS ]._edge;
2403 sides[ iS ]._face = botF;
2404 sides[ iS ]._leftSide = & sides[ botEdges[ iS ]._iR ];
2405 sides[ iS ]._rightSide = & sides[ botEdges[ iS ]._iL ];
2406 sides[ iS ]._faces = & facesOfSide[ iS ];
2407 sides[ iS ]._faces->Clear();
2410 bool isOK = true; // ok for a current botF
2411 bool isAdvanced = true;
2412 int nbFoundSideFaces = 0;
2413 for ( int iLoop = 0; isOK && isAdvanced; ++iLoop )
2416 for ( size_t iS = 0; iS < sides.size() && isOK; ++iS )
2418 PrismSide& side = sides[ iS ];
2419 if ( side._face.IsNull() )
2421 if ( side._topEdge.IsNull() )
2423 // find vertical EDGEs --- EGDEs shared with neighbor side FACEs
2424 for ( int is2nd = 0; is2nd < 2 && isOK; ++is2nd ) // 2 adjacent neighbors
2426 int di = is2nd ? 1 : -1;
2427 const PrismSide* adjSide = is2nd ? side._rightSide : side._leftSide;
2428 for ( size_t i = 1; i < side._edges->size(); ++i )
2430 int iE = SMESH_MesherHelper::WrapIndex( i*di + side._iBotEdge, side._edges->size());
2431 if ( side._isCheckedEdge[ iE ] ) continue;
2432 const TopoDS_Edge& vertE = side.Edge( iE );
2433 const TopoDS_Shape& neighborF = getAnotherFace( side._face, vertE, facesOfEdge );
2434 bool isEdgeShared = adjSide->_faces->Contains( neighborF );
2438 side._isCheckedEdge[ iE ] = true;
2439 side._nbCheckedEdges++;
2440 int nbNotCheckedE = side._edges->size() - side._nbCheckedEdges;
2441 if ( nbNotCheckedE == 1 )
2446 if ( i == 1 && iLoop == 0 ) isOK = false;
2452 int nbNotCheckedE = side._edges->size() - side._nbCheckedEdges;
2453 if ( nbNotCheckedE == 1 )
2455 vector<bool>::iterator ii = std::find( side._isCheckedEdge.begin(),
2456 side._isCheckedEdge.end(), false );
2457 if ( ii != side._isCheckedEdge.end() )
2459 size_t iE = std::distance( side._isCheckedEdge.begin(), ii );
2460 side._topEdge = side.Edge( iE );
2463 isOK = ( nbNotCheckedE >= 1 );
2465 else //if ( !side._topEdge.IsNull() )
2467 // get a next face of a side
2468 const TopoDS_Shape& f = getAnotherFace( side._face, side._topEdge, facesOfEdge );
2469 side._faces->Add( f );
2471 if ( f.IsSame( side._face ) || // _topEdge is a seam
2472 SMESH_MesherHelper::Count( f, TopAbs_WIRE, false ) != 1 )
2476 else if ( side._leftSide != & side ) // not closed side face
2478 if ( side._leftSide->_faces->Contains( f ))
2481 side._leftSide->_face.Nullify();
2482 side._leftSide->_topEdge.Nullify();
2484 if ( side._rightSide->_faces->Contains( f ))
2487 side._rightSide->_face.Nullify();
2488 side._rightSide->_topEdge.Nullify();
2493 side._face.Nullify();
2494 side._topEdge.Nullify();
2497 side._face = TopoDS::Face( f );
2498 int faceID = allFaces.FindIndex( side._face );
2499 side._edges = & faceEdgesVec[ faceID ];
2500 if ( side._edges->empty() )
2501 if ( !getEdges( side._face, * side._edges, /*noHoles=*/true ))
2503 const int nbE = side._edges->size();
2508 side._iBotEdge = side.FindEdge( side._topEdge );
2509 side._isCheckedEdge.clear();
2510 side._isCheckedEdge.resize( nbE, false );
2511 side._isCheckedEdge[ side._iBotEdge ] = true;
2512 side._nbCheckedEdges = 1; // bottom EDGE is known
2514 side._topEdge.Nullify();
2515 isOK = ( !side._edges->empty() || side._faces->Extent() > 1 );
2517 } //if ( !side._topEdge.IsNull() )
2519 } // loop on prism sides
2521 if ( nbFoundSideFaces > allFaces.Extent() )
2525 if ( iLoop > allFaces.Extent() * 10 )
2529 cerr << "BUG: infinite loop in StdMeshers_Prism_3D::IsApplicable()" << endl;
2532 } // while isAdvanced
2534 if ( isOK && sides[0]._faces->Extent() > 1 )
2536 const int nbFaces = sides[0]._faces->Extent();
2537 if ( botEdges.size() == 1 ) // cylinder
2539 prismDetected = ( nbFaces == allFaces.Extent()-1 );
2543 const TopoDS_Shape& topFace = sides[0]._faces->FindKey( nbFaces );
2545 for ( iS = 1; iS < sides.size(); ++iS )
2546 if ( !sides[ iS ]._faces->Contains( topFace ))
2548 prismDetected = ( iS == sides.size() );
2551 } // loop on allFaces
2553 if ( !prismDetected && toCheckAll ) return false;
2554 if ( prismDetected && !toCheckAll ) return true;
2563 //================================================================================
2565 * \brief Return true if this node and other one belong to one face
2567 //================================================================================
2569 bool Prism_3D::TNode::IsNeighbor( const Prism_3D::TNode& other ) const
2571 if ( !other.myNode || !myNode ) return false;
2573 SMDS_ElemIteratorPtr fIt = other.myNode->GetInverseElementIterator(SMDSAbs_Face);
2574 while ( fIt->more() )
2575 if ( fIt->next()->GetNodeIndex( myNode ) >= 0 )
2580 //================================================================================
2582 * \brief Prism initialization
2584 //================================================================================
2586 void TPrismTopo::Clear()
2588 myShape3D.Nullify();
2591 myWallQuads.clear();
2592 myBottomEdges.clear();
2593 myNbEdgesInWires.clear();
2594 myWallQuads.clear();
2597 //================================================================================
2599 * \brief Set upside-down
2601 //================================================================================
2603 void TPrismTopo::SetUpsideDown()
2605 std::swap( myBottom, myTop );
2606 myBottomEdges.clear();
2607 std::reverse( myBottomEdges.begin(), myBottomEdges.end() );
2608 for ( size_t i = 0; i < myWallQuads.size(); ++i )
2610 myWallQuads[i].reverse();
2611 TQuadList::iterator q = myWallQuads[i].begin();
2612 for ( ; q != myWallQuads[i].end(); ++q )
2614 (*q)->shift( 2, /*keepUnitOri=*/true );
2616 myBottomEdges.push_back( myWallQuads[i].front()->side[ QUAD_BOTTOM_SIDE ].grid->Edge(0) );
2620 } // namespace Prism_3D
2622 //================================================================================
2624 * \brief Constructor. Initialization is needed
2626 //================================================================================
2628 StdMeshers_PrismAsBlock::StdMeshers_PrismAsBlock()
2633 StdMeshers_PrismAsBlock::~StdMeshers_PrismAsBlock()
2637 void StdMeshers_PrismAsBlock::Clear()
2640 myShapeIDMap.Clear();
2644 delete mySide; mySide = 0;
2646 myParam2ColumnMaps.clear();
2647 myShapeIndex2ColumnMap.clear();
2650 //=======================================================================
2651 //function : initPrism
2652 //purpose : Analyse shape geometry and mesh.
2653 // If there are triangles on one of faces, it becomes 'bottom'.
2654 // thePrism.myBottom can be already set up.
2655 //=======================================================================
2657 bool StdMeshers_Prism_3D::initPrism(Prism_3D::TPrismTopo& thePrism,
2658 const TopoDS_Shape& shape3D)
2660 myHelper->SetSubShape( shape3D );
2662 SMESH_subMesh* mainSubMesh = myHelper->GetMesh()->GetSubMeshContaining( shape3D );
2663 if ( !mainSubMesh ) return toSM( error(COMPERR_BAD_INPUT_MESH,"Null submesh of shape3D"));
2665 // detect not-quad FACE sub-meshes of the 3D SHAPE
2666 list< SMESH_subMesh* > notQuadGeomSubMesh;
2667 list< SMESH_subMesh* > notQuadElemSubMesh;
2670 SMESH_subMesh* anyFaceSM = 0;
2671 SMESH_subMeshIteratorPtr smIt = mainSubMesh->getDependsOnIterator(false,true);
2672 while ( smIt->more() )
2674 SMESH_subMesh* sm = smIt->next();
2675 const TopoDS_Shape& face = sm->GetSubShape();
2676 if ( face.ShapeType() > TopAbs_FACE ) break;
2677 else if ( face.ShapeType() < TopAbs_FACE ) continue;
2681 // is quadrangle FACE?
2682 list< TopoDS_Edge > orderedEdges;
2683 list< int > nbEdgesInWires;
2684 int nbWires = SMESH_Block::GetOrderedEdges( TopoDS::Face( face ), orderedEdges,
2686 if ( nbWires != 1 || nbEdgesInWires.front() != 4 )
2687 notQuadGeomSubMesh.push_back( sm );
2689 // look for not quadrangle mesh elements
2690 if ( SMESHDS_SubMesh* smDS = sm->GetSubMeshDS() )
2691 if ( !myHelper->IsSameElemGeometry( smDS, SMDSGeom_QUADRANGLE ))
2692 notQuadElemSubMesh.push_back( sm );
2695 int nbNotQuadMeshed = notQuadElemSubMesh.size();
2696 int nbNotQuad = notQuadGeomSubMesh.size();
2697 bool hasNotQuad = ( nbNotQuad || nbNotQuadMeshed );
2700 if ( nbNotQuadMeshed > 2 )
2702 return toSM( error(COMPERR_BAD_INPUT_MESH,
2703 TCom("More than 2 faces with not quadrangle elements: ")
2704 <<nbNotQuadMeshed));
2706 if ( nbNotQuad > 2 || !thePrism.myBottom.IsNull() )
2708 // Issue 0020843 - one of side FACEs is quasi-quadrilateral (not 4 EDGEs).
2709 // Remove from notQuadGeomSubMesh faces meshed with regular grid
2710 int nbQuasiQuads = removeQuasiQuads( notQuadGeomSubMesh, myHelper,
2711 TQuadrangleAlgo::instance(this,myHelper) );
2712 nbNotQuad -= nbQuasiQuads;
2713 if ( nbNotQuad > 2 )
2714 return toSM( error(COMPERR_BAD_SHAPE,
2715 TCom("More than 2 not quadrilateral faces: ") <<nbNotQuad));
2716 hasNotQuad = ( nbNotQuad || nbNotQuadMeshed );
2719 // Analyse mesh and topology of FACEs: choose the bottom sub-mesh.
2720 // If there are not quadrangle FACEs, they are top and bottom ones.
2721 // Not quadrangle FACEs must be only on top and bottom.
2723 SMESH_subMesh * botSM = 0;
2724 SMESH_subMesh * topSM = 0;
2726 if ( hasNotQuad ) // can choose a bottom FACE
2728 if ( nbNotQuadMeshed > 0 ) botSM = notQuadElemSubMesh.front();
2729 else botSM = notQuadGeomSubMesh.front();
2730 if ( nbNotQuadMeshed > 1 ) topSM = notQuadElemSubMesh.back();
2731 else if ( nbNotQuad > 1 ) topSM = notQuadGeomSubMesh.back();
2733 if ( topSM == botSM ) {
2734 if ( nbNotQuadMeshed > 1 ) topSM = notQuadElemSubMesh.front();
2735 else topSM = notQuadGeomSubMesh.front();
2738 // detect mesh triangles on wall FACEs
2739 if ( nbNotQuad == 2 && nbNotQuadMeshed > 0 ) {
2741 if ( nbNotQuadMeshed == 1 )
2742 ok = ( find( notQuadGeomSubMesh.begin(),
2743 notQuadGeomSubMesh.end(), botSM ) != notQuadGeomSubMesh.end() );
2745 ok = ( notQuadGeomSubMesh == notQuadElemSubMesh );
2747 return toSM( error(COMPERR_BAD_INPUT_MESH,
2748 "Side face meshed with not quadrangle elements"));
2752 thePrism.myNotQuadOnTop = ( nbNotQuadMeshed > 1 );
2754 // use thePrism.myBottom
2755 if ( !thePrism.myBottom.IsNull() )
2758 if ( ! botSM->GetSubShape().IsSame( thePrism.myBottom )) {
2759 std::swap( botSM, topSM );
2760 if ( !botSM || ! botSM->GetSubShape().IsSame( thePrism.myBottom ))
2761 return toSM( error( COMPERR_BAD_INPUT_MESH,
2762 "Incompatible non-structured sub-meshes"));
2766 botSM = myHelper->GetMesh()->GetSubMesh( thePrism.myBottom );
2769 else if ( !botSM ) // find a proper bottom
2771 // composite walls or not prism shape
2772 for ( TopExp_Explorer f( shape3D, TopAbs_FACE ); f.More(); f.Next() )
2774 int minNbFaces = 2 + myHelper->Count( f.Current(), TopAbs_EDGE, false);
2775 if ( nbFaces >= minNbFaces)
2778 thePrism.myBottom = TopoDS::Face( f.Current() );
2779 if ( initPrism( thePrism, shape3D ))
2782 return toSM( error( COMPERR_BAD_SHAPE ));
2786 // find vertex 000 - the one with smallest coordinates (for easy DEBUG :-)
2788 double minVal = DBL_MAX, minX, val;
2789 for ( TopExp_Explorer exp( botSM->GetSubShape(), TopAbs_VERTEX );
2790 exp.More(); exp.Next() )
2792 const TopoDS_Vertex& v = TopoDS::Vertex( exp.Current() );
2793 gp_Pnt P = BRep_Tool::Pnt( v );
2794 val = P.X() + P.Y() + P.Z();
2795 if ( val < minVal || ( val == minVal && P.X() < minX )) {
2802 thePrism.myShape3D = shape3D;
2803 if ( thePrism.myBottom.IsNull() )
2804 thePrism.myBottom = TopoDS::Face( botSM->GetSubShape() );
2805 thePrism.myBottom.Orientation( myHelper->GetSubShapeOri( shape3D,
2806 thePrism.myBottom ));
2807 // Get ordered bottom edges
2808 TopoDS_Face reverseBottom = // to have order of top EDGEs as in the top FACE
2809 TopoDS::Face( thePrism.myBottom.Reversed() );
2810 SMESH_Block::GetOrderedEdges( reverseBottom,
2811 thePrism.myBottomEdges,
2812 thePrism.myNbEdgesInWires, V000 );
2814 // Get Wall faces corresponding to the ordered bottom edges and the top FACE
2815 if ( !getWallFaces( thePrism, nbFaces ))
2816 return false; //toSM( error(COMPERR_BAD_SHAPE, "Can't find side faces"));
2820 if ( !thePrism.myTop.IsSame( topSM->GetSubShape() ))
2822 (notQuadGeomSubMesh.empty() ? COMPERR_BAD_INPUT_MESH : COMPERR_BAD_SHAPE,
2823 "Non-quadrilateral faces are not opposite"));
2825 // check that the found top and bottom FACEs are opposite
2826 list< TopoDS_Edge >::iterator edge = thePrism.myBottomEdges.begin();
2827 for ( ; edge != thePrism.myBottomEdges.end(); ++edge )
2828 if ( myHelper->IsSubShape( *edge, thePrism.myTop ))
2830 (notQuadGeomSubMesh.empty() ? COMPERR_BAD_INPUT_MESH : COMPERR_BAD_SHAPE,
2831 "Non-quadrilateral faces are not opposite"));
2834 if ( thePrism.myBottomEdges.size() > thePrism.myWallQuads.size() )
2836 // composite bottom sides => set thePrism upside-down
2837 thePrism.SetUpsideDown();
2843 //================================================================================
2845 * \brief Initialization.
2846 * \param helper - helper loaded with mesh and 3D shape
2847 * \param thePrism - a prism data
2848 * \retval bool - false if a mesh or a shape are KO
2850 //================================================================================
2852 bool StdMeshers_PrismAsBlock::Init(SMESH_MesherHelper* helper,
2853 const Prism_3D::TPrismTopo& thePrism)
2856 SMESHDS_Mesh* meshDS = myHelper->GetMeshDS();
2857 SMESH_Mesh* mesh = myHelper->GetMesh();
2860 delete mySide; mySide = 0;
2862 vector< TSideFace* > sideFaces( NB_WALL_FACES, 0 );
2863 vector< pair< double, double> > params( NB_WALL_FACES );
2864 mySide = new TSideFace( *mesh, sideFaces, params );
2867 SMESH_Block::init();
2868 myShapeIDMap.Clear();
2869 myShapeIndex2ColumnMap.clear();
2871 int wallFaceIds[ NB_WALL_FACES ] = { // to walk around a block
2872 SMESH_Block::ID_Fx0z, SMESH_Block::ID_F1yz,
2873 SMESH_Block::ID_Fx1z, SMESH_Block::ID_F0yz
2876 myError = SMESH_ComputeError::New();
2878 myNotQuadOnTop = thePrism.myNotQuadOnTop;
2880 // Find columns of wall nodes and calculate edges' lengths
2881 // --------------------------------------------------------
2883 myParam2ColumnMaps.clear();
2884 myParam2ColumnMaps.resize( thePrism.myBottomEdges.size() ); // total nb edges
2886 size_t iE, nbEdges = thePrism.myNbEdgesInWires.front(); // nb outer edges
2887 vector< double > edgeLength( nbEdges );
2888 multimap< double, int > len2edgeMap;
2890 // for each EDGE: either split into several parts, or join with several next EDGEs
2891 vector<int> nbSplitPerEdge( nbEdges, 0 );
2892 vector<int> nbUnitePerEdge( nbEdges, 0 ); // -1 means "joined to a previous"
2894 // consider continuous straight EDGEs as one side
2895 const int nbSides = countNbSides( thePrism, nbUnitePerEdge, edgeLength );
2897 list< TopoDS_Edge >::const_iterator edgeIt = thePrism.myBottomEdges.begin();
2898 for ( iE = 0; iE < nbEdges; ++iE, ++edgeIt )
2900 TParam2ColumnMap & faceColumns = myParam2ColumnMaps[ iE ];
2902 Prism_3D::TQuadList::const_iterator quad = thePrism.myWallQuads[ iE ].begin();
2903 for ( ; quad != thePrism.myWallQuads[ iE ].end(); ++quad )
2905 const TopoDS_Edge& quadBot = (*quad)->side[ QUAD_BOTTOM_SIDE ].grid->Edge( 0 );
2906 if ( !myHelper->LoadNodeColumns( faceColumns, (*quad)->face, quadBot, meshDS ))
2907 return error(COMPERR_BAD_INPUT_MESH, TCom("Can't find regular quadrangle mesh ")
2908 << "on a side face #" << MeshDS()->ShapeToIndex( (*quad)->face ));
2910 SHOWYXZ("\np1 F " <<iE, gpXYZ(faceColumns.begin()->second.front() ));
2911 SHOWYXZ("p2 F " <<iE, gpXYZ(faceColumns.rbegin()->second.front() ));
2912 SHOWYXZ("V First "<<iE, BRep_Tool::Pnt( TopExp::FirstVertex(*edgeIt,true )));
2914 if ( nbSides < NB_WALL_FACES ) // fill map used to split faces
2915 len2edgeMap.insert( make_pair( edgeLength[ iE ], iE )); // sort edges by length
2917 // Load columns of internal edges (forming holes)
2918 // and fill map ShapeIndex to TParam2ColumnMap for them
2919 for ( ; edgeIt != thePrism.myBottomEdges.end() ; ++edgeIt, ++iE )
2921 TParam2ColumnMap & faceColumns = myParam2ColumnMaps[ iE ];
2923 Prism_3D::TQuadList::const_iterator quad = thePrism.myWallQuads[ iE ].begin();
2924 for ( ; quad != thePrism.myWallQuads[ iE ].end(); ++quad )
2926 const TopoDS_Edge& quadBot = (*quad)->side[ QUAD_BOTTOM_SIDE ].grid->Edge( 0 );
2927 if ( !myHelper->LoadNodeColumns( faceColumns, (*quad)->face, quadBot, meshDS ))
2928 return error(COMPERR_BAD_INPUT_MESH, TCom("Can't find regular quadrangle mesh ")
2929 << "on a side face #" << MeshDS()->ShapeToIndex( (*quad)->face ));
2932 int id = MeshDS()->ShapeToIndex( *edgeIt );
2933 bool isForward = true; // meaningless for intenal wires
2934 myShapeIndex2ColumnMap[ id ] = make_pair( & faceColumns, isForward );
2935 // columns for vertices
2937 const SMDS_MeshNode* n0 = faceColumns.begin()->second.front();
2938 id = n0->getshapeId();
2939 myShapeIndex2ColumnMap[ id ] = make_pair( & faceColumns, isForward );
2941 const SMDS_MeshNode* n1 = faceColumns.rbegin()->second.front();
2942 id = n1->getshapeId();
2943 myShapeIndex2ColumnMap[ id ] = make_pair( & faceColumns, isForward );
2945 // SHOWYXZ("\np1 F " <<iE, gpXYZ(faceColumns.begin()->second.front() ));
2946 // SHOWYXZ("p2 F " <<iE, gpXYZ(faceColumns.rbegin()->second.front() ));
2947 // SHOWYXZ("V First "<<iE, BRep_Tool::Pnt( TopExp::FirstVertex(*edgeIt,true )));
2950 // Create 4 wall faces of a block
2951 // -------------------------------
2953 if ( nbSides <= NB_WALL_FACES ) // ************* Split faces if necessary
2955 if ( nbSides != NB_WALL_FACES ) // define how to split
2957 if ( len2edgeMap.size() != nbEdges )
2958 RETURN_BAD_RESULT("Uniqueness of edge lengths not assured");
2960 multimap< double, int >::reverse_iterator maxLen_i = len2edgeMap.rbegin();
2961 multimap< double, int >::reverse_iterator midLen_i = ++len2edgeMap.rbegin();
2963 double maxLen = maxLen_i->first;
2964 double midLen = ( len2edgeMap.size() == 1 ) ? 0 : midLen_i->first;
2965 switch ( nbEdges ) {
2966 case 1: // 0-th edge is split into 4 parts
2967 nbSplitPerEdge[ 0 ] = 4;
2969 case 2: // either the longest edge is split into 3 parts, or both edges into halves
2970 if ( maxLen / 3 > midLen / 2 ) {
2971 nbSplitPerEdge[ maxLen_i->second ] = 3;
2974 nbSplitPerEdge[ maxLen_i->second ] = 2;
2975 nbSplitPerEdge[ midLen_i->second ] = 2;
2980 // split longest into 3 parts
2981 nbSplitPerEdge[ maxLen_i->second ] = 3;
2983 // split longest into halves
2984 nbSplitPerEdge[ maxLen_i->second ] = 2;
2988 else // **************************** Unite faces
2990 int nbExraFaces = nbSides - 4; // nb of faces to fuse
2991 for ( iE = 0; iE < nbEdges; ++iE )
2993 if ( nbUnitePerEdge[ iE ] < 0 )
2995 // look for already united faces
2996 for ( int i = iE; i < iE + nbExraFaces; ++i )
2998 if ( nbUnitePerEdge[ i ] > 0 ) // a side including nbUnitePerEdge[i]+1 edge
2999 nbExraFaces += nbUnitePerEdge[ i ];
3000 nbUnitePerEdge[ i ] = -1;
3002 nbUnitePerEdge[ iE ] = nbExraFaces;
3007 // Create TSideFace's
3009 list< TopoDS_Edge >::const_iterator botE = thePrism.myBottomEdges.begin();
3010 for ( iE = 0; iE < nbEdges; ++iE, ++botE )
3012 TFaceQuadStructPtr quad = thePrism.myWallQuads[ iE ].front();
3013 const int nbSplit = nbSplitPerEdge[ iE ];
3014 const int nbExraFaces = nbUnitePerEdge[ iE ] + 1;
3015 if ( nbSplit > 0 ) // split
3017 vector< double > params;
3018 splitParams( nbSplit, &myParam2ColumnMaps[ iE ], params );
3019 const bool isForward =
3020 StdMeshers_PrismAsBlock::IsForwardEdge( myHelper->GetMeshDS(),
3021 myParam2ColumnMaps[iE],
3022 *botE, SMESH_Block::ID_Fx0z );
3023 for ( int i = 0; i < nbSplit; ++i ) {
3024 double f = ( isForward ? params[ i ] : params[ nbSplit - i-1 ]);
3025 double l = ( isForward ? params[ i+1 ] : params[ nbSplit - i ]);
3026 TSideFace* comp = new TSideFace( *mesh, wallFaceIds[ iSide ],
3027 thePrism.myWallQuads[ iE ], *botE,
3028 &myParam2ColumnMaps[ iE ], f, l );
3029 mySide->SetComponent( iSide++, comp );
3032 else if ( nbExraFaces > 1 ) // unite
3034 double u0 = 0, sumLen = 0;
3035 for ( int i = iE; i < iE + nbExraFaces; ++i )
3036 sumLen += edgeLength[ i ];
3038 vector< TSideFace* > components( nbExraFaces );
3039 vector< pair< double, double> > params( nbExraFaces );
3040 bool endReached = false;
3041 for ( int i = 0; i < nbExraFaces; ++i, ++botE, ++iE )
3043 if ( iE == nbEdges )
3046 botE = thePrism.myBottomEdges.begin();
3049 components[ i ] = new TSideFace( *mesh, wallFaceIds[ iSide ],
3050 thePrism.myWallQuads[ iE ], *botE,
3051 &myParam2ColumnMaps[ iE ]);
3052 double u1 = u0 + edgeLength[ iE ] / sumLen;
3053 params[ i ] = make_pair( u0 , u1 );
3056 TSideFace* comp = new TSideFace( *mesh, components, params );
3057 mySide->SetComponent( iSide++, comp );
3060 --iE; // for increment in an external loop on iE
3063 else if ( nbExraFaces < 0 ) // skip already united face
3068 TSideFace* comp = new TSideFace( *mesh, wallFaceIds[ iSide ],
3069 thePrism.myWallQuads[ iE ], *botE,
3070 &myParam2ColumnMaps[ iE ]);
3071 mySide->SetComponent( iSide++, comp );
3076 // Fill geometry fields of SMESH_Block
3077 // ------------------------------------
3079 vector< int > botEdgeIdVec;
3080 SMESH_Block::GetFaceEdgesIDs( ID_BOT_FACE, botEdgeIdVec );
3082 bool isForward[NB_WALL_FACES] = { true, true, true, true };
3083 Adaptor2d_Curve2d* botPcurves[NB_WALL_FACES];
3084 Adaptor2d_Curve2d* topPcurves[NB_WALL_FACES];
3086 for ( int iF = 0; iF < NB_WALL_FACES; ++iF )
3088 TSideFace * sideFace = mySide->GetComponent( iF );
3090 RETURN_BAD_RESULT("NULL TSideFace");
3091 int fID = sideFace->FaceID(); // in-block ID
3093 // fill myShapeIDMap
3094 if ( sideFace->InsertSubShapes( myShapeIDMap ) != 8 &&
3095 !sideFace->IsComplex())
3096 MESSAGE( ": Warning : InsertSubShapes() < 8 on side " << iF );
3098 // side faces geometry
3099 Adaptor2d_Curve2d* pcurves[NB_WALL_FACES];
3100 if ( !sideFace->GetPCurves( pcurves ))
3101 RETURN_BAD_RESULT("TSideFace::GetPCurves() failed");
3103 SMESH_Block::TFace& tFace = myFace[ fID - ID_FirstF ];
3104 tFace.Set( fID, sideFace->Surface(), pcurves, isForward );
3106 SHOWYXZ( endl<<"F "<< iF << " id " << fID << " FRW " << sideFace->IsForward(), sideFace->Value(0,0));
3107 // edges 3D geometry
3108 vector< int > edgeIdVec;
3109 SMESH_Block::GetFaceEdgesIDs( fID, edgeIdVec );
3110 for ( int isMax = 0; isMax < 2; ++isMax ) {
3112 int eID = edgeIdVec[ isMax ];
3113 SMESH_Block::TEdge& tEdge = myEdge[ eID - ID_FirstE ];
3114 tEdge.Set( eID, sideFace->HorizCurve(isMax), true);
3115 SHOWYXZ(eID<<" HOR"<<isMax<<"(0)", sideFace->HorizCurve(isMax)->Value(0));
3116 SHOWYXZ(eID<<" HOR"<<isMax<<"(1)", sideFace->HorizCurve(isMax)->Value(1));
3119 int eID = edgeIdVec[ isMax+2 ];
3120 SMESH_Block::TEdge& tEdge = myEdge[ eID - ID_FirstE ];
3121 tEdge.Set( eID, sideFace->VertiCurve(isMax), true);
3122 SHOWYXZ(eID<<" VER"<<isMax<<"(0)", sideFace->VertiCurve(isMax)->Value(0));
3123 SHOWYXZ(eID<<" VER"<<isMax<<"(1)", sideFace->VertiCurve(isMax)->Value(1));
3126 vector< int > vertexIdVec;
3127 SMESH_Block::GetEdgeVertexIDs( eID, vertexIdVec );
3128 myPnt[ vertexIdVec[0] - ID_FirstV ] = tEdge.GetCurve()->Value(0).XYZ();
3129 myPnt[ vertexIdVec[1] - ID_FirstV ] = tEdge.GetCurve()->Value(1).XYZ();
3132 // pcurves on horizontal faces
3133 for ( iE = 0; iE < NB_WALL_FACES; ++iE ) {
3134 if ( edgeIdVec[ BOTTOM_EDGE ] == botEdgeIdVec[ iE ] ) {
3135 botPcurves[ iE ] = sideFace->HorizPCurve( false, thePrism.myBottom );
3136 topPcurves[ iE ] = sideFace->HorizPCurve( true, thePrism.myTop );
3140 //sideFace->dumpNodes( 4 ); // debug
3142 // horizontal faces geometry
3144 SMESH_Block::TFace& tFace = myFace[ ID_BOT_FACE - ID_FirstF ];
3145 tFace.Set( ID_BOT_FACE, new BRepAdaptor_Surface( thePrism.myBottom ), botPcurves, isForward );
3146 SMESH_Block::Insert( thePrism.myBottom, ID_BOT_FACE, myShapeIDMap );
3149 SMESH_Block::TFace& tFace = myFace[ ID_TOP_FACE - ID_FirstF ];
3150 tFace.Set( ID_TOP_FACE, new BRepAdaptor_Surface( thePrism.myTop ), topPcurves, isForward );
3151 SMESH_Block::Insert( thePrism.myTop, ID_TOP_FACE, myShapeIDMap );
3153 //faceGridToPythonDump( SMESH_Block::ID_Fxy0, 50 );
3154 //faceGridToPythonDump( SMESH_Block::ID_Fxy1 );
3156 // Fill map ShapeIndex to TParam2ColumnMap
3157 // ----------------------------------------
3159 list< TSideFace* > fList;
3160 list< TSideFace* >::iterator fListIt;
3161 fList.push_back( mySide );
3162 for ( fListIt = fList.begin(); fListIt != fList.end(); ++fListIt)
3164 int nb = (*fListIt)->NbComponents();
3165 for ( int i = 0; i < nb; ++i ) {
3166 if ( TSideFace* comp = (*fListIt)->GetComponent( i ))
3167 fList.push_back( comp );
3169 if ( TParam2ColumnMap* cols = (*fListIt)->GetColumns()) {
3170 // columns for a base edge
3171 int id = MeshDS()->ShapeToIndex( (*fListIt)->BaseEdge() );
3172 bool isForward = (*fListIt)->IsForward();
3173 myShapeIndex2ColumnMap[ id ] = make_pair( cols, isForward );
3175 // columns for vertices
3176 const SMDS_MeshNode* n0 = cols->begin()->second.front();
3177 id = n0->getshapeId();
3178 myShapeIndex2ColumnMap[ id ] = make_pair( cols, isForward );
3180 const SMDS_MeshNode* n1 = cols->rbegin()->second.front();
3181 id = n1->getshapeId();
3182 myShapeIndex2ColumnMap[ id ] = make_pair( cols, !isForward );
3186 // #define SHOWYXZ(msg, xyz) { \
3187 // gp_Pnt p (xyz); \
3188 // cout << msg << " ("<< p.X() << "; " <<p.Y() << "; " <<p.Z() << ") " <<endl; \
3190 // double _u[]={ 0.1, 0.1, 0.9, 0.9 };
3191 // double _v[]={ 0.1, 0.9, 0.1, 0.9 };
3192 // for ( int z = 0; z < 2; ++z )
3193 // for ( int i = 0; i < 4; ++i )
3195 // //gp_XYZ testPar(0.25, 0.25, 0), testCoord;
3196 // int iFace = (z ? ID_TOP_FACE : ID_BOT_FACE);
3197 // gp_XYZ testPar(_u[i], _v[i], z), testCoord;
3198 // if ( !FacePoint( iFace, testPar, testCoord ))
3199 // RETURN_BAD_RESULT("TEST FacePoint() FAILED");
3200 // SHOWYXZ("IN TEST PARAM" , testPar);
3201 // SHOWYXZ("OUT TEST CORD" , testCoord);
3202 // if ( !ComputeParameters( testCoord, testPar , iFace))
3203 // RETURN_BAD_RESULT("TEST ComputeParameters() FAILED");
3204 // SHOWYXZ("OUT TEST PARAM" , testPar);
3209 //================================================================================
3211 * \brief Return pointer to column of nodes
3212 * \param node - bottom node from which the returned column goes up
3213 * \retval const TNodeColumn* - the found column
3215 //================================================================================
3217 const TNodeColumn* StdMeshers_PrismAsBlock::GetNodeColumn(const SMDS_MeshNode* node) const
3219 int sID = node->getshapeId();
3221 map<int, pair< TParam2ColumnMap*, bool > >::const_iterator col_frw =
3222 myShapeIndex2ColumnMap.find( sID );
3223 if ( col_frw != myShapeIndex2ColumnMap.end() ) {
3224 const TParam2ColumnMap* cols = col_frw->second.first;
3225 TParam2ColumnIt u_col = cols->begin();
3226 for ( ; u_col != cols->end(); ++u_col )
3227 if ( u_col->second[ 0 ] == node )
3228 return & u_col->second;
3233 //=======================================================================
3234 //function : GetLayersTransformation
3235 //purpose : Return transformations to get coordinates of nodes of each layer
3236 // by nodes of the bottom. Layer is a set of nodes at a certain step
3237 // from bottom to top.
3238 // Transformation to get top node from bottom ones is computed
3239 // only if the top FACE is not meshed.
3240 //=======================================================================
3242 bool StdMeshers_PrismAsBlock::GetLayersTransformation(vector<gp_Trsf> & trsf,
3243 const Prism_3D::TPrismTopo& prism) const
3245 const bool itTopMeshed = !SubMesh( ID_BOT_FACE )->IsEmpty();
3246 const int zSize = VerticalSize();
3247 if ( zSize < 3 && !itTopMeshed ) return true;
3248 trsf.resize( zSize - 1 );
3250 // Select some node columns by which we will define coordinate system of layers
3252 vector< const TNodeColumn* > columns;
3255 list< TopoDS_Edge >::const_iterator edgeIt = prism.myBottomEdges.begin();
3256 for ( int iE = 0; iE < prism.myNbEdgesInWires.front(); ++iE, ++edgeIt )
3258 if ( SMESH_Algo::isDegenerated( *edgeIt )) continue;
3259 const TParam2ColumnMap* u2colMap =
3260 GetParam2ColumnMap( MeshDS()->ShapeToIndex( *edgeIt ), isReverse );
3261 if ( !u2colMap ) return false;
3262 double f = u2colMap->begin()->first, l = u2colMap->rbegin()->first;
3263 //isReverse = ( edgeIt->Orientation() == TopAbs_REVERSED );
3264 //if ( isReverse ) swap ( f, l ); -- u2colMap takes orientation into account
3265 const int nbCol = 5;
3266 for ( int i = 0; i < nbCol; ++i )
3268 double u = f + i/double(nbCol) * ( l - f );
3269 const TNodeColumn* col = & getColumn( u2colMap, u )->second;
3270 if ( columns.empty() || col != columns.back() )
3271 columns.push_back( col );
3276 // Find tolerance to check transformations
3281 for ( int i = 0; i < columns.size(); ++i )
3282 bndBox.Add( gpXYZ( columns[i]->front() ));
3283 tol2 = bndBox.SquareExtent() * 1e-5;
3286 // Compute transformations
3289 gp_Trsf fromCsZ, toCs0;
3290 gp_Ax3 cs0 = getLayerCoordSys(0, columns, xCol );
3291 //double dist0 = cs0.Location().Distance( gpXYZ( (*columns[0])[0]));
3292 toCs0.SetTransformation( cs0 );
3293 for ( int z = 1; z < zSize; ++z )
3295 gp_Ax3 csZ = getLayerCoordSys(z, columns, xCol );
3296 //double distZ = csZ.Location().Distance( gpXYZ( (*columns[0])[z]));
3297 fromCsZ.SetTransformation( csZ );
3299 gp_Trsf& t = trsf[ z-1 ];
3300 t = fromCsZ * toCs0;
3301 //t.SetScaleFactor( distZ/dist0 ); - it does not work properly, wrong base point
3303 // check a transformation
3304 for ( int i = 0; i < columns.size(); ++i )
3306 gp_Pnt p0 = gpXYZ( (*columns[i])[0] );
3307 gp_Pnt pz = gpXYZ( (*columns[i])[z] );
3308 t.Transforms( p0.ChangeCoord() );
3309 if ( p0.SquareDistance( pz ) > tol2 )
3312 return ( z == zSize - 1 ); // OK if fails only botton->top trsf
3319 //================================================================================
3321 * \brief Check curve orientation of a bootom edge
3322 * \param meshDS - mesh DS
3323 * \param columnsMap - node columns map of side face
3324 * \param bottomEdge - the bootom edge
3325 * \param sideFaceID - side face in-block ID
3326 * \retval bool - true if orientation coinside with in-block forward orientation
3328 //================================================================================
3330 bool StdMeshers_PrismAsBlock::IsForwardEdge(SMESHDS_Mesh* meshDS,
3331 const TParam2ColumnMap& columnsMap,
3332 const TopoDS_Edge & bottomEdge,
3333 const int sideFaceID)
3335 bool isForward = false;
3336 if ( SMESH_MesherHelper::IsClosedEdge( bottomEdge ))
3338 isForward = ( bottomEdge.Orientation() == TopAbs_FORWARD );
3342 const TNodeColumn& firstCol = columnsMap.begin()->second;
3343 const SMDS_MeshNode* bottomNode = firstCol[0];
3344 TopoDS_Shape firstVertex = SMESH_MesherHelper::GetSubShapeByNode( bottomNode, meshDS );
3345 isForward = ( firstVertex.IsSame( TopExp::FirstVertex( bottomEdge, true )));
3347 // on 2 of 4 sides first vertex is end
3348 if ( sideFaceID == ID_Fx1z || sideFaceID == ID_F0yz )
3349 isForward = !isForward;
3353 //=======================================================================
3354 //function : faceGridToPythonDump
3355 //purpose : Prints a script creating a normal grid on the prism side
3356 //=======================================================================
3358 void StdMeshers_PrismAsBlock::faceGridToPythonDump(const SMESH_Block::TShapeID face,
3362 gp_XYZ pOnF[6] = { gp_XYZ(0,0,0), gp_XYZ(0,0,1),
3363 gp_XYZ(0,0,0), gp_XYZ(0,1,0),
3364 gp_XYZ(0,0,0), gp_XYZ(1,0,0) };
3366 cout << "mesh = smesh.Mesh( 'Face " << face << "')" << endl;
3367 SMESH_Block::TFace& f = myFace[ face - ID_FirstF ];
3368 gp_XYZ params = pOnF[ face - ID_FirstF ];
3369 //const int nb = 10; // nb face rows
3370 for ( int j = 0; j <= nb; ++j )
3372 params.SetCoord( f.GetVInd(), double( j )/ nb );
3373 for ( int i = 0; i <= nb; ++i )
3375 params.SetCoord( f.GetUInd(), double( i )/ nb );
3376 gp_XYZ p = f.Point( params );
3377 gp_XY uv = f.GetUV( params );
3378 cout << "mesh.AddNode( " << p.X() << ", " << p.Y() << ", " << p.Z() << " )"
3379 << " # " << 1 + i + j * ( nb + 1 )
3380 << " ( " << i << ", " << j << " ) "
3381 << " UV( " << uv.X() << ", " << uv.Y() << " )" << endl;
3382 ShellPoint( params, p2 );
3383 double dist = ( p2 - p ).Modulus();
3385 cout << "#### dist from ShellPoint " << dist
3386 << " (" << p2.X() << ", " << p2.Y() << ", " << p2.Z() << " ) " << endl;
3389 for ( int j = 0; j < nb; ++j )
3390 for ( int i = 0; i < nb; ++i )
3392 int n = 1 + i + j * ( nb + 1 );
3393 cout << "mesh.AddFace([ "
3394 << n << ", " << n+1 << ", "
3395 << n+nb+2 << ", " << n+nb+1 << "]) " << endl;
3401 //================================================================================
3403 * \brief Constructor
3404 * \param faceID - in-block ID
3405 * \param face - geom FACE
3406 * \param baseEdge - EDGE proreply oriented in the bottom EDGE !!!
3407 * \param columnsMap - map of node columns
3408 * \param first - first normalized param
3409 * \param last - last normalized param
3411 //================================================================================
3413 StdMeshers_PrismAsBlock::TSideFace::TSideFace(SMESH_Mesh& mesh,
3415 const Prism_3D::TQuadList& quadList,
3416 const TopoDS_Edge& baseEdge,
3417 TParam2ColumnMap* columnsMap,
3421 myParamToColumnMap( columnsMap ),
3424 myParams.resize( 1 );
3425 myParams[ 0 ] = make_pair( first, last );
3426 mySurface = PSurface( new BRepAdaptor_Surface( quadList.front()->face ));
3427 myBaseEdge = baseEdge;
3428 myIsForward = StdMeshers_PrismAsBlock::IsForwardEdge( myHelper.GetMeshDS(),
3429 *myParamToColumnMap,
3431 myHelper.SetSubShape( quadList.front()->face );
3433 if ( quadList.size() > 1 ) // side is vertically composite
3435 // fill myShapeID2Surf map to enable finding a right surface by any sub-shape ID
3437 SMESHDS_Mesh* meshDS = myHelper.GetMeshDS();
3439 TopTools_IndexedDataMapOfShapeListOfShape subToFaces;
3440 Prism_3D::TQuadList::const_iterator quad = quadList.begin();
3441 for ( ; quad != quadList.end(); ++quad )
3443 const TopoDS_Face& face = (*quad)->face;
3444 TopExp::MapShapesAndAncestors( face, TopAbs_VERTEX, TopAbs_FACE, subToFaces );
3445 TopExp::MapShapesAndAncestors( face, TopAbs_EDGE, TopAbs_FACE, subToFaces );
3446 myShapeID2Surf.insert( make_pair( meshDS->ShapeToIndex( face ),
3447 PSurface( new BRepAdaptor_Surface( face ))));
3449 for ( int i = 1; i <= subToFaces.Extent(); ++i )
3451 const TopoDS_Shape& sub = subToFaces.FindKey( i );
3452 TopTools_ListOfShape& faces = subToFaces( i );
3453 int subID = meshDS->ShapeToIndex( sub );
3454 int faceID = meshDS->ShapeToIndex( faces.First() );
3455 myShapeID2Surf.insert ( make_pair( subID, myShapeID2Surf[ faceID ]));
3460 //================================================================================
3462 * \brief Constructor of a complex side face
3464 //================================================================================
3466 StdMeshers_PrismAsBlock::TSideFace::
3467 TSideFace(SMESH_Mesh& mesh,
3468 const vector< TSideFace* >& components,
3469 const vector< pair< double, double> > & params)
3470 :myID( components[0] ? components[0]->myID : 0 ),
3471 myParamToColumnMap( 0 ),
3473 myIsForward( true ),
3474 myComponents( components ),
3477 if ( myID == ID_Fx1z || myID == ID_F0yz )
3479 // reverse components
3480 std::reverse( myComponents.begin(), myComponents.end() );
3481 std::reverse( myParams.begin(), myParams.end() );
3482 for ( size_t i = 0; i < myParams.size(); ++i )
3484 const double f = myParams[i].first;
3485 const double l = myParams[i].second;
3486 myParams[i] = make_pair( 1. - l, 1. - f );
3490 //================================================================================
3492 * \brief Copy constructor
3493 * \param other - other side
3495 //================================================================================
3497 StdMeshers_PrismAsBlock::TSideFace::TSideFace( const TSideFace& other ):
3498 myID ( other.myID ),
3499 myParamToColumnMap ( other.myParamToColumnMap ),
3500 mySurface ( other.mySurface ),
3501 myBaseEdge ( other.myBaseEdge ),
3502 myShapeID2Surf ( other.myShapeID2Surf ),
3503 myParams ( other.myParams ),
3504 myIsForward ( other.myIsForward ),
3505 myComponents ( other.myComponents.size() ),
3506 myHelper ( *other.myHelper.GetMesh() )
3508 for (int i = 0 ; i < myComponents.size(); ++i )
3509 myComponents[ i ] = new TSideFace( *other.myComponents[ i ]);
3512 //================================================================================
3514 * \brief Deletes myComponents
3516 //================================================================================
3518 StdMeshers_PrismAsBlock::TSideFace::~TSideFace()
3520 for (int i = 0 ; i < myComponents.size(); ++i )
3521 if ( myComponents[ i ] )
3522 delete myComponents[ i ];
3525 //================================================================================
3527 * \brief Return geometry of the vertical curve
3528 * \param isMax - true means curve located closer to (1,1,1) block point
3529 * \retval Adaptor3d_Curve* - curve adaptor
3531 //================================================================================
3533 Adaptor3d_Curve* StdMeshers_PrismAsBlock::TSideFace::VertiCurve(const bool isMax) const
3535 if ( !myComponents.empty() ) {
3537 return myComponents.back()->VertiCurve(isMax);
3539 return myComponents.front()->VertiCurve(isMax);
3541 double f = myParams[0].first, l = myParams[0].second;
3542 if ( !myIsForward ) std::swap( f, l );
3543 return new TVerticalEdgeAdaptor( myParamToColumnMap, isMax ? l : f );
3546 //================================================================================
3548 * \brief Return geometry of the top or bottom curve
3550 * \retval Adaptor3d_Curve* -
3552 //================================================================================
3554 Adaptor3d_Curve* StdMeshers_PrismAsBlock::TSideFace::HorizCurve(const bool isTop) const
3556 return new THorizontalEdgeAdaptor( this, isTop );
3559 //================================================================================
3561 * \brief Return pcurves
3562 * \param pcurv - array of 4 pcurves
3563 * \retval bool - is a success
3565 //================================================================================
3567 bool StdMeshers_PrismAsBlock::TSideFace::GetPCurves(Adaptor2d_Curve2d* pcurv[4]) const
3569 int iEdge[ 4 ] = { BOTTOM_EDGE, TOP_EDGE, V0_EDGE, V1_EDGE };
3571 for ( int i = 0 ; i < 4 ; ++i ) {
3572 Handle(Geom2d_Line) line;
3573 switch ( iEdge[ i ] ) {
3575 line = new Geom2d_Line( gp_Pnt2d( 0, 1 ), gp::DX2d() ); break;
3577 line = new Geom2d_Line( gp::Origin2d(), gp::DX2d() ); break;
3579 line = new Geom2d_Line( gp::Origin2d(), gp::DY2d() ); break;
3581 line = new Geom2d_Line( gp_Pnt2d( 1, 0 ), gp::DY2d() ); break;
3583 pcurv[ i ] = new Geom2dAdaptor_Curve( line, 0, 1 );
3588 //================================================================================
3590 * \brief Returns geometry of pcurve on a horizontal face
3591 * \param isTop - is top or bottom face
3592 * \param horFace - a horizontal face
3593 * \retval Adaptor2d_Curve2d* - curve adaptor
3595 //================================================================================
3598 StdMeshers_PrismAsBlock::TSideFace::HorizPCurve(const bool isTop,
3599 const TopoDS_Face& horFace) const
3601 return new TPCurveOnHorFaceAdaptor( this, isTop, horFace );
3604 //================================================================================
3606 * \brief Return a component corresponding to parameter
3607 * \param U - parameter along a horizontal size
3608 * \param localU - parameter along a horizontal size of a component
3609 * \retval TSideFace* - found component
3611 //================================================================================
3613 StdMeshers_PrismAsBlock::TSideFace*
3614 StdMeshers_PrismAsBlock::TSideFace::GetComponent(const double U,double & localU) const
3617 if ( myComponents.empty() )
3618 return const_cast<TSideFace*>( this );
3621 for ( i = 0; i < myComponents.size(); ++i )
3622 if ( U < myParams[ i ].second )
3624 if ( i >= myComponents.size() )
3625 i = myComponents.size() - 1;
3627 double f = myParams[ i ].first, l = myParams[ i ].second;
3628 localU = ( U - f ) / ( l - f );
3629 return myComponents[ i ];
3632 //================================================================================
3634 * \brief Find node columns for a parameter
3635 * \param U - parameter along a horizontal edge
3636 * \param col1 - the 1st found column
3637 * \param col2 - the 2nd found column
3638 * \retval r - normalized position of U between the found columns
3640 //================================================================================
3642 double StdMeshers_PrismAsBlock::TSideFace::GetColumns(const double U,
3643 TParam2ColumnIt & col1,
3644 TParam2ColumnIt & col2) const
3646 double u = U, r = 0;
3647 if ( !myComponents.empty() ) {
3648 TSideFace * comp = GetComponent(U,u);
3649 return comp->GetColumns( u, col1, col2 );
3654 double f = myParams[0].first, l = myParams[0].second;
3655 u = f + u * ( l - f );
3657 col1 = col2 = getColumn( myParamToColumnMap, u );
3658 if ( ++col2 == myParamToColumnMap->end() ) {
3663 double uf = col1->first;
3664 double ul = col2->first;
3665 r = ( u - uf ) / ( ul - uf );
3670 //================================================================================
3672 * \brief Return all nodes at a given height together with their normalized parameters
3673 * \param [in] Z - the height of interest
3674 * \param [out] nodes - map of parameter to node
3676 //================================================================================
3678 void StdMeshers_PrismAsBlock::
3679 TSideFace::GetNodesAtZ(const int Z,
3680 map<double, const SMDS_MeshNode* >& nodes ) const
3682 if ( !myComponents.empty() )
3685 for ( size_t i = 0; i < myComponents.size(); ++i )
3687 map<double, const SMDS_MeshNode* > nn;
3688 myComponents[i]->GetNodesAtZ( Z, nn );
3689 map<double, const SMDS_MeshNode* >::iterator u2n = nn.begin();
3690 if ( !nodes.empty() && nodes.rbegin()->second == u2n->second )
3692 const double uRange = myParams[i].second - myParams[i].first;
3693 for ( ; u2n != nn.end(); ++u2n )
3694 nodes.insert( nodes.end(), make_pair( u0 + uRange * u2n->first, u2n->second ));
3700 double f = myParams[0].first, l = myParams[0].second;
3703 const double uRange = l - f;
3704 if ( Abs( uRange ) < std::numeric_limits<double>::min() )
3706 TParam2ColumnIt u2col = getColumn( myParamToColumnMap, myParams[0].first + 1e-3 );
3707 for ( ; u2col != myParamToColumnMap->end(); ++u2col )
3708 if ( u2col->first > myParams[0].second + 1e-9 )
3711 nodes.insert( nodes.end(),
3712 make_pair( ( u2col->first - f ) / uRange, u2col->second[ Z ] ));
3716 //================================================================================
3718 * \brief Return coordinates by normalized params
3719 * \param U - horizontal param
3720 * \param V - vertical param
3721 * \retval gp_Pnt - result point
3723 //================================================================================
3725 gp_Pnt StdMeshers_PrismAsBlock::TSideFace::Value(const Standard_Real U,
3726 const Standard_Real V) const
3728 if ( !myComponents.empty() ) {
3730 TSideFace * comp = GetComponent(U,u);
3731 return comp->Value( u, V );
3734 TParam2ColumnIt u_col1, u_col2;
3735 double vR, hR = GetColumns( U, u_col1, u_col2 );
3737 const SMDS_MeshNode* nn[4];
3739 // BEGIN issue 0020680: Bad cell created by Radial prism in center of torus
3740 // Workaround for a wrongly located point returned by mySurface.Value() for
3741 // UV located near boundary of BSpline surface.
3742 // To bypass the problem, we take point from 3D curve of EDGE.
3743 // It solves pb of the bloc_fiss_new.py
3744 const double tol = 1e-3;
3745 if ( V < tol || V+tol >= 1. )
3747 nn[0] = V < tol ? u_col1->second.front() : u_col1->second.back();
3748 nn[2] = V < tol ? u_col2->second.front() : u_col2->second.back();
3756 TopoDS_Shape s = myHelper.GetSubShapeByNode( nn[0], myHelper.GetMeshDS() );
3757 if ( s.ShapeType() != TopAbs_EDGE )
3758 s = myHelper.GetSubShapeByNode( nn[2], myHelper.GetMeshDS() );
3759 if ( s.ShapeType() == TopAbs_EDGE )
3760 edge = TopoDS::Edge( s );
3762 if ( !edge.IsNull() )
3764 double u1 = myHelper.GetNodeU( edge, nn[0], nn[2] );
3765 double u3 = myHelper.GetNodeU( edge, nn[2], nn[0] );
3766 double u = u1 * ( 1 - hR ) + u3 * hR;
3767 TopLoc_Location loc; double f,l;
3768 Handle(Geom_Curve) curve = BRep_Tool::Curve( edge,loc,f,l );
3769 return curve->Value( u ).Transformed( loc );
3772 // END issue 0020680: Bad cell created by Radial prism in center of torus
3774 vR = getRAndNodes( & u_col1->second, V, nn[0], nn[1] );
3775 vR = getRAndNodes( & u_col2->second, V, nn[2], nn[3] );
3777 if ( !myShapeID2Surf.empty() ) // side is vertically composite
3779 // find a FACE on which the 4 nodes lie
3780 TSideFace* me = (TSideFace*) this;
3781 int notFaceID1 = 0, notFaceID2 = 0;
3782 for ( int i = 0; i < 4; ++i )
3783 if ( nn[i]->GetPosition()->GetTypeOfPosition() == SMDS_TOP_FACE ) // node on FACE
3785 me->mySurface = me->myShapeID2Surf[ nn[i]->getshapeId() ];
3789 else if ( notFaceID1 == 0 ) // node on EDGE or VERTEX
3791 me->mySurface = me->myShapeID2Surf[ nn[i]->getshapeId() ];
3792 notFaceID1 = nn[i]->getshapeId();
3794 else if ( notFaceID1 != nn[i]->getshapeId() ) // node on other EDGE or VERTEX
3796 if ( mySurface != me->myShapeID2Surf[ nn[i]->getshapeId() ])
3797 notFaceID2 = nn[i]->getshapeId();
3799 if ( notFaceID2 ) // no nodes of FACE and nodes are on different FACEs
3801 SMESHDS_Mesh* meshDS = myHelper.GetMeshDS();
3802 TopoDS_Shape face = myHelper.GetCommonAncestor( meshDS->IndexToShape( notFaceID1 ),
3803 meshDS->IndexToShape( notFaceID2 ),
3804 *myHelper.GetMesh(),
3806 if ( face.IsNull() )
3807 throw SALOME_Exception("StdMeshers_PrismAsBlock::TSideFace::Value() face.IsNull()");
3808 int faceID = meshDS->ShapeToIndex( face );
3809 me->mySurface = me->myShapeID2Surf[ faceID ];
3811 throw SALOME_Exception("StdMeshers_PrismAsBlock::TSideFace::Value() !mySurface");
3814 ((TSideFace*) this)->myHelper.SetSubShape( mySurface->Face() );
3816 gp_XY uv1 = myHelper.GetNodeUV( mySurface->Face(), nn[0], nn[2]);
3817 gp_XY uv2 = myHelper.GetNodeUV( mySurface->Face(), nn[1], nn[3]);
3818 gp_XY uv12 = uv1 * ( 1 - vR ) + uv2 * vR;
3820 gp_XY uv3 = myHelper.GetNodeUV( mySurface->Face(), nn[2], nn[0]);
3821 gp_XY uv4 = myHelper.GetNodeUV( mySurface->Face(), nn[3], nn[1]);
3822 gp_XY uv34 = uv3 * ( 1 - vR ) + uv4 * vR;
3824 gp_XY uv = uv12 * ( 1 - hR ) + uv34 * hR;
3826 gp_Pnt p = mySurface->Value( uv.X(), uv.Y() );
3831 //================================================================================
3833 * \brief Return boundary edge
3834 * \param edge - edge index
3835 * \retval TopoDS_Edge - found edge
3837 //================================================================================
3839 TopoDS_Edge StdMeshers_PrismAsBlock::TSideFace::GetEdge(const int iEdge) const
3841 if ( !myComponents.empty() ) {
3843 case V0_EDGE : return myComponents.front()->GetEdge( iEdge );
3844 case V1_EDGE : return myComponents.back() ->GetEdge( iEdge );
3845 default: return TopoDS_Edge();
3849 const SMDS_MeshNode* node = 0;
3850 SMESHDS_Mesh * meshDS = myHelper.GetMesh()->GetMeshDS();
3851 TNodeColumn* column;
3856 column = & (( ++myParamToColumnMap->begin())->second );
3857 node = ( iEdge == TOP_EDGE ) ? column->back() : column->front();
3858 edge = myHelper.GetSubShapeByNode ( node, meshDS );
3859 if ( edge.ShapeType() == TopAbs_VERTEX ) {
3860 column = & ( myParamToColumnMap->begin()->second );
3861 node = ( iEdge == TOP_EDGE ) ? column->back() : column->front();
3866 bool back = ( iEdge == V1_EDGE );
3867 if ( !myIsForward ) back = !back;
3869 column = & ( myParamToColumnMap->rbegin()->second );
3871 column = & ( myParamToColumnMap->begin()->second );
3872 if ( column->size() > 0 )
3873 edge = myHelper.GetSubShapeByNode( (*column)[ 1 ], meshDS );
3874 if ( edge.IsNull() || edge.ShapeType() == TopAbs_VERTEX )
3875 node = column->front();
3880 if ( !edge.IsNull() && edge.ShapeType() == TopAbs_EDGE )
3881 return TopoDS::Edge( edge );
3883 // find edge by 2 vertices
3884 TopoDS_Shape V1 = edge;
3885 TopoDS_Shape V2 = myHelper.GetSubShapeByNode( node, meshDS );
3886 if ( !V2.IsNull() && V2.ShapeType() == TopAbs_VERTEX && !V2.IsSame( V1 ))
3888 TopoDS_Shape ancestor = myHelper.GetCommonAncestor( V1, V2, *myHelper.GetMesh(), TopAbs_EDGE);
3889 if ( !ancestor.IsNull() )
3890 return TopoDS::Edge( ancestor );
3892 return TopoDS_Edge();
3895 //================================================================================
3897 * \brief Fill block sub-shapes
3898 * \param shapeMap - map to fill in
3899 * \retval int - nb inserted sub-shapes
3901 //================================================================================
3903 int StdMeshers_PrismAsBlock::TSideFace::InsertSubShapes(TBlockShapes& shapeMap) const
3908 vector< int > edgeIdVec;
3909 SMESH_Block::GetFaceEdgesIDs( myID, edgeIdVec );
3911 for ( int i = BOTTOM_EDGE; i <=V1_EDGE ; ++i ) {
3912 TopoDS_Edge e = GetEdge( i );
3913 if ( !e.IsNull() ) {
3914 nbInserted += SMESH_Block::Insert( e, edgeIdVec[ i ], shapeMap);
3918 // Insert corner vertices
3920 TParam2ColumnIt col1, col2 ;
3921 vector< int > vertIdVec;
3924 SMESH_Block::GetEdgeVertexIDs( edgeIdVec[ V0_EDGE ], vertIdVec);
3925 GetColumns(0, col1, col2 );
3926 const SMDS_MeshNode* node0 = col1->second.front();
3927 const SMDS_MeshNode* node1 = col1->second.back();
3928 TopoDS_Shape v0 = myHelper.GetSubShapeByNode( node0, myHelper.GetMeshDS());
3929 TopoDS_Shape v1 = myHelper.GetSubShapeByNode( node1, myHelper.GetMeshDS());
3930 if ( v0.ShapeType() == TopAbs_VERTEX ) {
3931 nbInserted += SMESH_Block::Insert( v0, vertIdVec[ 0 ], shapeMap);
3933 if ( v1.ShapeType() == TopAbs_VERTEX ) {
3934 nbInserted += SMESH_Block::Insert( v1, vertIdVec[ 1 ], shapeMap);
3938 SMESH_Block::GetEdgeVertexIDs( edgeIdVec[ V1_EDGE ], vertIdVec);
3939 GetColumns(1, col1, col2 );
3940 node0 = col2->second.front();
3941 node1 = col2->second.back();
3942 v0 = myHelper.GetSubShapeByNode( node0, myHelper.GetMeshDS());
3943 v1 = myHelper.GetSubShapeByNode( node1, myHelper.GetMeshDS());
3944 if ( v0.ShapeType() == TopAbs_VERTEX ) {
3945 nbInserted += SMESH_Block::Insert( v0, vertIdVec[ 0 ], shapeMap);
3947 if ( v1.ShapeType() == TopAbs_VERTEX ) {
3948 nbInserted += SMESH_Block::Insert( v1, vertIdVec[ 1 ], shapeMap);
3951 // TopoDS_Vertex V0, V1, Vcom;
3952 // TopExp::Vertices( myBaseEdge, V0, V1, true );
3953 // if ( !myIsForward ) std::swap( V0, V1 );
3955 // // bottom vertex IDs
3956 // SMESH_Block::GetEdgeVertexIDs( edgeIdVec[ _u0 ], vertIdVec);
3957 // SMESH_Block::Insert( V0, vertIdVec[ 0 ], shapeMap);
3958 // SMESH_Block::Insert( V1, vertIdVec[ 1 ], shapeMap);
3960 // TopoDS_Edge sideEdge = GetEdge( V0_EDGE );
3961 // if ( sideEdge.IsNull() || !TopExp::CommonVertex( botEdge, sideEdge, Vcom ))
3964 // // insert one side edge
3966 // if ( Vcom.IsSame( V0 )) edgeID = edgeIdVec[ _v0 ];
3967 // else edgeID = edgeIdVec[ _v1 ];
3968 // SMESH_Block::Insert( sideEdge, edgeID, shapeMap);
3970 // // top vertex of the side edge
3971 // SMESH_Block::GetEdgeVertexIDs( edgeID, vertIdVec);
3972 // TopoDS_Vertex Vtop = TopExp::FirstVertex( sideEdge );
3973 // if ( Vcom.IsSame( Vtop ))
3974 // Vtop = TopExp::LastVertex( sideEdge );
3975 // SMESH_Block::Insert( Vtop, vertIdVec[ 1 ], shapeMap);
3977 // // other side edge
3978 // sideEdge = GetEdge( V1_EDGE );
3979 // if ( sideEdge.IsNull() )
3981 // if ( edgeID = edgeIdVec[ _v1 ]) edgeID = edgeIdVec[ _v0 ];
3982 // else edgeID = edgeIdVec[ _v1 ];
3983 // SMESH_Block::Insert( sideEdge, edgeID, shapeMap);
3986 // TopoDS_Edge topEdge = GetEdge( TOP_EDGE );
3987 // SMESH_Block::Insert( topEdge, edgeIdVec[ _u1 ], shapeMap);
3989 // // top vertex of the other side edge
3990 // if ( !TopExp::CommonVertex( topEdge, sideEdge, Vcom ))
3992 // SMESH_Block::GetEdgeVertexIDs( edgeID, vertIdVec );
3993 // SMESH_Block::Insert( Vcom, vertIdVec[ 1 ], shapeMap);
3998 //================================================================================
4000 * \brief Dump ids of nodes of sides
4002 //================================================================================
4004 void StdMeshers_PrismAsBlock::TSideFace::dumpNodes(int nbNodes) const
4007 cout << endl << "NODES OF FACE "; SMESH_Block::DumpShapeID( myID, cout ) << endl;
4008 THorizontalEdgeAdaptor* hSize0 = (THorizontalEdgeAdaptor*) HorizCurve(0);
4009 cout << "Horiz side 0: "; hSize0->dumpNodes(nbNodes); cout << endl;
4010 THorizontalEdgeAdaptor* hSize1 = (THorizontalEdgeAdaptor*) HorizCurve(1);
4011 cout << "Horiz side 1: "; hSize1->dumpNodes(nbNodes); cout << endl;
4012 TVerticalEdgeAdaptor* vSide0 = (TVerticalEdgeAdaptor*) VertiCurve(0);
4013 cout << "Verti side 0: "; vSide0->dumpNodes(nbNodes); cout << endl;
4014 TVerticalEdgeAdaptor* vSide1 = (TVerticalEdgeAdaptor*) VertiCurve(1);
4015 cout << "Verti side 1: "; vSide1->dumpNodes(nbNodes); cout << endl;
4016 delete hSize0; delete hSize1; delete vSide0; delete vSide1;
4020 //================================================================================
4022 * \brief Creates TVerticalEdgeAdaptor
4023 * \param columnsMap - node column map
4024 * \param parameter - normalized parameter
4026 //================================================================================
4028 StdMeshers_PrismAsBlock::TVerticalEdgeAdaptor::
4029 TVerticalEdgeAdaptor( const TParam2ColumnMap* columnsMap, const double parameter)
4031 myNodeColumn = & getColumn( columnsMap, parameter )->second;
4034 //================================================================================
4036 * \brief Return coordinates for the given normalized parameter
4037 * \param U - normalized parameter
4038 * \retval gp_Pnt - coordinates
4040 //================================================================================
4042 gp_Pnt StdMeshers_PrismAsBlock::TVerticalEdgeAdaptor::Value(const Standard_Real U) const
4044 const SMDS_MeshNode* n1;
4045 const SMDS_MeshNode* n2;
4046 double r = getRAndNodes( myNodeColumn, U, n1, n2 );
4047 return gpXYZ(n1) * ( 1 - r ) + gpXYZ(n2) * r;
4050 //================================================================================
4052 * \brief Dump ids of nodes
4054 //================================================================================
4056 void StdMeshers_PrismAsBlock::TVerticalEdgeAdaptor::dumpNodes(int nbNodes) const
4059 for ( int i = 0; i < nbNodes && i < myNodeColumn->size(); ++i )
4060 cout << (*myNodeColumn)[i]->GetID() << " ";
4061 if ( nbNodes < myNodeColumn->size() )
4062 cout << myNodeColumn->back()->GetID();
4066 //================================================================================
4068 * \brief Return coordinates for the given normalized parameter
4069 * \param U - normalized parameter
4070 * \retval gp_Pnt - coordinates
4072 //================================================================================
4074 gp_Pnt StdMeshers_PrismAsBlock::THorizontalEdgeAdaptor::Value(const Standard_Real U) const
4076 return mySide->TSideFace::Value( U, myV );
4079 //================================================================================
4081 * \brief Dump ids of <nbNodes> first nodes and the last one
4083 //================================================================================
4085 void StdMeshers_PrismAsBlock::THorizontalEdgeAdaptor::dumpNodes(int nbNodes) const
4088 // Not bedugged code. Last node is sometimes incorrect
4089 const TSideFace* side = mySide;
4091 if ( mySide->IsComplex() )
4092 side = mySide->GetComponent(0,u);
4094 TParam2ColumnIt col, col2;
4095 TParam2ColumnMap* u2cols = side->GetColumns();
4096 side->GetColumns( u , col, col2 );
4098 int j, i = myV ? mySide->ColumnHeight()-1 : 0;
4100 const SMDS_MeshNode* n = 0;
4101 const SMDS_MeshNode* lastN
4102 = side->IsForward() ? u2cols->rbegin()->second[ i ] : u2cols->begin()->second[ i ];
4103 for ( j = 0; j < nbNodes && n != lastN; ++j )
4105 n = col->second[ i ];
4106 cout << n->GetID() << " ";
4107 if ( side->IsForward() )
4115 if ( mySide->IsComplex() )
4116 side = mySide->GetComponent(1,u);
4118 side->GetColumns( u , col, col2 );
4119 if ( n != col->second[ i ] )
4120 cout << col->second[ i ]->GetID();
4124 //================================================================================
4126 * \brief Costructor of TPCurveOnHorFaceAdaptor fills its map of
4127 * normalized parameter to node UV on a horizontal face
4128 * \param [in] sideFace - lateral prism side
4129 * \param [in] isTop - is \a horFace top or bottom of the prism
4130 * \param [in] horFace - top or bottom face of the prism
4132 //================================================================================
4134 StdMeshers_PrismAsBlock::
4135 TPCurveOnHorFaceAdaptor::TPCurveOnHorFaceAdaptor( const TSideFace* sideFace,
4137 const TopoDS_Face& horFace)
4139 if ( sideFace && !horFace.IsNull() )
4141 //cout << "\n\t FACE " << sideFace->FaceID() << endl;
4142 const int Z = isTop ? sideFace->ColumnHeight() - 1 : 0;
4143 map<double, const SMDS_MeshNode* > u2nodes;
4144 sideFace->GetNodesAtZ( Z, u2nodes );
4145 if ( u2nodes.empty() )
4148 SMESH_MesherHelper helper( *sideFace->GetMesh() );
4149 helper.SetSubShape( horFace );
4154 Handle(Geom2d_Curve) C2d;
4156 const double tol = 10 * helper.MaxTolerance( horFace );
4157 const SMDS_MeshNode* prevNode = u2nodes.rbegin()->second;
4159 map<double, const SMDS_MeshNode* >::iterator u2n = u2nodes.begin();
4160 for ( ; u2n != u2nodes.end(); ++u2n )
4162 const SMDS_MeshNode* n = u2n->second;
4164 if ( n->GetPosition()->GetTypeOfPosition() == SMDS_TOP_EDGE )
4166 if ( n->getshapeId() != edgeID )
4169 edgeID = n->getshapeId();
4170 TopoDS_Shape S = helper.GetSubShapeByNode( n, helper.GetMeshDS() );
4171 if ( !S.IsNull() && S.ShapeType() == TopAbs_EDGE )
4173 C2d = BRep_Tool::CurveOnSurface( TopoDS::Edge( S ), horFace, f,l );
4176 if ( !C2d.IsNull() )
4178 double u = static_cast< const SMDS_EdgePosition* >( n->GetPosition() )->GetUParameter();
4179 if ( f <= u && u <= l )
4181 uv = C2d->Value( u ).XY();
4182 okUV = helper.CheckNodeUV( horFace, n, uv, tol );
4187 uv = helper.GetNodeUV( horFace, n, prevNode, &okUV );
4189 myUVmap.insert( myUVmap.end(), make_pair( u2n->first, uv ));
4190 // cout << n->getshapeId() << " N " << n->GetID()
4191 // << " \t" << uv.X() << ", " << uv.Y() << " \t" << u2n->first << endl;
4198 //================================================================================
4200 * \brief Return UV on pcurve for the given normalized parameter
4201 * \param U - normalized parameter
4202 * \retval gp_Pnt - coordinates
4204 //================================================================================
4206 gp_Pnt2d StdMeshers_PrismAsBlock::TPCurveOnHorFaceAdaptor::Value(const Standard_Real U) const
4208 map< double, gp_XY >::const_iterator i1 = myUVmap.upper_bound( U );
4210 if ( i1 == myUVmap.end() )
4211 return myUVmap.empty() ? gp_XY(0,0) : myUVmap.rbegin()->second;
4213 if ( i1 == myUVmap.begin() )
4214 return (*i1).second;
4216 map< double, gp_XY >::const_iterator i2 = i1--;
4218 double r = ( U - i1->first ) / ( i2->first - i1->first );
4219 return i1->second * ( 1 - r ) + i2->second * r;