smesh/src/DriverMED/DriverMED_R_SMESHDS_Mesh.cxx

870 lines
28 KiB
C++
Raw Normal View History

2003-07-10 15:06:41 +06:00
// 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"
2004-12-01 15:48:31 +05:00
#include "SMESHDS_Mesh.hxx"
#include "utilities.h"
2004-06-18 14:34:31 +06:00
#include "DriverMED_Family.h"
2004-06-18 14:34:31 +06:00
#include "SMESHDS_Group.hxx"
2005-01-20 11:25:54 +05:00
#include "MED_Factory.hxx"
2004-06-18 14:34:31 +06:00
#include "MED_Utilities.hxx"
#include <stdlib.h>
#ifdef _DEBUG_
static int MYDEBUG = 0;
#else
static int MYDEBUG = 0;
#endif
2004-07-20 13:56:05 +06:00
#define _EDF_NODE_IDS_
2005-01-20 11:25:54 +05:00
using namespace MED;
2004-06-18 14:34:31 +06:00
void DriverMED_R_SMESHDS_Mesh::SetMeshName(string theMeshName)
2003-09-04 18:03:27 +06:00
{
2004-06-18 14:34:31 +06:00
myMeshName = theMeshName;
}
2004-06-18 14:34:31 +06:00
static const SMDS_MeshNode*
2005-01-20 11:25:54 +05:00
FindNode(const SMDS_Mesh* theMesh, TInt theId){
2004-06-18 14:34:31 +06:00
const SMDS_MeshNode* aNode = theMesh->FindNode(theId);
if(aNode) return aNode;
EXCEPTION(runtime_error,"SMDS_Mesh::FindNode - cannot find a SMDS_MeshNode for ID = "<<theId);
}
2003-09-04 18:03:27 +06:00
2004-07-20 13:56:05 +06:00
enum ECoordName{eX, eY, eZ, eNone};
2005-01-20 11:25:54 +05:00
typedef TFloat (*TGetCoord)(MED::PNodeInfo&, TInt);
2004-07-20 13:56:05 +06:00
template<ECoordName TheCoordId>
2005-01-20 11:25:54 +05:00
TFloat GetCoord(MED::PNodeInfo& thePNodeInfo, TInt theElemId){
2004-07-20 13:56:05 +06:00
return thePNodeInfo->GetNodeCoord(theElemId,TheCoordId);
}
template<>
2005-01-20 11:25:54 +05:00
TFloat GetCoord<eNone>(MED::PNodeInfo& thePNodeInfo, TInt theElemId){
2004-07-20 13:56:05 +06:00
return 0.0;
}
static TGetCoord aXYZGetCoord[3] = {
&GetCoord<eX>,
&GetCoord<eY>,
&GetCoord<eZ>
};
static TGetCoord aXYGetCoord[3] = {
&GetCoord<eX>,
&GetCoord<eY>,
&GetCoord<eNone>
};
static TGetCoord aYZGetCoord[3] = {
&GetCoord<eNone>,
&GetCoord<eX>,
&GetCoord<eY>
};
static TGetCoord aXZGetCoord[3] = {
&GetCoord<eX>,
&GetCoord<eNone>,
&GetCoord<eY>
};
static TGetCoord aXGetCoord[3] = {
&GetCoord<eX>,
&GetCoord<eNone>,
&GetCoord<eNone>
};
static TGetCoord aYGetCoord[3] = {
&GetCoord<eNone>,
&GetCoord<eX>,
&GetCoord<eNone>
};
static TGetCoord aZGetCoord[3] = {
&GetCoord<eNone>,
&GetCoord<eNone>,
&GetCoord<eX>
};
class TCoordHelper{
2005-01-20 11:25:54 +05:00
MED::PNodeInfo myPNodeInfo;
2004-07-20 13:56:05 +06:00
TGetCoord* myGetCoord;
public:
2005-01-20 11:25:54 +05:00
TCoordHelper(const MED::PNodeInfo& thePNodeInfo,
2004-07-20 13:56:05 +06:00
TGetCoord* theGetCoord):
myPNodeInfo(thePNodeInfo),
myGetCoord(theGetCoord)
{}
virtual ~TCoordHelper(){}
2005-01-20 11:25:54 +05:00
TFloat GetCoord(TInt theElemId, TInt theCoodId){
2004-07-20 13:56:05 +06:00
return (*myGetCoord[theCoodId])(myPNodeInfo,theElemId);
}
};
typedef boost::shared_ptr<TCoordHelper> TCoordHelperPtr;
2004-12-01 15:48:31 +05:00
Driver_Mesh::Status DriverMED_R_SMESHDS_Mesh::Perform()
2004-06-18 14:34:31 +06:00
{
2004-12-01 15:48:31 +05:00
Status aResult = DRS_FAIL;
2004-06-18 14:34:31 +06:00
try{
myFamilies.clear();
if(MYDEBUG) MESSAGE("Perform - myFile : "<<myFile);
2005-01-20 11:25:54 +05:00
PWrapper aMed = CrWrapper(myFile);
2004-06-18 14:34:31 +06:00
2004-12-01 15:48:31 +05:00
aResult = DRS_EMPTY;
2005-01-20 11:25:54 +05:00
if(TInt aNbMeshes = aMed->GetNbMeshes()){
2004-06-18 14:34:31 +06:00
for(int iMesh = 0; iMesh < aNbMeshes; iMesh++){
// Reading the MED mesh
//---------------------
2005-01-20 11:25:54 +05:00
PMeshInfo aMeshInfo = aMed->GetPMeshInfo(iMesh+1);
2004-06-18 14:34:31 +06:00
string aMeshName;
if (myMeshId != -1) {
ostringstream aMeshNameStr;
aMeshNameStr<<myMeshId;
aMeshName = aMeshNameStr.str();
} else {
aMeshName = myMeshName;
}
if(MYDEBUG) MESSAGE("Perform - aMeshName : "<<aMeshName<<"; "<<aMeshInfo->GetName());
2004-06-18 14:34:31 +06:00
if(aMeshName != aMeshInfo->GetName()) continue;
2004-12-01 15:48:31 +05:00
aResult = DRS_OK;
2005-01-20 11:25:54 +05:00
//TInt aMeshDim = aMeshInfo->GetDim();
2004-12-01 15:48:31 +05:00
2004-06-18 14:34:31 +06:00
// Reading MED families to the temporary structure
//------------------------------------------------
TErr anErr;
TInt aNbFams = aMed->GetNbFamilies(aMeshInfo);
if(MYDEBUG) MESSAGE("Read " << aNbFams << " families");
2005-01-20 11:25:54 +05:00
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 << " :");
2004-06-18 14:34:31 +06:00
DriverMED_FamilyPtr aFamily (new DriverMED_Family);
2005-01-20 11:25:54 +05:00
TInt aNbGrp = aFamilyInfo->GetNbGroup();
if(MYDEBUG) MESSAGE("belong to " << aNbGrp << " groups");
2005-01-20 11:25:54 +05:00
for (TInt iGr = 0; iGr < aNbGrp; iGr++) {
2004-06-18 14:34:31 +06:00
string aGroupName = aFamilyInfo->GetGroupName(iGr);
if(MYDEBUG) MESSAGE(aGroupName);
2004-06-18 14:34:31 +06:00
aFamily->AddGroupName(aGroupName);
}
aFamily->SetId( aFamId );
2004-06-18 14:34:31 +06:00
myFamilies[aFamId] = aFamily;
}
2004-06-18 14:34:31 +06:00
}
// Reading MED nodes to the corresponding SMDS structure
//------------------------------------------------------
2005-01-20 11:25:54 +05:00
PNodeInfo aNodeInfo = aMed->GetPNodeInfo(aMeshInfo);
2004-07-20 13:56:05 +06:00
TCoordHelperPtr aCoordHelperPtr;
{
2005-01-20 11:25:54 +05:00
TInt aMeshDimension = aMeshInfo->GetDim();
2004-07-20 13:56:05 +06:00
bool anIsDimPresent[3] = {false, false, false};
2005-01-20 11:25:54 +05:00
for(TInt iDim = 0; iDim < aMeshDimension; iDim++){
2004-07-20 13:56:05 +06:00
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;
}
}
2005-01-20 11:25:54 +05:00
EBooleen anIsNodeNum = aNodeInfo->IsElemNum();
TInt aNbElems = aNodeInfo->GetNbElem();
if(MYDEBUG) MESSAGE("Perform - aNodeInfo->GetNbElem() = "<<aNbElems<<"; anIsNodeNum = "<<anIsNodeNum);
DriverMED_FamilyPtr aFamily = myFamilies.begin()->second;
2005-01-20 11:25:54 +05:00
for(TInt iElem = 0; iElem < aNbElems; iElem++){
2004-06-18 14:34:31 +06:00
double aCoords[3] = {0.0, 0.0, 0.0};
2005-01-20 11:25:54 +05:00
for(TInt iDim = 0; iDim < 3; iDim++)
2004-07-20 13:56:05 +06:00
aCoords[iDim] = aCoordHelperPtr->GetCoord(iElem,iDim);
2004-06-18 14:34:31 +06:00
const SMDS_MeshNode* aNode;
if(anIsNodeNum) {
aNode = myMesh->AddNodeWithID
(aCoords[0],aCoords[1],aCoords[2],aNodeInfo->GetElemNum(iElem));
} else {
aNode = myMesh->AddNode
(aCoords[0],aCoords[1],aCoords[2]);
}
//cout<<aNode->GetID()<<": "<<aNode->X()<<", "<<aNode->Y()<<", "<<aNode->Z()<<endl;
// Save reference to this node from its family
2005-01-20 11:25:54 +05:00
TInt aFamNum = aNodeInfo->GetFamNum(iElem);
if ( checkFamilyID ( aFamily, aFamNum ))
2004-06-18 14:34:31 +06:00
{
aFamily->AddElement(aNode);
aFamily->SetType(SMDSAbs_Node);
2004-06-18 14:34:31 +06:00
}
}
// Reading pre information about all MED cells
//--------------------------------------------
bool takeNumbers = true; // initially we trust the numbers from file
2005-01-20 11:25:54 +05:00
MED::TEntityInfo aEntityInfo = aMed->GetEntityInfo(aMeshInfo);
2004-06-18 14:34:31 +06:00
MED::TEntityInfo::iterator anEntityIter = aEntityInfo.begin();
for(; anEntityIter != aEntityInfo.end(); anEntityIter++){
2005-01-20 11:25:54 +05:00
const EEntiteMaillage& anEntity = anEntityIter->first;
if(anEntity == eNOEUD) continue;
2004-06-18 14:34:31 +06:00
// 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++){
2005-01-20 11:25:54 +05:00
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<int> 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<const SMDS_MeshNode*> 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<int> 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<int> 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<const SMDS_MeshNode*> 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 {
}
2005-01-20 11:25:54 +05:00
PCellInfo aCellInfo = aMed->GetPCellInfo(aMeshInfo,anEntity,aGeom);
EBooleen anIsElemNum = takeNumbers ? aCellInfo->IsElemNum() : eFAUX;
TInt aNbElems = aCellInfo->GetNbElem();
if(MYDEBUG) MESSAGE("Perform - anEntity = "<<anEntity<<"; anIsElemNum = "<<anIsElemNum);
if(MYDEBUG) MESSAGE("Perform - aGeom = "<<aGeom<<"; aNbElems = "<<aNbElems);
2004-06-18 14:34:31 +06:00
for(int iElem = 0; iElem < aNbElems; iElem++){
2005-01-20 11:25:54 +05:00
TInt aNbNodes = -1;
2004-06-18 14:34:31 +06:00
switch(aGeom){
2005-01-20 11:25:54 +05:00
case eSEG2:
case eSEG3:
2004-06-18 14:34:31 +06:00
aNbNodes = 2;
break;
2005-01-20 11:25:54 +05:00
case eTRIA3:
case eTRIA6:
2004-06-18 14:34:31 +06:00
aNbNodes = 3;
break;
break;
2005-01-20 11:25:54 +05:00
case eQUAD4:
case eQUAD8:
2004-06-18 14:34:31 +06:00
aNbNodes = 4;
break;
2005-01-20 11:25:54 +05:00
case eTETRA4:
case eTETRA10:
2004-06-18 14:34:31 +06:00
aNbNodes = 4;
break;
2005-01-20 11:25:54 +05:00
case ePYRA5:
case ePYRA13:
2004-06-18 14:34:31 +06:00
aNbNodes = 5;
break;
2005-01-20 11:25:54 +05:00
case ePENTA6:
case ePENTA15:
2004-06-18 14:34:31 +06:00
aNbNodes = 6;
break;
2005-01-20 11:25:54 +05:00
case eHEXA8:
case eHEXA20:
2004-06-18 14:34:31 +06:00
aNbNodes = 8;
break;
}
2005-01-20 11:25:54 +05:00
vector<TInt> aNodeIds(aNbNodes);
bool anIsValidConnect = false;
try{
2004-07-20 13:56:05 +06:00
#ifdef _EDF_NODE_IDS_
if(anIsNodeNum) {
for(int i = 0; i < aNbNodes; i++){
aNodeIds[i] = aNodeInfo->GetElemNum(aCellInfo->GetConn(iElem,i)-1);
}
}else{
for(int i = 0; i < aNbNodes; i++){
aNodeIds[i] = aCellInfo->GetConn(iElem,i);
}
2003-09-04 18:03:27 +06:00
}
#else
2004-06-18 14:34:31 +06:00
for(int i = 0; i < aNbNodes; i++){
2004-07-20 13:56:05 +06:00
aNodeIds[i] = aCellInfo->GetConn(iElem,i);
2003-09-04 18:03:27 +06:00
}
2004-07-20 13:56:05 +06:00
#endif
anIsValidConnect = true;
}catch(const std::exception& exc){
//INFOS("Follow exception was cought:\n\t"<<exc.what());
aResult = DRS_FAIL;
}catch(...){
//INFOS("Unknown exception was cought !!!");
aResult = DRS_FAIL;
}
if(!anIsValidConnect)
continue;
2004-06-18 14:34:31 +06:00
bool isRenum = false;
SMDS_MeshElement* anElement = NULL;
2005-01-20 11:25:54 +05:00
TInt aFamNum = aCellInfo->GetFamNum(iElem);
2004-06-18 14:34:31 +06:00
try{
//MESSAGE("Try to create element # " << iElem << " with id = "
// << aCellInfo->GetElemNum(iElem));
2004-06-18 14:34:31 +06:00
switch(aGeom){
2005-01-20 11:25:54 +05:00
case eSEG2:
case eSEG3:
2004-06-18 14:34:31 +06:00
if(anIsElemNum)
2004-07-20 13:56:05 +06:00
anElement = myMesh->AddEdgeWithID(aNodeIds[0],
aNodeIds[1],
2004-06-18 14:34:31 +06:00
aCellInfo->GetElemNum(iElem));
if (!anElement) {
2004-07-20 13:56:05 +06:00
anElement = myMesh->AddEdge(FindNode(myMesh,aNodeIds[0]),
FindNode(myMesh,aNodeIds[1]));
2004-06-18 14:34:31 +06:00
isRenum = anIsElemNum;
}
break;
2005-01-20 11:25:54 +05:00
case eTRIA3:
case eTRIA6:
2004-06-18 14:34:31 +06:00
aNbNodes = 3;
if(anIsElemNum)
2004-07-20 13:56:05 +06:00
anElement = myMesh->AddFaceWithID(aNodeIds[0],
aNodeIds[1],
aNodeIds[2],
2004-06-18 14:34:31 +06:00
aCellInfo->GetElemNum(iElem));
if (!anElement) {
2004-07-20 13:56:05 +06:00
anElement = myMesh->AddFace(FindNode(myMesh,aNodeIds[0]),
FindNode(myMesh,aNodeIds[1]),
FindNode(myMesh,aNodeIds[2]));
2004-06-18 14:34:31 +06:00
isRenum = anIsElemNum;
}
break;
2005-01-20 11:25:54 +05:00
case eQUAD4:
case eQUAD8:
2004-06-18 14:34:31 +06:00
aNbNodes = 4;
// There is some differnce between SMDS and MED
if(anIsElemNum)
2004-07-20 13:56:05 +06:00
anElement = myMesh->AddFaceWithID(aNodeIds[0],
aNodeIds[1],
aNodeIds[2],
aNodeIds[3],
2004-06-18 14:34:31 +06:00
aCellInfo->GetElemNum(iElem));
if (!anElement) {
2004-07-20 13:56:05 +06:00
anElement = myMesh->AddFace(FindNode(myMesh,aNodeIds[0]),
FindNode(myMesh,aNodeIds[1]),
FindNode(myMesh,aNodeIds[2]),
FindNode(myMesh,aNodeIds[3]));
2004-06-18 14:34:31 +06:00
isRenum = anIsElemNum;
}
break;
2005-01-20 11:25:54 +05:00
case eTETRA4:
case eTETRA10:
2004-06-18 14:34:31 +06:00
aNbNodes = 4;
if(anIsElemNum)
2004-07-20 13:56:05 +06:00
anElement = myMesh->AddVolumeWithID(aNodeIds[0],
aNodeIds[1],
aNodeIds[2],
aNodeIds[3],
2004-06-18 14:34:31 +06:00
aCellInfo->GetElemNum(iElem));
if (!anElement) {
2004-07-20 13:56:05 +06:00
anElement = myMesh->AddVolume(FindNode(myMesh,aNodeIds[0]),
FindNode(myMesh,aNodeIds[1]),
FindNode(myMesh,aNodeIds[2]),
FindNode(myMesh,aNodeIds[3]));
2004-06-18 14:34:31 +06:00
isRenum = anIsElemNum;
}
break;
2005-01-20 11:25:54 +05:00
case ePYRA5:
case ePYRA13:
2004-06-18 14:34:31 +06:00
aNbNodes = 5;
// There is some differnce between SMDS and MED
if(anIsElemNum)
2004-07-20 13:56:05 +06:00
anElement = myMesh->AddVolumeWithID(aNodeIds[0],
aNodeIds[1],
aNodeIds[2],
aNodeIds[3],
aNodeIds[4],
2004-06-18 14:34:31 +06:00
aCellInfo->GetElemNum(iElem));
if (!anElement) {
2004-07-20 13:56:05 +06:00
anElement = myMesh->AddVolume(FindNode(myMesh,aNodeIds[0]),
FindNode(myMesh,aNodeIds[1]),
FindNode(myMesh,aNodeIds[2]),
FindNode(myMesh,aNodeIds[3]),
FindNode(myMesh,aNodeIds[4]));
2004-06-18 14:34:31 +06:00
isRenum = anIsElemNum;
}
break;
2005-01-20 11:25:54 +05:00
case ePENTA6:
case ePENTA15:
2004-06-18 14:34:31 +06:00
aNbNodes = 6;
if(anIsElemNum)
2004-07-20 13:56:05 +06:00
anElement = myMesh->AddVolumeWithID(aNodeIds[0],
aNodeIds[1],
aNodeIds[2],
aNodeIds[3],
aNodeIds[4],
aNodeIds[5],
2004-06-18 14:34:31 +06:00
aCellInfo->GetElemNum(iElem));
if (!anElement) {
2004-07-20 13:56:05 +06:00
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]));
2004-06-18 14:34:31 +06:00
isRenum = anIsElemNum;
}
break;
2005-01-20 11:25:54 +05:00
case eHEXA8:
case eHEXA20:
2004-06-18 14:34:31 +06:00
aNbNodes = 8;
if(anIsElemNum)
2004-07-20 13:56:05 +06:00
anElement = myMesh->AddVolumeWithID(aNodeIds[0],
aNodeIds[1],
aNodeIds[2],
aNodeIds[3],
aNodeIds[4],
aNodeIds[5],
aNodeIds[6],
aNodeIds[7],
2004-06-18 14:34:31 +06:00
aCellInfo->GetElemNum(iElem));
if (!anElement) {
2004-07-20 13:56:05 +06:00
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]));
2004-06-18 14:34:31 +06:00
isRenum = anIsElemNum;
}
break;
2003-09-04 18:03:27 +06:00
}
2004-06-18 14:34:31 +06:00
}catch(const std::exception& exc){
//INFOS("Follow exception was cought:\n\t"<<exc.what());
2004-12-01 15:48:31 +05:00
aResult = DRS_FAIL;
2004-06-18 14:34:31 +06:00
}catch(...){
//INFOS("Unknown exception was cought !!!");
2004-12-01 15:48:31 +05:00
aResult = DRS_FAIL;
2004-06-18 14:34:31 +06:00
}
if (!anElement) {
2004-12-01 15:48:31 +05:00
aResult = DRS_WARN_SKIP_ELEM;
2004-06-18 14:34:31 +06:00
}
else {
if (isRenum) {
2005-01-20 11:25:54 +05:00
anIsElemNum = eFAUX;
2004-06-18 14:34:31 +06:00
takeNumbers = false;
2004-12-01 15:48:31 +05:00
if (aResult < DRS_WARN_RENUMBER)
aResult = DRS_WARN_RENUMBER;
2004-06-18 14:34:31 +06:00
}
if ( checkFamilyID ( aFamily, aFamNum ))
{
2004-06-18 14:34:31 +06:00
// Save reference to this element from its family
aFamily->AddElement(anElement);
aFamily->SetType(anElement->GetType());
2004-06-18 14:34:31 +06:00
}
}
}
}
2003-09-04 18:03:27 +06:00
}
2004-06-18 14:34:31 +06:00
break;
}
}
}catch(const std::exception& exc){
INFOS("Follow exception was cought:\n\t"<<exc.what());
2004-12-01 15:48:31 +05:00
aResult = DRS_FAIL;
2004-06-18 14:34:31 +06:00
}catch(...){
INFOS("Unknown exception was cought !!!");
2004-12-01 15:48:31 +05:00
aResult = DRS_FAIL;
2004-06-18 14:34:31 +06:00
}
if(MYDEBUG) MESSAGE("Perform - aResult status = "<<aResult);
2004-12-01 15:48:31 +05:00
return aResult;
2004-06-18 14:34:31 +06:00
}
2003-09-04 18:03:27 +06:00
2004-12-01 15:48:31 +05:00
list<string> DriverMED_R_SMESHDS_Mesh::GetMeshNames(Status& theStatus)
2004-06-18 14:34:31 +06:00
{
list<string> aMeshNames;
try {
if(MYDEBUG) MESSAGE("GetMeshNames - myFile : " << myFile);
2004-12-01 15:48:31 +05:00
theStatus = DRS_OK;
2005-01-20 11:25:54 +05:00
PWrapper aMed = CrWrapper(myFile);
2004-06-18 14:34:31 +06:00
2005-01-20 11:25:54 +05:00
if (TInt aNbMeshes = aMed->GetNbMeshes()) {
2004-06-18 14:34:31 +06:00
for (int iMesh = 0; iMesh < aNbMeshes; iMesh++) {
// Reading the MED mesh
//---------------------
2005-01-20 11:25:54 +05:00
PMeshInfo aMeshInfo = aMed->GetPMeshInfo(iMesh+1);
2004-06-18 14:34:31 +06:00
aMeshNames.push_back(aMeshInfo->GetName());
}
}
}catch(const std::exception& exc){
INFOS("Follow exception was cought:\n\t"<<exc.what());
2004-12-01 15:48:31 +05:00
theStatus = DRS_FAIL;
2004-06-18 14:34:31 +06:00
}catch(...){
INFOS("Unknown exception was cought !!!");
2004-12-01 15:48:31 +05:00
theStatus = DRS_FAIL;
2004-06-18 14:34:31 +06:00
}
return aMeshNames;
}
list<TNameAndType> DriverMED_R_SMESHDS_Mesh::GetGroupNamesAndTypes()
2003-09-04 18:03:27 +06:00
{
list<TNameAndType> aResult;
set<TNameAndType> aResGroupNames;
2004-06-18 14:34:31 +06:00
map<int, DriverMED_FamilyPtr>::iterator aFamsIter = myFamilies.begin();
for (; aFamsIter != myFamilies.end(); aFamsIter++)
{
DriverMED_FamilyPtr aFamily = (*aFamsIter).second;
const MED::TStringSet& aGroupNames = aFamily->GetGroupNames();
2005-08-31 16:10:42 +06:00
set<string>::const_iterator aGrNamesIter = aGroupNames.begin();
2004-06-18 14:34:31 +06:00
for (; aGrNamesIter != aGroupNames.end(); aGrNamesIter++)
{
TNameAndType aNameAndType = make_pair( *aGrNamesIter, aFamily->GetType() );
2004-06-18 14:34:31 +06:00
// 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 );
2004-06-18 14:34:31 +06:00
}
// }
}
}
return aResult;
}
2004-06-18 14:34:31 +06:00
void DriverMED_R_SMESHDS_Mesh::GetGroup(SMESHDS_Group* theGroup)
{
string aGroupName (theGroup->GetStoreName());
if(MYDEBUG) MESSAGE("Get Group " << aGroupName);
2004-06-18 14:34:31 +06:00
map<int, DriverMED_FamilyPtr>::iterator aFamsIter = myFamilies.begin();
for (; aFamsIter != myFamilies.end(); aFamsIter++)
{
DriverMED_FamilyPtr aFamily = (*aFamsIter).second;
if (aFamily->GetType() == theGroup->GetType() && aFamily->MemberOf(aGroupName))
2004-06-18 14:34:31 +06:00
{
const set<const SMDS_MeshElement *>& anElements = aFamily->GetElements();
2005-08-31 16:10:42 +06:00
set<const SMDS_MeshElement *>::const_iterator anElemsIter = anElements.begin();
2004-12-01 15:48:31 +05:00
const SMDS_MeshElement * element = 0;
2004-06-18 14:34:31 +06:00
for (; anElemsIter != anElements.end(); anElemsIter++)
{
2004-12-01 15:48:31 +05:00
element = *anElemsIter;
theGroup->SMDSGroup().Add(element);
2004-06-18 14:34:31 +06:00
}
2004-12-01 15:48:31 +05:00
if ( element )
theGroup->SetType( element->GetType() );
2004-06-18 14:34:31 +06:00
}
}
}
2003-09-04 18:03:27 +06:00
2004-06-18 14:34:31 +06:00
void DriverMED_R_SMESHDS_Mesh::GetSubMesh (SMESHDS_SubMesh* theSubMesh,
const int theId)
{
char submeshGrpName[ 30 ];
sprintf( submeshGrpName, "SubMesh %d", theId );
string aName (submeshGrpName);
map<int, DriverMED_FamilyPtr>::iterator aFamsIter = myFamilies.begin();
for (; aFamsIter != myFamilies.end(); aFamsIter++)
{
DriverMED_FamilyPtr aFamily = (*aFamsIter).second;
if (aFamily->MemberOf(aName))
{
const set<const SMDS_MeshElement *>& anElements = aFamily->GetElements();
2005-08-31 16:10:42 +06:00
set<const SMDS_MeshElement *>::const_iterator anElemsIter = anElements.begin();
2004-06-18 14:34:31 +06:00
if (aFamily->GetType() == SMDSAbs_Node)
{
for (; anElemsIter != anElements.end(); anElemsIter++)
{
const SMDS_MeshNode* node = static_cast<const SMDS_MeshNode*>(*anElemsIter);
theSubMesh->AddNode(node);
}
}
else
{
for (; anElemsIter != anElements.end(); anElemsIter++)
{
theSubMesh->AddElement(*anElemsIter);
}
}
}
}
}
2004-06-18 14:34:31 +06:00
void DriverMED_R_SMESHDS_Mesh::CreateAllSubMeshes ()
{
map<int, DriverMED_FamilyPtr>::iterator aFamsIter = myFamilies.begin();
for (; aFamsIter != myFamilies.end(); aFamsIter++)
{
DriverMED_FamilyPtr aFamily = (*aFamsIter).second;
MED::TStringSet aGroupNames = aFamily->GetGroupNames();
set<string>::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<const SMDS_MeshElement *> anElements = aFamily->GetElements();
set<const SMDS_MeshElement *>::iterator anElemsIter = anElements.begin();
if (aFamily->GetType() == SMDSAbs_Node)
{
for (; anElemsIter != anElements.end(); anElemsIter++)
{
2004-12-01 15:48:31 +05:00
SMDS_MeshNode* node = const_cast<SMDS_MeshNode*>
( static_cast<const SMDS_MeshNode*>( *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);
}
2004-06-18 14:34:31 +06:00
}
}
else
{
for (; anElemsIter != anElements.end(); anElemsIter++)
{
2004-12-01 15:48:31 +05:00
myMesh->SetMeshElementOnShape(*anElemsIter, Id);
2004-06-18 14:34:31 +06:00
}
}
}
}
}
2003-09-04 18:03:27 +06:00
}
/*!
* \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<int, DriverMED_FamilyPtr>::const_iterator i_fam = myFamilies.find(anID);
if ( i_fam == myFamilies.end() )
return false;
aFamily = i_fam->second;
}
return ( aFamily->GetId() == anID );
}