animaMaskedImageToImageFilter.h

Go to the documentation of this file.

#pragma once

#include <iostream>
#include <animaNumberedThreadImageToImageFilter.h>
#include <itkNumericTraits.h>

namespace anima
{

template <typename TInputImage, typename TOutputImage>
class MaskedImageToImageFilter :
        public anima::NumberedThreadImageToImageFilter <TInputImage, TOutputImage>
{
public:
    typedef MaskedImageToImageFilter Self;
    typedef anima::NumberedThreadImageToImageFilter <TInputImage, TOutputImage> Superclass;
    typedef itk::SmartPointer<Self> Pointer;
    typedef itk::SmartPointer<const Self>  ConstPointer;

    itkNewMacro(Self)

    
    itkTypeMacro(MaskedImageToImageFilter, anima::NumberedThreadImageToImageFilter)

    
    typedef typename Superclass::InputImageRegionType InputImageRegionType;
    typedef typename Superclass::OutputImageRegionType OutputImageRegionType;
    typedef typename itk::ImageSource<TOutputImage>::ThreadStruct ThreadStruct;

    typedef itk::Image <unsigned char,TInputImage::ImageDimension> MaskImageType;
    typedef typename MaskImageType::RegionType MaskRegionType;
    typedef typename MaskImageType::Pointer MaskImagePointer;
    typedef typename MaskImageType::IndexType MaskIndexType;
    typedef typename MaskImageType::SizeType MaskSizeType;

    itkSetMacro(ComputationMask, MaskImagePointer)
    itkGetMacro(ComputationMask, MaskImageType *)

protected:
    MaskedImageToImageFilter()
    {
        m_ComputationMask = ITK_NULLPTR;
    }

    virtual ~MaskedImageToImageFilter() {}

    virtual void CheckComputationMask();
    void InitializeComputationRegionFromMask();
    virtual void BeforeThreadedGenerateData() ITK_OVERRIDE;

private:
    ITK_DISALLOW_COPY_AND_ASSIGN(MaskedImageToImageFilter);

    MaskImagePointer m_ComputationMask;
};

} //end namespace anima

#include "animaMaskedImageToImageFilter.hxx"