mitkNrrdTbssRoiImageWriter.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_NRRDTBSSROIVOL_WRITER__H_
#define _MITK_NRRDTBSSROIVOL_WRITER__H_

#include <itkProcessObject.h>
#include <mitkFileWriterWithInformation.h>
#include <mitkTbssRoiImage.h>
#include "MitkQuantificationExports.h"

namespace mitk
{

class MITKQUANTIFICATION_EXPORT NrrdTbssRoiImageWriter : public mitk::FileWriterWithInformation
{
public:

    typedef itk::Image<unsigned char,3>     ImageType;

    mitkClassMacro( NrrdTbssRoiImageWriter, mitk::FileWriterWithInformation )

    mitkWriterMacro

    itkFactorylessNewMacro(Self)
    itkCloneMacro(Self)

    typedef mitk::TbssRoiImage InputType;

    itkSetStringMacro( FileName );

    itkGetStringMacro( FileName );

    itkSetStringMacro( FilePrefix );

    itkGetStringMacro( FilePrefix );

    itkSetStringMacro( FilePattern );

    itkGetStringMacro( FilePattern );

    using ProcessObject::SetInput;
    void SetInput( InputType* input );

    InputType* GetInput();

    itkGetMacro( Success, bool );

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

    std::string GetSupportedBaseData() const override;

    // FileWriterWithInformation methods
    virtual const char * GetDefaultFilename() override { return "TbssRoiImage.roi"; }
    virtual const char * GetFileDialogPattern() override { return "Tbss Roi Images (*.roi)"; }
    virtual const char * GetDefaultExtension() override { return ".roi"; }
    virtual bool CanWriteBaseDataType(BaseData::Pointer data) override
    {
      return (dynamic_cast<mitk::TbssRoiImage*>(data.GetPointer()) != nullptr);
    }


    virtual void DoWrite(BaseData::Pointer data) override {
      if (CanWriteBaseDataType(data)) {
        this->SetInput(dynamic_cast<mitk::TbssRoiImage*>(data.GetPointer()));
        this->Update();
      }
    }

protected:

    NrrdTbssRoiImageWriter();

    virtual ~NrrdTbssRoiImageWriter();

    virtual void GenerateData() override;

    std::string m_FileName;

    std::string m_FilePrefix;

    std::string m_FilePattern;

    bool m_Success;

};


} // end of namespace mitk

#endif