src/OBJECT/SMESH_ExtractGeometry.h

   1 // Copyright (C) 2007-2020  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 #ifndef SALOME_ExtractGeometry_HeaderFile
  24 #define SALOME_ExtractGeometry_HeaderFile
  25
  26 #include "SMESH_Object.h"
  27
  28 #include <vtkExtractGeometry.h>
  29 #include <vector>
  30
  31 #include "VTKViewer.h"
  32
  33 class SMESHOBJECT_EXPORT SMESH_ExtractGeometry : public vtkExtractGeometry{
  34 public:
  35   vtkTypeMacro(SMESH_ExtractGeometry,vtkExtractGeometry)
  36
  37   static SMESH_ExtractGeometry *New();
  38
  39   void SetStoreMapping(bool theStoreMapping){
  40     myStoreMapping = theStoreMapping;
  41     Modified();
  42   }
  43   bool GetStoreMapping(){ return myStoreMapping;}
  44
  45   virtual vtkIdType GetNodeObjId(int theVtkID);
  46   virtual vtkIdType GetElemObjId(int theVtkID);
  47
  48 protected:
  49   SMESH_ExtractGeometry();
  50   ~SMESH_ExtractGeometry();
  51
  52   // Usual data generation method
  53   virtual int RequestData(vtkInformation *, vtkInformationVector **, vtkInformationVector *);
  54
  55 private:
  56   bool myStoreMapping;
  57   typedef std::vector<vtkIdType> TVectorId;
  58   TVectorId myElemVTK2ObjIds;
  59   TVectorId myNodeVTK2ObjIds;
  60
  61   SMESH_ExtractGeometry(const SMESH_ExtractGeometry&);  // Not implemented.
  62   void operator=(const SMESH_ExtractGeometry&);  // Not implemented.
  63 };
  64
  65
  66 #endif
  67
  68