Salome HOME
Copyright update 2020
[modules/smesh.git] / src / DriverUNV / DriverUNV_W_SMDS_Mesh.cxx
index 6bf42e674fe428a4c2537c31c417f626c7238905..6b507b00e84a3ec8c0a6ad1c79065cd9c0119431 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 <algorithm>
+
 #include "DriverUNV_W_SMDS_Mesh.h"
 
-#include "SMDS_MeshElement.hxx"
-#include "SMDS_MeshNode.hxx"
-#include "SMDS_MeshEdgesIterator.hxx"
-#include "SMDS_MeshFacesIterator.hxx"
-#include "SMDS_MeshNodesIterator.hxx"
-#include "SMDS_MeshVolumesIterator.hxx"
-
-#include <utilities.h>
-
-#define sNODE_UNV_ID "  2411"
-#define sELT_UNV_ID  "  2412"
-#define sUNV_SEPARATOR "    -1"
-#define sNODE_UNV_DESCR "%10d         1         1        11\n"
-#define sELT_SURF_DESC  "%10d        %2d         1         1        11         %1d\n"
-#define sELT_VOLU_DESC  "%10d        %2d         1         1         9         %1d\n"
-#define sELT_BEAM_DESC1 "%10d        %2d         1         1         7         %1d\n"
-#define sELT_BEAM_DESC2 "         0         1         1\n"
-
-DriverUNV_W_SMDS_Mesh::DriverUNV_W_SMDS_Mesh() {
-;
-}
+#include "SMDS_Mesh.hxx"
+#include "SMESHDS_GroupBase.hxx"
 
-DriverUNV_W_SMDS_Mesh::~DriverUNV_W_SMDS_Mesh() {
-;
-}
+#include "utilities.h"
 
-void DriverUNV_W_SMDS_Mesh::SetMesh(Handle(SMDS_Mesh)& aMesh) {
-  myMesh = aMesh;
-}
+#include "UNV164_Structure.hxx"
+#include "UNV2411_Structure.hxx"
+#include "UNV2412_Structure.hxx"
+#include "UNV2417_Structure.hxx"
+#include "UNV2420_Structure.hxx"
+#include "UNV_Utilities.hxx"
 
-void DriverUNV_W_SMDS_Mesh::SetFile(string aFile) {
-  myFile = aFile;
-}
+#include <Basics_Utils.hxx>
 
