itkFFDConjugateGradientDescentOptimizer.h

Go to the documentation of this file.

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

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). All rights reserved.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

  See LICENSE.txt in the top level directory for details.

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

#ifndef itkFFDConjugateGradientDescentOptimizer_h
#define itkFFDConjugateGradientDescentOptimizer_h


#include "itkFFDGradientDescentOptimizer.h"
#include <itkImageToImageMetricWithConstraint.h>
#include <itkUCLBSplineTransform.h>
#include <itkRegistrationForceFilter.h>
#include <itkBSplineSmoothVectorFieldFilter.h>
#include <itkInterpolateVectorFieldFilter.h>
#include <itkScaleVectorFieldFilter.h>

namespace itk
{
  
template <class TFixedImage, class TMovingImage, class TScalarType, class TDeformationScalar>
class ITK_EXPORT FFDConjugateGradientDescentOptimizer : 
    public FFDGradientDescentOptimizer<TFixedImage, TMovingImage, TScalarType, TDeformationScalar>
{
public:
  
  typedef FFDConjugateGradientDescentOptimizer                         Self;
  typedef FFDGradientDescentOptimizer<TFixedImage, TMovingImage, 
                                      TScalarType, TDeformationScalar> Superclass;
  typedef SmartPointer<Self>                                           Pointer;
  typedef SmartPointer<const Self>                                     ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro( FFDConjugateGradientDescentOptimizer, FFDGradientDescentOptimizer );

  itkStaticConstMacro(Dimension, unsigned int, TFixedImage::ImageDimension);

  typedef typename Superclass::SimilarityMeasureType                   SimilarityMeasureType;
  typedef typename SimilarityMeasureType::TransformParametersType      ParametersType;
  typedef typename Superclass::UCLBSplineTransformPointer              UCLBSplineTransformPointer;
  typedef typename Superclass::GridImagePointer                        GridImagePointer;
  typedef typename Superclass::OutputImageSizeType                     OutputImageSizeType;
  
protected:
  
  FFDConjugateGradientDescentOptimizer(); 
  virtual ~FFDConjugateGradientDescentOptimizer() {};
  void PrintSelf(std::ostream& os, Indent indent) const;

  virtual void OptimizeNextStep(int iterationNumber, int numberOfGridVoxels, const ParametersType& current, ParametersType& next);
  
private:

  FFDConjugateGradientDescentOptimizer(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented

  virtual void Initialize();
  
  virtual void StoreGradient(const ParametersType& gradient);
  
  void GetNextGradient(const ParametersType& currentGradient, ParametersType& nextGradient);
  
  virtual void CleanUp();

  struct float3{
    float x,y,z;
  };  
  
  float3 *conjugateG;
  float3 *conjugateH;
  unsigned long int m_NumberOfGridVoxels;
  ParametersType m_DerivativeAtCurrentPosition;
  ParametersType m_DerivativeAtNextPosition;
};

} // namespace itk.

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkFFDConjugateGradientDescentOptimizer.txx"
#endif

#endif /*ITKFFDCONJUGATEGRADIENTDESCENTOPTIMIZER_H_*/