vtkRandomHyperTreeGridSource.h

Go to the documentation of this file.

/*=========================================================================

  Program:   Visualization Toolkit
  Module:    vtkRandomHyperTreeGridSource.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.

=========================================================================*/
#ifndef vtkRandomHyperTreeGridSource_h
#define vtkRandomHyperTreeGridSource_h

#include "vtkFiltersSourcesModule.h" // For export macro
#include "vtkHyperTreeGridAlgorithm.h"

#include <vtkNew.h> // For vtkNew

class vtkDoubleArray;
class vtkExtentTranslator;
class vtkHyperTreeCursor;
class vtkMinimalStandardRandomSequence;

class VTKFILTERSSOURCES_EXPORT vtkRandomHyperTreeGridSource
    : public vtkHyperTreeGridAlgorithm
{
public:
  static vtkRandomHyperTreeGridSource* New();
  vtkTypeMacro(vtkRandomHyperTreeGridSource, vtkHyperTreeGridAlgorithm)
  void PrintSelf(ostream &os, vtkIndent indent) override;

  vtkGetVector3Macro(GridSize, unsigned int)
  vtkSetVector3Macro(GridSize, unsigned int)
  vtkGetVector6Macro(OutputBounds, double)
  vtkSetVector6Macro(OutputBounds, double)

  vtkGetMacro(Seed, vtkTypeUInt32)
  vtkSetMacro(Seed, vtkTypeUInt32)
  vtkGetMacro(MaxDepth, vtkIdType)
  vtkSetClampMacro(MaxDepth, vtkIdType, 1, VTK_ID_MAX)
  vtkGetMacro(SplitFraction, double)
  vtkSetClampMacro(SplitFraction, double, 0., 1.)
protected:
  vtkRandomHyperTreeGridSource();
  ~vtkRandomHyperTreeGridSource() override;

  int RequestInformation(vtkInformation *req,
                         vtkInformationVector **inInfo,
                         vtkInformationVector *outInfo) override;

  int RequestData(vtkInformation *req,
                  vtkInformationVector **inInfo,
                  vtkInformationVector *outInfo) override;

  // We just do the work in RequestData.
  int ProcessTrees(vtkHyperTreeGrid *, vtkDataObject *) final { return 1; }

  int FillOutputPortInformation(int port, vtkInformation *info) override;

  void SubdivideLeaves(vtkHyperTreeCursor *cursor, vtkIdType treeId);

  bool ShouldRefine(vtkIdType level);

  unsigned int GridSize[3];
  double OutputBounds[6];
  vtkTypeUInt32 Seed;
  vtkIdType MaxDepth;
  double SplitFraction;

private:
  vtkRandomHyperTreeGridSource(const vtkRandomHyperTreeGridSource&) = delete;
  void operator=(const vtkRandomHyperTreeGridSource&) = delete;

  vtkNew<vtkMinimalStandardRandomSequence> RNG;
  vtkNew<vtkExtentTranslator> ExtentTranslator;
  vtkHyperTreeGrid *HTG;
  vtkDoubleArray *Levels;
};

#endif // vtkRandomHyperTreeGridSource_h