better filtering in getRandomTask

This commit is contained in:
David Rose 2008-10-14 02:18:32 +00:00
parent 02d98e41c7
commit dd0efe66fe

View File

@ -608,37 +608,25 @@ class TaskManager:
return self._profileInfo.session return self._profileInfo.session
def _getRandomTask(self): def _getRandomTask(self):
tasks = self.mgr.getActiveTasks() # Figure out when the next frame is likely to expire, so we
numTasks = tasks.getNumTasks() # won't grab any tasks that are sleeping for a long time.
doLaters = self.mgr.getSleepingTasks() now = globalClock.getFrameTime()
numDoLaters = doLaters.getNumTasks()
if random.random() < (numDoLaters / float(numTasks + numDoLaters)):
# grab a doLater that will most likely trigger in the next frame
tNow = globalClock.getFrameTime()
avgFrameRate = globalClock.getAverageFrameRate() avgFrameRate = globalClock.getAverageFrameRate()
if avgFrameRate < .00001: if avgFrameRate < .00001:
avgFrameDur = 0. avgFrameDur = 0.
else: else:
avgFrameDur = (1. / globalClock.getAverageFrameRate()) avgFrameDur = (1. / globalClock.getAverageFrameRate())
tNext = tNow + avgFrameDur next = now + avgFrameDur
# binary search to find doLaters that are likely to trigger on the next frame
curIndex = int(numDoLaters / 2) # Now grab a task at random, until we find one that we like.
rangeStart = 0 tasks = self.mgr.getTasks()
rangeEnd = numDoLaters i = random.randrange(tasks.getNumTasks())
while True: task = tasks.getTask(i)
if tNext < doLaters[curIndex].wakeTime: while not isinstance(task, PythonTask) or \
rangeEnd = curIndex task.getWakeTime() > next:
else: tasks.removeTask(i)
rangeStart = curIndex i = random.randrange(tasks.getNumTasks())
prevIndex = curIndex task = tasks.getTask(i)
curIndex = int((rangeStart + rangeEnd) / 2)
if curIndex == prevIndex:
break
index = curIndex
task = doLaters[random.randrange(index+1)]
else:
# grab a task
task = tasks[random.randint(0, numTasks - 1)]
return task return task
def __repr__(self): def __repr__(self):