toontown-just-works/toontown/coghq/DistributedMint.py
2024-07-07 18:08:39 -05:00

221 lines
8.2 KiB
Python

from panda3d.core import *
from direct.distributed.ClockDelta import *
from direct.distributed import DistributedObject
from direct.directnotify import DirectNotifyGlobal
from direct.showbase import BulletinBoardWatcher
from otp.otpbase import OTPGlobals
from toontown.toonbase.ToontownGlobals import *
from toontown.toonbase import TTLocalizer
from toontown.coghq import DistributedMintRoom, MintLayout, MintRoom
class DistributedMint(DistributedObject.DistributedObject):
notify = DirectNotifyGlobal.directNotify.newCategory('DistributedMint')
ReadyPost = 'MintReady'
WinEvent = 'MintWinEvent'
def __init__(self, cr):
DistributedObject.DistributedObject.__init__(self, cr)
def generate(self):
self.notify.debug('generate: %s' % self.doId)
DistributedObject.DistributedObject.generate(self)
bboard.post('mint', self)
self.roomWatcher = None
self.geom = None
self.rooms = []
self.hallways = []
self.allRooms = []
self.curToonRoomNum = None
base.localAvatar.setCameraCollisionsCanMove(1)
base.localAvatar.reparentTo(render)
base.localAvatar.setPosHpr(0, 0, 0, 0, 0, 0)
self.accept('SOSPanelEnter', self.handleSOSPanel)
return
def setZoneId(self, zoneId):
self.zoneId = zoneId
def setMintId(self, id):
DistributedMint.notify.debug('setMintId: %s' % id)
self.mintId = id
def setFloorNum(self, num):
DistributedMint.notify.debug('floorNum: %s' % num)
self.floorNum = num
self.layout = MintLayout.MintLayout(self.mintId, self.floorNum)
def setRoomDoIds(self, roomDoIds):
self.roomDoIds = roomDoIds
self.roomWatcher = BulletinBoardWatcher.BulletinBoardWatcher('roomWatcher-%s' % self.doId, [ DistributedMintRoom.getMintRoomReadyPostName(doId) for doId in self.roomDoIds ], self.gotAllRooms)
def gotAllRooms(self):
self.notify.debug('mint %s: got all rooms' % self.doId)
if self.roomWatcher:
self.roomWatcher.destroy()
self.roomWatcher = None
self.geom = render.attachNewNode('mint%s' % self.doId)
for doId in self.roomDoIds:
self.rooms.append(base.cr.doId2do[doId])
self.rooms[-1].setMint(self)
self.notify.info('mintId %s, floor %s, %s' % (self.mintId, self.floorNum, self.rooms[0].avIdList))
rng = self.layout.getRng()
numRooms = self.layout.getNumRooms()
for i, room in enumerate(self.rooms):
if i == 0:
room.getGeom().reparentTo(self.geom)
else:
room.attachTo(self.hallways[i - 1], rng)
self.allRooms.append(room)
self.listenForFloorEvents(room)
if i < numRooms - 1:
hallway = MintRoom.MintRoom(self.layout.getHallwayModel(i))
hallway.attachTo(room, rng)
hallway.setRoomNum(i * 2 + 1)
hallway.initFloorCollisions()
hallway.enter()
self.hallways.append(hallway)
self.allRooms.append(hallway)
self.listenForFloorEvents(hallway)
def handleCameraRayFloorCollision(collEntry, self = self):
name = collEntry.getIntoNode().getName()
self.notify.debug('camera floor ray collided with: %s' % name)
prefix = MintRoom.MintRoom.FloorCollPrefix
prefixLen = len(prefix)
if name[:prefixLen] == prefix:
try:
roomNum = int(name[prefixLen:])
except:
DistributedLevel.notify.warning('Invalid zone floor collision node: %s' % name)
else:
self.camEnterRoom(roomNum)
self.accept('on-floor', handleCameraRayFloorCollision)
if bboard.has('mintRoom'):
self.warpToRoom(bboard.get('mintRoom'))
firstSetZoneDoneEvent = self.cr.getNextSetZoneDoneEvent()
def handleFirstSetZoneDone():
self.notify.debug('mintHandleFirstSetZoneDone')
bboard.post(DistributedMint.ReadyPost, self)
self.acceptOnce(firstSetZoneDoneEvent, handleFirstSetZoneDone)
zoneList = [OTPGlobals.UberZone, self.zoneId]
for room in self.rooms:
zoneList.extend(room.zoneIds)
base.cr.sendSetZoneMsg(self.zoneId, zoneList)
self.accept('takingScreenshot', self.handleScreenshot)
return
def listenForFloorEvents(self, room):
roomNum = room.getRoomNum()
floorCollName = room.getFloorCollName()
def handleZoneEnter(collisionEntry, self = self, roomNum = roomNum):
self.toonEnterRoom(roomNum)
floorNode = collisionEntry.getIntoNode()
if floorNode.hasTag('ouch'):
room = self.allRooms[roomNum]
ouchLevel = room.getFloorOuchLevel()
room.startOuch(ouchLevel)
self.accept('enter%s' % floorCollName, handleZoneEnter)
def handleZoneExit(collisionEntry, self = self, roomNum = roomNum):
floorNode = collisionEntry.getIntoNode()
if floorNode.hasTag('ouch'):
self.allRooms[roomNum].stopOuch()
self.accept('exit%s' % floorCollName, handleZoneExit)
def getAllRoomsTimeout(self):
self.notify.warning('mint %s: timed out waiting for room objs' % self.doId)
def toonEnterRoom(self, roomNum):
self.notify.debug('toonEnterRoom: %s' % roomNum)
if roomNum != self.curToonRoomNum:
if self.curToonRoomNum is not None:
self.allRooms[self.curToonRoomNum].localToonFSM.request('notPresent')
self.allRooms[roomNum].localToonFSM.request('present')
self.curToonRoomNum = roomNum
return
def camEnterRoom(self, roomNum):
self.notify.debug('camEnterRoom: %s' % roomNum)
if roomNum % 2 == 1:
minVis = roomNum - 2
maxVis = roomNum + 2
else:
minVis = roomNum - 1
maxVis = roomNum + 1
for i, room in enumerate(self.allRooms):
if i < minVis or i > maxVis:
room.getGeom().stash()
else:
room.getGeom().unstash()
def setBossConfronted(self, avId):
if avId == base.localAvatar.doId:
return
av = base.cr.identifyFriend(avId)
if av is None:
return
base.localAvatar.setSystemMessage(avId, TTLocalizer.MintBossConfrontedMsg % av.getName())
return
def warpToRoom(self, roomId):
for i in xrange(len(self.rooms)):
room = self.rooms[i]
if room.roomId == roomId:
break
else:
return False
base.localAvatar.setPosHpr(room.getGeom(), 0, 0, 0, 0, 0, 0)
self.camEnterRoom(i * 2)
return True
def disable(self):
self.notify.debug('disable')
self.ignoreAll()
for hallway in self.hallways:
hallway.exit()
self.rooms = []
for hallway in self.hallways:
hallway.delete()
self.hallways = []
self.allRooms = []
if self.roomWatcher:
self.roomWatcher.destroy()
self.roomWatcher = None
if self.geom is not None:
self.geom.removeNode()
self.geom = None
base.localAvatar.setCameraCollisionsCanMove(0)
if hasattr(self, 'relatedObjectMgrRequest') and self.relatedObjectMgrRequest:
self.cr.relatedObjectMgr.abortRequest(self.relatedObjectMgrRequest)
del self.relatedObjectMgrRequest
DistributedObject.DistributedObject.disable(self)
return
def delete(self):
DistributedObject.DistributedObject.delete(self)
self.ignore('SOSPanelEnter')
bboard.remove('mint')
def handleSOSPanel(self, panel):
avIds = []
for avId in self.rooms[0].avIdList:
if base.cr.doId2do.get(avId):
avIds.append(avId)
panel.setFactoryToonIdList(avIds)
def handleScreenshot(self):
base.addScreenshotString('mintId: %s, floor (from 1): %s' % (self.mintId, self.floorNum + 1))
if hasattr(self, 'currentRoomName'):
base.addScreenshotString('%s' % self.currentRoomName)