Salome HOME
IPAL54452: Compact mesh after merge operations
[modules/smesh.git] / src / DriverDAT / DriverDAT_W_SMDS_Mesh.cxx
index d99a124fa89f4500f8ce028f6234da23a92c14b1..8157374954176867f4a0093b794a0b3420cc9b2a 100644 (file)
-//  SMESH DriverDAT : driver to read and write 'dat' files
+// Copyright (C) 2007-2019  CEA/DEN, EDF R&D, OPEN CASCADE
 //
-//  Copyright (C) 2003  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.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org 
+// 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
 //
-//  File   : DriverDAT_W_SMDS_Mesh.cxx
-//  Module : SMESH
-
-#include "DriverDAT_W_SMDS_Mesh.h"
-#include "SMDS_MeshElement.hxx"
-#include "SMDS_MeshNode.hxx"
-#include "utilities.h"
 
-extern "C"
-{
+#include <stdio.h>
 
-/**
- * Factory function which will be called by SMESHDriver
- */
-void * SMESH_createDATMeshWriter()
-{
-       return new DriverDAT_W_SMDS_Mesh();
-}
-}
-
-DriverDAT_W_SMDS_Mesh::DriverDAT_W_SMDS_Mesh()
-{
-}
+#include "DriverDAT_W_SMDS_Mesh.h"
 
-DriverDAT_W_SMDS_Mesh::~DriverDAT_W_SMDS_Mesh()
-{
-       ;
-}
+#include "SMDS_Mesh.hxx"
 
-void DriverDAT_W_SMDS_Mesh::SetMesh(SMDS_Mesh * aMesh)
-{
-       myMesh = aMesh;
-}
+#include "utilities.h"
 
-void DriverDAT_W_SMDS_Mesh::SetFile(string aFile)
-{
-       myFile = aFile;
-}
+#include <Basics_Utils.hxx>
 
-void DriverDAT_W_SMDS_Mesh::SetFileId(FILE * aFileId)
-{
-       myFileId = aFileId;
-}
+using namespace std;
 
-void DriverDAT_W_SMDS_Mesh::SetMeshId(int aMeshId)
+Driver_Mesh::Status DriverDAT_W_SMDS_Mesh::Perform()
 {
-       myMeshId = aMeshId;
-}
+  Kernel_Utils::Localizer loc;
+  Status aResult = DRS_OK;
+
+  int nbNodes, nbCells;
+#if defined(WIN32) && defined(UNICODE)
+  std::wstring file2Read = Kernel_Utils::utf8_decode_s(myFile);
+  FILE* aFileId = _wfopen(file2Read.c_str(), L"w+");
+
+#else
+  char *file2Read = (char *)myFile.c_str();
+  FILE* aFileId = fopen(file2Read, "w+");
+#endif
+  if ( !aFileId )
+  {
+    fprintf(stderr, ">> ERREUR : ouverture du fichier %s \n", file2Read);
+    return DRS_FAIL;
+  }
+  SCRUTE(myMesh);
+  /****************************************************************************
+   *                       NOMBRES D'OBJETS                                    *
+   ****************************************************************************/
 
-void DriverDAT_W_SMDS_Mesh::Add()
-{
-       MESSAGE("Adding a mesh to a DAT document. As DAT do not support more than one mesh in a file, the previous mesh is deleted");
-       Write();
-}
+  /* Combien de noeuds ? */
+  nbNodes = myMesh->NbNodes();
 
-void DriverDAT_W_SMDS_Mesh::Write()
-{
-       int nbNodes, nbCells;
-       int i;
+  /* Combien de mailles, faces ou aretes ? */
+  int 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);
 
-       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);
-       }
+  //fprintf(stdout, "%d %d\n", nbNodes, nbCells);
+  fprintf(aFileId, "%d %d\n", nbNodes, nbCells);
 
   /****************************************************************************
-  *                       NOMBRES D'OBJETS                                    *
-  ****************************************************************************/
-       fprintf(stdout, "\n(****************************)\n");
-       fprintf(stdout, "(* INFORMATIONS GENERALES : *)\n");
-       fprintf(stdout, "(****************************)\n");
-
-       /* Combien de noeuds ? */
-       nbNodes = myMesh->NbNodes();
+   *                       ECRITURE DES NOEUDS                                 *
+   ****************************************************************************/
 
