From abfa12f3fce68392cf28966909f2732c8a17b086 Mon Sep 17 00:00:00 2001 From: eap Date: Thu, 4 May 2006 09:28:59 +0000 Subject: [PATCH] Implement ConvertFromQuadratic() --- src/SMESH/SMESH_MeshEditor.cxx | 144 ++++++++++++++++++++++++++--- src/SMESH/SMESH_MeshEditor.hxx | 8 +- src/SMESH_I/SMESH_MeshEditor_i.cxx | 16 +++- src/SMESH_I/SMESH_MeshEditor_i.hxx | 1 + 4 files changed, 155 insertions(+), 14 deletions(-) diff --git a/src/SMESH/SMESH_MeshEditor.cxx b/src/SMESH/SMESH_MeshEditor.cxx index ca4d2c72c..731cbfdf5 100644 --- a/src/SMESH/SMESH_MeshEditor.cxx +++ b/src/SMESH/SMESH_MeshEditor.cxx @@ -5858,14 +5858,14 @@ void SMESH_MeshEditor::ConvertElemToQuadratic(SMESHDS_SubMesh *theSm, } SMDSAbs_ElementType aType = elem->GetType(); + const SMDS_MeshElement* NewElem = 0; switch( aType ) { case SMDSAbs_Edge : { meshDS->RemoveFreeElement(elem, theSm); - const SMDS_QuadraticEdge* NewEdge = theHelper->AddQuadraticEdge(aNds[0], aNds[1], id, theForce3d); - AddToSameGroups(NewEdge, elem, meshDS); + NewElem = theHelper->AddQuadraticEdge(aNds[0], aNds[1], id, theForce3d); break; } case SMDSAbs_Face : @@ -5873,19 +5873,17 @@ void SMESH_MeshEditor::ConvertElemToQuadratic(SMESHDS_SubMesh *theSm, if(elem->IsQuadratic()) continue; meshDS->RemoveFreeElement(elem, theSm); - SMDS_MeshFace * NewFace = 0; switch(nbNodes) { case 3: - NewFace = theHelper->AddFace(aNds[0], aNds[1], aNds[2], id, theForce3d); + NewElem = theHelper->AddFace(aNds[0], aNds[1], aNds[2], id, theForce3d); break; case 4: - NewFace = theHelper->AddFace(aNds[0], aNds[1], aNds[2], aNds[3], id, theForce3d); + NewElem = theHelper->AddFace(aNds[0], aNds[1], aNds[2], aNds[3], id, theForce3d); break; default: continue; } - AddToSameGroups(NewFace, elem, meshDS); break; } case SMDSAbs_Volume : @@ -5893,28 +5891,31 @@ void SMESH_MeshEditor::ConvertElemToQuadratic(SMESHDS_SubMesh *theSm, if( elem->IsQuadratic() ) continue; meshDS->RemoveFreeElement(elem, theSm); - SMDS_MeshVolume * NewVolume = 0; switch(nbNodes) { case 4: - NewVolume = theHelper->AddVolume(aNds[0], aNds[1], aNds[2], aNds[3], id, true); + NewElem = theHelper->AddVolume(aNds[0], aNds[1], aNds[2], aNds[3], id, true); break; case 6: - NewVolume = theHelper->AddVolume(aNds[0], aNds[1], aNds[2], aNds[3], aNds[4], aNds[5], id, true); + NewElem = theHelper->AddVolume(aNds[0], aNds[1], aNds[2], aNds[3], aNds[4], aNds[5], id, true); break; case 8: - NewVolume = theHelper->AddVolume(aNds[0], aNds[1], aNds[2], aNds[3], - aNds[4], aNds[5], aNds[6], aNds[7], id, true); + NewElem = theHelper->AddVolume(aNds[0], aNds[1], aNds[2], aNds[3], + aNds[4], aNds[5], aNds[6], aNds[7], id, true); break; default: continue; } - AddToSameGroups(NewVolume, elem, meshDS); break; } default : continue; } + if( NewElem ) + { + AddToSameGroups( NewElem, elem, meshDS); + theSm->AddElement( NewElem ); + } } } @@ -6037,12 +6038,131 @@ void SMESH_MeshEditor::ConvertToQuadratic(const bool theForce3d) delete aHelper; } +//======================================================================= +//function : RemoveQuadElem +//purpose : +//======================================================================= +void SMESH_MeshEditor::RemoveQuadElem(SMESHDS_SubMesh *theSm, + SMDS_ElemIteratorPtr theItr) +{ + SMESHDS_Mesh* meshDS = GetMeshDS(); + while( theItr->more() ) + { + const SMDS_MeshElement* elem = theItr->next(); + if( elem ) + { + if( !elem->IsQuadratic() ) + continue; + + int id = elem->GetID(); + + int nbNodes = elem->NbNodes(), idx = 0; + vector aNds; + vector aQuadNds; + + //remove all quadratic nodes + for(int i = 0; i < nbNodes; i++) + { + const SMDS_MeshNode* n = elem->GetNode(i); + + if( elem->IsMediumNode( n ) ) + { + aQuadNds.push_back( n ); + /* ItRemoveQuadNodeMap itRNM = myRemoveNodeMap.find( n ); + if( itRNM == myRemoveNodeMap.end() ) + { + aQuadNds.push_back( n ); + myRemoveNodeMap.insert(RemoveQuadNodeMap::value_type( n,theSm )); + }*/ + } + else + aNds.push_back( n ); + } + + idx = aNds.size(); + if( !idx ) continue; + SMDSAbs_ElementType aType = elem->GetType(); + + //remove old quadratic elements + meshDS->RemoveFreeElement( elem, theSm ); + + for( int j = 0; j < aQuadNds.size(); j++ ) + { + meshDS->RemoveFreeNode( aQuadNds[j], theSm ); + } + SMDS_MeshElement * NewElem = 0; + switch(aType) + { + case SMDSAbs_Edge: + NewElem = meshDS->AddEdgeWithID( aNds[0], aNds[1] ,id ); + break; + case SMDSAbs_Face: + if( idx==3 ) NewElem = meshDS->AddFaceWithID( aNds[0], + aNds[1], aNds[2], id ); + if( idx==4 ) NewElem = meshDS->AddFaceWithID( aNds[0], + aNds[1], aNds[2], aNds[3],id ); + break; + case SMDSAbs_Volume: + if( idx==4 ) NewElem = meshDS->AddVolumeWithID( aNds[0], + aNds[1], aNds[2], aNds[3], id ); + if( idx==6 ) NewElem = meshDS->AddVolumeWithID( aNds[0], + aNds[1], aNds[2], aNds[3], + aNds[4], aNds[5], id ); + if( idx==8 ) NewElem = meshDS->AddVolumeWithID(aNds[0], + aNds[1], aNds[2], aNds[3], + aNds[4], aNds[5], aNds[6], + aNds[7] ,id ); + break; + default: + break; + } + + AddToSameGroups(NewElem, elem, meshDS); + if( theSm ) + theSm->AddElement( NewElem ); + } + } +} //======================================================================= //function : ConvertFromQuadratic //purpose : //======================================================================= bool SMESH_MeshEditor::ConvertFromQuadratic() { + SMESHDS_Mesh* meshDS = GetMeshDS(); + // myRemoveNodeMap.clear(); + + const TopoDS_Shape& aShape = meshDS->ShapeToMesh(); + + if ( !aShape.IsNull() && GetMesh()->GetSubMeshContaining(aShape) ) + { + SMESH_subMesh *aSubMesh = GetMesh()->GetSubMeshContaining(aShape); + + const map < int, SMESH_subMesh * >& aMapSM = aSubMesh->DependsOn(); + map < int, SMESH_subMesh * >::const_iterator itsub; + for (itsub = aMapSM.begin(); itsub != aMapSM.end(); itsub++) + { + SMESHDS_SubMesh *sm = ((*itsub).second)->GetSubMeshDS(); + if( sm ) + RemoveQuadElem( sm, sm->GetElements() ); + } + SMESHDS_SubMesh *Sm = aSubMesh->GetSubMeshDS(); + if( Sm ) + RemoveQuadElem( Sm, Sm->GetElements() ); + } + else + { + SMESHDS_SubMesh *aSM = 0; + RemoveQuadElem( aSM, meshDS->elementsIterator() ); + } + + /* ItRemoveQuadNodeMap itRNM = myRemoveNodeMap.begin(); + for ( ; itRNM != myRemoveNodeMap.end(); itRNM++ ) { + { + meshDS->RemoveFreeNode( (*itRNM), (*itRNM).second ); + }*/ + + return true; } //======================================================================= diff --git a/src/SMESH/SMESH_MeshEditor.hxx b/src/SMESH/SMESH_MeshEditor.hxx index bd284c574..c95ded87b 100644 --- a/src/SMESH/SMESH_MeshEditor.hxx +++ b/src/SMESH/SMESH_MeshEditor.hxx @@ -394,6 +394,7 @@ class SMESH_MeshEditor { SMESH_Mesh * GetMesh() { return myMesh; } SMESHDS_Mesh * GetMeshDS() { return myMesh->GetMeshDS(); } + private: void ConvertElemToQuadratic(SMESHDS_SubMesh *theSm, @@ -402,7 +403,12 @@ private: //Auxiliary function for "ConvertToQuadratic" is intended to convert //elements contained in submesh to quadratic - private: + void RemoveQuadElem( SMESHDS_SubMesh *theSm, + SMDS_ElemIteratorPtr theItr); + //Auxiliary function for "ConvertFromQuadratic" is intended to convert quadratic + //element to ordinary and for removing quadratic nodes + +private: SMESH_Mesh * myMesh; diff --git a/src/SMESH_I/SMESH_MeshEditor_i.cxx b/src/SMESH_I/SMESH_MeshEditor_i.cxx index d85ee6295..f17fbdd21 100644 --- a/src/SMESH_I/SMESH_MeshEditor_i.cxx +++ b/src/SMESH_I/SMESH_MeshEditor_i.cxx @@ -1743,5 +1743,19 @@ void SMESH_MeshEditor_i::ConvertToQuadratic(CORBA::Boolean theForce3d) anEditor.ConvertToQuadratic(theForce3d); // Update Python script TPythonDump() << this << ".ConvertToQuadratic( " - << theForce3d << ") "; + << theForce3d << " )"; +} + +//======================================================================= +//function : ConvertFromQuadratic +//purpose : +//======================================================================= + +CORBA::Boolean SMESH_MeshEditor_i::ConvertFromQuadratic() +{ + ::SMESH_MeshEditor anEditor( _myMesh ); + CORBA::Boolean isDone = anEditor.ConvertFromQuadratic(); + // Update Python script + TPythonDump() << this << ".ConvertFromQuadratic( )"; + return isDone; } diff --git a/src/SMESH_I/SMESH_MeshEditor_i.hxx b/src/SMESH_I/SMESH_MeshEditor_i.hxx index f5f12943c..4b50c66cf 100644 --- a/src/SMESH_I/SMESH_MeshEditor_i.hxx +++ b/src/SMESH_I/SMESH_MeshEditor_i.hxx @@ -118,6 +118,7 @@ class SMESH_MeshEditor_i: public POA_SMESH::SMESH_MeshEditor void ConvertToQuadratic(CORBA::Boolean Force3d); + CORBA::Boolean ConvertFromQuadratic(); void RenumberNodes(); void RenumberElements(); -- 2.39.2