vtkSTLWriter.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 vtkSTLWriter_h
#define vtkSTLWriter_h
 
#include "vtkIOGeometryModule.h" // For export macro
#include "vtkWriter.h"
 
VTK_ABI_NAMESPACE_BEGIN
class vtkCellArray;
class vtkPoints;
class vtkPolyData;
class vtkUnsignedCharArray;
 
class VTKIOGEOMETRY_EXPORT vtkSTLWriter : public vtkWriter
{
public:
  static vtkSTLWriter* New();
  vtkTypeMacro(vtkSTLWriter, vtkWriter);
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
 
  vtkPolyData* GetInput();
  vtkPolyData* GetInput(int port);
 
 
  vtkSetFilePathMacro(FileName);
  vtkGetFilePathMacro(FileName);
 
 
  vtkSetStringMacro(Header);
  vtkGetStringMacro(Header);
 
 
  virtual void SetBinaryHeader(vtkUnsignedCharArray* binaryHeader);
  vtkGetObjectMacro(BinaryHeader, vtkUnsignedCharArray);
 
 
  vtkSetClampMacro(FileType, int, VTK_ASCII, VTK_BINARY);
  vtkGetMacro(FileType, int);
  void SetFileTypeToASCII() { this->SetFileType(VTK_ASCII); }
  void SetFileTypeToBinary() { this->SetFileType(VTK_BINARY); }
 
protected:
  vtkSTLWriter();
  ~vtkSTLWriter() override;
 
  void WriteData() override;
 
  void WriteBinarySTL(vtkPoints* pts, vtkCellArray* polys, vtkCellArray* strips);
  void WriteAsciiSTL(vtkPoints* pts, vtkCellArray* polys, vtkCellArray* strips);
 
  char* FileName;
  char* Header;
  vtkUnsignedCharArray* BinaryHeader;
  int FileType;
 
  int FillInputPortInformation(int port, vtkInformation* info) override;
 
private:
  vtkSTLWriter(const vtkSTLWriter&) = delete;
  void operator=(const vtkSTLWriter&) = delete;
};
 
VTK_ABI_NAMESPACE_END
#endif