Salome HOME
Copyright update 2020
[modules/smesh.git] / src / DriverUNV / DriverUNV_W_SMDS_Mesh.cxx
index 01b112baa9058f229e1371eecd8ba304920722b8..6b507b00e84a3ec8c0a6ad1c79065cd9c0119431 100644 (file)
-using namespace std;
-#include "DriverUNV_W_SMDS_Mesh.h"
-
-#include "SMDS_MeshElement.hxx"
-#include "SMDS_MeshNode.hxx"
-
-
-
-
+// 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 <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()
-{
-       ;
-}
-
-DriverUNV_W_SMDS_Mesh::~DriverUNV_W_SMDS_Mesh()
-{
-       ;
-}
+#include "DriverUNV_W_SMDS_Mesh.h"
 
-void DriverUNV_W_SMDS_Mesh::SetMesh(SMDS_Mesh * aMesh)
-{
-       myMesh = aMesh;
-}
+#include "SMDS_Mesh.hxx"
+#include "SMESHDS_GroupBase.hxx"
 
-void DriverUNV_W_SMDS_Mesh::SetFile(string aFile)
-{
-       myFile = aFile;
-}
+#include "utilities.h"
 
-void DriverUNV_W_SMDS_Mesh::SetFileId(FILE * aFileId)
-{
-       myFileId = aFileId;
-}
+#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::SetMeshId(int aMeshId)
-{
-       myMeshId = aMeshId;
-}
+#include <Basics_Utils.hxx>
 
-void DriverUNV_W_SMDS_Mesh::Add()
-{
-       ;
-}
+using namespace std;
+using namespace UNV;
 
-void DriverUNV_W_SMDS_Mesh::Write()
+Driver_Mesh::Status DriverUNV_W_SMDS_Mesh::Perform()
 {
-
-       int nbNodes, nbCells;
-       int i;
-
-       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 ? */
-       int 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");
-
-       fprintf(myFileId, "%s\n", sUNV_SEPARATOR);
-       fprintf(myFileId, "%s\n", sNODE_UNV_ID);
-
-       SMDS_Iterator<const SMDS_MeshNode *> * itNodes=myMesh->nodesIterator();
-       while(itNodes->more())
-       {
-               const SMDS_MeshNode * node = itNodes->next();
-
-               fprintf(myFileId, sNODE_UNV_DESCR, node->GetID());
-               fprintf(myFileId, "%25.16E%25.16E%25.16E\n", node->X(), node->Y(),
-                       node->Z());
-       }
-       delete itNodes;
-       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_Iterator<const SMDS_MeshEdge *> * itEdges=myMesh->edgesIterator();
-       while(itEdges->more())
-       {
-               const SMDS_MeshElement * elem = itEdges->next();
-               SMDS_Iterator<const SMDS_MeshElement*> *itn=elem->nodesIterator();
-
-               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", itn->next()->GetID(),
-                               itn->next()->GetID());
-                       break;
-
-               case 3:
-                       fprintf(myFileId, sELT_BEAM_DESC1, elem->GetID(), 24,
-                               elem->NbNodes());
-                       fprintf(myFileId, sELT_BEAM_DESC2);
-                       fprintf(myFileId, "%10d%10d%10d\n", itn->next()->GetID(),
-                               itn->next()->GetID(), itn->next()->GetID());
-
-                       break;
-               }
-               delete itn;
-       }
-       delete itEdges;
-
-       SMDS_Iterator<const SMDS_MeshFace*> * itFaces=myMesh->facesIterator();
-       while(itFaces->more())
-       {
-               const SMDS_MeshElement * elem = itFaces->next();
-
-               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");
-               }
-
-               SMDS_Iterator<const SMDS_MeshElement*> *itn=elem->nodesIterator();
-               while(itn->more()) fprintf(myFileId, "%10d", itn->next()->GetID());
-               delete itn;
-
-               fprintf(myFileId, "\n");
-       }
-       delete itFaces;
-
-       SMDS_Iterator<const SMDS_MeshVolume*> * itVolumes=myMesh->volumesIterator();
-       while(itVolumes->more())
-       {
-               const SMDS_MeshElement * elem = itVolumes->next();
-
-               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;
-               }
-
-               SMDS_Iterator<const SMDS_MeshElement*> *itn=elem->nodesIterator();
-               while(itn->more()) fprintf(myFileId, "%10d", itn->next()->GetID());
-               delete itn;
-
-               fprintf(myFileId, "\n");
-       }
-       delete itVolumes;
-
-       fprintf(myFileId, "%s\n", sUNV_SEPARATOR);
-
-       fclose(myFileId);
+  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 );
+
+    {
+      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);
+    }
+
+    std::vector< size_t > elemLabelByID;
+    if ( !myGroups.empty() )
+      elemLabelByID.resize( myMesh->MaxElementID() + 1 );
+
+    {
+      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);
+        }
+      }
+
+      // -------------------
+      // 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);
+        }
+      }
+
+      // ---------------------
+      // 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);
+    }
+
+    // --------------------
+    // 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();
+      }
+    }
+
+    out_stream.flush();
+    out_stream.close();
+    if (!check_file(myFile))
+      EXCEPTION(runtime_error,"ERROR: Output file not good.");
+  }
+  catch(const std::exception& exc){
+    INFOS("Follow exception was cought:\n\t"<<exc.what());
+    throw;
+  }
+  catch(...){
+    INFOS("Unknown exception was cought !!!");
+    throw;
+  }
+  return aResult;
 }