src/DriverGMF/DriverGMF_Write.hxx

   1 // Copyright (C) 2007-2015  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 // File      : DriverGMF_Write.hxx
  23 // Created   : Mon Sep 17 15:36:47 2012
  24 // Author    : Edward AGAPOV (eap)
  25
  26
  27 #ifndef __DriverGMF_Write_HXX__
  28 #define __DriverGMF_Write_HXX__
  29
  30 #include "SMESH_DriverGMF.hxx"
  31
  32 #include "Driver_SMESHDS_Mesh.h"
  33 #include "SMDSAbs_ElementType.hxx"
  34 #include "SMDS_ElemIterator.hxx"
  35
  36 #include <gp_Pnt.hxx>
  37
  38 /*!
  39  * \brief Class for storing control points for writing GMF size maps
  40  */
  41 class MESHDriverGMF_EXPORT Control_Pnt : public gp_Pnt
  42 {
  43 public:
  44   Control_Pnt();
  45   Control_Pnt(const gp_Pnt& aPnt, double theSize);
  46   Control_Pnt(double x, double y, double z);
  47   Control_Pnt(double x, double y, double z, double size);
  48
  49   double Size() const { return size; };
  50   void SetSize( double theSize ) { size = theSize; };
  51
  52 private:
  53   double size;
  54 };
  55
  56 /*!
  57  * \brief Driver Writing a mesh into a GMF file.
  58  */
  59 class MESHDriverGMF_EXPORT DriverGMF_Write : public Driver_SMESHDS_Mesh
  60 {
  61 public:
  62
  63   DriverGMF_Write();
  64   ~DriverGMF_Write();
  65
  66   void SetExportRequiredGroups( bool toExport )
  67   {
  68     _exportRequiredGroups = toExport;
  69   }
  70
  71   virtual Status Perform();
  72
  73   // Size Maps
  74   Status PerformSizeMap( const std::vector<Control_Pnt>& points );
  75   void SetSizeMapPrefix( std::string prefix )
  76   {
  77     myVerticesFile = prefix + ".mesh";
  78     mySolFile = prefix + ".sol";
  79   };
  80   std::vector<std::string> GetSizeMapFiles();
  81
  82  private:
  83
  84   SMDS_ElemIteratorPtr elementIterator(SMDSAbs_ElementType  type);
  85   SMDS_ElemIteratorPtr elementIterator(SMDSAbs_EntityType   type);
  86   SMDS_ElemIteratorPtr elementIterator(SMDSAbs_GeometryType type);
  87
  88   bool _exportRequiredGroups;
  89   std::string myVerticesFile;
  90   std::string mySolFile;
  91 };
  92
  93 #endif