itkKernelTransform.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  Copyright NumFOCUS
 *
 *  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
 *
 *         https://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 itkKernelTransform_h
#define itkKernelTransform_h
 
#include "itkTransform.h"
#include "itkMatrix.h"
#include "itkPointSet.h"
#include <deque>
#include <cmath>
#include "vnl/vnl_matrix_fixed.h"
#include "vnl/vnl_matrix.h"
#include "vnl/vnl_vector.h"
#include "vnl/vnl_vector_fixed.h"
#include "vnl/algo/vnl_svd.h"
 
namespace itk
{
template <typename TParametersValueType, unsigned int VDimension>
class ITK_TEMPLATE_EXPORT KernelTransform : public Transform<TParametersValueType, VDimension, VDimension>
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(KernelTransform);
 
  using Self = KernelTransform;
  using Superclass = Transform<TParametersValueType, VDimension, VDimension>;
  using Pointer = SmartPointer<Self>;
  using ConstPointer = SmartPointer<const Self>;
 
  itkOverrideGetNameOfClassMacro(KernelTransform);
 
  itkNewMacro(Self);
 
  static constexpr unsigned int SpaceDimension = VDimension;
 
  using typename Superclass::ScalarType;
 
  using typename Superclass::FixedParametersType;
  using typename Superclass::ParametersType;
 
  using typename Superclass::JacobianType;
  using typename Superclass::JacobianPositionType;
  using typename Superclass::InverseJacobianPositionType;
 
  using typename Superclass::TransformCategoryEnum;
 
  using typename Superclass::InputPointType;
  using typename Superclass::OutputPointType;
 
  using typename Superclass::InputVectorType;
  using typename Superclass::OutputVectorType;
 
  using typename Superclass::InputCovariantVectorType;
  using typename Superclass::OutputCovariantVectorType;
 
  using typename Superclass::InputVnlVectorType;
  using typename Superclass::OutputVnlVectorType;
 
  using typename Superclass::NumberOfParametersType;
 
  using PointSetTraitsType =
    DefaultStaticMeshTraits<TParametersValueType, VDimension, VDimension, TParametersValueType, TParametersValueType>;
  using PointSetType = PointSet<InputPointType, VDimension, PointSetTraitsType>;
 
  using PointSetPointer = typename PointSetType::Pointer;
  using PointsContainer = typename PointSetType::PointsContainer;
  using PointsIterator = typename PointSetType::PointsContainerIterator;
  using PointsConstIterator = typename PointSetType::PointsContainerConstIterator;
  using PointIdentifier = typename PointSetType::PointIdentifier;
 
  using VectorSetType = itk::VectorContainer<SizeValueType, InputVectorType>;
  using VectorSetPointer = typename VectorSetType::Pointer;
 
  itkGetModifiableObjectMacro(SourceLandmarks, PointSetType); // NOTE: This is used to circumvent the SetTargetLandmarks
  virtual void
  SetSourceLandmarks(PointSetType *);
  itkGetModifiableObjectMacro(TargetLandmarks, PointSetType); // NOTE: This is used to circumvent the SetTargetLandmarks
  virtual void
  SetTargetLandmarks(PointSetType *);
  itkGetModifiableObjectMacro(Displacements, VectorSetType);
 
  void
  ComputeWMatrix();
 
  OutputPointType
  TransformPoint(const InputPointType & thisPoint) const override;
 
  using Superclass::TransformVector;
  OutputVectorType
  TransformVector(const InputVectorType &) const override
  {
    itkExceptionMacro("TransformVector(const InputVectorType &) is not implemented for KernelTransform");
  }
  OutputVnlVectorType
  TransformVector(const InputVnlVectorType &) const override
  {
    itkExceptionMacro("TransformVector(const InputVnlVectorType &) is not implemented for KernelTransform");
  }
 
