src/ElectromagnetismFluxDisc/plugin/ElectromagnetismFluxDiscModule/vtkElectromagnetismFluxDisc.h

   1 // Copyright (C) 2021  CEA/DEN, EDF R&D
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19 // Author : Anthony Geay (EDF R&D)
  20
  21 #pragma once
  22
  23 #include <vtkUnstructuredGridAlgorithm.h>
  24 //#include <vtkPolyDataAlgorithm.h>
  25
  26 class vtkMutableDirectedGraph;
  27 class vtkImplicitFunction;
  28 class vtkCylinder;
  29
  30 class VTK_EXPORT vtkElectromagnetismFluxDisc : public vtkUnstructuredGridAlgorithm
  31 {
  32 public:
  33   static vtkElectromagnetismFluxDisc* New();
  34   vtkTypeMacro(vtkElectromagnetismFluxDisc, vtkUnstructuredGridAlgorithm);
  35   void PrintSelf(ostream& os, vtkIndent indent) override;
  36
  37   void SetCutFunction(vtkImplicitFunction* func);
  38
  39   vtkMTimeType GetMTime();
  40
  41   vtkSetClampMacro(RadialResolution, int, 1, VTK_INT_MAX);
  42   vtkGetMacro(RadialResolution, int);
  43
  44   vtkSetClampMacro(CircumferentialResolution, int, 3, VTK_INT_MAX);
  45   vtkGetMacro(CircumferentialResolution, int);
  46
  47 protected:
  48   vtkElectromagnetismFluxDisc();
  49   ~vtkElectromagnetismFluxDisc();
  50
  51
  52   int RequestInformation(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
  53   int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
  54
  55   vtkCylinder* _cyl;
  56   class vtkInternals;
  57   vtkInternals* Internal;
  58
  59 private:
  60   vtkElectromagnetismFluxDisc(const vtkElectromagnetismFluxDisc&) = delete;
  61   void operator=(const vtkElectromagnetismFluxDisc&) = delete;
  62
  63   int RadialResolution;
  64   int CircumferentialResolution;
  65 };