Poodletooth-iLand/toontown/coghq/LawOfficeLayout.py

43 lines
1.4 KiB
Python

import random
from direct.directnotify import DirectNotifyGlobal
from direct.showbase.PythonUtil import invertDictLossless
from direct.showbase.PythonUtil import normalDistrib, lerp
from toontown.coghq import MintRoomSpecs
from toontown.toonbase import ToontownGlobals
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():
LawbotFloorSpecs[floorIndex] = __import__('toontown.coghq.' + 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