  using Superclass::TransformCovariantVector;
  OutputCovariantVectorType
  TransformCovariantVector(const InputCovariantVectorType &) const override
  {
    itkExceptionMacro(
      << "TransformCovariantVector(const InputCovariantVectorType &) is not implemented for KernelTransform");
  }
  using IMatrixType = vnl_matrix_fixed<TParametersValueType, VDimension, VDimension>;
 
  void
  ComputeJacobianWithRespectToParameters(const InputPointType & p, JacobianType & jacobian) const override;
 
  void
  ComputeJacobianWithRespectToPosition(const InputPointType &, JacobianPositionType &) const override
  {
    itkExceptionMacro("ComputeJacobianWithRespectToPosition not yet implemented "
                      "for "
                      << this->GetNameOfClass());
  }
  using Superclass::ComputeJacobianWithRespectToPosition;
 
  void
  SetParameters(const ParametersType &) override;
 
  void
  SetFixedParameters(const FixedParametersType &) override;
 
  virtual void
  UpdateParameters() const;
 
  const ParametersType &
  GetParameters() const override;
 
  const FixedParametersType &
  GetFixedParameters() const override;
 
  TransformCategoryEnum
  GetTransformCategory() const override
  {
    return Self::TransformCategoryEnum::Spline;
  }
 
  itkSetClampMacro(Stiffness, double, 0.0, NumericTraits<double>::max());
  itkGetConstMacro(Stiffness, double);
protected:
  KernelTransform();
  ~KernelTransform() override = default;
  void
  PrintSelf(std::ostream & os, Indent indent) const override;
 
public:
  using GMatrixType = vnl_matrix_fixed<TParametersValueType, VDimension, VDimension>;
 
  using LMatrixType = vnl_matrix<TParametersValueType>;
 
  using KMatrixType = vnl_matrix<TParametersValueType>;
 
  using PMatrixType = vnl_matrix<TParametersValueType>;
 
  using YMatrixType = vnl_matrix<TParametersValueType>;
 
  using WMatrixType = vnl_matrix<TParametersValueType>;
 
  using DMatrixType = vnl_matrix<TParametersValueType>;
 
  using AMatrixType = vnl_matrix_fixed<TParametersValueType, VDimension, VDimension>;
 
  using BMatrixType = vnl_vector_fixed<TParametersValueType, VDimension>;
 
  using RowMatrixType = vnl_matrix_fixed<TParametersValueType, 1, VDimension>;
 
  using ColumnMatrixType = vnl_matrix_fixed<TParametersValueType, VDimension, 1>;
 
protected:
  virtual void
  ComputeG(const InputVectorType & landmarkVector, GMatrixType & gmatrix) const;
 
  virtual const GMatrixType & ComputeReflexiveG(PointsIterator) const;
 
  virtual void
  ComputeDeformationContribution(const InputPointType & thisPoint, OutputPointType & result) const;
 
  void
  ComputeK();
 
  void
  ComputeL();
 
  void
  ComputeP();
 
  void
  ComputeY();
 
  void
  ComputeD();
 
  void
  ReorganizeW();
 
  double m_Stiffness{};
 
  VectorSetPointer m_Displacements{};
 
  LMatrixType m_LMatrix{};
 
  KMatrixType m_KMatrix{};
 
  PMatrixType m_PMatrix{};
 
  YMatrixType m_YMatrix{};
 
  WMatrixType m_WMatrix{};
 
  DMatrixType m_DMatrix{};
 
  AMatrixType m_AMatrix{};
 
  BMatrixType m_BVector{};
 
  mutable GMatrixType m_GMatrix{};
 
  bool m_WMatrixComputed{};
 
  IMatrixType m_I{};
 
  PointSetPointer m_SourceLandmarks{};
 
  PointSetPointer m_TargetLandmarks{};
 
private:
};
} // end namespace itk
 
#ifndef ITK_MANUAL_INSTANTIATION
#  include "itkKernelTransform.hxx"
#endif
 
#endif // itkKernelTransform_h