/*=========================================================================

  Program:   Visualization Toolkit
  Module:    vtkExtractPiece.h

  Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
  All rights reserved.
  See Copyright.txt or http://www.kitware.com/Copyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notice for more information.

=========================================================================*/
/**
 * @class   vtkExtractPiece
 *
 * vtkExtractPiece returns the appropriate piece of each
 * sub-dataset in the vtkCompositeDataSet.
 * This filter can handle sub-datasets of type vtkImageData, vtkPolyData,
 * vtkRectilinearGrid, vtkStructuredGrid, and vtkUnstructuredGrid; it does
 * not handle sub-grids of type vtkCompositeDataSet.
 */

#ifndef vtkExtractPiece_h
#define vtkExtractPiece_h

#include "vtkCompositeDataSetAlgorithm.h"
#include "vtkFiltersParallelImagingModule.h" // For export macro

class vtkImageData;
class vtkPolyData;
class vtkRectilinearGrid;
class vtkStructuredGrid;
class vtkUnstructuredGrid;
class vtkCompositeDataIterator;

class VTKFILTERSPARALLELIMAGING_EXPORT vtkExtractPiece : public vtkCompositeDataSetAlgorithm
{
public:
  static vtkExtractPiece* New();
  vtkTypeMacro(vtkExtractPiece, vtkCompositeDataSetAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;

protected:
  vtkExtractPiece() {}
  ~vtkExtractPiece() override {}

  int RequestDataObject(vtkInformation* request, vtkInformationVector** inputVector,
    vtkInformationVector* outputVector) override;

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

  void ExtractImageData(vtkImageData* imageData, vtkCompositeDataSet* output, int piece,
    int numberOfPieces, int ghostLevel, vtkCompositeDataIterator* iter);
  void ExtractPolyData(vtkPolyData* polyData, vtkCompositeDataSet* output, int piece,
    int numberOfPieces, int ghostLevel, vtkCompositeDataIterator* iter);
  void ExtractRectilinearGrid(vtkRectilinearGrid* rGrid, vtkCompositeDataSet* output, int piece,
    int numberOfPieces, int ghostLevel, vtkCompositeDataIterator* iter);
  void ExtractStructuredGrid(vtkStructuredGrid* sGrid, vtkCompositeDataSet* output, int piece,
    int numberOfPieces, int ghostLevel, vtkCompositeDataIterator* iter);
  void ExtractUnstructuredGrid(vtkUnstructuredGrid* uGrid, vtkCompositeDataSet* output, int piece,
    int numberOfPieces, int ghostLevel, vtkCompositeDataIterator* iter);

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

#endif
