freSimilarity2DTransformController.cxx

Go to the documentation of this file.

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

  Program:   F.R.E.E. - flexible registration evaluation engine
  Version:   v.1.0.0
  Date:      $Date: 2006/09/01 12:00:00 $
  Module:    $RCSfile: freSimilarity2DTransformController.cxx,v $
  Language:  C++



  Copyright (c) 2007 Ralf o Floca (Department of Medical Informatics,
  Institute for Medical Biometry and Informatics, University of Heidelberg,
  Germany). All rights reserved.
  See FREECopyright.txt or http://www.mi.med.uni-hd.de/free/copyright.htm
  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.

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

#include "freSimilarity2DTransformController.h"
#include "freTransformSetupAdaptor.h"

#include "itkPoint.h"
#include "itkCenteredTransformInitializer.h"

namespace FREE
{



Similarity2DTransformController::
Similarity2DTransformController()
{
  this->UpdateControllerID(ControllerID::Similarity2DTransformController);
  this->m_Description = "A combination of rotation, translation and homogeneous scaling in 2D.";
};

void
Similarity2DTransformController::
GenerateProfile(CtrlProfile::ControllerProfile& profile,
                const SessionComponentCache* pComponentCache,
                bool bRegardOldSetup) const
{
  Superclass::GenerateProfile(profile,pComponentCache,bRegardOldSetup);

  //Parameters
  profile.Parameters().AddParameter(cParam_InitialisationMethod,Parameter::PVTInteger,std::string(cParamDsc_InitialisationMethod)+"\n1:estimate rotation by initial reference points\n2: estimate scaling by initial reference points\n3: estimate translation by initial reference points",1,"0");
  profile.Parameters().AddParameter(cParam_InitialTransformValues,CtrlProfile::Parameter::PVTDouble,"1: rotation angel (in radian)\n2: scaling factor\n3: x translation\n4: y translation",4,"0");
  profile.Parameters().AddParameter(cParam_CurrentTransformValues,CtrlProfile::Parameter::PVTDouble,"1: rotation angel (in radian)\n2: scaling factor\n3: x translation\n4: y translation",4,"0");
  profile.Parameters().AddParameter(cParam_TransformScale,CtrlProfile::Parameter::PVTDouble,"1: rotation angel (in radian)\n2: scaling factor\n3: x translation\n4: y translation",4,"1",-1,true);
};

bool
Similarity2DTransformController::
SetInitialTransformValues( ComponentType* pTransformComponent,
                           SessionComponentCache* pComponentCache,
                           SessionInfo* pSessionInfo,
                           const int& iInitializeByITV) const
{
  if (!pComponentCache->SetupIsAssigned()) throwCtrlExceptionMacro("","Cannot initialize transform values, transform setup is not assigned in cache.");
  ComponentSetup* pTransformSetup = pComponentCache->Setup();

  TransformSetupAdaptor adaptor(pTransformSetup);
  ReferencePointsAccessor<2> pointsAccessor(pComponentCache);

        //Calculate the translation
  if (iInitializeByITV==3)
  {
    if (!pointsAccessor.MovingPointsExist()) throwCtrlExceptionMacro("","Cannot estimate rotation via reference points. No moving reference point set defined. Please check media linkage.");
    if (!pointsAccessor.FixedPointsExist()) throwCtrlExceptionMacro("","Cannot estimate rotation via via reference points. No fixed reference point set defined. Please check media linkage.");

    ParameterArrayType translation(2);
    for (int iRefIndex=0; iRefIndex<pointsAccessor.GetFixedPointsCount(); iRefIndex++)
    {
      ParameterArrayType movingPoint = pointsAccessor.GetMovingPointArray(iRefIndex);
      ParameterArrayType fixedPoint = pointsAccessor.GetFixedPointArray(iRefIndex);

      for (int iIndex=0; iIndex<2; iIndex++)
      {
        translation[iIndex] += movingPoint[iIndex]-fixedPoint[iIndex];
      };
    };

    for (int iIndex=0; iIndex<2; iIndex++)
    {
      translation[iIndex] = translation[iIndex]/pointsAccessor.GetFixedPointsCount();
      pTransformSetup->Parameters().SetParameterValue(cParam_CurrentTransformValues,translation[iIndex],3+iIndex);
    };

  }
  else if (iInitializeByITV==2)
  {
  }
  else if (iInitializeByITV==1)
  {
    if (!pointsAccessor.MovingPointsExist()) throwCtrlExceptionMacro("","Cannot estimate rotation via reference points. No moving reference point set defined. Please check media linkage.");
    if (!pointsAccessor.FixedPointsExist()) throwCtrlExceptionMacro("","Cannot estimate rotation via via reference points. No fixed reference point set defined. Please check media linkage.");

    RotationInitializer<2> rotationInitializer;
    ParameterArrayType center(2);
    center.Fill(0.0);
          
          rotationInitializer.SetCenterPair(center,center);

          //Add all references
          for (int iIndex = 0; iIndex<pointsAccessor.GetFixedPointsCount(); iIndex++)
          {
                rotationInitializer.AddReferencePair(pointsAccessor.GetMovingPointArray(iIndex),pointsAccessor.GetFixedPointArray(iIndex));
          };

          //set schedule
          rotationInitializer.ClearSchedule(false);
          rotationInitializer.AddToSchedule(RA_Z);

          rotationInitializer.ComputeRotations();

          //add angel to the parameters. Angel must be inverted, because itk uses back transform
          //(fixed to moving) and the initial calculation uses forward transform to allow an intuitive
          //parameter setting
          pTransformSetup->Parameters().SetParameterValue(cParam_CurrentTransformValues,-1*rotationInitializer.GetRotation(RA_Z),0);
        };
 
  return true;
};

} //end of namespace free

---