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.
157 lines
3.9 KiB
C++
157 lines
3.9 KiB
C++
/*=========================================================================
|
|
|
|
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.
|
|
|
|
=========================================================================*/
|
|
/**
|
|
* @class vtkAMREnzoReader
|
|
*
|
|
*
|
|
* A concrete instance of vtkAMRBaseReader that implements functionality
|
|
* for reading Enzo AMR datasets.
|
|
*/
|
|
|
|
#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
|
|
|
|
class vtkOverlappingAMR;
|
|
class vtkEnzoReaderInternal;
|
|
|
|
class VTKIOAMR_EXPORT vtkAMREnzoReader : public vtkAMRBaseReader
|
|
{
|
|
public:
|
|
static vtkAMREnzoReader* New();
|
|
vtkTypeMacro(vtkAMREnzoReader, vtkAMRBaseReader);
|
|
void PrintSelf(ostream& os, vtkIndent indent) override;
|
|
|
|
//@{
|
|
/**
|
|
* Set/Get whether data should be converted to CGS
|
|
*/
|
|
vtkSetMacro(ConvertToCGS, vtkTypeBool);
|
|
vtkGetMacro(ConvertToCGS, vtkTypeBool);
|
|
vtkBooleanMacro(ConvertToCGS, vtkTypeBool);
|
|
//@}
|
|
|
|
/**
|
|
* See vtkAMRBaseReader::GetNumberOfBlocks
|
|
*/
|
|
int GetNumberOfBlocks() override;
|
|
|
|
/**
|
|
* See vtkAMRBaseReader::GetNumberOfLevels
|
|
*/
|
|
int GetNumberOfLevels() override;
|
|
|
|
/**
|
|
* See vtkAMRBaseReader::SetFileName
|
|
*/
|
|
void SetFileName(const char* fileName) override;
|
|
|
|
protected:
|
|
vtkAMREnzoReader();
|
|
~vtkAMREnzoReader() override;
|
|
|
|
/**
|
|
* Parses the parameters file and extracts the
|
|
* conversion factors that are used to convert
|
|
* to CGS units.
|
|
*/
|
|
void ParseConversionFactors();
|
|
|
|
/**
|
|
* Given an array name of the form "array[idx]" this method
|
|
* extracts and returns the corresponding index idx.
|
|
*/
|
|
int GetIndexFromArrayName(std::string arrayName);
|
|
|
|
/**
|
|
* Given the label string, this method parses the attribute label and
|
|
* the string index.
|
|
*/
|
|
void ParseLabel(const std::string& labelString, int& idx, std::string& label);
|
|
|
|
/**
|
|
* Given the label string, this method parses the corresponding attribute
|
|
* index and conversion factor
|
|
*/
|
|
void ParseCFactor(const std::string& labelString, int& idx, double& factor);
|
|
|
|
/**
|
|
* Given the variable name, return the conversion factor used to convert
|
|
* the data to CGS. These conversion factors are read directly from the
|
|
* parameters file when the filename is set.
|
|
*/
|
|
double GetConversionFactor(const std::string& name);
|
|
|
|
/**
|
|
* See vtkAMRBaseReader::ReadMetaData
|
|
*/
|
|
void ReadMetaData() override;
|
|
|
|
/**
|
|
* See vtkAMRBaseReader::GetBlockLevel
|
|
*/
|
|
int GetBlockLevel(const int blockIdx) override;
|
|
|
|
void ComputeStats(
|
|
vtkEnzoReaderInternal* internal, std::vector<int>& blocksPerLevel, double min[3]);
|
|
|
|
/**
|
|
* See vtkAMRBaseReader::FillMetaData
|
|
*/
|
|
int FillMetaData() override;
|
|
|
|
/**
|
|
* See vtkAMRBaseReader::GetAMRGrid
|
|
*/
|
|
vtkUniformGrid* GetAMRGrid(const int blockIdx) override;
|
|
|
|
/**
|
|
* See vtkAMRBaseReader::GetAMRGridData
|
|
*/
|
|
void GetAMRGridData(const int blockIdx, vtkUniformGrid* block, const char* field) override;
|
|
|
|
/**
|
|
* See vtkAMRBaseReader::GetAMRGridData
|
|
*/
|
|
void GetAMRGridPointData(const int vtkNotUsed(blockIdx), vtkUniformGrid* vtkNotUsed(block),
|
|
const char* vtkNotUsed(field)) override
|
|
{
|
|
}
|
|
|
|
/**
|
|
* See vtkAMRBaseReader::SetUpDataArraySelections
|
|
*/
|
|
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 */
|