mitkImageMaskGenerator.h

Go to the documentation of this file.

#ifndef mitkBinaryMaskGenerator
#define mitkBinaryMaskGenerator

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

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

    itkNewMacro(Self)

    
    itkTypeMacro(BinaryImageMaskGenerator, MaskGenerator)

    mitk::Image::Pointer GetMask();

    void SetTimeStep(unsigned int timeStep);

    void SetImageMask(mitk::Image::Pointer maskImage);

protected:
    ImageMaskGenerator():Superclass(){
        m_InternalMaskUpdateTime = 0;
        m_InternalMask = mitk::Image::New();
    }

private:
    bool IsUpdateRequired() const;
    void UpdateInternalMask();

    mitk::Image::Pointer m_internalMaskImage;
    unsigned long m_InternalMaskUpdateTime;

};


}

#endif