third_party/kashmir/randomstream.h

/********************************************************************\
 * randomstream.h -- random number generator                        *
 *                                                                  *
 * Copyright (C) 2009 Kenneth Laskoski                              *
 *                                                                  *
\********************************************************************/
/** @file randomstream.h
    @brief random number generator
    @author Copyright (C) 2008 Kenneth Laskoski

    Use, modification, and distribution are subject
    to the Boost Software License, Version 1.0.  (See accompanying file
    LICENSE_1_0.txt or a copy at <http://www.boost.org/LICENSE_1_0.txt>.)
*/

#ifndef KL_RANDOMSTREAM_H
#define KL_RANDOMSTREAM_H

#include <cstddef>

namespace kashmir { namespace user {

template<class user_impl>
class randomstream
{
 public:
  randomstream<user_impl>() : self(static_cast<user_impl*>(this)) {}

  void 
  read(char* buffer, std::size_t count) { self->read(buffer, count); }

  randomstream<user_impl>& 
  operator>>(char& c) { read(&c, 1); return *this; }

  randomstream<user_impl>& 
  operator>>(signed char& c) { read(reinterpret_cast<char*>(&c), 1); return *this; }

  randomstream<user_impl>& 
  operator>>(unsigned char& c) { read(reinterpret_cast<char*>(&c), 1); return *this; }

  randomstream<user_impl>& 
  operator>>(int& n) { read(reinterpret_cast<char*>(&n), sizeof(int)); return *this; }

  randomstream<user_impl>& 
  operator>>(long& n) { read(reinterpret_cast<char*>(&n), sizeof(long)); return *this; }

  randomstream<user_impl>& 
  operator>>(short& n) { read(reinterpret_cast<char*>(&n), sizeof(short)); return *this; }

  randomstream<user_impl>& 
  operator>>(unsigned int& u) { read(reinterpret_cast<char*>(&u), sizeof(unsigned int)); return *this; }

  randomstream<user_impl>& 
  operator>>(unsigned long& u) { read(reinterpret_cast<char*>(&u), sizeof(unsigned long)); return *this; }

  randomstream<user_impl>& 
  operator>>(unsigned short& u) { read(reinterpret_cast<char*>(&u), sizeof(unsigned short)); return *this; }

  randomstream<user_impl>& 
  operator>>(float& f) { read(reinterpret_cast<char*>(&f), sizeof(float)); return *this; }

  randomstream<user_impl>& 
  operator>>(double& f) { read(reinterpret_cast<char*>(&f), sizeof(double)); return *this; }

  randomstream<user_impl>& 
  operator>>(long double& f) { read(reinterpret_cast<char*>(&f), sizeof(long double)); return *this; }

  randomstream<user_impl>& 
  operator>>(bool& b) { read(reinterpret_cast<char*>(&b), sizeof(bool)); b &= 1; return *this; }

  randomstream<user_impl>& 
  operator>>(void*& p) { read(reinterpret_cast<char*>(&p), sizeof(void*)); return *this; }

 private:
    user_impl* const self;
};

}}

#endif