src/libengrid/tvtkoperation.h

   1 // ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
   2 // +                                                                      +
   3 // + This file is part of enGrid.                                         +
   4 // +                                                                      +
   5 // + Copyright 2008-2014 enGits GmbH                                      +
   6 // +                                                                      +
   7 // + enGrid is free software: you can redistribute it and/or modify       +
   8 // + it under the terms of the GNU General Public License as published by +
   9 // + the Free Software Foundation, either version 3 of the License, or    +
  10 // + (at your option) any later version.                                  +
  11 // +                                                                      +
  12 // + enGrid is distributed in the hope that it will be useful,            +
  13 // + but WITHOUT ANY WARRANTY; without even the implied warranty of       +
  14 // + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the        +
  15 // + GNU General Public License for more details.                         +
  16 // +                                                                      +
  17 // + You should have received a copy of the GNU General Public License    +
  18 // + along with enGrid. If not, see <http://www.gnu.org/licenses/>.       +
  19 // +                                                                      +
  20 // ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
  21 #ifndef tvtkoperation_H
  22 #define tvtkoperation_H
  23
  24
  25 template <class T>
  26 class TVtkOperation;
  27
  28 #include "operation.h"
  29
  30 template <class T>
  31 class TVtkOperation : public Operation
  32 {
  33
  34 protected: // attributes
  35
  36   T *vtk_filter;
  37
  38 public: // methods
  39
  40   TVtkOperation();
  41   virtual ~TVtkOperation();
  42   T* getFilter() { return vtk_filter; };
  43
  44 protected: // methods
  45
  46   virtual void operate();
  47
  48 };
  49
  50
  51 template <class T>
  52 TVtkOperation<T>::TVtkOperation()
  53 {
  54   vtk_filter = T::New();
  55 };
  56
  57 template <class T>
  58 TVtkOperation<T>::~TVtkOperation()
  59 {
  60   vtk_filter->Delete();
  61 };
  62
  63 template <class T>
  64 void TVtkOperation<T>::operate()
  65 {
  66   vtkSmartPointer<vtkUnstructuredGrid> ug = vtkSmartPointer<vtkUnstructuredGrid>::New();
  67   ug->DeepCopy(grid);
  68   vtk_filter->SetInput(ug);
  69   vtk_filter->Update();
  70   grid->DeepCopy(vtk_filter->GetOutput());
  71 };
  72
  73
  74 #endif