itkOptBSplineInterpolateImageFunction.h

Go to the documentation of this file.

/*=========================================================================

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkOptBSplineInterpolateImageFunction.h,v $
  Language:  C++
  Date:      $Date: 2009-02-07 16:05:47 $
  Version:   $Revision: 1.10 $

  Copyright (c) Insight Software Consortium. All rights reserved.
  See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.

  Portions of this code are covered under the VTK copyright.
  See VTKCopyright.txt or http://www.kitware.com/VTKCopyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/
#ifndef __itkOptBSplineInterpolateImageFunction_h
#define __itkOptBSplineInterpolateImageFunction_h

#include <vector>

#include "itkImageLinearIteratorWithIndex.h"
#include "itkInterpolateImageFunction.h"
#include "vnl/vnl_matrix.h"

#include "itkBSplineDecompositionImageFilter.h"
#include "itkConceptChecking.h"
#include "itkCovariantVector.h"

namespace itk
{
template <
  class TImageType,
  class TCoordRep = double,
  class TCoefficientType = double >
class ITK_EXPORT BSplineInterpolateImageFunction :
    public InterpolateImageFunction<TImageType,TCoordRep>
{
public:
  typedef BSplineInterpolateImageFunction                   Self;
  typedef InterpolateImageFunction<TImageType,TCoordRep>    Superclass;
  typedef SmartPointer<Self>                                Pointer;
  typedef SmartPointer<const Self>                          ConstPointer;

  itkTypeMacro(BSplineInterpolateImageFunction, InterpolateImageFunction);


  itkNewMacro( Self );

  typedef typename Superclass::OutputType OutputType;

  typedef typename Superclass::InputImageType InputImageType;

  itkStaticConstMacro(ImageDimension, unsigned int,Superclass::ImageDimension);

  typedef typename Superclass::IndexType IndexType;

  typedef typename Superclass::ContinuousIndexType ContinuousIndexType;

  typedef typename Superclass::PointType PointType;

  typedef ImageLinearIteratorWithIndex<TImageType> Iterator;

  typedef TCoefficientType CoefficientDataType;
  typedef Image<CoefficientDataType,
                     itkGetStaticConstMacro(ImageDimension) >
                                                           CoefficientImageType;

  typedef BSplineDecompositionImageFilter<TImageType,
                                               CoefficientImageType>
                                                              CoefficientFilter;
  typedef typename CoefficientFilter::Pointer CoefficientFilterPointer;

  typedef CovariantVector<OutputType,
                          itkGetStaticConstMacro(ImageDimension) >
                                                            CovariantVectorType;


  virtual OutputType Evaluate( const PointType & point ) const
    {
    ContinuousIndexType index;
    this->GetInputImage()->TransformPhysicalPointToContinuousIndex( point,
                                                                    index );
    // No thread info passed in, so call method that doesn't need thread ID.
    return ( this->EvaluateAtContinuousIndex( index ) );
    }

  virtual OutputType Evaluate( const PointType & point,
                               unsigned int threadID ) const
    {
    ContinuousIndexType index;
    this->GetInputImage()->TransformPhysicalPointToContinuousIndex( point,
                                                                    index );
    return ( this->EvaluateAtContinuousIndex( index, threadID ) );
    }

  virtual OutputType EvaluateAtContinuousIndex( const ContinuousIndexType &
                                                                 index ) const
    {
    // Don't know thread information, make evaluateIndex, weights on the stack.
    // Slower, but safer.
    vnl_matrix<long>        evaluateIndex(ImageDimension, ( m_SplineOrder + 1 ));
    vnl_matrix<double>      weights(ImageDimension, ( m_SplineOrder + 1 ));

    // Pass evaluateIndex, weights by reference. They're only good as long
    // as this method is in scope.
    return this->EvaluateAtContinuousIndexInternal( index,
                                                    evaluateIndex,
                                                    weights);
    }

  virtual OutputType EvaluateAtContinuousIndex( const ContinuousIndexType &
                                                                        index,
                                                unsigned int threadID ) const;

  CovariantVectorType EvaluateDerivative( const PointType & point ) const
    {
    ContinuousIndexType index;
    this->GetInputImage()->TransformPhysicalPointToContinuousIndex( point,
                                                                    index );
    // No thread info passed in, so call method that doesn't need thread ID.
    return ( this->EvaluateDerivativeAtContinuousIndex( index ) );
    }

  CovariantVectorType EvaluateDerivative( const PointType & point,
                                          unsigned int threadID ) const
    {
    ContinuousIndexType index;
    this->GetInputImage()->TransformPhysicalPointToContinuousIndex( point,
                                                                    index );
    return ( this->EvaluateDerivativeAtContinuousIndex( index, threadID ) );
    }

  CovariantVectorType EvaluateDerivativeAtContinuousIndex(
                                         const ContinuousIndexType & x ) const
    {
    // Don't know thread information, make evaluateIndex, weights, weightsDerivative
    // on the stack.
    // Slower, but safer.
    vnl_matrix<long>          evaluateIndex(ImageDimension, ( m_SplineOrder + 1 ));
    vnl_matrix<double>        weights(ImageDimension, ( m_SplineOrder + 1 ));
    vnl_matrix<double>        weightsDerivative(ImageDimension, ( m_SplineOrder + 1));

    // Pass evaluateIndex, weights, weightsDerivative by reference. They're only good
    // as long as this method is in scope.
    return this->EvaluateDerivativeAtContinuousIndexInternal( x,
                                                              evaluateIndex,
                                                              weights,
                                                              weightsDerivative );
    }

  CovariantVectorType EvaluateDerivativeAtContinuousIndex(
                                         const ContinuousIndexType & x,
                                         unsigned int threadID ) const;

  void EvaluateValueAndDerivative( const PointType & point,
                                   OutputType & value,
                                   CovariantVectorType & deriv ) const
    {
    ContinuousIndexType index;
    this->GetInputImage()->TransformPhysicalPointToContinuousIndex( point,
                                                                    index );

    // No thread info passed in, so call method that doesn't need thread ID.
    this->EvaluateValueAndDerivativeAtContinuousIndex( index,
                                                       value,
                                                       deriv );
    }

  void EvaluateValueAndDerivative( const PointType & point,
                                   OutputType & value,
                                   CovariantVectorType & deriv,
                                   unsigned int threadID ) const
    {
    ContinuousIndexType index;
    this->GetInputImage()->TransformPhysicalPointToContinuousIndex( point,
                                                                    index );
    this->EvaluateValueAndDerivativeAtContinuousIndex( index,
                                                       value,
                                                       deriv,
                                                       threadID );
    }

  void EvaluateValueAndDerivativeAtContinuousIndex(
                                                const ContinuousIndexType & x,
                                                OutputType & value,
                                                CovariantVectorType & deriv
                                                ) const
    {
    // Don't know thread information, make evaluateIndex, weights, weightsDerivative
    // on the stack.
    // Slower, but safer.
    vnl_matrix<long>          evaluateIndex(ImageDimension, ( m_SplineOrder + 1 ));
    vnl_matrix<double>        weights(ImageDimension, ( m_SplineOrder + 1 ));
    vnl_matrix<double>        weightsDerivative(ImageDimension, ( m_SplineOrder + 1));

    // Pass evaluateIndex, weights, weightsDerivative by reference. They're only good
    // as long as this method is in scope.
    this->EvaluateValueAndDerivativeAtContinuousIndexInternal(x,
                                                              value,
                                                              deriv,
                                                              evaluateIndex,
                                                              weights,
                                                              weightsDerivative );
    }

  void EvaluateValueAndDerivativeAtContinuousIndex(
                                                const ContinuousIndexType & x,
                                                OutputType & value,
                                                CovariantVectorType & deriv,
                                                unsigned int threadID ) const;


  void SetSplineOrder(unsigned int SplineOrder);
  itkGetConstMacro(SplineOrder, int);

  void SetNumberOfThreads(unsigned int numThreads);
  itkGetMacro(NumberOfThreads, int);

  virtual void SetInputImage(const TImageType * inputData);


  itkSetMacro( UseImageDirection, bool );
  itkGetMacro( UseImageDirection, bool );
  itkBooleanMacro( UseImageDirection );


protected:

  virtual OutputType EvaluateAtContinuousIndexInternal( const ContinuousIndexType & index,
                                                        vnl_matrix<long>& evaluateIndex,
                                                        vnl_matrix<double>& weights) const;

  virtual void EvaluateValueAndDerivativeAtContinuousIndexInternal( const ContinuousIndexType & x,
                                                       OutputType & value,
                                                       CovariantVectorType & derivativeValue,
                                                       vnl_matrix<long>& evaluateIndex,
                                                       vnl_matrix<double>& weights,
                                                       vnl_matrix<double>& weightsDerivative
                                                       ) const;

  virtual CovariantVectorType EvaluateDerivativeAtContinuousIndexInternal( const ContinuousIndexType & x,
                                                                           vnl_matrix<long>& evaluateIndex,
                                                                           vnl_matrix<double>& weights,
                                                                           vnl_matrix<double>& weightsDerivative
                                                                           ) const;


  BSplineInterpolateImageFunction();
  ~BSplineInterpolateImageFunction();
  void PrintSelf(std::ostream& os, Indent indent) const;

  // These are needed by the smoothing spline routine.
  // temp storage for processing of Coefficients
  std::vector<CoefficientDataType>    m_Scratch;
  // Image size
  typename TImageType::SizeType       m_DataLength;
  // User specified spline order (3rd or cubic is the default)
  unsigned int                        m_SplineOrder;

  // Spline coefficients
  typename CoefficientImageType::ConstPointer       m_Coefficients;

private:
  BSplineInterpolateImageFunction( const Self& ); //purposely not implemented
  void operator=( const Self& ); //purposely not implemented

  void SetInterpolationWeights( const ContinuousIndexType & x,
                                const vnl_matrix<long> & EvaluateIndex,
                                vnl_matrix<double> & weights,
                                unsigned int splineOrder ) const;

  void SetDerivativeWeights( const ContinuousIndexType & x,
                             const vnl_matrix<long> & EvaluateIndex,
                             vnl_matrix<double> & weights,
                             unsigned int splineOrder ) const;

  void GeneratePointsToIndex(  );

  void DetermineRegionOfSupport( vnl_matrix<long> & evaluateIndex,
                                 const ContinuousIndexType & x,
                                 unsigned int splineOrder ) const;

  void ApplyMirrorBoundaryConditions(vnl_matrix<long> & evaluateIndex,
                                     unsigned int splineOrder) const;


  Iterator                  m_CIterator;    // Iterator for traversing spline coefficients.
  unsigned long             m_MaxNumberInterpolationPoints; // number of neighborhood points used for interpolation
  std::vector<IndexType>    m_PointsToIndex;  // Preallocation of interpolation neighborhood indicies

  CoefficientFilterPointer     m_CoefficientFilter;

  // flag to take or not the image direction into account when computing the
  // derivatives.
  bool m_UseImageDirection;

  unsigned int         m_NumberOfThreads;
  vnl_matrix<long>   * m_ThreadedEvaluateIndex;
  vnl_matrix<double> * m_ThreadedWeights;
  vnl_matrix<double> * m_ThreadedWeightsDerivative;
};

} // namespace itk

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

#endif