mitkNrrdTbssImageWriter.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_NRRDTBSSVOL_WRITER__H_
#define _MITK_NRRDTBSSVOL_WRITER__H_

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

namespace mitk
{

class MITKQUANTIFICATION_EXPORT NrrdTbssImageWriter : public mitk::FileWriterWithInformation
{
public:

    mitkClassMacro( NrrdTbssImageWriter, mitk::FileWriterWithInformation )

    mitkWriterMacro

    itkFactorylessNewMacro(Self)
    itkCloneMacro(Self)

    typedef mitk::TbssImage 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 "TbssImage.tbss"; }
    virtual const char * GetFileDialogPattern() override { return "Tbss Images (*.tbss)"; }
    virtual const char * GetDefaultExtension() override { return ".tbss"; }
    virtual bool CanWriteBaseDataType(BaseData::Pointer data) override
    {
      return (dynamic_cast<mitk::TbssImage*>(data.GetPointer()) != nullptr);
    }


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

protected:

    NrrdTbssImageWriter();

    virtual ~NrrdTbssImageWriter();

    virtual void GenerateData() override;

    std::string m_FileName;

    std::string m_FilePrefix;

    std::string m_FilePattern;

    bool m_Success;




};


} // end of namespace mitk



#endif