Source code for direct.distributed.DoHierarchy

from direct.directnotify.DirectNotifyGlobal import directNotify

[docs]class DoHierarchy: """ This table has been a source of memory leaks, with DoIds getting left in the table indefinitely. DoHierarchy guards access to the table and ensures correctness. """ notify = directNotify.newCategory("DoHierarchy")
[docs] def __init__(self): # parentId->zoneId->set(child DoIds) self._table = {} self._allDoIds = set()
[docs] def isEmpty(self): assert ((len(self._table) == 0) == (len(self._allDoIds) == 0)) return len(self._table) == 0 and len(self._allDoIds) == 0
def __len__(self): return len(self._allDoIds)
[docs] def clear(self): assert self.notify.debugCall() self._table = {} self._allDoIds = set()
[docs] def getDoIds(self, getDo, parentId, zoneId=None, classType=None): """ Args: parentId: any distributed object id. zoneId: a uint32, defaults to None (all zones). Try zone 2 if you're not sure which zone to use (0 is a bad/null zone and 1 has had reserved use in the past as a no messages zone, while 2 has traditionally been a global, uber, misc stuff zone). dclassType: a distributed class type filter, defaults to None (no filter). If dclassName is None then all objects in the zone are returned; otherwise the list is filtered to only include objects of that type. """ parent=self._table.get(parentId) if parent is None: return [] if zoneId is None: r = [] for zone in parent.values(): for obj in zone: r.append(obj) else: r = parent.get(zoneId, []) if classType is not None: a = [] for doId in r: obj = getDo(doId) if isinstance(obj, classType): a.append(doId) r = a return r
[docs] def storeObjectLocation(self, do, parentId, zoneId): doId = do.doId if doId in self._allDoIds: self.notify.error( 'storeObjectLocation(%s %s) already in _allDoIds; duplicate generate()? or didn\'t clean up previous instance of DO?' % ( do.__class__.__name__, do.doId)) parentZoneDict = self._table.setdefault(parentId, {}) zoneDoSet = parentZoneDict.setdefault(zoneId, set()) zoneDoSet.add(doId) self._allDoIds.add(doId) self.notify.debug('storeObjectLocation: %s(%s) @ (%s, %s)' % ( do.__class__.__name__, doId, parentId, zoneId))
[docs] def deleteObjectLocation(self, do, parentId, zoneId): doId = do.doId if doId not in self._allDoIds: self.notify.error( 'deleteObjectLocation(%s %s) not in _allDoIds; duplicate delete()? or invalid previous location on a new object?' % ( do.__class__.__name__, do.doId)) # jbutler: temp hack to get by the assert, this will be fixed soon if (doId not in self._allDoIds): return parentZoneDict = self._table.get(parentId) if parentZoneDict is not None: zoneDoSet = parentZoneDict.get(zoneId) if zoneDoSet is not None: if doId in zoneDoSet: zoneDoSet.remove(doId) self._allDoIds.remove(doId) self.notify.debug('deleteObjectLocation: %s(%s) @ (%s, %s)' % ( do.__class__.__name__, doId, parentId, zoneId)) if len(zoneDoSet) == 0: del parentZoneDict[zoneId] if len(parentZoneDict) == 0: del self._table[parentId] else: self.notify.error( "deleteObjectLocation: objId: %s not found" % doId) else: self.notify.error( "deleteObjectLocation: zoneId: %s not found" % zoneId) else: self.notify.error( "deleteObjectLocation: parentId: %s not found" % parentId)