mitkImageWriter.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_IMAGE_WRITER__H_
#define _MITK_IMAGE_WRITER__H_

#include <MitkLegacyIOExports.h>
#include <mitkFileWriterWithInformation.h>

namespace mitk
{
  class Image;
  class MITKLEGACYIO_EXPORT ImageWriter : public mitk::FileWriterWithInformation
  {
  public:
    mitkClassMacro(ImageWriter, mitk::FileWriter);

    itkFactorylessNewMacro(Self) itkCloneMacro(Self)

      mitkWriterMacro;

    virtual void SetFileName(const char *fileName) override;
    virtual void SetFileName(const std::string &fileName);

    itkGetStringMacro(FileName);

    virtual void SetExtension(const char *extension);
    virtual void SetExtension(const std::string &extension);

    itkGetStringMacro(Extension);

    void SetDefaultExtension();

    itkSetStringMacro(FilePrefix);

    itkGetStringMacro(FilePrefix);

    itkSetStringMacro(FilePattern);

    itkGetStringMacro(FilePattern);

    void SetInput(mitk::Image *input);

    //##Documentation
    //## @brief Return the possible file extensions for the data type associated with the writer
    virtual std::vector<std::string> GetPossibleFileExtensions() override;

    virtual std::string GetSupportedBaseData() const override;

    virtual std::string GetFileExtension() override;

    virtual bool CanWriteDataType(DataNode *) override;

    virtual std::string GetWritenMIMEType() override;

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

    const mitk::Image *GetInput();

    // FileWriterWithInformation methods
    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;

    void SetUseCompression(bool useCompression);

  protected:
    ImageWriter();

    virtual ~ImageWriter();

    virtual void GenerateData() override;

    virtual void WriteByITK(mitk::Image *image, const std::string &fileName);

    std::string m_FileName;

    std::string m_FileNameWithoutExtension;

    std::string m_FilePrefix;

    std::string m_FilePattern;

    std::string m_Extension;

    std::string m_MimeType;

    bool m_UseCompression;
  };
}

#endif //_MITK_IMAGE_WRITER__H_