Salome HOME
Fix for IPAL12172 - import volume group from UNV
[modules/smesh.git] / src / DriverUNV / DriverUNV_R_SMDS_Mesh.cxx
index b9de018aa527e1924b51caa26a54fbbe7b1c8d08..173339ac3a02cd57fdbab2bb1fc05b07d5f4cb19 100644 (file)
-using namespace std;
+// Copyright (C) 2005  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.salome-platform.org/
+//
 #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 "UNV2411_Structure.hxx"
+#include "UNV2412_Structure.hxx"
+#include "UNV2417_Structure.hxx"
+#include "UNV_Utilities.hxx"
 
-DriverUNV_R_SMDS_Mesh::~DriverUNV_R_SMDS_Mesh() {
-;
-}
+using namespace std;
 
-void DriverUNV_R_SMDS_Mesh::SetMesh(Handle(SMDS_Mesh)& aMesh) {
-  myMesh = aMesh;
-}
 
-void DriverUNV_R_SMDS_Mesh::SetFile(string aFile) {
-  myFile = aFile;
-}
+#ifdef _DEBUG_
+static int MYDEBUG = 0;
+#else
+static int MYDEBUG = 0;
+#endif
 
-void DriverUNV_R_SMDS_Mesh::SetFileId(FILE* aFileId) {
-  myFileId = aFileId;
-}
 
-void DriverUNV_R_SMDS_Mesh::SetMeshId(int aMeshId) {
-  myMeshId = aMeshId;
+DriverUNV_R_SMDS_Mesh::~DriverUNV_R_SMDS_Mesh()
+{
+  if (myGroup != 0) 
+    delete myGroup;
 }
 
-void DriverUNV_R_SMDS_Mesh::Add() {
-  ;
-}
-
-void DriverUNV_R_SMDS_Mesh::Read() {
-
-  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;
 
-  /****************************************************************************
-  *                      OUVERTURE DU FICHIER EN LECTURE                      *
-  ****************************************************************************/
-  char* file2Read = (char*)myFile.c_str();
-  myFileId = fopen(file2Read,"r");
-  if (myFileId < 0)
+Driver_Mesh::Status DriverUNV_R_SMDS_Mesh::Perform()
+{
+  Status aResult = DRS_OK;
+  std::ifstream in_stream(myFile.c_str());
+  try{
     {
-      fprintf(stderr,">> ERREUR : ouverture du fichier %s \n",file2Read);
-      exit(EXIT_FAILURE);
+      using namespace UNV2411;
+      TDataSet aDataSet2411;
+      UNV2411::Read(in_stream,aDataSet2411);
+      if(MYDEBUG) MESSAGE("Perform - aDataSet2411.size() = "<<aDataSet2411.size());
+      TDataSet::const_iterator anIter = aDataSet2411.begin();
+      for(; anIter != aDataSet2411.end(); anIter++){
+       const TNodeLab& aLabel = anIter->first;
+       const TRecord& aRec = anIter->second;
+       myMesh->AddNodeWithID(aRec.coord[0],aRec.coord[1],aRec.coord[2],aLabel);
+      }
     }
-
-  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);
+    {
+      using namespace UNV2412;
+      in_stream.seekg(0);
+      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 TElementLab& aLabel = anIter->first;
+       const TRecord& aRec = anIter->second;
+       if(IsBeam(aRec.fe_descriptor_id)) {
+          switch ( aRec.node_labels.size() ) {
+          case 2: // edge with two nodes
+            anElement = myMesh->AddEdgeWithID(aRec.node_labels[0],
+                                              aRec.node_labels[1],
+                                              aLabel);
+            break;
+          case 3: // quadratic edge (with 3 nodes)
+            anElement = myMesh->AddEdgeWithID(aRec.node_labels[0],
+                                              aRec.node_labels[2],
+                                              aRec.node_labels[1],
+                                              aLabel);
+          }
        }
-       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(IsFace(aRec.fe_descriptor_id)) {
+         switch(aRec.fe_descriptor_id){
+         case 71: // TRI3
+         case 72:
+         case 74:
+           
+         case 41: // Plane Stress Linear Triangle - TRI3
+         case 91: // Thin Shell Linear Triangle - TRI3
+           anElement = myMesh->AddFaceWithID(aRec.node_labels[0],
+                                             aRec.node_labels[1],
+                                             aRec.node_labels[2],
+                                             aLabel);
+           break;
+           
+         case 42: // Plane Stress Quadratic Triangle - TRI6
+         case 92: // Thin Shell Quadratic Triangle - TRI6
+           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],
+                                             aLabel);
+           break;
+           
+         case 44: // Plane Stress Linear Quadrilateral - QUAD4
+         case 94: // Thin Shell   Linear Quadrilateral -  QUAD4
+           anElement = myMesh->AddFaceWithID(aRec.node_labels[0],
+                                             aRec.node_labels[1],
+                                             aRec.node_labels[2],
+                                             aRec.node_labels[3],
+                                             aLabel);
+           break;
+           
+         case 45: // Plane Stress Quadratic Quadrilateral - QUAD8
+         case 95: // Thin Shell   Quadratic Quadrilateral - QUAD8
+           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],
+                                             aLabel);
+           break;
          }
+       }
+        else if(IsVolume(aRec.fe_descriptor_id)){
+         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],
+                                               aLabel);
+           break;
 
