vtkCellGridWriter.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 vtkCellGridWriter_h
#define vtkCellGridWriter_h

#include "vtkIOCellGridModule.h" // For export macro.
#include "vtkWriter.h"

// clang-format off
#include <vtk_nlohmannjson.h>        // For API.
#include VTK_NLOHMANN_JSON(json.hpp) // For API.
// clang-format on

VTK_ABI_NAMESPACE_BEGIN

class vtkCellGrid;

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

  vtkSetStringMacro(FileName);
  vtkGetStringMacro(FileName);

  vtkCellGrid* GetInput();
  vtkCellGrid* GetInput(int port);

  bool ToJSON(nlohmann::json& data, vtkCellGrid* grid);

  enum Format : int
  {
    PlainText,      
    MessagePack,    
    NumberOfFormats 
  };

  vtkSetEnumMacro(FileFormat, Format);
  vtkGetEnumMacro(FileFormat, Format);

protected:
  vtkCellGridWriter();
  ~vtkCellGridWriter() override;

  int FillInputPortInformation(int port, vtkInformation* info) override;
  bool WriteDataAndReturn() override;

  char* FileName{ nullptr };
  Format FileFormat{ Format::PlainText };

private:
  vtkCellGridWriter(const vtkCellGridWriter&) = delete;
  void operator=(const vtkCellGridWriter&) = delete;
};

VTK_ABI_NAMESPACE_END

#endif // vtkCellGridWriter_h