def f(x, y):
x = r_uint(x)
y = r_uint(y)
rnd = Random(x)
rnd.init_by_array([x, y])
rnd.jumpahead(intmask(y))
return float(rnd.genrand32()) + rnd.random()
def f(x, y):
x = r_uint(x)
y = r_uint(y)
rnd = Random(x)
rnd.init_by_array([x, y])
rnd.jumpahead(intmask(y))
return float(rnd.genrand32()) + rnd.random()
def test_jumpahead_badstate():
rnd = Random()
s, j = 4043161618, 2674112291824205302
rnd.init_by_array([s])
rnd.jumpahead(j)
for i in range(500):
r = rnd.random()
assert r <= 1.0, (r, i)
def test_jumpahead_badstate():
rnd = Random()
s, j = 4043161618, 2674112291824205302
rnd.init_by_array([s])
rnd.jumpahead(j)
for i in range(500):
r = rnd.random()
assert r <= 1.0, (r, i)
class rand(object):
def seed(self, seed):
self.r = Random(seed)
def random(self):
return self.r.random()
def randint(self, a, b):
r32 = intmask(self.r.genrand32())
r = a + r32 % (b - a)
return intmask(r)
class W_RandomObject(W_Object):
classdef = ClassDef("Random", W_Object.classdef, filepath=__file__)
def __init__(self, space, klass=None):
W_Object.__init__(self, space, klass)
self.random = Random()
@classdef.singleton_method("allocate")
def method_allocate(self, space):
return W_RandomObject(space, self)
@classdef.method("rand")
def method_rand(self, space):
return space.newfloat(self.random.random())
def rrandom_example():
rnd1 = Random()
print rnd1.genrand32() # get a 32-bit random number
print rnd1.random() # get a float random number x, 0.0 <= x < 1.0
seed = 1546979228
rnd2 = Random(seed) # set a seed
print rnd2.random()
print rnd1.state # you can access the internal state of the generator
# change the internal state to one different from and likely far away from
# the current state, n is a non-negative integer which is used to scramble
# the current state vector.
n = 10
rnd1.jumpahead(n)
rnd1.random()
class W_RandomObject(W_Object):
classdef = ClassDef("Random", W_Object.classdef)
def __init__(self, space, seed=0, klass=None):
W_Object.__init__(self, space, klass)
self.w_seed = None
self.random = Random(abs(seed))
@classdef.setup_class
def setup_class(cls, space, w_cls):
default = space.send(w_cls, "new")
space.set_const(w_cls, "DEFAULT", default)
@classdef.singleton_method("allocate")
def method_allocate(self, space):
return W_RandomObject(space, 0, self)
@classdef.method("initialize")
def method_initialize(self, space, w_seed=None):
self.srand(space, w_seed)
@classdef.method("seed")
def method_seed(self, space):
return self.w_seed
def srand(self, space, seed=None):
previous_seed = self.w_seed
if seed is None:
seed = intmask(self._generate_seed())
else:
seed = Coerce.int(space, seed)
self.w_seed = space.newint(seed)
if previous_seed is None:
value = space.newfloat(self.random.random())
else:
value = previous_seed
self.random = Random(abs(seed))
return value
@classdef.method("rand")
def method_rand(self, space, w_max=None):
if w_max is None:
return space.newfloat(self.random.random())
elif space.is_kind_of(w_max, space.w_float):
return self._rand_float(space, w_max)
elif space.is_kind_of(w_max, space.getclassfor(W_RangeObject)):
return self._rand_range(space, w_max)
else:
return self._rand_int(space, w_max)
@classdef.singleton_method("rand")
def method_singleton_rand(self, space, args_w):
default = space.find_const(self, "DEFAULT")
return space.send(default, "rand", args_w)
def _rand_range(self, space, range):
random = self.random.random()
first = space.send(range, "first")
last = space.send(range, "last")
if space.is_true(space.send(range, "include?", [last])):
last = space.send(last, "+", [space.newint(1)])
diff = space.send(last, "-", [first])
offset = space.send(diff, "*", [space.newfloat(random)])
choice = space.send(offset, "+", [first])
if (not space.is_kind_of(first, space.w_float)
and not space.is_kind_of(last, space.w_float)):
choice = space.send(choice, "to_i")
return choice
def _rand_float(self, space, float):
random = self.random.random()
max = Coerce.float(space, float)
if max <= 0:
raise space.error(space.w_ArgumentError, "invalid argument")
return space.newfloat(random * max)
def _rand_int(self, space, integer):
random = self.random.random()
max = Coerce.int(space, integer)
if max <= 0:
raise space.error(space.w_ArgumentError, "invalid argument")
else:
return space.newint(int(random * max))
def _generate_seed(self):
seed = 0
if os.access('/dev/urandom', os.R_OK):
file = os.open('/dev/urandom', os.R_OK, 0644)
bytes = os.read(file, 4)
os.close(file)
for b in bytes:
seed = seed * 0xff + ord(b)
return seed + int(time.time()) + os.getpid()
class W_RandomObject(W_Object):
classdef = ClassDef("Random", W_Object.classdef, filepath=__file__)
def __init__(self, space, seed=0, klass=None):
W_Object.__init__(self, space, klass)
self.w_seed = None
self.random = Random(abs(seed))
@classdef.setup_class
def setup_class(cls, space, w_cls):
default = space.send(w_cls, "new")
space.set_const(w_cls, "DEFAULT", default)
@classdef.singleton_method("allocate", seed="int")
def method_allocate(self, space, seed=0):
return W_RandomObject(space, seed, self)
@classdef.method("initialize")
def method_initialize(self, space, w_seed=None):
self.srand(space, w_seed)
@classdef.method("seed")
def method_seed(self, space):
return self.w_seed
def srand(self, space, seed=None):
previous_seed = self.w_seed
if seed is None:
seed = intmask(self._generate_seed())
else:
seed = Coerce.int(space, seed)
self.w_seed = space.newint(seed)
if previous_seed is None:
value = space.newfloat(self.random.random())
else:
value = previous_seed
self.random = Random(abs(seed))
return value
@classdef.method("rand")
def method_rand(self, space, w_max=None):
if w_max is None:
return space.newfloat(self.random.random())
elif space.is_kind_of(w_max, space.w_float):
return self._rand_float(space, w_max)
elif space.is_kind_of(w_max, space.getclassfor(W_RangeObject)):
return self._rand_range(space, w_max)
else:
return self._rand_int(space, w_max)
@classdef.singleton_method("rand")
def method_singleton_rand(self, space, args_w):
default = space.find_const(self, "DEFAULT")
return space.send(default, "rand", args_w)
def _rand_range(self, space, range):
random = self.random.random()
first = space.send(range, "first")
last = space.send(range, "last")
if space.is_true(space.send(range, "include?", [last])):
last = space.send(last, "+", [space.newint(1)])
diff = space.send(last, "-", [first])
offset = space.send(diff, "*", [space.newfloat(random)])
choice = space.send(offset, "+", [first])
if (not space.is_kind_of(first, space.w_float) and
not space.is_kind_of(last, space.w_float)):
choice = space.send(choice, "to_i")
return choice
def _rand_float(self, space, float):
random = self.random.random()
max = Coerce.float(space, float)
if max <= 0:
raise space.error(space.w_ArgumentError, "invalid argument")
return space.newfloat(random * max)
def _rand_int(self, space, integer):
random = self.random.random()
max = Coerce.int(space, integer)
if max <= 0:
raise space.error(space.w_ArgumentError, "invalid argument")
else:
return space.newint(int(random * max))
def _generate_seed(self):
seed = 0
if os.access('/dev/urandom', os.R_OK):
file = os.open('/dev/urandom', os.R_OK, 0644)
bytes = os.read(file, 4)
os.close(file)
for b in bytes:
seed = seed * 0xff + ord(b)
return seed + int(time.time()) + os.getpid()
class W_RandomObject(W_Object):
classdef = ClassDef("Random", W_Object.classdef, filepath=__file__)
def __init__(self, space, seed=0, klass=None):
W_Object.__init__(self, space, klass)
self.random = Random(abs(seed))
@classdef.setup_class
def setup_class(cls, space, w_cls):
default = space.send(w_cls, space.newsymbol("new"))
space.set_const(w_cls, "DEFAULT", default)
@classdef.singleton_method("allocate", seed="int")
def method_allocate(self, space, seed=0):
return W_RandomObject(space, seed, self)
@classdef.method("initialize")
def method_initialize(self, space, w_seed=None):
pass
@classdef.method("srand")
def method_srand(self, space, args_w):
self.random = Random()
@classdef.singleton_method("srand")
def method_singleton_srand(self, space, args_w):
default = space.find_const(self, "DEFAULT")
return space.send(default, space.newsymbol("srand"), args_w)
@classdef.method("rand")
def method_rand(self, space, w_max=None):
if w_max is None:
return space.newfloat(self.random.random())
elif space.is_kind_of(w_max, space.w_float):
return self._rand_float(space, w_max)
elif space.is_kind_of(w_max, space.getclassfor(W_RangeObject)):
return self._rand_range(space, w_max)
else:
return self._rand_int(space, w_max)
@classdef.singleton_method("rand")
def method_singleton_rand(self, space, args_w):
default = space.find_const(self, "DEFAULT")
return space.send(default, space.newsymbol("rand"), args_w)
def _rand_range(self, space, range):
random = self.random.random()
first = space.send(range, space.newsymbol("first"))
last = space.send(range, space.newsymbol("last"))
if space.is_true(space.send(range, space.newsymbol("include?"), [last])):
last = space.send(last, space.newsymbol("+"), [space.newint(1)])
diff = space.send(last, space.newsymbol("-"), [first])
offset = space.send(diff, space.newsymbol("*"), [space.newfloat(random)])
choice = space.send(offset, space.newsymbol("+"), [first])
if (not space.is_kind_of(first, space.w_float) and
not space.is_kind_of(last, space.w_float)):
choice = space.send(choice, space.newsymbol("to_i"))
return choice
def _rand_float(self, space, float):
random = self.random.random()
max = Coerce.float(space, float)
if max <= 0:
raise space.error(space.w_ArgumentError, "invalid argument")
return space.newfloat(random * max)
def _rand_int(self, space, integer):
random = self.random.random()
max = Coerce.int(space, integer)
if max <= 0:
raise space.error(space.w_ArgumentError, "invalid argument")
else:
return space.newint(int(random * max))
