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