itkBSplineBaseTransform.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.
 *
 *=========================================================================*/
#ifndef itkBSplineBaseTransform_h
#define itkBSplineBaseTransform_h

#include <iostream>
#include "itkTransform.h"
#include "itkImage.h"
#include "itkBSplineInterpolationWeightFunction.h"

namespace itk
{
template<typename TParametersValueType=double,
           unsigned int NDimensions = 3,
          unsigned int VSplineOrder = 3>
class ITK_TEMPLATE_EXPORT BSplineBaseTransform :
  public Transform<TParametersValueType, NDimensions, NDimensions>
{
public:
  typedef BSplineBaseTransform                                      Self;
  typedef Transform<TParametersValueType, NDimensions, NDimensions> Superclass;
  typedef SmartPointer<Self>                                        Pointer;
  typedef SmartPointer<const Self>                                  ConstPointer;

  itkTypeMacro( BSplineBaseTransform, Transform );

  itkStaticConstMacro( SpaceDimension, unsigned int, NDimensions );

  itkStaticConstMacro( SplineOrder, unsigned int, VSplineOrder );

  itkCloneMacro(Self);

  typedef typename Superclass::ScalarType ScalarType;

  typedef typename Superclass::FixedParametersType FixedParametersType;
  typedef typename Superclass::ParametersType      ParametersType;

  typedef typename Superclass::JacobianType JacobianType;

  typedef typename Superclass::TransformCategoryType TransformCategoryType;

  typedef typename Superclass::NumberOfParametersType NumberOfParametersType;

  typedef Vector<TParametersValueType, itkGetStaticConstMacro( SpaceDimension )> InputVectorType;
  typedef Vector<TParametersValueType, itkGetStaticConstMacro( SpaceDimension )> OutputVectorType;

  typedef CovariantVector<TParametersValueType, itkGetStaticConstMacro( SpaceDimension )> InputCovariantVectorType;
  typedef CovariantVector<TParametersValueType, itkGetStaticConstMacro( SpaceDimension )> OutputCovariantVectorType;

  typedef vnl_vector_fixed<TParametersValueType, SpaceDimension> InputVnlVectorType;
  typedef vnl_vector_fixed<TParametersValueType, SpaceDimension> OutputVnlVectorType;

  typedef Point <TParametersValueType, itkGetStaticConstMacro( SpaceDimension )> InputPointType;
  typedef Point <TParametersValueType, itkGetStaticConstMacro( SpaceDimension )> OutputPointType;

  void SetParameters( const ParametersType & parameters ) ITK_OVERRIDE;

  virtual void SetFixedParameters( const FixedParametersType & parameters ) ITK_OVERRIDE = 0;

  void SetParametersByValue( const ParametersType & parameters ) ITK_OVERRIDE;

  void SetIdentity();

  virtual const ParametersType & GetParameters() const ITK_OVERRIDE;

  virtual const FixedParametersType & GetFixedParameters() const ITK_OVERRIDE;

  typedef typename ParametersType::ValueType           ParametersValueType;
  typedef Image<ParametersValueType, itkGetStaticConstMacro( SpaceDimension )> ImageType;
  typedef typename ImageType::Pointer                  ImagePointer;
  typedef FixedArray<ImagePointer, NDimensions>        CoefficientImageArray;

  virtual void SetCoefficientImages( const CoefficientImageArray & images ) = 0;

  const CoefficientImageArray GetCoefficientImages() const
  {
    return this->m_CoefficientImages;
  }

  typedef typename Superclass::DerivativeType DerivativeType;

  virtual void UpdateTransformParameters( const DerivativeType & update, TParametersValueType factor = 1.0 ) ITK_OVERRIDE;

  typedef ImageRegion<itkGetStaticConstMacro( SpaceDimension )> RegionType;

  typedef typename RegionType::IndexType    IndexType;
  typedef typename RegionType::SizeType     SizeType;
  typedef typename ImageType::SpacingType   SpacingType;
  typedef typename ImageType::DirectionType DirectionType;
  typedef typename ImageType::PointType     OriginType;

  OutputPointType  TransformPoint( const InputPointType & point ) const ITK_OVERRIDE;

  typedef BSplineInterpolationWeightFunction<ScalarType,
    itkGetStaticConstMacro( SpaceDimension ),
     itkGetStaticConstMacro( SplineOrder )> WeightsFunctionType;

  typedef typename WeightsFunctionType::WeightsType         WeightsType;
  typedef typename WeightsFunctionType::ContinuousIndexType ContinuousIndexType;

  typedef Array<unsigned long> ParameterIndexArrayType;

  virtual void TransformPoint( const InputPointType & inputPoint, OutputPointType & outputPoint,
    WeightsType & weights, ParameterIndexArrayType & indices, bool & inside ) const = 0;

  unsigned long GetNumberOfWeights() const
  {
    return m_WeightsFunction->GetNumberOfWeights();
  }

  using Superclass::TransformVector;
  virtual OutputVectorType TransformVector( const InputVectorType & ) const ITK_OVERRIDE
  {
    itkExceptionMacro( << "Method not applicable for deformable transform." );
  }

  virtual OutputVnlVectorType TransformVector( const InputVnlVectorType & ) const ITK_OVERRIDE
  {
    itkExceptionMacro( << "Method not applicable for deformable transform. " );
  }

  using Superclass::TransformCovariantVector;
  virtual OutputCovariantVectorType TransformCovariantVector(
    const InputCovariantVectorType & ) const ITK_OVERRIDE
  {
    itkExceptionMacro( << "Method not applicable for deformable transform. " );
  }

  void ComputeJacobianFromBSplineWeightsWithRespectToPosition(
    const InputPointType &, WeightsType &, ParameterIndexArrayType & ) const;

  virtual void ComputeJacobianWithRespectToParameters( const InputPointType &, JacobianType & ) const ITK_OVERRIDE = 0;

  virtual void ComputeJacobianWithRespectToPosition( const InputPointType &, JacobianType & ) const ITK_OVERRIDE
  {
    itkExceptionMacro( << "ComputeJacobianWithRespectToPosition not yet implemented "
                       "for " << this->GetNameOfClass() );
  }

  virtual NumberOfParametersType GetNumberOfParameters() const ITK_OVERRIDE = 0;

  virtual NumberOfParametersType GetNumberOfParametersPerDimension() const = 0;

  virtual TransformCategoryType GetTransformCategory() const ITK_OVERRIDE
  {
    return Self::BSpline;
  }

  unsigned int GetNumberOfAffectedWeights() const;

  typedef typename ImageType::SpacingType   PhysicalDimensionsType;
  typedef typename ImageType::PixelType     PixelType;

  typedef SizeType MeshSizeType;

  virtual NumberOfParametersType GetNumberOfLocalParameters() const ITK_OVERRIDE
  {
    return this->GetNumberOfParameters();
  }

protected:
  void PrintSelf( std::ostream & os, Indent indent ) const ITK_OVERRIDE;

  BSplineBaseTransform();
  virtual ~BSplineBaseTransform() ITK_OVERRIDE;

  itkSetObjectMacro( WeightsFunction, WeightsFunctionType );
  itkGetModifiableObjectMacro(WeightsFunction, WeightsFunctionType );

  void WrapAsImages();

protected:
  void SetFixedParametersFromTransformDomainInformation() const;

  virtual void SetFixedParametersGridSizeFromTransformDomainInformation() const = 0;

  virtual void SetFixedParametersGridOriginFromTransformDomainInformation() const = 0;

  virtual void SetFixedParametersGridSpacingFromTransformDomainInformation() const = 0;

  virtual void SetFixedParametersGridDirectionFromTransformDomainInformation() const = 0;

  virtual void SetCoefficientImageInformationFromFixedParameters() =0;

  virtual bool InsideValidRegion( ContinuousIndexType & ) const = 0;

  // NOTE:  There is a natural duality between the
  //       two representations of of the coefficients
  //       whereby the m_InternalParametersBuffer is
  //       needed to fit into the optimization framework
  //       and the m_CoefficientImages is needed for
  //       the Jacobian computations.  This implementation
  //       is an attempt to remove as much redundancy as possible
  //       and share as much information between the two
  //       instances as possible.
  //
  CoefficientImageArray m_CoefficientImages;

  ParametersType m_InternalParametersBuffer;

  typename WeightsFunctionType::Pointer m_WeightsFunction;

private:
  ITK_DISALLOW_COPY_AND_ASSIGN(BSplineBaseTransform);

  static CoefficientImageArray ArrayOfImagePointerGeneratorHelper();
}; // class BSplineBaseTransform
}  // namespace itk

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

#endif /* itkBSplineBaseTransform_h */