vtkHyperTreeGridCellSizeStrategy.h

Go to the documentation of this file.

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

#include "vtkFiltersHyperTreeModule.h" // For export macro
#include "vtkHyperTreeGridGenerateFieldStrategy.h"
#include "vtkIndexedArray.h"
#include "vtkNew.h"

#include <unordered_map>

VTK_ABI_NAMESPACE_BEGIN

class vtkDoubleArray;
class vtkUnsignedCharArray;

class VTKFILTERSHYPERTREE_EXPORT vtkHyperTreeGridCellSizeStrategy
  : public vtkHyperTreeGridGenerateFieldStrategy
{
public:
  static vtkHyperTreeGridCellSizeStrategy* New();
  vtkTypeMacro(vtkHyperTreeGridCellSizeStrategy, vtkHyperTreeGridGenerateFieldStrategy);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  using vtkHyperTreeGridGenerateFieldStrategy::Initialize;
  void Initialize(vtkHyperTreeGrid* inputHTG) override;

  using vtkHyperTreeGridGenerateFieldStrategy::Compute;
  void Compute(vtkHyperTreeGridNonOrientedGeometryCursor* cursor) override;

  vtkDataArray* GetAndFinalizeArray() override;

private:
  vtkHyperTreeGridCellSizeStrategy();
  ~vtkHyperTreeGridCellSizeStrategy() override;

  bool InsertSize(double cellSize, vtkIdType currentIndex);

  void ConvertSizes();

  bool UseIndexedVolume = true;
  std::unordered_map<double, unsigned char> VolumeLookup;
  vtkNew<vtkUnsignedCharArray> SizeIndirectionTable;
  vtkNew<vtkDoubleArray> SizeDiscreteValues;
  vtkNew<vtkDoubleArray> SizeFullValues;

  vtkNew<vtkIndexedArray<double>> OutputSizeArray;
};

VTK_ABI_NAMESPACE_END
#endif // vtkHyperTreeGridCellSizeStrategy_h