kazu
4f511d907e
needed interface changes [new options] logger for android wifi-ap-optimization new test-cases
122 lines
2.5 KiB
C++
122 lines
2.5 KiB
C++
#ifndef DRAWLIST_H
|
|
#define DRAWLIST_H
|
|
|
|
#include <vector>
|
|
|
|
#include "Random.h"
|
|
#include "../Assertions.h"
|
|
|
|
/**
|
|
* add elements of a certain probability
|
|
* and randomly draw from them
|
|
*/
|
|
template <typename T> class DrawList {
|
|
|
|
/** one entry */
|
|
struct Entry {
|
|
|
|
/** the user element */
|
|
T element;
|
|
|
|
/** the cumulative probability up to this element */
|
|
double cumProbability;
|
|
|
|
/** the element's own probability */
|
|
double probability;
|
|
|
|
/** ctor */
|
|
Entry(T element, const double cumProbability, const double probability) : element(element), cumProbability(cumProbability), probability(probability) {;}
|
|
|
|
/** compare for searches */
|
|
bool operator < (const double val) const {return cumProbability < val;}
|
|
|
|
};
|
|
|
|
private:
|
|
|
|
/** current cumulative probability */
|
|
double cumProbability;
|
|
|
|
/** all contained elements */
|
|
std::vector<Entry> elements;
|
|
|
|
/** the used random number generator */
|
|
RandomGenerator& gen;
|
|
|
|
|
|
private:
|
|
|
|
/** default random generator. fallback */
|
|
RandomGenerator defRndGen;
|
|
|
|
|
|
public:
|
|
|
|
/** ctor with random seed */
|
|
DrawList() : cumProbability(0), gen(defRndGen) {
|
|
;
|
|
}
|
|
|
|
/** ctor with custom seed */
|
|
DrawList(const uint32_t seed) : cumProbability(0), gen(defRndGen(seed)) {
|
|
;
|
|
}
|
|
|
|
/** ctor with custom RandomNumberGenerator */
|
|
DrawList(RandomGenerator& gen) : cumProbability(0), gen(gen) {
|
|
;
|
|
}
|
|
|
|
/** change the seed */
|
|
void setSeed(const uint64_t seed) {
|
|
gen.seed(seed);
|
|
}
|
|
|
|
/** reset */
|
|
void reset() {
|
|
cumProbability = 0;
|
|
elements.clear();
|
|
}
|
|
|
|
/** adjust the reserved list size */
|
|
void reserve(const size_t numElements) {
|
|
elements.reserve(numElements);
|
|
}
|
|
|
|
/** add a new user-element and its probability */
|
|
void add(T element, const double probability) {
|
|
Assert::isTrue(probability >= 0, "probability must not be negative!");
|
|
cumProbability += probability;
|
|
elements.push_back(Entry(element, cumProbability, probability));
|
|
}
|
|
|
|
/** get a random element based on its probability */
|
|
T get(double& elemProbability) {
|
|
|
|
// generate random number between [0:cumProbability]
|
|
std::uniform_real_distribution<> dist(0, cumProbability);
|
|
|
|
// get a random value
|
|
const double rndVal = dist(gen);
|
|
|
|
// binary search for the matching entry O(log(n))
|
|
const auto tmp = std::lower_bound(elements.begin(), elements.end(), rndVal);
|
|
|
|
// sanity check
|
|
Assert::isFalse(tmp == elements.end(), "draw() did not find a valid element");
|
|
|
|
// done
|
|
elemProbability = (*tmp).probability;
|
|
return (*tmp).element;
|
|
|
|
}
|
|
|
|
/** get the current, cumulative probability */
|
|
double getCumProbability() const {
|
|
return cumProbability;
|
|
}
|
|
|
|
};
|
|
|
|
#endif // DRAWLIST_H
|