55 lines
2.4 KiB
Python
55 lines
2.4 KiB
Python
from direct.distributed.DistributedObjectGlobalUD import DistributedObjectGlobalUD
|
|
from direct.directnotify.DirectNotifyGlobal import directNotify
|
|
import random
|
|
|
|
class NonRepeatableRandomSourceUD(DistributedObjectGlobalUD):
|
|
notify = directNotify.newCategory('NonRepeatableRandomSourceUD')
|
|
RandomNumberCacheSize = 2000000
|
|
|
|
class Request(ScratchPad):
|
|
|
|
def __init__(self, replyTo, replyToClass, context, num):
|
|
ScratchPad.__init__(self, replyTo=replyTo, replyToClass=replyToClass, context=context, num=num, randoms=[])
|
|
|
|
def __init__(self, air):
|
|
DistributedObjectGlobalUD.__init__(self, air)
|
|
|
|
def announceGenerate(self):
|
|
DistributedObjectGlobalUD.announceGenerate(self)
|
|
self._randoms = []
|
|
self._requests = []
|
|
self._fakeIt = 0
|
|
if __dev__:
|
|
NonRepeatableRandomSourceUD.RandomNumberCacheSize = config.GetInt('random-source-cache-size', 5000)
|
|
self._fakeIt = config.GetBool('fake-non-repeatable-random-source', self._fakeIt)
|
|
|
|
def randomSample(self, nrrsDoId, random):
|
|
self._randoms = [random] + self._randoms
|
|
if len(self._randoms) > self.RandomNumberCacheSize:
|
|
self._randoms.pop()
|
|
self._processRequests()
|
|
self.air.sendUpdateToDoId('NonRepeatableRandomSource', 'randomSampleAck', nrrsDoId, [])
|
|
|
|
def getRandomSamples(self, replyTo, replyToClass, context, num):
|
|
self._requests.append(self.Request(replyTo, replyToClass, context, num))
|
|
self._processRequests()
|
|
|
|
def _processRequests(self):
|
|
while len(self._requests):
|
|
request = self._requests[0]
|
|
if not self._fakeIt and len(request.randoms) < request.num:
|
|
if len(self._randoms) == 0:
|
|
break
|
|
needed = request.num - len(request.randoms)
|
|
if needed > 0:
|
|
numRandoms = min(needed, len(self._randoms))
|
|
if self._fakeIt:
|
|
for i in range(numRandoms):
|
|
request.randoms.append(random.random() * 4294967295)
|
|
|
|
else:
|
|
request.randoms += self._randoms[:numRandoms]
|
|
self._randoms = self._randoms[numRandoms:]
|
|
if request.num == len(request.randoms):
|
|
self._requests.pop(0)
|
|
self.air.dispatchUpdateToDoId(request.replyToClass, 'getRandomSamplesReply', request.replyTo, [request.context, request.randoms])
|