vtkNanoVDBWriter.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 vtkNanoVDBWriter_h
#define vtkNanoVDBWriter_h

#include "vtkIONanoVDBModule.h" // needed for exports
#include "vtkWriter.h"

VTK_ABI_NAMESPACE_BEGIN
class vtkImageData;
class vtkInformation;
class vtkMultiProcessController;

class VTKIONANOVDB_EXPORT vtkNanoVDBWriter : public vtkWriter
{
public:
  static vtkNanoVDBWriter* New();
  vtkTypeMacro(vtkNanoVDBWriter, vtkWriter);
  void PrintSelf(ostream& os, vtkIndent indent) override;


  vtkSetFilePathMacro(FileName);
  vtkGetFilePathMacro(FileName);


  vtkSetMacro(WriteAllTimeSteps, bool);
  vtkGetMacro(WriteAllTimeSteps, bool);


  void SetController(vtkMultiProcessController*);
  vtkGetObjectMacro(Controller, vtkMultiProcessController);

protected:
  vtkNanoVDBWriter();
  ~vtkNanoVDBWriter() override;

  bool WriteDataAndReturn() override;

  // This writer only accepts vtkImageData input.
  int FillInputPortInformation(int port, vtkInformation* info) override;

  // Needed so we can request pieces and time steps.
  int ProcessRequest(vtkInformation* request, vtkInformationVector** inputVector,
    vtkInformationVector* outputVector) override;

private:
  vtkNanoVDBWriter(const vtkNanoVDBWriter&) = delete;
  void operator=(const vtkNanoVDBWriter&) = delete;

  bool WriteImageData(vtkImageData* imageData);

  char* FileName = nullptr;
  bool WriteAllTimeSteps = false;

  // The controller for the writer to work in parallel.
  vtkMultiProcessController* Controller = nullptr;

  int CurrentTimeIndex = 0;
  int NumberOfTimeSteps = 1;
};

VTK_ABI_NAMESPACE_END
#endif