zxhMetricFFD2Base.h

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

  Program:   ZXH Registration Software
  Author:    Xiahai Zhuang
  Module:    $RCSfle: zxhMetricFFD2Base.h    $
  Language:  C++
  Date:      $Date: From  2008-10 $
  Version:   $Revision: 2.0 $
  revision:
=========================================================================*/
#ifndef zxhMetricFFD2Base_h
#define zxhMetricFFD2Base_h

#include "zxh.h"
#include "zxhMetricFFDBase.h"
#include <time.h>
#include <math.h>
class zxhMetricFFD2Base :   public zxhMetricFFDBase
{
public:
    zxhMetricFFD2Base(void);
    ~zxhMetricFFD2Base(void);

    virtual std::string GetMetricType() {return zxhMetricFFDBase::GetMetricType()+"_FFD2_"; } ;

    virtual zxhMetricFFDBase* Clone( zxhMetricFFDBase * & pRet );

    /* Re-implemented */
    ;
    virtual int GetValueByFFDGrid(zxhlfloat&fValue,int aiCtrPnt[ImageDimensionMax])
    {
        std::cerr<<"error: FFD2 does not use GetValueByFFDGrid\n" ;
        return -1 ;
    };
    virtual int ComputeConstanceForFFDGradient()
    {
        //int num=this->ComputeConstanceForGradient();
        this->InitMetricFFDConstant();
        bool b = this->ComputeAllControlPointDerivativeBySumLocalEstimation();
        if( m_fWeightFFDBendingEnergy>0 )
            b= b& this->AddAllCtrPntOnTargetForceForBendingEnergy() ;
        return (b==true? 1: 0 );
    }; 
    virtual int GetDiffByFFDGrid(float * afPD, // output partial differentiation 
                                    zxhTransformFFDBase * pFFDGrid,
                                    int * aiCtrPnt,
                                    zxhTransformFFDBase * pDifferential )
    {return this->GetDerivativeFFDGrid(afPD,pFFDGrid,aiCtrPnt,pDifferential); } ;
    virtual int GetDerivativeFFDGrid(float * afPD, // output partial differentiation
                                     zxhTransformFFDBase * pFFDGrid,
                                     int * aiCtrPnt,
                                     zxhTransformFFDBase * pDifferential);



    /***   methods for ComputeConstanceForFFDGradient    ***/
protected:

    virtual bool    EvaluateAllPixelOnTargetForceForMetric() = 0;

    virtual bool    ComputeAllControlPointDerivativeBySumLocalEstimation();

    virtual void InitMetricFFDConstant() ;


    /*  variables */
protected:
    zxhImageDataT<float>    m_imgVoxelForceConst[ImageDimensionMax];

};


#endif