220 lines
7.3 KiB
Python
220 lines
7.3 KiB
Python
|
from direct.directnotify import DirectNotifyGlobal
|
||
|
|
import string
|
||
|
|
import LevelConstants
|
||
|
|
from direct.showbase.PythonUtil import lineInfo, uniqueElements
|
||
|
|
import types
|
||
|
|
|
||
|
|
class Level:
|
||
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('Level')
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
self.levelSpec = None
|
||
|
|
self.initialized = 0
|
||
|
|
return
|
||
|
|
|
||
|
|
def initializeLevel(self, levelId, levelSpec, scenarioIndex):
|
||
|
|
self.levelId = levelId
|
||
|
|
self.levelSpec = levelSpec
|
||
|
|
self.scenarioIndex = scenarioIndex
|
||
|
|
self.levelSpec.setScenario(self.scenarioIndex)
|
||
|
|
self.entranceId2entity = {}
|
||
|
|
self.entId2createCallbacks = {}
|
||
|
|
self.createdEntIds = []
|
||
|
|
self.nonlocalEntIds = {}
|
||
|
|
self.nothingEntIds = {}
|
||
|
|
self.entityCreator = self.createEntityCreator()
|
||
|
|
self.entType2ids = self.levelSpec.getEntType2ids(self.levelSpec.getAllEntIds())
|
||
|
|
for entType in self.entityCreator.getEntityTypes():
|
||
|
|
self.entType2ids.setdefault(entType, [])
|
||
|
|
|
||
|
|
self.createAllEntities(priorityTypes=['levelMgr', 'zone', 'propSpinner'])
|
||
|
|
self.levelMgrEntity = self.getEntity(LevelConstants.LevelMgrEntId)
|
||
|
|
self.uberZoneEntity = self.getEntity(LevelConstants.UberZoneEntId)
|
||
|
|
self.initialized = 1
|
||
|
|
|
||
|
|
def isInitialized(self):
|
||
|
|
return self.initialized
|
||
|
|
|
||
|
|
def getLevelId(self):
|
||
|
|
return self.levelId
|
||
|
|
|
||
|
|
def destroyLevel(self):
|
||
|
|
self.destroyAllEntities()
|
||
|
|
if self.initialized:
|
||
|
|
del self.levelMgrEntity
|
||
|
|
del self.uberZoneEntity
|
||
|
|
del self.entityCreator
|
||
|
|
del self.entId2createCallbacks
|
||
|
|
del self.entranceId2entity
|
||
|
|
self.levelSpec.destroy()
|
||
|
|
del self.levelSpec
|
||
|
|
self.initialized = 0
|
||
|
|
del self.createdEntIds
|
||
|
|
del self.nonlocalEntIds
|
||
|
|
del self.nothingEntIds
|
||
|
|
if hasattr(self, 'entities'):
|
||
|
|
del self.entities
|
||
|
|
if hasattr(self, 'levelSpec'):
|
||
|
|
self.levelSpec.destroy()
|
||
|
|
del self.levelSpec
|
||
|
|
|
||
|
|
def createEntityCreator(self):
|
||
|
|
Level.notify.error('concrete Level class must override %s' % lineInfo()[2])
|
||
|
|
|
||
|
|
def createAllEntities(self, priorityTypes = []):
|
||
|
|
self.entities = {}
|
||
|
|
entTypes = self.entityCreator.getEntityTypes()
|
||
|
|
self.onLevelPreCreate()
|
||
|
|
for type in priorityTypes:
|
||
|
|
self.createAllEntitiesOfType(type)
|
||
|
|
entTypes.remove(type)
|
||
|
|
|
||
|
|
for type in entTypes:
|
||
|
|
self.createAllEntitiesOfType(type)
|
||
|
|
|
||
|
|
self.onLevelPostCreate()
|
||
|
|
|
||
|
|
def destroyAllEntities(self):
|
||
|
|
self.nonlocalEntIds = {}
|
||
|
|
self.nothingEntIds = {}
|
||
|
|
if not uniqueElements(self.createdEntIds):
|
||
|
|
Level.notify.warning('%s: self.createdEntIds is not unique: %s' % (getattr(self, 'doId', None), self.createdEntIds))
|
||
|
|
while len(self.createdEntIds) > 0:
|
||
|
|
entId = self.createdEntIds.pop()
|
||
|
|
entity = self.getEntity(entId)
|
||
|
|
if entity is not None:
|
||
|
|
Level.notify.debug('destroying %s %s' % (self.getEntityType(entId), entId))
|
||
|
|
entity.destroy()
|
||
|
|
else:
|
||
|
|
Level.notify.error('trying to destroy entity %s, but it is already gone' % entId)
|
||
|
|
|
||
|
|
return
|
||
|
|
|
||
|
|
def createAllEntitiesOfType(self, entType):
|
||
|
|
self.onEntityTypePreCreate(entType)
|
||
|
|
for entId in self.entType2ids[entType]:
|
||
|
|
self.createEntity(entId)
|
||
|
|
|
||
|
|
self.onEntityTypePostCreate(entType)
|
||
|
|
|
||
|
|
def createEntity(self, entId):
|
||
|
|
spec = self.levelSpec.getEntitySpec(entId)
|
||
|
|
Level.notify.debug('creating %s %s' % (spec['type'], entId))
|
||
|
|
entity = self.entityCreator.createEntity(entId)
|
||
|
|
announce = False
|
||
|
|
if entity is 'nonlocal':
|
||
|
|
self.nonlocalEntIds[entId] = None
|
||
|
|
elif entity is 'nothing':
|
||
|
|
self.nothingEntIds[entId] = None
|
||
|
|
announce = True
|
||
|
|
else:
|
||
|
|
self.createdEntIds.append(entId)
|
||
|
|
announce = True
|
||
|
|
if announce:
|
||
|
|
self.onEntityCreate(entId)
|
||
|
|
return entity
|
||
|
|
|
||
|
|
def initializeEntity(self, entity):
|
||
|
|
entId = entity.entId
|
||
|
|
spec = self.levelSpec.getEntitySpec(entId)
|
||
|
|
for key, value in spec.items():
|
||
|
|
if key in ('type', 'name', 'comment'):
|
||
|
|
continue
|
||
|
|
entity.setAttribInit(key, value)
|
||
|
|
|
||
|
|
self.entities[entId] = entity
|
||
|
|
|
||
|
|
def getEntity(self, entId):
|
||
|
|
if hasattr(self, 'entities'):
|
||
|
|
return self.entities.get(entId)
|
||
|
|
else:
|
||
|
|
return None
|
||
|
|
return None
|
||
|
|
|
||
|
|
def getEntityType(self, entId):
|
||
|
|
return self.levelSpec.getEntityType(entId)
|
||
|
|
|
||
|
|
def getEntityZoneEntId(self, entId):
|
||
|
|
return self.levelSpec.getEntityZoneEntId(entId)
|
||
|
|
|
||
|
|
def getEntityZoneId(self, entId):
|
||
|
|
zoneEntId = self.getEntityZoneEntId(entId)
|
||
|
|
if not hasattr(self, 'zoneNum2zoneId'):
|
||
|
|
return None
|
||
|
|
return self.zoneNum2zoneId.get(zoneEntId)
|
||
|
|
|
||
|
|
def getZoneId(self, zoneEntId):
|
||
|
|
return self.zoneNum2zoneId[zoneEntId]
|
||
|
|
|
||
|
|
def getZoneNumFromId(self, zoneId):
|
||
|
|
return self.zoneId2zoneNum[zoneId]
|
||
|
|
|
||
|
|
def getParentTokenForEntity(self, entId):
|
||
|
|
return entId
|
||
|
|
|
||
|
|
def getLevelPreCreateEvent(self):
|
||
|
|
return 'levelPreCreate-%s' % self.levelId
|
||
|
|
|
||
|
|
def getLevelPostCreateEvent(self):
|
||
|
|
return 'levelPostCreate-%s' % self.levelId
|
||
|
|
|
||
|
|
def getEntityTypePreCreateEvent(self, entType):
|
||
|
|
return 'entityTypePreCreate-%s-%s' % (self.levelId, entType)
|
||
|
|
|
||
|
|
def getEntityTypePostCreateEvent(self, entType):
|
||
|
|
return 'entityTypePostCreate-%s-%s' % (self.levelId, entType)
|
||
|
|
|
||
|
|
def getEntityCreateEvent(self, entId):
|
||
|
|
return 'entityCreate-%s-%s' % (self.levelId, entId)
|
||
|
|
|
||
|
|
def getEntityOfTypeCreateEvent(self, entType):
|
||
|
|
return 'entityOfTypeCreate-%s-%s' % (self.levelId, entType)
|
||
|
|
|
||
|
|
def onLevelPreCreate(self):
|
||
|
|
messenger.send(self.getLevelPreCreateEvent())
|
||
|
|
|
||
|
|
def onLevelPostCreate(self):
|
||
|
|
messenger.send(self.getLevelPostCreateEvent())
|
||
|
|
|
||
|
|
def onEntityTypePreCreate(self, entType):
|
||
|
|
messenger.send(self.getEntityTypePreCreateEvent(entType))
|
||
|
|
|
||
|
|
def onEntityTypePostCreate(self, entType):
|
||
|
|
messenger.send(self.getEntityTypePostCreateEvent(entType))
|
||
|
|
|
||
|
|
def onEntityCreate(self, entId):
|
||
|
|
messenger.send(self.getEntityCreateEvent(entId))
|
||
|
|
messenger.send(self.getEntityOfTypeCreateEvent(self.getEntityType(entId)), [entId])
|
||
|
|
if entId in self.entId2createCallbacks:
|
||
|
|
for callback in self.entId2createCallbacks[entId]:
|
||
|
|
callback()
|
||
|
|
|
||
|
|
del self.entId2createCallbacks[entId]
|
||
|
|
|
||
|
|
def setEntityCreateCallback(self, entId, callback):
|
||
|
|
ent = self.getEntity(entId)
|
||
|
|
if ent is not None:
|
||
|
|
callNow = True
|
||
|
|
elif entId in self.nothingEntIds:
|
||
|
|
callNow = True
|
||
|
|
else:
|
||
|
|
callNow = False
|
||
|
|
if callNow:
|
||
|
|
callback()
|
||
|
|
else:
|
||
|
|
self.entId2createCallbacks.setdefault(entId, [])
|
||
|
|
self.entId2createCallbacks[entId].append(callback)
|
||
|
|
return
|
||
|
|
|
||
|
|
def getEntityDestroyEvent(self, entId):
|
||
|
|
return 'entityDestroy-%s-%s' % (self.levelId, entId)
|
||
|
|
|
||
|
|
def onEntityDestroy(self, entId):
|
||
|
|
messenger.send(self.getEntityDestroyEvent(entId))
|
||
|
|
del self.entities[entId]
|
||
|
|
if entId in self.createdEntIds:
|
||
|
|
self.createdEntIds.remove(entId)
|
||
|
|
|
||
|
|
def handleVisChange(self):
|
||
|
|
pass
|