vtkPassSelectedArrays.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 vtkPassSelectedArrays_h
#define vtkPassSelectedArrays_h

#include "vtkDataObject.h"           // for vtkDataObject::FieldAssociations
#include "vtkFiltersGeneralModule.h" // For export macro
#include "vtkPassInputTypeAlgorithm.h"
#include "vtkSmartPointer.h" // for ivar

VTK_ABI_NAMESPACE_BEGIN
class vtkCellGrid;
class vtkDataArraySelection;

class VTKFILTERSGENERAL_EXPORT vtkPassSelectedArrays : public vtkPassInputTypeAlgorithm
{
public:
  static vtkPassSelectedArrays* New();
  vtkTypeMacro(vtkPassSelectedArrays, vtkPassInputTypeAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;


  vtkSetMacro(Enabled, bool);
  vtkGetMacro(Enabled, bool);
  vtkBooleanMacro(Enabled, bool);

  vtkDataArraySelection* GetArraySelection(int association);


  vtkDataArraySelection* GetPointDataArraySelection()
  {
    return this->GetArraySelection(vtkDataObject::FIELD_ASSOCIATION_POINTS);
  }
  vtkDataArraySelection* GetCellDataArraySelection()
  {
    return this->GetArraySelection(vtkDataObject::FIELD_ASSOCIATION_CELLS);
  }
  vtkDataArraySelection* GetFieldDataArraySelection()
  {
    return this->GetArraySelection(vtkDataObject::FIELD_ASSOCIATION_NONE);
  }
  vtkDataArraySelection* GetVertexDataArraySelection()
  {
    return this->GetArraySelection(vtkDataObject::FIELD_ASSOCIATION_VERTICES);
  }
  vtkDataArraySelection* GetEdgeDataArraySelection()
  {
    return this->GetArraySelection(vtkDataObject::FIELD_ASSOCIATION_EDGES);
  }
  vtkDataArraySelection* GetRowDataArraySelection()
  {
    return this->GetArraySelection(vtkDataObject::FIELD_ASSOCIATION_ROWS);
  }

protected:
  vtkPassSelectedArrays();
  ~vtkPassSelectedArrays() override;

  int FillInputPortInformation(int port, vtkInformation* info) override;
  int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;

  int HandleCellGridAttributes(vtkCellGrid* output);

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

  bool Enabled;
  vtkSmartPointer<vtkDataArraySelection> ArraySelections[vtkDataObject::NUMBER_OF_ASSOCIATIONS];
};

VTK_ABI_NAMESPACE_END
#endif