vtkOBJWriter.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 vtkOBJWriter_h
#define vtkOBJWriter_h

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

VTK_ABI_NAMESPACE_BEGIN
class vtkDataSet;
class vtkDataSetAttributes;
class vtkImageData;
class vtkPolyData;
class vtkUnsignedCharArray;

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


  vtkPolyData* GetInputGeometry();
  vtkImageData* GetInputTexture();
  vtkDataSet* GetInput(int port);


  vtkSetFilePathMacro(TextureFileName);
  vtkGetFilePathMacro(TextureFileName);


  vtkSetFilePathMacro(FileName);
  vtkGetFilePathMacro(FileName);


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


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


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

protected:
  vtkOBJWriter();
  ~vtkOBJWriter() override;

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

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

  char* FileName{ nullptr };
  char* TextureFileName{ nullptr };

  std::string ColorArrayName;
  bool WriteColorArray{ false };

  bool UseRelativeTexturePath{ false };
};

VTK_ABI_NAMESPACE_END
#endif