vtkHyperTreeGridPreConfiguredSource.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause

#include "vtkFiltersSourcesModule.h" //for export macro
#include "vtkHyperTreeGridAlgorithm.h"
#include "vtkWrappingHints.h" // For VTK_MARSHALAUTO

#ifndef vtkHyperTreeGridPreConfiguredSource_h
#define vtkHyperTreeGridPreConfiguredSource_h

VTK_ABI_NAMESPACE_BEGIN
class vtkHyperTreeGrid;
class vtkHyperTreeGridNonOrientedCursor;
class vtkDoubleArray;

class VTKFILTERSSOURCES_EXPORT VTK_MARSHALAUTO vtkHyperTreeGridPreConfiguredSource
  : public vtkHyperTreeGridAlgorithm
{
public:
  vtkTypeMacro(vtkHyperTreeGridPreConfiguredSource, vtkHyperTreeGridAlgorithm);
  static vtkHyperTreeGridPreConfiguredSource* New();


  void GenerateUnbalanced(vtkHyperTreeGrid* HTG, unsigned int dim, unsigned int factor,
    unsigned int depth, const std::vector<double>& extent,
    const std::vector<unsigned int>& subdivisions);

  void GenerateBalanced(vtkHyperTreeGrid* HTG, unsigned int dim, unsigned int factor,
    unsigned int depth, const std::vector<double>& extent,
    const std::vector<unsigned int>& subdivisions);


  enum HTGType
  {
    UNBALANCED_3DEPTH_2BRANCH_2X3,
    BALANCED_3DEPTH_2BRANCH_2X3,
    UNBALANCED_2DEPTH_3BRANCH_3X3,
    BALANCED_4DEPTH_3BRANCH_2X2,
    UNBALANCED_3DEPTH_2BRANCH_3X2X3,
    BALANCED_2DEPTH_3BRANCH_3X3X2,
    CUSTOM
  };


  enum HTGArchitecture
  {
    UNBALANCED,
    BALANCED
  };

  vtkGetEnumMacro(HTGMode, HTGType);
  vtkSetEnumMacro(HTGMode, HTGType);


  vtkGetEnumMacro(CustomArchitecture, HTGArchitecture);
  vtkSetEnumMacro(CustomArchitecture, HTGArchitecture);

  vtkGetMacro(CustomDim, unsigned int);
  vtkSetMacro(CustomDim, unsigned int);

  vtkGetMacro(CustomFactor, unsigned int);
  vtkSetMacro(CustomFactor, unsigned int);

  vtkGetMacro(CustomDepth, unsigned int);
  vtkSetMacro(CustomDepth, unsigned int);

  vtkGetVector6Macro(CustomExtent, double);
  vtkSetVector6Macro(CustomExtent, double);

  vtkGetVector3Macro(CustomSubdivisions, unsigned int);
  vtkSetVector3Macro(CustomSubdivisions, unsigned int);


  void GenerateUnbalanced3DepthQuadTree2x3(vtkHyperTreeGrid* HTG);

  void GenerateBalanced3DepthQuadTree2x3(vtkHyperTreeGrid* HTG);

  void GenerateUnbalanced2Depth3BranchTree3x3(vtkHyperTreeGrid* HTG);

  void GenerateBalanced4Depth3BranchTree2x2(vtkHyperTreeGrid* HTG);

  void GenerateUnbalanced3DepthOctTree3x2x3(vtkHyperTreeGrid* HTG);

  void GenerateBalanced2Depth3BranchTree3x3x2(vtkHyperTreeGrid* HTG);

  int GenerateCustom(vtkHyperTreeGrid* HTG);

protected:
  vtkHyperTreeGridPreConfiguredSource();
  ~vtkHyperTreeGridPreConfiguredSource() override = default;
  vtkHyperTreeGridPreConfiguredSource(const vtkHyperTreeGridPreConfiguredSource&) = delete;

  void operator=(const vtkHyperTreeGridPreConfiguredSource&) = delete;

  int FillOutputPortInformation(int, vtkInformation*) override;

  int RequestInformation(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
  int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;

  int ProcessTrees(vtkHyperTreeGrid*, vtkDataObject*) override;


  void Preprocess(vtkHyperTreeGrid* HTG, unsigned int dim, unsigned int factor,
    const std::vector<double>& extent, const std::vector<unsigned int>& subdivisions);

  void RecurseBalanced(
    vtkHyperTreeGridNonOrientedCursor* cursor, vtkDoubleArray* levels, int maxDepth);


  HTGType HTGMode;

  HTGArchitecture CustomArchitecture;
  unsigned int CustomDim;
  unsigned int CustomFactor;
  unsigned int CustomDepth;
  double CustomExtent[6];
  unsigned int CustomSubdivisions[3];

}; // vtkHyperTreeGridPreConfiguredSource
VTK_ABI_NAMESPACE_END

#endif // vtkHyperTreeGridPreConfiguredSource_h