Salome HOME
Copyright update 2020
[modules/smesh.git] / src / DriverUNV / DriverUNV_R_SMDS_Mesh.cxx
index b9de018aa527e1924b51caa26a54fbbe7b1c8d08..6eb1d66e3917254384a4653d742087d07c1fbf92 100644 (file)
-using namespace std;
+// Copyright (C) 2007-2020  CEA/DEN, EDF R&D, OPEN CASCADE
+//
+// Copyright (C) 2003-2007  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, or (at your option) any later version.
+//
+// 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.salome-platform.org/ or email : webmaster.salome@opencascade.com
+//
+
 #include "DriverUNV_R_SMDS_Mesh.h"
+#include "SMDS_Mesh.hxx"
+#include "SMDS_MeshGroup.hxx"
 
 #include "utilities.h"
 
-DriverUNV_R_SMDS_Mesh::DriverUNV_R_SMDS_Mesh() {
-;
-}
+#include "UNV164_Structure.hxx"
+#include "UNV2411_Structure.hxx"
+#include "UNV2412_Structure.hxx"
+#include "UNV2417_Structure.hxx"
+#include "UNV2420_Structure.hxx"
+#include "UNV_Utilities.hxx"
 
-DriverUNV_R_SMDS_Mesh::~DriverUNV_R_SMDS_Mesh() {
-;
-}
+#include <Basics_Utils.hxx>
 
-void DriverUNV_R_SMDS_Mesh::SetMesh(Handle(SMDS_Mesh)& aMesh) {
-  myMesh = aMesh;
-}
+using namespace std;
 
-void DriverUNV_R_SMDS_Mesh::SetFile(string aFile) {
-  myFile = aFile;
-}
 
-void DriverUNV_R_SMDS_Mesh::SetFileId(FILE* aFileId) {
-  myFileId = aFileId;
-}
+#ifdef _DEBUG_
+static int MYDEBUG = 0;
+#else
+static int MYDEBUG = 0;
+#endif
 
