Kaydet (Commit) e4a3e999 authored tarafından Raymond Hettinger's avatar Raymond Hettinger

In the case where only a user supplied random() method is available,

adopt a strategy that makes the fewest calls to random().
üst 51e01a6f
...@@ -212,33 +212,33 @@ class Random(_random.Random): ...@@ -212,33 +212,33 @@ class Random(_random.Random):
return self.randrange(a, b+1) return self.randrange(a, b+1)
def _randbelow(self, n, int=int, bpf=BPF, type=type, def _randbelow(self, n, int=int, maxsize=1<<BPF, type=type,
Method=_MethodType, BuiltinMethod=_BuiltinMethodType): Method=_MethodType, BuiltinMethod=_BuiltinMethodType):
"""Return a random int in the range [0,n). Raises ValueError if n==0. "Return a random int in the range [0,n). Raises ValueError if n==0."
"""
k = n.bit_length() # don't use (n-1) here because n can be 1
getrandbits = self.getrandbits getrandbits = self.getrandbits
# Only call self.getrandbits if the original random() builtin method # Only call self.getrandbits if the original random() builtin method
# has not been overridden or if a new getrandbits() was supplied. # has not been overridden or if a new getrandbits() was supplied.
if type(self.random) is BuiltinMethod or type(getrandbits) is Method: if type(self.random) is BuiltinMethod or type(getrandbits) is Method:
k = n.bit_length() # don't use (n-1) here because n can be 1
r = getrandbits(k) # 0 <= r < 2**k r = getrandbits(k) # 0 <= r < 2**k
while r >= n: while r >= n:
r = getrandbits(k) r = getrandbits(k)
return r return r
# There's an overriden random() method but no new getrandbits() method, # There's an overriden random() method but no new getrandbits() method,
# so we can only use random() from here. # so we can only use random() from here.
if k > bpf: random = self.random
if n >= maxsize:
_warn("Underlying random() generator does not supply \n" _warn("Underlying random() generator does not supply \n"
"enough bits to choose from a population range this large.\n" "enough bits to choose from a population range this large.\n"
"To remove the range limitation, add a getrandbits() method.") "To remove the range limitation, add a getrandbits() method.")
return int(self.random() * n) return int(random() * n)
random = self.random rem = maxsize % n
N = 1 << k limit = (maxsize - rem) / maxsize # int(limit * maxsize) % n == 0
r = int(N * random()) # 0 <= r < 2**k r = random()
while r >= n: while r >= limit:
r = int(N * random()) r = random()
return r return int(r*maxsize) % n
## -------------------- sequence methods ------------------- ## -------------------- sequence methods -------------------
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment