vtkAMREnzoParticlesReader.h

Go to the documentation of this file.

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

 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.

 =========================================================================*/
#ifndef vtkAMREnzoParticlesReader_h
#define vtkAMREnzoParticlesReader_h

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


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;


  vtkSetMacro( ParticleType, int );
  vtkGetMacro( ParticleType, int );

  int GetTotalNumberOfParticles() override;

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

  vtkPolyData* GetParticles( const char* file, const int blockIdx );

  void ReadMetaData() override;

  void SetupParticleDataSelections() override;

  bool CheckParticleType( const int pIdx, vtkIntArray *ptypes );

  vtkDataArray *GetParticlesTypeArray( const int blockIdx );

  vtkPolyData* ReadParticles( const int blkidx ) override;

  int ParticleType;

  vtkEnzoReaderInternal *Internal;

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

#endif /* vtkAMREnzoParticlesReader_h */