vtkCellGridChangeBasis.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 vtkCellGridChangeBasis_h
#define vtkCellGridChangeBasis_h
 
#include "vtkCellGridAlgorithm.h"
 
#include "vtkFiltersCellGrid.h" // For export macro
#include <map>                  // For ivar.
 
VTK_ABI_NAMESPACE_BEGIN

class VTKFILTERSCELLGRID_EXPORT vtkCellGridChangeBasis : public vtkCellGridAlgorithm
{
public:

  static vtkCellGridChangeBasis* New();
  vtkTypeMacro(vtkCellGridChangeBasis, vtkCellGridAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  vtkMTimeType GetMTime() override;

  class Query : public vtkCellGridQuery
  {
  public:

    static Query* New();
    vtkTypeMacro(vtkCellGridChangeBasis::Query, vtkCellGridQuery);
    void PrintSelf(ostream& os, vtkIndent indent) override;

    Query();
    ~Query() override;

    vtkMTimeType GetMTime() override;


    virtual void SetCellAttribute(vtkCellAttribute*);
    vtkGetObjectMacro(CellAttribute, vtkCellAttribute);


    vtkSetStringTokenMacro(SourceFunctionSpace);
    vtkGetStringTokenMacro(SourceFunctionSpace);


    vtkSetStringTokenMacro(SourceBasis);
    vtkGetStringTokenMacro(SourceBasis);


    vtkSetMacro(SourceOrder, int);
    vtkGetMacro(SourceOrder, int);


    vtkSetStringTokenMacro(TargetFunctionSpace);
    vtkGetStringTokenMacro(TargetFunctionSpace);


    vtkSetStringTokenMacro(TargetBasis);
    vtkGetStringTokenMacro(TargetBasis);


    vtkSetMacro(TargetOrder, int);
    vtkGetMacro(TargetOrder, int);

    vtkDataArray* CreateNewDataArray(vtkDataArray* input) const;

    std::map<vtkDataArray*, vtkSmartPointer<vtkDataArray>>& GetSharedOutputs()
    {
      return this->SharedOutputs;
    }

    bool Initialize() override;
 
  protected:
    vtkCellAttribute* CellAttribute{ nullptr };
    vtkStringToken SourceFunctionSpace;
    vtkStringToken SourceBasis;
    int SourceOrder{ -1 };
    vtkStringToken TargetFunctionSpace;
    vtkStringToken TargetBasis;
    int TargetOrder{ -1 };
    int OutputPointsPrecision{ vtkAlgorithm::DesiredOutputPrecision::DEFAULT_PRECISION };
    std::map<vtkDataArray*, vtkSmartPointer<vtkDataArray>> SharedOutputs;
  };

  void SetCellAttribute(vtkCellAttribute* att);
  void SetSourceFunctionSpace(const char* functionSpace);
  void SetSourceBasis(const char* basis);
  void SetSourceOrder(int order);
  void SetTargetFunctionSpace(const char* functionSpace);
  void SetTargetBasis(const char* basis);
  void SetTargetOrder(int order);
 
protected:
  vtkCellGridChangeBasis();
  ~vtkCellGridChangeBasis() override;
 
  int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
 
  vtkNew<Query> Request;
 
private:
  vtkCellGridChangeBasis(const vtkCellGridChangeBasis&) = delete;
  void operator=(const vtkCellGridChangeBasis&) = delete;
};
 
VTK_ABI_NAMESPACE_END
#endif