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

 Program:   Visualization Toolkit
 Module:    vtkAMREnzoParticlesReader.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   vtkAMREnzoParticlesReader
 *
 *
 *  A concrete instance of the vtkAMRBaseParticlesReader which provides
 *  functionality for loading ENZO AMR particle datasets from ENZO.
 *
 * @sa
 *  vtkAMRBaseParticlesReader
 */

#ifndef vtkAMREnzoParticlesReader_h
#define vtkAMREnzoParticlesReader_h

#include "vtkAMRBaseParticlesReader.h"
#include "vtkIOAMRModule.h" // For export macro

class vtkPolyData;
class vtkDataArray;
class vtkIntArray;
class vtkEnzoReaderInternal;

class VTKIOAMR_EXPORT vtkAMREnzoParticlesReader : public vtkAMRBaseParticlesReader
{
public:
  static vtkAMREnzoParticlesReader* New();
  vtkTypeMacro(vtkAMREnzoParticlesReader, vtkAMRBaseParticlesReader);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  //@{
  /**
   * Returns the requested particle type.
   */
  vtkSetMacro(ParticleType, int);
  vtkGetMacro(ParticleType, int);
  //@}

  /**
   * See vtkAMRBaseParticlesReader::GetTotalNumberOfParticles.
   */
  int GetTotalNumberOfParticles() override;

protected:
  vtkAMREnzoParticlesReader();
  ~vtkAMREnzoParticlesReader() override;

  /**
   * Read the particles from the given particles file for the block
   * corresponding to the given block index.
   */
  vtkPolyData* GetParticles(const char* file, const int blockIdx);

  /**
   * See vtkAMRBaseParticlesReader::ReadMetaData()
   */
  void ReadMetaData() override;

  /**
   * See vtkAMRBaseParticlesReader::SetupParticleDataSelections
   */
  void SetupParticleDataSelections() override;

  /**
   * Filter's by particle type, iff particle_type is included in
   * the given file.
   */
  bool CheckParticleType(const int pIdx, vtkIntArray* ptypes);

  /**
   * Returns the ParticlesType Array
   */
  vtkDataArray* GetParticlesTypeArray(const int blockIdx);

  /**
   * Reads the particles.
   */
  vtkPolyData* ReadParticles(const int blkidx) override;

  int ParticleType;

  vtkEnzoReaderInternal* Internal;

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

#endif /* vtkAMREnzoParticlesReader_h */