-       /* 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;
+  std::vector< size_t > nodeNumByID;
+  if ( myMesh->HasNumerationHoles() )
+    nodeNumByID.resize( myMesh->MaxNodeID() + 1 );
 
-       fprintf(stdout, "%d %d\n", nbNodes, nbCells);
-       fprintf(myFileId, "%d %d\n", nbNodes, nbCells);
+  int num;
+  SMDS_NodeIteratorPtr itNodes=myMesh->nodesIterator();
+  for ( num = 1; itNodes->more(); ++num )
+  {
+    const SMDS_MeshNode * node = itNodes->next();
+    fprintf(aFileId, "%d %.14e %.14e %.14e\n", num, node->X(), node->Y(), node->Z());
 
-  /****************************************************************************
-  *                       ECRITURE DES NOEUDS                                 *
-  ****************************************************************************/
-       fprintf(stdout, "\n(************************)\n");
-       fprintf(stdout, "(* NOEUDS DU MAILLAGE : *)\n");
-       fprintf(stdout, "(************************)\n");
-
-       SMDS_Iterator<const SMDS_MeshNode *> * itNodes=myMesh->nodesIterator();
-       while(itNodes->more())
-       {               
-               const SMDS_MeshNode * node = itNodes->next();
-               fprintf(myFileId, "%d %e %e %e\n", node->GetID(), node->X(),
-                       node->Y(), node->Z());
-       }
-       delete itNodes;
+    if ( !nodeNumByID.empty() )
+      nodeNumByID[ node->GetID() ] = num;
+  }
 
   /****************************************************************************
-  *                       ECRITURE DES ELEMENTS                                *
-  ****************************************************************************/
-       fprintf(stdout, "\n(**************************)\n");
-       fprintf(stdout, "(* ELEMENTS DU MAILLAGE : *)\n");
-       fprintf(stdout, "(**************************)");
-       /* Ecriture des connectivites, noms, numeros des mailles */
-
-       SMDS_Iterator<const SMDS_MeshEdge*> * itEdges=myMesh->edgesIterator();
-       while(itEdges->more())
-       {
-               const SMDS_MeshEdge * elem = itEdges->next();
-
-               switch (elem->NbNodes())
-               {
-               case 2:
-               {
-                       fprintf(myFileId, "%d %d ", elem->GetID(), 102);
-                       break;
-               }
-               case 3:
-               {
-                       fprintf(myFileId, "%d %d ", elem->GetID(), 103);
-                       break;
-               }
-               }
-
-               SMDS_Iterator<const SMDS_MeshElement *> * itNodes=elem->nodesIterator();
-               while(itNodes->more())
-                       fprintf(myFileId, "%d ", itNodes->next()->GetID());
-               
-               fprintf(myFileId, "\n");
-       }
-       delete itEdges;
-
-       SMDS_Iterator<const SMDS_MeshFace *> * itFaces=myMesh->facesIterator();
-       while(itFaces->more())
-       {
-               const SMDS_MeshElement * elem = itFaces->next();
-
-               switch (elem->NbNodes())
-               {
-               case 3:
-               {
-                       fprintf(myFileId, "%d %d ", elem->GetID(), 203);
-                       break;
-               }
-               case 4:
-               {
-                       fprintf(myFileId, "%d %d ", elem->GetID(), 204);
-                       break;
-               }
-               case 6:
-               {
-                       fprintf(myFileId, "%d %d ", elem->GetID(), 206);
-                       break;
-               }
-               }
-
-               SMDS_Iterator<const SMDS_MeshElement *> * itNodes=elem->nodesIterator();
-               while(itNodes->more())
-                       fprintf(myFileId, "%d ", itNodes->next()->GetID());
-               delete itNodes;
-       
-               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 8:
-               {
-                       fprintf(myFileId, "%d %d ", elem->GetID(), 308);
-                       break;
-               }
-               }
-
-               SMDS_Iterator<const SMDS_MeshElement *> * itNodes=elem->nodesIterator();
-               while(itNodes->more())
-                       fprintf(myFileId, "%d ", itNodes->next()->GetID());
-               delete itNodes; 
-
-               fprintf(myFileId, "\n");
-       }
-       delete itVolumes;
-
-       fclose(myFileId);
+   *                       ECRITURE DES ELEMENTS                                *
+   ****************************************************************************/
+  /* Ecriture des connectivites, noms, numeros des mailles */
+
+  num = 1;
+  for ( SMDS_EdgeIteratorPtr itEdges = myMesh->edgesIterator(); itEdges->more(); ++num )
+  {
+    const SMDS_MeshElement * elem = itEdges->next();
+    fprintf(aFileId, "%d %d ", num, 100 + elem->NbNodes());
+
+    for ( SMDS_ElemIteratorPtr it = elem->nodesIterator(); it->more(); )
+    {
+      int nodeID = it->next()->GetID();
+      if ( !nodeNumByID.empty() )
+        nodeID = nodeNumByID[ nodeID ];
+      fprintf(aFileId, "%d ", nodeID );
+    }
+    fprintf(aFileId, "\n");
+  }
+
+  for ( SMDS_FaceIteratorPtr itFaces = myMesh->facesIterator(); itFaces->more(); ++num )
+  {
+    const SMDS_MeshElement * elem = itFaces->next();
+
+    fprintf(aFileId, "%d %d ", num, (elem->IsPoly() ? 400 : 200 ) + elem->NbNodes() );
+
+    for( SMDS_ElemIteratorPtr it = elem->nodesIterator(); it->more(); )
+    {
+      int nodeID = it->next()->GetID();
+      if ( !nodeNumByID.empty() )
+        nodeID = nodeNumByID[ nodeID ];
+      fprintf(aFileId, "%d ", nodeID );
+    }
+    fprintf(aFileId, "\n");
+  }
+
+
+  const SMDS_MeshVolume* v;
+  for ( SMDS_VolumeIteratorPtr itVolumes=myMesh->volumesIterator(); itVolumes->more(); ++num )
+  {
+    const SMDS_MeshElement * elem = itVolumes->next();
+    if ( elem->IsPoly() )
+    {
+      fprintf(aFileId, "%d %d ", num, 500 + elem->NbNodes());
+
+      if (( v = myMesh->DownCast< SMDS_MeshVolume >( elem )))
+      {
+        std::vector<int> quant = v->GetQuantities();
+        if ( !quant.empty() )
+        {
+          fprintf(aFileId, "%d %d ", (int)quant.size(), quant[0]);
+          for ( size_t i = 1; i < quant.size(); ++i )
+            fprintf(aFileId, "%d ", quant[i]);
+        }
+      }
+    }
+    else
+    {
+      fprintf(aFileId, "%d %d ", num, 300 + elem->NbNodes());
+    }
+
+    for( SMDS_ElemIteratorPtr it = elem->nodesIterator(); it->more(); )
+    {
+      int nodeID = it->next()->GetID();
+      if ( !nodeNumByID.empty() )
+        nodeID = nodeNumByID[ nodeID ];
+      fprintf(aFileId, "%d ", nodeID );
+    }
+
+    fprintf(aFileId, "\n");
+  }
+
+  fclose(aFileId);
+
+  return aResult;
 }