mirror of
https://github.com/Sneed-Group/Poodletooth-iLand
synced 2024-12-26 13:12:30 -06:00
257 lines
10 KiB
Python
257 lines
10 KiB
Python
"""RelatedObjectMgr module: contains the RelatedObjectMgr class"""
|
|
|
|
# from direct.showbase.ShowBaseGlobal import *
|
|
from direct.showbase import DirectObject
|
|
from direct.directnotify import DirectNotifyGlobal
|
|
|
|
class RelatedObjectMgr(DirectObject.DirectObject):
|
|
"""
|
|
This class manages a relationship between DistributedObjects that
|
|
know about each other, and are expected to be generated together.
|
|
Ideally, we should be able to guarantee the ordering of the
|
|
generate calls, but there are certain cases in which the objects
|
|
may not be generated in the correct order as defined by the
|
|
toon.dc file.
|
|
|
|
To handle cases like these robustly, it is necessary for each
|
|
object to deal with the possibility that its companion object has
|
|
not yet been generated. This may mean deferring some operations
|
|
until the expected companion object has been generated.
|
|
|
|
This class helps manage that process. To use it, an object should
|
|
register its desire to be associated with the other object's doId.
|
|
When the other object is generated (or immediately, if the object
|
|
already exists), the associated callback will be called. There is
|
|
also a timeout callback in case the object never appears.
|
|
"""
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('RelatedObjectMgr')
|
|
|
|
doLaterSequence = 1
|
|
|
|
def __init__(self, cr):
|
|
self.cr = cr
|
|
self.pendingObjects = {}
|
|
|
|
def destroy(self):
|
|
self.abortAllRequests()
|
|
del self.cr
|
|
del self.pendingObjects
|
|
|
|
def requestObjects(self, doIdList, allCallback = None, eachCallback = None,
|
|
timeout = None, timeoutCallback = None):
|
|
"""
|
|
Requests a callback to be called when the objects in the
|
|
doIdList are generated. The allCallback will be called only
|
|
when all the objects have been generated (and it receives a
|
|
list of objects, in the order given in doIdList). The
|
|
eachCallback is called as each object is generated, and
|
|
receives only the object itself.
|
|
|
|
If the objects already exist, the appropriate callback is
|
|
called immediately.
|
|
|
|
If all of the objects are not generated within the indicated
|
|
timeout time, the timeoutCallback is called instead, with the
|
|
original doIdList as the parameter. If the timeoutCallback is
|
|
None, then allCallback is called on timeout, with the list of
|
|
objects that have been generated so far, and None for objects
|
|
that have not been generated.
|
|
|
|
If any element of doIdList is None or 0, it is ignored, and
|
|
None is passed in its place in the object list passed to the
|
|
callback.
|
|
|
|
The return value may be saved and passed to a future call to
|
|
abortRequest(), in order to abort a pending request before the
|
|
timeout expires.
|
|
|
|
Actually, you should be careful to call abortRequest() if you
|
|
have made a call to requestObjects() that has not been resolved.
|
|
To find examples, do a search for abortRequest() to find out
|
|
how other code is using it. A common idiom is to store the
|
|
result from requestObjects() and call abortRequest() if delete()
|
|
or destroy() is called on the requesting object.
|
|
|
|
See Also: abortRequest()
|
|
"""
|
|
assert self.notify.debug("requestObjects(%s, timeout=%s)" % (doIdList, timeout))
|
|
|
|
# First, see if we have all of the objects already.
|
|
objects, doIdsPending = self.__generateObjectList(doIdList)
|
|
|
|
# Call the eachCallback immediately on any objects we already
|
|
# have.
|
|
if eachCallback:
|
|
for object in objects:
|
|
if object:
|
|
eachCallback(object)
|
|
|
|
if len(doIdsPending) == 0:
|
|
# All the objects exist, so just call the callback
|
|
# immediately.
|
|
assert self.notify.debug("All objects already exist.")
|
|
if allCallback:
|
|
allCallback(objects)
|
|
return
|
|
|
|
# Some objects don't exist yet, so start listening for them, and
|
|
# also set a timeout in case they don't come.
|
|
assert self.notify.debug("Some objects pending: %s" % (doIdsPending))
|
|
|
|
# Make a copy of the original doIdList, so we can save it over
|
|
# a period of time without worrying about the caller modifying
|
|
# it.
|
|
doIdList = doIdList[:]
|
|
|
|
doLaterName = None
|
|
if timeout != None:
|
|
doLaterName = "RelatedObject-%s" % (RelatedObjectMgr.doLaterSequence)
|
|
assert self.notify.debug("doLaterName = %s" % (doLaterName))
|
|
|
|
RelatedObjectMgr.doLaterSequence += 1
|
|
|
|
tuple = (allCallback, eachCallback, timeoutCallback,
|
|
doIdsPending, doIdList, doLaterName)
|
|
|
|
for doId in doIdsPending:
|
|
pendingList = self.pendingObjects.get(doId)
|
|
if pendingList == None:
|
|
pendingList = []
|
|
self.pendingObjects[doId] = pendingList
|
|
self.__listenFor(doId)
|
|
|
|
pendingList.append(tuple)
|
|
|
|
if doLaterName:
|
|
# Now spawn a do-later to catch the timeout.
|
|
taskMgr.doMethodLater(timeout, self.__timeoutExpired, doLaterName,
|
|
extraArgs = [tuple])
|
|
|
|
return tuple
|
|
|
|
def abortRequest(self, tuple):
|
|
"""
|
|
Aborts a previous request. The parameter is the return value
|
|
from a previous call to requestObjects(). The pending request
|
|
is removed from the queue and no further callbacks will be called.
|
|
|
|
See Also: requestObjects()
|
|
"""
|
|
if tuple:
|
|
allCallback, eachCallback, timeoutCallback, doIdsPending, doIdList, doLaterName = tuple
|
|
assert self.notify.debug("aborting request for %s (remaining: %s)" % (doIdList, doIdsPending))
|
|
|
|
if doLaterName:
|
|
taskMgr.remove(doLaterName)
|
|
self.__removePending(tuple, doIdsPending)
|
|
|
|
def abortAllRequests(self):
|
|
"""
|
|
Call this method to abruptly abort all pending requests, but
|
|
leave the RelatedObjectMgr in a state for accepting more
|
|
requests.
|
|
"""
|
|
|
|
# Stop listening for all events.
|
|
self.ignoreAll()
|
|
|
|
# Iterate through all the pendingObjects and stop any pending
|
|
# tasks.
|
|
for pendingList in self.pendingObjects.values():
|
|
for tuple in pendingList:
|
|
allCallback, eachCallback, timeoutCallback, doIdsPending, doIdList, doLaterName = tuple
|
|
if doLaterName:
|
|
taskMgr.remove(doLaterName)
|
|
|
|
self.pendingObjects = {}
|
|
|
|
|
|
def __timeoutExpired(self, tuple):
|
|
allCallback, eachCallback, timeoutCallback, doIdsPending, doIdList, doLaterName = tuple
|
|
assert self.notify.debug("timeout expired for %s (remaining: %s)" % (doIdList, doIdsPending))
|
|
|
|
self.__removePending(tuple, doIdsPending)
|
|
|
|
if timeoutCallback:
|
|
timeoutCallback(doIdList)
|
|
else:
|
|
objects, doIdsPending = self.__generateObjectList(doIdList)
|
|
if allCallback:
|
|
allCallback(objects)
|
|
|
|
def __removePending(self, tuple, doIdsPending):
|
|
# Removes all the pending events for the doIdsPending list.
|
|
while len(doIdsPending) > 0:
|
|
# We pop doId's off the list instead of simply iterating
|
|
# through the list, so that we will shorten the list (and
|
|
# all other outstanding instances of the list) as we go.
|
|
doId = doIdsPending.pop()
|
|
pendingList = self.pendingObjects[doId]
|
|
pendingList.remove(tuple)
|
|
if len(pendingList) == 0:
|
|
del self.pendingObjects[doId]
|
|
self.__noListenFor(doId)
|
|
|
|
|
|
def __listenFor(self, doId):
|
|
# Start listening for the indicated object to be generated.
|
|
assert self.notify.debug("Now listening for generate from %s" % (doId))
|
|
announceGenerateName = "generate-%s" % (doId)
|
|
self.acceptOnce(announceGenerateName, self.__generated)
|
|
|
|
def __noListenFor(self, doId):
|
|
# Stop listening for the indicated object to be generated.
|
|
assert self.notify.debug("No longer listening for generate from %s" % (doId))
|
|
announceGenerateName = "generate-%s" % (doId)
|
|
self.ignore(announceGenerateName)
|
|
|
|
def __generated(self, object):
|
|
# The indicated object has been generated.
|
|
doId = object.doId
|
|
assert self.notify.debug("Got generate from %s" % (doId))
|
|
pendingList = self.pendingObjects[doId]
|
|
del self.pendingObjects[doId]
|
|
|
|
for tuple in pendingList:
|
|
allCallback, eachCallback, timeoutCallback, doIdsPending, doIdList, doLaterName = tuple
|
|
|
|
# Here we are depending on Python to unify this one list
|
|
# across all objects that share it. When we remove our
|
|
# doId from our reference to the list, it is also removed
|
|
# from all the other references.
|
|
doIdsPending.remove(doId)
|
|
|
|
if eachCallback:
|
|
eachCallback(object)
|
|
|
|
if len(doIdsPending) == 0:
|
|
# That was the last doId on the list. Call the
|
|
# allCallback!
|
|
assert self.notify.debug("All objects generated on list: %s" % (doIdList,))
|
|
if doLaterName:
|
|
taskMgr.remove(doLaterName)
|
|
|
|
objects, doIdsPending = self.__generateObjectList(doIdList)
|
|
if None in objects:
|
|
assert self.notify.warning('calling %s with None.\n objects=%s\n doIdsPending=%s\n doIdList=%s\n' % (allCallback,objects,doIdsPending,doIdList))
|
|
if allCallback:
|
|
allCallback(objects)
|
|
|
|
else:
|
|
assert self.notify.debug("Objects still pending: %s" % (doIdsPending))
|
|
|
|
def __generateObjectList(self, doIdList):
|
|
objects = []
|
|
doIdsPending = []
|
|
|
|
for doId in doIdList:
|
|
if doId:
|
|
object = self.cr.doId2do.get(doId)
|
|
objects.append(object)
|
|
if object == None:
|
|
doIdsPending.append(doId)
|
|
else:
|
|
objects.append(None)
|
|
|
|
return objects, doIdsPending
|
|
|