"""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