itkDiffusionTensor3DZeroCorrection.h

Go to the documentation of this file.

/*=========================================================================
 
  Program:   Diffusion Applications
  Module:    $HeadURL$
  Language:  C++
  Date:      $Date$
  Version:   $Revision$
 
  Copyright (c) Brigham and Women's Hospital (BWH) All Rights Reserved.
 
  See License.txt or http://www.slicer.org/copyright/copyright.txt for details.
 
==========================================================================*/
#ifndef itkDiffusionTensor3DZeroCorrection_h
#define itkDiffusionTensor3DZeroCorrection_h
 
#include "itkUnaryFunctorImageFilter.h"
#include "itkMath.h"
#include <itkMatrix.h>
#include "itkDiffusionTensor3DExtended.h"
#include "itkDiffusionTensor3DConstants.h"
 
namespace itk
{

namespace Functor
{
 
template <class TInput, class TOutput>
class DiffusionTensor3DZero
{
public:
  DiffusionTensor3DZero() = default;
  ~DiffusionTensor3DZero() = default;
  bool operator!=(const DiffusionTensor3DZero& other) const { return *this != other; }
 
  bool operator==(const DiffusionTensor3DZero& other) const { return !(*this != other); }
 
  inline DiffusionTensor3D<TOutput> operator()(const DiffusionTensor3D<TInput>& A)
  {
    DiffusionTensor3D<TOutput> tensor;
    Matrix<double, 3, 3> mat;
    Matrix<double, 3, 3> matcorrect;
    typename DiffusionTensor3DExtended<double>::EigenValuesArrayType eigenValues;
    typename DiffusionTensor3DExtended<double>::EigenVectorsMatrixType eigenVectors;
    DiffusionTensor3DExtended<double> tensorDouble(A);
    tensorDouble.ComputeEigenAnalysis(eigenValues, eigenVectors);
    for (int i = 0; i < 3; i++)
    {
      mat[i][i] = (eigenValues[i] <= 0 ? ITK_DIFFUSION_TENSOR_3D_ZERO : eigenValues[i]);
    }
    eigenVectors = eigenVectors.GetTranspose();
    matcorrect = eigenVectors * mat * eigenVectors.GetInverse();
    tensorDouble.SetTensorFromMatrix(matcorrect);
    for (int i = 0; i < 6; i++)
    {
      tensor[i] = (TOutput)tensorDouble[i];
    }
    return tensor;
  }
};
} // namespace Functor
 
template <class TInputImage, class TOutputImage>
class DiffusionTensor3DZeroCorrectionFilter
  : public UnaryFunctorImageFilter<TInputImage,
                                   TOutputImage,
                                   Functor::DiffusionTensor3DZero<typename TInputImage::PixelType::ComponentType, typename TOutputImage::PixelType::ComponentType>>
{
public:
  typedef DiffusionTensor3DZeroCorrectionFilter Self;
  typedef UnaryFunctorImageFilter<TInputImage, TOutputImage, Functor::DiffusionTensor3DZero<typename TInputImage::PixelType, typename TOutputImage::PixelType>> Superclass;
  typedef SmartPointer<Self> Pointer;
  typedef SmartPointer<const Self> ConstPointer;

  DiffusionTensor3DZeroCorrectionFilter(const Self&) = delete;
  void operator=(const Self&) = delete;

  itkTypeMacro(DiffusionTensor3DZeroCorrectionFilter, UnaryFunctorImageFilter);

  itkNewMacro(Self);
 
#ifdef ITK_USE_CONCEPT_CHECKING
  itkConceptMacro(InputTensorTypeCheck, (Concept::SameType<DiffusionTensor3D<typename TInputImage::PixelType::ComponentType>, typename TInputImage::PixelType>));
  itkConceptMacro(OutputTensorTypeCheck, (Concept::SameType<DiffusionTensor3D<typename TOutputImage::PixelType::ComponentType>, typename TOutputImage::PixelType>));

#endif
protected:
  DiffusionTensor3DZeroCorrectionFilter() = default;
  ~DiffusionTensor3DZeroCorrectionFilter() override = default;
};
 
} // end namespace itk
 
#endif