vtkPVArrayCalculator.h

Go to the documentation of this file.

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

#include "vtkArrayCalculator.h"
#include "vtkPVVTKExtensionsFiltersGeneralModule.h" //needed for exports

class vtkDataObject;
class vtkDataSetAttributes;

class VTKPVVTKEXTENSIONSFILTERSGENERAL_EXPORT vtkPVArrayCalculator : public vtkArrayCalculator
{
public:
  vtkTypeMacro(vtkPVArrayCalculator, vtkArrayCalculator);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  static vtkPVArrayCalculator* New();


  void SetFunctionParserTypeFromInt(int type)
  {
    this->SetFunctionParserType(static_cast<FunctionParserTypes>(type));
  }

protected:
  vtkPVArrayCalculator();
  ~vtkPVArrayCalculator() override;

  int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;

  int GetAttributeTypeFromInput(vtkDataObject* input);

  void ResetArrayAndVariableNames();

  void AddCoordinateVariableNames();

  void AddArrayAndVariableNames(vtkDataObject* theInputObj, vtkDataSetAttributes* inDataAttrs);

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

#endif