Sneed-Group/Poodletooth-iLand
1
0
Fork 0
mirror of https://github.com/Sneed-Group/Poodletooth-iLand synced 2024-12-23 03:35:12 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions
Poodletooth-iLand/toontown/coghq/MintLayout.py

848 lines
17 KiB
Python
Raw Permalink Blame History

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
def printAllCashbotInfo():
print 'roomId: roomName'
for roomId, roomName in MintRoomSpecs.CashbotMintRoomId2RoomName.items():
print '%s: %s' % (roomId, roomName)
print '\nroomId: numBattles'
for roomId, numBattles in MintRoomSpecs.roomId2numBattles.items():
print '%s: %s' % (roomId, numBattles)
print '\nmintId floor roomIds'
printMintRoomIds()
print '\nmintId floor numRooms'
printNumRooms()
print '\nmintId floor numForcedBattles'
printNumBattles()
def iterateCashbotMints(func):
from toontown.toonbase import ToontownGlobals
for mintId in [ToontownGlobals.CashbotMintIntA, ToontownGlobals.CashbotMintIntB, ToontownGlobals.CashbotMintIntC]:
for floorNum in xrange(ToontownGlobals.MintNumFloors[mintId]):
func(MintLayout(mintId, floorNum))
def printMintInfo():
def func(ml):
print ml
iterateCashbotMints(func)
def printMintRoomIds():
def func(ml):
print ml.getMintId(), ml.getFloorNum(), ml.getRoomIds()
iterateCashbotMints(func)
def printMintRoomNames():
def func(ml):
print ml.getMintId(), ml.getFloorNum(), ml.getRoomNames()
iterateCashbotMints(func)
def printNumRooms():
def func(ml):
print ml.getMintId(), ml.getFloorNum(), ml.getNumRooms()
iterateCashbotMints(func)
def printNumBattles():
def func(ml):
print ml.getMintId(), ml.getFloorNum(), ml.getNumBattles()
iterateCashbotMints(func)
BakedFloorLayouts = {12500: {0: (0,
4,
9,
6,
5,
8,
17),
1: (0,
15,
13,
16,
7,
6,
22),
2: (0,
4,
11,
3,
9,
6,
14,
19),
3: (0,
1,
3,
4,
16,
14,
15,
24),
4: (0,
15,
5,
8,
9,
11,
10,
21),
5: (0,
13,
12,
8,
7,
16,
10,
18),
6: (0,
16,
13,
5,
12,
7,
1,
23),
7: (0,
10,
12,
7,
3,
13,
16,
8,
20),
8: (0,
3,
5,
7,
6,
1,
4,
9,
25),
9: (0,
6,
9,
10,
13,
16,
8,
4,
22),
10: (0,
13,
1,
7,
2,
16,
11,
3,
19),
11: (0,
3,
1,
6,
4,
14,
8,
9,
24),
12: (0,
7,
14,
2,
1,
8,
5,
10,
11,
21),
13: (0,
13,
6,
4,
11,
3,
9,
10,
8,
17),
14: (0,
15,
5,
1,
14,
10,
4,
7,
16,
23),
15: (0,
16,
10,
11,
2,
1,
3,
14,
5,
20),
16: (0,
5,
8,
10,
6,
3,
15,
14,
7,
25),
17: (0,
12,
13,
5,
8,
14,
11,
7,
16,
10,
22),
18: (0,
11,
3,
15,
7,
16,
14,
6,
1,
5,
18),
19: (0,
10,
16,
11,
3,
5,
12,
13,
7,
14,
24)},
12600: {0: (0,
8,
1,
6,
14,
2,
5,
9,
17),
1: (0,
4,
14,
7,
2,
13,
8,
9,
18),
2: (0,
7,
9,
6,
5,
14,
12,
3,
20),
3: (0,
6,
2,
13,
16,
7,
5,
3,
9,
22),
4: (0,
15,
4,
9,
8,
6,
13,
5,
11,
23),
5: (0,
13,
7,
14,
15,
11,
3,
2,
8,
25),
6: (0,
5,
14,
2,
11,
7,
16,
10,
15,
18),
7: (0,
10,
9,
5,
4,
2,
7,
13,
11,
19),
8: (0,
11,
4,
12,
6,
1,
13,
7,
3,
21),
9: (0,
15,
16,
5,
13,
9,
14,
4,
6,
3,
23),
10: (0,
16,
15,
7,
6,
8,
3,
4,
9,
10,
24),
11: (0,
5,
8,
4,
12,
13,
9,
11,
16,
3,
17),
12: (0,
13,
16,
7,
4,
12,
3,
6,
5,
1,
19),
13: (0,
14,
6,
12,
13,
7,
10,
3,
16,
9,
20),
14: (0,
9,
15,
13,
5,
6,
3,
14,
11,
4,
22),
15: (0,
13,
14,
3,
12,
16,
11,
9,
4,
5,
7,
24),
16: (0,
3,
6,
1,
7,
5,
10,
9,
4,
13,
15,
25),
17: (0,
3,
6,
14,
4,
13,
16,
12,
8,
5,
7,
18),
18: (0,
11,
13,
4,
1,
15,
6,
3,
8,
9,
16,
20),
19: (0,
11,
5,
8,
7,
2,
6,
13,
3,
14,
9,
21)},
12700: {0: (0,
16,
14,
6,
1,
5,
9,
2,
15,
8,
17),
1: (0,
3,
2,
12,
14,
8,
13,
6,
10,
7,
23),
2: (0,
15,
9,
5,
12,
7,
4,
11,
14,
16,
21),
3: (0,
2,
13,
7,
6,
8,
15,
4,
1,
11,
19),
4: (0,
12,
7,
4,
6,
10,
14,
13,
16,
15,
11,
17),
5: (0,
10,
2,
9,
13,
4,
8,
1,
15,
14,
11,
23),
6: (0,
2,
14,
4,
10,
16,
15,
1,
3,
8,
6,
21),
7: (0,
14,
11,
1,
7,
9,
10,
12,
8,
5,
2,
19),
8: (0,
9,
11,
8,
5,
1,
4,
3,
7,
15,
2,
17),
9: (0,
2,
9,
7,
11,
16,
10,
15,
3,
8,
6,
23),
10: (0,
4,
10,
6,
8,
7,
15,
2,
1,
3,
13,
21),
11: (0,
10,
14,
8,
6,
9,
15,
5,
1,
2,
13,
19),
12: (0,
16,
5,
12,
10,
6,
9,
11,
3,
15,
13,
17),
13: (0,
1,
3,
6,
14,
4,
10,
12,
15,
13,
16,
24),
14: (0,
8,
7,
14,
9,
1,
2,
6,
16,
10,
15,
13,
21),
15: (0,
4,
1,
8,
11,
12,
3,
10,
16,
13,
6,
15,
19),
16: (0,
6,
3,
10,
4,
1,
2,
13,
11,
5,
15,
16,
17),
17: (0,
6,
16,
5,
12,
11,
1,
8,
14,
15,
9,
10,
24),
18: (0,
15,
8,
12,
10,
1,
7,
11,
9,
16,
4,
5,
21),
19: (0,
10,
2,
16,
5,
6,
11,
13,
7,
12,
1,
3,
19)}}
class MintLayout:
notify = DirectNotifyGlobal.directNotify.newCategory('MintLayout')
def __init__(self, mintId, floorNum):
self.mintId = mintId
self.floorNum = floorNum
self.roomIds = []
self.hallways = []
self.numRooms = 1 + ToontownGlobals.MintNumRooms[self.mintId][self.floorNum]
self.numHallways = self.numRooms - 1
if self.mintId in BakedFloorLayouts and self.floorNum in BakedFloorLayouts[self.mintId]:
self.roomIds = list(BakedFloorLayouts[self.mintId][self.floorNum])
else:
self.roomIds = self._genFloorLayout()
hallwayRng = self.getRng()
connectorRoomNames = MintRoomSpecs.CashbotMintConnectorRooms
for i in xrange(self.numHallways):
self.hallways.append(hallwayRng.choice(connectorRoomNames))
def _genFloorLayout(self):
rng = self.getRng()
startingRoomIDs = MintRoomSpecs.CashbotMintEntranceIDs
middleRoomIDs = MintRoomSpecs.CashbotMintMiddleRoomIDs
finalRoomIDs = MintRoomSpecs.CashbotMintFinalRoomIDs
numBattlesLeft = ToontownGlobals.MintNumBattles[self.mintId]
finalRoomId = rng.choice(finalRoomIDs)
numBattlesLeft -= MintRoomSpecs.getNumBattles(finalRoomId)
middleRoomIds = []
middleRoomsLeft = self.numRooms - 2
numBattles2middleRoomIds = invertDictLossless(MintRoomSpecs.middleRoomId2numBattles)
allBattleRooms = []
for num, roomIds in numBattles2middleRoomIds.items():
if num > 0:
allBattleRooms.extend(roomIds)
while 1:
allBattleRoomIds = list(allBattleRooms)
rng.shuffle(allBattleRoomIds)
battleRoomIds = self._chooseBattleRooms(numBattlesLeft,
allBattleRoomIds)
if battleRoomIds is not None:
break
MintLayout.notify.info('could not find a valid set of battle rooms, trying again')
middleRoomIds.extend(battleRoomIds)
middleRoomsLeft -= len(battleRoomIds)
if middleRoomsLeft > 0:
actionRoomIds = numBattles2middleRoomIds[0]
for i in xrange(middleRoomsLeft):
roomId = rng.choice(actionRoomIds)
actionRoomIds.remove(roomId)
middleRoomIds.append(roomId)
roomIds = []
roomIds.append(rng.choice(startingRoomIDs))
rng.shuffle(middleRoomIds)
roomIds.extend(middleRoomIds)
roomIds.append(finalRoomId)
return roomIds
def getNumRooms(self):
return len(self.roomIds)
def getRoomId(self, n):
return self.roomIds[n]
def getRoomIds(self):
return self.roomIds[:]
def getRoomNames(self):
names = []
for roomId in self.roomIds:
names.append(MintRoomSpecs.CashbotMintRoomId2RoomName[roomId])
return names
def getNumHallways(self):
return len(self.hallways)
def getHallwayModel(self, n):
return self.hallways[n]
def getNumBattles(self):
numBattles = 0
for roomId in self.getRoomIds():
numBattles += MintRoomSpecs.roomId2numBattles[roomId]
return numBattles
def getMintId(self):
return self.mintId
def getFloorNum(self):
return self.floorNum
def getRng(self):
return random.Random(self.mintId * self.floorNum)
def _chooseBattleRooms(self, numBattlesLeft, allBattleRoomIds, baseIndex = 0, chosenBattleRooms = None):
if chosenBattleRooms is None:
chosenBattleRooms = []
while baseIndex < len(allBattleRoomIds):
nextRoomId = allBattleRoomIds[baseIndex]
baseIndex += 1
newNumBattlesLeft = numBattlesLeft - MintRoomSpecs.middleRoomId2numBattles[nextRoomId]
if newNumBattlesLeft < 0:
continue
elif newNumBattlesLeft == 0:
chosenBattleRooms.append(nextRoomId)
return chosenBattleRooms
chosenBattleRooms.append(nextRoomId)
result = self._chooseBattleRooms(newNumBattlesLeft, allBattleRoomIds, baseIndex, chosenBattleRooms)
if result is not None:
return result
else:
del chosenBattleRooms[-1:]
else:
return
return
def __str__(self):
return 'MintLayout: id=%s, floor=%s, numRooms=%s, numBattles=%s' % (self.mintId,
self.floorNum,
self.getNumRooms(),
self.getNumBattles())
def __repr__(self):
return str(self)
Reference in a new issue View git blame Copy permalink