animaPyramidImageFilter.h

Go to the documentation of this file.

#pragma once

#include <itkImageToImageFilter.h>
#include <itkVectorImage.h>

namespace anima
{

template <class TInputImage, class TOutputImage>
class PyramidImageFilter :
public itk::ImageToImageFilter< TInputImage, TOutputImage >
{
public:
    typedef PyramidImageFilter Self;
    typedef itk::ImageToImageFilter<TInputImage,TOutputImage> Superclass;
    typedef itk::SmartPointer<Self> Pointer;
    typedef itk::SmartPointer<const Self> ConstPointer;

    itkNewMacro(Self)

    
    itkTypeMacro(PyramidImageFilter, ImageToImageFilter)

    typedef TInputImage InputImageType;
    typedef typename InputImageType::IOPixelType InputInternalScalarType;
    typedef typename InputImageType::RegionType RegionType;
    typedef typename InputImageType::SpacingType SpacingType;

    typedef TOutputImage OutputImageType;
    typedef typename OutputImageType::Pointer OutputImagePointer;
    typedef typename OutputImageType::IOPixelType OutputInternalScalarType;

    typedef itk::Image <typename TInputImage::IOPixelType,TInputImage::ImageDimension> ScalarInputImageType;
    typedef itk::VectorImage <typename TInputImage::IOPixelType,TInputImage::ImageDimension> VectorInputImageType;

    typedef itk::Image <typename TInputImage::IOPixelType,TInputImage::ImageDimension> ScalarOutputImageType;
    typedef typename ScalarOutputImageType::Pointer ScalarOutputImagePointer;

    typedef itk::VectorImage <typename TInputImage::IOPixelType,TInputImage::ImageDimension> VectorOutputImageType;
    typedef typename VectorOutputImageType::Pointer VectorOutputImagePointer;

    typedef itk::ImageToImageFilter <TInputImage,TOutputImage> BaseResamplerType;
    typedef typename BaseResamplerType::Pointer BaseResamplerPointer;

    itkSetMacro(NumberOfLevels, unsigned int)
    itkGetConstMacro(NumberOfLevels, unsigned int)

    itkSetObjectMacro(ImageResampler, BaseResamplerType)

protected:
    PyramidImageFilter();
    virtual ~PyramidImageFilter() {}

    void GenerateData() ITK_OVERRIDE;

    void CheckNumberOfLevels();
    double AnisotropyMeasure(SpacingType &sp, std::vector <bool> &changeableSizes);

    void CreateLevelVectorImage(unsigned int level);
    void CreateLevelImage(unsigned int level);

private:
    ITK_DISALLOW_COPY_AND_ASSIGN(PyramidImageFilter);

    unsigned int m_NumberOfLevels;

    BaseResamplerPointer m_ImageResampler;

    // Internal variables to compute images
    std::vector <RegionType> m_LevelRegions;
    std::vector <SpacingType> m_LevelSpacings;
};

} //end of namespace anima

#include "animaPyramidImageFilter.hxx"