mitkSurfaceVtkWriter.h

Go to the documentation of this file.

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

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

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

#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();

    virtual std::string GetFileExtension() override;

    virtual bool CanWriteDataType(DataNode *) override;

    virtual std::string GetWritenMIMEType() override;

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

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

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

  protected:
    SurfaceVtkWriter();

    virtual ~SurfaceVtkWriter();

    virtual 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_