src/DriverUNV/DriverUNV_R_SMDS_Mesh.cxx

   1 //  Copyright (C) 2007-2010  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 //  Copyright (C) 2003-2007  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.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22
  23 #include "DriverUNV_R_SMDS_Mesh.h"
  24 #include "SMDS_Mesh.hxx"
  25 #include "SMDS_MeshGroup.hxx"
  26
  27 #include "utilities.h"
  28
  29 #include "UNV2411_Structure.hxx"
  30 #include "UNV2412_Structure.hxx"
  31 #include "UNV2417_Structure.hxx"
  32 #include "UNV_Utilities.hxx"
  33
  34 #include <Basics_Utils.hxx>
  35
  36 using namespace std;
  37
  38
  39 #ifdef _DEBUG_
  40 static int MYDEBUG = 0;
  41 #else
  42 static int MYDEBUG = 0;
  43 #endif
  44
  45
  46 DriverUNV_R_SMDS_Mesh::~DriverUNV_R_SMDS_Mesh()
  47 {
  48   if (myGroup != 0)
  49     delete myGroup;
  50 }
  51
  52
  53 Driver_Mesh::Status DriverUNV_R_SMDS_Mesh::Perform()
  54 {
  55   Kernel_Utils::Localizer loc;
  56   Status aResult = DRS_OK;
  57   std::ifstream in_stream(myFile.c_str());
  58   try{
  59     {
  60       using namespace UNV2411;
  61       TDataSet aDataSet2411;
  62       UNV2411::Read(in_stream,aDataSet2411);
  63       if(MYDEBUG) MESSAGE("Perform - aDataSet2411.size() = "<<aDataSet2411.size());
  64       TDataSet::const_iterator anIter = aDataSet2411.begin();
  65       for(; anIter != aDataSet2411.end(); anIter++){
  66         const TNodeLab& aLabel = anIter->first;
  67         const TRecord& aRec = anIter->second;
  68         myMesh->AddNodeWithID(aRec.coord[0],aRec.coord[1],aRec.coord[2],aLabel);
  69       }
  70     }
  71     {
  72       using namespace UNV2412;
  73       in_stream.seekg(0);
  74       TDataSet aDataSet2412;
  75       UNV2412::Read(in_stream,aDataSet2412);
  76       TDataSet::const_iterator anIter = aDataSet2412.begin();
  77       if(MYDEBUG) MESSAGE("Perform - aDataSet2412.size() = "<<aDataSet2412.size());
  78       for(; anIter != aDataSet2412.end(); anIter++){
  79         SMDS_MeshElement* anElement = NULL;
  80         const TElementLab& aLabel = anIter->first;
  81         const TRecord& aRec = anIter->second;
  82         if(IsBeam(aRec.fe_descriptor_id)) {
  83           switch ( aRec.node_labels.size() ) {
  84           case 2: // edge with two nodes
  85             anElement = myMesh->AddEdgeWithID(aRec.node_labels[0],
  86                                               aRec.node_labels[1],
  87                                               aLabel);
  88             break;
  89           case 3: // quadratic edge (with 3 nodes)
  90             anElement = myMesh->AddEdgeWithID(aRec.node_labels[0],
  91                                               aRec.node_labels[2],
  92                                               aRec.node_labels[1],
  93                                               aLabel);
  94           }
  95         }
  96         else if(IsFace(aRec.fe_descriptor_id)) {
  97           switch(aRec.fe_descriptor_id){
  98           case 71: // TRI3
  99           case 72:
 100           case 74:
 101
 102           case 41: // Plane Stress Linear Triangle - TRI3
 103           case 91: // Thin Shell Linear Triangle - TRI3
 104             anElement = myMesh->AddFaceWithID(aRec.node_labels[0],
 105                                               aRec.node_labels[1],
 106                                               aRec.node_labels[2],
 107                                               aLabel);
 108             break;
 109
 110           case 42: // Plane Stress Quadratic Triangle - TRI6
 111           case 92: // Thin Shell Quadratic Triangle - TRI6
 112             anElement = myMesh->AddFaceWithID(aRec.node_labels[0],
 113                                               aRec.node_labels[2],
 114                                               aRec.node_labels[4],
 115                                               aRec.node_labels[1],
 116                                               aRec.node_labels[3],
 117                                               aRec.node_labels[5],
 118                                               aLabel);
 119             break;
 120
 121           case 44: // Plane Stress Linear Quadrilateral - QUAD4
 122           case 94: // Thin Shell   Linear Quadrilateral -  QUAD4
 123             anElement = myMesh->AddFaceWithID(aRec.node_labels[0],
 124                                               aRec.node_labels[1],
 125                                               aRec.node_labels[2],
 126                                               aRec.node_labels[3],
 127                                               aLabel);
 128             break;
 129
 130           case 45: // Plane Stress Quadratic Quadrilateral - QUAD8
 131           case 95: // Thin Shell   Quadratic Quadrilateral - QUAD8
 132             anElement = myMesh->AddFaceWithID(aRec.node_labels[0],
 133                                               aRec.node_labels[2],
 134                                               aRec.node_labels[4],
 135                                               aRec.node_labels[6],
 136                                               aRec.node_labels[1],
 137                                               aRec.node_labels[3],
 138                                               aRec.node_labels[5],
 139                                               aRec.node_labels[7],
 140                                               aLabel);
 141             break;
 142           }
 143         }
 144         else if(IsVolume(aRec.fe_descriptor_id)){
 145           switch(aRec.fe_descriptor_id){
 146
 147           case 111: // Solid Linear Tetrahedron - TET4
 148             anElement = myMesh->AddVolumeWithID(aRec.node_labels[0],
 149                                                 aRec.node_labels[2],
 150                                                 aRec.node_labels[1],
 151                                                 aRec.node_labels[3],
 152                                                 aLabel);
 153             break;
 154
 155           case 118: // Solid Quadratic Tetrahedron - TET10
 156             anElement = myMesh->AddVolumeWithID(aRec.node_labels[0],
 157                                                 aRec.node_labels[4],
 158                                                 aRec.node_labels[2],
 159
 160                                                 aRec.node_labels[9],
 161
 162                                                 aRec.node_labels[5],
 163                                                 aRec.node_labels[3],
 164                                                 aRec.node_labels[1],
 165
 166                                                 aRec.node_labels[6],
 167                                                 aRec.node_labels[8],
 168                                                 aRec.node_labels[7],
 169                                                 aLabel);
 170             break;
 171
 172           case 112: // Solid Linear Prism - PRISM6
 173             anElement = myMesh->AddVolumeWithID(aRec.node_labels[0],
 174                                                 aRec.node_labels[2],
 175                                                 aRec.node_labels[1],
 176                                                 aRec.node_labels[3],
 177                                                 aRec.node_labels[5],
 178                                                 aRec.node_labels[4],
 179                                                 aLabel);
 180             break;
 181
 182           case 113: // Solid Quadratic Prism - PRISM15
 183             anElement = myMesh->AddVolumeWithID(aRec.node_labels[0],
 184                                                 aRec.node_labels[4],
 185                                                 aRec.node_labels[2],
 186
 187                                                 aRec.node_labels[9],
 188                                                 aRec.node_labels[13],
 189                                                 aRec.node_labels[11],
 190
 191                                                 aRec.node_labels[5],
 192                                                 aRec.node_labels[3],
 193                                                 aRec.node_labels[1],
 194
 195                                                 aRec.node_labels[14],
 196                                                 aRec.node_labels[12],
 197                                                 aRec.node_labels[10],
 198
 199                                                 aRec.node_labels[6],
 200                                                 aRec.node_labels[8],
 201                                                 aRec.node_labels[7],
 202                                                 aLabel);
 203             break;
 204
 205           case 115: // Solid Linear Brick - HEX8
 206             anElement = myMesh->AddVolumeWithID(aRec.node_labels[0],
 207                                                 aRec.node_labels[3],
 208                                                 aRec.node_labels[2],
 209                                                 aRec.node_labels[1],
 210                                                 aRec.node_labels[4],
 211                                                 aRec.node_labels[7],
 212                                                 aRec.node_labels[6],
 213                                                 aRec.node_labels[5],
 214                                                 aLabel);
 215             break;
 216
 217           case 116: // Solid Quadratic Brick - HEX20
 218             anElement = myMesh->AddVolumeWithID(aRec.node_labels[0],
 219                                                 aRec.node_labels[6],
 220                                                 aRec.node_labels[4],
 221                                                 aRec.node_labels[2],
 222
 223                                                 aRec.node_labels[12],
 224                                                 aRec.node_labels[18],
 225                                                 aRec.node_labels[16],
 226                                                 aRec.node_labels[14],
 227
 228                                                 aRec.node_labels[7],
 229                                                 aRec.node_labels[5],
 230                                                 aRec.node_labels[3],
 231                                                 aRec.node_labels[1],
 232
 233                                                 aRec.node_labels[19],
 234                                                 aRec.node_labels[17],
 235                                                 aRec.node_labels[15],
 236                                                 aRec.node_labels[13],
 237
 238                                                 aRec.node_labels[8],
 239                                                 aRec.node_labels[11],
 240                                                 aRec.node_labels[10],
 241                                                 aRec.node_labels[9],
 242                                                 aLabel);
 243             break;
 244
 245           case 114: // pyramid of 13 nodes (quadratic) - PIRA13
 246             anElement = myMesh->AddVolumeWithID(aRec.node_labels[0],
 247                                                 aRec.node_labels[6],
 248                                                 aRec.node_labels[4],
 249                                                 aRec.node_labels[2],
 250                                                 aRec.node_labels[7],
 251                                                 aRec.node_labels[5],
 252                                                 aRec.node_labels[3],
 253                                                 aRec.node_labels[1],
 254
 255                                                 aRec.node_labels[8],
 256                                                 aRec.node_labels[11],
 257                                                 aRec.node_labels[10],
 258                                                 aRec.node_labels[9],
 259                                                 aRec.node_labels[12],
 260                                                 aLabel);
 261             break;
 262
 263           }
 264         }
 265         //      if(!anElement)
 266         //        MESSAGE("DriverUNV_R_SMDS_Mesh::Perform - can not add element with ID = "<<aLabel<<" and type = "<<aRec.fe_descriptor_id);
 267       }
 268     }
 269     {
 270       using namespace UNV2417;
 271       in_stream.seekg(0);
 272       TDataSet aDataSet2417;
 273       UNV2417::Read(in_stream,aDataSet2417);
 274       if(MYDEBUG) MESSAGE("Perform - aDataSet2417.size() = "<<aDataSet2417.size());
 275       if  (aDataSet2417.size() > 0) {
 276         myGroup = new SMDS_MeshGroup(myMesh);
 277         TDataSet::const_iterator anIter = aDataSet2417.begin();
 278         for(; anIter != aDataSet2417.end(); anIter++){
 279           const TGroupId& aLabel = anIter->first;
 280           const TRecord& aRec = anIter->second;
 281
 282           int aNodesNb = aRec.NodeList.size();
 283           int aElementsNb = aRec.ElementList.size();
 284
 285           bool useSuffix = ((aNodesNb > 0) && (aElementsNb > 0));
 286           int i;
 287           if (aNodesNb > 0) {
 288             SMDS_MeshGroup* aNodesGroup = (SMDS_MeshGroup*) myGroup->AddSubGroup(SMDSAbs_Node);
 289             std::string aGrName = (useSuffix) ? aRec.GroupName + "_Nodes" : aRec.GroupName;
 290             int i = aGrName.find( "\r" );
 291             if (i > 0)
 292               aGrName.erase (i, 2);
 293             myGroupNames.insert(TGroupNamesMap::value_type(aNodesGroup, aGrName));
 294             myGroupId.insert(TGroupIdMap::value_type(aNodesGroup, aLabel));
 295
 296             for (i = 0; i < aNodesNb; i++) {
 297               const SMDS_MeshNode* aNode = myMesh->FindNode(aRec.NodeList[i]);
 298               if (aNode)
 299                 aNodesGroup->Add(aNode);
 300             }
 301           }
 302           if (aElementsNb > 0){
 303             SMDS_MeshGroup* aEdgesGroup = 0;
 304             SMDS_MeshGroup* aFacesGroup = 0;
 305             SMDS_MeshGroup* aVolumeGroup = 0;
 306             bool createdGroup = false;
 307
 308             for (i = 0; i < aElementsNb; i++) {
 309               const SMDS_MeshElement* aElement = myMesh->FindElement(aRec.ElementList[i]);
 310               if (aElement) {
 311                 switch (aElement->GetType()) {
 312                 case SMDSAbs_Edge:
 313                   if (!aEdgesGroup) {
 314                     aEdgesGroup = (SMDS_MeshGroup*) myGroup->AddSubGroup(SMDSAbs_Edge);
 315                     if (!useSuffix && createdGroup) useSuffix = true;
 316                     std::string aEdgesGrName = (useSuffix) ? aRec.GroupName + "_Edges" : aRec.GroupName;
 317                     int i = aEdgesGrName.find( "\r" );
 318                     if (i > 0)
 319                       aEdgesGrName.erase (i, 2);
 320                     myGroupNames.insert(TGroupNamesMap::value_type(aEdgesGroup, aEdgesGrName));
 321                     myGroupId.insert(TGroupIdMap::value_type(aEdgesGroup, aLabel));
 322                     createdGroup = true;
 323                   }
 324                   aEdgesGroup->Add(aElement);
 325                   break;
 326                 case SMDSAbs_Face:
 327                   if (!aFacesGroup) {
 328                     aFacesGroup = (SMDS_MeshGroup*) myGroup->AddSubGroup(SMDSAbs_Face);
 329                     if (!useSuffix && createdGroup) useSuffix = true;
 330                     std::string aFacesGrName = (useSuffix) ? aRec.GroupName + "_Faces" : aRec.GroupName;
 331                     int i = aFacesGrName.find( "\r" );
 332                     if (i > 0)
 333                       aFacesGrName.erase (i, 2);
 334                     myGroupNames.insert(TGroupNamesMap::value_type(aFacesGroup, aFacesGrName));
 335                     myGroupId.insert(TGroupIdMap::value_type(aFacesGroup, aLabel));
 336                     createdGroup = true;
 337                   }
 338                   aFacesGroup->Add(aElement);
 339                   break;
 340                 case SMDSAbs_Volume:
 341                   if (!aVolumeGroup) {
 342                     aVolumeGroup = (SMDS_MeshGroup*) myGroup->AddSubGroup(SMDSAbs_Volume);
 343                     if (!useSuffix && createdGroup) useSuffix = true;
 344                     std::string aVolumeGrName = (useSuffix) ? aRec.GroupName + "_Volumes" : aRec.GroupName;
 345                     int i = aVolumeGrName.find( "\r" );
 346                     if (i > 0)
 347                       aVolumeGrName.erase (i, 2);
 348                     myGroupNames.insert(TGroupNamesMap::value_type(aVolumeGroup, aVolumeGrName));
 349                     myGroupId.insert(TGroupIdMap::value_type(aVolumeGroup, aLabel));
 350                     createdGroup = true;
 351                   }
 352                   aVolumeGroup->Add(aElement);
 353                   break;
 354                 }
 355               }
 356             }
 357           }
 358         }
 359       }
 360     }
 361   }
 362   catch(const std::exception& exc){
 363     INFOS("Follow exception was cought:\n\t"<<exc.what());
 364   }
 365   catch(...){
 366     INFOS("Unknown exception was cought !!!");
 367   }
 368   return aResult;
 369 }