// 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); } 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 (myFamilies.find(aFamNum) != myFamilies.end()) { myFamilies[aFamNum]->AddElement(aNode); myFamilies[aFamNum]->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; PCellInfo aCellInfo = aMed->GetPCellInfo(aMeshInfo,anEntity,aGeom); EBooleen anIsElemNum = takeNumbers ? aCellInfo->IsElemNum() : eFAUX; TInt aNbElems = aCellInfo->GetNbElem(); if(MYDEBUG) MESSAGE("Perform - anEntity = "<AddElement(anElement); myFamilies[aFamNum]->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::GetGroupNames() { 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::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, 5) == string("Group")) { if (aResGroupNames.find(aName) == aResGroupNames.end()) { aResGroupNames.insert(aName); aResult.push_back(aName); } // } } } 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->MemberOf(aGroupName)) { const set& anElements = aFamily->GetElements(); set::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::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); } } } } } }