zxhMetricLogProbability.h

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

  Program:   ZXH Registration Software
  Author:    GAO, Gang
  Language:  Visual C++
  OS:        Windows 7
  Date:      $Date: From  2010-08 $
  Version:   $Revision: 2.0 $

  Description: this function requires gsl library
=========================================================================*/

#pragma once
#include "zxh.h"
#include "zxhMetricBase.h"

#define PI                                                  3.1415926;
#define BOUNDARY_RANGE                                      5.0

#define MAX_INTENSITY                                       255
#define MIN_DIST_PARAM_1                                    1
#define MAX_DIST_PARAM_1                                    MAX_INTENSITY
#define MIN_DIST_PARAM_2                                    1
#define MAX_DIST_PARAM_2                                    100

#define GAMMA_DISTRIBUTION                                  0
#define GAUSSIAN_DISTRIBUTION                               1
#define BEAM_THICKNESS_GAUSSIAN_SMOOTHING_2_SIGMA_SQUARED   15.0
#define BOUNDARY_REFLECTION_SIGMA_SQUARED                   1800
#define AIR_ACOUSTIC_IMPEDENCE                              0.0004

class zxhMetricLogProbability : public zxhMetricBase
{
public:
        zxhMetricLogProbability(void){};
        ~zxhMetricLogProbability(void){};
/*
    int GetRefImageCount();
    
    virtual std::string GetMetricType() {return "_LogProbability_"; } ;
    
    virtual bool SetImage(zxhImageData **pImageRef, int nImageRef);
    
    virtual int GetNoOfEchoImages()             {return _nImageRef-1;};
    virtual zxhImageData* GetMRImage()          { return m_ppImageRef[0] ; };

    virtual zxhImageData* GetEchoImage( int index) 
    {
        int d = m_ppImageRef[index+1]->GetDimension();
        return m_ppImageRef[index+1] ; 
    };
    virtual zxhImageData* GetEchoMaskImage( )
    {
        if(m_pMaskRef==0) return 0 ;
        return m_pMaskRef ;
    };
    //todo virtual bool SetImageModelingType()
    virtual zxhImageModelingBase* GetEchoModeling(int i)    { return & m_pModelingRef[i+1] ; } ;
    virtual zxhImageModelingBase* GetMRModeling(void)       { return & m_pModelingRef[0] ; } ;
    virtual bool SetDistanceImage(zxhImageData *pImageDist) { _pImageDist = pImageDist; return true; };
    virtual bool SetNearestNeighborImage(zxhImageData *pImageNeighbor) {_pImageNeighbor = pImageNeighbor; return true; };


    virtual bool SetGradientImages(zxhImageData *x, zxhImageData *y, zxhImageData *z)
    {
        _pImageGradX = x;
        _pImageGradY = y;
        _pImageGradZ = z;

        return true;
    };
    virtual int GetValue(zxhlfloat &fValue);

    virtual bool ComputeLogProbability();

    virtual bool SetLabels(int *labels, int n)
    {
        _numOfRegs = n;
        _regLabels = labels;

        return true;
    };

    virtual bool SetRegionDistanceIndices(int *idx)
    {
        _regDistIndices = idx;

        return true;
    }

    virtual bool SetDistanceParameters(double **p)
    {
        _distParams = p;

        return true;
    }

    virtual bool SetDistanceType(int *distType)
    {
        _distType = distType;

        return true;
    }

    virtual bool SetEchoProbePosition(double **pos)
    {
        _echoProbePos = pos;

        return true;
    }

    virtual bool SetRegionZValue(double *zvalues)
    {
        _regZValues = zvalues;

        return true;
    };

    virtual bool SetBoundaryWeights();


    void DebugOn()
    {
        _debug = true;
    }

    void DebugOff()
    {
        _debug = false;
    }

protected:
    bool CreateGradientImages(zxhImageData *image);

    double N(const double);

protected:
    int _nImageRef;

    zxhlfloat _fLogProb;

    zxhImageData **m_ppImageRef;

    zxhImageData *_pImageDist;
    zxhImageData *_pImageNeighbor;
    zxhImageData *_pImageGradX;
    zxhImageData *_pImageGradY;
    zxhImageData *_pImageGradZ;

    zxhImageData *_pImageProb;


    double _uniformRegProb;

    double **_distParams;
    double **_echoProbePos;
    double **_boundaryWeights;
    double **_gaussianDistCumulativeSum;
    double **_gammaDistCumulativeSum;

    double *_regZValues;
    double *_boundaryCumulativeSum;
    int    *_labelIndices;
    int    *_regDistIndices;
    int    *_regLabels;
    int    *_distType;

    int    _numOfRegs;

    bool   _debug;
        */
};