Salome HOME
1) Remove writing into stdout
[modules/smesh.git] / src / DriverDAT / DriverDAT_W_SMDS_Mesh.cxx
index d99a124fa89f4500f8ce028f6234da23a92c14b1..ae1632dca8b45d2594294b2c9e52b87aa7f9171e 100644 (file)
-//  SMESH DriverDAT : driver to read and write 'dat' files
+// Copyright (C) 2007-2013  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.
 //
+// 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"
-{
-
-/**
- * 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()
-{
-}
 
-DriverDAT_W_SMDS_Mesh::~DriverDAT_W_SMDS_Mesh()
-{
-       ;
-}
+#include <stdio.h>
 
-void DriverDAT_W_SMDS_Mesh::SetMesh(SMDS_Mesh * aMesh)
-{
-       myMesh = aMesh;
-}
+#include "DriverDAT_W_SMDS_Mesh.h"
 
-void DriverDAT_W_SMDS_Mesh::SetFile(string aFile)
-{
-       myFile = aFile;
-}
+#include "SMDS_Mesh.hxx"
 
-void DriverDAT_W_SMDS_Mesh::SetFileId(FILE * aFileId)
-{
-       myFileId = aFileId;
-}
+#include "utilities.h"
 
-void DriverDAT_W_SMDS_Mesh::SetMeshId(int aMeshId)
-{
-       myMeshId = aMeshId;
-}
+#include <Basics_Utils.hxx>
 
-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();
-}
+using namespace std;
 
-void DriverDAT_W_SMDS_Mesh::Write()
+Driver_Mesh::Status DriverDAT_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);
-       }
-
+  Kernel_Utils::Localizer loc;
+  Status aResult = DRS_OK;
+
+  int nbNodes, nbCells;
+  //int i;
+  
+  char *file2Read = (char *)myFile.c_str();
+  FILE* aFileId = fopen(file2Read, "w+");
+  if (aFileId < 0) {
+    fprintf(stderr, ">> ERREUR : ouverture du fichier %s \n", file2Read);
+    return DRS_FAIL;
+  }
+  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;
-
-       fprintf(stdout, "%d %d\n", nbNodes, nbCells);
-       fprintf(myFileId, "%d %d\n", nbNodes, nbCells);
-
+   *                       NOMBRES D'OBJETS                                    *
+   ****************************************************************************/
+  
+  /* 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(aFileId, "%d %d\n", nbNodes, nbCells);
+  
   /****************************************************************************
-  *                       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;
-
+   *                       ECRITURE DES NOEUDS                                 *
+   ****************************************************************************/
+  
+  SMDS_NodeIteratorPtr itNodes=myMesh->nodesIterator();
+  while(itNodes->more()){               
+    const SMDS_MeshNode * node = itNodes->next();
+    fprintf(aFileId, "%d %e %e %e\n", node->GetID(), node->X(), node->Y(), node->Z());
+  }
+        
   /****************************************************************************
-  *                       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 */
+  
+  SMDS_EdgeIteratorPtr itEdges=myMesh->edgesIterator();
+  while(itEdges->more()){
+    const SMDS_MeshElement * elem = itEdges->next();
+    switch (elem->NbNodes()) {
+    case 2:
+      fprintf(aFileId, "%d %d ", elem->GetID(), 102);
+      break;
+    case 3:
+      fprintf(aFileId, "%d %d ", elem->GetID(), 103);
+      break;
+    }
+    SMDS_ElemIteratorPtr it=elem->nodesIterator();
+    while(it->more()) 
+      fprintf(aFileId, "%d ", it->next()->GetID());
+    fprintf(aFileId, "\n");
+  }
+  
+  SMDS_FaceIteratorPtr itFaces=myMesh->facesIterator();
+  while(itFaces->more()){
+    const SMDS_MeshElement * elem = itFaces->next();
+    if ( elem->IsPoly() )
+      fprintf(aFileId, "%d %d ", elem->GetID(), 400+elem->NbNodes());
+    else
+      fprintf(aFileId, "%d %d ", elem->GetID(), 200+elem->NbNodes());
+    SMDS_ElemIteratorPtr it=elem->nodesIterator();
+    while(it->more()) 
+      fprintf(aFileId, "%d ", it->next()->GetID());
+    fprintf(aFileId, "\n");
+  }
+
+  SMDS_VolumeIteratorPtr itVolumes=myMesh->volumesIterator();
+  while(itVolumes->more()){
+    const SMDS_MeshElement * elem = itVolumes->next();
+    if ( elem->IsPoly() )
+      fprintf(aFileId, "%d %d ", elem->GetID(), 500+elem->NbNodes());
+    else
+      fprintf(aFileId, "%d %d ", elem->GetID(), 300+elem->NbNodes());
+    SMDS_ElemIteratorPtr it=elem->nodesIterator();
+    while(it->more()) 
+      fprintf(aFileId, "%d ", it->next()->GetID());
+
+    fprintf(aFileId, "\n");
+  }
+  
+  fclose(aFileId);
+
+  return aResult;
 }