OpenGTA/math/weighted_set.cpp

#include "weighted_set.h"
#include <cassert>
#include <ctime>

namespace Math {
  WeightedSet::WeightedSet(unsigned int seed) :
    elements()
    {
	  std::srand(unsigned(std::time(0)));
    }

  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 = std::rand() % (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;
}*/