1 // SMESH DriverMED : driver to read and write 'med' files
3 // Copyright (C) 2003 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License.
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 // Lesser General Public License for more details.
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20 // See http://www.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org
24 // File : DriverMED_W_SMESHDS_Mesh.cxx
27 #include "DriverMED_W_SMESHDS_Mesh.h"
28 #include "DriverMED_W_SMDS_Mesh.h"
29 #include "DriverMED_Family.h"
31 #include "SMDS_MeshElement.hxx"
32 #include "SMDS_MeshNode.hxx"
33 #include "utilities.h"
35 #include "MEDA_Wrapper.hxx"
38 #include "MED_Utilities.hxx"
40 DriverMED_W_SMESHDS_Mesh::DriverMED_W_SMESHDS_Mesh()
46 myAllSubMeshes (false),
47 myDoGroupOfNodes (false),
48 myDoGroupOfEdges (false),
49 myDoGroupOfFaces (false),
50 myDoGroupOfVolumes (false)
54 DriverMED_W_SMESHDS_Mesh::~DriverMED_W_SMESHDS_Mesh()
58 void DriverMED_W_SMESHDS_Mesh::SetMesh(SMDS_Mesh * aMesh)
63 void DriverMED_W_SMESHDS_Mesh::SetFile(string aFile)
68 void DriverMED_W_SMESHDS_Mesh::SetFileId(med_idt aFileId)
73 void DriverMED_W_SMESHDS_Mesh::SetMeshId(int aMeshId)
78 void DriverMED_W_SMESHDS_Mesh::SetMeshName(string theMeshName)
80 myMeshName = theMeshName;
83 void DriverMED_W_SMESHDS_Mesh::AddGroup(SMESHDS_Group* theGroup)
85 myGroups.push_back(theGroup);
88 void DriverMED_W_SMESHDS_Mesh::AddAllSubMeshes()
90 myAllSubMeshes = true;
93 void DriverMED_W_SMESHDS_Mesh::AddSubMesh(SMESHDS_SubMesh* theSubMesh, int theID)
95 mySubMeshes[theID] = theSubMesh;
98 void DriverMED_W_SMESHDS_Mesh::AddGroupOfNodes()
100 myDoGroupOfNodes = true;
103 void DriverMED_W_SMESHDS_Mesh::AddGroupOfEdges()
105 myDoGroupOfEdges = true;
108 void DriverMED_W_SMESHDS_Mesh::AddGroupOfFaces()
110 myDoGroupOfFaces = true;
113 void DriverMED_W_SMESHDS_Mesh::AddGroupOfVolumes()
115 myDoGroupOfVolumes = true;
118 void DriverMED_W_SMESHDS_Mesh::Write()
120 string myClass = string("SMDS_Mesh");
121 string myExtension = string("MED");
123 DriverMED_W_SMDS_Mesh *myWriter = new DriverMED_W_SMDS_Mesh;
125 myWriter->SetMesh(myMesh);
126 // myWriter->SetFile(myFile);
127 myWriter->SetMeshId(myMeshId);
128 myWriter->SetFileId(myFileId);
133 void DriverMED_W_SMESHDS_Mesh::Add()
135 if (myMesh->hasConstructionEdges() || myMesh->hasConstructionFaces()) {
136 INFOS("SMDS_MESH with hasConstructionEdges() or hasConstructionFaces() do not supports!!!");
140 using namespace MEDA;
141 using namespace boost;
143 MESSAGE("Add - myFile : "<<myFile);
144 TWrapper aMed(myFile);
146 // Creating the MED mesh for corresponding SMDS structure
147 //-------------------------------------------------------
149 if (myMeshId != -1) {
150 ostringstream aMeshNameStr;
151 aMeshNameStr<<myMeshId;
152 aMeshName = aMeshNameStr.str();
154 aMeshName = myMeshName;
156 const int SMDS_MESH_DIM = 3;
157 PMeshInfo aMeshInfo = TWrapper::CrMeshInfo(SMDS_MESH_DIM,aMeshName);
158 MESSAGE("Add - aMeshName : "<<aMeshName<<"; "<<aMeshInfo->GetName());
159 aMed.SetMeshInfo(aMeshInfo);
161 // Storing SMDS groups and sub-meshes
162 //-----------------------------------
163 int myNodesDefaultFamilyId = 0;
164 int myEdgesDefaultFamilyId = 0;
165 int myFacesDefaultFamilyId = 0;
166 int myVolumesDefaultFamilyId = 0;
167 if (myDoGroupOfNodes)
168 myNodesDefaultFamilyId = REST_NODES_FAMILY;
169 if (myDoGroupOfEdges)
170 myEdgesDefaultFamilyId = REST_EDGES_FAMILY;
171 if (myDoGroupOfFaces)
172 myFacesDefaultFamilyId = REST_FACES_FAMILY;
173 if (myDoGroupOfVolumes)
174 myVolumesDefaultFamilyId = REST_VOLUMES_FAMILY;
176 MESSAGE("Add - aFamilyInfo");
177 map<const SMDS_MeshElement *, int> anElemFamMap;
178 list<DriverMED_FamilyPtr> aFamilies;
179 if (myAllSubMeshes) {
180 SMESHDS_Mesh* aSMESHDSMesh = dynamic_cast<SMESHDS_Mesh*>(myMesh);
182 EXCEPTION(runtime_error,"Can not cast SMDS_Mesh to SMESHDS_Mesh");
184 aFamilies = DriverMED_Family::MakeFamilies
185 (aSMESHDSMesh->SubMeshes(), myGroups,
186 myDoGroupOfNodes, myDoGroupOfEdges, myDoGroupOfFaces, myDoGroupOfVolumes);
188 aFamilies = DriverMED_Family::MakeFamilies
189 (mySubMeshes, myGroups,
190 myDoGroupOfNodes, myDoGroupOfEdges, myDoGroupOfFaces, myDoGroupOfVolumes);
192 list<DriverMED_FamilyPtr>::iterator aFamsIter = aFamilies.begin();
194 for (; aFamsIter != aFamilies.end(); aFamsIter++)
196 PFamilyInfo aFamilyInfo = (*aFamsIter)->GetFamilyInfo(aMeshInfo);
197 aMed.SetFamilyInfo(aFamilyInfo);
198 int aFamId = (*aFamsIter)->GetId();
200 const set<const SMDS_MeshElement *>& anElems = (*aFamsIter)->GetElements();
201 set<const SMDS_MeshElement *>::iterator anElemsIter = anElems.begin();
202 for (; anElemsIter != anElems.end(); anElemsIter++)
204 anElemFamMap[*anElemsIter] = aFamId;
206 // delete (*aFamsIter);
209 // Storing SMDS nodes to the MED file for the MED mesh
210 //----------------------------------------------------
211 typedef map<med_int,med_int> TNodeIdMap;
212 TNodeIdMap aNodeIdMap;
214 med_int aNbElems = myMesh->NbNodes();
215 MED::TIntVector anElemNums(aNbElems);
216 MED::TIntVector aFamilyNums(aNbElems);
217 MED::TFloatVector aCoordinates(aNbElems*SMDS_MESH_DIM);
218 SMDS_NodeIteratorPtr aNodesIter = myMesh->nodesIterator();
219 for(med_int iNode = 0, iCoord = 0; aNodesIter->more(); iNode++, iCoord+=SMDS_MESH_DIM){
220 const SMDS_MeshNode* aNode = aNodesIter->next();
221 aCoordinates[iCoord] = aNode->X();
222 aCoordinates[iCoord+1] = aNode->Y();
223 aCoordinates[iCoord+2] = aNode->Z();
224 TNodeIdMap::key_type aNodeId = aNode->GetID();
225 anElemNums[iNode] = aNodeId;
226 aNodeIdMap[aNodeId] = iNode+1;
227 //cout<<aNode->GetID()<<": "<<aNode->X()<<", "<<aNode->Y()<<", "<<aNode->Z()<<endl;
229 if (anElemFamMap.find(aNode) != anElemFamMap.end())
230 aFamilyNums[iNode] = anElemFamMap[aNode];
232 aFamilyNums[iNode] = myNodesDefaultFamilyId;
235 MED::TStringVector aCoordNames(3);
236 aCoordNames[0] = "x"; aCoordNames[1] = "y"; aCoordNames[2] = "z";
238 MED::TStringVector aCoordUnits(3);
239 aCoordUnits[0] = "m"; aCoordUnits[1] = "m"; aCoordUnits[2] = "m";
241 const med_repere SMDS_COORDINATE_SYSTEM = MED_CART;
243 PNodeInfo aNodeInfo = TWrapper::CrNodeInfo(aMeshInfo,
244 SMDS_COORDINATE_SYSTEM,
250 MESSAGE("Add - aNodeInfo->GetNbElem() = "<<aNbElems);
251 aMed.SetNodeInfo(aNodeInfo);
254 // Storing others SMDS elements to the MED file for the MED mesh
255 //--------------------------------------------------------------
256 const med_entite_maillage SMDS_MED_ENTITY = MED_MAILLE;
257 const med_connectivite SMDS_MED_CONNECTIVITY = MED_NOD;
259 // Storing SMDS Edges
260 if(med_int aNbElems = myMesh->NbEdges()){
261 SMDS_EdgeIteratorPtr anIter = myMesh->edgesIterator();
262 med_int aNbConnectivity = MED::GetNbConn(SMDS_MED_ENTITY,MED_SEG2,SMDS_MESH_DIM);
263 MED::TIntVector anElemNums(aNbElems);
264 MED::TIntVector aFamilyNums(aNbElems);
265 MED::TIntVector aConnectivity(aNbElems*aNbConnectivity);
267 for(med_int iElem = 0, iConn = 0; anIter->more(); iElem++, iConn+=aNbConnectivity){
268 const SMDS_MeshEdge* anElem = anIter->next();
269 SMDS_ElemIteratorPtr aNodesIter = anElem->nodesIterator();
270 for(med_int iNode = 0; iNode < aNbConnectivity && aNodesIter->more(); iNode++){
271 const SMDS_MeshElement* aNode = aNodesIter->next();
272 aConnectivity[iConn+iNode] = aNodeIdMap[aNode->GetID()];
274 anElemNums[iElem] = anElem->GetID();
276 if (anElemFamMap.find(anElem) != anElemFamMap.end())
277 aFamilyNums[iElem] = anElemFamMap[anElem];
279 aFamilyNums[iElem] = myEdgesDefaultFamilyId;
282 PCellInfo aCellInfo = TWrapper::CrCellInfo(aMeshInfo,
285 SMDS_MED_CONNECTIVITY,
289 aMed.SetCellInfo(aCellInfo);
292 // Storing SMDS Faces
293 if(med_int aNbElems = myMesh->NbFaces()){
294 SMDS_FaceIteratorPtr anIter = myMesh->facesIterator();
296 med_int aNbTriaConn = MED::GetNbConn(SMDS_MED_ENTITY,MED_TRIA3,SMDS_MESH_DIM);
297 MED::TIntVector anTriaElemNums;
298 anTriaElemNums.reserve(aNbElems);
299 MED::TIntVector aTriaFamilyNums;
300 aTriaFamilyNums.reserve(aNbElems);
301 MED::TIntVector aTriaConn;
302 aTriaConn.reserve(aNbElems*aNbTriaConn);
304 med_int aNbQuadConn = MED::GetNbConn(SMDS_MED_ENTITY,MED_QUAD4,SMDS_MESH_DIM);
305 MED::TIntVector aQuadElemNums;
306 aQuadElemNums.reserve(aNbElems);
307 MED::TIntVector aQuadFamilyNums;
308 aQuadFamilyNums.reserve(aNbElems);
309 MED::TIntVector aQuadConn;
310 aQuadConn.reserve(aNbElems*aNbQuadConn);
312 for(med_int iElem = 0; iElem < aNbElems && anIter->more(); iElem++){
313 const SMDS_MeshFace* anElem = anIter->next();
314 med_int aNbNodes = anElem->NbNodes();
315 SMDS_ElemIteratorPtr aNodesIter = anElem->nodesIterator();
316 med_int aNbConnectivity;
317 MED::TIntVector* anElemNums;
318 MED::TIntVector* aFamilyNums;
319 MED::TIntVector* aConnectivity;
322 aNbConnectivity = aNbTriaConn;
323 anElemNums = &anTriaElemNums;
324 aFamilyNums = &aTriaFamilyNums;
325 aConnectivity = &aTriaConn;
328 aNbConnectivity = aNbQuadConn;
329 anElemNums = &aQuadElemNums;
330 aFamilyNums = &aQuadFamilyNums;
331 aConnectivity = &aQuadConn;
334 MED::TIntVector aVector(aNbNodes);
335 for(med_int iNode = 0; aNodesIter->more(); iNode++){
336 const SMDS_MeshElement* aNode = aNodesIter->next();
337 aVector[iNode] = aNode->GetID();
340 med_int aSize = aConnectivity->size();
341 aConnectivity->resize(aSize+aNbConnectivity);
342 // There is some differnce between SMDS and MED in cells mapping
345 (*aConnectivity)[aSize+0] = aNodeIdMap[aVector[0]];
346 (*aConnectivity)[aSize+1] = aNodeIdMap[aVector[1]];
347 (*aConnectivity)[aSize+2] = aNodeIdMap[aVector[3]];
348 (*aConnectivity)[aSize+3] = aNodeIdMap[aVector[2]];
350 for(med_int iNode = 0; iNode < aNbNodes; iNode++)
351 (*aConnectivity)[aSize+iNode] = aNodeIdMap[aVector[iNode]];
353 anElemNums->push_back(anElem->GetID());
355 if (anElemFamMap.find(anElem) != anElemFamMap.end())
356 aFamilyNums->push_back(anElemFamMap[anElem]);
358 aFamilyNums->push_back(myFacesDefaultFamilyId);
360 if(med_int aNbElems = anTriaElemNums.size()){
361 PCellInfo aCellInfo = TWrapper::CrCellInfo(aMeshInfo,
364 SMDS_MED_CONNECTIVITY,
368 MESSAGE("Add - anEntity = "<<SMDS_MED_ENTITY<<"; aGeom = "<<MED_TRIA3<<"; aNbElems = "<<aNbElems);
369 aMed.SetCellInfo(aCellInfo);
371 if(med_int aNbElems = aQuadElemNums.size()){
372 PCellInfo aCellInfo = TWrapper::CrCellInfo(aMeshInfo,
375 SMDS_MED_CONNECTIVITY,
379 MESSAGE("Add - anEntity = "<<SMDS_MED_ENTITY<<"; aGeom = "<<MED_QUAD4<<"; aNbElems = "<<aNbElems);
380 aMed.SetCellInfo(aCellInfo);
384 // Storing SMDS Volumes
385 if(med_int aNbElems = myMesh->NbVolumes()){
386 SMDS_VolumeIteratorPtr anIter = myMesh->volumesIterator();
388 med_int aNbTetraConn = MED::GetNbConn(SMDS_MED_ENTITY,MED_TETRA4,SMDS_MESH_DIM);
389 MED::TIntVector anTetraElemNums;
390 anTetraElemNums.reserve(aNbElems);
391 MED::TIntVector aTetraFamilyNums;
392 aTetraFamilyNums.reserve(aNbElems);
393 MED::TIntVector aTetraConn;
394 aTetraConn.reserve(aNbElems*aNbTetraConn);
396 med_int aNbPyraConn = MED::GetNbConn(SMDS_MED_ENTITY,MED_PYRA5,SMDS_MESH_DIM);
397 MED::TIntVector anPyraElemNums;
398 anPyraElemNums.reserve(aNbElems);
399 MED::TIntVector aPyraFamilyNums;
400 aPyraFamilyNums.reserve(aNbElems);
401 MED::TIntVector aPyraConn;
402 aPyraConn.reserve(aNbElems*aNbPyraConn);
404 med_int aNbPentaConn = MED::GetNbConn(SMDS_MED_ENTITY,MED_PENTA6,SMDS_MESH_DIM);
405 MED::TIntVector anPentaElemNums;
406 anPentaElemNums.reserve(aNbElems);
407 MED::TIntVector aPentaFamilyNums;
408 aPentaFamilyNums.reserve(aNbElems);
409 MED::TIntVector aPentaConn;
410 aPentaConn.reserve(aNbElems*aNbPentaConn);
412 med_int aNbHexaConn = MED::GetNbConn(SMDS_MED_ENTITY,MED_HEXA8,SMDS_MESH_DIM);
413 MED::TIntVector aHexaElemNums;
414 aHexaElemNums.reserve(aNbElems);
415 MED::TIntVector aHexaFamilyNums;
416 aHexaFamilyNums.reserve(aNbElems);
417 MED::TIntVector aHexaConn;
418 aHexaConn.reserve(aNbElems*aNbHexaConn);
420 for(med_int iElem = 0; iElem < aNbElems && anIter->more(); iElem++){
421 const SMDS_MeshVolume* anElem = anIter->next();
422 med_int aNbNodes = anElem->NbNodes();
423 SMDS_ElemIteratorPtr aNodesIter = anElem->nodesIterator();
424 med_int aNbConnectivity;
425 MED::TIntVector* anElemNums;
426 MED::TIntVector* aFamilyNums;
427 MED::TIntVector* aConnectivity;
430 aNbConnectivity = aNbTetraConn;
431 anElemNums = &anTetraElemNums;
432 aFamilyNums = &aTetraFamilyNums;
433 aConnectivity = &aTetraConn;
436 aNbConnectivity = aNbPyraConn;
437 anElemNums = &anPyraElemNums;
438 aFamilyNums = &aPyraFamilyNums;
439 aConnectivity = &aPyraConn;
442 aNbConnectivity = aNbPentaConn;
443 anElemNums = &anPentaElemNums;
444 aFamilyNums = &aPentaFamilyNums;
445 aConnectivity = &aPentaConn;
448 aNbConnectivity = aNbHexaConn;
449 anElemNums = &aHexaElemNums;
450 aFamilyNums = &aHexaFamilyNums;
451 aConnectivity = &aHexaConn;
454 MED::TIntVector aVector(aNbNodes);
455 for(med_int iNode = 0; aNodesIter->more(); iNode++){
456 const SMDS_MeshElement* aNode = aNodesIter->next();
457 aVector[iNode] = aNode->GetID();
459 med_int aSize = aConnectivity->size();
460 aConnectivity->resize(aSize+aNbConnectivity);
461 // There is some difference between SMDS and MED in cells mapping
464 (*aConnectivity)[aSize+0] = aNodeIdMap[aVector[0]];
465 (*aConnectivity)[aSize+1] = aNodeIdMap[aVector[3]];
466 (*aConnectivity)[aSize+2] = aNodeIdMap[aVector[2]];
467 (*aConnectivity)[aSize+3] = aNodeIdMap[aVector[1]];
468 (*aConnectivity)[aSize+4] = aNodeIdMap[aVector[4]];
470 for(med_int iNode = 0; iNode < aNbNodes; iNode++)
471 (*aConnectivity)[aSize+iNode] = aNodeIdMap[aVector[iNode]];
473 anElemNums->push_back(anElem->GetID());
475 if (anElemFamMap.find(anElem) != anElemFamMap.end())
476 aFamilyNums->push_back(anElemFamMap[anElem]);
478 aFamilyNums->push_back(myVolumesDefaultFamilyId);
481 if(med_int aNbElems = anTetraElemNums.size()){
482 PCellInfo aCellInfo = TWrapper::CrCellInfo(aMeshInfo,
485 SMDS_MED_CONNECTIVITY,
489 MESSAGE("Add - anEntity = "<<SMDS_MED_ENTITY<<"; aGeom = "<<MED_TETRA4<<"; aNbElems = "<<aNbElems);
490 aMed.SetCellInfo(aCellInfo);
492 if(med_int aNbElems = anPyraElemNums.size()){
493 PCellInfo aCellInfo = TWrapper::CrCellInfo(aMeshInfo,
496 SMDS_MED_CONNECTIVITY,
500 MESSAGE("Add - anEntity = "<<SMDS_MED_ENTITY<<"; aGeom = "<<MED_PYRA5<<"; aNbElems = "<<aNbElems);
501 aMed.SetCellInfo(aCellInfo);
503 if(med_int aNbElems = anPentaElemNums.size()){
504 PCellInfo aCellInfo = TWrapper::CrCellInfo(aMeshInfo,
507 SMDS_MED_CONNECTIVITY,
511 MESSAGE("Add - anEntity = "<<SMDS_MED_ENTITY<<"; aGeom = "<<MED_PENTA6<<"; aNbElems = "<<aNbElems);
512 aMed.SetCellInfo(aCellInfo);
514 if(med_int aNbElems = aHexaElemNums.size()){
515 PCellInfo aCellInfo = TWrapper::CrCellInfo(aMeshInfo,
518 SMDS_MED_CONNECTIVITY,
522 MESSAGE("Add - anEntity = "<<SMDS_MED_ENTITY<<"; aGeom = "<<MED_HEXA8<<"; aNbElems = "<<aNbElems);
523 aMed.SetCellInfo(aCellInfo);
526 }catch(const std::exception& exc){
527 INFOS("Follow exception was cought:\n\t"<<exc.what());
529 INFOS("Unknown exception was cought !!!");