vtkOpenVDBReader.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 vtkOpenVDBReader_h
#define vtkOpenVDBReader_h
 
#include "vtkDataArraySelection.h" // needed for vtkDataArraySelection
#include "vtkIOOpenVDBModule.h"    //needed for exports
#include "vtkPartitionedDataSetCollectionAlgorithm.h"
#include "vtkSmartPointer.h" // needed for smart pointers
 
VTK_ABI_NAMESPACE_BEGIN
class vtkResourceStream;
class vtkOpenVDBReaderInternals;
 
class VTKIOOPENVDB_EXPORT vtkOpenVDBReader : public vtkPartitionedDataSetCollectionAlgorithm
{
public:
  static vtkOpenVDBReader* New();
  vtkTypeMacro(vtkOpenVDBReader, vtkPartitionedDataSetCollectionAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;


  bool CanReadFile(VTK_FILEPATH const char* name);
  static bool CanReadFile(vtkResourceStream* stream);

  const char* GetFileExtensions();


  vtkSetFilePathMacro(FileName);
  vtkGetFilePathMacro(FileName);


  void SetStream(vtkResourceStream* stream);
  vtkResourceStream* GetStream() const;

  const char* GetDescriptiveName();


  vtkSetClampMacro(DownsamplingFactor, float, 0.01, 1.0);
  vtkGetMacro(DownsamplingFactor, float);


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


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


  vtkGetObjectMacro(GridSelection, vtkDataArraySelection);
  int GetNumberOfGridsSelectionArrays();
  const char* GetGridsSelectionArrayName(int index);
  int GetGridsSelectionArrayStatus(const char* name);
  void SetGridsSelectionArrayStatus(const char* name, int status);

  const char* GetGridArrayName(int index);

  int GetGridArrayType(int index);

  int NumberOfGrids();

  vtkMTimeType GetMTime() override;
 
protected:
  vtkOpenVDBReader();
  ~vtkOpenVDBReader() override;
 
  bool LoadFile();
  void InitializeData();
 
  float DownsamplingFactor = 1.0;
 
  char* FileName = nullptr;
 
  bool MergeImageVolumes = false;
  bool MergePointSets = false;
 
  bool DataCorrect = true;
 
  vtkNew<vtkDataArraySelection> GridSelection;
 
  constexpr static const char* FILE_EXTENSIONS = ".vdb";
  constexpr static const char* DESCRIPTIVE_NAME = "OpenVDB volumetric data file format";
 
  int RequestInformation(vtkInformation* request, vtkInformationVector** inputVector,
    vtkInformationVector* outputVector) override;
 
  int RequestData(vtkInformation* request, vtkInformationVector** inputVector,
    vtkInformationVector* outputVector) override;
 
  int RequestDataObject(vtkInformation* request, vtkInformationVector** inputVector,
    vtkInformationVector* outputVector) override;
 
private:
  vtkOpenVDBReader(const vtkOpenVDBReader&) = delete;
  void operator=(const vtkOpenVDBReader&) = delete;
 
  vtkSmartPointer<vtkResourceStream> Stream;
 
  std::unique_ptr<vtkOpenVDBReaderInternals> Internals;
};
 
VTK_ABI_NAMESPACE_END
#endif