src/DriverDAT/DriverDAT_W_SMDS_Mesh.cxx

   1 // Copyright (C) 2007-2016  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 // This library is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU Lesser General Public
   8 // License as published by the Free Software Foundation; either
   9 // version 2.1 of the License, or (at your option) any later version.
  10 //
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 // Lesser General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU Lesser General Public
  17 // License along with this library; if not, write to the Free Software
  18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22
  23 #include <stdio.h>
  24
  25 #include "DriverDAT_W_SMDS_Mesh.h"
  26
  27 #include "SMDS_Mesh.hxx"
  28
  29 #include "utilities.h"
  30
  31 #include <Basics_Utils.hxx>
  32
  33 using namespace std;
  34
  35 Driver_Mesh::Status DriverDAT_W_SMDS_Mesh::Perform()
  36 {
  37   Kernel_Utils::Localizer loc;
  38   Status aResult = DRS_OK;
  39
  40   int nbNodes, nbCells;
  41
  42   char *file2Read = (char *)myFile.c_str();
  43   FILE* aFileId = fopen(file2Read, "w+");
  44   if ( !aFileId )
  45   {
  46     fprintf(stderr, ">> ERREUR : ouverture du fichier %s \n", file2Read);
  47     return DRS_FAIL;
  48   }
  49   SCRUTE(myMesh);
  50   /****************************************************************************
  51    *                       NOMBRES D'OBJETS                                    *
  52    ****************************************************************************/
  53
  54   /* Combien de noeuds ? */
  55   nbNodes = myMesh->NbNodes();
  56
  57   /* Combien de mailles, faces ou aretes ? */
  58   int nb_of_edges, nb_of_faces, nb_of_volumes;
  59   nb_of_edges = myMesh->NbEdges();
  60   nb_of_faces = myMesh->NbFaces();
  61   nb_of_volumes = myMesh->NbVolumes();
  62   nbCells = nb_of_edges + nb_of_faces + nb_of_volumes;
  63   SCRUTE(nb_of_edges);
  64   SCRUTE(nb_of_faces);
  65   SCRUTE(nb_of_volumes);
  66
  67   //fprintf(stdout, "%d %d\n", nbNodes, nbCells);
  68   fprintf(aFileId, "%d %d\n", nbNodes, nbCells);
  69
  70   /****************************************************************************
  71    *                       ECRITURE DES NOEUDS                                 *
  72    ****************************************************************************/
  73
  74   SMDS_NodeIteratorPtr itNodes=myMesh->nodesIterator();
  75   while(itNodes->more())
  76   {
  77     const SMDS_MeshNode * node = itNodes->next();
  78     fprintf(aFileId, "%d %.14e %.14e %.14e\n", node->GetID(), node->X(), node->Y(), node->Z());
  79   }
  80
  81   /****************************************************************************
  82    *                       ECRITURE DES ELEMENTS                                *
  83    ****************************************************************************/
  84   /* Ecriture des connectivites, noms, numeros des mailles */
  85
  86   SMDS_EdgeIteratorPtr itEdges=myMesh->edgesIterator();
  87   while(itEdges->more())
  88   {
  89     const SMDS_MeshElement * elem = itEdges->next();
  90     switch (elem->NbNodes())
  91     {
  92     case 2:
  93       fprintf(aFileId, "%d %d ", elem->GetID(), 102);
  94       break;
  95     case 3:
  96       fprintf(aFileId, "%d %d ", elem->GetID(), 103);
  97       break;
  98     }
  99     SMDS_ElemIteratorPtr it=elem->nodesIterator();
 100     while(it->more())
 101       fprintf(aFileId, "%d ", it->next()->GetID());
 102     fprintf(aFileId, "\n");
 103   }
 104
 105   SMDS_FaceIteratorPtr itFaces=myMesh->facesIterator();
 106   while(itFaces->more())
 107   {
 108     const SMDS_MeshElement * elem = itFaces->next();
 109     if ( elem->IsPoly() )
 110       fprintf(aFileId, "%d %d ", elem->GetID(), 400+elem->NbNodes());
 111     else
 112       fprintf(aFileId, "%d %d ", elem->GetID(), 200+elem->NbNodes());
 113     SMDS_ElemIteratorPtr it=elem->nodesIterator();
 114     while(it->more())
 115       fprintf(aFileId, "%d ", it->next()->GetID());
 116     fprintf(aFileId, "\n");
 117   }
 118
 119   SMDS_VolumeIteratorPtr itVolumes=myMesh->volumesIterator();
 120   const SMDS_VtkVolume* v;
 121   while(itVolumes->more())
 122   {
 123     const SMDS_MeshElement * elem = itVolumes->next();
 124     if ( elem->IsPoly() )
 125     {
 126       fprintf(aFileId, "%d %d ", elem->GetID(), 500+elem->NbNodes());
 127
 128       if (( v = dynamic_cast< const SMDS_VtkVolume*>( elem )))
 129       {
 130         std::vector<int> quant = v->GetQuantities();
 131         if ( !quant.empty() )
 132         {
 133           fprintf(aFileId, "%d %d ", quant.size(), quant[0]);
 134           for ( size_t i = 1; i < quant.size(); ++i )
 135             fprintf(aFileId, "%d ", quant[i]);
 136         }
 137       }
 138     }
 139     else
 140     {
 141       fprintf(aFileId, "%d %d ", elem->GetID(), 300+elem->NbNodes());
 142     }
 143     SMDS_ElemIteratorPtr it=elem->nodesIterator();
 144     while(it->more())
 145       fprintf(aFileId, "%d ", it->next()->GetID());
 146
 147     fprintf(aFileId, "\n");
 148   }
 149
 150   fclose(aFileId);
 151
 152   return aResult;
 153 }