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 mitkSurfaceVtkWriter_h
#define mitkSurfaceVtkWriter_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