/vol/vipdata/irtk/packages/registration/include/irtkHistogramSimilarityMetric.h

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

  Library   : Image Registration Toolkit (IRTK)
  Module    : $Id: irtkHistogramSimilarityMetric.h 2 2008-12-23 12:40:14Z dr $
  Copyright : Imperial College, Department of Computing
              Visual Information Processing (VIP), 2008 onwards
  Date      : $Date: 2008-12-23 12:40:14 +0000 (Tue, 23 Dec 2008) $
  Version   : $Revision: 2 $
  Changes   : $Author: dr $

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

#ifndef _IRTKHISTOGRAMSIMILARITYMETRIC_H

#define _IRTKHISTOGRAMSIMILARITYMETRIC_H

class irtkHistogramSimilarityMetric : public irtkSimilarityMetric
{

protected:

  irtkHistogram_2D *_histogram;

public:

  irtkHistogramSimilarityMetric(int = 64, int = 64);

  ~irtkHistogramSimilarityMetric();

  virtual void Add(int, int);

  virtual void Delete(int, int);

  virtual void Combine(irtkSimilarityMetric *);

  virtual void Reset();

  virtual void Reset(irtkSimilarityMetric *);

  int NumberOfBinsX();

  int NumberOfBinsY();

};

inline irtkHistogramSimilarityMetric::irtkHistogramSimilarityMetric(int nbins_x, int nbins_y)
{
  _histogram = new irtkHistogram_2D(nbins_x, nbins_y);
}

inline irtkHistogramSimilarityMetric::~irtkHistogramSimilarityMetric()
{
  delete _histogram;
}

inline void irtkHistogramSimilarityMetric::Add(int x, int y)
{
  _histogram->Add(x, y);
}

inline void irtkHistogramSimilarityMetric::Delete(int x, int y)
{
  _histogram->Delete(x, y);
}

inline void irtkHistogramSimilarityMetric::Combine(irtkSimilarityMetric *metric)
{
  int i, j;
  irtkHistogramSimilarityMetric *m = dynamic_cast<irtkHistogramSimilarityMetric *>(metric);

  if (m == NULL) {
    cerr << "irtkHistogramSimilarityMetric::Combine: Dynamic cast failed" << endl;
    exit(1);
  }

  if ((_histogram->GetNumberOfBinsX() != m->_histogram->GetNumberOfBinsX()) ||
      (_histogram->GetNumberOfBinsY() != m->_histogram->GetNumberOfBinsY())) {
    cerr << "irtkHistogramSimilarityMetric::Combine: Number of bins differs" << endl;
    exit(1);
  }
  for (j = 0; j < _histogram->GetNumberOfBinsY(); j++) {
    for (i = 0; i < _histogram->GetNumberOfBinsX(); i++) {
      _histogram->Add(i, j, m->_histogram->irtkHistogram_2D::operator()(i, j));
    }
  }
}

inline void irtkHistogramSimilarityMetric::Reset()
{
  _histogram->Reset();
}

inline void irtkHistogramSimilarityMetric::Reset(irtkSimilarityMetric *metric)
{
  irtkHistogramSimilarityMetric *m = dynamic_cast<irtkHistogramSimilarityMetric *>(metric);

  if (m == NULL) {
    cerr << "irtkHistogramSimilarityMetric::Reset: Dynamic cast failed" << endl;
    exit(1);
  }

  _histogram->Reset(*m->_histogram);
}

inline int irtkHistogramSimilarityMetric::NumberOfBinsX()
{
  return _histogram->GetNumberOfBinsX();
}

inline int irtkHistogramSimilarityMetric::NumberOfBinsY()
{
  return _histogram->GetNumberOfBinsY();
}

#include <irtkGenericHistogramSimilarityMetric.h>
#include <irtkMutualInformationSimilarityMetric.h>
#include <irtkNormalisedMutualInformationSimilarityMetric.h>
#include <irtkJointEntropySimilarityMetric.h>
#include <irtkCorrelationRatioXYSimilarityMetric.h>
#include <irtkCorrelationRatioYXSimilarityMetric.h>
#include <irtkKappaSimilarityMetric.h>
#include <irtkLabelConsistencySimilarityMetric.h>

#endif

Generated on Mon Mar 23 16:55:53 2009 for IRTK by Doxygen 1.5.4