itkDiffusionTensor3DFSAffineTransform.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 itkDiffusionTensor3DFSAffineTransform_h
#define itkDiffusionTensor3DFSAffineTransform_h

#include "itkDiffusionTensor3DAffineTransform.h"
#include <vnl/vnl_matrix.h>
#include <vnl/algo/vnl_real_eigensystem.h>
#include <vnl/vnl_real.h>

namespace itk
{
template <class TData>
class DiffusionTensor3DFSAffineTransform :
  public DiffusionTensor3DAffineTransform<TData>
{
public:
  typedef TData                                            DataType;
  typedef DiffusionTensor3DFSAffineTransform               Self;
  typedef DiffusionTensor3DAffineTransform<DataType>       Superclass;
  typedef typename Superclass::MatrixTransformType         MatrixTransformType;
  typedef typename Superclass::InternalMatrixTransformType InternalMatrixTransformType;
  typedef SmartPointer<Self>                               Pointer;
  typedef SmartPointer<const Self>                         ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(DiffusionTensor3DFSAffineTransform, DiffusionTensor3DAffineTransform);

protected:
  void PreCompute() override;

private:
  MatrixTransformType ComputeMatrixSquareRoot( MatrixTransformType matrix );

  MatrixTransformType ComputeRotationMatrixFromTransformationMatrix();

};

} // end namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkDiffusionTensor3DFSAffineTransform.txx"
#endif

#endif