mitkUnstructuredGridVtkWriter.h

Go to the documentation of this file.

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

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

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_UNSTRUCTURED_GRID_VTK_WRITER__H_
#define _MITK_UNSTRUCTURED_GRID_VTK_WRITER__H_

#include <iomanip>

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

class vtkTransformFilter;

namespace mitk
{

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

    mitkClassMacro( UnstructuredGridVtkWriter, 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::UnstructuredGrid* input );

    const mitk::UnstructuredGrid* GetInput();

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

    virtual std::string GetFileExtension();

    virtual bool CanWriteDataType( DataNode* );

    virtual std::string GetWritenMIMEType();

    virtual void SetInput( DataNode* );

    VtkWriterType* GetVtkWriter()
    {
      return m_VtkWriter;
    }

protected:

    UnstructuredGridVtkWriter();

    virtual ~UnstructuredGridVtkWriter();

    virtual void GenerateData();
      
    void ExecuteWrite( VtkWriterType* m_VtkWriter, vtkTransformFilter* transformPointSet );

    std::string m_FileName;

    std::string m_FilePrefix;

    std::string m_FilePattern;

    std::string m_Extension;

    std::string m_MimeType;

    VtkWriterType* m_VtkWriter;
  
    bool m_WriterWriteHasReturnValue;
};

}

#endif // _MITK_UNSTRUCTURED_GRID_VTK_WRITER__H_