src/DriverMED/DriverMED_R_SMESHDS_Mesh.cxx

   1 //  SMESH DriverMED : driver to read and write 'med' files
   2 //
   3 //  Copyright (C) 2003  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   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.
  10 //
  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.
  15 //
  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
  19 //
  20 //  See http://www.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org
  21 //
  22 //
  23 //
  24 //  File   : DriverMED_R_SMESHDS_Mesh.cxx
  25 //  Module : SMESH
  26
  27 #include "DriverMED_R_SMESHDS_Mesh.h"
  28 #include "DriverMED_R_SMDS_Mesh.h"
  29 #include "SMESHDS_Mesh.hxx"
  30 #include "utilities.h"
  31
  32 #include "DriverMED_Family.h"
  33
  34 #include "SMESHDS_Group.hxx"
  35
  36 #include "MED_Factory.hxx"
  37 #include "MED_Utilities.hxx"
  38
  39 #include <stdlib.h>
  40
  41 #ifdef _DEBUG_
  42 static int MYDEBUG = 0;
  43 #else
  44 static int MYDEBUG = 0;
  45 #endif
  46
  47 #define _EDF_NODE_IDS_
  48
  49 using namespace MED;
  50
  51 void DriverMED_R_SMESHDS_Mesh::SetMeshName(string theMeshName)
  52 {
  53   myMeshName = theMeshName;
  54 }
  55
  56 static const SMDS_MeshNode*
  57 FindNode(const SMDS_Mesh* theMesh, TInt theId){
  58   const SMDS_MeshNode* aNode = theMesh->FindNode(theId);
  59   if(aNode) return aNode;
  60   EXCEPTION(runtime_error,"SMDS_Mesh::FindNode - cannot find a SMDS_MeshNode for ID = "<<theId);
  61 }
  62
  63
  64 enum ECoordName{eX, eY, eZ, eNone};
  65 typedef TFloat (*TGetCoord)(MED::PNodeInfo&, TInt);
  66
  67 template<ECoordName TheCoordId>
  68 TFloat GetCoord(MED::PNodeInfo& thePNodeInfo, TInt theElemId){
  69   return thePNodeInfo->GetNodeCoord(theElemId,TheCoordId);
  70 }
  71
  72 template<>
  73 TFloat GetCoord<eNone>(MED::PNodeInfo& thePNodeInfo, TInt theElemId){
  74   return 0.0;
  75 }
  76
  77
  78 static TGetCoord aXYZGetCoord[3] = {
  79   &GetCoord<eX>,
  80   &GetCoord<eY>,
  81   &GetCoord<eZ>
  82 };
  83
  84
  85 static TGetCoord aXYGetCoord[3] = {
  86   &GetCoord<eX>,
  87   &GetCoord<eY>,
  88   &GetCoord<eNone>
  89 };
  90
  91 static TGetCoord aYZGetCoord[3] = {
  92   &GetCoord<eNone>,
  93   &GetCoord<eX>,
  94   &GetCoord<eY>
  95 };
  96
  97 static TGetCoord aXZGetCoord[3] = {
  98   &GetCoord<eX>,
  99   &GetCoord<eNone>,
 100   &GetCoord<eY>
 101 };
 102
 103
 104 static TGetCoord aXGetCoord[3] = {
 105   &GetCoord<eX>,
 106   &GetCoord<eNone>,
 107   &GetCoord<eNone>
 108 };
 109
 110 static TGetCoord aYGetCoord[3] = {
 111   &GetCoord<eNone>,
 112   &GetCoord<eX>,
 113   &GetCoord<eNone>
 114 };
 115
 116 static TGetCoord aZGetCoord[3] = {
 117   &GetCoord<eNone>,
 118   &GetCoord<eNone>,
 119   &GetCoord<eX>
 120 };
 121
 122
 123 class TCoordHelper{
 124   MED::PNodeInfo myPNodeInfo;
 125   TGetCoord* myGetCoord;
 126 public:
 127   TCoordHelper(const MED::PNodeInfo& thePNodeInfo,
 128                TGetCoord* theGetCoord):
 129     myPNodeInfo(thePNodeInfo),
 130     myGetCoord(theGetCoord)
 131   {}
 132   virtual ~TCoordHelper(){}
 133   TFloat GetCoord(TInt theElemId, TInt theCoodId){
 134     return (*myGetCoord[theCoodId])(myPNodeInfo,theElemId);
 135   }
 136 };
 137 typedef boost::shared_ptr<TCoordHelper> TCoordHelperPtr;
 138
 139
 140 Driver_Mesh::Status DriverMED_R_SMESHDS_Mesh::Perform()
 141 {
 142   Status aResult = DRS_FAIL;
 143   try{
 144     myFamilies.clear();
 145     if(MYDEBUG) MESSAGE("Perform - myFile : "<<myFile);
 146     PWrapper aMed = CrWrapper(myFile);
 147
 148     aResult = DRS_EMPTY;
 149     if(TInt aNbMeshes = aMed->GetNbMeshes()){
 150       for(int iMesh = 0; iMesh < aNbMeshes; iMesh++){
 151         // Reading the MED mesh
 152         //---------------------
 153         PMeshInfo aMeshInfo = aMed->GetPMeshInfo(iMesh+1);
 154         string aMeshName;
 155         if (myMeshId != -1) {
 156           ostringstream aMeshNameStr;
 157           aMeshNameStr<<myMeshId;
 158           aMeshName = aMeshNameStr.str();
 159         } else {
 160           aMeshName = myMeshName;
 161         }
 162         if(MYDEBUG) MESSAGE("Perform - aMeshName : "<<aMeshName<<"; "<<aMeshInfo->GetName());
 163         if(aMeshName != aMeshInfo->GetName()) continue;
 164         aResult = DRS_OK;
 165         //TInt aMeshDim = aMeshInfo->GetDim();
 166
 167         // Reading MED families to the temporary structure
 168         //------------------------------------------------
 169         TErr anErr;
 170         TInt aNbFams = aMed->GetNbFamilies(aMeshInfo);
 171         if(MYDEBUG) MESSAGE("Read " << aNbFams << " families");
 172         for (TInt iFam = 0; iFam < aNbFams; iFam++) {
 173           PFamilyInfo aFamilyInfo = aMed->GetPFamilyInfo(aMeshInfo,iFam+1,&anErr);
 174           if(anErr >= 0){
 175             TInt aFamId = aFamilyInfo->GetId();
 176             if(MYDEBUG) MESSAGE("Family " << aFamId << " :");
 177
 178             DriverMED_FamilyPtr aFamily (new DriverMED_Family);
 179
 180             TInt aNbGrp = aFamilyInfo->GetNbGroup();
 181             if(MYDEBUG) MESSAGE("belong to " << aNbGrp << " groups");
 182             for (TInt iGr = 0; iGr < aNbGrp; iGr++) {
 183               string aGroupName = aFamilyInfo->GetGroupName(iGr);
 184               if(MYDEBUG) MESSAGE(aGroupName);
 185               aFamily->AddGroupName(aGroupName);
 186             }
 187             myFamilies[aFamId] = aFamily;
 188           }
 189         }
 190
 191         // Reading MED nodes to the corresponding SMDS structure
 192         //------------------------------------------------------
 193         PNodeInfo aNodeInfo = aMed->GetPNodeInfo(aMeshInfo);
 194
 195         TCoordHelperPtr aCoordHelperPtr;
 196         {
 197           TInt aMeshDimension = aMeshInfo->GetDim();
 198           bool anIsDimPresent[3] = {false, false, false};
 199           for(TInt iDim = 0; iDim < aMeshDimension; iDim++){
 200             string aDimName = aNodeInfo->GetCoordName(iDim);
 201             if(aDimName == "x" || aDimName == "X")
 202               anIsDimPresent[eX] = true;
 203             else if(aDimName == "y" || aDimName == "Y")
 204               anIsDimPresent[eY] = true;
 205             else if(aDimName == "z" || aDimName == "Z")
 206               anIsDimPresent[eZ] = true;
 207           }
 208           switch(aMeshDimension){
 209           case 3:
 210             aCoordHelperPtr.reset(new TCoordHelper(aNodeInfo,aXYZGetCoord));
 211             break;
 212           case 2:
 213             if(anIsDimPresent[eY] && anIsDimPresent[eZ])
 214               aCoordHelperPtr.reset(new TCoordHelper(aNodeInfo,aYZGetCoord));
 215             else if(anIsDimPresent[eX] && anIsDimPresent[eZ])
 216               aCoordHelperPtr.reset(new TCoordHelper(aNodeInfo,aXZGetCoord));
 217             else
 218               aCoordHelperPtr.reset(new TCoordHelper(aNodeInfo,aXYGetCoord));
 219             break;
 220           case 1:
 221             if(anIsDimPresent[eY])
 222               aCoordHelperPtr.reset(new TCoordHelper(aNodeInfo,aYGetCoord));
 223             else if(anIsDimPresent[eZ])
 224               aCoordHelperPtr.reset(new TCoordHelper(aNodeInfo,aZGetCoord));
 225             else
 226               aCoordHelperPtr.reset(new TCoordHelper(aNodeInfo,aXGetCoord));
 227             break;
 228           }
 229         }
 230
 231         EBooleen anIsNodeNum = aNodeInfo->IsElemNum();
 232         TInt aNbElems = aNodeInfo->GetNbElem();
 233         if(MYDEBUG) MESSAGE("Perform - aNodeInfo->GetNbElem() = "<<aNbElems<<"; anIsNodeNum = "<<anIsNodeNum);
 234         for(TInt iElem = 0; iElem < aNbElems; iElem++){
 235           double aCoords[3] = {0.0, 0.0, 0.0};
 236           for(TInt iDim = 0; iDim < 3; iDim++)
 237             aCoords[iDim] = aCoordHelperPtr->GetCoord(iElem,iDim);
 238           const SMDS_MeshNode* aNode;
 239           if(anIsNodeNum) {
 240             aNode = myMesh->AddNodeWithID
 241               (aCoords[0],aCoords[1],aCoords[2],aNodeInfo->GetElemNum(iElem));
 242           } else {
 243             aNode = myMesh->AddNode
 244               (aCoords[0],aCoords[1],aCoords[2]);
 245           }
 246           //cout<<aNode->GetID()<<": "<<aNode->X()<<", "<<aNode->Y()<<", "<<aNode->Z()<<endl;
 247
 248           // Save reference to this node from its family
 249           TInt aFamNum = aNodeInfo->GetFamNum(iElem);
 250           if (myFamilies.find(aFamNum) != myFamilies.end())
 251           {
 252             myFamilies[aFamNum]->AddElement(aNode);
 253             myFamilies[aFamNum]->SetType(SMDSAbs_Node);
 254           }
 255         }
 256
 257         // Reading pre information about all MED cells
 258         //--------------------------------------------
 259         bool takeNumbers = true;  // initially we trust the numbers from file
 260         MED::TEntityInfo aEntityInfo = aMed->GetEntityInfo(aMeshInfo);
 261         MED::TEntityInfo::iterator anEntityIter = aEntityInfo.begin();
 262         for(; anEntityIter != aEntityInfo.end(); anEntityIter++){
 263           const EEntiteMaillage& anEntity = anEntityIter->first;
 264           if(anEntity == eNOEUD) continue;
 265           // Reading MED cells to the corresponding SMDS structure
 266           //------------------------------------------------------
 267           const MED::TGeom& aTGeom = anEntityIter->second;
 268           MED::TGeom::const_iterator anTGeomIter = aTGeom.begin();
 269           for(; anTGeomIter != aTGeom.end(); anTGeomIter++){
 270             const EGeometrieElement& aGeom = anTGeomIter->first;
 271             if(aGeom == ePOINT1) continue;
 272             PCellInfo aCellInfo = aMed->GetPCellInfo(aMeshInfo,anEntity,aGeom);
 273             EBooleen anIsElemNum = takeNumbers ? aCellInfo->IsElemNum() : eFAUX;
 274             TInt aNbElems = aCellInfo->GetNbElem();
 275             if(MYDEBUG) MESSAGE("Perform - anEntity = "<<anEntity<<"; anIsElemNum = "<<anIsElemNum);
 276             if(MYDEBUG) MESSAGE("Perform - aGeom = "<<aGeom<<"; aNbElems = "<<aNbElems);
 277
 278             for(int iElem = 0; iElem < aNbElems; iElem++){
 279               TInt aNbNodes = -1;
 280               switch(aGeom){
 281               case eSEG2:
 282               case eSEG3:
 283                 aNbNodes = 2;
 284                 break;
 285               case eTRIA3:
 286               case eTRIA6:
 287                 aNbNodes = 3;
 288                 break;
 289                 break;
 290               case eQUAD4:
 291               case eQUAD8:
 292                 aNbNodes = 4;
 293                 break;
 294               case eTETRA4:
 295               case eTETRA10:
 296                 aNbNodes = 4;
 297                 break;
 298               case ePYRA5:
 299               case ePYRA13:
 300                 aNbNodes = 5;
 301                 break;
 302               case ePENTA6:
 303               case ePENTA15:
 304                 aNbNodes = 6;
 305                 break;
 306               case eHEXA8:
 307               case eHEXA20:
 308                 aNbNodes = 8;
 309                 break;
 310               }
 311               vector<TInt> aNodeIds(aNbNodes);
 312 #ifdef _EDF_NODE_IDS_
 313               if(anIsNodeNum) {
 314                 for(int i = 0; i < aNbNodes; i++){
 315                   aNodeIds[i] = aNodeInfo->GetElemNum(aCellInfo->GetConn(iElem,i)-1);
 316                 }
 317               }else{
 318                 for(int i = 0; i < aNbNodes; i++){
 319                   aNodeIds[i] = aCellInfo->GetConn(iElem,i);
 320                 }
 321               }
 322 #else
 323               for(int i = 0; i < aNbNodes; i++){
 324                 aNodeIds[i] = aCellInfo->GetConn(iElem,i);
 325               }
 326 #endif
 327
 328               bool isRenum = false;
 329               SMDS_MeshElement* anElement = NULL;
 330               TInt aFamNum = aCellInfo->GetFamNum(iElem);
 331               try{
 332                 switch(aGeom){
 333                 case eSEG2:
 334                 case eSEG3:
 335                   if(anIsElemNum)
 336                     anElement = myMesh->AddEdgeWithID(aNodeIds[0],
 337                                                       aNodeIds[1],
 338                                                       aCellInfo->GetElemNum(iElem));
 339                   if (!anElement) {
 340                     anElement = myMesh->AddEdge(FindNode(myMesh,aNodeIds[0]),
 341                                                 FindNode(myMesh,aNodeIds[1]));
 342                     isRenum = anIsElemNum;
 343                   }
 344                   break;
 345                 case eTRIA3:
 346                 case eTRIA6:
 347                   aNbNodes = 3;
 348                   if(anIsElemNum)
 349                     anElement = myMesh->AddFaceWithID(aNodeIds[0],
 350                                                       aNodeIds[1],
 351                                                       aNodeIds[2],
 352                                                       aCellInfo->GetElemNum(iElem));
 353                   if (!anElement) {
 354                     anElement = myMesh->AddFace(FindNode(myMesh,aNodeIds[0]),
 355                                                 FindNode(myMesh,aNodeIds[1]),
 356                                                 FindNode(myMesh,aNodeIds[2]));
 357                     isRenum = anIsElemNum;
 358                   }
 359                   break;
 360                 case eQUAD4:
 361                 case eQUAD8:
 362                   aNbNodes = 4;
 363                   // There is some differnce between SMDS and MED
 364                   if(anIsElemNum)
 365                     anElement = myMesh->AddFaceWithID(aNodeIds[0],
 366                                                       aNodeIds[1],
 367                                                       aNodeIds[2],
 368                                                       aNodeIds[3],
 369                                                       aCellInfo->GetElemNum(iElem));
 370                   if (!anElement) {
 371                     anElement = myMesh->AddFace(FindNode(myMesh,aNodeIds[0]),
 372                                                 FindNode(myMesh,aNodeIds[1]),
 373                                                 FindNode(myMesh,aNodeIds[2]),
 374                                                 FindNode(myMesh,aNodeIds[3]));
 375                     isRenum = anIsElemNum;
 376                   }
 377                   break;
 378                 case eTETRA4:
 379                 case eTETRA10:
 380                   aNbNodes = 4;
 381                   if(anIsElemNum)
 382                     anElement = myMesh->AddVolumeWithID(aNodeIds[0],
 383                                                         aNodeIds[1],
 384                                                         aNodeIds[2],
 385                                                         aNodeIds[3],
 386                                                         aCellInfo->GetElemNum(iElem));
 387                   if (!anElement) {
 388                     anElement = myMesh->AddVolume(FindNode(myMesh,aNodeIds[0]),
 389                                                   FindNode(myMesh,aNodeIds[1]),
 390                                                   FindNode(myMesh,aNodeIds[2]),
 391                                                   FindNode(myMesh,aNodeIds[3]));
 392                     isRenum = anIsElemNum;
 393                   }
 394                   break;
 395                 case ePYRA5:
 396                 case ePYRA13:
 397                   aNbNodes = 5;
 398                   // There is some differnce between SMDS and MED
 399                   if(anIsElemNum)
 400                     anElement = myMesh->AddVolumeWithID(aNodeIds[0],
 401                                                         aNodeIds[1],
 402                                                         aNodeIds[2],
 403                                                         aNodeIds[3],
 404                                                         aNodeIds[4],
 405                                                         aCellInfo->GetElemNum(iElem));
 406                   if (!anElement) {
 407                     anElement = myMesh->AddVolume(FindNode(myMesh,aNodeIds[0]),
 408                                                   FindNode(myMesh,aNodeIds[1]),
 409                                                   FindNode(myMesh,aNodeIds[2]),
 410                                                   FindNode(myMesh,aNodeIds[3]),
 411                                                   FindNode(myMesh,aNodeIds[4]));
 412                     isRenum = anIsElemNum;
 413                   }
 414                   break;
 415                 case ePENTA6:
 416                 case ePENTA15:
 417                   aNbNodes = 6;
 418                   if(anIsElemNum)
 419                     anElement = myMesh->AddVolumeWithID(aNodeIds[0],
 420                                                         aNodeIds[1],
 421                                                         aNodeIds[2],
 422                                                         aNodeIds[3],
 423                                                         aNodeIds[4],
 424                                                         aNodeIds[5],
 425                                                         aCellInfo->GetElemNum(iElem));
 426                   if (!anElement) {
 427                     anElement = myMesh->AddVolume(FindNode(myMesh,aNodeIds[0]),
 428                                                   FindNode(myMesh,aNodeIds[1]),
 429                                                   FindNode(myMesh,aNodeIds[2]),
 430                                                   FindNode(myMesh,aNodeIds[3]),
 431                                                   FindNode(myMesh,aNodeIds[4]),
 432                                                   FindNode(myMesh,aNodeIds[5]));
 433                     isRenum = anIsElemNum;
 434                   }
 435                   break;
 436                 case eHEXA8:
 437                 case eHEXA20:
 438                   aNbNodes = 8;
 439                   if(anIsElemNum)
 440                     anElement = myMesh->AddVolumeWithID(aNodeIds[0],
 441                                                         aNodeIds[1],
 442                                                         aNodeIds[2],
 443                                                         aNodeIds[3],
 444                                                         aNodeIds[4],
 445                                                         aNodeIds[5],
 446                                                         aNodeIds[6],
 447                                                         aNodeIds[7],
 448                                                         aCellInfo->GetElemNum(iElem));
 449                   if (!anElement) {
 450                     anElement = myMesh->AddVolume(FindNode(myMesh,aNodeIds[0]),
 451                                                   FindNode(myMesh,aNodeIds[1]),
 452                                                   FindNode(myMesh,aNodeIds[2]),
 453                                                   FindNode(myMesh,aNodeIds[3]),
 454                                                   FindNode(myMesh,aNodeIds[4]),
 455                                                   FindNode(myMesh,aNodeIds[5]),
 456                                                   FindNode(myMesh,aNodeIds[6]),
 457                                                   FindNode(myMesh,aNodeIds[7]));
 458                     isRenum = anIsElemNum;
 459                   }
 460                   break;
 461                 }
 462               }catch(const std::exception& exc){
 463                 //INFOS("Follow exception was cought:\n\t"<<exc.what());
 464                 aResult = DRS_FAIL;
 465               }catch(...){
 466                 //INFOS("Unknown exception was cought !!!");
 467                 aResult = DRS_FAIL;
 468               }
 469
 470               if (!anElement) {
 471                 aResult = DRS_WARN_SKIP_ELEM;
 472               }
 473               else {
 474                 if (isRenum) {
 475                   anIsElemNum = eFAUX;
 476                   takeNumbers = false;
 477                   if (aResult < DRS_WARN_RENUMBER)
 478                     aResult = DRS_WARN_RENUMBER;
 479                 }
 480                 if (myFamilies.find(aFamNum) != myFamilies.end()) {
 481                   // Save reference to this element from its family
 482                   myFamilies[aFamNum]->AddElement(anElement);
 483                   myFamilies[aFamNum]->SetType(anElement->GetType());
 484                 }
 485               }
 486             }
 487           }
 488         }
 489         break;
 490       }
 491     }
 492   }catch(const std::exception& exc){
 493     INFOS("Follow exception was cought:\n\t"<<exc.what());
 494     aResult = DRS_FAIL;
 495   }catch(...){
 496     INFOS("Unknown exception was cought !!!");
 497     aResult = DRS_FAIL;
 498   }
 499   if(MYDEBUG) MESSAGE("Perform - aResult status = "<<aResult);
 500   return aResult;
 501 }
 502
 503 list<string> DriverMED_R_SMESHDS_Mesh::GetMeshNames(Status& theStatus)
 504 {
 505   list<string> aMeshNames;
 506
 507   try {
 508     if(MYDEBUG) MESSAGE("GetMeshNames - myFile : " << myFile);
 509     theStatus = DRS_OK;
 510     PWrapper aMed = CrWrapper(myFile);
 511
 512     if (TInt aNbMeshes = aMed->GetNbMeshes()) {
 513       for (int iMesh = 0; iMesh < aNbMeshes; iMesh++) {
 514         // Reading the MED mesh
 515         //---------------------
 516         PMeshInfo aMeshInfo = aMed->GetPMeshInfo(iMesh+1);
 517         aMeshNames.push_back(aMeshInfo->GetName());
 518       }
 519     }
 520   }catch(const std::exception& exc){
 521     INFOS("Follow exception was cought:\n\t"<<exc.what());
 522     theStatus = DRS_FAIL;
 523   }catch(...){
 524     INFOS("Unknown exception was cought !!!");
 525     theStatus = DRS_FAIL;
 526   }
 527
 528   return aMeshNames;
 529 }
 530
 531 list<string> DriverMED_R_SMESHDS_Mesh::GetGroupNames()
 532 {
 533   list<string> aResult;
 534   set<string> aResGroupNames;
 535
 536   map<int, DriverMED_FamilyPtr>::iterator aFamsIter = myFamilies.begin();
 537   for (; aFamsIter != myFamilies.end(); aFamsIter++)
 538   {
 539     DriverMED_FamilyPtr aFamily = (*aFamsIter).second;
 540     const MED::TStringSet& aGroupNames = aFamily->GetGroupNames();
 541     set<string>::iterator aGrNamesIter = aGroupNames.begin();
 542     for (; aGrNamesIter != aGroupNames.end(); aGrNamesIter++)
 543     {
 544       string aName = *aGrNamesIter;
 545       // Check, if this is a Group or SubMesh name
 546 //if (aName.substr(0, 5) == string("Group")) {
 547         if (aResGroupNames.find(aName) == aResGroupNames.end()) {
 548           aResGroupNames.insert(aName);
 549           aResult.push_back(aName);
 550         }
 551 //    }
 552     }
 553   }
 554
 555   return aResult;
 556 }
 557
 558 void DriverMED_R_SMESHDS_Mesh::GetGroup(SMESHDS_Group* theGroup)
 559 {
 560   string aGroupName (theGroup->GetStoreName());
 561   if(MYDEBUG) MESSAGE("Get Group " << aGroupName);
 562
 563   map<int, DriverMED_FamilyPtr>::iterator aFamsIter = myFamilies.begin();
 564   for (; aFamsIter != myFamilies.end(); aFamsIter++)
 565   {
 566     DriverMED_FamilyPtr aFamily = (*aFamsIter).second;
 567     if (aFamily->MemberOf(aGroupName))
 568     {
 569       const set<const SMDS_MeshElement *>& anElements = aFamily->GetElements();
 570       set<const SMDS_MeshElement *>::iterator anElemsIter = anElements.begin();
 571       const SMDS_MeshElement * element = 0;
 572       for (; anElemsIter != anElements.end(); anElemsIter++)
 573       {
 574         element = *anElemsIter;
 575         theGroup->SMDSGroup().Add(element);
 576       }
 577       if ( element )
 578         theGroup->SetType( element->GetType() );
 579     }
 580   }
 581 }
 582
 583 void DriverMED_R_SMESHDS_Mesh::GetSubMesh (SMESHDS_SubMesh* theSubMesh,
 584                                            const int theId)
 585 {
 586   char submeshGrpName[ 30 ];
 587   sprintf( submeshGrpName, "SubMesh %d", theId );
 588   string aName (submeshGrpName);
 589   map<int, DriverMED_FamilyPtr>::iterator aFamsIter = myFamilies.begin();
 590   for (; aFamsIter != myFamilies.end(); aFamsIter++)
 591   {
 592     DriverMED_FamilyPtr aFamily = (*aFamsIter).second;
 593     if (aFamily->MemberOf(aName))
 594     {
 595       const set<const SMDS_MeshElement *>& anElements = aFamily->GetElements();
 596       set<const SMDS_MeshElement *>::iterator anElemsIter = anElements.begin();
 597       if (aFamily->GetType() == SMDSAbs_Node)
 598       {
 599         for (; anElemsIter != anElements.end(); anElemsIter++)
 600         {
 601           const SMDS_MeshNode* node = static_cast<const SMDS_MeshNode*>(*anElemsIter);
 602           theSubMesh->AddNode(node);
 603         }
 604       }
 605       else
 606       {
 607         for (; anElemsIter != anElements.end(); anElemsIter++)
 608         {
 609           theSubMesh->AddElement(*anElemsIter);
 610         }
 611       }
 612     }
 613   }
 614 }
 615
 616 void DriverMED_R_SMESHDS_Mesh::CreateAllSubMeshes ()
 617 {
 618   map<int, DriverMED_FamilyPtr>::iterator aFamsIter = myFamilies.begin();
 619   for (; aFamsIter != myFamilies.end(); aFamsIter++)
 620   {
 621     DriverMED_FamilyPtr aFamily = (*aFamsIter).second;
 622     MED::TStringSet aGroupNames = aFamily->GetGroupNames();
 623     set<string>::iterator aGrNamesIter = aGroupNames.begin();
 624     for (; aGrNamesIter != aGroupNames.end(); aGrNamesIter++)
 625     {
 626       string aName = *aGrNamesIter;
 627       // Check, if this is a Group or SubMesh name
 628       if (aName.substr(0, 7) == string("SubMesh"))
 629       {
 630         int Id = atoi(string(aName).substr(7).c_str());
 631         set<const SMDS_MeshElement *> anElements = aFamily->GetElements();
 632         set<const SMDS_MeshElement *>::iterator anElemsIter = anElements.begin();
 633         if (aFamily->GetType() == SMDSAbs_Node)
 634         {
 635           for (; anElemsIter != anElements.end(); anElemsIter++)
 636           {
 637             SMDS_MeshNode* node = const_cast<SMDS_MeshNode*>
 638               ( static_cast<const SMDS_MeshNode*>( *anElemsIter ));
 639             // find out a shape type
 640             TopoDS_Shape aShape = myMesh->IndexToShape( Id );
 641             int aShapeType = ( aShape.IsNull() ? -1 : aShape.ShapeType() );
 642             switch ( aShapeType ) {
 643             case TopAbs_FACE:
 644               myMesh->SetNodeOnFace(node, Id); break;
 645             case TopAbs_EDGE:
 646               myMesh->SetNodeOnEdge(node, Id); break;
 647             case TopAbs_VERTEX:
 648               myMesh->SetNodeOnVertex(node, Id); break;
 649             default:
 650               myMesh->SetNodeInVolume(node, Id);
 651             }
 652           }
 653         }
 654         else
 655         {
 656           for (; anElemsIter != anElements.end(); anElemsIter++)
 657           {
 658             myMesh->SetMeshElementOnShape(*anElemsIter, Id);
 659           }
 660         }
 661       }
 662     }
 663   }
 664 }