Blame Imath/ImathRandom.cpp

Packit 8dc392
///////////////////////////////////////////////////////////////////////////
Packit 8dc392
//
Packit 8dc392
// Copyright (c) 2002-2012, Industrial Light & Magic, a division of Lucas
Packit 8dc392
// Digital Ltd. LLC
Packit 8dc392
// 
Packit 8dc392
// All rights reserved.
Packit 8dc392
// 
Packit 8dc392
// Redistribution and use in source and binary forms, with or without
Packit 8dc392
// modification, are permitted provided that the following conditions are
Packit 8dc392
// met:
Packit 8dc392
// *       Redistributions of source code must retain the above copyright
Packit 8dc392
// notice, this list of conditions and the following disclaimer.
Packit 8dc392
// *       Redistributions in binary form must reproduce the above
Packit 8dc392
// copyright notice, this list of conditions and the following disclaimer
Packit 8dc392
// in the documentation and/or other materials provided with the
Packit 8dc392
// distribution.
Packit 8dc392
// *       Neither the name of Industrial Light & Magic nor the names of
Packit 8dc392
// its contributors may be used to endorse or promote products derived
Packit 8dc392
// from this software without specific prior written permission. 
Packit 8dc392
// 
Packit 8dc392
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
Packit 8dc392
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
Packit 8dc392
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
Packit 8dc392
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
Packit 8dc392
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
Packit 8dc392
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
Packit 8dc392
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
Packit 8dc392
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
Packit 8dc392
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
Packit 8dc392
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
Packit 8dc392
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
Packit 8dc392
//
Packit 8dc392
///////////////////////////////////////////////////////////////////////////
Packit 8dc392
Packit 8dc392
//-----------------------------------------------------------------------------
Packit 8dc392
//
Packit 8dc392
//	Routines that generate pseudo-random numbers compatible
Packit 8dc392
//	with the standard erand48(), nrand48(), etc. functions.
Packit 8dc392
//
Packit 8dc392
//-----------------------------------------------------------------------------
Packit 8dc392
Packit 8dc392
#include "ImathRandom.h"
Packit 8dc392
#include "ImathInt64.h"
Packit 8dc392
Packit 8dc392
IMATH_INTERNAL_NAMESPACE_SOURCE_ENTER
Packit 8dc392
namespace {
Packit 8dc392
Packit 8dc392
//
Packit 8dc392
// Static state used by Imath::drand48(), Imath::lrand48() and Imath::srand48()
Packit 8dc392
//
Packit 8dc392
Packit 8dc392
unsigned short staticState[3] = {0, 0, 0};
Packit 8dc392
Packit 8dc392
Packit 8dc392
void
Packit 8dc392
rand48Next (unsigned short state[3])
Packit 8dc392
{
Packit 8dc392
    //
Packit 8dc392
    // drand48() and friends are all based on a linear congruential
Packit 8dc392
    // sequence,
Packit 8dc392
    //
Packit 8dc392
    //   x[n+1] = (a * x[n] + c) % m,
Packit 8dc392
    // 
Packit 8dc392
    // where a and c are as specified below, and m == (1 << 48)
Packit 8dc392
    //
Packit 8dc392
Packit 8dc392
    static const Int64 a = Int64 (0x5deece66dLL);
Packit 8dc392
    static const Int64 c = Int64 (0xbLL);
Packit 8dc392
Packit 8dc392
    //
Packit 8dc392
    // Assemble the 48-bit value x[n] from the
Packit 8dc392
    // three 16-bit values stored in state.
Packit 8dc392
    //
Packit 8dc392
Packit 8dc392
    Int64 x = (Int64 (state[2]) << 32) |
Packit 8dc392
	      (Int64 (state[1]) << 16) |
Packit 8dc392
	       Int64 (state[0]);
Packit 8dc392
Packit 8dc392
    //
Packit 8dc392
    // Compute x[n+1], except for the "modulo m" part.
Packit 8dc392
    //
Packit 8dc392
Packit 8dc392
    x = a * x + c;
Packit 8dc392
Packit 8dc392
    //
Packit 8dc392
    // Disassemble the 48 least significant bits of x[n+1] into
Packit 8dc392
    // three 16-bit values.  Discard the 16 most significant bits;
Packit 8dc392
    // this takes care of the "modulo m" operation.
Packit 8dc392
    //
Packit 8dc392
    // We assume that sizeof (unsigned short) == 2.
Packit 8dc392
    //
Packit 8dc392
Packit 8dc392
    state[2] = (unsigned short)(x >> 32);
Packit 8dc392
    state[1] = (unsigned short)(x >> 16);
Packit 8dc392
    state[0] = (unsigned short)(x);
Packit 8dc392
}
Packit 8dc392
Packit 8dc392
} // namespace
Packit 8dc392
Packit 8dc392
Packit 8dc392
double
Packit 8dc392
erand48 (unsigned short state[3])
Packit 8dc392
{
Packit 8dc392
    //
Packit 8dc392
    // Generate double-precision floating-point values between 0.0 and 1.0:
Packit 8dc392
    // 
Packit 8dc392
    // The exponent is set to 0x3ff, which indicates a value greater
Packit 8dc392
    // than or equal to 1.0, and less than 2.0.  The 48 most significant
Packit 8dc392
    // bits of the significand (mantissa) are filled with pseudo-random
Packit 8dc392
    // bits generated by rand48Next().  The remaining 4 bits are a copy
Packit 8dc392
    // of the 4 most significant bits of the significand.  This results
Packit 8dc392
    // in bit patterns between 0x3ff0000000000000 and 0x3fffffffffffffff,
Packit 8dc392
    // which correspond to uniformly distributed floating-point values
Packit 8dc392
    // between 1.0 and 1.99999999999999978.  Subtracting 1.0 from those
Packit 8dc392
    // values produces numbers between 0.0 and 0.99999999999999978, that
Packit 8dc392
    // is, between 0.0 and 1.0-DBL_EPSILON.
Packit 8dc392
    // 
Packit 8dc392
Packit 8dc392
    rand48Next (state);
Packit 8dc392
Packit 8dc392
    union {double d; Int64 i;} u;
Packit 8dc392
Packit 8dc392
    u.i = (Int64 (0x3ff)    << 52) |	// sign and exponent
Packit 8dc392
	  (Int64 (state[2]) << 36) |	// significand
Packit 8dc392
	  (Int64 (state[1]) << 20) |
Packit 8dc392
	  (Int64 (state[0]) <<  4) |
Packit 8dc392
	  (Int64 (state[2]) >> 12);
Packit 8dc392
Packit 8dc392
    return u.d - 1;
Packit 8dc392
}
Packit 8dc392
Packit 8dc392
Packit 8dc392
double
Packit 8dc392
drand48 ()
Packit 8dc392
{
Packit 8dc392
    return IMATH_INTERNAL_NAMESPACE::erand48 (staticState);
Packit 8dc392
}
Packit 8dc392
Packit 8dc392
Packit 8dc392
long int
Packit 8dc392
nrand48 (unsigned short state[3])
Packit 8dc392
{
Packit 8dc392
    //
Packit 8dc392
    // Generate uniformly distributed integers between 0 and 0x7fffffff.
Packit 8dc392
    // 
Packit 8dc392
Packit 8dc392
    rand48Next (state);
Packit 8dc392
Packit 8dc392
    return ((long int) (state[2]) << 15) |
Packit 8dc392
	   ((long int) (state[1]) >>  1);
Packit 8dc392
}
Packit 8dc392
Packit 8dc392
Packit 8dc392
long int
Packit 8dc392
lrand48 ()
Packit 8dc392
{
Packit 8dc392
    return IMATH_INTERNAL_NAMESPACE::nrand48 (staticState);
Packit 8dc392
}
Packit 8dc392
Packit 8dc392
Packit 8dc392
void
Packit 8dc392
srand48 (long int seed)
Packit 8dc392
{
Packit 8dc392
    staticState[2] = (unsigned short)(seed >> 16);
Packit 8dc392
    staticState[1] = (unsigned short)(seed);
Packit 8dc392
    staticState[0] = 0x330e;
Packit 8dc392
}
Packit 8dc392
Packit 8dc392
Packit 8dc392
float
Packit 8dc392
Rand32::nextf ()
Packit 8dc392
{
Packit 8dc392
    //
Packit 8dc392
    // Generate single-precision floating-point values between 0.0 and 1.0:
Packit 8dc392
    // 
Packit 8dc392
    // The exponent is set to 0x7f, which indicates a value greater than
Packit 8dc392
    // or equal to 1.0, and less than 2.0.  The 23 bits of the significand
Packit 8dc392
    // (mantissa) are filled with pseudo-random bits generated by
Packit 8dc392
    // Rand32::next().  This results in in bit patterns between 0x3f800000
Packit 8dc392
    // and 0x3fffffff, which correspond to uniformly distributed floating-
Packit 8dc392
    // point values between 1.0 and 1.99999988.  Subtracting 1.0 from
Packit 8dc392
    // those values produces numbers between 0.0 and 0.99999988, that is,
Packit 8dc392
    // between 0.0 and 1.0-FLT_EPSILON.
Packit 8dc392
    // 
Packit 8dc392
Packit 8dc392
    next ();
Packit 8dc392
Packit 8dc392
    union {float f; unsigned int i;} u;
Packit 8dc392
Packit 8dc392
    u.i = 0x3f800000 | (_state & 0x7fffff);
Packit 8dc392
    return u.f - 1;
Packit 8dc392
}
Packit 8dc392
Packit 8dc392
IMATH_INTERNAL_NAMESPACE_SOURCE_EXIT