mitkSurfaceVtkWriter.h

Go to the documentation of this file.

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

Program:   Medical Imaging & Interaction Toolkit
Language:  C++
Date:      $Date$
Version:   $Revision$

Copyright (c) German Cancer Research Center, Division of Medical and
Biological Informatics. All rights reserved.
See MITKCopyright.txt or https://www.mitk.org/copyright.html for details.

This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE.  See the above copyright notices for more information.

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


#ifndef _MITK_SURFACE_VTK_WRITER__H_
#define _MITK_SURFACE_VTK_WRITER__H_

#include <iomanip>

#include <itkProcessObject.h>
#include <mitkFileWriter.h>
#include <mitkPointSet.h>
#include <mitkSurface.h>

#include <vtkSmartPointer.h>

class vtkTransformPolyDataFilter;

namespace mitk
{

template <class VTKWRITER>
class SurfaceVtkWriter : public mitk::FileWriter
{
public:

    mitkClassMacro( SurfaceVtkWriter, mitk::FileWriter );

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

    virtual bool CanWriteDataType( DataNode* );

    virtual std::string GetWritenMIMEType();

    virtual void SetInput( DataNode* );

    VtkWriterType* GetVtkWriter()
    {
      return m_VtkWriter;
    }

    virtual std::vector<std::string> GetPossibleFileExtensions();

protected:

    SurfaceVtkWriter();

    virtual ~SurfaceVtkWriter();

    virtual void GenerateData();
      
    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;
};

}

#endif //_MITK_SURFACE_VTK_WRITER__H_