MeshCacheMockAlgorithms.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 MeshCacheMockAlgorithms_h
#define MeshCacheMockAlgorithms_h

#include "vtkNew.h"
#include "vtkPartitionedDataSetCollectionAlgorithm.h"
#include "vtkPassInputTypeAlgorithm.h"
#include "vtkPolyDataAlgorithm.h"
#include "vtkSetGet.h"

#include <array>

VTK_ABI_NAMESPACE_BEGIN

class vtkCompositeDataSet;
class vtkPartitionedDataSetCollection;

//------------------------------------------------------------------------------
namespace mockArraysName
{
const std::string pointIds = "pointIds";
const std::string pointData = "pointData";
};

//------------------------------------------------------------------------------
class vtkStaticDataSource : public vtkPolyDataAlgorithm
{
  vtkNew<vtkPolyData> SourceOutput;
  int StartData = 0;
  bool GenerateGhosts = false;

public:
  vtkTypeMacro(vtkStaticDataSource, vtkPolyDataAlgorithm);
  static vtkStaticDataSource* New();

  vtkStaticDataSource();
  ~vtkStaticDataSource() override = default;

  vtkSetMacro(StartData, int);

  vtkSetMacro(GenerateGhosts, bool);

  void MarkGhostsModified();

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

  // mark polydata points as modified
  void MarkMeshModified();
};

//------------------------------------------------------------------------------
class vtkStaticCompositeSource : public vtkPartitionedDataSetCollectionAlgorithm
{
  vtkNew<vtkPartitionedDataSetCollection> SourceOutput;
  vtkNew<vtkStaticDataSource> FirstData;
  vtkNew<vtkStaticDataSource> SecondData;

public:
  vtkTypeMacro(vtkStaticCompositeSource, vtkPartitionedDataSetCollectionAlgorithm);
  static vtkStaticCompositeSource* New();

  vtkStaticCompositeSource();
  ~vtkStaticCompositeSource() override = default;

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

  void SetStartData(int start);

  void MarkMeshModified();
};

//------------------------------------------------------------------------------
class vtkConsumerDataFilter : public vtkPassInputTypeAlgorithm
{
public:
  vtkTypeMacro(vtkConsumerDataFilter, vtkPassInputTypeAlgorithm);
  static vtkConsumerDataFilter* New();

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

  vtkCompositeDataSet* GetCompositeOutput();
};

VTK_ABI_NAMESPACE_END

#endif