NICE_VisLearning/math/distances/emd.h

#ifndef _EMD_H
#define _EMD_H
/*
    emd.h

    Last update: 3/24/98

    An implementation of the Earth Movers Distance.
    Based of the solution for the Transportation problem as described in
    "Introduction to Mathematical Programming" by F. S. Hillier and 
    G. J. Lieberman, McGraw-Hill, 1990.

    Copyright (C) 1998 Yossi Rubner
    Computer Science Department, Stanford University
    E-Mail: rubner@cs.stanford.edu   URL: http://vision.stanford.edu/~rubner
*/

#ifdef __cplusplus
extern "C" {
#endif

/* DEFINITIONS */
#define MAX_SIG_SIZE   100
#define MAX_ITERATIONS 5000
#undef INFINITY
#define INFINITY       1e20
#define EPSILON        1e-6

/*****************************************************************************/
/* feature_t SHOULD BE MODIFIED BY THE USER TO REFLECT THE FEATURE TYPE      */
typedef double* feature_t;
/*****************************************************************************/


typedef struct
{
  int n;                /* Number of features in the signature */
  // refactor-nice.pl: check this substitution
  // old: feature_t *Features;  /* Pointer to the features vector */
  feature_t *Features;  /* Pointer to the features std::vector */
  float *Weights;       /* Pointer to the weights of the features */
} signature_t;


typedef struct
{
  int from;             /* Feature number in signature 1 */
  int to;               /* Feature number in signature 2 */
  float amount;         /* Amount of flow from "from" to "to" */
} flow_t;


float emd(signature_t *Signature1, signature_t *Signature2,
	  float (*func)(feature_t *, feature_t *),
	  flow_t *Flow, int *FlowSize);

#ifdef __cplusplus
} /* extern C */
#endif

#endif