48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
from direct.showbase.PythonUtil import Functor
|
|
import LevelMgrBase
|
|
|
|
class LevelMgr(LevelMgrBase.LevelMgrBase):
|
|
|
|
def __init__(self, level, entId):
|
|
LevelMgrBase.LevelMgrBase.__init__(self, level, entId)
|
|
self.geom = loader.loadModel(self.modelFilename)
|
|
|
|
if hasattr(self, 'removeNodes'):
|
|
for node in self.removeNodes:
|
|
self.geom.find(node).removeNode()
|
|
|
|
self.zoneNums = []
|
|
self.level.zoneNum2zoneId = {}
|
|
self.level.zoneId2zoneNum = {}
|
|
self.accept(self.level.getEntityOfTypeCreateEvent('zone'), self.handleZoneCreated)
|
|
|
|
def destroy(self):
|
|
del self.level.zoneIds
|
|
del self.level.zoneId2zoneNum
|
|
del self.level.zoneNum2zoneId
|
|
self.geom.removeNode()
|
|
del self.geom
|
|
LevelMgrBase.LevelMgrBase.destroy(self)
|
|
|
|
def handleZoneCreated(self, entId):
|
|
zoneEnt = self.level.getEntity(entId)
|
|
self.zoneNums.append(zoneEnt.entId)
|
|
self.privAssignZoneIds()
|
|
self.accept(self.level.getEntityDestroyEvent(entId), Functor(self.handleZoneDestroy, entId))
|
|
|
|
def handleZoneDestroy(self, entId):
|
|
zoneEnt = self.level.getEntity(entId)
|
|
del self.level.zoneId2zoneNum[self.level.zoneNum2zoneId[zoneEnt.entId]]
|
|
del self.level.zoneNum2zoneId[zoneEnt.entId]
|
|
self.zoneNums.remove(zoneEnt.entId)
|
|
self.privAssignZoneIds()
|
|
|
|
def privAssignZoneIds(self):
|
|
self.zoneNums.sort()
|
|
for i in xrange(len(self.zoneNums)):
|
|
zoneNum = self.zoneNums[i]
|
|
zoneEnt = self.level.getEntity(zoneNum)
|
|
zoneId = self.level.zoneIds[i]
|
|
zoneEnt.setZoneId(zoneId)
|
|
self.level.zoneNum2zoneId[zoneNum] = zoneId
|
|
self.level.zoneId2zoneNum[zoneId] = zoneNum
|