40 lines
994 B
Python
40 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
|