-void DriverUNV_R_SMDS_Mesh::SetMeshId(int aMeshId) {
-  myMeshId = aMeshId;
+namespace
+{
+  /*!
+   * \brief Move node coordinates to the global Cartesian CS
+   */
+  void transformNodes( UNV2411::TDataSet::const_iterator fromNode,
+                       UNV2411::TDataSet::const_iterator endNode,
+                       const UNV2420::TRecord &          csRecord )
+  {
+    const int csLabel = fromNode->exp_coord_sys_num;
+
+    UNV2411::TDataSet::const_iterator nodeIt;
+
+    // apply Transformation Matrix
+    if ( !csRecord.isIdentityMatrix() )
+    {
+      for ( nodeIt = fromNode; nodeIt != endNode; ++nodeIt )
+      {
+        const UNV2411::TRecord& nodeRec = *nodeIt;
+        if ( nodeRec.exp_coord_sys_num == csLabel )
+          csRecord.ApplyMatrix( (double*) nodeRec.coord );
+      }
+    }
+
+    // transform from Cylindrical CS
+    if ( csRecord.coord_sys_type == UNV2420::Cylindrical )
+    {
+      for ( nodeIt = fromNode; nodeIt != endNode; ++nodeIt )
+      {
+        const UNV2411::TRecord& nodeRec = *nodeIt;
+        if ( nodeRec.exp_coord_sys_num == csLabel )
+          csRecord.FromCylindricalCS( (double*) nodeRec.coord );
+      }
+    }
+    // transform from Spherical CS
+    else if ( csRecord.coord_sys_type == UNV2420::Spherical )
+    {
+      for ( nodeIt = fromNode; nodeIt != endNode; ++nodeIt )
+      {
+        const UNV2411::TRecord& nodeRec = *nodeIt;
+        if ( nodeRec.exp_coord_sys_num == csLabel )
+          csRecord.FromSphericalCS( (double*) nodeRec.coord );
+      }
+    }
+  }
 }
 
-void DriverUNV_R_SMDS_Mesh::Add() {
-  ;
+DriverUNV_R_SMDS_Mesh::~DriverUNV_R_SMDS_Mesh()
+{
+  TGroupNamesMap::iterator grp2name = myGroupNames.begin();
+  for ( ; grp2name != myGroupNames.end(); ++grp2name )
+    delete grp2name->first;
 }
 
-void DriverUNV_R_SMDS_Mesh::Read() {
+Driver_Mesh::Status DriverUNV_R_SMDS_Mesh::Perform()
+{
+  Kernel_Utils::Localizer loc;
+  Status aResult = DRS_OK;
+#if defined(WIN32) && defined(UNICODE)
+  std::wstring aFile = Kernel_Utils::utf8_decode_s(myFile);
+  std::ifstream in_stream(aFile.c_str());
+#else
+  std::ifstream in_stream(myFile.c_str());
+#endif
+  try
+  {
+    {
+      // Read Units
+      UNV164::TRecord aUnitsRecord;
+      UNV164::Read( in_stream, aUnitsRecord );
 
-  int cell=0,node=0,n1,n2,n3,n4,n_nodes,nodes[6],blockId,i;
-  char *s1,*s2,*s3;
-  string str1,str2,str3;
-  int i1=0;
-  bool ok, found_block2411, found_block2412;
+      // Read Coordinate systems
+      UNV2420::TDataSet aCoordSysDataSet;
+      UNV2420::Read(in_stream, myMeshName, aCoordSysDataSet);
 
-  /****************************************************************************
-  *                      OUVERTURE DU FICHIER EN LECTURE                      *
-  ****************************************************************************/
-  char* file2Read = (char*)myFile.c_str();
-  myFileId = fopen(file2Read,"r");
-  if (myFileId < 0)
-    {
-      fprintf(stderr,">> ERREUR : ouverture du fichier %s \n",file2Read);
-      exit(EXIT_FAILURE);
-    }
+      // Read nodes
+      using namespace UNV2411;
+      TDataSet aDataSet2411;
+      UNV2411::Read(in_stream,aDataSet2411);
+      if(MYDEBUG) MESSAGE("Perform - aDataSet2411.size() = "<<aDataSet2411.size());
 
-  s1 = (char*) malloc(sizeof(char)*100);
-  s2 = (char*) malloc(sizeof(char)*100);
-  s3 = (char*) malloc(sizeof(char)*100);
-
-  found_block2411 = false;
-  found_block2412 = false;
-
-  do {
-
-    while (i1==-1) {
-      fscanf(myFileId,"%d\n",&blockId);
-      switch (blockId) {
-      case 2411 : {
-      MESSAGE("BlockId "<<blockId);
-       fscanf(myFileId,"%d",&node);
-       //MESSAGE("Node "<<node);
-       while (node!=-1) {
-         fscanf(myFileId,"%d %d %d\n",&n1,&n2,&n3);
-         fscanf(myFileId,"%s %s %s\n",s1,s2,s3);
-         str1=string(s1);
-         str2=string(s2);
-         str3=string(s3);
-         if (str1.find("D")!=string::npos) str1.replace(str1.find("D"),1,"E");
-         if (str2.find("D")!=string::npos) str2.replace(str2.find("D"),1,"E");
-         if (str3.find("D")!=string::npos) str3.replace(str3.find("D"),1,"E");
-         ok = myMesh->AddNodeWithID(atof(str1.c_str()),atof(str2.c_str()),atof(str3.c_str()),node);
-         fscanf(myFileId,"%d",&node);
-       }
-       i1=0;
-       found_block2411 = true;
-       break;
-      }        
-      case 2412 : {
-      MESSAGE("BlockId "<<blockId);
-       fscanf(myFileId,"%d",&cell);
-       //MESSAGE("Cell "<<cell);
-       while (cell!=-1) {
-         fscanf(myFileId,"%d %d %d %d %d\n",&n1,&n2,&n3,&n4,&n_nodes);
-
-         if ((n1==71)||(n1==72)||(n1==74)||(n1==91)||(n1==92)) {//203
-           if (n_nodes==3) {
-             for (i=1;i<=n_nodes;i++)
-               fscanf(myFileId,"%d",&nodes[i-1]);
-             ok = myMesh->AddFaceWithID(nodes[0],nodes[1],nodes[2],cell);
-           }
-           else if (n_nodes==6) {//206
-             for (i=1;i<=n_nodes;i++)
-               fscanf(myFileId,"%d",&nodes[i-1]);
-             ok = myMesh->AddFaceWithID(nodes[0],nodes[2],nodes[4],cell);
-           }
-         }
-
-         else if ((n1==11)||(n1==21)||(n1==24)||(n1==25)) {//103
-           fgets(s2,100,myFileId);
-           if (n_nodes==3) {
-             for (i=1;i<=n_nodes;i++)
-               fscanf(myFileId,"%d",&nodes[i-1]);
-             ok = myMesh->AddEdgeWithID(nodes[0],nodes[1],cell);
-             //MESSAGE("in 103 "<<cell);
-           }
-           else if (n_nodes==2) {//102
-             for (i=1;i<=n_nodes;i++)
-               fscanf(myFileId,"%d",&nodes[i-1]);
-             ok = myMesh->AddEdgeWithID(nodes[0],nodes[1],cell);
-             //MESSAGE("in 102 "<<cell);
-           }
-         }
-
-         fscanf(myFileId,"\n");
-         fscanf(myFileId,"%d",&cell);
-       }
-
-       i1=0;
-       found_block2412 = true;
-       break;
+      // Move nodes in a global CS
+      if ( !aCoordSysDataSet.empty() )
+      {
+        UNV2420::TDataSet::const_iterator csIter = aCoordSysDataSet.begin();
+        for ( ; csIter != aCoordSysDataSet.end(); ++csIter )
+        {
+          // find any node in this CS
+          TDataSet::const_iterator nodeIter = aDataSet2411.begin();
+          for (; nodeIter != aDataSet2411.end(); nodeIter++)
+            if ( nodeIter->exp_coord_sys_num == csIter->coord_sys_label )
+            {
+              transformNodes( nodeIter, aDataSet2411.end(), *csIter );
+              break;
+            }
+        }
       }
-      case -1 : {
-       break;
+      // Move nodes to SI unit system
+      const double lenFactor = aUnitsRecord.factors[ UNV164::LENGTH_FACTOR ];
+      if ( lenFactor != 1. )
+      {
+        TDataSet::iterator nodeIter = aDataSet2411.begin(), nodeEnd;
+        for ( nodeEnd = aDataSet2411.end(); nodeIter != nodeEnd; nodeIter++)
+        {
+          UNV2411::TRecord& nodeRec = *nodeIter;
+          nodeRec.coord[0] *= lenFactor;
+          nodeRec.coord[1] *= lenFactor;
+          nodeRec.coord[2] *= lenFactor;
+        }
       }
-      default:
-       MESSAGE("BlockId "<<blockId);
-       i1=0;
-       break;
+
+      // Create nodes in the mesh
+      TDataSet::const_iterator anIter = aDataSet2411.begin();
+      for(; anIter != aDataSet2411.end(); anIter++)
+      {
+        const TRecord& aRec = *anIter;
+        myMesh->AddNodeWithID(aRec.coord[0],aRec.coord[1],aRec.coord[2],aRec.label);
       }
     }
-  
-    fscanf(myFileId,"%s\n",s1);
-    i1 = atoi(s1);
+    {
+      using namespace UNV2412;
+      TDataSet aDataSet2412;
+      UNV2412::Read(in_stream,aDataSet2412);
+      TDataSet::const_iterator anIter = aDataSet2412.begin();
+      if(MYDEBUG) MESSAGE("Perform - aDataSet2412.size() = "<<aDataSet2412.size());
+      for(; anIter != aDataSet2412.end(); anIter++)
+      {
+        SMDS_MeshElement* anElement = NULL;
+        const TRecord& aRec = *anIter;
+        if(IsBeam(aRec.fe_descriptor_id)) {
+          switch ( aRec.node_labels.size() ) {
+          case 2: // edge with two nodes
+            //MESSAGE("add edge " << aLabel << " " << aRec.node_labels[0] << " " << aRec.node_labels[1]);
+            anElement = myMesh->AddEdgeWithID(aRec.node_labels[0],
+                                              aRec.node_labels[1],
+                                              aRec.label);
+            break;
+          case 3: // quadratic edge (with 3 nodes)
+            //MESSAGE("add edge " << aRec.label << " " << aRec.node_labels[0] << " " << aRec.node_labels[1] << " " << aRec.node_labels[2]);
+            anElement = myMesh->AddEdgeWithID(aRec.node_labels[0],
+                                              aRec.node_labels[2],
+                                              aRec.node_labels[1],
+                                              aRec.label);
+          }
+        }
+        else if(IsFace(aRec.fe_descriptor_id)) {
+          //MESSAGE("add face " << aRec.label);
+          switch(aRec.fe_descriptor_id){
+          case 41: // Plane Stress Linear Triangle
+          case 51: // Plane Strain Linear Triangle
+          case 61: // Plate Linear Triangle
+          case 74: // Membrane Linear Triangle
+          case 81: // Axisymmetric Solid Linear Triangle
+          case 91: // Thin Shell Linear Triangle
+            anElement = myMesh->AddFaceWithID(aRec.node_labels[0],
+                                              aRec.node_labels[1],
+                                              aRec.node_labels[2],
+                                              aRec.label);
+            break;
 
-  }
-  while ((!feof(myFileId))&&((!found_block2411)||(!found_block2412)));
+          case 42: //  Plane Stress Parabolic Triangle
+          case 52: //  Plane Strain Parabolic Triangle
+          case 62: //  Plate Parabolic Triangle
+          case 72: //  Membrane Parabolic Triangle
+          case 82: //  Axisymmetric Solid Parabolic Triangle
+          case 92: //  Thin Shell Parabolic Triangle
+            if ( aRec.node_labels.size() == 7 )
+              anElement = myMesh->AddFaceWithID(aRec.node_labels[0],
+                                                aRec.node_labels[2],
+                                                aRec.node_labels[4],
+                                                aRec.node_labels[1],
+                                                aRec.node_labels[3],
+                                                aRec.node_labels[5],
+                                                aRec.node_labels[6],
+                                                aRec.label);
+            else
+              anElement = myMesh->AddFaceWithID(aRec.node_labels[0],
+                                                aRec.node_labels[2],
+                                                aRec.node_labels[4],
+                                                aRec.node_labels[1],
+                                                aRec.node_labels[3],
+                                                aRec.node_labels[5],
+                                                aRec.label);
+            break;
+
+          case 44: // Plane Stress Linear Quadrilateral
+          case 54: // Plane Strain Linear Quadrilateral
+          case 64: // Plate Linear Quadrilateral
+          case 71: // Membrane Linear Quadrilateral
+          case 84: // Axisymmetric Solid Linear Quadrilateral
+          case 94: // Thin Shell Linear Quadrilateral
+            anElement = myMesh->AddFaceWithID(aRec.node_labels[0],
+                                              aRec.node_labels[1],
+                                              aRec.node_labels[2],
+                                              aRec.node_labels[3],
+                                              aRec.label);
+            break;
+
+          case 45: // Plane Stress Parabolic Quadrilateral
+          case 55: // Plane Strain Parabolic Quadrilateral
+          case 65: // Plate Parabolic Quadrilateral
+          case 75: // Membrane Parabolic Quadrilateral
+          case 85: // Axisymmetric Solid Parabolic Quadrilateral
+          case 95: // Thin Shell Parabolic Quadrilateral
+            if ( aRec.node_labels.size() == 9 )
+              anElement = myMesh->AddFaceWithID(aRec.node_labels[0],
+                                                aRec.node_labels[2],
+                                                aRec.node_labels[4],
+                                                aRec.node_labels[6],
+                                                aRec.node_labels[1],
+                                                aRec.node_labels[3],
+                                                aRec.node_labels[5],
+                                                aRec.node_labels[7],
+                                                aRec.node_labels[8],
+                                                aRec.label);
+            else
+              anElement = myMesh->AddFaceWithID(aRec.node_labels[0],
+                                                aRec.node_labels[2],
+                                                aRec.node_labels[4],
+                                                aRec.node_labels[6],
+                                                aRec.node_labels[1],
+                                                aRec.node_labels[3],
+                                                aRec.node_labels[5],
+                                                aRec.node_labels[7],
+                                                aRec.label);
+            break;
+          }
+        }
+        else if(IsVolume(aRec.fe_descriptor_id)){
+          //MESSAGE("add volume " << aRec.label);
+          switch(aRec.fe_descriptor_id){
+
+          case 111: // Solid Linear Tetrahedron - TET4
+            anElement = myMesh->AddVolumeWithID(aRec.node_labels[0],
+                                                aRec.node_labels[2],
+                                                aRec.node_labels[1],
+                                                aRec.node_labels[3],
+                                                aRec.label);
+            break;
+
+          case 118: // Solid Quadratic Tetrahedron - TET10
+            anElement = myMesh->AddVolumeWithID(aRec.node_labels[0],
+                                                aRec.node_labels[4],
+                                                aRec.node_labels[2],
 
-  /****************************************************************************
-  *                      FERMETURE DU FICHIER                      *
-  ****************************************************************************/
-  free(s1);
-  free(s2);
-  free(s3);
-  fclose(myFileId);
+                                                aRec.node_labels[9],
 
+                                                aRec.node_labels[5],
+                                                aRec.node_labels[3],
+                                                aRec.node_labels[1],
+
+                                                aRec.node_labels[6],
+                                                aRec.node_labels[8],
+                                                aRec.node_labels[7],
+                                                aRec.label);
+            break;
+
+          case 112: // Solid Linear Prism - PRISM6
+            anElement = myMesh->AddVolumeWithID(aRec.node_labels[0],
+                                                aRec.node_labels[2],
+                                                aRec.node_labels[1],
+                                                aRec.node_labels[3],
+                                                aRec.node_labels[5],
+                                                aRec.node_labels[4],
+                                                aRec.label);
+            break;
+
+          case 113: // Solid Quadratic Prism - PRISM15
+            anElement = myMesh->AddVolumeWithID(aRec.node_labels[0],
+                                                aRec.node_labels[4],
+                                                aRec.node_labels[2],
+
+                                                aRec.node_labels[9],
+                                                aRec.node_labels[13],
+                                                aRec.node_labels[11],
+
+                                                aRec.node_labels[5],
+                                                aRec.node_labels[3],
+                                                aRec.node_labels[1],
+
+                                                aRec.node_labels[14],
+                                                aRec.node_labels[12],
+                                                aRec.node_labels[10],
+
+                                                aRec.node_labels[6],
+                                                aRec.node_labels[8],
+                                                aRec.node_labels[7],
+                                                aRec.label);
+            break;
+
+          case 115: // Solid Linear Brick - HEX8
+            anElement = myMesh->AddVolumeWithID(aRec.node_labels[0],
+                                                aRec.node_labels[3],
+                                                aRec.node_labels[2],
+                                                aRec.node_labels[1],
+                                                aRec.node_labels[4],
+                                                aRec.node_labels[7],
+                                                aRec.node_labels[6],
+                                                aRec.node_labels[5],
+                                                aRec.label);
+            break;
+
+          case 116: // Solid Quadratic Brick - HEX20
+            anElement = myMesh->AddVolumeWithID(aRec.node_labels[0],
+                                                aRec.node_labels[6],
+                                                aRec.node_labels[4],
+                                                aRec.node_labels[2],
+
+                                                aRec.node_labels[12],
+                                                aRec.node_labels[18],
+                                                aRec.node_labels[16],
+                                                aRec.node_labels[14],
+
+                                                aRec.node_labels[7],
+                                                aRec.node_labels[5],
+                                                aRec.node_labels[3],
+                                                aRec.node_labels[1],
+
+                                                aRec.node_labels[19],
+                                                aRec.node_labels[17],
+                                                aRec.node_labels[15],
+                                                aRec.node_labels[13],
+
+                                                aRec.node_labels[8],
+                                                aRec.node_labels[11],
+                                                aRec.node_labels[10],
+                                                aRec.node_labels[9],
+                                                aRec.label);
+            break;
+
+          case 114: // pyramid of 13 nodes (quadratic) - PIRA13
+            anElement = myMesh->AddVolumeWithID(aRec.node_labels[0],
+                                                aRec.node_labels[6],
+                                                aRec.node_labels[4],
+                                                aRec.node_labels[2],
+
+                                                aRec.node_labels[12],
+
+                                                aRec.node_labels[7],
+                                                aRec.node_labels[5],
+                                                aRec.node_labels[3],
+                                                aRec.node_labels[1],
+
+                                                aRec.node_labels[8],
+                                                aRec.node_labels[11],
+                                                aRec.node_labels[10],
+                                                aRec.node_labels[9],
+                                                aRec.label);
+            break;
+
+          }
+        }
+        if(!anElement)
+          MESSAGE("DriverUNV_R_SMDS_Mesh::Perform - can not add element with ID = "<<aRec.label<<" and type = "<<aRec.fe_descriptor_id);
+      }
+    }
+    {
+      using namespace UNV2417;
+      TDataSet aDataSet2417;
+      UNV2417::Read(in_stream,aDataSet2417);
+      if(MYDEBUG) MESSAGE("Perform - aDataSet2417.size() = "<<aDataSet2417.size());
+      if (aDataSet2417.size() > 0)
+      {
+        TDataSet::const_iterator anIter = aDataSet2417.begin();
+        for ( ; anIter != aDataSet2417.end(); anIter++ )
+        {
+          const TRecord& aRec = anIter->second;
+          int        aNodesNb = aRec.NodeList.size();
+          int     aElementsNb = aRec.ElementList.size();
+
+          bool useSuffix = ((aNodesNb > 0) && (aElementsNb > 0));
+          if ( aNodesNb > 0 )
+          {
+            SMDS_MeshGroup* aNodesGroup = new SMDS_MeshGroup( myMesh );
+            std::string aGrName = (useSuffix) ? aRec.GroupName + "_Nodes" : aRec.GroupName;
+            int i = aGrName.find( "\r" );
+            if (i > 0)
+              aGrName.erase (i, 2);
+            myGroupNames.insert( std::make_pair( aNodesGroup, aGrName ));
+
+            for ( int i = 0; i < aNodesNb; i++ )
+              if ( const SMDS_MeshNode* aNode = myMesh->FindNode( aRec.NodeList[i] ))
+                aNodesGroup->Add( aNode );
+          }
+          if ( aElementsNb > 0 )
+          {
+            std::vector< SMDS_MeshGroup* > aGroupVec( SMDSAbs_NbElementTypes, (SMDS_MeshGroup*)0 );
+            const char* aSuffix[] = { "", "", "_Edges", "_Faces", "_Volumes", "_0D", "_Balls" };
+            bool createdGroup = false;
+            for ( int i = 0; i < aElementsNb; i++)
+            {
+              const SMDS_MeshElement* aElement = myMesh->FindElement( aRec.ElementList[i] );
+              if ( !aElement ) continue;
+
+              SMDS_MeshGroup * & aGroup = aGroupVec[ aElement->GetType() ];
+              if ( !aGroup )
+              {
+                aGroup = new SMDS_MeshGroup( myMesh );
+                if (!useSuffix && createdGroup) useSuffix = true;
+                std::string aGrName = aRec.GroupName;
+                int i = aGrName.find( "\r" );
+                if ( i > 0 )
+                  aGrName.erase (i, 2);
+                if ( useSuffix )
+                  aGrName += aSuffix[ aElement->GetType() ];
+                myGroupNames.insert( std::make_pair( aGroup, aGrName ));
+                createdGroup = true;
+              }
+              aGroup->Add(aElement);
+            }
+          }
+        }
+      }
+    }
+  }
+  catch(const std::exception& exc){
+    INFOS("Follow exception was cought:\n\t"<<exc.what());
+  }
+  catch(...){
+    INFOS("Unknown exception was cought !!!");
+  }
+  if (myMesh)
+  {
+    myMesh->Modified();
+    myMesh->CompactMesh();
+  }
+  return aResult;
 }