vtkADIOS2CoreImageReader.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 vtkADIOS2CoreImageReader_h
#define vtkADIOS2CoreImageReader_h
 
#include <map>    // For independently time stepped array indexing
#include <memory> // For std::unique_ptr
#include <string> // For variable name index mapping
#include <vector> // For independently time stepped array indexing
 
#include "vtkDataObjectAlgorithm.h"
#include "vtkSetGet.h"       // For property get/set macros
#include "vtkSmartPointer.h" // For the object cache
 
#include "vtkIOADIOS2Module.h" // For export macro
 
VTK_ABI_NAMESPACE_BEGIN
class vtkCellArray;
class vtkDataArray;
class vtkDataObject;
class vtkDataSet;
class vtkDataSetAttributes;
class vtkFieldData;
class vtkImageData;
class vtkMultiBlockDataSet;
class vtkMultiProcessController;
class vtkStringArray;
 
//----------------------------------------------------------------------------
 
class VTKIOADIOS2_EXPORT vtkADIOS2CoreImageReader : public vtkDataObjectAlgorithm
{
public:
  enum class VarType
  {
    PointData,
    CellData
  };
  using Params = std::map<std::string, std::string>;
  using StringToParams = std::map<std::string, Params>;
  using InquireVariablesType = std::map<std::string, VarType>;
  static vtkADIOS2CoreImageReader* New();
  vtkTypeMacro(vtkADIOS2CoreImageReader, vtkDataObjectAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  int CanReadFile(VTK_FILEPATH const std::string& name);
 
  virtual int CanReadFile(VTK_FILEPATH const char* filename);


  vtkSetMacro(FileName, std::string);
  vtkGetMacro(FileName, std::string);
 
  void SetFileName(VTK_FILEPATH const char* filename);


  vtkSetVector3Macro(Origin, double);
  vtkGetVector3Macro(Origin, double);


  vtkSetVector3Macro(Spacing, double);
  vtkGetVector3Macro(Spacing, double);


  vtkStringArray* GetAllDimensionArrays();
  vtkSetMacro(DimensionArray, std::string);
  vtkGetMacro(DimensionArray, std::string);


  vtkSetMacro(DimensionArrayAsCell, bool);
  vtkGetMacro(DimensionArrayAsCell, bool);
  vtkBooleanMacro(DimensionArrayAsCell, bool);
 
  vtkStringArray* GetAllTimeStepArrays();

  vtkSetMacro(TimeStepArray, std::string);
  vtkGetMacro(TimeStepArray, std::string);


  int GetNumberOfArrays();
  const char* GetArrayName(int index);


  void SetArrayStatus(const char* name, int status);
  int GetArrayStatus(const char* name);

  vtkMTimeType GetMTime() override;


  vtkSetMacro(IsColumnMajor, bool);
  vtkGetMacro(IsColumnMajor, bool);
  vtkBooleanMacro(IsColumnMajor, bool);


  void SetActiveScalar(const std::pair<std::string, VarType>& inqVars);
  std::pair<std::string, VarType>& GetActiveScalar();
  const std::pair<std::string, VarType>& GetActiveScalar() const;


  StringToParams& GetAvilableVariables();
  const StringToParams& GetAvilableVariables() const;


  StringToParams& GetAvailableAttributes();
  const StringToParams& GetAvailableAttributes() const;


  void SetController(vtkMultiProcessController*);

  int ProcessRequest(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
 
protected:
  vtkADIOS2CoreImageReader();
  ~vtkADIOS2CoreImageReader() override;
 
  int RequestDataObjectInternal(vtkInformationVector*);
 
  int RequestInformation(
    vtkInformation* request, vtkInformationVector** input, vtkInformationVector* output) override;
  int RequestData(
    vtkInformation* request, vtkInformationVector** input, vtkInformationVector* output) override;
 
  std::string FetchTypeStringFromVarName(const std::string& name);
 
  void UpdateDimensionFromDimensionArray();
 
  // Read available variables and attributes in the file
  bool OpenAndReadMetaData();
 
  // Convert the array selection into inquire variables.
  void ConvertArraySelectionToInqVar();
 
  // Init the workDistribution based on the first inquired variable
  bool InitWorkDistribution();
 
  void ReadImageBlocks(vtkMultiBlockDataSet* mbds);
 
  // Gather time steps info from the time step array
  bool GatherTimeSteps();
 
  // Helper function for InitWorkDistribution to calculate how many blocks each process shall read
  template <typename T>
  void CalculateWorkDistribution(const std::string& varName);
 
  // Helper function for ReadImageBlocks to populate vtk data array from adios variable
  template <typename T>
  vtkSmartPointer<vtkAbstractArray> PopulateDataArrayFromVar(
    const std::string& varName, size_t blockIndex);
 
  // Helper function to gather time steps from adios time array
  template <typename T>
  void GatherTimeStepsFromADIOSTimeArray();
 
  std::string FileName;
 
  bool DimensionArrayAsCell;
  bool IsColumnMajor;
 
  std::string DimensionArray;
  std::string TimeStepArray;
 
  double Origin[3];
  double Spacing[3];
  int Dimension[3];
 
  double RequestTimeStep;
 
  vtkSmartPointer<vtkMultiProcessController> Controller;
 
  struct vtkADIOS2CoreImageReaderImpl;
  std::unique_ptr<vtkADIOS2CoreImageReaderImpl> Impl;
 
private:
  vtkADIOS2CoreImageReader(const vtkADIOS2CoreImageReader&) = delete;
  void operator=(const vtkADIOS2CoreImageReader&) = delete;
};
VTK_ABI_NAMESPACE_END
#endif