mirror of
https://github.com/Sneed-Group/Poodletooth-iLand
synced 2024-12-26 21:22:27 -06:00
354 lines
13 KiB
Python
354 lines
13 KiB
Python
|
from direct.gui.DirectGui import *
|
||
|
|
from pandac.PandaModules import *
|
||
|
|
from direct.interval.IntervalGlobal import *
|
||
|
|
from direct.distributed.ClockDelta import *
|
||
|
|
from direct.fsm import FSM
|
||
|
|
from direct.distributed import DistributedObject
|
||
|
|
from direct.showutil import Rope
|
||
|
|
from direct.showbase import PythonUtil
|
||
|
|
from direct.task import Task
|
||
|
|
from toontown.toonbase import ToontownGlobals
|
||
|
|
from otp.otpbase import OTPGlobals
|
||
|
|
from direct.actor import Actor
|
||
|
|
from toontown.suit import Suit
|
||
|
|
from toontown.suit import SuitDNA
|
||
|
|
import random
|
||
|
|
from toontown.battle import BattleProps
|
||
|
|
from toontown.toon import NPCToons
|
||
|
|
|
||
|
|
class DistributedLawbotChair(DistributedObject.DistributedObject, FSM.FSM):
|
||
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('DistributedLawbotChair')
|
||
|
|
chairCushionSurface = Point3(0, -0.75, 2.25)
|
||
|
|
landingPt = Point3(0, -1.5, 0)
|
||
|
|
courtroomCeiling = 30
|
||
|
|
|
||
|
|
def __init__(self, cr):
|
||
|
|
DistributedObject.DistributedObject.__init__(self, cr)
|
||
|
|
FSM.FSM.__init__(self, 'DistributedLawbotBossChair')
|
||
|
|
self.boss = None
|
||
|
|
self.index = None
|
||
|
|
self.avId = 0
|
||
|
|
self.modelPath = 'phase_11/models/lawbotHQ/JuryBoxChair'
|
||
|
|
self.modelFindString = None
|
||
|
|
self.nodePath = None
|
||
|
|
self.ival = None
|
||
|
|
self.origHpr = Point3(0, 0, 0)
|
||
|
|
self.downTime = 0.5
|
||
|
|
self.upTime = 5
|
||
|
|
self.cogJuror = None
|
||
|
|
self.propInSound = None
|
||
|
|
self.propOutSound = None
|
||
|
|
self.propTrack = None
|
||
|
|
self.cogJurorTrack = None
|
||
|
|
self.cogJurorSound = None
|
||
|
|
self.toonJurorIndex = -1
|
||
|
|
self.toonJuror = None
|
||
|
|
return
|
||
|
|
|
||
|
|
def announceGenerate(self):
|
||
|
|
self.notify.debug('announceGenerate: %s' % self.doId)
|
||
|
|
DistributedObject.DistributedObject.announceGenerate(self)
|
||
|
|
self.name = 'Chair-%s' % self.doId
|
||
|
|
self.loadModel(self.modelPath, self.modelFindString)
|
||
|
|
self.randomGenerator = random.Random()
|
||
|
|
self.randomGenerator.seed(self.doId)
|
||
|
|
self.loadSounds()
|
||
|
|
self.loadCogJuror()
|
||
|
|
self.cogJuror.stash()
|
||
|
|
origPos = self.computePos()
|
||
|
|
self.nodePath.setPos(origPos)
|
||
|
|
self.nodePath.setHpr(-90, 0, 0)
|
||
|
|
chairParent = self.boss.getChairParent()
|
||
|
|
self.nodePath.wrtReparentTo(chairParent)
|
||
|
|
self.boss.chairs[self.index] = self
|
||
|
|
|
||
|
|
def delete(self):
|
||
|
|
DistributedObject.DistributedObject.delete(self)
|
||
|
|
loader.unloadModel(self.modelPath)
|
||
|
|
self.unloadSounds()
|
||
|
|
self.nodePath.removeNode()
|
||
|
|
|
||
|
|
def loadModel(self, modelPath, modelFindString = None):
|
||
|
|
if self.nodePath == None:
|
||
|
|
self.makeNodePath()
|
||
|
|
else:
|
||
|
|
self.chair.getChildren().detach()
|
||
|
|
model = loader.loadModel(modelPath)
|
||
|
|
if modelFindString != None:
|
||
|
|
model = model.find('**/' + modelFindString)
|
||
|
|
model.instanceTo(self.chair)
|
||
|
|
trigger_chair = self.chair.find('**/trigger_chair')
|
||
|
|
if not trigger_chair.isEmpty():
|
||
|
|
trigger_chair.stash()
|
||
|
|
collision_chair = self.chair.find('**/collision_chair')
|
||
|
|
if not collision_chair.isEmpty():
|
||
|
|
collision_chair.stash()
|
||
|
|
shadow = self.chair.find('**/shadow')
|
||
|
|
if not shadow.isEmpty():
|
||
|
|
pass
|
||
|
|
self.scale = 0.5
|
||
|
|
self.chair.setScale(self.scale)
|
||
|
|
self.attachColSphere()
|
||
|
|
return
|
||
|
|
|
||
|
|
def loadSounds(self):
|
||
|
|
if self.propInSound == None:
|
||
|
|
self.propInSound = base.loadSfx('phase_5/audio/sfx/ENC_propeller_in.ogg')
|
||
|
|
if self.propOutSound == None:
|
||
|
|
self.propOutSound = base.loadSfx('phase_5/audio/sfx/ENC_propeller_out.ogg')
|
||
|
|
if self.cogJurorSound == None:
|
||
|
|
self.cogJurorSound = base.loadSfx('phase_11/audio/sfx/LB_cog_jury.ogg')
|
||
|
|
return
|
||
|
|
|
||
|
|
def unloadSounds(self):
|
||
|
|
if self.propInSound:
|
||
|
|
del self.propInSound
|
||
|
|
self.propInSound = None
|
||
|
|
if self.propOutSound:
|
||
|
|
del self.propOutSound
|
||
|
|
self.propOutSound = None
|
||
|
|
if self.cogJurorSound:
|
||
|
|
del self.cogJurorSound
|
||
|
|
self.cogJurorSound = None
|
||
|
|
return
|
||
|
|
|
||
|
|
def loadCogJuror(self):
|
||
|
|
self.cleanupCogJuror()
|
||
|
|
self.cogJuror = Suit.Suit()
|
||
|
|
level = self.randomGenerator.randrange(len(SuitDNA.suitsPerLevel))
|
||
|
|
self.cogJuror.dna = SuitDNA.SuitDNA()
|
||
|
|
self.cogJuror.dna.newSuitRandom(level=level, dept='l')
|
||
|
|
self.cogJuror.setDNA(self.cogJuror.dna)
|
||
|
|
self.cogJuror.pose('landing', 0)
|
||
|
|
self.cogJuror.reparentTo(self.nodePath)
|
||
|
|
self.cogJuror.prop = None
|
||
|
|
if self.cogJuror.prop == None:
|
||
|
|
self.cogJuror.prop = BattleProps.globalPropPool.getProp('propeller')
|
||
|
|
head = self.cogJuror.find('**/joint_head')
|
||
|
|
self.cogJuror.prop.reparentTo(head)
|
||
|
|
self.propTrack = Sequence(ActorInterval(self.cogJuror.prop, 'propeller', startFrame=8, endFrame=25))
|
||
|
|
return
|
||
|
|
|
||
|
|
def attachColSphere(self):
|
||
|
|
chairTop = self.nodePath.find('**/top*')
|
||
|
|
chairHandle = self.nodePath.find('**/handle*')
|
||
|
|
collNode = CollisionNode(self.uniqueName('headSphere'))
|
||
|
|
topBounds = self.chair.getBounds()
|
||
|
|
center = topBounds.getCenter()
|
||
|
|
radius = topBounds.getRadius()
|
||
|
|
radius *= 0.65
|
||
|
|
adjustedZ = center[2]
|
||
|
|
adjustedZ += 0.6
|
||
|
|
sphere1 = CollisionSphere(center[0], center[1], adjustedZ, radius)
|
||
|
|
sphere1.setTangible(1)
|
||
|
|
collNode.addSolid(sphere1)
|
||
|
|
collNode.setName('Chair-%s' % self.index)
|
||
|
|
self.collNodePath = self.nodePath.attachNewNode(collNode)
|
||
|
|
|
||
|
|
def makeNodePath(self):
|
||
|
|
self.nodePath = Actor.Actor()
|
||
|
|
self.chair = self.nodePath.attachNewNode('myChair')
|
||
|
|
|
||
|
|
def disable(self):
|
||
|
|
DistributedObject.DistributedObject.disable(self)
|
||
|
|
self.nodePath.detachNode()
|
||
|
|
if self.ival:
|
||
|
|
self.ival.finish()
|
||
|
|
self.ival = None
|
||
|
|
self.ignoreAll()
|
||
|
|
del self.boss.chairs[self.index]
|
||
|
|
self.cleanup()
|
||
|
|
if self.propTrack:
|
||
|
|
self.propTrack.finish()
|
||
|
|
self.propTrack = None
|
||
|
|
if self.cogJurorTrack:
|
||
|
|
self.cogJurorTrack.finish()
|
||
|
|
self.cogJurorTrack = None
|
||
|
|
self.cleanupCogJuror()
|
||
|
|
self.cleanupToonJuror()
|
||
|
|
return
|
||
|
|
|
||
|
|
def stopCogsFlying(self):
|
||
|
|
if self.ival:
|
||
|
|
self.ival.finish()
|
||
|
|
self.ival = None
|
||
|
|
if self.propTrack:
|
||
|
|
self.propTrack.finish()
|
||
|
|
self.propTrack = None
|
||
|
|
if self.cogJurorTrack:
|
||
|
|
self.cogJurorTrack.finish()
|
||
|
|
self.cogJurorTrack = None
|
||
|
|
return
|
||
|
|
|
||
|
|
def cleanupCogJuror(self):
|
||
|
|
if self.cogJuror:
|
||
|
|
self.cogJuror.detachNode()
|
||
|
|
self.cogJuror.delete()
|
||
|
|
del self.cogJuror
|
||
|
|
self.cogJuror = None
|
||
|
|
return
|
||
|
|
|
||
|
|
def cleanupToonJuror(self):
|
||
|
|
if self.toonJuror:
|
||
|
|
self.toonJuror.detachNode()
|
||
|
|
self.toonJuror.delete()
|
||
|
|
del self.toonJuror
|
||
|
|
self.toonJuror = None
|
||
|
|
return
|
||
|
|
|
||
|
|
def cleanup(self):
|
||
|
|
self.boss = None
|
||
|
|
return
|
||
|
|
|
||
|
|
def startCogJuror(self, duration, y):
|
||
|
|
if self.cogJurorTrack:
|
||
|
|
self.cogJurorTrack.finish()
|
||
|
|
self.loadCogJuror()
|
||
|
|
self.cogJuror.stash()
|
||
|
|
x = 0
|
||
|
|
curPos = self.nodePath.getPos(render)
|
||
|
|
z = self.courtroomCeiling - curPos[2]
|
||
|
|
self.notify.debug('curPos =%s\nz=%f' % (curPos, z))
|
||
|
|
cogTrack = Sequence(Func(self.cogJuror.setPos, x, y, z), Func(self.cogJuror.unstash), Func(self.propTrack.loop), self.cogJuror.posInterval(duration, self.landingPt, Point3(x, y, z)), Func(self.propTrack.finish), Func(self.stashCogJuror))
|
||
|
|
audioTrack = SoundInterval(self.propInSound, duration=duration, node=self.cogJuror, loop=1)
|
||
|
|
self.cogJurorTrack = Parallel(audioTrack, cogTrack)
|
||
|
|
self.cogJurorTrack.start()
|
||
|
|
|
||
|
|
def stashCogJuror(self):
|
||
|
|
if self.cogJuror and not self.cogJuror.isEmpty():
|
||
|
|
self.cogJuror.stash()
|
||
|
|
|
||
|
|
def putCogJurorOnSeat(self):
|
||
|
|
self.stopCogsFlying()
|
||
|
|
if self.cogJuror and not self.cogJuror.isEmpty():
|
||
|
|
base.playSfx(self.cogJurorSound, node=self.chair)
|
||
|
|
self.cogJuror.unstash()
|
||
|
|
self.cogJuror.prop.stash()
|
||
|
|
self.cogJuror.pose('landing', 47)
|
||
|
|
self.cogJuror.setH(180)
|
||
|
|
self.cogJuror.setPos(0, -1.25, 0.95)
|
||
|
|
if self.toonJuror:
|
||
|
|
self.toonJuror.hide()
|
||
|
|
else:
|
||
|
|
self.notify.warning('putCogJurorOnSeat invalid cogJuror')
|
||
|
|
|
||
|
|
def putToonJurorOnSeat(self):
|
||
|
|
if self.toonJuror and not self.toonJuror.isEmpty():
|
||
|
|
self.toonJuror.show()
|
||
|
|
self.toonJuror.reparentTo(self.nodePath)
|
||
|
|
self.toonJuror.setH(180)
|
||
|
|
self.toonJuror.setPos(0, -2.5, 0.95)
|
||
|
|
self.toonJuror.animFSM.request('Sit')
|
||
|
|
else:
|
||
|
|
self.notify.warning('putToonJurorOnSeat invalid toonJuror')
|
||
|
|
|
||
|
|
def showCogJurorFlying(self):
|
||
|
|
self.notify.debug('showCogJurorFlying')
|
||
|
|
self.startCogJuror(ToontownGlobals.LawbotBossCogJurorFlightTime, -ToontownGlobals.LawbotBossCogJurorDistance)
|
||
|
|
|
||
|
|
def setBossCogId(self, bossCogId):
|
||
|
|
self.bossCogId = bossCogId
|
||
|
|
self.boss = base.cr.doId2do[bossCogId]
|
||
|
|
|
||
|
|
def setIndex(self, index):
|
||
|
|
self.index = index
|
||
|
|
|
||
|
|
def setState(self, state):
|
||
|
|
avId = 0
|
||
|
|
if state == 'C':
|
||
|
|
self.demand('Controlled', avId)
|
||
|
|
elif state == 'F':
|
||
|
|
self.demand('Free')
|
||
|
|
elif state == 'N':
|
||
|
|
self.demand('On')
|
||
|
|
elif state == 'T':
|
||
|
|
self.demand('ToonJuror')
|
||
|
|
elif state == 'S':
|
||
|
|
self.demand('SuitJuror')
|
||
|
|
elif state == 'E':
|
||
|
|
self.demand('EmptyJuror')
|
||
|
|
elif state == 'E':
|
||
|
|
self.demand('StopCogs')
|
||
|
|
else:
|
||
|
|
self.notify.error('Invalid state from AI: %s' % state)
|
||
|
|
|
||
|
|
def __touchedChair(self, entry):
|
||
|
|
self.notify.debug('__touchedChair')
|
||
|
|
self.notify.debug('self=%s entry=%s' % (self, entry))
|
||
|
|
self.boss.touchedChair(self, entry)
|
||
|
|
|
||
|
|
def __touchedChairHandle(self, entry):
|
||
|
|
self.notify.debug('__touchedChairHandle')
|
||
|
|
self.boss.touchedChairHandle(self, entry)
|
||
|
|
|
||
|
|
def enterToonJuror(self):
|
||
|
|
self.chair.setColorScale(0.2, 0.2, 1.0, 1.0)
|
||
|
|
self.boss.countToonJurors()
|
||
|
|
if not self.cogJurorTrack:
|
||
|
|
self.cogJuror.stash()
|
||
|
|
self.putToonJurorOnSeat()
|
||
|
|
|
||
|
|
def enterSuitJuror(self):
|
||
|
|
self.chair.setColorScale(0.5, 0.5, 0.5, 1.0)
|
||
|
|
self.boss.countToonJurors()
|
||
|
|
if self.toonJuror:
|
||
|
|
self.toonJuror.hide()
|
||
|
|
self.putCogJurorOnSeat()
|
||
|
|
|
||
|
|
def enterEmptyJuror(self):
|
||
|
|
self.chair.setColorScale(1.0, 1.0, 1.0, 1.0)
|
||
|
|
|
||
|
|
def enterStopCogs(self):
|
||
|
|
self.stopCogs()
|
||
|
|
|
||
|
|
def exitStopCogs(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def enterOn(self):
|
||
|
|
self.notify.debug('enterOn for chair %d' % self.index)
|
||
|
|
myHeadings = ToontownGlobals.LawbotBossChairHeadings[self.index]
|
||
|
|
seqName = 'LawbotBossChair-%s' % self.doId
|
||
|
|
self.ival = Sequence(name=seqName)
|
||
|
|
downAngle = -80
|
||
|
|
for index in xrange(len(myHeadings)):
|
||
|
|
nextIndex = index + 1
|
||
|
|
if nextIndex == len(myHeadings):
|
||
|
|
nextIndex = 0
|
||
|
|
goingDown = self.nodePath.hprInterval(self.downTime, Point3(myHeadings[index] + self.origHpr[0], downAngle, self.origHpr[2]), startHpr=Point3(myHeadings[index] + self.origHpr[0], 0, self.origHpr[2]))
|
||
|
|
self.ival.append(goingDown)
|
||
|
|
self.ival.append(Wait(self.stayDownTime))
|
||
|
|
goingUp = self.nodePath.hprInterval(self.upTime, Point3(myHeadings[nextIndex] + self.origHpr[0], 0, self.origHpr[2]), startHpr=Point3(myHeadings[index] + self.origHpr[0], downAngle, self.origHpr[2]))
|
||
|
|
self.ival.append(goingUp)
|
||
|
|
|
||
|
|
self.ival.loop()
|
||
|
|
self.accept('enterChairZap', self.__touchedChair)
|
||
|
|
self.accept('enterChairHandleZap', self.__touchedChairHandle)
|
||
|
|
|
||
|
|
def computePos(self):
|
||
|
|
rowIndex = self.index % 6
|
||
|
|
if self.index < 6:
|
||
|
|
startPt = Point3(*ToontownGlobals.LawbotBossChairRow1PosA)
|
||
|
|
endPt = Point3(*ToontownGlobals.LawbotBossChairRow1PosB)
|
||
|
|
else:
|
||
|
|
startPt = Point3(*ToontownGlobals.LawbotBossChairRow2PosA)
|
||
|
|
endPt = Point3(*ToontownGlobals.LawbotBossChairRow2PosB)
|
||
|
|
totalDisplacement = endPt - startPt
|
||
|
|
stepDisplacement = totalDisplacement / (6 - 1)
|
||
|
|
newPos = stepDisplacement * rowIndex
|
||
|
|
self.notify.debug('curDisplacement = %s' % newPos)
|
||
|
|
newPos += startPt
|
||
|
|
self.notify.debug('newPos before offset = %s' % newPos)
|
||
|
|
newPos -= Point3(*ToontownGlobals.LawbotBossJuryBoxRelativeEndPos)
|
||
|
|
self.notify.debug('newPos = %s' % newPos)
|
||
|
|
return newPos
|
||
|
|
|
||
|
|
def loadToonJuror(self):
|
||
|
|
self.cleanupToonJuror()
|
||
|
|
self.toonJuror = NPCToons.createLocalNPC(ToontownGlobals.LawbotBossBaseJurorNpcId + self.toonJurorIndex)
|
||
|
|
self.toonJuror.hide()
|
||
|
|
|
||
|
|
def setToonJurorIndex(self, newVal):
|
||
|
|
if not self.toonJurorIndex == newVal:
|
||
|
|
self.toonJurorIndex = newVal
|
||
|
|
self.loadToonJuror()
|