00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017 #ifndef __itkKernelTransform_h
00018 #define __itkKernelTransform_h
00019
00020 #include "itkTransform.h"
00021 #include "itkPoint.h"
00022 #include "itkVector.h"
00023 #include "itkMatrix.h"
00024 #include "itkPointSet.h"
00025 #include <deque>
00026 #include <math.h>
00027 #include "vnl/vnl_matrix_fixed.h"
00028 #include "vnl/vnl_matrix.h"
00029 #include "vnl/vnl_vector.h"
00030 #include "vnl/vnl_vector_fixed.h"
00031 #include "vnl/algo/vnl_svd.h"
00032 #include "vnl/vnl_sample.h"
00033
00034 namespace itk
00035 {
00036
00062 template <class TScalarType,
00063 unsigned int NDimensions>
00064 class ITK_EXPORT KernelTransform :
00065 public Transform<TScalarType, NDimensions,NDimensions>
00066 {
00067 public:
00069 typedef KernelTransform Self;
00070 typedef Transform<TScalarType, NDimensions, NDimensions > Superclass;
00071 typedef SmartPointer<Self> Pointer;
00072 typedef SmartPointer<const Self> ConstPointer;
00073
00075 itkTypeMacro( KernelTransform, Transform );
00076
00078 itkNewMacro( Self );
00079
00081 itkStaticConstMacro(SpaceDimension, unsigned int, NDimensions);
00082
00084 typedef typename Superclass::ScalarType ScalarType;
00085
00087 typedef typename Superclass::ParametersType ParametersType;
00088
00090 typedef typename Superclass::JacobianType JacobianType;
00091
00093 typedef typename Superclass::InputPointType InputPointType;
00094 typedef typename Superclass::OutputPointType OutputPointType;
00095
00097 typedef typename Superclass::InputVectorType InputVectorType;
00098 typedef typename Superclass::OutputVectorType OutputVectorType;
00099
00102 typedef DefaultStaticMeshTraits<TScalarType,
00103 NDimensions,
00104 NDimensions,
00105 TScalarType,
00106 TScalarType> PointSetTraitsType;
00107 typedef PointSet<InputPointType, NDimensions, PointSetTraitsType> PointSetType;
00108 typedef typename PointSetType::Pointer PointSetPointer;
00109 typedef typename PointSetType::PointsContainer PointsContainer;
00110 typedef typename PointSetType::PointsContainerIterator PointsIterator;
00111 typedef typename PointSetType::PointsContainerConstIterator PointsConstIterator;
00112
00114 typedef itk::VectorContainer<unsigned long,InputVectorType> VectorSetType;
00115 typedef typename VectorSetType::Pointer VectorSetPointer;
00116
00118 itkGetObjectMacro( SourceLandmarks, PointSetType);
00119
00121 virtual void SetSourceLandmarks(PointSetType *);
00122
00124 itkGetObjectMacro( TargetLandmarks, PointSetType);
00125
00127 virtual void SetTargetLandmarks(PointSetType *);
00128
00131 itkGetObjectMacro( Displacements, VectorSetType );
00132
00134 void ComputeWMatrix(void);
00135
00137 virtual OutputPointType TransformPoint(const InputPointType& thisPoint) const;
00138
00140 typedef vnl_matrix_fixed<TScalarType, NDimensions, NDimensions> IMatrixType;
00141
00142
00144 virtual const JacobianType & GetJacobian(const InputPointType &point ) const;
00145
00150 virtual void SetParameters(const ParametersType &);
00151
00153 virtual void UpdateParameters(void) const;
00154
00156 virtual const ParametersType& GetParameters(void) const;
00157
00168 itkSetClampMacro(Stiffness, double, 0.0, NumericTraits<double>::max());
00169 itkGetMacro(Stiffness, double);
00170
00171
00172 protected:
00173 KernelTransform();
00174 virtual ~KernelTransform();
00175 void PrintSelf(std::ostream& os, Indent indent) const;
00176
00177 public:
00179 typedef vnl_matrix_fixed<TScalarType, NDimensions, NDimensions> GMatrixType;
00180
00182 typedef vnl_matrix<TScalarType> LMatrixType;
00183
00185 typedef vnl_matrix<TScalarType> KMatrixType;
00186
00188 typedef vnl_matrix<TScalarType> PMatrixType;
00189
00191 typedef vnl_matrix<TScalarType> YMatrixType;
00192
00194 typedef vnl_matrix<TScalarType> WMatrixType;
00195
00197 typedef vnl_matrix<TScalarType> DMatrixType;
00198
00200 typedef vnl_matrix_fixed<TScalarType,NDimensions,NDimensions> AMatrixType;
00201
00203 typedef vnl_vector_fixed<TScalarType,NDimensions> BMatrixType;
00204
00206 typedef vnl_matrix_fixed<TScalarType, 1, NDimensions> RowMatrixType;
00207
00209 typedef vnl_matrix_fixed<TScalarType, NDimensions, 1> ColumnMatrixType;
00210
00212 PointSetPointer m_SourceLandmarks;
00213
00215 PointSetPointer m_TargetLandmarks;
00216
00217 protected:
00224 virtual const GMatrixType & ComputeG(const InputVectorType & landmarkVector) const;
00225
00232 virtual const GMatrixType & ComputeReflexiveG(PointsIterator) const;
00233
00234
00237 virtual void ComputeDeformationContribution( const InputPointType & inputPoint,
00238 OutputPointType & result ) const;
00239
00241 void ComputeK();
00242
00244 void ComputeL();
00245
00247 void ComputeP();
00248
00250 void ComputeY();
00251
00253 void ComputeD();
00254
00259 void ReorganizeW(void);
00260
00262 double m_Stiffness;
00263
00266 VectorSetPointer m_Displacements;
00267
00269 LMatrixType m_LMatrix;
00270
00272 KMatrixType m_KMatrix;
00273
00275 PMatrixType m_PMatrix;
00276
00278 YMatrixType m_YMatrix;
00279
00281 WMatrixType m_WMatrix;
00282
00288 DMatrixType m_DMatrix;
00289
00291 AMatrixType m_AMatrix;
00292
00294 BMatrixType m_BVector;
00295
00299 mutable GMatrixType m_GMatrix;
00300
00302 bool m_WMatrixComputed;
00303
00305 IMatrixType m_I;
00306
00307 private:
00308 KernelTransform(const Self&);
00309 void operator=(const Self&);
00310
00311 };
00312
00313 }
00314
00315 #ifndef ITK_MANUAL_INSTANTIATION
00316 #include "itkKernelTransform.txx"
00317 #endif
00318
00319 #endif // __itkKernelTransform_h