1 | #ifndef _EMD_H |
---|

2 | #define _EMD_H |
---|

3 | /* |
---|

4 | emd.h |
---|

5 | |
---|

6 | Last update: 3/24/98 (but see below for newer changes) |
---|

7 | |
---|

8 | An implementation of the Earth Movers Distance. |
---|

9 | Based of the solution for the Transportation problem as described in |
---|

10 | "Introduction to Mathematical Programming" by F. S. Hillier and |
---|

11 | G. J. Lieberman, McGraw-Hill, 1990. |
---|

12 | |
---|

13 | Copyright (C) 1998 Yossi Rubner |
---|

14 | Computer Science Department, Stanford University |
---|

15 | E-Mail: rubner@cs.stanford.edu URL: http://robotics.stanford.edu/~rubner/emd/default.htm |
---|

16 | */ |
---|

17 | |
---|

18 | // List of changes since 2020, compared to the original implementation: |
---|

19 | // r1050: Renamed variables that caused problems with g++ 7.3.0 |
---|

20 | // r1062: Global static arrays moved to a function and now allocated dynamically; removed #defined limit of MAX_SIG_SIZE=1000 |
---|

21 | |
---|

22 | |
---|

23 | /* DEFINITIONS */ |
---|

24 | #define MAX_ITERATIONS 500 |
---|

25 | //#define INFINITY 1e20 |
---|

26 | #define EPSILON 1e-6 |
---|

27 | |
---|

28 | /*****************************************************************************/ |
---|

29 | /* feature_t SHOULD BE MODIFIED BY THE USER TO REFLECT THE FEATURE TYPE */ |
---|

30 | //typedef struct { |
---|

31 | // double X,Y,Z; |
---|

32 | //} |
---|

33 | typedef double feature_t; |
---|

34 | /*****************************************************************************/ |
---|

35 | |
---|

36 | |
---|

37 | typedef struct |
---|

38 | { |
---|

39 | int n; /* Number of features in the signature */ |
---|

40 | feature_t *Features; /* Pointer to the features vector */ |
---|

41 | float *Weights; /* Pointer to the weights of the features */ |
---|

42 | } signature_t; |
---|

43 | |
---|

44 | |
---|

45 | typedef struct |
---|

46 | { |
---|

47 | int from; /* Feature number in signature 1 */ |
---|

48 | int to; /* Feature number in signature 2 */ |
---|

49 | float amount; /* Amount of flow from "from" to "to" */ |
---|

50 | } flow_t; |
---|

51 | |
---|

52 | |
---|

53 | |
---|

54 | float emd(signature_t *Signature1, signature_t *Signature2, |
---|

55 | float (*func)(feature_t *, feature_t *), |
---|

56 | flow_t *Flow, int *FlowSize); |
---|

57 | |
---|

58 | #endif |
---|