302 lines
12 KiB
Python
302 lines
12 KiB
Python
|
import math
|
||
|
|
import random
|
||
|
|
from direct.directnotify import DirectNotifyGlobal
|
||
|
|
from direct.interval.IntervalGlobal import *
|
||
|
|
from direct.fsm import ClassicFSM, State
|
||
|
|
from panda3d.core import *
|
||
|
|
from pandac.PandaModules import NodePath
|
||
|
|
from toontown.toonbase.ToontownGlobals import *
|
||
|
|
from toontown.safezone import SafeZoneLoader
|
||
|
|
from toontown.parties import Party
|
||
|
|
from toontown.parties.PartyGlobals import FireworksStartedEvent, FireworksFinishedEvent
|
||
|
|
|
||
|
|
class PartyLoader(SafeZoneLoader.SafeZoneLoader):
|
||
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('PartyLoader')
|
||
|
|
|
||
|
|
def __init__(self, hood, parentFSM, doneEvent):
|
||
|
|
SafeZoneLoader.SafeZoneLoader.__init__(self, hood, parentFSM, doneEvent)
|
||
|
|
del self.fsm
|
||
|
|
self.fsm = ClassicFSM.ClassicFSM('PartyLoader', [State.State('start', self.enterStart, self.exitStart, ['quietZone', 'party', 'planning']),
|
||
|
|
State.State('party', self.enterParty, self.exitParty, ['quietZone']),
|
||
|
|
State.State('quietZone', self.enterQuietZone, self.exitQuietZone, ['planning', 'party']),
|
||
|
|
State.State('final', self.enterFinal, self.exitFinal, ['start'])], 'start', 'final')
|
||
|
|
self.musicFile = 'phase_13/audio/bgm/party_original_theme.ogg'
|
||
|
|
self.activityMusicFile = 'phase_13/audio/bgm/party_waltz_dance.ogg'
|
||
|
|
self.dnaFile = 'phase_13/dna/party_sz.pdna'
|
||
|
|
self.safeZoneStorageDNAFile = None
|
||
|
|
self.cloudSwitch = 0
|
||
|
|
self.id = PartyHood
|
||
|
|
self.partyOwnerId = None
|
||
|
|
self.branchZone = None
|
||
|
|
self.partyDoneEvent = 'partyDone'
|
||
|
|
self.barrel = None
|
||
|
|
self.clouds = []
|
||
|
|
self.cloudTrack = None
|
||
|
|
self.sunMoonNode = None
|
||
|
|
self.fsm.enterInitialState()
|
||
|
|
return
|
||
|
|
|
||
|
|
def load(self):
|
||
|
|
self.oldClear = base.win.getClearColor()
|
||
|
|
base.win.setClearColor(Vec4(0.47, 0.69, 0.3, 1.0))
|
||
|
|
SafeZoneLoader.SafeZoneLoader.load(self)
|
||
|
|
self.underwaterSound = base.loadSfx('phase_4/audio/sfx/AV_ambient_water.ogg')
|
||
|
|
self.swimSound = base.loadSfx('phase_4/audio/sfx/AV_swim_single_stroke.ogg')
|
||
|
|
self.submergeSound = base.loadSfx('phase_5.5/audio/sfx/AV_jump_in_water.ogg')
|
||
|
|
self.birdSound = map(base.loadSfx, ['phase_4/audio/sfx/SZ_TC_bird1.ogg', 'phase_4/audio/sfx/SZ_TC_bird2.ogg', 'phase_4/audio/sfx/SZ_TC_bird3.ogg'])
|
||
|
|
self.cricketSound = map(base.loadSfx, ['phase_4/audio/sfx/SZ_TC_bird1.ogg', 'phase_4/audio/sfx/SZ_TC_bird2.ogg', 'phase_4/audio/sfx/SZ_TC_bird3.ogg'])
|
||
|
|
|
||
|
|
def unload(self):
|
||
|
|
self.ignoreAll()
|
||
|
|
base.win.setClearColor(self.oldClear)
|
||
|
|
if base.cr.partyManager:
|
||
|
|
base.cr.partyManager.leaveParty()
|
||
|
|
self.partyOwnerId = None
|
||
|
|
self.partyZoneId = None
|
||
|
|
if self.place:
|
||
|
|
self.place.exit()
|
||
|
|
self.place.unload()
|
||
|
|
del self.place
|
||
|
|
del self.underwaterSound
|
||
|
|
del self.swimSound
|
||
|
|
del self.submergeSound
|
||
|
|
del self.birdSound
|
||
|
|
del self.cricketSound
|
||
|
|
self.__cleanupCloudFadeInterval()
|
||
|
|
if self.sunMoonNode:
|
||
|
|
self.sunMoonNode.removeNode()
|
||
|
|
del self.sunMoonNode
|
||
|
|
self.sunMoonNode = None
|
||
|
|
if self.clouds:
|
||
|
|
for cloud in self.clouds:
|
||
|
|
cloud[0].removeNode()
|
||
|
|
del cloud[1]
|
||
|
|
|
||
|
|
del self.clouds
|
||
|
|
if self.barrel:
|
||
|
|
self.barrel.removeNode()
|
||
|
|
SafeZoneLoader.SafeZoneLoader.unload(self)
|
||
|
|
return
|
||
|
|
|
||
|
|
def loadClouds(self):
|
||
|
|
self.loadCloudPlatforms()
|
||
|
|
if base.cloudPlatformsEnabled and 0:
|
||
|
|
self.setCloudSwitch(1)
|
||
|
|
if self.cloudSwitch:
|
||
|
|
self.setCloudSwitch(self.cloudSwitch)
|
||
|
|
|
||
|
|
def enter(self, requestStatus):
|
||
|
|
self.partyOwnerId = requestStatus.get('ownerId', base.localAvatar.doId)
|
||
|
|
base.localAvatar.inParty = 1
|
||
|
|
self.accept(FireworksStartedEvent, self.__handleFireworksStarted)
|
||
|
|
self.accept(FireworksFinishedEvent, self.__handleFireworksFinished)
|
||
|
|
SafeZoneLoader.SafeZoneLoader.enter(self, requestStatus)
|
||
|
|
|
||
|
|
def exit(self):
|
||
|
|
self.ignoreAll()
|
||
|
|
base.cr.cache.flush()
|
||
|
|
base.localAvatar.stopChat()
|
||
|
|
base.localAvatar.inParty = 0
|
||
|
|
self.ignore(FireworksStartedEvent)
|
||
|
|
self.ignore(FireworksFinishedEvent)
|
||
|
|
SafeZoneLoader.SafeZoneLoader.exit(self)
|
||
|
|
|
||
|
|
def createSafeZone(self, dnaFile):
|
||
|
|
SafeZoneLoader.SafeZoneLoader.createSafeZone(self, dnaFile)
|
||
|
|
parent = self.geom.getParent()
|
||
|
|
geom = self.geom
|
||
|
|
n = NodePath('PartyGroundRoot')
|
||
|
|
n.reparentTo(parent)
|
||
|
|
geom.reparentTo(n)
|
||
|
|
geom.setPos(-10.0, 0.0, 0.0)
|
||
|
|
self.geom = n
|
||
|
|
self.loadSunMoon()
|
||
|
|
|
||
|
|
def loadSunMoon(self):
|
||
|
|
self.sun = loader.loadModel('phase_4/models/props/sun.bam')
|
||
|
|
self.moon = loader.loadModel('phase_5.5/models/props/moon.bam')
|
||
|
|
self.sunMoonNode = self.geom.attachNewNode('sunMoon')
|
||
|
|
self.sunMoonNode.setPosHpr(0, 0, 0, 0, 0, 0)
|
||
|
|
if self.sun:
|
||
|
|
self.sun.reparentTo(self.sunMoonNode)
|
||
|
|
self.sun.setY(270)
|
||
|
|
self.sun.setScale(2)
|
||
|
|
self.sun.setBillboardPointEye()
|
||
|
|
if self.moon:
|
||
|
|
self.moon.reparentTo(self.sunMoonNode)
|
||
|
|
self.moon.setY(-270)
|
||
|
|
self.moon.setScale(15)
|
||
|
|
self.moon.setBillboardPointEye()
|
||
|
|
self.sunMoonNode.setP(30)
|
||
|
|
|
||
|
|
def enterParty(self, requestStatus):
|
||
|
|
self.notify.debug('enterParty: requestStatus = %s' % requestStatus)
|
||
|
|
ownerId = requestStatus.get('ownerId')
|
||
|
|
if ownerId:
|
||
|
|
self.partyOwnerId = ownerId
|
||
|
|
zoneId = requestStatus['zoneId']
|
||
|
|
self.notify.debug('enterParty, ownerId = %s, zoneId = %s' % (self.partyOwnerId, zoneId))
|
||
|
|
self.accept(self.partyDoneEvent, self.handlePartyDone)
|
||
|
|
self.place = Party.Party(self, self.partyOwnerId, zoneId, self.fsm.getStateNamed('party'), self.partyDoneEvent)
|
||
|
|
base.cr.playGame.setPlace(self.place)
|
||
|
|
self.place.load()
|
||
|
|
self.place.enter(requestStatus)
|
||
|
|
self.partyZoneId = zoneId
|
||
|
|
|
||
|
|
def exitParty(self):
|
||
|
|
self.notify.debug('exitParty')
|
||
|
|
self.ignore(self.partyDoneEvent)
|
||
|
|
self.place.exit()
|
||
|
|
self.place.unload()
|
||
|
|
self.place = None
|
||
|
|
base.cr.playGame.setPlace(self.place)
|
||
|
|
base.cr.cache.flush()
|
||
|
|
return
|
||
|
|
|
||
|
|
def handlePartyDone(self, doneStatus = None):
|
||
|
|
PartyLoader.notify.debug('handlePartyDone doneStatus = %s' % doneStatus)
|
||
|
|
if not doneStatus:
|
||
|
|
doneStatus = self.place.getDoneStatus()
|
||
|
|
how = doneStatus['how']
|
||
|
|
shardId = doneStatus['shardId']
|
||
|
|
hoodId = doneStatus['hoodId']
|
||
|
|
zoneId = doneStatus['zoneId']
|
||
|
|
avId = doneStatus.get('avId', -1)
|
||
|
|
ownerId = doneStatus.get('ownerId', -1)
|
||
|
|
self.notify.debug('hoodId = %s, avId = %s' % (hoodId, avId))
|
||
|
|
self.doneStatus = doneStatus
|
||
|
|
messenger.send(self.doneEvent)
|
||
|
|
|
||
|
|
def handleQuietZoneDone(self):
|
||
|
|
status = self.quietZoneStateData.getRequestStatus()
|
||
|
|
self.fsm.request(status['where'], [status])
|
||
|
|
|
||
|
|
def atMyParty(self):
|
||
|
|
if self.partyOwnerId != None:
|
||
|
|
if self.partyOwnerId == base.localAvatar.getDoId():
|
||
|
|
return 1
|
||
|
|
else:
|
||
|
|
return 0
|
||
|
|
else:
|
||
|
|
self.notify.warning("We aren't in an party")
|
||
|
|
return
|
||
|
|
|
||
|
|
def startCloudPlatforms(self):
|
||
|
|
return
|
||
|
|
if len(self.clouds):
|
||
|
|
self.cloudTrack = self.__cloudTrack()
|
||
|
|
self.cloudTrack.loop()
|
||
|
|
|
||
|
|
def stopCloudPlatforms(self):
|
||
|
|
if self.cloudTrack:
|
||
|
|
self.cloudTrack.pause()
|
||
|
|
del self.cloudTrack
|
||
|
|
self.cloudTrack = None
|
||
|
|
return
|
||
|
|
|
||
|
|
def __cloudTrack(self):
|
||
|
|
track = Parallel()
|
||
|
|
for cloud in self.clouds:
|
||
|
|
axis = cloud[1]
|
||
|
|
pos = cloud[0].getPos(render)
|
||
|
|
newPos = pos + axis * 30
|
||
|
|
reversePos = pos - axis * 30
|
||
|
|
track.append(Sequence(LerpPosInterval(cloud[0], 10, newPos), LerpPosInterval(cloud[0], 20, reversePos), LerpPosInterval(cloud[0], 10, pos)))
|
||
|
|
|
||
|
|
return track
|
||
|
|
|
||
|
|
def debugGeom(self, decomposed):
|
||
|
|
print 'numPrimitives = %d' % decomposed.getNumPrimitives()
|
||
|
|
for primIndex in xrange(decomposed.getNumPrimitives()):
|
||
|
|
prim = decomposed.getPrimitive(primIndex)
|
||
|
|
print 'prim = %s' % prim
|
||
|
|
print 'isIndexed = %d' % prim.isIndexed()
|
||
|
|
print 'prim.getNumPrimitives = %d' % prim.getNumPrimitives()
|
||
|
|
for basicPrim in xrange(prim.getNumPrimitives()):
|
||
|
|
print '%d start=%d' % (basicPrim, prim.getPrimitiveStart(basicPrim))
|
||
|
|
print '%d end=%d' % (basicPrim, prim.getPrimitiveEnd(basicPrim))
|
||
|
|
|
||
|
|
def loadCloud(self, version, radius, zOffset):
|
||
|
|
self.notify.debug('loadOnePlatform version=%d' % version)
|
||
|
|
cloud = NodePath('cloud-%d%d' % (radius, version))
|
||
|
|
cloudModel = loader.loadModel('phase_5.5/models/estate/bumper_cloud')
|
||
|
|
cc = cloudModel.copyTo(cloud)
|
||
|
|
colCube = cc.find('**/collision')
|
||
|
|
colCube.setName('cloudSphere-0')
|
||
|
|
dTheta = 2.0 * math.pi / self.numClouds
|
||
|
|
cloud.reparentTo(self.cloudOrigin)
|
||
|
|
axes = [Vec3(1, 0, 0), Vec3(0, 1, 0), Vec3(0, 0, 1)]
|
||
|
|
cloud.setPos(radius * math.cos(version * dTheta), radius * math.sin(version * dTheta), 4 * random.random() + zOffset)
|
||
|
|
cloud.setScale(4.0)
|
||
|
|
cloud.setTag('number', '%d%d' % (radius, version))
|
||
|
|
self.clouds.append([cloud, random.choice(axes)])
|
||
|
|
|
||
|
|
def loadSkyCollision(self):
|
||
|
|
plane = CollisionPlane(Plane(Vec3(0, 0, -1), Point3(0, 0, 200)))
|
||
|
|
plane.setTangible(0)
|
||
|
|
planeNode = CollisionNode('sky_collision')
|
||
|
|
planeNode.addSolid(plane)
|
||
|
|
self.cloudOrigin.attachNewNode(planeNode)
|
||
|
|
|
||
|
|
def loadCloudPlatforms(self):
|
||
|
|
self.cloudOrigin = self.geom.attachNewNode('cloudOrigin')
|
||
|
|
self.cloudOrigin.setZ(30)
|
||
|
|
self.loadSkyCollision()
|
||
|
|
self.numClouds = 12
|
||
|
|
for i in xrange(self.numClouds):
|
||
|
|
self.loadCloud(i, 50, 0)
|
||
|
|
|
||
|
|
for i in xrange(self.numClouds):
|
||
|
|
self.loadCloud(i, 70, 30)
|
||
|
|
|
||
|
|
for i in xrange(self.numClouds):
|
||
|
|
self.loadCloud(i, 30, 60)
|
||
|
|
|
||
|
|
self.cloudOrigin.stash()
|
||
|
|
|
||
|
|
def __cleanupCloudFadeInterval(self):
|
||
|
|
if hasattr(self, 'cloudFadeInterval'):
|
||
|
|
self.cloudFadeInterval.pause()
|
||
|
|
self.cloudFadeInterval = None
|
||
|
|
return
|
||
|
|
|
||
|
|
def fadeClouds(self):
|
||
|
|
self.__cleanupCloudFadeInterval()
|
||
|
|
self.cloudOrigin.setTransparency(1)
|
||
|
|
self.cloudFadeInterval = self.cloudOrigin.colorInterval(0.5, Vec4(1, 1, 1, int(self.cloudOrigin.isStashed())), blendType='easeIn')
|
||
|
|
if self.cloudOrigin.isStashed():
|
||
|
|
self.cloudOrigin.setColor(Vec4(1, 1, 1, 0))
|
||
|
|
self.setCloudSwitch(1)
|
||
|
|
else:
|
||
|
|
self.cloudFadeInterval = Sequence(self.cloudFadeInterval, Func(self.setCloudSwitch, 0), Func(self.cloudOrigin.setTransparency, 0))
|
||
|
|
self.cloudFadeInterval.start()
|
||
|
|
|
||
|
|
def setCloudSwitch(self, on):
|
||
|
|
self.cloudSwitch = on
|
||
|
|
if hasattr(self, 'cloudOrigin'):
|
||
|
|
if on:
|
||
|
|
self.cloudOrigin.unstash()
|
||
|
|
else:
|
||
|
|
self.cloudOrigin.stash()
|
||
|
|
|
||
|
|
def _clearDayChangeInterval(self):
|
||
|
|
if hasattr(self, 'dayChangeInterval'):
|
||
|
|
self.dayChangeInterval.pause()
|
||
|
|
self.dayChangeInterval = None
|
||
|
|
return
|
||
|
|
|
||
|
|
def switchToNight(self):
|
||
|
|
self._clearDayChangeInterval()
|
||
|
|
self.dayChangeInterval = Sequence(self.sunMoonNode.hprInterval(5.0, Point3(0, -30, 0), blendType='easeInOut'), Func(base.win.setClearColor, Vec4(0.15, 0.22, 0.14, 1.0)))
|
||
|
|
self.dayChangeInterval.start()
|
||
|
|
|
||
|
|
def switchToDay(self):
|
||
|
|
self.dayChangeInterval = Sequence(Func(base.win.setClearColor, Vec4(0.47, 0.69, 0.3, 1.0)), self.sunMoonNode.hprInterval(5.0, Point3(0, 30, 0), blendType='easeInOut'))
|
||
|
|
self.dayChangeInterval.start()
|
||
|
|
|
||
|
|
def __handleFireworksStarted(self):
|
||
|
|
self.sunMoonNode.hide()
|
||
|
|
|
||
|
|
def __handleFireworksFinished(self):
|
||
|
|
self.sunMoonNode.show()
|