vtkHyperTreeGridSource.h

Go to the documentation of this file.

/*=========================================================================
 
  Program:   Visualization Toolkit
  Module:    vtkHyperTreeGridSource.h
 
  Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
  All rights reserved.
  See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
 
     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notice for more information.
 
=========================================================================*/
 The top level of the tree is not considered a grid level
 NB: For ease of legibility, white spaces are allowed and ignored.
 
    @par Thanks:
 This class was written by Philippe Pebay and Joachim Pouderoux,
 Kitware 2013
 This work was supported in part by Commissariat a l'Energie Atomique (CEA/DIF)
 
    @par      Examples:
              @ref c2_vtk_e_vtkHyperTreeGridSource "vtkHyperTreeGridSource (Examples)"
 
    @par      Tests:
              @ref c2_vtk_t_vtkHyperTreeGridSource "vtkHyperTreeGridSource (Tests)"
*/
 
#ifndef vtkHyperTreeGridSource_h
#define vtkHyperTreeGridSource_h
 
#include "vtkFiltersSourcesModule.h" // For export macro
#include "vtkHyperTreeGridAlgorithm.h"
 
#include <string> // STL Header
#include <map> // STL Header
#include <vector> // STL Header
 
class vtkDataArray;
class vtkBitArray;
class vtkImplicitFunction;
class vtkHyperTreeGrid;
class vtkQuadric;
 
class VTKFILTERSSOURCES_EXPORT vtkHyperTreeGridSource : public vtkHyperTreeGridAlgorithm
{
public:
  vtkTypeMacro(vtkHyperTreeGridSource,vtkHyperTreeGridAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent);
 
  static vtkHyperTreeGridSource* New();
 
  unsigned int GetMaximumLevel();
 
  void SetMaximumLevel( unsigned int levels );
 
 
  vtkSetVector3Macro(Origin, double);
  vtkGetVector3Macro(Origin, double);
 
 
  vtkSetVector3Macro(GridScale, double);
  vtkGetVector3Macro(GridScale, double);
 
 
  vtkSetVector3Macro(GridSize, unsigned int);
  vtkGetVector3Macro(GridSize, unsigned int);
 
 
  vtkSetMacro(TransposedRootIndexing, bool);
  vtkGetMacro(TransposedRootIndexing, bool);
  void SetIndexingModeToKJI();
  void SetIndexingModeToIJK();
 
 
  vtkSetClampMacro(BranchFactor, unsigned int, 2, 3);
  vtkGetMacro(BranchFactor, unsigned int);
 
 
  vtkSetClampMacro(Dimension, unsigned int, 2, 3);
  vtkGetMacro(Dimension, unsigned int);
 
 
  vtkSetMacro(UseDescriptor, bool);
  vtkGetMacro(UseDescriptor, bool);
  vtkBooleanMacro(UseDescriptor, bool);
 
 
  vtkSetMacro(UseMaterialMask, bool);
  vtkGetMacro(UseMaterialMask, bool);
  vtkBooleanMacro(UseMaterialMask, bool);
 
 
  vtkSetStringMacro(Descriptor);
  vtkGetStringMacro(Descriptor);
 
 
  vtkSetStringMacro(MaterialMask);
  vtkGetStringMacro(MaterialMask);
 
 
  virtual void SetDescriptorBits( vtkBitArray* );
  vtkGetObjectMacro( DescriptorBits, vtkBitArray );
 
  virtual void SetLevelZeroMaterialIndex( vtkIdTypeArray* );
 
 
  virtual void SetMaterialMaskBits( vtkBitArray* );
  vtkGetObjectMacro( MaterialMaskBits, vtkBitArray );
 
 
  virtual void SetQuadric( vtkQuadric* );
  vtkGetObjectMacro(Quadric, vtkQuadric);
 
 
  void SetQuadricCoefficients( double[10] );
  void GetQuadricCoefficients( double[10] );
  double* GetQuadricCoefficients();
 
  unsigned long GetMTime();
 
 
  vtkBitArray* ConvertDescriptorStringToBitArray( const std::string& );
  vtkBitArray* ConvertMaterialMaskStringToBitArray( const std::string& );
 
protected:
  vtkHyperTreeGridSource();
  ~vtkHyperTreeGridSource();
 
  int RequestInformation ( vtkInformation*,
                           vtkInformationVector**,
                           vtkInformationVector* );
 
  virtual int RequestData( vtkInformation*,
                           vtkInformationVector**,
                           vtkInformationVector* );
 
  int InitializeFromStringDescriptor();
 
  int InitializeFromBitsDescriptor();
 
 
  void InitTreeFromDescriptor( vtkHyperTreeCursor* cursor,
                                int treeIdx,
                                int idx[3] );
 
 
  void SubdivideFromStringDescriptor( vtkHyperTreeCursor* cursor,
                                unsigned int level,
                                int treeIdx,
                                int childIdx,
                                int idx[3],
                                int parentPos );
 
 
  void SubdivideFromBitsDescriptor( vtkHyperTreeCursor* cursor,
                                unsigned int level,
                                int treeIdx,
                                int childIdx,
                                int idx[3],
                                int parentPos );
 
 
  void SubdivideFromQuadric( vtkHyperTreeCursor* cursor,
                             unsigned int level,
                             int treeIdx,
                             const int idx[3],
                             double origin[3],
                             double size[3] );
 
  double EvaluateQuadric( double[3] );
 
  double Origin[3];
  double GridScale[3];
  unsigned int GridSize[3];
  bool TransposedRootIndexing;
  unsigned int MaximumLevel;
  unsigned int Dimension;
  unsigned int BranchFactor;
  unsigned int BlockSize;
  bool UseDescriptor;
  bool UseMaterialMask;
 
  vtkDataArray* XCoordinates;
  vtkDataArray* YCoordinates;
  vtkDataArray* ZCoordinates;
 
  char* Descriptor;
  char* MaterialMask;
  std::vector<std::string> LevelDescriptors;
  std::vector<std::string> LevelMaterialMasks;
 
  vtkBitArray* DescriptorBits;
  vtkBitArray* MaterialMaskBits;
  std::vector<vtkIdType> LevelBitsIndex;
  std::vector<vtkIdType> LevelBitsIndexCnt;
 
  vtkIdTypeArray* LevelZeroMaterialIndex;
  std::map<vtkIdType, vtkIdType> LevelZeroMaterialMap;
 
  std::vector<int> LevelCounters;
 
  vtkQuadric* Quadric;
 
  vtkHyperTreeGrid* Output;
 
private:
  vtkHyperTreeGridSource(const vtkHyperTreeGridSource&);  // Not implemented.
  void operator=(const vtkHyperTreeGridSource&);  // Not implemented.
};
 
#endif