-         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);
-           }
-         }
+         case 118: // Solid Quadratic Tetrahedron - TET10
+           anElement = myMesh->AddVolumeWithID(aRec.node_labels[0],
+                                               aRec.node_labels[4],
+                                               aRec.node_labels[2],
 
-         fscanf(myFileId,"\n");
-         fscanf(myFileId,"%d",&cell);
-       }
+                                               aRec.node_labels[9],
 
-       i1=0;
-       found_block2412 = true;
-       break;
-      }
-      case -1 : {
-       break;
-      }
-      default:
-       MESSAGE("BlockId "<<blockId);
-       i1=0;
-       break;
+                                               aRec.node_labels[5],
+                                               aRec.node_labels[3],
+                                                aRec.node_labels[1],
+
+                                                aRec.node_labels[6],
+                                               aRec.node_labels[8],
+                                               aRec.node_labels[7],
+                                               aLabel);
+           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],
+                                               aLabel);
+           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],
+                                               aLabel);
+           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],
+                                               aLabel);
+           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],
+                                               aLabel);
+           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[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.node_labels[12],
+                                               aLabel);
+           break;
+
+         }
+       }
+       //      if(!anElement)
+       //        MESSAGE("DriverUNV_R_SMDS_Mesh::Perform - can not add element with ID = "<<aLabel<<" and type = "<<aRec.fe_descriptor_id);
       }
     }
