animaNLinksFilter.h

Go to the documentation of this file.

#pragma once

#include <itkImageToImageFilter.h>
#include <itkImageRegionIterator.h>
#include <itkImageRegionConstIterator.h>
#include <itkVariableSizeMatrix.h>
#include <itkCSVArray2DFileReader.h>
#include "animaGraph.h"

namespace anima
{
template <typename TInput, typename TOutput>
class NLinksFilter :
        public itk::ImageToImageFilter< TInput , TOutput >
{
public:
    typedef NLinksFilter Self;
    typedef itk::ImageToImageFilter< TInput , TOutput> Superclass;
    typedef itk::SmartPointer<Self> Pointer;
    typedef itk::SmartPointer<const Self>  ConstPointer;

    itkNewMacro(Self)

    
    itkTypeMacro(NLinksFilter, ImageToImageFilter)

    
    typedef typename TInput::Pointer InputImagePointer;
    typedef typename TInput::ConstPointer InputImageConstPointer;
    typedef itk::ImageRegionConstIterator< TInput > InRegionConstIteratorType;
    typedef typename TInput::PixelType InputPixelType;

    typedef typename TOutput::Pointer OutputImagePointer;
    typedef typename TOutput::PixelType OutputPixelType;
    typedef typename itk::ImageRegionIterator < TOutput > OutputIteratorType;

    typedef double  PixelTypeD;
    typedef itk::Image <PixelTypeD,3> TSeedProba;
    typedef TSeedProba::Pointer ImagePointerProba;
    typedef itk::ImageRegionIterator <TSeedProba> ImageIteratorTypeProba;
    typedef TSeedProba::PixelType SeedProbaPixelType;

    typedef int     PixelTypeInt;
    typedef itk::Image <PixelTypeInt,3> ImageTypeInt;
    typedef ImageTypeInt::IndexType pixelIndexInt;
    typedef itk::ImageRegionIterator <ImageTypeInt> ImageIteratorTypeInt;

    typedef unsigned char   PixelTypeUC;
    typedef itk::Image <PixelTypeUC,3> TMask;
    typedef TMask::Pointer TMaskPointer;
    typedef itk::ImageRegionIterator< TMask > MaskRegionIteratorType;
    typedef itk::ImageRegionConstIterator< TMask > MaskRegionConstIteratorType;

    typedef Graph<double,double,double> GraphType;

    typedef double NumericType;
    typedef itk::VariableSizeMatrix<NumericType> doubleVariableSizeMatrixType;

    void SetInputImage1(const TInput* image);
    void SetInputImage2(const TInput* image);
    void SetInputImage3(const TInput* image);
    void SetInputImage4(const TInput* image);
    void SetInputImage5(const TInput* image);

    void SetMask(const TMask* mask);

    void SetInputSeedProbaSources(const TSeedProba* mask);
    void SetInputSeedProbaSinks(const TSeedProba* mask);

    void SetMatFilename(std::string mat){m_MatFilename=mat;}
    void SetMatrix(doubleVariableSizeMatrixType mat){m_Matrix=mat;}

    OutputImagePointer GetOutput();
    OutputImagePointer GetOutputBackground();

    typedef typename Superclass::OutputImageRegionType OutputImageRegionType;

    void SetTol(const double tol)
    {
        this->SetCoordinateTolerance(tol);
        this->SetDirectionTolerance(tol);
        m_Tol = tol;
    }

    itkSetMacro(Sigma, double)
    itkGetMacro(Sigma, double)

    itkSetMacro(UseSpectralGradient, bool)
    itkGetMacro(UseSpectralGradient, bool)

    itkSetMacro(NbModalities, unsigned int)
    itkGetMacro(NbModalities, unsigned int)

    itkSetMacro(Verbose, bool)
    itkGetMacro(Verbose, bool)

protected:
    NLinksFilter()
    {
        this->SetNthOutput( 0, this->MakeOutput(0) );
        this->SetNthOutput( 1, this->MakeOutput(1) );

        m_Sigma = 0.6;
        m_UseSpectralGradient=true;
        m_Verbose=false;
        m_NbModalities = 0;
        m_NbInputs = 4;
        m_NbMaxImage = 10;
        m_IndexImage1=m_NbMaxImage,m_IndexImage2=m_NbMaxImage,m_IndexImage3=m_NbMaxImage, m_IndexImage4=m_NbMaxImage,m_IndexImage5=m_NbMaxImage;

        m_Tol = 0.0001;

        this->SetNumberOfRequiredOutputs(2);
        this->SetNumberOfRequiredInputs(4);
    }

    virtual ~NLinksFilter()
    {
    }

    itk::DataObject::Pointer MakeOutput(unsigned int idx);

    typename TInput::ConstPointer GetInputImage1();
    typename TInput::ConstPointer GetInputImage2();
    typename TInput::ConstPointer GetInputImage3();
    typename TInput::ConstPointer GetInputImage4();
    typename TInput::ConstPointer GetInputImage5();
    typename TInput::ConstPointer GetInputImage6();

    TMask::ConstPointer GetMask();

    TSeedProba::ConstPointer GetInputSeedProbaSources();
    TSeedProba::ConstPointer GetInputSeedProbaSinks();

    doubleVariableSizeMatrixType GetMatrix(void){return m_Matrix;}
    std::string GetMatFilename(void){return m_MatFilename;}

    bool readMatrixFile();

    void CheckSpectralGradient(void);
    void GenerateData() ITK_OVERRIDE;
    void SetGraph();
    bool isInside (unsigned int x,unsigned int y,unsigned int z ) const;
    void CreateGraph();
    double computeNLink(int i1, int j1, int k1, int i2, int j2, int k2);

private:
    ITK_DISALLOW_COPY_AND_ASSIGN(NLinksFilter);

    bool m_UseSpectralGradient;

    double m_Sigma;

    GraphType *m_graph;

    std::string m_MatFilename;
    doubleVariableSizeMatrixType m_Matrix;

    typename TOutput::SizeType m_size;

    std::vector<InputImageConstPointer > m_ListImages;

    bool m_Verbose;

    ImageTypeInt::Pointer pix;

    TSeedProba::Pointer m_e1, m_e2; // Precomputed spectral grad quantities (keep track of 2 images instead of 3...
    unsigned int m_NbModalities;
    unsigned int m_NbInputs, m_NbMaxImage;
    unsigned int m_IndexImage1, m_IndexImage2, m_IndexImage3, m_IndexImage4,m_IndexImage5;

    double m_Tol;
};

} // end of namespace anima

#include "animaNLinksFilter.hxx"