itkPSMProcrustesFunction.h

/*=========================================================================
 *
 *  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 __itkPSMProcrustesFunction_h
#define __itkPSMProcrustesFunction_h

#include <vector>
#include <vnl/vnl_matrix.h>
#include <vnl/vnl_vector_fixed.h>
#include <vnl/vnl_matrix_fixed.h>
#include "itkSmartPointer.h"
#include "itkMacro.h"
#include "itkObject.h"
#include "itkObjectFactory.h"
#include "itkPoint.h"

namespace itk
{

template <unsigned int VDimension>
class ITK_EXPORT PSMProcrustesFunction : public itk::Object
{
  struct SimilarityTransform3D
  {
    vnl_matrix_fixed<double, VDimension, VDimension> rotation;
    double scale;
    vnl_vector_fixed<double, VDimension> translation;
  };
    
public:
  typedef double                                RealType;
  typedef vnl_vector_fixed<double, VDimension>  PointType;
  typedef std::vector<PointType>                ShapeType;
  typedef typename ShapeType::iterator          ShapeIteratorType;
  typedef std::vector<ShapeType>                ShapeListType;
  typedef typename ShapeListType::iterator      ShapeListIteratorType;
  typedef std::vector<SimilarityTransform3D>    SimilarityTransformListType;
  typedef typename SimilarityTransformListType::iterator SimilarityTransformListIteratorType;
  
  typedef PSMProcrustesFunction                 Self;
  typedef Object                                Superclass;
  typedef SmartPointer< Self >                  Pointer;
  typedef SmartPointer< const Self >            ConstPointer;
  
  itkNewMacro(Self);

  itkTypeMacro(PSMProcrustesFunction, Object);

  void RunGeneralizedProcrustes(SimilarityTransformListType & transform,
                                ShapeListType & shapes);

  RealType ComputeSumOfSquares(ShapeListType & shapes);

  ShapeType TransformShape(ShapeType shape, SimilarityTransform3D & transform);
  
private:        
  bool CheckDegenerateCase(PointType ssqShape, PointType ssqMean, PointType muShape,
                           PointType muMean, int rows);

  void LeaveOneOutMean(ShapeType & mean, ShapeListType & shapeList,
                       ShapeListIteratorType & leaveOutIt);

  ShapeType RunProcrustes(SimilarityTransform3D & transform, ShapeType mean,
                          ShapeListIteratorType & leaveOutIt);
};
} // end namespace itk
#endif