prng.h

Go to the documentation of this file.

#ifndef CVMFS_UTIL_PRNG_H_
#define CVMFS_UTIL_PRNG_H_

#include <stdint.h>
#include <sys/time.h>

#include <limits> // NOLINT

#include <cassert>
#include <cmath>
#include <cstdlib>


#ifdef CVMFS_NAMESPACE_GUARD
namespace CVMFS_NAMESPACE_GUARD {
#endif

class Prng {
 public:
  // Cannot throw an exception
  Prng() throw() {
    state_ = 0;
  }

  void InitSeed(const uint64_t seed) {
    state_ = seed;
  }

  void InitLocaltime() {
    struct timeval tv_now;
    int retval = gettimeofday(&tv_now, NULL);
    assert(retval == 0);
    state_ = tv_now.tv_usec;
  }

  uint32_t Next(const uint64_t boundary) {
    state_ = a*state_ + c;
    double scaled_val =
      static_cast<double>(state_) * static_cast<double>(boundary) /
      static_cast<double>(18446744073709551616.0);
    return static_cast<uint32_t>(static_cast<uint64_t>(scaled_val) % boundary);
  }

  double NextDouble() {
    state_ = a*state_ + c;
    double unit_val = static_cast<double>(state_) /
      static_cast<double>(18446744073709551616.0);
    return unit_val;
  }
  double NextNormal() {
    double z, u1, u2;
    double pi = atan(1) * 4;
    u1 = NextDouble();
    u2 = NextDouble();
    z = sqrt(-2.0 * log(u1)) * cos(2 * pi * u2);
    return z;
  }


 private:
  // Magic numbers from MMIX
  // static const uint64_t m = 2^64;
  static const uint64_t a = 6364136223846793005LLU;
  static const uint64_t c = 1442695040888963407LLU;
  uint64_t state_;
};  // class Prng

#ifdef CVMFS_NAMESPACE_GUARD
}  // namespace CVMFS_NAMESPACE_GUARD
#endif

#endif  // CVMFS_UTIL_PRNG_H_