blob: 0ab4c068a8d55d0d21aabae73816c8b73fa886b5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
"""
Simple interface to SyncRNG. This file defines a SyncRNG object which can be
used to seed and pull numbers from the RNG.
"""
from __future__ import division
from copy import deepcopy
from warnings import warn as _warn
import syncrng
class SyncRNG(object):
def __init__(self, seed=0):
self.BPF = 32
self.seed = seed
self.state = syncrng.seed(seed)
def randi(self):
tmp = syncrng.rand(self.state)
self.state = tmp[:-1]
return(tmp[-1])
def rand(self):
return self.randi() * 2.3283064365387e-10
def randbelow(self, n):
maxsize = 1<<self.BPF
if n >= maxsize:
_warn("Underlying random generator does not supply \n"
"enough bits to choose from a population range this "
"large.\n")
return int(self.rand() * n)
rem = maxsize % n
limit = (maxsize - rem) / maxsize
r = self.rand()
while r >= limit:
r = self.rand()
return int(r*maxsize) % n
def shuffle(self, x):
y = deepcopy(x)
for i in reversed(range(1, len(y))):
j = self.randbelow(i+1)
y[i], y[j] = y[j], y[i]
return y
|
