mitkMaskGenerator.h

Go to the documentation of this file.

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

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.

Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.

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

#ifndef MITKMASKGENERATOR
#define MITKMASKGENERATOR

#include <MitkImageStatisticsExports.h>
#include <mitkImage.h>
#include <itkRegion.h>
#include <itkObject.h>
#include <itkSmartPointer.h>

namespace mitk
{
class MITKIMAGESTATISTICS_EXPORT MaskGenerator: public itk::Object
{
public:
    typedef MaskGenerator                       Self;
    typedef itk::Object                         Superclass;
    typedef itk::SmartPointer< Self >           Pointer;
    typedef itk::SmartPointer< const Self >     ConstPointer;

    itkNewMacro(Self); 
    itkTypeMacro(MaskGenerator, itk::Object);

    //~MaskGenerator();

    virtual mitk::Image::Pointer GetMask();

    virtual mitk::Image::ConstPointer GetReferenceImage();

    void SetInputImage(mitk::Image::ConstPointer inputImg);

    virtual void SetTimeStep(unsigned int timeStep);

protected:
    MaskGenerator();

    unsigned int m_TimeStep;
    mitk::Image::Pointer m_InternalMask;
    mitk::Image::ConstPointer m_inputImage;

private:

};
}

#endif // MITKMASKGENERATOR