source: cpp/frams/model/similarity/measure-hungarian.h @ 1243

// This file is a part of Framsticks SDK.  http://www.framsticks.com/
// Copyright (C) 1999-2020  Maciej Komosinski and Szymon Ulatowski.
// See LICENSE.txt for details.

#ifndef HUNGARIAN_MEASURE_H
#define HUNGARIAN_MEASURE_H

#include "measure-mds-based.h"
#include "frams/genetics/geno.h"
#include "frams/model/model.h"
#include "hungarian/hungarian.h"

class SimilMeasureHungarian : public SimilMeasureMDSBased
{
public:
        SimilMeasureHungarian();
        ~SimilMeasureHungarian(){};
        
        double distanceForTransformation();
        double distanceWithoutAlignment();
        
        static int getNOFactors();
        int setParams(std::vector<double> params);

        /// Table of weights for weighted distance function.
        /// Weights are for factors in the following order:
        /// [0]: m_iDV (difference in the number of vertices)
        /// [1]: m_iDD (difference in degrees over matching)
        /// [2]: m_iDN (difference in neurons over matching)
        /// [3]: m_dDG (difference in geometry over matching)
        /// @sa EvaluateDistance
        double m_adFactors[4];
        
        /// Interface to local parameters
        Param localpar;

protected:
        void prepareData();
        void beforeTransformation();
        void copyMatching();
        void cleanData();
        
        void countDegrees();
        void countNeurons();
        void fillPartsDistances(double*& dist, int bigger, int smaller, bool geo);
        double addNeuronsPartsDiff(double dist);
        
        //Smaller and greater structures sizes
        int nSmaller;
        int nBigger;
        
        //Vertex degrees for both structures
        int *degrees[2];
        
        //Count of neurons attached to each parts for both structures
        int *neurons[2];
        
        //Number of on-joint and unattached neurons for both structures
        int on_joint[2];
        int anywhere[2];
        
        //Temporary and final assignment
        int *assignment;
        std::vector<int> min_assignment;
        
        //Final and temporary parts distances
        double* parts_distances;
        double* temp_parts_distances;
        
        HungarianAlgorithm hungarian;
        
        /// Number of weights in the function which evaluates distance.
        static const int iNOFactors;
};


#endif /* HUNGARIAN_MEASURE_H */