vtkCellAttributeCalculator.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 vtkCellAttributeCalculator_h
#define vtkCellAttributeCalculator_h

#include "vtkCommonDataModelModule.h" // for export
#include "vtkObject.h"
#include "vtkSmartPointer.h" // for API.

#include <unordered_map>

VTK_ABI_NAMESPACE_BEGIN
class vtkCellGrid;
class vtkCellMetadata;
class vtkCellAttribute;

class VTKCOMMONDATAMODEL_EXPORT vtkCellAttributeCalculator : public vtkObject
{
public:
  vtkTypeMacro(vtkCellAttributeCalculator, vtkObject);
  static vtkCellAttributeCalculator* New();
  void PrintSelf(ostream& os, vtkIndent indent) override;

  template <typename CalculatorType>
  vtkSmartPointer<CalculatorType> Prepare(vtkCellMetadata* cell, vtkCellAttribute* field)
  {
    auto result = this->PrepareForGrid(cell, field);
    return CalculatorType::SafeDownCast(result);
  }

  virtual vtkSmartPointer<vtkCellAttributeCalculator> PrepareForGrid(
    vtkCellMetadata* vtkNotUsed(cell), vtkCellAttribute* vtkNotUsed(field))
  {
    return nullptr;
  }

protected:
  vtkCellAttributeCalculator() = default;
  ~vtkCellAttributeCalculator() override = default;

private:
  vtkCellAttributeCalculator(const vtkCellAttributeCalculator&) = delete;
  void operator=(const vtkCellAttributeCalculator&) = delete;
};

VTK_ABI_NAMESPACE_END
#endif