You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

103 lines
2.6 KiB
C++

/*=========================================================================
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 */