Poodletooth-iLand/toontown/coghq/StageRoomSpecs.py

126 lines
5.2 KiB
Python
Raw Normal View History

2015-11-14 13:28:53 -06:00
from direct.showbase.PythonUtil import invertDict
from toontown.toonbase import ToontownGlobals
from toontown.coghq import NullCogs
from toontown.coghq import LawbotOfficeOilRoom_Battle00_Cogs
from toontown.coghq import LawbotOfficeOilRoom_Battle01_Cogs
from toontown.coghq import LawbotOfficeBoilerRoom_Battle00_Cogs
from toontown.coghq import LawbotOfficeBoilerRoom_Trap00_Cogs
from toontown.coghq import LawbotOfficeLobby_Trap00_Cogs
from toontown.coghq import LawbotOfficeDiamondRoom_Trap00_Cogs
from toontown.coghq import LawbotOfficeDiamondRoom_Battle00_Cogs
from toontown.coghq import LawbotOfficeGearRoom_Battle00_Cogs
# Explicit room imports...
from toontown.coghq import LawbotOfficeEntrance_Action00
from toontown.coghq import LawbotOfficeOilRoom_Battle00
from toontown.coghq import LawbotOfficeOilRoom_Battle01
from toontown.coghq import LawbotOfficeBoilerRoom_Security00
from toontown.coghq import LawbotOfficeBoilerRoom_Battle00
from toontown.coghq import LawbotOfficeGearRoom_Action00
from toontown.coghq import LawbotOfficeLobby_Action00
from toontown.coghq import LawbotOfficeGearRoom_Security00
from toontown.coghq import LawbotOfficeLobby_Trap00
from toontown.coghq import LawbotOfficeDiamondRoom_Security00
from toontown.coghq import LawbotOfficeDiamondRoom_Trap00
from toontown.coghq import LawbotOfficeGearRoom_Platform00
from toontown.coghq import LawbotOfficeLobby_Lights00
from toontown.coghq import LawbotOfficeBoilerRoom_Action01
from toontown.coghq import LawbotOfficeDiamondRoom_Action00
from toontown.coghq import LawbotOfficeDiamondRoom_Action01
from toontown.coghq import LawbotOfficeLobby_Action01
from toontown.coghq import LawbotOfficeDiamondRoom_Battle00
from toontown.coghq import LawbotOfficeGearRoom_Battle00
def getStageRoomSpecModule(roomId):
return CashbotStageSpecModules[roomId]
def getCogSpecModule(roomId):
roomName = CashbotStageRoomId2RoomName[roomId]
return CogSpecModules.get(roomName, NullCogs)
def getNumBattles(roomId):
return roomId2numBattles[roomId]
CashbotStageRoomId2RoomName = {0: 'LawbotOfficeEntrance_Action00',
1: 'LawbotOfficeOilRoom_Battle00',
2: 'LawbotOfficeOilRoom_Battle01',
3: 'LawbotOfficeBoilerRoom_Security00',
4: 'LawbotOfficeBoilerRoom_Battle00',
5: 'LawbotOfficeGearRoom_Action00',
6: 'LawbotOfficeLobby_Action00',
7: 'LawbotOfficeGearRoom_Security00',
8: 'LawbotOfficeLobby_Trap00',
9: 'LawbotOfficeDiamondRoom_Security00',
10: 'LawbotOfficeDiamondRoom_Trap00',
11: 'LawbotOfficeGearRoom_Platform00',
12: 'LawbotOfficeLobby_Lights00',
100: 'LawbotOfficeBoilerRoom_Action01',
101: 'LawbotOfficeDiamondRoom_Action00',
102: 'LawbotOfficeDiamondRoom_Action01',
103: 'LawbotOfficeLobby_Action01',
104: 'LawbotOfficeDiamondRoom_Battle00',
105: 'LawbotOfficeGearRoom_Battle00'}
CashbotStageRoomName2RoomId = invertDict(CashbotStageRoomId2RoomName)
CashbotStageEntranceIDs = (0,)
CashbotStageMiddleRoomIDs = (1,)
CashbotStageFinalRoomIDs = (2,)
CashbotStageConnectorRooms = ('phase_11/models/lawbotHQ/LB_connector_7cubeL2', 'phase_11/models/lawbotHQ/LB_connector_7cubeLR')
CashbotStageSpecModules = {}
for roomName, roomId in CashbotStageRoomName2RoomId.items():
CashbotStageSpecModules[roomId] = locals()[roomName]
CogSpecModules = {'LawbotOfficeOilRoom_Battle00': LawbotOfficeOilRoom_Battle00_Cogs,
'LawbotOfficeOilRoom_Battle01': LawbotOfficeOilRoom_Battle01_Cogs,
'LawbotOfficeBoilerRoom_Battle00': LawbotOfficeBoilerRoom_Battle00_Cogs,
'LawbotOfficeBoilerRoom_Trap00': LawbotOfficeBoilerRoom_Trap00_Cogs,
'LawbotOfficeLobby_Trap00': LawbotOfficeLobby_Trap00_Cogs,
'LawbotOfficeDiamondRoom_Trap00': LawbotOfficeDiamondRoom_Trap00_Cogs,
'LawbotOfficeDiamondRoom_Battle00': LawbotOfficeDiamondRoom_Battle00_Cogs,
'LawbotOfficeGearRoom_Battle00': LawbotOfficeGearRoom_Battle00_Cogs}
roomId2numBattles = {}
for roomName, roomId in CashbotStageRoomName2RoomId.items():
if roomName not in CogSpecModules:
roomId2numBattles[roomId] = 0
else:
cogSpecModule = CogSpecModules[roomName]
roomId2numBattles[roomId] = len(cogSpecModule.BattleCells)
roomId2numCogs = {}
for roomName, roomId in CashbotStageRoomName2RoomId.items():
if roomName not in CogSpecModules:
roomId2numCogs[roomId] = 0
else:
cogSpecModule = CogSpecModules[roomName]
roomId2numCogs[roomId] = len(cogSpecModule.CogData)
roomId2numCogLevels = {}
for roomName, roomId in CashbotStageRoomName2RoomId.items():
if roomName not in CogSpecModules:
roomId2numCogLevels[roomId] = 0
else:
cogSpecModule = CogSpecModules[roomName]
levels = 0
for cogData in cogSpecModule.CogData:
levels += cogData['level']
roomId2numCogLevels[roomId] = levels
roomId2numMeritCogLevels = {}
for roomName, roomId in CashbotStageRoomName2RoomId.items():
if roomName not in CogSpecModules or roomId in (8, 10):
roomId2numMeritCogLevels[roomId] = 0
else:
cogSpecModule = CogSpecModules[roomName]
levels = 0
for cogData in cogSpecModule.CogData:
levels += cogData['level']
roomId2numMeritCogLevels[roomId] = levels
middleRoomId2numBattles = {}
for roomId in CashbotStageMiddleRoomIDs:
middleRoomId2numBattles[roomId] = roomId2numBattles[roomId]