mitkPointSetKernelTransformEstimator.h

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

  Program:   3DMed
  Date:      $Date: 2014-02-25 18:30:00 +0800 $
  Version:   $Version: 4.6.0 $
  Copyright: MIPG, Institute of Automation, Chinese Academy of Sciences

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


#ifndef __mitkPointSetKernelTransformEstimator_h
#define __mitkPointSetKernelTransformEstimator_h

#include "mitkPointSetEstimator.h"
#include "mitkPointSet.h"

#define MITK_KERNEL_FUNC_TPS_R          1
#define MITK_KERNEL_FUNC_TPS_R2LOGR     2
#define MITK_KERNEL_FUNC_EBS            3
#define MITK_KERNEL_FUNC_VS             4




class MITK_REGISTRATION_API mitkPointSetKernelTransformEstimator : public mitkPointSetEstimator
{
public:
    MITK_TYPE(mitkPointSetKernelTransformEstimator, mitkPointSetEstimator)  
    virtual void PrintSelf(ostream &os);
    
    mitkPointSetKernelTransformEstimator();

    mitkPointSetKernelTransformEstimator(int tfMode);

    void SetMovingPointSet(mitkPointSet *movingPointSet);

    void SetKernelFunction(int m) { m_KernelFunction = m; }

    void SetKernelFunctionToTPSR() { this->SetKernelFunction(MITK_KERNEL_FUNC_TPS_R); }

    void SetKernelFunctionToTPSR2LOGR() { this->SetKernelFunction(MITK_KERNEL_FUNC_TPS_R2LOGR); }

    void SetKernelFunctionToEBS() { this->SetKernelFunction(MITK_KERNEL_FUNC_EBS); }

    void SetKernelFunctionToVS() { this->SetKernelFunction(MITK_KERNEL_FUNC_VS); }

    void SetAlphaCoefficient(ScalarParameterType a) { m_Alpha = a; }

    ScalarParameterType GetAlphaCoefficient() { return m_Alpha; }

    void SetStiffnessCoefficient(ScalarParameterType s) { m_Stiffness = s; }

    ScalarParameterType GetStiffnessCoefficient() { return m_Stiffness; }

    MatrixType& GetDeformMatrix() { return m_DMatrix; }

    bool IsTransformSupported(mitkTransform* transform = NULL);

    bool ComputeWMatrix();

protected:
    virtual ~mitkPointSetKernelTransformEstimator();
    virtual bool Execute();
    void ComputeLMatrix();
    void ComputeKMatrix();
    void ComputePMatrix();
    void ComputeYMatrix();
    void ComputeDisplacement();
    void ComputeGMatrix(const mitkPoint<double> & s);

    mitkPointSet                m_Displacements;

    MatrixType                  m_LMatrix;
    MatrixType                  m_KMatrix;
    MatrixType                  m_PMatrix;
    MatrixType                  m_YMatrix;
    MatrixType                  m_WMatrix;
    MatrixType                  m_GMatrix;
    MatrixType                  m_DMatrix;

    int                         m_KernelFunction;
    ScalarParameterType         m_Stiffness;
    ScalarParameterType         m_Alpha;

    //affine and translation parts can use mitkTransformMatrix


private:
    mitkPointSetKernelTransformEstimator(const mitkPointSetKernelTransformEstimator&);
    void operator = (const mitkPointSetKernelTransformEstimator&);

};


//#define DEFINED_mitkPointSetKernelTransformEstimator



#endif

---