-  
-    fscanf(myFileId,"%s\n",s1);
-    i1 = atoi(s1);
+    {
+      using namespace UNV2417;      
+      in_stream.seekg(0);
+      TDataSet aDataSet2417;
+      UNV2417::Read(in_stream,aDataSet2417);
+      if(MYDEBUG) MESSAGE("Perform - aDataSet2417.size() = "<<aDataSet2417.size());
+      if  (aDataSet2417.size() > 0) {
+       myGroup = new SMDS_MeshGroup(myMesh);
+       TDataSet::const_iterator anIter = aDataSet2417.begin();
+       for(; anIter != aDataSet2417.end(); anIter++){
+         const TGroupId& aLabel = anIter->first;
+         const TRecord& aRec = anIter->second;
 
-  }
-  while ((!feof(myFileId))&&((!found_block2411)||(!found_block2412)));
+         int aNodesNb = aRec.NodeList.size();
+         int aElementsNb = aRec.ElementList.size();
+
+         bool useSuffix = ((aNodesNb > 0) && (aElementsNb > 0));
+         int i;
+         if (aNodesNb > 0) {
+           SMDS_MeshGroup* aNodesGroup = (SMDS_MeshGroup*) myGroup->AddSubGroup(SMDSAbs_Node);
+           std::string aGrName = (useSuffix) ? aRec.GroupName + "_Nodes" : aRec.GroupName;
+           myGroupNames.insert(TGroupNamesMap::value_type(aNodesGroup, aGrName));
+           myGroupId.insert(TGroupIdMap::value_type(aNodesGroup, aLabel));
 
-  /****************************************************************************
-  *                      FERMETURE DU FICHIER                      *
-  ****************************************************************************/
-  free(s1);
-  free(s2);
-  free(s3);
-  fclose(myFileId);
+           for (i = 0; i < aNodesNb; i++) {
+             const SMDS_MeshNode* aNode = myMesh->FindNode(aRec.NodeList[i]);
+             if (aNode)
+               aNodesGroup->Add(aNode);
+           }
+         }
+         if (aElementsNb > 0){
+           SMDS_MeshGroup* aEdgesGroup = 0;
+           SMDS_MeshGroup* aFacesGroup = 0;
+           SMDS_MeshGroup* aVolumeGroup = 0;
+           bool createdGroup = false;
 
+           for (i = 0; i < aElementsNb; i++) {
+             const SMDS_MeshElement* aElement = myMesh->FindElement(aRec.ElementList[i]);
+             if (aElement) {
+               switch (aElement->GetType()) {
+               case SMDSAbs_Edge:
+                 if (!aEdgesGroup) {
+                   aEdgesGroup = (SMDS_MeshGroup*) myGroup->AddSubGroup(SMDSAbs_Edge);
+                   if (!useSuffix && createdGroup) useSuffix = true;
+                   std::string aEdgesGrName = (useSuffix) ? aRec.GroupName + "_Edges" : aRec.GroupName;
+                   myGroupNames.insert(TGroupNamesMap::value_type(aEdgesGroup, aEdgesGrName));
+                   myGroupId.insert(TGroupIdMap::value_type(aEdgesGroup, aLabel));
+                   createdGroup = true;
+                 }
+                 aEdgesGroup->Add(aElement);
+                 break;
+               case SMDSAbs_Face:
+                 if (!aFacesGroup) {
+                   aFacesGroup = (SMDS_MeshGroup*) myGroup->AddSubGroup(SMDSAbs_Face);
+                   if (!useSuffix && createdGroup) useSuffix = true;
+                   std::string aFacesGrName = (useSuffix) ? aRec.GroupName + "_Faces" : aRec.GroupName;
+                   myGroupNames.insert(TGroupNamesMap::value_type(aFacesGroup, aFacesGrName));
+                   myGroupId.insert(TGroupIdMap::value_type(aFacesGroup, aLabel));
+                   createdGroup = true;
+                 }
+                 aFacesGroup->Add(aElement);
+                 break;
+               case SMDSAbs_Volume:
+                 if (!aVolumeGroup) {
+                   aVolumeGroup = (SMDS_MeshGroup*) myGroup->AddSubGroup(SMDSAbs_Volume);
+                   if (!useSuffix && createdGroup) useSuffix = true;
+                   std::string aVolumeGrName = (useSuffix) ? aRec.GroupName + "_Volumes" : aRec.GroupName;
+                   myGroupNames.insert(TGroupNamesMap::value_type(aVolumeGroup, aVolumeGrName));
+                   myGroupId.insert(TGroupIdMap::value_type(aVolumeGroup, aLabel));
+                   createdGroup = true;
+                 }
+                 aVolumeGroup->Add(aElement);
+                 break;
+               }
+             } 
+           }
+         }
+       }
+      }
+    } 
+  }
+  catch(const std::exception& exc){
+    INFOS("Follow exception was cought:\n\t"<<exc.what());
+  }
+  catch(...){
+    INFOS("Unknown exception was cought !!!");
+  }
+  return aResult;
 }