// SMESH DriverMED : driver to read and write 'med' files // // Copyright (C) 2003 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN, // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS // // This library is free software; you can redistribute it and/or // modify it under the terms of the GNU Lesser General Public // License as published by the Free Software Foundation; either // version 2.1 of the License. // // This library is distributed in the hope that it will be useful, // but WITHOUT ANY WARRANTY; without even the implied warranty of // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU // Lesser General Public License for more details. // // You should have received a copy of the GNU Lesser General Public // License along with this library; if not, write to the Free Software // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA // // See http://www.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org // // // // File : DriverMED_R_SMESHDS_Mesh.cxx // Module : SMESH #include "DriverMED_R_SMESHDS_Mesh.h" #include "DriverMED_R_SMDS_Mesh.h" #include "SMESHDS_Mesh.hxx" #include "utilities.h" #include "DriverMED_Family.h" #include "SMESHDS_Group.hxx" #include "MED_Factory.hxx" #include "MED_Utilities.hxx" #include #ifdef _DEBUG_ static int MYDEBUG = 0; #else static int MYDEBUG = 0; #endif #define _EDF_NODE_IDS_ using namespace MED; void DriverMED_R_SMESHDS_Mesh::SetMeshName(string theMeshName) { myMeshName = theMeshName; } static const SMDS_MeshNode* FindNode(const SMDS_Mesh* theMesh, TInt theId){ const SMDS_MeshNode* aNode = theMesh->FindNode(theId); if(aNode) return aNode; EXCEPTION(runtime_error,"SMDS_Mesh::FindNode - cannot find a SMDS_MeshNode for ID = "< TFloat GetCoord(MED::PNodeInfo& thePNodeInfo, TInt theElemId){ return thePNodeInfo->GetNodeCoord(theElemId,TheCoordId); } template<> TFloat GetCoord(MED::PNodeInfo& thePNodeInfo, TInt theElemId){ return 0.0; } static TGetCoord aXYZGetCoord[3] = { &GetCoord, &GetCoord, &GetCoord }; static TGetCoord aXYGetCoord[3] = { &GetCoord, &GetCoord, &GetCoord }; static TGetCoord aYZGetCoord[3] = { &GetCoord, &GetCoord, &GetCoord }; static TGetCoord aXZGetCoord[3] = { &GetCoord, &GetCoord, &GetCoord }; static TGetCoord aXGetCoord[3] = { &GetCoord, &GetCoord, &GetCoord }; static TGetCoord aYGetCoord[3] = { &GetCoord, &GetCoord, &GetCoord }; static TGetCoord aZGetCoord[3] = { &GetCoord, &GetCoord, &GetCoord }; class TCoordHelper{ MED::PNodeInfo myPNodeInfo; TGetCoord* myGetCoord; public: TCoordHelper(const MED::PNodeInfo& thePNodeInfo, TGetCoord* theGetCoord): myPNodeInfo(thePNodeInfo), myGetCoord(theGetCoord) {} virtual ~TCoordHelper(){} TFloat GetCoord(TInt theElemId, TInt theCoodId){ return (*myGetCoord[theCoodId])(myPNodeInfo,theElemId); } }; typedef boost::shared_ptr TCoordHelperPtr; Driver_Mesh::Status DriverMED_R_SMESHDS_Mesh::Perform() { Status aResult = DRS_FAIL; try{ myFamilies.clear(); if(MYDEBUG) MESSAGE("Perform - myFile : "<GetNbMeshes()){ for(int iMesh = 0; iMesh < aNbMeshes; iMesh++){ // Reading the MED mesh //--------------------- PMeshInfo aMeshInfo = aMed->GetPMeshInfo(iMesh+1); string aMeshName; if (myMeshId != -1) { ostringstream aMeshNameStr; aMeshNameStr<GetName()); if(aMeshName != aMeshInfo->GetName()) continue; aResult = DRS_OK; //TInt aMeshDim = aMeshInfo->GetDim(); // Reading MED families to the temporary structure //------------------------------------------------ TErr anErr; TInt aNbFams = aMed->GetNbFamilies(aMeshInfo); if(MYDEBUG) MESSAGE("Read " << aNbFams << " families"); for (TInt iFam = 0; iFam < aNbFams; iFam++) { PFamilyInfo aFamilyInfo = aMed->GetPFamilyInfo(aMeshInfo,iFam+1,&anErr); if(anErr >= 0){ TInt aFamId = aFamilyInfo->GetId(); if(MYDEBUG) MESSAGE("Family " << aFamId << " :"); DriverMED_FamilyPtr aFamily (new DriverMED_Family); TInt aNbGrp = aFamilyInfo->GetNbGroup(); if(MYDEBUG) MESSAGE("belong to " << aNbGrp << " groups"); for (TInt iGr = 0; iGr < aNbGrp; iGr++) { string aGroupName = aFamilyInfo->GetGroupName(iGr); if(MYDEBUG) MESSAGE(aGroupName); aFamily->AddGroupName(aGroupName); } aFamily->SetId( aFamId ); myFamilies[aFamId] = aFamily; } } // Reading MED nodes to the corresponding SMDS structure //------------------------------------------------------ PNodeInfo aNodeInfo = aMed->GetPNodeInfo(aMeshInfo); TCoordHelperPtr aCoordHelperPtr; { TInt aMeshDimension = aMeshInfo->GetDim(); bool anIsDimPresent[3] = {false, false, false}; for(TInt iDim = 0; iDim < aMeshDimension; iDim++){ string aDimName = aNodeInfo->GetCoordName(iDim); if(aDimName == "x" || aDimName == "X") anIsDimPresent[eX] = true; else if(aDimName == "y" || aDimName == "Y") anIsDimPresent[eY] = true; else if(aDimName == "z" || aDimName == "Z") anIsDimPresent[eZ] = true; } switch(aMeshDimension){ case 3: aCoordHelperPtr.reset(new TCoordHelper(aNodeInfo,aXYZGetCoord)); break; case 2: if(anIsDimPresent[eY] && anIsDimPresent[eZ]) aCoordHelperPtr.reset(new TCoordHelper(aNodeInfo,aYZGetCoord)); else if(anIsDimPresent[eX] && anIsDimPresent[eZ]) aCoordHelperPtr.reset(new TCoordHelper(aNodeInfo,aXZGetCoord)); else aCoordHelperPtr.reset(new TCoordHelper(aNodeInfo,aXYGetCoord)); break; case 1: if(anIsDimPresent[eY]) aCoordHelperPtr.reset(new TCoordHelper(aNodeInfo,aYGetCoord)); else if(anIsDimPresent[eZ]) aCoordHelperPtr.reset(new TCoordHelper(aNodeInfo,aZGetCoord)); else aCoordHelperPtr.reset(new TCoordHelper(aNodeInfo,aXGetCoord)); break; } } EBooleen anIsNodeNum = aNodeInfo->IsElemNum(); TInt aNbElems = aNodeInfo->GetNbElem(); if(MYDEBUG) MESSAGE("Perform - aNodeInfo->GetNbElem() = "<X()<<", "<Y()<<", "<Z()<GetFamNum(iElem); if ( checkFamilyID ( aFamily, aFamNum )) { aFamily->AddElement(aNode); aFamily->SetType(SMDSAbs_Node); } } // Reading pre information about all MED cells //-------------------------------------------- bool takeNumbers = true; // initially we trust the numbers from file MED::TEntityInfo aEntityInfo = aMed->GetEntityInfo(aMeshInfo); MED::TEntityInfo::iterator anEntityIter = aEntityInfo.begin(); for(; anEntityIter != aEntityInfo.end(); anEntityIter++){ const EEntiteMaillage& anEntity = anEntityIter->first; if(anEntity == eNOEUD) continue; // Reading MED cells to the corresponding SMDS structure //------------------------------------------------------ const MED::TGeom& aTGeom = anEntityIter->second; MED::TGeom::const_iterator anTGeomIter = aTGeom.begin(); for(; anTGeomIter != aTGeom.end(); anTGeomIter++){ const EGeometrieElement& aGeom = anTGeomIter->first; if (aGeom == ePOINT1) { continue; } else if (aGeom == ePOLYGONE) { PPolygoneInfo aPolygoneInfo = aMed->GetPPolygoneInfo(aMeshInfo,anEntity,aGeom); EBooleen anIsElemNum = takeNumbers ? aPolygoneInfo->IsElemNum() : eFAUX; TElemNum aConn = aPolygoneInfo->GetConnectivite(); TElemNum aIndex = aPolygoneInfo->GetIndex(); TInt nbPolygons = aPolygoneInfo->GetNbElem(); for (TInt iPG = 0; iPG < nbPolygons; iPG++) { // get nodes TInt aCurrPG_FirstNodeIndex = aIndex[iPG] - 1; int nbNodes = aPolygoneInfo->GetNbConn(iPG); std::vector nodes_ids (nbNodes); //for (TInt inode = 0; inode < nbNodes; inode++) { // nodes_ids[inode] = aConn[aCurrPG_FirstNodeIndex + inode]; //} #ifdef _EDF_NODE_IDS_ if (anIsNodeNum) { for (TInt inode = 0; inode < nbNodes; inode++) { nodes_ids[inode] = aNodeInfo->GetElemNum(aConn[aCurrPG_FirstNodeIndex + inode] - 1); } } else { for (TInt inode = 0; inode < nbNodes; inode++) { nodes_ids[inode] = aConn[aCurrPG_FirstNodeIndex + inode]; } } #else for (TInt inode = 0; inode < nbNodes; inode++) { nodes_ids[inode] = aConn[aCurrPG_FirstNodeIndex + inode]; } #endif bool isRenum = false; SMDS_MeshElement* anElement = NULL; TInt aFamNum = aPolygoneInfo->GetFamNum(iPG); try { if (anIsElemNum) { anElement = myMesh->AddPolygonalFaceWithID (nodes_ids, aPolygoneInfo->GetElemNum(iPG)); } if (!anElement) { std::vector nodes (nbNodes); for (int inode = 0; inode < nbNodes; inode++) { nodes[inode] = FindNode(myMesh, nodes_ids[inode]); } anElement = myMesh->AddPolygonalFace(nodes); isRenum = anIsElemNum; } } catch (const std::exception& exc) { aResult = DRS_FAIL; } catch (...) { aResult = DRS_FAIL; } if (!anElement) { aResult = DRS_WARN_SKIP_ELEM; } else { if (isRenum) { anIsElemNum = eFAUX; takeNumbers = false; if (aResult < DRS_WARN_RENUMBER) aResult = DRS_WARN_RENUMBER; } if ( checkFamilyID ( aFamily, aFamNum )) { // Save reference to this element from its family aFamily->AddElement(anElement); aFamily->SetType(anElement->GetType()); } } } // for (TInt iPG = 0; iPG < nbPolygons; iPG++) continue; } else if (aGeom == ePOLYEDRE) { PPolyedreInfo aPolyedreInfo = aMed->GetPPolyedreInfo(aMeshInfo,anEntity,aGeom); EBooleen anIsElemNum = takeNumbers ? aPolyedreInfo->IsElemNum() : eFAUX; TElemNum aConn = aPolyedreInfo->GetConnectivite(); TElemNum aFacesIndex = aPolyedreInfo->GetFacesIndex(); TElemNum aIndex = aPolyedreInfo->GetIndex(); TInt nbPolyedres = aPolyedreInfo->GetNbElem(); for (int iPE = 0; iPE < nbPolyedres; iPE++) { // get faces int aCurrPE_FirstFaceIndex = aIndex[iPE] - 1; int aNextPE_FirstFaceIndex = aIndex[iPE + 1] - 1; int nbFaces = aNextPE_FirstFaceIndex - aCurrPE_FirstFaceIndex; std::vector quantities (nbFaces); for (int iFa = 0; iFa < nbFaces; iFa++) { int aCurrFace_FirstNodeIndex = aFacesIndex[aCurrPE_FirstFaceIndex + iFa] - 1; int aNextFace_FirstNodeIndex = aFacesIndex[aCurrPE_FirstFaceIndex + iFa + 1] - 1; int nbNodes = aNextFace_FirstNodeIndex - aCurrFace_FirstNodeIndex; quantities[iFa] = nbNodes; } // get nodes int aCurrPE_FirstNodeIndex = aFacesIndex[aCurrPE_FirstFaceIndex] - 1; int nbPENodes = aPolyedreInfo->GetNbConn(iPE); std::vector nodes_ids (nbPENodes); //for (int inode = 0; inode < nbPENodes; inode++) { // nodes_ids[inode] = aConn[aCurrPE_FirstNodeIndex + inode]; //} #ifdef _EDF_NODE_IDS_ if (anIsNodeNum) { for (int inode = 0; inode < nbPENodes; inode++) { nodes_ids[inode] = aNodeInfo->GetElemNum(aConn[aCurrPE_FirstNodeIndex + inode] - 1); } } else { for (int inode = 0; inode < nbPENodes; inode++) { nodes_ids[inode] = aConn[aCurrPE_FirstNodeIndex + inode]; } } #else for (int inode = 0; inode < nbPENodes; inode++) { nodes_ids[inode] = aConn[aCurrPE_FirstNodeIndex + inode]; } #endif bool isRenum = false; SMDS_MeshElement* anElement = NULL; TInt aFamNum = aPolyedreInfo->GetFamNum(iPE); try { if (anIsElemNum) { anElement = myMesh->AddPolyhedralVolumeWithID (nodes_ids, quantities, aPolyedreInfo->GetElemNum(iPE)); } if (!anElement) { std::vector nodes (nbPENodes); for (int inode = 0; inode < nbPENodes; inode++) { nodes[inode] = FindNode(myMesh, nodes_ids[inode]); } anElement = myMesh->AddPolyhedralVolume(nodes, quantities); isRenum = anIsElemNum; } } catch (const std::exception& exc) { aResult = DRS_FAIL; } catch (...) { aResult = DRS_FAIL; } if (!anElement) { aResult = DRS_WARN_SKIP_ELEM; } else { if (isRenum) { anIsElemNum = eFAUX; takeNumbers = false; if (aResult < DRS_WARN_RENUMBER) aResult = DRS_WARN_RENUMBER; } if ( checkFamilyID ( aFamily, aFamNum )) { // Save reference to this element from its family aFamily->AddElement(anElement); aFamily->SetType(anElement->GetType()); } } } // for (int iPE = 0; iPE < nbPolyedres; iPE++) continue; } else { } PCellInfo aCellInfo = aMed->GetPCellInfo(aMeshInfo,anEntity,aGeom); EBooleen anIsElemNum = takeNumbers ? aCellInfo->IsElemNum() : eFAUX; TInt aNbElems = aCellInfo->GetNbElem(); if(MYDEBUG) MESSAGE("Perform - anEntity = "<GetFamNum(iElem); try{ //MESSAGE("Try to create element # " << iElem << " with id = " // << aCellInfo->GetElemNum(iElem)); switch(aGeom){ case eSEG2: case eSEG3: if(anIsElemNum) anElement = myMesh->AddEdgeWithID(aNodeIds[0], aNodeIds[1], aCellInfo->GetElemNum(iElem)); if (!anElement) { anElement = myMesh->AddEdge(FindNode(myMesh,aNodeIds[0]), FindNode(myMesh,aNodeIds[1])); isRenum = anIsElemNum; } break; case eTRIA3: case eTRIA6: aNbNodes = 3; if(anIsElemNum) anElement = myMesh->AddFaceWithID(aNodeIds[0], aNodeIds[1], aNodeIds[2], aCellInfo->GetElemNum(iElem)); if (!anElement) { anElement = myMesh->AddFace(FindNode(myMesh,aNodeIds[0]), FindNode(myMesh,aNodeIds[1]), FindNode(myMesh,aNodeIds[2])); isRenum = anIsElemNum; } break; case eQUAD4: case eQUAD8: aNbNodes = 4; // There is some differnce between SMDS and MED if(anIsElemNum) anElement = myMesh->AddFaceWithID(aNodeIds[0], aNodeIds[1], aNodeIds[2], aNodeIds[3], aCellInfo->GetElemNum(iElem)); if (!anElement) { anElement = myMesh->AddFace(FindNode(myMesh,aNodeIds[0]), FindNode(myMesh,aNodeIds[1]), FindNode(myMesh,aNodeIds[2]), FindNode(myMesh,aNodeIds[3])); isRenum = anIsElemNum; } break; case eTETRA4: case eTETRA10: aNbNodes = 4; if(anIsElemNum) anElement = myMesh->AddVolumeWithID(aNodeIds[0], aNodeIds[1], aNodeIds[2], aNodeIds[3], aCellInfo->GetElemNum(iElem)); if (!anElement) { anElement = myMesh->AddVolume(FindNode(myMesh,aNodeIds[0]), FindNode(myMesh,aNodeIds[1]), FindNode(myMesh,aNodeIds[2]), FindNode(myMesh,aNodeIds[3])); isRenum = anIsElemNum; } break; case ePYRA5: case ePYRA13: aNbNodes = 5; // There is some differnce between SMDS and MED if(anIsElemNum) anElement = myMesh->AddVolumeWithID(aNodeIds[0], aNodeIds[1], aNodeIds[2], aNodeIds[3], aNodeIds[4], aCellInfo->GetElemNum(iElem)); if (!anElement) { anElement = myMesh->AddVolume(FindNode(myMesh,aNodeIds[0]), FindNode(myMesh,aNodeIds[1]), FindNode(myMesh,aNodeIds[2]), FindNode(myMesh,aNodeIds[3]), FindNode(myMesh,aNodeIds[4])); isRenum = anIsElemNum; } break; case ePENTA6: case ePENTA15: aNbNodes = 6; if(anIsElemNum) anElement = myMesh->AddVolumeWithID(aNodeIds[0], aNodeIds[1], aNodeIds[2], aNodeIds[3], aNodeIds[4], aNodeIds[5], aCellInfo->GetElemNum(iElem)); if (!anElement) { anElement = myMesh->AddVolume(FindNode(myMesh,aNodeIds[0]), FindNode(myMesh,aNodeIds[1]), FindNode(myMesh,aNodeIds[2]), FindNode(myMesh,aNodeIds[3]), FindNode(myMesh,aNodeIds[4]), FindNode(myMesh,aNodeIds[5])); isRenum = anIsElemNum; } break; case eHEXA8: case eHEXA20: aNbNodes = 8; if(anIsElemNum) anElement = myMesh->AddVolumeWithID(aNodeIds[0], aNodeIds[1], aNodeIds[2], aNodeIds[3], aNodeIds[4], aNodeIds[5], aNodeIds[6], aNodeIds[7], aCellInfo->GetElemNum(iElem)); if (!anElement) { anElement = myMesh->AddVolume(FindNode(myMesh,aNodeIds[0]), FindNode(myMesh,aNodeIds[1]), FindNode(myMesh,aNodeIds[2]), FindNode(myMesh,aNodeIds[3]), FindNode(myMesh,aNodeIds[4]), FindNode(myMesh,aNodeIds[5]), FindNode(myMesh,aNodeIds[6]), FindNode(myMesh,aNodeIds[7])); isRenum = anIsElemNum; } break; } }catch(const std::exception& exc){ //INFOS("Follow exception was cought:\n\t"<AddElement(anElement); aFamily->SetType(anElement->GetType()); } } } } } break; } } }catch(const std::exception& exc){ INFOS("Follow exception was cought:\n\t"<GetNbMeshes()) { for (int iMesh = 0; iMesh < aNbMeshes; iMesh++) { // Reading the MED mesh //--------------------- PMeshInfo aMeshInfo = aMed->GetPMeshInfo(iMesh+1); aMeshNames.push_back(aMeshInfo->GetName()); } } }catch(const std::exception& exc){ INFOS("Follow exception was cought:\n\t"< DriverMED_R_SMESHDS_Mesh::GetGroupNamesAndTypes() { list aResult; set aResGroupNames; map::iterator aFamsIter = myFamilies.begin(); for (; aFamsIter != myFamilies.end(); aFamsIter++) { DriverMED_FamilyPtr aFamily = (*aFamsIter).second; const MED::TStringSet& aGroupNames = aFamily->GetGroupNames(); set::const_iterator aGrNamesIter = aGroupNames.begin(); for (; aGrNamesIter != aGroupNames.end(); aGrNamesIter++) { TNameAndType aNameAndType = make_pair( *aGrNamesIter, aFamily->GetType() ); // Check, if this is a Group or SubMesh name //if (aName.substr(0, 5) == string("Group")) { if ( aResGroupNames.insert( aNameAndType ).second ) { aResult.push_back( aNameAndType ); } // } } } return aResult; } void DriverMED_R_SMESHDS_Mesh::GetGroup(SMESHDS_Group* theGroup) { string aGroupName (theGroup->GetStoreName()); if(MYDEBUG) MESSAGE("Get Group " << aGroupName); map::iterator aFamsIter = myFamilies.begin(); for (; aFamsIter != myFamilies.end(); aFamsIter++) { DriverMED_FamilyPtr aFamily = (*aFamsIter).second; if (aFamily->GetType() == theGroup->GetType() && aFamily->MemberOf(aGroupName)) { const set& anElements = aFamily->GetElements(); set::const_iterator anElemsIter = anElements.begin(); const SMDS_MeshElement * element = 0; for (; anElemsIter != anElements.end(); anElemsIter++) { element = *anElemsIter; theGroup->SMDSGroup().Add(element); } if ( element ) theGroup->SetType( element->GetType() ); } } } void DriverMED_R_SMESHDS_Mesh::GetSubMesh (SMESHDS_SubMesh* theSubMesh, const int theId) { char submeshGrpName[ 30 ]; sprintf( submeshGrpName, "SubMesh %d", theId ); string aName (submeshGrpName); map::iterator aFamsIter = myFamilies.begin(); for (; aFamsIter != myFamilies.end(); aFamsIter++) { DriverMED_FamilyPtr aFamily = (*aFamsIter).second; if (aFamily->MemberOf(aName)) { const set& anElements = aFamily->GetElements(); set::const_iterator anElemsIter = anElements.begin(); if (aFamily->GetType() == SMDSAbs_Node) { for (; anElemsIter != anElements.end(); anElemsIter++) { const SMDS_MeshNode* node = static_cast(*anElemsIter); theSubMesh->AddNode(node); } } else { for (; anElemsIter != anElements.end(); anElemsIter++) { theSubMesh->AddElement(*anElemsIter); } } } } } void DriverMED_R_SMESHDS_Mesh::CreateAllSubMeshes () { map::iterator aFamsIter = myFamilies.begin(); for (; aFamsIter != myFamilies.end(); aFamsIter++) { DriverMED_FamilyPtr aFamily = (*aFamsIter).second; MED::TStringSet aGroupNames = aFamily->GetGroupNames(); set::iterator aGrNamesIter = aGroupNames.begin(); for (; aGrNamesIter != aGroupNames.end(); aGrNamesIter++) { string aName = *aGrNamesIter; // Check, if this is a Group or SubMesh name if (aName.substr(0, 7) == string("SubMesh")) { int Id = atoi(string(aName).substr(7).c_str()); set anElements = aFamily->GetElements(); set::iterator anElemsIter = anElements.begin(); if (aFamily->GetType() == SMDSAbs_Node) { for (; anElemsIter != anElements.end(); anElemsIter++) { SMDS_MeshNode* node = const_cast ( static_cast( *anElemsIter )); // find out a shape type TopoDS_Shape aShape = myMesh->IndexToShape( Id ); int aShapeType = ( aShape.IsNull() ? -1 : aShape.ShapeType() ); switch ( aShapeType ) { case TopAbs_FACE: myMesh->SetNodeOnFace(node, Id); break; case TopAbs_EDGE: myMesh->SetNodeOnEdge(node, Id); break; case TopAbs_VERTEX: myMesh->SetNodeOnVertex(node, Id); break; default: myMesh->SetNodeInVolume(node, Id); } } } else { for (; anElemsIter != anElements.end(); anElemsIter++) { myMesh->SetMeshElementOnShape(*anElemsIter, Id); } } } } } } /*! * \brief Ensure aFamily to have required ID * \param aFamily - a family to check and update * \param anID - an ID aFamily should have * \retval bool - true if successful */ bool DriverMED_R_SMESHDS_Mesh::checkFamilyID(DriverMED_FamilyPtr & aFamily, int anID) const { if ( !aFamily || aFamily->GetId() != anID ) { map::const_iterator i_fam = myFamilies.find(anID); if ( i_fam == myFamilies.end() ) return false; aFamily = i_fam->second; } return ( aFamily->GetId() == anID ); }