acopredictor.h

Go to the documentation of this file.

#pragma once

#include <vector>
#include <random>

#include "hpchain.h"
#include "acosolution.h"
#include "config.h"

class ACOPredictor {
    HPChain dHPChain;
    int dCycles;
    int dNAnts;
    double dAlpha;
    double dBeta;
    double dEvap;
    int dLSFreq;
    int dExchangedAnts;
    int dRandSeed;
    int dNMovElems;
    int dHCount; 
    double *dPheromone; 
    std::vector<std::mt19937> dRandGen; 
    std::uniform_real_distribution<> dRandDist; 
    ACOSolution dBestSol; 
    int dBestContacts; 
    double &pheromone(int i, int d) const;
    double random(int tid = 0);
    std::vector<double> get_heuristics(
            const std::vector<vec3<int>> &possiblePos,
            const std::vector<vec3<int>> &beadVector
        );
    std::vector<double> get_probabilities(int movIndex, std::vector<double> heuristics) const;
    ACOSolution ant_develop_solution(int tid = 0);
    void ant_deposit_pheromone(const std::vector<char> &directions, int nContacts);

    void perform_cycle(std::vector<ACOSolution> &antsSolutions, int *nContacts);

public:
    ACOPredictor(const Config &config);

    ~ACOPredictor();

    struct Results {
        ACOSolution solution; 
        int contacts; 
    };

    struct Results predict();
};

inline
void ACOPredictor::ant_deposit_pheromone(const std::vector<char> &directions, int nContacts){
    for(unsigned i = 0; i < directions.size(); i++){
        pheromone(i, directions[i]) += nContacts / dHCount;
    }
}

inline double &ACOPredictor::pheromone(int i, int d) const {
    return dPheromone[i*5 + d];
}

inline double ACOPredictor::random(int tid) {
    double retval = dRandDist(dRandGen[tid]);
    return retval;
}