1 | // This file is a part of Framsticks SDK. http://www.framsticks.com/ |
---|

2 | // Copyright (C) 1999-2020 Maciej Komosinski and Szymon Ulatowski. |
---|

3 | // See LICENSE.txt for details. |
---|

4 | |
---|

5 | #ifndef HUNGARIAN_MEASURE_H |
---|

6 | #define HUNGARIAN_MEASURE_H |
---|

7 | |
---|

8 | #include "measure-mds-based.h" |
---|

9 | #include "frams/genetics/geno.h" |
---|

10 | #include "frams/model/model.h" |
---|

11 | #include "hungarian/hungarian.h" |
---|

12 | |
---|

13 | class SimilMeasureHungarian : public SimilMeasureMDSBased |
---|

14 | { |
---|

15 | public: |
---|

16 | SimilMeasureHungarian(); |
---|

17 | ~SimilMeasureHungarian(){}; |
---|

18 | |
---|

19 | double distanceForTransformation(); |
---|

20 | double distanceWithoutAlignment(); |
---|

21 | |
---|

22 | static int getNOFactors(); |
---|

23 | int setParams(std::vector<double> params); |
---|

24 | |
---|

25 | /// Table of weights for weighted distance function. |
---|

26 | /// Weights are for factors in the following order: |
---|

27 | /// [0]: m_iDV (difference in the number of vertices) |
---|

28 | /// [1]: m_iDD (difference in degrees over matching) |
---|

29 | /// [2]: m_iDN (difference in neurons over matching) |
---|

30 | /// [3]: m_dDG (difference in geometry over matching) |
---|

31 | /// @sa EvaluateDistance |
---|

32 | double m_adFactors[4]; |
---|

33 | |
---|

34 | /// Interface to local parameters |
---|

35 | Param localpar; |
---|

36 | |
---|

37 | protected: |
---|

38 | void prepareData(); |
---|

39 | void beforeTransformation(); |
---|

40 | void copyMatching(); |
---|

41 | void cleanData(); |
---|

42 | |
---|

43 | void countDegrees(); |
---|

44 | void countNeurons(); |
---|

45 | void fillPartsDistances(double*& dist, int bigger, int smaller, bool geo); |
---|

46 | double addNeuronsPartsDiff(double dist); |
---|

47 | |
---|

48 | //Smaller and greater structures sizes |
---|

49 | int nSmaller; |
---|

50 | int nBigger; |
---|

51 | |
---|

52 | //Vertex degrees for both structures |
---|

53 | int *degrees[2]; |
---|

54 | |
---|

55 | //Count of neurons attached to each parts for both structures |
---|

56 | int *neurons[2]; |
---|

57 | |
---|

58 | //Number of on-joint and unattached neurons for both structures |
---|

59 | int on_joint[2]; |
---|

60 | int anywhere[2]; |
---|

61 | |
---|

62 | //Temporary and final assignment |
---|

63 | int *assignment; |
---|

64 | std::vector<int> min_assignment; |
---|

65 | |
---|

66 | //Final and temporary parts distances |
---|

67 | double* parts_distances; |
---|

68 | double* temp_parts_distances; |
---|

69 | |
---|

70 | HungarianAlgorithm hungarian; |
---|

71 | |
---|

72 | /// Number of weights in the function which evaluates distance. |
---|

73 | static const int iNOFactors; |
---|

74 | }; |
---|

75 | |
---|

76 | |
---|

77 | #endif /* HUNGARIAN_MEASURE_H */ |
---|

78 | |
---|