mitkNrrdDiffusionImageWriter.h

Go to the documentation of this file.

/*=========================================================================
 
Program:   Medical Imaging & Interaction Toolkit
Language:  C++
Date:      $Date: 2008-08-27 17:18:46 +0200 (Mi, 27 Aug 2008) $
Version:   $Revision: 15096 $
 
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_NRRDDIFFVOL_WRITER__H_
#define _MITK_NRRDDIFFVOL_WRITER__H_

#include <itkProcessObject.h>
#include <mitkFileWriterWithInformation.h>
#include <mitkDiffusionImage.h>


namespace mitk
{

template < class TPixelType >
class NrrdDiffusionImageWriter : public mitk::FileWriterWithInformation
{
public:

    mitkClassMacro( NrrdDiffusionImageWriter, mitk::FileWriterWithInformation );

    mitkWriterMacro;

    itkNewMacro( Self );
    
    typedef mitk::DiffusionImage<TPixelType> InputType;
    
    itkSetStringMacro( FileName );

    itkGetStringMacro( FileName );

    itkSetStringMacro( FilePrefix );

    itkGetStringMacro( FilePrefix );

    itkSetStringMacro( FilePattern );

    itkGetStringMacro( FilePattern );

    void SetInput( InputType* input );

    InputType* GetInput();

    itkGetMacro( Success, bool );
   
    virtual std::vector<std::string> GetPossibleFileExtensions();

    // FileWriterWithInformation methods
    virtual const char * GetDefaultFilename() { return "DiffusionWeightedImages.dwi"; }
    virtual const char * GetFileDialogPattern() { return "Diffusion Weighted Images (*.dwi *.hdwi)"; }
    virtual const char * GetDefaultExtension() { return ".dwi"; }
    virtual bool CanWriteDataType(BaseData::Pointer data) { return (dynamic_cast<mitk::DiffusionImage<TPixelType>*>(data.GetPointer()) != NULL); };  
    virtual void DoWrite(BaseData::Pointer data) { 
      if (CanWriteDataType(data)) {
        this->SetInput(dynamic_cast<mitk::DiffusionImage<TPixelType>*>(data.GetPointer())); 
        this->Update(); 
      }
    };

protected:
        
    NrrdDiffusionImageWriter();

    virtual ~NrrdDiffusionImageWriter();

    virtual void GenerateData();
    
    std::string m_FileName;

    std::string m_FilePrefix;

    std::string m_FilePattern;
    
    bool m_Success;
            
};    
       

} // end of namespace mitk

#include "mitkNrrdDiffusionImageWriter.cpp"

#endif