mitkImageMaskGenerator.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 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::ConstPointer GetMask() override;

    void SetTimeStep(unsigned int timeStep) override;

    void SetImageMask(const mitk::Image* maskImage);

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

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

    mitk::Image::ConstPointer m_InternalMaskImage;
    mitk::Image::ConstPointer m_InternalMask;
    unsigned long m_InternalMaskUpdateTime;

};


}

#endif