vtkThinPlateSplineTransform.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#ifndef vtkThinPlateSplineTransform_h
#define vtkThinPlateSplineTransform_h
 
#include "vtkCommonTransformsModule.h" // For export macro
#include "vtkWarpTransform.h"
 
#define VTK_RBF_CUSTOM 0
#define VTK_RBF_R 1
#define VTK_RBF_R2LOGR 2
 
VTK_ABI_NAMESPACE_BEGIN
class VTKCOMMONTRANSFORMS_EXPORT vtkThinPlateSplineTransform : public vtkWarpTransform
{
public:
  vtkTypeMacro(vtkThinPlateSplineTransform, vtkWarpTransform);
  void PrintSelf(ostream& os, vtkIndent indent) override;
  static vtkThinPlateSplineTransform* New();
 
 
  vtkGetMacro(Sigma, double);
  vtkSetMacro(Sigma, double);
 
 
  void SetBasis(int basis);
  vtkGetMacro(Basis, int);
  void SetBasisToR() { this->SetBasis(VTK_RBF_R); }
  void SetBasisToR2LogR() { this->SetBasis(VTK_RBF_R2LOGR); }
  const char* GetBasisAsString();
 
 
  void SetBasisFunction(double (*U)(double r))
  {
    if (this->BasisFunction == U)
    {
      return;
    }
    this->SetBasis(VTK_RBF_CUSTOM);
    this->BasisFunction = U;
    this->Modified();
  }
  void SetBasisDerivative(double (*dUdr)(double r, double& dU))
  {
    this->BasisDerivative = dUdr;
    this->Modified();
  }
 
 
  void SetSourceLandmarks(vtkPoints* source);
  vtkGetObjectMacro(SourceLandmarks, vtkPoints);
 
 
  void SetTargetLandmarks(vtkPoints* target);
  vtkGetObjectMacro(TargetLandmarks, vtkPoints);
 
  vtkMTimeType GetMTime() override;
 
  vtkAbstractTransform* MakeTransform() override;
 
 
  vtkGetMacro(RegularizeBulkTransform, bool);
  vtkSetMacro(RegularizeBulkTransform, bool);
  vtkBooleanMacro(RegularizeBulkTransform, bool);
 
protected:
  vtkThinPlateSplineTransform();
  ~vtkThinPlateSplineTransform() override;
 
  void InternalUpdate() override;
 
  void InternalDeepCopy(vtkAbstractTransform* transform) override;
 
  void ForwardTransformPoint(const float in[3], float out[3]) override;
  void ForwardTransformPoint(const double in[3], double out[3]) override;
 
  void ForwardTransformDerivative(const float in[3], float out[3], float derivative[3][3]) override;
  void ForwardTransformDerivative(
    const double in[3], double out[3], double derivative[3][3]) override;
 
  double Sigma;
  vtkPoints* SourceLandmarks;
  vtkPoints* TargetLandmarks;
 
  // the radial basis function to use
  double (*BasisFunction)(double r);
  double (*BasisDerivative)(double r, double& dUdr);
 
  int Basis;
 
  int NumberOfPoints;
  double** MatrixW;
 
  bool RegularizeBulkTransform;
 
private:
  vtkThinPlateSplineTransform(const vtkThinPlateSplineTransform&) = delete;
  void operator=(const vtkThinPlateSplineTransform&) = delete;
};
 
VTK_ABI_NAMESPACE_END
#endif