animaGeneralizedFAImageFilter.h

Go to the documentation of this file.

#pragma once

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

namespace anima
{

template <typename TInputPixelType>
class GeneralizedFAImageFilter :
public itk::ImageToImageFilter< itk::VectorImage<TInputPixelType, 3> , itk::Image <TInputPixelType, 3> >
{
public:
    typedef GeneralizedFAImageFilter Self;
    typedef itk::VectorImage <TInputPixelType, 3> TInputImage;
    typedef itk::Image <TInputPixelType, 3> TOutputImage;
    typedef itk::ImageToImageFilter< TInputImage, TOutputImage > Superclass;
    typedef itk::SmartPointer<Self> Pointer;
    typedef itk::SmartPointer<const Self>  ConstPointer;

    itkNewMacro(Self);

    itkTypeMacro(GeneralizedFAImageFilter, ImageToImageFilter);

    typedef typename TInputImage::Pointer InputImagePointer;
    typedef typename TInputImage::PixelType InputImagePixel;
    typedef typename TOutputImage::Pointer OutputImagePointer;

    typedef typename Superclass::OutputImageRegionType OutputImageRegionType;

protected:
    GeneralizedFAImageFilter()
    {
    }

    virtual ~GeneralizedFAImageFilter() {}

    void DynamicThreadedGenerateData(const OutputImageRegionType &outputRegionForThread) ITK_OVERRIDE;

    bool isZero(InputImagePixel &testVal)
    {
        bool resVal = true;
        for (unsigned int i = 0;i < testVal.GetSize();++i)
        {
            if (testVal[i] != 0)
            {
                resVal = false;
                break;
            }
        }

        return resVal;
    }

private:
    ITK_DISALLOW_COPY_AND_ASSIGN(GeneralizedFAImageFilter);
};
    
} // end of namespace anima

#include "animaGeneralizedFAImageFilter.hxx"