X-Git-Url: http://git.salome-platform.org/gitweb/?p=modules%2Fsmesh.git;a=blobdiff_plain;f=src%2FDriverMED%2FDriverMED_R_SMESHDS_Mesh.cxx;h=8388327d91af7d64048b6ee04b50f72f003a7606;hp=ada67b7bacfd79aee9f436fbfb9ee0c45d1c6eda;hb=87fd8ca50b1fe004fc046c66c5b5df6199861981;hpb=8b4ba0cdff7660465a9ea7b26ef0df0757afed97 diff --git a/src/DriverMED/DriverMED_R_SMESHDS_Mesh.cxx b/src/DriverMED/DriverMED_R_SMESHDS_Mesh.cxx index ada67b7ba..8388327d9 100644 --- a/src/DriverMED/DriverMED_R_SMESHDS_Mesh.cxx +++ b/src/DriverMED/DriverMED_R_SMESHDS_Mesh.cxx @@ -1,545 +1,869 @@ -using namespace std; +// 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_R_SMESHDS_Mesh::DriverMED_R_SMESHDS_Mesh() { - myFileId=-1; -} - -DriverMED_R_SMESHDS_Mesh::~DriverMED_R_SMESHDS_Mesh() { -; -} +#include "DriverMED_Family.h" -void DriverMED_R_SMESHDS_Mesh::SetMesh(Handle(SMDS_Mesh)& aMesh) { - //myMesh = Handle(SMESHDS_Mesh)::DownCast(aMesh); - myMesh = aMesh; -} +#include "SMESHDS_Group.hxx" -void DriverMED_R_SMESHDS_Mesh::SetFile(string aFile) { - myFile = aFile; -} +#include "MED_Factory.hxx" +#include "MED_Utilities.hxx" -void DriverMED_R_SMESHDS_Mesh::SetFileId(med_idt aFileId) { - myFileId = aFileId; -} - -void DriverMED_R_SMESHDS_Mesh::SetMeshId(int aMeshId) { - myMeshId = aMeshId; -} - -void DriverMED_R_SMESHDS_Mesh::Read() { - - string myClass = string("SMDS_Mesh"); - string myExtension = string("MED"); +#include - DriverMED_R_SMDS_Mesh* myReader = new DriverMED_R_SMDS_Mesh; +#ifdef _DEBUG_ +static int MYDEBUG = 0; +#else +static int MYDEBUG = 0; +#endif - myReader->SetMesh(myMesh); - myReader->SetMeshId(myMeshId); - myReader->SetFile(myFile); - myReader->SetFileId(-1); +#define _EDF_NODE_IDS_ - myReader->Read(); +using namespace MED; +void DriverMED_R_SMESHDS_Mesh::SetMeshName(string theMeshName) +{ + myMeshName = theMeshName; } -void DriverMED_R_SMESHDS_Mesh::Add() { - - string myClass = string("SMDS_Mesh"); - string myExtension = string("MED"); - - DriverMED_R_SMDS_Mesh* myReader = new DriverMED_R_SMDS_Mesh; +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 = "<SetMesh(myMesh); - myReader->SetMeshId(myMeshId); - SCRUTE(myFileId); - myReader->SetFileId(myFileId); +enum ECoordName{eX, eY, eZ, eNone}; +typedef TFloat (*TGetCoord)(MED::PNodeInfo&, TInt); - myReader->Read(); +template +TFloat GetCoord(MED::PNodeInfo& thePNodeInfo, TInt theElemId){ + return thePNodeInfo->GetNodeCoord(theElemId,TheCoordId); +} +template<> +TFloat GetCoord(MED::PNodeInfo& thePNodeInfo, TInt theElemId){ + return 0.0; } -void DriverMED_R_SMESHDS_Mesh::ReadMySelf() { - - med_err ret = 0; - int i,j,k,l; - int numero; - char message[200]; - Standard_Boolean ok; - /* nombre d'objets MED */ - char nom_universel[MED_TAILLE_LNOM+1]; - med_int long_fichier_en_tete; - char *fichier_en_tete; - char version_hdf[10]; - char version_med[10]; - med_int nmaa,mdim,nnoe; - med_int nmai[MED_NBR_GEOMETRIE_MAILLE],nfac[MED_NBR_GEOMETRIE_FACE]; - med_int nare[MED_NBR_GEOMETRIE_ARETE]; - /* nom du maillage */ - char nommaa[MED_TAILLE_NOM+1]; - /* noeuds */ - med_float *coo; - char nomcoo[3*MED_TAILLE_PNOM+1]; - char unicoo[3*MED_TAILLE_PNOM+1]; - char *nomnoe; - med_int *numnoe; - med_int *nufano; - med_repere rep; - med_booleen inonoe,inunoe; - med_mode_switch mode_coo; - char str[MED_TAILLE_PNOM+1]; - /* elements */ - med_int nsup; - med_int edim; - med_int taille; - med_int elem_id; - med_int cmpt = 0; - med_int *connectivite; - char *nomele; - med_int *numele; - med_int *nufael; - med_booleen inoele, inuele; - med_connectivite typ_con; - med_geometrie_element typgeo; - med_geometrie_element typmai[MED_NBR_GEOMETRIE_MAILLE] = {MED_POINT1,MED_SEG2, - MED_SEG3,MED_TRIA3, - MED_TRIA6,MED_QUAD4, - MED_QUAD8,MED_TETRA4, - MED_TETRA10,MED_HEXA8, - MED_HEXA20,MED_PENTA6, - MED_PENTA15,MED_PYRA5, - MED_PYRA13}; - med_int desmai[MED_NBR_GEOMETRIE_MAILLE] = {0,2,3,3,3,4,4,4,4,6,6,5,5,5,5}; - med_int nmailles[MED_NBR_GEOMETRIE_MAILLE]; - char nommai[MED_NBR_GEOMETRIE_MAILLE] [MED_TAILLE_NOM+1] = {"MED_POINT1", - "MED_SEG2", - "MED_SEG3", - "MED_TRIA3", - "MED_TRIA6", - "MED_QUAD4", - "MED_QUAD8", - "MED_TETRA4", - "MED_TETRA10", - "MED_HEXA8", - "MED_HEXA20", - "MED_PENTA6", - "MED_PENTA15", - "MED_PYRA5", - "MED_PYRA13"}; - med_geometrie_element typfac[MED_NBR_GEOMETRIE_FACE] = {MED_TRIA3,MED_TRIA6, - MED_QUAD4,MED_QUAD8}; - med_int desfac[MED_NBR_GEOMETRIE_FACE] = {3,3,4,4}; - med_int nfaces[MED_NBR_GEOMETRIE_FACE]; - char nomfac[MED_NBR_GEOMETRIE_FACE][MED_TAILLE_NOM+1] = {"MED_TRIA3","MED_TRIA6", - "MED_QUAD4","MED_QUAD8"}; - med_geometrie_element typare[MED_NBR_GEOMETRIE_ARETE] = {MED_SEG2,MED_SEG3}; - med_int desare[MED_NBR_GEOMETRIE_ARETE] = {2,3}; - med_int naretes[MED_NBR_GEOMETRIE_ARETE]; - char nomare[MED_NBR_GEOMETRIE_ARETE] [MED_TAILLE_NOM+1] = {"MED_SEG2","MED_SEG3"}; - /* familles */ - med_int nfam; - med_int natt,ngro; - char *attdes,*gro; - med_int *attval,*attide; - char nomfam[MED_TAILLE_NOM+1]; - med_int numfam; - char str1[MED_TAILLE_DESC+1]; - char str2[MED_TAILLE_LNOM+1]; - string fam; - string fam_type; - string fam_id; - - char* file2Read; - bool locally_managed; - - if (myFileId==-1) - locally_managed = true; - else - locally_managed = false; - - if (locally_managed) - { - file2Read = (char*)myFile.c_str(); - myFileId = MEDouvrir(file2Read,MED_LECT); - if (myFileId < 0) - { - fprintf(stderr,">> ERREUR : ouverture du fichier %s \n",file2Read); - exit(EXIT_FAILURE); - } - numero = 1; - } - else - numero = myMeshId; - sprintf(nommaa,"Mesh %d",myMeshId);//pour load - SCRUTE(nommaa); - - typ_con = MED_NOD; - mode_coo = MED_FULL_INTERLACE; - mdim = 3; - - Handle(SMESHDS_Mesh) mySMESHDSMesh = Handle(SMESHDS_Mesh)::DownCast(myMesh); - - //TopoDS_Shape myShape = mySMESHDSMesh->ShapeToMesh(); - - /**************************************************************************** - * NOMBRES D'OBJETS MED * - ****************************************************************************/ - fprintf(stdout,"\n(****************************)\n"); - fprintf(stdout,"(* INFORMATIONS GENERALES : *)\n"); - fprintf(stdout,"(****************************)\n"); - - /* lecture du nom et de la dimension du maillage */ - /*! fprintf(stdout,"%d %d\n",myFileId,numero); - ret = MEDmaaInfo(myFileId,numero,nommaa,&mdim); - fprintf(stdout,"%d\n",ret); - if (ret < 0) - { - fprintf(stderr,">> ERREUR : lecture du nom du maillage \n"); - exit(EXIT_FAILURE); - } - fprintf(stdout,"- Nom du maillage : <<%s>>\n",nommaa); - fprintf(stdout,"- Dimension du maillage : %d\n",mdim); - */ - /* Combien de noeuds ? */ - nnoe = MEDnEntMaa(myFileId,nommaa,MED_COOR,MED_NOEUD,MED_POINT1,typ_con); - if (nnoe < 0) - { - fprintf(stderr,">> ERREUR : lecture du nombre de noeuds \n"); - exit(EXIT_FAILURE); - } - fprintf(stdout,"- Nombre de noeuds : %d \n",nnoe); - /* Combien de mailles, faces ou aretes ? */ - for (i=0;i> ERREUR : lecture du nombre de mailles \n"); - exit(EXIT_FAILURE); - } - fprintf (stdout,"- Nombre de mailles de type %s : %d \n",nommai[i],nmailles[i]); - } +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; + } + } - for (i=0;i> ERREUR : lecture du nombre de faces \n"); - exit(EXIT_FAILURE); - } - fprintf (stdout,"- Nombre de faces de type %s : %d \n",nomfac[i],nfaces[i]); - } + // Reading MED nodes to the corresponding SMDS structure + //------------------------------------------------------ + PNodeInfo aNodeInfo = aMed->GetPNodeInfo(aMeshInfo); - for (i=0;i> ERREUR : lecture du nombre d'aretes \n"); - exit(EXIT_FAILURE); + 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; + } } - fprintf (stdout,"- Nombre d'aretes de type %s : %d \n",nomare[i],naretes[i]); - } - /* nombre de familles */ - nfam = MEDnFam(myFileId,nommaa,0,MED_FAMILLE); - if (nfam < 0) - { - fprintf(stderr,">> ERREUR : lecture du nombre de familles \n"); - exit(EXIT_FAILURE); - } - fprintf(stdout,"- Nombre de familles : %d \n",nfam); - - vector family[nfam]; - - /**************************************************************************** - * LECTURE DES NOEUDS * - ****************************************************************************/ - fprintf(stdout,"\n(************************)\n"); - fprintf(stdout,"(* NOEUDS DU MAILLAGE : *)\n"); - fprintf(stdout,"(************************)\n"); - - /* Allocations memoires */ - /* table des coordonnees - profil : (dimension * nombre de noeuds ) */ - coo = (med_float*) malloc(sizeof(med_float)*nnoe*mdim); - /* table des numeros, des numeros de familles des noeuds - profil : (nombre de noeuds) */ - numnoe = (med_int*) malloc(sizeof(med_int)*nnoe); - nufano = (med_int*) malloc(sizeof(med_int)*nnoe); - /* table des noms des noeuds - profil : (nnoe*MED_TAILLE_PNOM+1) */ - nomnoe = (char*) malloc(MED_TAILLE_PNOM*nnoe+1); - - /* lecture des noeuds : - - coordonnees - - noms (optionnel dans un fichier MED) - - numeros (optionnel dans un fichier MED) - - numeros des familles */ - ret = MEDnoeudsLire(myFileId,nommaa,mdim,coo,mode_coo,&rep, - nomcoo,unicoo,nomnoe,&inonoe,numnoe,&inunoe, - nufano,nnoe); - if (ret < 0) - strcpy(message,">> ERREUR : lecture des noeuds \n"); - - if (inunoe) { - for (int i=0;iAddNodeWithID(coo[i*3],coo[i*3+1],coo[i*3+2],numnoe[i]); - //fprintf(Out,"%d %f %f %f\n",numnoe[i],coo[i*3],coo[i*3+1],coo[i*3+2]); - } - } - else { - for (int i=0;iAddNodeWithID(coo[i*3],coo[i*3+1],coo[i*3+2],i+1); - //fprintf(Out,"%d %f %f %f\n",numnoe[i],coo[i*3],coo[i*3+1],i); - family[*(nufano+i)].push_back(numnoe[i]); - } - } - - fprintf(stdout,"\n- Numeros des familles des noeuds : \n"); - for (i=0;i 0 && ret == 0) - { - /* dimension de la maille */ - edim = typmai[i] / 100; - nsup = 0; - if (mdim == 2 || mdim == 3) - if (edim == 1) - nsup = 1; - if (mdim == 3) - if (edim == 2) - nsup = 1; - - taille = nsup+typmai[i]%100; - //taille = typmai[i]%100; - - /* allocation memoire */ - connectivite = (med_int*)malloc(sizeof(med_int)* - taille*nmailles[i]); - nomele = (char*)malloc(sizeof(char)*MED_TAILLE_PNOM* - nmailles[i]+1); - numele = (med_int*)malloc(sizeof(med_int)* - nmailles[i]); - nufael = (med_int*)malloc(sizeof(med_int)* - nmailles[i]); - - /* lecture des données */ - ret = MEDelementsLire(myFileId,nommaa,mdim,connectivite,mode_coo, - nomele,&inoele,numele,&inuele,nufael, - nmailles[i],MED_MAILLE,typmai[i], - typ_con); - - switch (typmai[i]) - { - case MED_TRIA3 : { - if (inuele) { - for (j=0;jAddFaceWithID(*(connectivite+j*(taille-nsup)),*(connectivite+j*(taille-nsup)+1),*(connectivite+j*(taille-nsup)+2),elem_id); - //fprintf(Out,"%d %d %d %d\n",elem_id,*(connectivite+j*(taille-nsup)),*(connectivite+j*(taille-nsup)+1),*(connectivite+j*(taille-nsup)+2)); + 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 (j=0;jAddFaceWithID(*(connectivite+j*(taille)),*(connectivite+j*(taille)+1),*(connectivite+j*(taille)+2),cmpt); - //fprintf(Out,"%d %d %d %d\n",j,*(connectivite+j*(taille)),*(connectivite+j*(taille)+1),*(connectivite+j*(taille)+2)); +#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 = "< aNodeIds(aNbNodes); + bool anIsValidConnect = false; + + try{ +#ifdef _EDF_NODE_IDS_ + if(anIsNodeNum) { + for(int i = 0; i < aNbNodes; i++){ + aNodeIds[i] = aNodeInfo->GetElemNum(aCellInfo->GetConn(iElem,i)-1); } - } - else { - for (j=0;jAddFaceWithID(*(connectivite+j*(taille)),*(connectivite+j*(taille)+1),*(connectivite+j*(taille)+2),*(connectivite+j*(taille)+3),cmpt); - //fprintf(Out,"%d %d %d %d\n",j,*(connectivite+j*(taille)),*(connectivite+j*(taille)+1),*(connectivite+j*(taille)+2),*(connectivite+j*(taille)+3)); + }else{ + for(int i = 0; i < aNbNodes; i++){ + aNodeIds[i] = aCellInfo->GetConn(iElem,i); } } - break; +#else + for(int i = 0; i < aNbNodes; i++){ + aNodeIds[i] = aCellInfo->GetConn(iElem,i); + } +#endif + anIsValidConnect = true; + }catch(const std::exception& exc){ + //INFOS("Follow exception was cought:\n\t"<AddVolumeWithID(*(connectivite+j*(taille-nsup)),*(connectivite+j*(taille-nsup)+1),*(connectivite+j*(taille-nsup)+2),*(connectivite+j*(taille-nsup)+3),*(connectivite+j*(taille-nsup)+4),*(connectivite+j*(taille-nsup)+5),*(connectivite+j*(taille-nsup)+6),*(connectivite+j*(taille-nsup)+7),elem_id); - //fprintf(Out,"%d %d %d %d\n",elem_id,*(connectivite+j*(taille-nsup)),*(connectivite+j*(taille-nsup)+1),*(connectivite+j*(taille-nsup)+2),*(connectivite+j*(taille-nsup)+3),*(connectivite+j*(taille-nsup)+4),*(connectivite+j*(taille-nsup)+5),*(connectivite+j*(taille-nsup)+6),*(connectivite+j*(taille-nsup)+7)); + + if(!anIsValidConnect) + continue; + + bool isRenum = false; + SMDS_MeshElement* anElement = NULL; + TInt aFamNum = aCellInfo->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; } - } - else { - for (j=0;jAddVolumeWithID(*(connectivite+j*(taille)),*(connectivite+j*(taille)+1),*(connectivite+j*(taille)+2),*(connectivite+j*(taille)+3),*(connectivite+j*(taille)+4),*(connectivite+j*(taille)+5),*(connectivite+j*(taille)+6),*(connectivite+j*(taille)+7),cmpt); - //fprintf(Out,"%d %d %d %d\n",j,*(connectivite+j*(taille)),*(connectivite+j*(taille)+1),*(connectivite+j*(taille)+2),*(connectivite+j*(taille)+3),*(connectivite+j*(taille)+4),*(connectivite+j*(taille)+5),*(connectivite+j*(taille)+6),*(connectivite+j*(taille)+7)); + 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; } - break; - } - default : { - 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"<> ERREUR : lecture du nombre de groupes d'une famille \n"); - } - - /* nombre d'attributs */ - if (ret == 0) - { - natt = MEDnFam(myFileId,nommaa,i+1,MED_ATTR); - if (natt < 0) - { - ret = -1; - strcpy(message, - ">> ERREUR : lecture du nombre d'attributs d'une famille\n"); - } - } - - if (ret == 0) - fprintf(stdout,"- Famille %d a %d attributs et %d groupes \n",i+1,natt,ngro); - - /* nom,numero,attributs,groupes */ - if (ret == 0) - { - attide = (med_int*) malloc(sizeof(med_int)*natt); - attval = (med_int*) malloc(sizeof(med_int)*natt); - attdes = (char *) malloc(MED_TAILLE_DESC*natt+1); - gro = (char*) malloc(MED_TAILLE_LNOM*ngro+1); - ret = MEDfamInfo(myFileId,nommaa,i+1,nomfam,&numfam,attide,attval, - attdes,&natt,gro,&ngro); - - fam = string(nomfam); - fam_type = fam.substr(1,1); - fam_id = fam.substr(2,1); - if ((fam_type==string("V"))||(fam_type==string("A"))||(fam_type==string("F"))) - LinkMeshToShape(fam_type, fam_id, family[i]); - - fprintf(stdout," - Famille de nom %s et de numero %d : \n",nomfam,numfam); - fprintf(stdout," - Attributs : \n"); - for (j=0;j DriverMED_R_SMESHDS_Mesh::GetMeshNames(Status& theStatus) +{ + list aMeshNames; + + try { + if(MYDEBUG) MESSAGE("GetMeshNames - myFile : " << myFile); + theStatus = DRS_OK; + PWrapper aMed = CrWrapper(myFile); + + if (TInt aNbMeshes = aMed->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"< myNodes) { +list 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; +} - Handle(SMESHDS_Mesh) mySMESHDSMesh = Handle(SMESHDS_Mesh)::DownCast(myMesh); +void DriverMED_R_SMESHDS_Mesh::GetGroup(SMESHDS_Group* theGroup) +{ + string aGroupName (theGroup->GetStoreName()); + if(MYDEBUG) MESSAGE("Get Group " << aGroupName); - int id = atoi(fam_id.c_str()); - if (fam_type==string("V")) { //Linked to a vertex - for (int i=0;iFindNode(myNodes[i]); - const Handle(SMDS_MeshNode)& node = mySMESHDSMesh->GetNode(1,elem); - //const TopoDS_Vertex& S;//le recuperer !!! - //mySMESHDSMesh->SetNodeOnVertex (node,S); + 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() ); } } - else - if (fam_type==string("E")) { //Linked to an edge - for (int i=0;iFindNode(myNodes[i]); - const Handle(SMDS_MeshNode)& node = mySMESHDSMesh->GetNode(1,elem); - //const TopoDS_Edge& S;//le recuperer !!! - //mySMESHDSMesh->SetNodeOnEdge (node,S); +} + +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); + } + } } } - else - if (fam_type==string("F")) { //Linked to a face - for (int i=0;iFindNode(myNodes[i]); - const Handle(SMDS_MeshNode)& node = mySMESHDSMesh->GetNode(1,elem); - //const TopoDS_Face& S;//le recuperer !!! - //mySMESHDSMesh->SetNodeOnFace (node,S); +} + +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 ); }