mitkSurfaceVtkWriter.h

Go to the documentation of this file.

/*============================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.

Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.

============================================================================*/

#ifndef _MITK_SURFACE_VTK_WRITER__H_
#define _MITK_SURFACE_VTK_WRITER__H_

#include <iomanip>

#include <vtkPolyDataWriter.h>
#include <vtkSTLWriter.h>
#include <vtkXMLPolyDataWriter.h>

#include <MitkLegacyIOExports.h>
#include <itkProcessObject.h>
#include <mitkFileWriterWithInformation.h>
#include <mitkPointSet.h>
#include <mitkSurface.h>

#include <vtkSmartPointer.h>

class vtkTransformPolyDataFilter;

namespace mitk
{
  template <class VTKWRITER>
  class MITKLEGACYIO_EXPORT SurfaceVtkWriter : public mitk::FileWriterWithInformation
  {
  public:
    mitkClassMacro(SurfaceVtkWriter, mitk::FileWriter);

    itkFactorylessNewMacro(Self);

    itkCloneMacro(Self);

      mitkWriterMacro;

    typedef VTKWRITER VtkWriterType;

    itkSetStringMacro(FileName);

    itkGetStringMacro(FileName);

    itkSetStringMacro(Extension);

    itkGetStringMacro(Extension);

    void SetDefaultExtension();

    itkSetStringMacro(FilePrefix);

    itkGetStringMacro(FilePrefix);

    itkSetStringMacro(FilePattern);

    itkGetStringMacro(FilePattern);

    void SetInput(mitk::Surface *input);

    const mitk::Surface *GetInput();

    std::string GetFileExtension() override;

    bool CanWriteDataType(DataNode *) override;

    std::string GetWritenMIMEType() override;

    using Superclass::SetInput;
    virtual void SetInput(DataNode *);

    VtkWriterType *GetVtkWriter() { return m_VtkWriter; }
    std::vector<std::string> GetPossibleFileExtensions() override;

    std::string GetSupportedBaseData() const override { return Surface::GetStaticNameOfClass(); }
    const char *GetDefaultFilename() override;
    const char *GetFileDialogPattern() override;
    const char *GetDefaultExtension() override;
    bool CanWriteBaseDataType(BaseData::Pointer data) override;
    void DoWrite(BaseData::Pointer data) override;

  protected:
    SurfaceVtkWriter();

    ~SurfaceVtkWriter() override;

    void GenerateData() override;

    void ExecuteWrite(VtkWriterType *vtkWriter);

    std::string m_FileName;

    std::string m_FilePrefix;

    std::string m_FilePattern;

    std::string m_Extension;

    std::string m_MimeType;

    vtkSmartPointer<VtkWriterType> m_VtkWriter;

    bool m_WriterWriteHasReturnValue;
  };

#ifndef MitkLegacyIO_EXPORTS
  extern template class SurfaceVtkWriter<vtkSTLWriter>;
  extern template class SurfaceVtkWriter<vtkPolyDataWriter>;
  extern template class SurfaceVtkWriter<vtkXMLPolyDataWriter>;
#endif
}

#endif //_MITK_SURFACE_VTK_WRITER__H_