animaSmoothingRecursiveYvvGaussianImageFilter.h

Go to the documentation of this file.

#pragma once

#include <animaRecursiveLineYvvGaussianImageFilter.h>
#include <itkCastImageFilter.h>
#include <itkImage.h>
#include <itkPixelTraits.h>
#include <itkCommand.h>
#include <itkFixedArray.h>

namespace anima
{

template <typename TInputImage,
typename TOutputImage= TInputImage >
class SmoothingRecursiveYvvGaussianImageFilter:
public itk::InPlaceImageFilter<TInputImage,TOutputImage>
{
public:
    typedef SmoothingRecursiveYvvGaussianImageFilter          Self;
    typedef itk::InPlaceImageFilter<TInputImage,TOutputImage>      Superclass;
    typedef itk::SmartPointer<Self>                                Pointer;
    typedef itk::SmartPointer<const Self>                          ConstPointer;

    typedef TInputImage                                       InputImageType;
    typedef TOutputImage                                      OutputImageType;
    typedef typename TInputImage::PixelType                   PixelType;
    typedef typename itk::NumericTraits<PixelType>::RealType       RealType;
    typedef typename itk::NumericTraits<PixelType>::ScalarRealType ScalarRealType;

    itkTypeMacro(SmoothingRecursiveYvvGaussianImageFilter,
                 InPlaceImageFilter)

    
    itkStaticConstMacro(ImageDimension, unsigned int,
                        TInputImage::ImageDimension);

    typedef itk::FixedArray< ScalarRealType,
    itkGetStaticConstMacro(ImageDimension) > SigmaArrayType;

    typedef typename itk::NumericTraits< PixelType >::RealType   InternalRealType;
    typedef typename InputImageType::template Rebind<InternalRealType>::Type RealImageType;

    typedef anima::RecursiveLineYvvGaussianImageFilter<
    InputImageType,
    RealImageType
    >    FirstGaussianFilterType;

    typedef anima::RecursiveLineYvvGaussianImageFilter<
    RealImageType,
    RealImageType
    >    InternalGaussianFilterType;

    typedef itk::CastImageFilter<
    RealImageType,
    OutputImageType
    >    CastingFilterType;

    typedef typename InternalGaussianFilterType::Pointer    InternalGaussianFilterPointer;

    typedef typename FirstGaussianFilterType::Pointer       FirstGaussianFilterPointer;

    typedef typename CastingFilterType::Pointer             CastingFilterPointer;

    typedef typename OutputImageType::Pointer                  OutputImagePointer;

    itkNewMacro(Self)

    
    void SetSigmaArray(const SigmaArrayType & sigmas);
    void SetSigma( ScalarRealType sigma );

    SigmaArrayType GetSigmaArray() const;
    ScalarRealType GetSigma() const;

    void SetNormalizeAcrossScale(bool normalizeInScaleSpace);
    itkGetConstMacro(NormalizeAcrossScale, bool)

    void SetNumberOfWorkUnits(itk::ThreadIdType nb) ITK_OVERRIDE;

    // See super class for doxygen documentation
    //
    virtual bool CanRunInPlace() const ITK_OVERRIDE;

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

    void GenerateData() ITK_OVERRIDE;

    virtual void GenerateInputRequestedRegion() ITK_OVERRIDE;

    // Override since the filter produces the entire dataset
    void EnlargeOutputRequestedRegion(itk::DataObject *output) ITK_OVERRIDE;

private:
    ITK_DISALLOW_COPY_AND_ASSIGN(SmoothingRecursiveYvvGaussianImageFilter);

    InternalGaussianFilterPointer         m_SmoothingFilters[ImageDimension - 1];
    FirstGaussianFilterPointer            m_FirstSmoothingFilter;
    CastingFilterPointer                  m_CastingFilter;

    bool m_NormalizeAcrossScale;

    SigmaArrayType                        m_Sigma;
};

} // end of namespace anima

#include "animaSmoothingRecursiveYvvGaussianImageFilter.hxx"