source: cpp/frams/model/similarity/measure-distribution.h @ 1120

Last change on this file since 1120 was 1120, checked in by Maciej Komosinski, 9 days ago

Used a local random number generator for full determinism. Introduced a few refactorings for better performance.

File size: 1.8 KB
Line 
1// This file is a part of Framsticks SDK.  http://www.framsticks.com/
2// Copyright (C) 1999-2021  Maciej Komosinski and Szymon Ulatowski.
3// See LICENSE.txt for details.
4
5#ifndef DISTRIBUTION_MEASURE_H
6#define DISTRIBUTION_MEASURE_H
7
8#include "simil-measure-base.h"
9#include "EMD/emd.h"
10#include <random>
11
12class SimilMeasureDistribution : public SimilMeasureBase
13{
14public:
15        SimilMeasureDistribution();
16        ~SimilMeasureDistribution() {};
17        int setParams(std::vector<double> params);
18        void calculateFuns();
19        void calculateFun(std::pair<double, float> *fun, const Model &sampled);
20        double compareFuns();
21        double EMD(std::pair<double, float> *fun1, std::pair<double, float> *fun2);
22        double density;
23        int samples_num;
24        int bin_num;
25        /// Interface to local parameters
26        Param localpar;
27
28protected:
29        std::pair<double, float> *funs[2];
30        SolidsShapeTypeModel *sst_models[2];
31        void (SimilMeasureDistribution::*distribution_fun)(int samples_taken, std::uniform_int_distribution<> &distribution, const Model &sampled, std::vector<double> &dist_vect);
32
33        void fillPointsWeights(std::pair<double, float> *fun, feature_t *points, float *weights);
34        double getDistance();
35        void D2(int samples_taken, std::uniform_int_distribution<> &distribution, const Model &sampled, std::vector<double> &dist_vect);
36        void D1(int samples_taken, std::uniform_int_distribution<> &distribution, const Model &sampled, std::vector<double> &dist_vect);
37        void D3(int samples_taken, std::uniform_int_distribution<> &distribution, const Model &sampled, std::vector<double> &dist_vect);
38        void D4(int samples_taken, std::uniform_int_distribution<> &distribution, const Model &sampled, std::vector<double> &dist_vect);
39        void A3(int samples_taken, std::uniform_int_distribution<> &distribution, const Model &sampled, std::vector<double> &dist_vect);
40
41private:
42        std::mt19937 rndgen;
43};
44
45#endif /* DISTRIBUTION_MEASURE_H */
46
Note: See TracBrowser for help on using the repository browser.