vtkOpenVDBWriter.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 vtkOpenVDBWriter_h
#define vtkOpenVDBWriter_h
 
#include "vtkIOOpenVDBModule.h" //needed for exports
#include "vtkSmartPointer.h"    // For protected ivars
#include "vtkWriter.h"
 
VTK_ABI_NAMESPACE_BEGIN
class vtkDataSetAttributes;
class vtkImageData;
class vtkMultiProcessController;
class vtkOpenVDBWriterInternals;
class vtkPointSet;
class vtkScalarsToColors;
class vtkUnsignedCharArray;
 
class VTKIOOPENVDB_EXPORT vtkOpenVDBWriter : public vtkWriter
{
public:
  static vtkOpenVDBWriter* New();
  vtkTypeMacro(vtkOpenVDBWriter, vtkWriter);
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
 
  vtkSetFilePathMacro(FileName);
  vtkGetFilePathMacro(FileName);
 
 
  vtkSetMacro(WriteAllTimeSteps, bool);
  vtkGetMacro(WriteAllTimeSteps, bool);
 
 
  virtual void SetLookupTable(vtkScalarsToColors*);
  vtkGetObjectMacro(LookupTable, vtkScalarsToColors);
 
 
  vtkSetMacro(EnableColoring, bool);
  vtkGetMacro(EnableColoring, bool);
 
 
  vtkSetMacro(EnableAlpha, bool);
  vtkGetMacro(EnableAlpha, bool);
 
 
  void SetController(vtkMultiProcessController*);
  vtkGetObjectMacro(Controller, vtkMultiProcessController);
 
protected:
  vtkOpenVDBWriter();
  ~vtkOpenVDBWriter() override;
 
  void WriteData() override;
 
  void WriteImageData(vtkImageData* imageData);
  void WritePointSet(vtkPointSet* pointSet);
 
  // see algorithm for more info.
  // This writer takes in vtkTable, vtkDataSet or vtkCompositeDataSet.
  int FillInputPortInformation(int port, vtkInformation* info) override;
 
  // see algorithm for more info. needed here so we can request pieces.
  int ProcessRequest(vtkInformation* request, vtkInformationVector** inputVector,
    vtkInformationVector* outputVector) override;
 
  void SetRGBA(vtkIdType num, vtkUnsignedCharArray* rgba, vtkDataSetAttributes* attributes);
 
  char* FileName;
 
 
  bool WriteAllTimeSteps;
 
 
  vtkScalarsToColors* LookupTable;
  bool EnableColoring;
  bool EnableAlpha;
 
private:
  vtkOpenVDBWriter(const vtkOpenVDBWriter&) = delete;
  void operator=(const vtkOpenVDBWriter&) = delete;
 
 
  vtkMultiProcessController* Controller;
 
  vtkOpenVDBWriterInternals* Internals;
  friend class vtkOpenVDBWriterInternals;
};
 
VTK_ABI_NAMESPACE_END
#endif