def generate(self):
self.notify.debug('generate %s: room=%s' % (self.doId, self.roomId))
self.notify.debug('loading spec')
specModule = BoardOfficeRoomSpecs.getBoardOfficeRoomSpecModule(
self.roomId)
roomSpec = LevelSpec.LevelSpec(specModule)
if __dev__:
self.notify.debug('creating entity type registry')
typeReg = self.getBoardOfficeEntityTypeReg()
roomSpec.setEntityTypeReg(typeReg)
self.notify.debug('creating entities')
DistributedLevelAI.DistributedLevelAI.generate(self, roomSpec)
self.notify.debug('creating cogs')
cogSpecModule = BoardOfficeRoomSpecs.getCogSpecModule(self.roomId)
self.planner = LevelSuitPlannerAI.LevelSuitPlannerAI(
self.air,
self,
DistributedBoardOfficeSuitAI.DistributedBoardOfficeSuitAI,
DistributedBoardOfficeBattleAI.DistributedBoardOfficeBattleAI,
cogSpecModule.CogData,
cogSpecModule.ReserveCogData,
cogSpecModule.BattleCells,
battleExpAggreg=self.battleExpAggreg)
suitHandles = self.planner.genSuits()
messenger.send('plannerCreated-' + str(self.doId))
self.suits = suitHandles['activeSuits']
self.reserveSuits = suitHandles['reserveSuits']
self.d_setSuits()
self.notify.debug('finish boardoffice room %s %s creation' %
(self.roomId, self.doId))
def levelAnnounceGenerate(self):
self.notify.debug('levelAnnounceGenerate')
DistributedLevel.DistributedLevel.levelAnnounceGenerate(self)
specModule = BoardOfficeRoomSpecs.getBoardOfficeRoomSpecModule(
self.roomId)
roomSpec = LevelSpec.LevelSpec(specModule)
if __dev__:
typeReg = self.getEntityTypeReg()
roomSpec.setEntityTypeReg(typeReg)
DistributedLevel.DistributedLevel.initializeLevel(self, roomSpec)
def _genFloorLayout(self):
rng = self.getRng()
startingRoomIDs = BoardOfficeRoomSpecs.BoardOfficeEntranceIDs
middleRoomIDs = BoardOfficeRoomSpecs.BoardOfficeMiddleRoomIDs
finalRoomIDs = BoardOfficeRoomSpecs.BoardOfficeFinalRoomIDs
numBattlesLeft = ToontownGlobals.BoardOfficeNumBattles[
self.boardofficeId]
finalRoomId = rng.choice(finalRoomIDs)
numBattlesLeft -= BoardOfficeRoomSpecs.getNumBattles(finalRoomId)
middleRoomIds = []
middleRoomsLeft = self.numRooms - 2
numBattles2middleRoomIds = invertDictLossless(
BoardOfficeRoomSpecs.middleRoomId2numBattles)
allBattleRooms = []
for num, roomIds in numBattles2middleRoomIds.items():
if num > 0:
allBattleRooms.extend(roomIds)
while 1:
allBattleRoomIds = list(allBattleRooms)
rng.shuffle(allBattleRoomIds)
battleRoomIds = self._chooseBattleRooms(numBattlesLeft,
allBattleRoomIds)
if battleRoomIds is not None:
break
BoardOfficeLayout.notify.info('numBattlesLeft = ' +
str(numBattlesLeft) +
' allBattleRoomIds = ' +
str(allBattleRoomIds))
middleRoomIds.extend(battleRoomIds)
middleRoomsLeft -= len(battleRoomIds)
if middleRoomsLeft > 0:
actionRoomIds = numBattles2middleRoomIds[0]
for i in xrange(middleRoomsLeft):
roomId = rng.choice(actionRoomIds)
actionRoomIds.remove(roomId)
middleRoomIds.append(roomId)
roomIds = []
roomIds.append(rng.choice(startingRoomIDs))
rng.shuffle(middleRoomIds)
roomIds.extend(middleRoomIds)
roomIds.append(finalRoomId)
return roomIds
def getBattleCellSpec(self, battleCellId):
cogSpecModule = BoardOfficeRoomSpecs.getCogSpecModule(self.roomId)
return cogSpecModule.BattleCells[battleCellId]
def getReserveCogSpec(self, cogId):
cogSpecModule = BoardOfficeRoomSpecs.getCogSpecModule(self.roomId)
return cogSpecModule.ReserveCogData[cogId]