mirror of
https://github.com/Sneed-Group/Poodletooth-iLand
synced 2024-12-23 03:35:12 -06:00
115 lines
4.4 KiB
Python
115 lines
4.4 KiB
Python
from direct.directnotify import DirectNotifyGlobal
|
|
from direct.fsm import State
|
|
from direct.interval.IntervalGlobal import *
|
|
from panda3d.core import *
|
|
from toontown.building import Elevator
|
|
from toontown.coghq import CogHQExterior
|
|
from toontown.dna.DNAParser import loadDNAFileAI, DNAStorage
|
|
from toontown.hood import ZoneUtil
|
|
from toontown.safezone import Train
|
|
|
|
|
|
class CashbotHQExterior(CogHQExterior.CogHQExterior):
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('CashbotHQExterior')
|
|
TrackZ = -67
|
|
TrainTracks = [{'start': Point3(-1000, -54.45, TrackZ),
|
|
'end': Point3(2200, -54.45, TrackZ)},
|
|
{'start': Point3(1800, -133.45, TrackZ),
|
|
'end': Point3(-1200, -133.45, TrackZ)},
|
|
{'start': Point3(-1000, -212.45, TrackZ),
|
|
'end': Point3(2200, -212.45, TrackZ)},
|
|
{'start': Point3(1800, -291.45, TrackZ),
|
|
'end': Point3(-1200, -291.45, TrackZ)}]
|
|
|
|
def __init__(self, loader, parentFSM, doneEvent):
|
|
CogHQExterior.CogHQExterior.__init__(self, loader, parentFSM, doneEvent)
|
|
self.elevatorDoneEvent = 'elevatorDone'
|
|
self.trains = None
|
|
self.fsm.addState(State.State('elevator', self.enterElevator, self.exitElevator, ['walk', 'stopped']))
|
|
state = self.fsm.getStateNamed('walk')
|
|
state.addTransition('elevator')
|
|
state = self.fsm.getStateNamed('stopped')
|
|
state.addTransition('elevator')
|
|
state = self.fsm.getStateNamed('stickerBook')
|
|
state.addTransition('elevator')
|
|
state = self.fsm.getStateNamed('squished')
|
|
state.addTransition('elevator')
|
|
return
|
|
|
|
def load(self):
|
|
CogHQExterior.CogHQExterior.load(self)
|
|
if not self.trains:
|
|
self.trains = []
|
|
for track in self.TrainTracks:
|
|
train = Train.Train(track['start'], track['end'], self.TrainTracks.index(track), len(self.TrainTracks))
|
|
self.trains.append(train)
|
|
|
|
def unload(self):
|
|
CogHQExterior.CogHQExterior.unload(self)
|
|
for train in self.trains:
|
|
train.delete()
|
|
|
|
self.trains = None
|
|
return
|
|
|
|
def enter(self, requestStatus):
|
|
CogHQExterior.CogHQExterior.enter(self, requestStatus)
|
|
|
|
for train in self.trains:
|
|
train.show()
|
|
|
|
# Load the CogHQ DNA file:
|
|
dnaStore = DNAStorage()
|
|
dnaFileName = self.genDNAFileName(self.zoneId)
|
|
loadDNAFileAI(dnaStore, dnaFileName)
|
|
|
|
# Collect all of the vis group zone IDs:
|
|
self.zoneVisDict = {}
|
|
for i in xrange(dnaStore.getNumDNAVisGroupsAI()):
|
|
groupFullName = dnaStore.getDNAVisGroupName(i)
|
|
visGroup = dnaStore.getDNAVisGroupAI(i)
|
|
visZoneId = int(base.cr.hoodMgr.extractGroupName(groupFullName))
|
|
visibles = []
|
|
for i in xrange(visGroup.getNumVisibles()):
|
|
visibles.append(int(visGroup.getVisible(i)))
|
|
visibles.append(ZoneUtil.getBranchZone(visZoneId))
|
|
self.zoneVisDict[visZoneId] = visibles
|
|
|
|
# Next, we want interest in all vis groups due to this being a Cog HQ:
|
|
base.cr.sendSetZoneMsg(self.zoneId, self.zoneVisDict.values()[0])
|
|
|
|
def exit(self):
|
|
CogHQExterior.CogHQExterior.exit(self)
|
|
for train in self.trains:
|
|
train.hide()
|
|
|
|
def enterElevator(self, distElevator):
|
|
self.accept(self.elevatorDoneEvent, self.handleElevatorDone)
|
|
self.elevator = Elevator.Elevator(self.fsm.getStateNamed('elevator'), self.elevatorDoneEvent, distElevator)
|
|
self.elevator.load()
|
|
self.elevator.enter()
|
|
|
|
def exitElevator(self):
|
|
self.ignore(self.elevatorDoneEvent)
|
|
self.elevator.unload()
|
|
self.elevator.exit()
|
|
del self.elevator
|
|
|
|
def detectedElevatorCollision(self, distElevator):
|
|
self.fsm.request('elevator', [distElevator])
|
|
|
|
def handleElevatorDone(self, doneStatus):
|
|
self.notify.debug('handling elevator done event')
|
|
where = doneStatus['where']
|
|
if where == 'reject':
|
|
if hasattr(base.localAvatar, 'elevatorNotifier') and base.localAvatar.elevatorNotifier.isNotifierOpen():
|
|
pass
|
|
else:
|
|
self.fsm.request('walk')
|
|
elif where == 'exit':
|
|
self.fsm.request('walk')
|
|
elif where == 'mintInterior':
|
|
self.doneStatus = doneStatus
|
|
messenger.send(self.doneEvent)
|
|
else:
|
|
self.notify.error('Unknown mode: ' + where + ' in handleElevatorDone')
|