vtkAMReXParticlesReader.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#ifndef vtkAMReXParticlesReader_h
#define vtkAMReXParticlesReader_h
 
#include "vtkIOAMRModule.h" // For export macro
#include "vtkMultiBlockDataSetAlgorithm.h"
#include "vtkNew.h" // for vtkNew
 
#include <string> // for std::string.
 
VTK_ABI_NAMESPACE_BEGIN
class vtkDataArraySelection;
class vtkMultiPieceDataSet;
class vtkMultiProcessController;
 
class VTKIOAMR_EXPORT vtkAMReXParticlesReader : public vtkMultiBlockDataSetAlgorithm
{
public:
  static vtkAMReXParticlesReader* New();
  vtkTypeMacro(vtkAMReXParticlesReader, vtkMultiBlockDataSetAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
 
  void SetPlotFileName(VTK_FILEPATH const char* fname);
  VTK_FILEPATH const char* GetPlotFileName() const;
 
 
  void SetParticleType(const std::string& str);
  const std::string& GetParticleType() const { return this->ParticleType; }
 
  vtkDataArraySelection* GetPointDataArraySelection() const;
 
  static int CanReadFile(VTK_FILEPATH const char* fname, const char* particlesType = nullptr);
 
 
  void SetController(vtkMultiProcessController* controller);
  vtkGetObjectMacro(Controller, vtkMultiProcessController);
 
protected:
  vtkAMReXParticlesReader();
  ~vtkAMReXParticlesReader() override;
 
  int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
  int RequestInformation(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
 
  vtkMultiProcessController* Controller;
  std::string PlotFileName;
  bool GenerateGlobalIds;
 
private:
  vtkAMReXParticlesReader(const vtkAMReXParticlesReader&) = delete;
  void operator=(const vtkAMReXParticlesReader&) = delete;
 
  bool ReadMetaData();
 
  bool ReadLevel(int level, vtkMultiPieceDataSet* pdataset, int piece_idx, int num_pieces) const;
 
  vtkTimeStamp PlotFileNameMTime;
  vtkTimeStamp MetaDataMTime;
  std::string ParticleType;
  double dataTimeStep;
  vtkNew<vtkDataArraySelection> PointDataArraySelection;
 
  class AMReXParticleHeader;
  AMReXParticleHeader* Header;
  friend class AMReXParticleHeader;
};
 
VTK_ABI_NAMESPACE_END
#endif