animaDTIScalarMapsImageFilter.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 <animaNumberedThreadImageToImageFilter.h>
#include <itkVectorImage.h>
#include <itkImage.h>

namespace anima

{
template <unsigned int ImageDimension = 3>
class DTIScalarMapsImageFilter :
    public anima::NumberedThreadImageToImageFilter < itk::VectorImage <double, ImageDimension>, itk::Image <double, ImageDimension> >
{
public:

    typedef itk::VectorImage <double, ImageDimension> InputImageType;
    typedef itk::Image <double, ImageDimension> OutputImageType;
    typedef InputImageType TensorImageType;
    typedef OutputImageType FAImageType;
    typedef OutputImageType ADCImageType;

    itkStaticConstMacro(InputImageDimension, unsigned int,
                        InputImageType::ImageDimension);
    itkStaticConstMacro(OutputImageDimension, unsigned int,
                        OutputImageType::ImageDimension);


    typedef DTIScalarMapsImageFilter Self;
    typedef anima::NumberedThreadImageToImageFilter <InputImageType, OutputImageType> Superclass;
    typedef itk::SmartPointer<Self> Pointer;
    typedef itk::SmartPointer<const Self> ConstPointer;

    itkNewMacro(Self)

    
    itkTypeMacro(DTIScalarMapsImageFilter, anima::NumberedThreadImageToImageFilter)

    
    typedef typename TensorImageType::PixelType               TensorVectorType;

    typedef typename TensorImageType::RegionType  TensorImageRegionType;
    typedef typename OutputImageType::RegionType  OutputImageRegionType;
    typedef typename TensorImageType::SizeType    TensorImageSizeType;

    itk::DataObject::Pointer MakeOutput(itk::ProcessObject::DataObjectPointerArraySizeType idx) ITK_OVERRIDE;

    typename OutputImageType::Pointer GetADCImage() {return this->GetOutput(0);}
    typename OutputImageType::Pointer GetFAImage() {return this->GetOutput(1);}
    typename OutputImageType::Pointer GetAxialDiffusivityImage() {return this->GetOutput(2);}
    typename OutputImageType::Pointer GetRadialDiffusivityImage() {return this->GetOutput(3);}

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

    void DynamicThreadedGenerateData(const OutputImageRegionType& outputRegionForThread) ITK_OVERRIDE;

private:
    ITK_DISALLOW_COPY_AND_ASSIGN(DTIScalarMapsImageFilter);
};

} // end of namespace anima

#include "animaDTIScalarMapsImageFilter.hxx"