mitkMetric.h

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

  Program:   3DMed
  Date:      $Date: 2014-02-25 18:30:00 +0800 $
  Version:   $Version: 4.6.0 $
  Copyright: MIPG, Institute of Automation, Chinese Academy of Sciences

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


#ifndef __mitkMetric_h
#define __mitkMetric_h

#include "mitkProcessObject.h"
#include "mitkTransform.h"
#include "mitkInterpolateFilter.h"
#include "mitkRCPtr.h"

#define INVALID_PIXEL   1e10
#define YITA            1e-10


#ifdef MITK_REGISTRATION_USE_HISTOGRAM
class mitkHistogram;
#endif

class MITK_REGISTRATION_API mitkMetric : public mitkProcessObject
{
public:
    MITK_TYPE(mitkMetric, mitkProcessObject)    
    virtual void PrintSelf(ostream &os);

    void SetFixedVolume(mitkVolume *fixedVolume) {m_FixedVolume = fixedVolume;}

    mitkVolume* GetFixedVolume() {return m_FixedVolume;}

    void SetMovingVolume(mitkVolume *movingVolume) {m_MovingVolume = movingVolume;}

    mitkVolume* GetMovingVolume() {return m_MovingVolume;}

    void SetInterpolator(mitkInterpolateFilter *interpolator) {m_Interpolator = interpolator;}

    mitkInterpolateFilter* GetInterpolator();

    void SetTransform(mitkTransform *transform) {m_Transform = transform;}

    mitkTransform* GetTransform();

    void SetTransformParameters(const VectorParameterType & parameters);

    VectorParameterType & GetTransformParameters();

    virtual void Update();

    ScalarParameterType GetSimilarity() {return m_Similarity;}

    virtual bool GetSimilarity(const mitkMatrixType *matrix, ScalarParameterType &similarity);

    virtual bool GetSimilarity(const VectorParameterType & parameters, ScalarParameterType &similarity);

    virtual bool GetSimilarityAndDerivative(const VectorParameterType & parameters,
                                            ScalarParameterType &       similarity,
                                            VectorParameterType &       derivative);

    mitkInterpolateFilter* GetGradientImageInterpolator();

    void SetComputeImageGradientFlag(bool flag);

    void SetInitialTransformMatrix();

    void SetInitialTransformMatrix(mitkMatrixScalarType* matrix);

    void SetRegion(const VectorIndexType& r);

    const VectorIndexType& GetRegion() const;

    const VectorIndexType& GetFixedImageRegion();

#ifdef MITK_REGISTRATION_USE_HISTOGRAM
    void SetHistogramSize(unsigned int x, unsigned int y);
    void SetLowerBound(ScalarParameterType x, ScalarParameterType y);
    void SetUpperBound(ScalarParameterType x, ScalarParameterType y);
#endif

protected:
    mitkMetric();
    virtual ~mitkMetric();  
    virtual bool Execute();
    virtual void ComputeGradientImage();
#ifdef MITK_REGISTRATION_USE_HISTOGRAM
    virtual void ComputeHistogram(const VectorParameterType & parameters, mitkHistogram* histogram = NULL);
#endif

    mitkRCPtr<mitkTransform>                            m_Transform;
    mitkRCPtr<mitkInterpolateFilter>                    m_Interpolator;
    mitkRCPtr<mitkInterpolateFilter>                    m_GradientImageInterpolator;

    mitkRCPtr<mitkVolume>                               m_FixedVolume;
    mitkRCPtr<mitkVolume>                               m_MovingVolume;
    mitkVolumeIteratorBase*                             m_FixedIterator;

    ScalarParameterType                                 m_Similarity;
    VectorIndexType                                     m_Region;
    int                                                 m_Dimensions[3];

    unsigned int                                        m_ImageDimension;
    unsigned int                                        m_ParameterDimension;
    unsigned long                                       m_NumberOfPixelsCounted;

    bool                                                m_FlagFirstRun;
    bool                                                m_FlagComputeGradient;
    bool                                                m_FlagDefinedRegion;

#ifdef MITK_REGISTRATION_USE_HISTOGRAM
    bool                                                m_FlagComputeHistogram;
    mitkHistogram*                                      m_Histogram;
    VectorIndexType                                     m_HistogramSize;
    VectorParameterType                                 m_HistogramLowerBound;
    VectorParameterType                                 m_HistogramUpperBound;
    double                                              m_UpperBoundIncreaseFactor;
#endif


private:
    mitkMetric(const mitkMetric&);
    void operator = (const mitkMetric&);

};


//#define DEFINED_mitkMetric



#endif

---