freFEMRegistrationProcessor.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: freFEMRegistrationProcessor.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 __freFEMRegistrationProcessor_h
#define __freFEMRegistrationProcessor_h

#include "freImageRegistrationProcessorBase.h"

#include "fstream"

#include "freControllerCentral.h"
#include "freRegistrationObserver.h"

#include "itkFEM.h"
#include "itkFEMRegistrationFilter.h"
#include "itkHistogramMatchingImageFilter.h"

namespace FREE
{

template <unsigned int VImageDimension>
class FEMRegistrationProcessor
            : public ImageRegistrationProcessorBase<VImageDimension>
{
public:
    typedef FEMRegistrationProcessor<VImageDimension> Self;
    typedef ImageRegistrationProcessorBase<VImageDimension> Superclass;
    typedef itk::SmartPointer<Self>        Pointer;
    typedef itk::SmartPointer<const Self>  ConstPointer;

    typedef typename Superclass::ImageType              ImageType;
    typedef typename Superclass::ImagePointer           ImagePointer;
    typedef typename Superclass::RegionType             RegionType;
    typedef typename Superclass::SpacingType            SpacingType;
    typedef typename Superclass::PointType              PointType;
    typedef typename Superclass::TransformFieldType     TransformFieldType;
    typedef typename Superclass::TransformFieldPointer  TransformFieldPointer;

    itkTypeMacro( FEMRegistrationProcessor, ImageRegistrationProcessorBase );

    itkNewMacro(Self);


    typedef itk::ImageToImageMetric<  ImageType, ImageType > MetricType;
    typedef typename MetricType::Pointer MetricPointer;
    typedef itk::fem::FEMRegistrationFilter< ImageType, ImageType > RegistrationType;
    typedef typename RegistrationType::Pointer RegistrationPointer;
    typedef itk::HistogramMatchingImageFilter<ImageType, ImageType > HistoMatchFilterType;

    typedef itk::fem::Element2DC0LinearQuadrilateralMembrane   Element2DType;
    typedef itk::fem::Element2DC0LinearTriangularMembrane      Element2DType2;

    typedef itk::fem::Element3DC0LinearHexahedronMembrane   Element3DType;
    typedef itk::fem::Element3DC0LinearTetrahedronMembrane  Element3DType2;

    typedef itk::fem::FiniteDifferenceFunctionLoad<ImageType,ImageType>     ImageLoadType;
    //template class itk::fem::ImageMetricLoadImplementation<ImageLoadType>;

    typedef Element2DType::LoadImplementationFunctionPointer     LoadImp2DFP;
    typedef Element2DType::LoadType                              Element2DLoadType;

    typedef Element2DType2::LoadImplementationFunctionPointer    LoadImp2DFP2;
    typedef Element2DType2::LoadType                             Element2DLoadType2;

    typedef itk::fem::VisitorDispatcher<Element2DType,Element2DLoadType, LoadImp2DFP>
    Dispatcher2DType;

    typedef itk::fem::VisitorDispatcher<Element2DType2,Element2DLoadType2, LoadImp2DFP2>
    Dispatcher2DType2;

    typedef Element3DType::LoadImplementationFunctionPointer     LoadImp3DFP;
    typedef Element3DType::LoadType                              Element3DLoadType;

    typedef Element3DType2::LoadImplementationFunctionPointer    LoadImp3DFP2;
    typedef Element3DType2::LoadType                             Element3DLoadType2;

    typedef itk::fem::VisitorDispatcher<Element3DType,Element3DLoadType, LoadImp3DFP>
    Dispatcher3DType;

    typedef itk::fem::VisitorDispatcher<Element3DType2,Element3DLoadType2, LoadImp3DFP2>
    Dispatcher3DType2;

    MetricPointer smpMetric;

    RegistrationPointer smpRegistration;

    IterationObserver::Pointer m_IterationObserver;
    typename NotificationEvent<Self>::Pointer m_IterationNotificationEvent;

    virtual const int GetResolutionLevelCount() const;

    virtual const long GetMaxIterationCount();

protected:
    FEMRegistrationProcessor();

    virtual ~FEMRegistrationProcessor();

    virtual void ComputeRegistration();

    void SetRegistrationParameters();

    void SaveLandmarksToFile(const std::string& sFileName);

    void SetFEMElements();

    virtual TransformFieldPointer ComputeTransformationField(const PointType& origin,
                                                             const RegionType& region,
                                                             const SpacingType& spacing);

    void OnNewIteration(void* pSender, long threadID) ;
};

typedef FEMRegistrationProcessor<2> FEMRegistrationProcessor2D;
typedef FEMRegistrationProcessor<3> FEMRegistrationProcessor3D;

}//End of Namespace free

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

#endif

---