Poodletooth-iLand/toontown/pets/PetManager.py

39 lines
994 B
Python

from panda3d.core import *
from toontown.toonbase import ToontownGlobals
from direct.task import Task
def acquirePetManager():
if not hasattr(base, 'petManager'):
PetManager()
base.petManager.incRefCount()
def releasePetManager():
base.petManager.decRefCount()
class PetManager:
CollTaskName = 'petFloorCollisions'
def __init__(self):
base.petManager = self
self.refCount = 0
self.cTrav = CollisionTraverser('petFloorCollisions')
taskMgr.add(self._doCollisions, PetManager.CollTaskName, priority=ToontownGlobals.PetFloorCollPriority)
def _destroy(self):
taskMgr.remove(PetManager.CollTaskName)
del self.cTrav
def _doCollisions(self, task):
self.cTrav.traverse(render)
return Task.cont
def incRefCount(self):
self.refCount += 1
def decRefCount(self):
self.refCount -= 1
if self.refCount == 0:
self._destroy()
del base.petManager