vtkExtractUserDefinedPiece.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-FileCopyrightText: Copyright (c) Sandia Corporation
// SPDX-License-Identifier: BSD-3-Clause
 
#ifndef vtkExtractUserDefinedPiece_h
#define vtkExtractUserDefinedPiece_h
 
#include "vtkExtractUnstructuredGridPiece.h"
#include "vtkFiltersParallelModule.h" // For export macro
 
VTK_ABI_NAMESPACE_BEGIN
class VTKFILTERSPARALLEL_EXPORT vtkExtractUserDefinedPiece : public vtkExtractUnstructuredGridPiece
{
public:
  vtkTypeMacro(vtkExtractUserDefinedPiece, vtkExtractUnstructuredGridPiece);
  static vtkExtractUserDefinedPiece* New();
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
  typedef int (*UserDefFunc)(vtkIdType cellID, vtkUnstructuredGrid* grid, void* constantData);
 
  // Set the function used to identify the piece.  The function should
  // return 1 if the cell is in the piece, and 0 otherwise.
  void SetPieceFunction(UserDefFunc func)
  {
    this->InPiece = func;
    this->Modified();
  }
 
  // Set constant data to be used by the piece identifying function.
  void SetConstantData(void* data, int len);
 
  // Get constant data to be used by the piece identifying function.
  // Return the length of the data buffer.
  int GetConstantData(void** data);
 
  // The function should return 1 if the cell
  // is in the piece, and 0 otherwise.
 
protected:
  vtkExtractUserDefinedPiece();
  ~vtkExtractUserDefinedPiece() override;
 
  int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
 
  void ComputeCellTagsWithFunction(
    vtkIntArray* tags, vtkIdList* pointOwnership, vtkUnstructuredGrid* input);
 
private:
  vtkExtractUserDefinedPiece(const vtkExtractUserDefinedPiece&) = delete;
  void operator=(const vtkExtractUserDefinedPiece&) = delete;
 
  void* ConstantData;
  int ConstantDataLen;
 
  UserDefFunc InPiece;
};
VTK_ABI_NAMESPACE_END
#endif