animaCheckStructureNeighborFilter.h

Go to the documentation of this file.

#pragma once

#include <animaReadWriteFunctions.h>

#include <itkImageToImageFilter.h>
#include <itkImageRegionIterator.h>
#include <itkImageRegionConstIterator.h>
#include <itkRelabelComponentImageFilter.h>
#include <itkConnectedComponentImageFilter.h>
#include <itkGrayscaleDilateImageFilter.h>
#include <itkBinaryBallStructuringElement.h>
#include <itkBinaryImageToLabelMapFilter.h>
#include <itkLabelMapToLabelImageFilter.h>
#include <itkLabelContourImageFilter.h>

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

    itkNewMacro(Self)

    
    itkTypeMacro(CheckStructureNeighborFilter, ImageToImageFilter)

    
    typedef TInput InputImageType;
    typedef typename InputImageType::ConstPointer InputImageConstPointer;
    typedef typename InputImageType::PixelType InputPixelType;
    typedef typename itk::ImageRegionConstIterator< InputImageType > InputConstIteratorType;

    typedef TMask MaskImageType;
    typedef typename MaskImageType::ConstPointer MaskImageConstPointer;
    typedef typename MaskImageType::PixelType MaskPixelType;
    typedef typename itk::ImageRegionConstIterator< MaskImageType > MaskConstIteratorType;

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

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

    typedef typename itk::ConnectedComponentImageFilter <InputImageType,ImageTypeInt> ConnectedComponentFilterType;
    typedef itk::LabelContourImageFilter<ImageTypeInt,ImageTypeInt> LabelContourFilterType;
    typedef itk::BinaryBallStructuringElement<PixelTypeInt, 3> StructuringElementType;
    typedef itk::GrayscaleDilateImageFilter <ImageTypeInt,ImageTypeInt,StructuringElementType> DilateFilterType;

    void SetInputClassification(const TInput* image);
    void SetInputMap(const TMask* image);

    void WriteOutputs();

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

    typedef typename Superclass::OutputImageRegionType OutputImageRegionType;

    itkSetMacro(LabelToCheck, int)
    itkGetMacro(LabelToCheck, int)

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

    itkSetMacro(Ratio, double)
    itkGetMacro(Ratio, double)

    std::string GetOutputFilename() {return m_OutputFilename;}
    void SetOutputFilename(std::string filename) {m_OutputFilename=filename;}

protected:
    CheckStructureNeighborFilter()
    {
        this->SetNumberOfRequiredOutputs(1);
        this->SetNumberOfRequiredInputs(2);

        m_Verbose=false;
        m_Tol = 0.0001;
        m_Ratio = 0;

        this->SetNumberOfWorkUnits(itk::MultiThreaderBase::GetGlobalDefaultNumberOfThreads());
    }

    virtual ~CheckStructureNeighborFilter()
    {
    }

    typename TInput::ConstPointer GetInputClassification();
    typename TMask::ConstPointer GetInputMap();

    void GenerateData() ITK_OVERRIDE;

private:
    ITK_DISALLOW_COPY_AND_ASSIGN(CheckStructureNeighborFilter);

    std::string m_OutputFilename;
    int m_LabelToCheck;
    bool m_Verbose;
    double m_Tol;
    double m_Ratio;

};

} // end of namespace anima

#include "animaCheckStructureNeighborFilter.hxx"