39 lines
1.4 KiB
Python
39 lines
1.4 KiB
Python
|
from direct.directnotify import DirectNotifyGlobal
|
||
|
|
from direct.showbase.PythonUtil import invertDictLossless
|
||
|
|
from toontown.coghq import MintRoomSpecs
|
||
|
|
from toontown.toonbase import ToontownGlobals
|
||
|
|
from direct.showbase.PythonUtil import normalDistrib, lerp
|
||
|
|
import random
|
||
|
|
OfficeBuildingFloorSequences = {
|
||
|
|
13300: [
|
||
|
|
(0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0)]}
|
||
|
|
Index2Spec = {
|
||
|
|
0: 'LawOffice_Spec_Tier0_a',
|
||
|
|
1: 'LawOffice_Spec_Tier0_b'}
|
||
|
|
LawbotFloorSpecs = {}
|
||
|
|
for (floorIndex, floorSpec) in Index2Spec.items():
|
||
|
|
exec 'from toontown.coghq import %s' % floorSpec
|
||
|
|
LawbotFloorSpecs[floorIndex] = eval(floorSpec)
|
||
|
|
|
||
|
|
|
||
|
|
class LawOfficeLayout:
|
||
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('MintLayout')
|
||
|
|
|
||
|
|
def __init__(self, lawOfficeId):
|
||
|
|
self.lawOfficeId = lawOfficeId
|
||
|
|
self.floorIds = []
|
||
|
|
if self.lawOfficeId in OfficeBuildingFloorSequences:
|
||
|
|
self.floorIds = OfficeBuildingFloorSequences[self.lawOfficeId][random.randint(0, len(OfficeBuildingFloorSequences[self.lawOfficeId])) - 1]
|
||
|
|
else:
|
||
|
|
self.notify.warning('no layout for Law Office ID: using defaults')
|
||
|
|
self.floorIds = (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0)
|
||
|
|
|
||
|
|
def getFloorId(self, index):
|
||
|
|
return self.floorIds[index]
|
||
|
|
|
||
|
|
def getFloorSpec(self, index):
|
||
|
|
return LawbotFloorSpecs[self.floorIds[index]]
|
||
|
|
|
||
|
|
def getFloorIds(self):
|
||
|
|
return self.floorIds
|