itkBSplineInterpolateImageFunction.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  Copyright Insight Software Consortium
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0.txt
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 *=========================================================================*/
/*=========================================================================
 *
 *  Portions of this file are subject to the VTK Toolkit Version 3 copyright.
 *
 *  Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
 *
 *  For complete copyright, license and disclaimer of warranty information
 *  please refer to the NOTICE file at the top of the ITK source tree.
 *
 *=========================================================================*/
#ifndef itkBSplineInterpolateImageFunction_h
#define itkBSplineInterpolateImageFunction_h

#include <vector>

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

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

namespace itk
{
template<
  typename TImageType,
  typename TCoordRep = double,
  typename TCoefficientType = double >
class ITK_TEMPLATE_EXPORT BSplineInterpolateImageFunction:
  public InterpolateImageFunction< TImageType, TCoordRep >
{
public:
  ITK_DISALLOW_COPY_AND_ASSIGN(BSplineInterpolateImageFunction);

  using Self = BSplineInterpolateImageFunction;
  using Superclass = InterpolateImageFunction< TImageType, TCoordRep >;
  using Pointer = SmartPointer< Self >;
  using ConstPointer = SmartPointer< const Self >;

  itkTypeMacro(BSplineInterpolateImageFunction, InterpolateImageFunction);

  itkNewMacro(Self);

  using OutputType = typename Superclass::OutputType;

  using InputImageType = typename Superclass::InputImageType;

  static constexpr unsigned int ImageDimension = Superclass::ImageDimension;

  using IndexType = typename Superclass::IndexType;

  using ContinuousIndexType = typename Superclass::ContinuousIndexType;

  using PointType = typename Superclass::PointType;

  using Iterator = ImageLinearIteratorWithIndex< TImageType >;

  using CoefficientDataType = TCoefficientType;
  using CoefficientImageType = Image< CoefficientDataType,
                 Self::ImageDimension >;

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

  using CovariantVectorType = CovariantVector< OutputType,
                           Self::ImageDimension >;

  OutputType Evaluate(const PointType & point) const override
  {
    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,
                              ThreadIdType threadId) const
  {
    ContinuousIndexType index;

    this->GetInputImage()->TransformPhysicalPointToContinuousIndex(point,
                                                                   index);
    return ( this->EvaluateAtContinuousIndex(index, threadId) );
  }

  OutputType EvaluateAtContinuousIndex(const ContinuousIndexType &
                                               index) const override
  {
    // 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,
                                               ThreadIdType 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,
                                         ThreadIdType 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,
    ThreadIdType 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,
                                  ThreadIdType 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,
    ThreadIdType threadId) const;

  void SetSplineOrder(unsigned int SplineOrder);

  itkGetConstMacro(SplineOrder, int);

  void SetNumberOfWorkUnits(ThreadIdType numThreads);

  itkGetConstMacro(NumberOfWorkUnits, ThreadIdType);

  void SetInputImage(const TImageType *inputData) override;

  itkSetMacro(UseImageDirection, bool);
  itkGetConstMacro(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() override;
  void PrintSelf(std::ostream & os, Indent indent) const override;

  // 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:
  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
                                                           // indices

  CoefficientFilterPointer m_CoefficientFilter;

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

  ThreadIdType          m_NumberOfWorkUnits;
  vnl_matrix< long > *  m_ThreadedEvaluateIndex;
  vnl_matrix< double > *m_ThreadedWeights;
  vnl_matrix< double > *m_ThreadedWeightsDerivative;
};
} // namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkBSplineInterpolateImageFunction.hxx"
#endif

#endif