-void DriverUNV_W_SMDS_Mesh::SetFileId(FILE* aFileId) {
-  myFileId = aFileId;
-}
+using namespace std;
+using namespace UNV;
+
+Driver_Mesh::Status DriverUNV_W_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::ofstream out_stream(aFile.c_str());
+#else
+  std::ofstream out_stream(myFile.c_str());
+#endif
+  try{
+
+    UNV164::Write( out_stream ); // unit system
+    UNV2420::Write( out_stream, myMeshName ); // Coordinate system
+
+    std::vector< size_t > nodeLabelByID;
+    if ( myMesh->HasNumerationHoles() )
+      nodeLabelByID.resize( myMesh->MaxNodeID() + 1 );
 
-void DriverUNV_W_SMDS_Mesh::SetMeshId(int aMeshId) {
-  myMeshId = aMeshId;
-}
+    {
+      using namespace UNV2411;
+      TDataSet aDataSet2411;
+      // -----------------------------------
+      // Storing SMDS nodes to the UNV file
+      // -----------------------------------
+      SMDS_NodeIteratorPtr aNodesIter = myMesh->nodesIterator();
+      TRecord aRec;
+      for ( aRec.label = 1; aNodesIter->more(); ++aRec.label )
+      {
+        const SMDS_MeshNode* aNode = aNodesIter->next();
+        // aRec.label    = aNode->GetID(); -- IPAL54452
+        if ( !nodeLabelByID.empty() )
+          nodeLabelByID[ aNode->GetID() ] = aRec.label;
+        aRec.coord[0] = aNode->X();
+        aRec.coord[1] = aNode->Y();
+        aRec.coord[2] = aNode->Z();
+        aDataSet2411.push_back( aRec );
+      }
+      UNV2411::Write(out_stream,aDataSet2411);
+    }
 
-void DriverUNV_W_SMDS_Mesh::Add() {
-  ;
-}
+    std::vector< size_t > elemLabelByID;
+    if ( !myGroups.empty() )
+      elemLabelByID.resize( myMesh->MaxElementID() + 1 );
 
-void DriverUNV_W_SMDS_Mesh::Write() {
+    {
+      using namespace UNV2412;
+      TDataSet aDataSet2412;
+      TRecord aRec;
+      aRec.label = 0;
+
+      // -------------------
+      // Storing SMDS Edges
+      // -------------------
+      if ( myMesh->NbEdges() )
+      {
+        SMDS_EdgeIteratorPtr anIter = myMesh->edgesIterator();
+        while ( anIter->more() )
+        {
+          const SMDS_MeshEdge* anElem = anIter->next();
+          // aRec.label = anElem->GetID();  -- IPAL54452
+          ++aRec.label;
+          if ( !elemLabelByID.empty() )
+            elemLabelByID[ anElem->GetID() ] = aRec.label;
+
+          aRec.fe_descriptor_id = anElem->IsQuadratic() ? 22 : 11;
+
+          SMDS_NodeIteratorPtr aNodesIter = anElem->nodesIteratorToUNV();
+          for ( aRec.node_labels.clear(); aNodesIter->more(); )
+          {
+            const SMDS_MeshNode* aNode = aNodesIter->next();
+            if ( nodeLabelByID.empty() )
+              aRec.node_labels.push_back( aNode->GetID() );
+            else
+              aRec.node_labels.push_back( nodeLabelByID[ aNode->GetID() ]);
+          }
+
+          aDataSet2412.push_back(aRec);
+        }
+      }
 
-  int nbNodes,nbCells;
-  int i;
+      // -------------------
+      // Storing SMDS Faces
+      // -------------------
+      if ( myMesh->NbFaces() )
+      {
+        SMDS_FaceIteratorPtr anIter = myMesh->facesIterator();
+        while ( anIter->more() )
+        {
+          const SMDS_MeshFace* anElem = anIter->next();
+          if ( anElem->IsPoly() ) continue;
+
+          SMDS_NodeIteratorPtr aNodesIter = anElem->nodesIteratorToUNV();
+          for ( aRec.node_labels.clear(); aNodesIter->more();  ) {
+            const SMDS_MeshNode* aNode = aNodesIter->next();
+            if ( nodeLabelByID.empty() )
+              aRec.node_labels.push_back( aNode->GetID() );
+            else
+              aRec.node_labels.push_back( nodeLabelByID[ aNode->GetID() ]);
+          }
+          switch ( anElem->NbNodes() ) {
+          case 3: aRec.fe_descriptor_id = 41; break;
+          case 4: aRec.fe_descriptor_id = 44; break;
+          case 6: aRec.fe_descriptor_id = 42; break;
+          case 7: aRec.fe_descriptor_id = 42; break;
+          case 8: aRec.fe_descriptor_id = 45; break;
+          case 9: aRec.fe_descriptor_id = 45; aRec.node_labels.resize( 8 ); break;
+          default:
+            continue;
+          }
+          // aRec.label = anElem->GetID(); -- IPAL54452
+          ++aRec.label;
+          if ( !elemLabelByID.empty() )
+            elemLabelByID[ anElem->GetID() ] = aRec.label;
+
+          aDataSet2412.push_back(aRec);
+        }
+      }
 
-  char* file2Read = (char*)myFile.c_str();
-  myFileId = fopen(file2Read,"w+");
-  if (myFileId < 0)
-    {
-      fprintf(stderr,">> ERREUR : ouverture du fichier %s \n",file2Read);
-      exit(EXIT_FAILURE);
-    }
-  SCRUTE(myMesh);
-  /****************************************************************************
-  *                       NOMBRES D'OBJETS                                    *
-  ****************************************************************************/
-  fprintf(stdout,"\n(****************************)\n");
-  fprintf(stdout,"(* INFORMATIONS GENERALES : *)\n");
-  fprintf(stdout,"(****************************)\n");
-
-  /* Combien de noeuds ? */
-  nbNodes = myMesh->NbNodes();
-
-  /* Combien de mailles, faces ou aretes ? */
-  Standard_Integer nb_of_nodes, nb_of_edges,nb_of_faces, nb_of_volumes;
-  nb_of_edges = myMesh->NbEdges();
-  nb_of_faces = myMesh->NbFaces();
-  nb_of_volumes = myMesh->NbVolumes();
-  nbCells = nb_of_edges + nb_of_faces + nb_of_volumes;
-  SCRUTE(nb_of_edges);
-  SCRUTE(nb_of_faces);
-  SCRUTE(nb_of_volumes);
-
-  fprintf(stdout,"%d %d\n",nbNodes,nbCells);
-  fprintf(myFileId,"%d %d\n",nbNodes,nbCells);
-
-  /****************************************************************************
-  *                       ECRITURE DES NOEUDS                                 *
-  ****************************************************************************/
-  fprintf(stdout,"\n(************************)\n");
-  fprintf(stdout,"(* NOEUDS DU MAILLAGE : *)\n");
-  fprintf(stdout,"(************************)\n");
-
-  SMDS_MeshNodesIterator itNodes(myMesh);
-
-  fprintf(myFileId,"%s\n", sUNV_SEPARATOR);
-  fprintf(myFileId,"%s\n", sNODE_UNV_ID  );
-
-  for (;itNodes.More();itNodes.Next()) {
-    const Handle(SMDS_MeshElement)& elem = itNodes.Value();
-    const Handle(SMDS_MeshNode   )& node = myMesh->GetNode(1, elem);
-
-    fprintf(myFileId, sNODE_UNV_DESCR, node->GetID());
-    fprintf(myFileId, "%25.16E%25.16E%25.16E\n", node->X(), node->Y(), node->Z());
-  }
-  fprintf(myFileId,"%s\n", sUNV_SEPARATOR);
-
-  /****************************************************************************
-  *                       ECRITURE DES ELEMENTS                                *
-  ****************************************************************************/
-  fprintf(stdout,"\n(**************************)\n");
-  fprintf(stdout,"(* ELEMENTS DU MAILLAGE : *)\n");
-  fprintf(stdout,"(**************************)");
-  /* Ecriture des connectivites, noms, numeros des mailles */
-
-  fprintf(myFileId,"%s\n", sUNV_SEPARATOR);
-  fprintf(myFileId,"%s\n", sELT_UNV_ID   );
-
-  SMDS_MeshEdgesIterator itEdges(myMesh);
-  for (;itEdges.More();itEdges.Next()) {
-    const Handle(SMDS_MeshElement)& elem = itEdges.Value();
-
-    switch (elem->NbNodes()) {
-      case 2 : {
-        fprintf(myFileId, sELT_BEAM_DESC1, elem->GetID(), 21, elem->NbNodes());
-        fprintf(myFileId, sELT_BEAM_DESC2);
-        fprintf(myFileId, "%10d%10d\n", elem->GetConnection(1), elem->GetConnection(2));
-        break;
-      } 
-      case 3 : {
-        fprintf(myFileId, sELT_BEAM_DESC1, elem->GetID(), 24, elem->NbNodes());
-        fprintf(myFileId, sELT_BEAM_DESC2);
-        fprintf(myFileId, "%10d%10d%10d\n",elem->GetConnection(1), elem->GetConnection(2), elem->GetConnection(3));
-        break;
+      // ---------------------
+      // Storing SMDS Volumes
+      // ---------------------
+      if ( myMesh->NbVolumes() )
+      {
+        SMDS_VolumeIteratorPtr anIter = myMesh->volumesIterator();
+        while ( anIter->more() )
+        {
+          const SMDS_MeshVolume* anElem = anIter->next();
+          if ( anElem->IsPoly() )
+            continue;
+          size_t aNbNodes = anElem->NbNodes();
+          switch( aNbNodes ) {
+          case 4:  aRec.fe_descriptor_id = 111; break;
+          case 6:  aRec.fe_descriptor_id = 112; break;
+          case 8:  aRec.fe_descriptor_id = 115; break;
+          case 10: aRec.fe_descriptor_id = 118; break;
+          case 13: aRec.fe_descriptor_id = 114; break;
+          case 15: aRec.fe_descriptor_id = 113; break;
+          case 20:
+          case 27: aRec.fe_descriptor_id = 116; aNbNodes = 20; break;
+          default:
+            continue;
+          }
+          // aRec.label = anElem->GetID(); -- IPAL54452
+          ++aRec.label;
+          if ( !elemLabelByID.empty() )
+            elemLabelByID[  anElem->GetID() ] = aRec.label;
+
+          aRec.node_labels.clear();
+          SMDS_NodeIteratorPtr aNodesIter = anElem->nodesIteratorToUNV();
+          while ( aNodesIter->more() && aRec.node_labels.size() < aNbNodes )
+          {
+            const SMDS_MeshElement* aNode = aNodesIter->next();
+            if ( nodeLabelByID.empty() )
+              aRec.node_labels.push_back( aNode->GetID() );
+            else
+              aRec.node_labels.push_back( nodeLabelByID[ aNode->GetID() ]);
+          }
+          aDataSet2412.push_back(aRec);
+        }
       }
+      UNV2412::Write(out_stream,aDataSet2412);
     }
-  }
 
-  SMDS_MeshFacesIterator itFaces(myMesh);
-  for (;itFaces.More();itFaces.Next()) {
-    const Handle(SMDS_MeshElement)& elem = itFaces.Value();
-
-    switch (elem->NbNodes()) {
-      case 3 :
-        // linear triangle
-        fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 74, elem->NbNodes());
-        break;
-      case 4 :
-        // linear quadrilateral
-        fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 71, elem->NbNodes());
-        break;
-      case 6 :
-        // parabolic triangle
-        fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 72, elem->NbNodes());
-        break;
-      case 8 :
-        // parabolic quadrilateral
-        fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 75, elem->NbNodes());
-        break;
-      default:
-        fprintf(myFileId, "element not registered\n");
+    // --------------------
+    // Storing SMDS Groups
+    // --------------------
+    {
+      using namespace UNV2417;
+      if ( myGroups.size() > 0 ) {
+        TRecord aRec;
+        TDataSet aDataSet2417;
+        TGroupList::const_iterator aIter = myGroups.begin();
+        for ( ; aIter != myGroups.end(); aIter++ )
+        {
+          SMESHDS_GroupBase* aGroupDS = *aIter;
+          aRec.GroupName = aGroupDS->GetStoreName();
+          aRec.NodeList.clear();
+          aRec.ElementList.clear();
+
+          SMDS_ElemIteratorPtr aIter = aGroupDS->GetElements();
+          if ( aGroupDS->GetType() == SMDSAbs_Node ) {
+            while ( aIter->more() ) {
+              const SMDS_MeshElement* aNode = aIter->next();
+              if ( nodeLabelByID.empty() )
+                aRec.NodeList.push_back( aNode->GetID() );
+              else
+                aRec.NodeList.push_back( nodeLabelByID[ aNode->GetID() ]);
+            }
+          }
+          else
+          {
+            while ( aIter->more() ) {
+              const SMDS_MeshElement* aElem = aIter->next();
+              if ( elemLabelByID.empty() )
+                aRec.ElementList.push_back( aElem->GetID() );
+              else
+                aRec.ElementList.push_back( elemLabelByID[ aElem->GetID() ]);
+            }
+          }
+          // 0019936: EDF 794 SMESH : Export UNV : Node color and group id
+          //aDataSet2417.insert(TDataSet::value_type(aGroupDS->GetID(), aRec));
+          aDataSet2417.insert(TDataSet::value_type(aGroupDS->GetID()+1, aRec));
+        }
+        UNV2417::Write(out_stream,aDataSet2417);
+        myGroups.clear();
+      }
     }
 
-    for (i=0;i<elem->NbNodes();i++)
-      fprintf(myFileId,"%10d",elem->GetConnection(i+1));
-    
-    fprintf(myFileId,"\n");
+    out_stream.flush();
+    out_stream.close();
+    if (!check_file(myFile))
+      EXCEPTION(runtime_error,"ERROR: Output file not good.");
   }
-
-  SMDS_MeshVolumesIterator itVolumes(myMesh);
-  for (;itVolumes.More();itVolumes.Next()) {
-    const Handle(SMDS_MeshElement)& elem = itVolumes.Value();
-
-    switch (elem->NbNodes()) {
-      case 4 : 
-        // linear tetrahedron
-        fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 111, elem->NbNodes());
-        break;
-      case 6 : 
-        // linear tetrahedron
-        fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 112, elem->NbNodes());
-        break;
-      case 8 : 
-        // linear brick
-        fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 115, elem->NbNodes());
-        break;
-    }
-
-    for (i=0;i<elem->NbNodes();i++)
-      fprintf(myFileId,"%10d",elem->GetConnection(i+1));
-    
-    fprintf(myFileId,"\n");
+  catch(const std::exception& exc){
+    INFOS("Follow exception was cought:\n\t"<<exc.what());
+    throw;
   }
-  fprintf(myFileId,"%s\n", sUNV_SEPARATOR);
-
-  fclose (myFileId);
+  catch(...){
+    INFOS("Unknown exception was cought !!!");
+    throw;
+  }
+  return aResult;
 }