animaMeanAndVarianceImagesFilter.h

Go to the documentation of this file.

#pragma once

// First make sure that the configuration is available.
// This line can be removed once the optimized versions
// gets integrated into the main directories.
#include <itkConfigure.h>

#include <itkImageToImageFilter.h>
#include <itkImage.h>
#include <itkNumericTraits.h>

namespace anima
{
template <class TInputImage, class TOutputImage>
class MeanAndVarianceImagesFilter :
    public itk::ImageToImageFilter< TInputImage, TOutputImage >
{
public:
    itkStaticConstMacro(InputImageDimension, unsigned int,
                        TInputImage::ImageDimension);
    itkStaticConstMacro(OutputImageDimension, unsigned int,
                        TOutputImage::ImageDimension);

    typedef TInputImage  InputImageType;
    typedef TOutputImage OutputImageType;

    typedef MeanAndVarianceImagesFilter Self;
    typedef itk::ImageToImageFilter< InputImageType, OutputImageType> Superclass;
    typedef itk::SmartPointer<Self> Pointer;
    typedef itk::SmartPointer<const Self> ConstPointer;

    itkNewMacro(Self)

    
    itkTypeMacro(MeanAndVarianceImagesFilter, ImageToImageFilter)

    
    typedef typename InputImageType::PixelType               InputPixelType;
    typedef typename OutputImageType::PixelType              OutputPixelType;
    typedef typename itk::NumericTraits<InputPixelType>::RealType InputRealType;

    typedef typename InputImageType::RegionType  InputImageRegionType;
    typedef typename OutputImageType::RegionType OutputImageRegionType;
    typedef typename InputImageType::SizeType    InputSizeType;

    itkSetMacro(Radius, InputSizeType)

    itkGetConstReferenceMacro(Radius, InputSizeType)

    typename OutputImageType::Pointer GetMeanImage() {return this->GetOutput(0);}
    typename OutputImageType::Pointer GetVarImage() {return this->GetOutput(1);}

#ifdef ITK_USE_CONCEPT_CHECKING

    itkConceptMacro(InputHasNumericTraitsCheck,
                    (itk::Concept::HasNumericTraits<InputPixelType>));
#endif

protected:
    MeanAndVarianceImagesFilter();
    virtual ~MeanAndVarianceImagesFilter() {}
    void PrintSelf(std::ostream& os, itk::Indent indent) const ITK_OVERRIDE;

    void DynamicThreadedGenerateData(const OutputImageRegionType& outputRegionForThread) ITK_OVERRIDE;

private:
    ITK_DISALLOW_COPY_AND_ASSIGN(MeanAndVarianceImagesFilter);

    InputSizeType m_Radius;
};

} // end of namespace anima

#include "animaMeanAndVarianceImagesFilter.hxx"