#include "weighted_set.h"

namespace Math {
  WeightedSet::WeightedSet(unsigned int seed) :
    elements(),
    rng(seed) {
    }

  void WeightedSet::add(unsigned int w, unsigned int e) {
    while (w > 0) {
      elements.push_back(e);
      --w;
    }
  }

  unsigned int WeightedSet::getRandom() {
    //unsigned int rnd = (int) (totalWeight * (rand() / (RAND_MAX + 1.0)));
    unsigned int rnd = rng.nextUint(elements.size());
    return elements[rnd];
  }
}

/*
#include <iostream>
int main(int argc, char* argv[]) {
  Util::Math::WeightedSet set(int(getpid()) ^ int(time(0)));
  set.add(1, 1);
  set.add(2, 2);
  set.add(2, 3);
  set.add(3, 4);
  set.add(3, 5);

  std::cout << set.getRandom() << std::endl;
}*/