itkPSMProcrustesRegistration.h

/*=========================================================================
  Program:   ShapeWorks: Particle-based Shape Correspondence & Visualization
  Module:    $RCSfile: itkPSMProcrustesRegistration.h,v $
  Date:      $Date: 2011/03/24 01:17:33 $
  Version:   $Revision: 1.3 $
  Author:    $Author: wmartin $

  Copyright (c) 2009 Scientific Computing and Imaging Institute.
  See ShapeWorksLicense.txt for details.

     This software is distributed WITHOUT ANY WARRANTY; without even 
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
     PURPOSE.  See the above copyright notices for more information.
=========================================================================*/
#ifndef __itkPSMProcrustesRegistration_h
#define __itkPSMProcrustesRegistration_h

#include "itkDataObject.h"
#include "itkWeakPointer.h"
#include "vnl/vnl_matrix.h"
#include "itkPSMParticleSystem.h"
#include "itkPSMProcrustesFunction.h"

namespace itk
{
template <unsigned int VDimension>
class ITK_EXPORT PSMProcrustesRegistration: public DataObject
{
public:
  //  typedef TDataType DataType;
  typedef PSMProcrustesRegistration Self;
  typedef DataObject Superclass;
  typedef SmartPointer<Self>  Pointer;
  typedef SmartPointer<const Self>  ConstPointer;
  typedef WeakPointer<const Self>  ConstWeakPointer;

  typedef PSMParticleSystem<VDimension> PSMParticleSystemType;
  typedef typename PSMParticleSystemType::TransformType TransformType;
  
  typedef PSMProcrustesFunction<VDimension> PSMProcrustesFunctionType;
  typedef typename PSMProcrustesFunctionType::ShapeListType ShapeListType;
  typedef typename PSMProcrustesFunctionType::ShapeType ShapeType;
  typedef typename PSMProcrustesFunctionType::PointType PointType;
  typedef typename PSMProcrustesFunctionType::SimilarityTransformListType SimilarityTransformListType;
  itkNewMacro(Self);
  
  itkTypeMacro(PSMProcrustesRegistration, DataObject);

  void SetPSMParticleSystem(PSMParticleSystemType *p)
  { m_PSMParticleSystem = p;  }
  const PSMParticleSystemType *GetPSMParticleSystem() const
  { return m_PSMParticleSystem; }
  PSMParticleSystemType *GetPSMParticleSystem()
  { return m_PSMParticleSystem; }

  void IterationCallback(itk::Object *, const itk::EventObject &)
  {
    this->RunRegistration();
  }

  void RunRegistration(int i);
  void RunRegistration()
  {
    for (int i = 0; i < m_DomainsPerShape; i++)
      {
      this->RunRegistration(i);
      }
  }

  void SetDomainsPerShape(int i)
  { m_DomainsPerShape = i; }
  int GetDomainsPerShape() const
  { return m_DomainsPerShape; }

  bool GetScaling() const
  { return m_Scaling; }
  void ScalingOn()
  { m_Scaling = true; }
  void ScalingOff()
  { m_Scaling = false; }

  void SetProcrustesInterval(int i)
  { m_Procrustes_Interval = i; }
  int GetProcrustesInterval()
  { return m_Procrustes_Interval; }


protected:
  PSMProcrustesRegistration() : m_Procrustes_Interval(0),
     m_DomainsPerShape(1), m_Scaling(true)
    {  }
  virtual ~PSMProcrustesRegistration() {};

  void PrintSelf(std::ostream& os, Indent indent) const
  {  Superclass::PrintSelf(os,indent);  }

private:
  PSMProcrustesRegistration(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented

  int m_Procrustes_Interval;

  int m_DomainsPerShape;

  bool m_Scaling;
  
  PSMParticleSystemType *m_PSMParticleSystem;
};

} // end namespace

#endif