vtkMNIObjectWriter.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-FileCopyrightText: Copyright (c) 2006 Atamai, Inc.
// SPDX-License-Identifier: BSD-3-Clause
#ifndef vtkMNIObjectWriter_h
#define vtkMNIObjectWriter_h

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

VTK_ABI_NAMESPACE_BEGIN
class vtkMapper;
class vtkProperty;
class vtkLookupTable;
class vtkPolyData;
class vtkFloatArray;
class vtkIntArray;
class vtkPoints;
struct vtkMNIObjectWriterFunctor;

class VTKIOMINC_EXPORT vtkMNIObjectWriter : public vtkWriter
{
public:
  vtkTypeMacro(vtkMNIObjectWriter, vtkWriter);

  static vtkMNIObjectWriter* New();
  void PrintSelf(ostream& os, vtkIndent indent) override;

  virtual const char* GetFileExtensions() { return ".obj"; }

  virtual const char* GetDescriptiveName() { return "MNI object"; }


  virtual void SetProperty(vtkProperty* property);
  virtual vtkProperty* GetProperty() { return this->Property; }


  virtual void SetMapper(vtkMapper* mapper);
  virtual vtkMapper* GetMapper() { return this->Mapper; }


  virtual void SetLookupTable(vtkLookupTable* table);
  virtual vtkLookupTable* GetLookupTable() { return this->LookupTable; }


  vtkPolyData* GetInput();
  vtkPolyData* GetInput(int port);


  vtkSetFilePathMacro(FileName);
  vtkGetFilePathMacro(FileName);


  vtkSetClampMacro(FileType, int, VTK_ASCII, VTK_BINARY);
  vtkGetMacro(FileType, int);
  void SetFileTypeToASCII() { this->SetFileType(VTK_ASCII); }
  void SetFileTypeToBinary() { this->SetFileType(VTK_BINARY); }

protected:
  vtkMNIObjectWriter();
  ~vtkMNIObjectWriter() override;

  vtkProperty* Property;
  vtkMapper* Mapper;
  vtkLookupTable* LookupTable;

  ostream* OutputStream;

  int WriteObjectType(int objType);
  int WriteValues(vtkDataArray* array);
  int WriteIdValue(vtkIdType value);
  int WriteNewline();

  int WriteProperty(vtkProperty* property);
  int WriteLineThickness(vtkProperty* property);
  int WritePoints(vtkPolyData* polyData);
  int WriteNormals(vtkPolyData* polyData);
  int WriteColors(vtkProperty* property, vtkMapper* mapper, vtkPolyData* data);
  int WriteCells(vtkPolyData* data, int cellType);

  int WritePolygonObject(vtkPolyData* output);
  int WriteLineObject(vtkPolyData* output);

  bool WriteDataAndReturn() override;

  char* FileName;

  int FileType;

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

  ostream* OpenFile();
  void CloseFile(ostream* fp);

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

  friend struct vtkMNIObjectWriterFunctor;
};

VTK_ABI_NAMESPACE_END
#endif