freScaleTransformController.h

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: freScaleTransformController.h,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.

=========================================================================*/
#ifndef __freScaleTransformController_h
#define __freScaleTransformController_h

#include "freITKTransformControllerBase.h"
#include "freTransformInitializer.h"
#include "freTransformSetupAdaptor.h"

#include "itkScaleTransform.h"

namespace FREE
{

  freControllerIDMacro(ScaleTransformControllerBase, "ScaleTransformBase");
template <int iDimension>
class ScaleTransformControllerBase : public ITKTransformControllerBase< itk::ScaleTransform < ScalarType, iDimension >, iDimension >
{
public:  
  typedef itk::ScaleTransform < ScalarType, iDimension > ComponentType;
  typedef ITKTransformControllerBase<ComponentType, iDimension> Superclass;

  itkTypeMacro(ScaleTransformControllerBase, ITKTransformControllerBase);

protected:
  ScaleTransformControllerBase()
  {
    //Profile settings
    this->UpdateControllerID(ControllerID::ScaleTransformControllerBase);
    this->m_Description = "A rigid scale transform.";
  };

  virtual void 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: scales are set by itv",1,"0");
  };

  virtual bool SetInitialTransformValues( ComponentType* pTransformComponent,
                                          SessionComponentCache* pComponentCache,
                                          SessionInfo* pSessionInfo,
                                                                                                                                                            const int& iInitializeByITV) const
  {
    if (!iInitializeByITV)
    {
      ComponentSetup* pTransformSetup = pComponentCache->Setup();
      TransformSetupAdaptor adaptor(pTransformSetup);
      ReferencePointsAccessor<iDimension> pointsAccessor(pComponentCache);

      if (!pointsAccessor.MovingPointsExist()) throwCtrlExceptionMacro("","Cannot estimate scaling via reference points. No moving reference point set defined. Please check media linkage.");
      if (!pointsAccessor.FixedPointsExist()) throwCtrlExceptionMacro("","Cannot estimate scaling via via reference points. No fixed reference point set defined. Please check media linkage.");

      ParameterArrayType scaling(iDimension);

      scaling.fill(0);
      for (int iIndex = 0; iIndex<pointsAccessor.GetFixedPointsCount(); iIndex++)
      {
        ParameterArrayType movingRef = pointsAccessor.GetMovingPointArray(iIndex);
        ParameterArrayType fixedRef = pointsAccessor.GetFixedPointArray(iIndex);

        for (int iIndex=0; iIndex<iDimension; iIndex++)
        {
          scaling[iIndex] += movingRef[iIndex]/fixedRef[iIndex];
        };
      };

      int iCoupleCount = pointsAccessor.GetFixedPointsCount();
      if (iCoupleCount<1) iCoupleCount = 1;

      for (int iIndex=0; iIndex<iDimension; iIndex++)
      {
        scaling[iIndex] = scaling[iIndex]/iCoupleCount;
        pTransformSetup->Parameters().SetParameterValue(cParam_CurrentTransformValues,scaling[iIndex],iIndex);
      }
    };
    return true;
  };
};

  freControllerIDMacro(Scale2DTransformController, "Scale 2D Transform");
class Scale2DTransformController : public ScaleTransformControllerBase<2>
{
public:  
  itkTypeMacro(Scale2DTransformController, ScaleTransformControllerBase);

  Scale2DTransformController();

protected:
  virtual void GenerateProfile(CtrlProfile::ControllerProfile& profile,
                               const SessionComponentCache* pComponentCache,
                               bool bRegardOldSetup) const;
};

  freControllerIDMacro(Scale3DTransformController, "Scale 3D Transform");
class Scale3DTransformController : public ScaleTransformControllerBase<3>
{
public:  
  itkTypeMacro(Scale3DTransformController, ScaleTransformControllerBase);

  Scale3DTransformController();

protected:
  virtual void GenerateProfile(CtrlProfile::ControllerProfile& profile,
                               const SessionComponentCache* pComponentCache,
                               bool bRegardOldSetup) const;
};

} //end of namespace free

#endif

---