55 lines
2.1 KiB
Python
55 lines
2.1 KiB
Python
import random
|
|
|
|
import DistributedLawOfficeAI
|
|
import DistributedStageAI
|
|
from direct.directnotify import DirectNotifyGlobal
|
|
from direct.showbase import DirectObject
|
|
from toontown.coghq import StageLayout
|
|
from toontown.toonbase import ToontownGlobals
|
|
|
|
|
|
StageId2Layouts = {
|
|
ToontownGlobals.LawbotStageIntA: (0, 1, 2),
|
|
ToontownGlobals.LawbotStageIntB: (3, 4, 5),
|
|
ToontownGlobals.LawbotStageIntC: (6, 7, 8),
|
|
ToontownGlobals.LawbotStageIntD: (9, 10, 11)
|
|
}
|
|
|
|
|
|
class LawOfficeManagerAI(DirectObject.DirectObject):
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('LawOfficeManagerAI')
|
|
lawOfficeId = None
|
|
|
|
def __init__(self, air):
|
|
DirectObject.DirectObject.__init__(self)
|
|
self.air = air
|
|
|
|
def getDoId(self):
|
|
return 0
|
|
|
|
def createLawOffice(self, StageId, entranceId, players):
|
|
for avId in players:
|
|
if bboard.has('StageId-%s' % avId):
|
|
StageId = bboard.get('StageId-%s' % avId)
|
|
break
|
|
floor = 0
|
|
layoutIndex = None
|
|
for avId in players:
|
|
if bboard.has('stageRoom-%s' % avId):
|
|
roomId = bboard.get('stageRoom-%s' % avId)
|
|
for lt in StageId2Layouts[StageId]:
|
|
for i in xrange(StageLayout.getNumFloors(lt)):
|
|
layout = StageLayout.StageLayout(StageId, i, stageLayout = lt)
|
|
if roomId in layout.getRoomIds():
|
|
layoutIndex = lt
|
|
floor = i
|
|
else:
|
|
StageRoomSpecs = StageRoomSpecs
|
|
roomName = StageRoomSpecs.CashbotStageRoomId2RoomName[roomId]
|
|
LawOfficeManagerAI.notify.warning('room %s (%s) not found in any floor of Stage %s' % (roomId, roomName, StageId))
|
|
StageZone = self.air.allocateZone()
|
|
if layoutIndex is None:
|
|
layoutIndex = random.choice(StageId2Layouts[StageId])
|
|
Stage = DistributedStageAI.DistributedStageAI(self.air, StageId, StageZone, floor, players, layoutIndex)
|
|
Stage.generateWithRequired(StageZone)
|
|
return StageZone
|