vtkEgGridFilter.h

   1 //
   2 // ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
   3 // +                                                                      +
   4 // + This file is part of enGrid.                                         +
   5 // +                                                                      +
   6 // + Copyright 2008,2009 Oliver Gloth                                     +
   7 // +                                                                      +
   8 // + enGrid is free software: you can redistribute it and/or modify       +
   9 // + it under the terms of the GNU General Public License as published by +
  10 // + the Free Software Foundation, either version 3 of the License, or    +
  11 // + (at your option) any later version.                                  +
  12 // +                                                                      +
  13 // + enGrid is distributed in the hope that it will be useful,            +
  14 // + but WITHOUT ANY WARRANTY; without even the implied warranty of       +
  15 // + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the        +
  16 // + GNU General Public License for more details.                         +
  17 // +                                                                      +
  18 // + You should have received a copy of the GNU General Public License    +
  19 // + along with enGrid. If not, see <http://www.gnu.org/licenses/>.       +
  20 // +                                                                      +
  21 // ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
  22 //
  23 #ifndef __vtkEgGridFilter_h
  24 #define __vtkEgGridFilter_h
  25
  26 class vtkEgGridFilter;
  27
  28 #include <vtkUnstructuredGridAlgorithm.h>
  29 #include <vtkUnstructuredGrid.h>
  30 #include <vtkIntArray.h>
  31 #include <vtkLongArray.h>
  32 #include <vtkDoubleArray.h>
  33 #include <vtkCellData.h>
  34 #include <vtkCellType.h>
  35 #include <vtkCellLinks.h>
  36 #include <vtkObjectFactory.h>
  37
  38 #include <QSet>
  39 #include <QVector>
  40
  41 #include "engrid.h"
  42 #include "egvtkobject.h"
  43
  44 class vtkEgGridFilter : public vtkUnstructuredGridAlgorithm, public EgVtkObject
  45 {
  46
  47 protected: // attributes
  48
  49   QSet<int>           *BoundaryCodes;
  50   vtkUnstructuredGrid *input;
  51   vtkUnstructuredGrid *output;
  52
  53 protected: // methods
  54
  55   vtkEgGridFilter();
  56   ~vtkEgGridFilter() {};
  57
  58   virtual int FillInputPortInformation
  59     (
  60       int,
  61       vtkInformation* info
  62     );
  63
  64   virtual int FillOutputPortInformation
  65     (
  66       int,
  67       vtkInformation* info
  68     );
  69
  70   virtual int RequestData
  71     (
  72       vtkInformation*,
  73       vtkInformationVector** inputVector,
  74       vtkInformationVector*  outputVector
  75     );
  76
  77   virtual void ExecuteEg() = 0;
  78
  79   /**
  80    * Extract boundary elements and nodes with a given set of boundary conditions
  81    * @param cells On return, this will contain the indexes of the extracted boundary elements.
  82    * @param modes On return, this will contain the indexes of the extracted boundary nodes.
  83    * @param grid  The grid to operate on.
  84    * @param bc    A set of the boundary conditions to extract.
  85    */
  86   void ExtractBoundary
  87     (
  88       QVector<vtkIdType>  &cells,
  89       QVector<vtkIdType>  &nodes,
  90       QSet<int>           &bc,
  91       vtkUnstructuredGrid *grid
  92     );
  93
  94 public: // methods
  95
  96   void SetBoundaryCodes(QSet<int> *bc);
  97
  98 };
  99
 100 #endif