vtkAMREnzoReader.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 vtkAMREnzoReader_h
#define vtkAMREnzoReader_h

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

#include <map>    // For STL map
#include <string> // For std::string

VTK_ABI_NAMESPACE_BEGIN
class vtkOverlappingAMR;
class vtkEnzoReaderInternal;

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


  vtkSetMacro(ConvertToCGS, vtkTypeBool);
  vtkGetMacro(ConvertToCGS, vtkTypeBool);
  vtkBooleanMacro(ConvertToCGS, vtkTypeBool);

  int GetNumberOfBlocks() override;

  int GetNumberOfLevels() override;

  void SetFileName(VTK_FILEPATH const char* fileName) override;

protected:
  vtkAMREnzoReader();
  ~vtkAMREnzoReader() override;

  void ParseConversionFactors();

  int GetIndexFromArrayName(std::string arrayName);

  void ParseLabel(const std::string& labelString, int& idx, std::string& label);

  void ParseCFactor(const std::string& labelString, int& idx, double& factor);

  double GetConversionFactor(const std::string& name);

  void ReadMetaData() override;

  int GetBlockLevel(int blockIdx) override;

  int FillMetaData() override;

  vtkUniformGrid* GetAMRGrid(int blockIdx) override;

  void GetAMRGridData(int blockIdx, vtkUniformGrid* block, const char* field) override;

  void GetAMRGridPointData(int vtkNotUsed(blockIdx), vtkUniformGrid* vtkNotUsed(block),
    const char* vtkNotUsed(field)) override
  {
  }

  void SetUpDataArraySelections() override;

  vtkTypeBool ConvertToCGS;
  bool IsReady;

private:
  vtkAMREnzoReader(const vtkAMREnzoReader&) = delete;
  void operator=(const vtkAMREnzoReader&) = delete;

  void ComputeStats(
    vtkEnzoReaderInternal* internal, std::vector<unsigned int>& blocksPerLevel, double min[3]);

  vtkEnzoReaderInternal* Internal;

  std::map<std::string, int> label2idx;
  std::map<int, double> conversionFactors;
};

VTK_ABI_NAMESPACE_END
#endif /* vtkAMREnzoReader_h */