vtkAMREnzoReader.h

Go to the documentation of this file.

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

 Program:   Visualization Toolkit
 Module:    vtkAMREnzoReader.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 vtkAMREnzoReader_h
#define vtkAMREnzoReader_h

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

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

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( 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( const int blockIdx ) override;

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

  int FillMetaData( ) override;

  vtkUniformGrid* GetAMRGrid( const int blockIdx ) override;

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

  void GetAMRGridPointData(
      const 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;

  vtkEnzoReaderInternal *Internal;

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

#endif /* vtkAMREnzoReader_h */