vtkHyperTreeGridGenerateFieldStrategy.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 vtkHyperTreeGridGenerateFieldStrategy_h
#define vtkHyperTreeGridGenerateFieldStrategy_h

#include "vtkFiltersHyperTreeModule.h" // For export macro
#include "vtkHyperTreeGrid.h"          // For vtkHyperTreeGrid
#include "vtkObject.h"

#include <unordered_map>

VTK_ABI_NAMESPACE_BEGIN

class VTKFILTERSHYPERTREE_EXPORT vtkHyperTreeGridGenerateFieldStrategy : public vtkObject
{
public:
  struct Field
  {
    std::string name;
    vtkSmartPointer<vtkHyperTreeGridGenerateFieldStrategy> strategy;
    bool enabled;      // Whether the user asked for this array to be computed.
    bool valid = true; // Whether the array can be computed. Only make sense for Field Data (Cell
                       // Data fields are always valid if enabled).
  };

  vtkHyperTreeGridGenerateFieldStrategy() = default;
  vtkAbstractTypeMacro(vtkHyperTreeGridGenerateFieldStrategy, vtkObject);
  void PrintSelf(ostream& os, vtkIndent indent) override
  {
    this->Superclass::PrintSelf(os, indent);
    os << indent << "Array name: " << this->ArrayName << "\n";
    os << indent << "Array type: "
       << (this->ArrayType == vtkDataObject::AttributeTypes::CELL ? "CELL_DATA" : "FIELD_DATA")
       << "\n";
  }


  virtual void Initialize([[maybe_unused]] vtkHyperTreeGrid* inputHTG) {}
  virtual bool Initialize(std::unordered_map<std::string, Field>) { return true; }


  virtual void Compute(vtkHyperTreeGridNonOrientedGeometryCursor*) {}
  virtual void Compute(vtkHyperTreeGridNonOrientedGeometryCursor*, vtkCellData*,
    std::unordered_map<std::string, Field>)
  {
  }


  virtual vtkDataArray* GetAndFinalizeArray() = 0;


  vtkDataObject::AttributeTypes GetArrayType() { return this->ArrayType; }
  void SetArrayType(vtkDataObject::AttributeTypes arrayType)
  {
    assert(arrayType == vtkDataObject::AttributeTypes::CELL ||
      arrayType == vtkDataObject::AttributeTypes::FIELD);
    this->ArrayType = arrayType;
  }


  std::string GetArrayName() { return this->ArrayName; }
  void SetArrayName(std::string arrayName) { this->ArrayName = arrayName; }

protected:
  std::string ArrayName;
  vtkDataObject::AttributeTypes ArrayType = vtkDataObject::AttributeTypes::CELL;

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

VTK_ABI_NAMESPACE_END
#endif // vtkHyperTreeGridGenerateFieldStrategy_h