mirror of
https://github.com/Sneed-Group/Poodletooth-iLand
synced 2024-10-31 16:57:54 +00:00
81 lines
3.9 KiB
Python
81 lines
3.9 KiB
Python
from direct.showbase.PythonUtil import invertDict
|
|
from toontown.toonbase import ToontownGlobals
|
|
from toontown.coghq import BossbotCountryClubFairwayRoom_Battle00_Cogs
|
|
from toontown.coghq import BossbotCountryClubMazeRoom_Battle00_Cogs
|
|
from toontown.coghq import BossbotCountryClubMazeRoom_Battle01_Cogs
|
|
from toontown.coghq import BossbotCountryClubMazeRoom_Battle02_Cogs
|
|
from toontown.coghq import BossbotCountryClubMazeRoom_Battle03_Cogs
|
|
from toontown.coghq import NullCogs
|
|
from toontown.coghq import BossbotCountryClubKartRoom_Battle00_Cogs
|
|
from toontown.coghq import BossbotCountryClubPresidentRoom_Battle00_Cogs
|
|
|
|
# Explicit imports...
|
|
from toontown.coghq import BossbotCountryClubEntrance_Action00
|
|
from toontown.coghq import BossbotCountryClubTeeOffRoom_Action00
|
|
from toontown.coghq import BossbotCountryClubFairwayRoom_Battle00
|
|
from toontown.coghq import BossbotCountryClubMazeRoom_Battle00
|
|
from toontown.coghq import BossbotCountryClubMazeRoom_Battle01
|
|
from toontown.coghq import BossbotCountryClubMazeRoom_Battle02
|
|
from toontown.coghq import BossbotCountryClubGreenRoom_Action00
|
|
from toontown.coghq import BossbotCountryClubKartRoom_Battle00
|
|
from toontown.coghq import BossbotCountryClubPresidentRoom_Battle00
|
|
from toontown.coghq import BossbotCountryClubTeeOffRoom_Action01
|
|
from toontown.coghq import BossbotCountryClubTeeOffRoom_Action02
|
|
from toontown.coghq import BossbotCountryClubGreenRoom_Action01
|
|
from toontown.coghq import BossbotCountryClubGreenRoom_Action02
|
|
|
|
def getCountryClubRoomSpecModule(roomId):
|
|
return CashbotMintSpecModules[roomId]
|
|
|
|
|
|
def getCogSpecModule(roomId):
|
|
roomName = BossbotCountryClubRoomId2RoomName[roomId]
|
|
return CogSpecModules.get(roomName, NullCogs)
|
|
|
|
|
|
def getNumBattles(roomId):
|
|
return roomId2numBattles[roomId]
|
|
|
|
|
|
BossbotCountryClubRoomId2RoomName = {0: 'BossbotCountryClubEntrance_Action00',
|
|
2: 'BossbotCountryClubTeeOffRoom_Action00',
|
|
4: 'BossbotCountryClubFairwayRoom_Battle00',
|
|
5: 'BossbotCountryClubMazeRoom_Battle00',
|
|
6: 'BossbotCountryClubMazeRoom_Battle01',
|
|
7: 'BossbotCountryClubMazeRoom_Battle02',
|
|
9: 'BossbotCountryClubGreenRoom_Action00',
|
|
17: 'BossbotCountryClubKartRoom_Battle00',
|
|
18: 'BossbotCountryClubPresidentRoom_Battle00',
|
|
22: 'BossbotCountryClubTeeOffRoom_Action01',
|
|
32: 'BossbotCountryClubTeeOffRoom_Action02',
|
|
29: 'BossbotCountryClubGreenRoom_Action01',
|
|
39: 'BossbotCountryClubGreenRoom_Action02'}
|
|
BossbotCountryClubRoomName2RoomId = invertDict(BossbotCountryClubRoomId2RoomName)
|
|
BossbotCountryClubEntranceIDs = (0,)
|
|
BossbotCountryClubMiddleRoomIDs = (2, 5, 6)
|
|
BossbotCountryClubFinalRoomIDs = (18,)
|
|
BossbotCountryClubConnectorRooms = ('phase_12/models/bossbotHQ/Connector_Tunnel_A', 'phase_12/models/bossbotHQ/Connector_Tunnel_B')
|
|
CashbotMintSpecModules = {}
|
|
for roomName, roomId in BossbotCountryClubRoomName2RoomId.items():
|
|
CashbotMintSpecModules[roomId] = locals()[roomName]
|
|
|
|
CogSpecModules = {'BossbotCountryClubFairwayRoom_Battle00': BossbotCountryClubFairwayRoom_Battle00_Cogs,
|
|
'BossbotCountryClubMazeRoom_Battle00': BossbotCountryClubMazeRoom_Battle00_Cogs,
|
|
'BossbotCountryClubMazeRoom_Battle01': BossbotCountryClubMazeRoom_Battle01_Cogs,
|
|
'BossbotCountryClubMazeRoom_Battle02': BossbotCountryClubMazeRoom_Battle02_Cogs,
|
|
'BossbotCountryClubKartRoom_Battle00': BossbotCountryClubKartRoom_Battle00_Cogs,
|
|
'BossbotCountryClubPresidentRoom_Battle00': BossbotCountryClubPresidentRoom_Battle00_Cogs}
|
|
roomId2numBattles = {}
|
|
for roomName, roomId in BossbotCountryClubRoomName2RoomId.items():
|
|
if roomName not in CogSpecModules:
|
|
roomId2numBattles[roomId] = 0
|
|
else:
|
|
cogSpecModule = CogSpecModules[roomName]
|
|
roomId2numBattles[roomId] = len(cogSpecModule.BattleCells)
|
|
|
|
name2id = BossbotCountryClubRoomName2RoomId
|
|
roomId2numBattles[name2id['BossbotCountryClubTeeOffRoom_Action00']] = 1
|
|
del name2id
|
|
middleRoomId2numBattles = {}
|
|
for roomId in BossbotCountryClubMiddleRoomIDs:
|
|
middleRoomId2numBattles[roomId] = roomId2numBattles[roomId]
|