Poodletooth-iLand/toontown/battle/MovieUtil.py
2015-08-03 17:31:13 +03:00

607 lines
22 KiB
Python
Executable file

from direct.interval.IntervalGlobal import *
from BattleBase import *
from BattleProps import *
from direct.directnotify import DirectNotifyGlobal
import random
from direct.particles import ParticleEffect
import BattleParticles
import BattleProps
from toontown.toonbase import TTLocalizer
notify = DirectNotifyGlobal.directNotify.newCategory('MovieUtil')
SUIT_LOSE_DURATION = 8.5
SUIT_LOSE_REVIVE_DURATION = 6.0
SUIT_LURE_DISTANCE = 2.6
SUIT_LURE_DOLLAR_DISTANCE = 5.1
SUIT_EXTRA_REACH_DISTANCE = 0.9
SUIT_EXTRA_RAKE_DISTANCE = 1.1
SUIT_TRAP_DISTANCE = 2.6
SUIT_TRAP_RAKE_DISTANCE = 4.5
SUIT_TRAP_MARBLES_DISTANCE = 3.7
SUIT_TRAP_TNT_DISTANCE = 5.1
PNT3_NEARZERO = Point3(0.01, 0.01, 0.01)
PNT3_ZERO = Point3(0.0, 0.0, 0.0)
PNT3_ONE = Point3(1.0, 1.0, 1.0)
largeSuits = ['f',
'cc',
'gh',
'tw',
'bf',
'sc',
'ds',
'hh',
'cr',
'tbc',
'bs',
'sd',
'le',
'bw',
'nc',
'mb',
'ls',
'rb',
'ms',
'tf',
'm',
'mh']
shotDirection = 'left'
def avatarDodge(leftAvatars, rightAvatars, leftData, rightData):
if len(leftAvatars) > len(rightAvatars):
PoLR = rightAvatars
PoMR = leftAvatars
else:
PoLR = leftAvatars
PoMR = rightAvatars
upper = 1 + 4 * abs(len(leftAvatars) - len(rightAvatars))
if random.randint(0, upper) > 0:
avDodgeList = PoLR
else:
avDodgeList = PoMR
if avDodgeList is leftAvatars:
data = leftData
else:
data = rightData
return (avDodgeList, data)
def avatarHide(avatar):
notify.debug('avatarHide(%d)' % avatar.doId)
if hasattr(avatar, 'battleTrapProp'):
notify.debug('avatar.battleTrapProp = %s' % avatar.battleTrapProp)
avatar.detachNode()
def copyProp(prop):
from direct.actor import Actor
if isinstance(prop, Actor.Actor):
return Actor.Actor(other=prop)
else:
return prop.copyTo(hidden)
def showProp(prop, hand, pos = None, hpr = None, scale = None):
prop.reparentTo(hand)
if pos:
if callable(pos):
pos = pos()
prop.setPos(pos)
if hpr:
if callable(hpr):
hpr = hpr()
prop.setHpr(hpr)
if scale:
if callable(scale):
scale = scale()
prop.setScale(scale)
def showProps(props, hands, pos = None, hpr = None, scale = None):
index = 0
for prop in props:
prop.reparentTo(hands[index])
if pos:
prop.setPos(pos)
if hpr:
prop.setHpr(hpr)
if scale:
prop.setScale(scale)
index += 1
def hideProps(props):
for prop in props:
prop.detachNode()
def removeProp(prop):
from direct.actor import Actor
if prop.isEmpty() == 1 or prop == None:
return
prop.detachNode()
if isinstance(prop, Actor.Actor):
prop.cleanup()
else:
prop.removeNode()
return
def removeProps(props):
for prop in props:
removeProp(prop)
def getActorIntervals(props, anim):
tracks = Parallel()
for prop in props:
tracks.append(ActorInterval(prop, anim))
return tracks
def getScaleIntervals(props, duration, startScale, endScale):
tracks = Parallel()
for prop in props:
tracks.append(LerpScaleInterval(prop, duration, endScale, startScale=startScale))
return tracks
def avatarFacePoint(av, other = render):
pnt = av.getPos(other)
pnt.setZ(pnt[2] + av.getHeight())
return pnt
def insertDeathSuit(suit, deathSuit, battle = None, pos = None, hpr = None):
holdParent = suit.getParent()
if suit.getVirtual():
virtualize(deathSuit)
avatarHide(suit)
if deathSuit != None and not deathSuit.isEmpty():
if holdParent and 0:
deathSuit.reparentTo(holdParent)
else:
deathSuit.reparentTo(render)
if battle != None and pos != None:
deathSuit.setPos(battle, pos)
if battle != None and hpr != None:
deathSuit.setHpr(battle, hpr)
return
def removeDeathSuit(suit, deathSuit):
notify.debug('removeDeathSuit()')
if not deathSuit.isEmpty():
deathSuit.detachNode()
suit.cleanupLoseActor()
def insertReviveSuit(suit, deathSuit, battle = None, pos = None, hpr = None):
holdParent = suit.getParent()
if suit.getVirtual():
virtualize(deathSuit)
suit.hide()
if deathSuit != None and not deathSuit.isEmpty():
if holdParent and 0:
deathSuit.reparentTo(holdParent)
else:
deathSuit.reparentTo(render)
if battle != None and pos != None:
deathSuit.setPos(battle, pos)
if battle != None and hpr != None:
deathSuit.setHpr(battle, hpr)
return
def removeReviveSuit(suit, deathSuit):
notify.debug('removeDeathSuit()')
suit.setSkelecog(1)
suit.show()
if not deathSuit.isEmpty():
deathSuit.detachNode()
suit.cleanupLoseActor()
suit.healthBar.show()
suit.resetHealthBarForSkele()
def virtualize(deathsuit):
actorNode = deathsuit.find('**/__Actor_modelRoot')
actorCollection = actorNode.findAllMatches('*')
parts = ()
for thingIndex in xrange(0, actorCollection.getNumPaths()):
thing = actorCollection[thingIndex]
if thing.getName() not in ('joint_attachMeter', 'joint_nameTag', 'def_nameTag'):
thing.setColorScale(1.0, 0.0, 0.0, 1.0)
thing.setAttrib(ColorBlendAttrib.make(ColorBlendAttrib.MAdd))
thing.setDepthWrite(False)
thing.setBin('fixed', 1)
def createTrainTrackAppearTrack(dyingSuit, toon, battle, npcs):
retval = Sequence()
return retval
possibleSuits = []
for suitAttack in battle.movie.suitAttackDicts:
suit = suitAttack['suit']
if not suit == dyingSuit:
if hasattr(suit, 'battleTrapProp') and suit.battleTrapProp and suit.battleTrapProp.getName() == 'traintrack':
possibleSuits.append(suitAttack['suit'])
closestXDistance = 10000
closestSuit = None
for suit in possibleSuits:
suitPoint, suitHpr = battle.getActorPosHpr(suit)
xDistance = abs(suitPoint.getX())
if xDistance < closestXDistance:
closestSuit = suit
closestXDistance = xDistance
if closestSuit and closestSuit.battleTrapProp.isHidden():
closestSuit.battleTrapProp.setColorScale(1, 1, 1, 0)
closestSuit.battleTrapProp.show()
newRelativePos = dyingSuit.battleTrapProp.getPos(closestSuit)
newHpr = dyingSuit.battleTrapProp.getHpr(closestSuit)
closestSuit.battleTrapProp.setPos(newRelativePos)
closestSuit.battleTrapProp.setHpr(newHpr)
retval.append(LerpColorScaleInterval(closestSuit.battleTrapProp, 3.0, Vec4(1, 1, 1, 1)))
else:
notify.debug('could not find closest suit, returning empty sequence')
return retval
def createSuitReviveTrack(suit, toon, battle, npcs = []):
suitTrack = Sequence()
suitPos, suitHpr = battle.getActorPosHpr(suit)
if hasattr(suit, 'battleTrapProp') and suit.battleTrapProp and suit.battleTrapProp.getName() == 'traintrack' and not suit.battleTrapProp.isHidden():
suitTrack.append(createTrainTrackAppearTrack(suit, toon, battle, npcs))
deathSuit = suit.getLoseActor()
suitTrack.append(Func(notify.debug, 'before insertDeathSuit'))
suitTrack.append(Func(insertReviveSuit, suit, deathSuit, battle, suitPos, suitHpr))
suitTrack.append(Func(notify.debug, 'before actorInterval lose'))
suitTrack.append(ActorInterval(deathSuit, 'lose', duration=SUIT_LOSE_REVIVE_DURATION))
suitTrack.append(Func(notify.debug, 'before removeDeathSuit'))
suitTrack.append(Func(removeReviveSuit, suit, deathSuit, name='remove-death-suit'))
suitTrack.append(Func(notify.debug, 'after removeDeathSuit'))
suitTrack.append(ActorInterval(suit, 'slip-forward', startTime=2.48, duration=0.1))
suitTrack.append(ActorInterval(suit, 'slip-forward', startTime=2.58))
suitTrack.append(Func(suit.loop, 'neutral'))
suitTrack.append(Func(messenger.send, suit.uniqueName('hpChange')))
spinningSound = base.loadSfx('phase_3.5/audio/sfx/Cog_Death.ogg')
deathSound = base.loadSfx('phase_3.5/audio/sfx/ENC_cogfall_apart_%s.ogg' % random.randint(1, 6))
deathSoundTrack = Sequence(Wait(0.8), SoundInterval(spinningSound, duration=1.2, startTime=1.5, volume=0.2), SoundInterval(spinningSound, duration=3.0, startTime=0.6, volume=0.8), SoundInterval(deathSound, volume=0.32))
BattleParticles.loadParticles()
smallGears = BattleParticles.createParticleEffect(file='gearExplosionSmall')
singleGear = BattleParticles.createParticleEffect('GearExplosion', numParticles=1)
smallGearExplosion = BattleParticles.createParticleEffect('GearExplosion', numParticles=10)
bigGearExplosion = BattleParticles.createParticleEffect('BigGearExplosion', numParticles=30)
gearPoint = Point3(suitPos.getX(), suitPos.getY(), suitPos.getZ() + suit.height - 0.2)
smallGears.setPos(gearPoint)
singleGear.setPos(gearPoint)
smallGears.setDepthWrite(False)
singleGear.setDepthWrite(False)
smallGearExplosion.setPos(gearPoint)
bigGearExplosion.setPos(gearPoint)
smallGearExplosion.setDepthWrite(False)
bigGearExplosion.setDepthWrite(False)
explosionTrack = Sequence()
explosionTrack.append(Wait(5.4))
explosionTrack.append(createKapowExplosionTrack(battle, explosionPoint=gearPoint))
gears1Track = Sequence(Wait(2.1), ParticleInterval(smallGears, battle, worldRelative=0, duration=4.3, cleanup=True), name='gears1Track')
gears2MTrack = Track((0.0, explosionTrack), (0.7, ParticleInterval(singleGear, battle, worldRelative=0, duration=5.7, cleanup=True)), (5.2, ParticleInterval(smallGearExplosion, battle, worldRelative=0, duration=1.2, cleanup=True)), (5.4, ParticleInterval(bigGearExplosion, battle, worldRelative=0, duration=1.0, cleanup=True)), name='gears2MTrack')
toonMTrack = Parallel(name='toonMTrack')
for mtoon in battle.toons:
toonMTrack.append(Sequence(Wait(1.0), ActorInterval(mtoon, 'duck'), ActorInterval(mtoon, 'duck', startTime=1.8), Func(mtoon.loop, 'neutral')))
for mtoon in npcs:
toonMTrack.append(Sequence(Wait(1.0), ActorInterval(mtoon, 'duck'), ActorInterval(mtoon, 'duck', startTime=1.8), Func(mtoon.loop, 'neutral')))
return Parallel(suitTrack, deathSoundTrack, gears1Track, gears2MTrack, toonMTrack)
def createSuitDeathTrack(suit, toon, battle, npcs = []):
suitTrack = Sequence()
suitPos, suitHpr = battle.getActorPosHpr(suit)
if hasattr(suit, 'battleTrapProp') and suit.battleTrapProp and suit.battleTrapProp.getName() == 'traintrack' and not suit.battleTrapProp.isHidden():
suitTrack.append(createTrainTrackAppearTrack(suit, toon, battle, npcs))
deathSuit = suit.getLoseActor()
suitTrack.append(Func(notify.debug, 'before insertDeathSuit'))
suitTrack.append(Func(insertDeathSuit, suit, deathSuit, battle, suitPos, suitHpr))
suitTrack.append(Func(notify.debug, 'before actorInterval lose'))
suitTrack.append(ActorInterval(deathSuit, 'lose', duration=SUIT_LOSE_DURATION))
suitTrack.append(Func(notify.debug, 'before removeDeathSuit'))
suitTrack.append(Func(removeDeathSuit, suit, deathSuit, name='remove-death-suit'))
suitTrack.append(Func(notify.debug, 'after removeDeathSuit'))
spinningSound = base.loadSfx('phase_3.5/audio/sfx/Cog_Death.ogg')
deathSound = base.loadSfx('phase_3.5/audio/sfx/ENC_cogfall_apart_%s.ogg' % random.randint(1, 6))
deathSoundTrack = Sequence(Wait(0.8), SoundInterval(spinningSound, duration=1.2, startTime=1.5, volume=0.2), SoundInterval(spinningSound, duration=3.0, startTime=0.6, volume=0.8), SoundInterval(deathSound, volume=0.32))
BattleParticles.loadParticles()
smallGears = BattleParticles.createParticleEffect(file='gearExplosionSmall')
singleGear = BattleParticles.createParticleEffect('GearExplosion', numParticles=1)
smallGearExplosion = BattleParticles.createParticleEffect('GearExplosion', numParticles=10)
bigGearExplosion = BattleParticles.createParticleEffect('BigGearExplosion', numParticles=30)
gearPoint = Point3(suitPos.getX(), suitPos.getY(), suitPos.getZ() + suit.height - 0.2)
smallGears.setPos(gearPoint)
singleGear.setPos(gearPoint)
smallGears.setDepthWrite(False)
singleGear.setDepthWrite(False)
smallGearExplosion.setPos(gearPoint)
bigGearExplosion.setPos(gearPoint)
smallGearExplosion.setDepthWrite(False)
bigGearExplosion.setDepthWrite(False)
explosionTrack = Sequence()
explosionTrack.append(Wait(5.4))
explosionTrack.append(createKapowExplosionTrack(battle, explosionPoint=gearPoint))
gears1Track = Sequence(Wait(2.1), ParticleInterval(smallGears, battle, worldRelative=0, duration=4.3, cleanup=True), name='gears1Track')
gears2MTrack = Track((0.0, explosionTrack), (0.7, ParticleInterval(singleGear, battle, worldRelative=0, duration=5.7, cleanup=True)), (5.2, ParticleInterval(smallGearExplosion, battle, worldRelative=0, duration=1.2, cleanup=True)), (5.4, ParticleInterval(bigGearExplosion, battle, worldRelative=0, duration=1.0, cleanup=True)), name='gears2MTrack')
toonMTrack = Parallel(name='toonMTrack')
for mtoon in battle.toons:
toonMTrack.append(Sequence(Wait(1.0), ActorInterval(mtoon, 'duck'), ActorInterval(mtoon, 'duck', startTime=1.8), Func(mtoon.loop, 'neutral')))
for mtoon in npcs:
toonMTrack.append(Sequence(Wait(1.0), ActorInterval(mtoon, 'duck'), ActorInterval(mtoon, 'duck', startTime=1.8), Func(mtoon.loop, 'neutral')))
return Parallel(suitTrack, deathSoundTrack, gears1Track, gears2MTrack, toonMTrack)
def createSuitDodgeMultitrack(tDodge, suit, leftSuits, rightSuits):
suitTracks = Parallel()
suitDodgeList, sidestepAnim = avatarDodge(leftSuits, rightSuits, 'sidestep-left', 'sidestep-right')
for s in suitDodgeList:
suitTracks.append(Sequence(ActorInterval(s, sidestepAnim), Func(s.loop, 'neutral')))
suitTracks.append(Sequence(ActorInterval(suit, sidestepAnim), Func(suit.loop, 'neutral')))
suitTracks.append(Func(indicateMissed, suit))
return Sequence(Wait(tDodge), suitTracks)
def createToonDodgeMultitrack(tDodge, toon, leftToons, rightToons):
toonTracks = Parallel()
if len(leftToons) > len(rightToons):
PoLR = rightToons
PoMR = leftToons
else:
PoLR = leftToons
PoMR = rightToons
upper = 1 + 4 * abs(len(leftToons) - len(rightToons))
if random.randint(0, upper) > 0:
toonDodgeList = PoLR
else:
toonDodgeList = PoMR
if toonDodgeList is leftToons:
sidestepAnim = 'sidestep-left'
for t in toonDodgeList:
toonTracks.append(Sequence(ActorInterval(t, sidestepAnim), Func(t.loop, 'neutral')))
else:
sidestepAnim = 'sidestep-right'
toonTracks.append(Sequence(ActorInterval(toon, sidestepAnim), Func(toon.loop, 'neutral')))
toonTracks.append(Func(indicateMissed, toon))
return Sequence(Wait(tDodge), toonTracks)
def createSuitTeaseMultiTrack(suit, delay = 0.01):
suitTrack = Sequence(Wait(delay), ActorInterval(suit, 'victory', startTime=0.5, endTime=1.9), Func(suit.loop, 'neutral'))
missedTrack = Sequence(Wait(delay + 0.2), Func(indicateMissed, suit, 0.9))
return Parallel(suitTrack, missedTrack)
SPRAY_LEN = 1.5
def getSprayTrack(battle, color, origin, target, dScaleUp, dHold, dScaleDown, horizScale = 1.0, vertScale = 1.0, parent = render):
track = Sequence()
sprayProp = globalPropPool.getProp('spray')
sprayScale = hidden.attachNewNode('spray-parent')
sprayRot = hidden.attachNewNode('spray-rotate')
spray = sprayRot
spray.setColor(color)
if color[3] < 1.0:
spray.setTransparency(1)
def showSpray(sprayScale, sprayRot, sprayProp, origin, target, parent):
if callable(origin):
origin = origin()
if callable(target):
target = target()
sprayRot.reparentTo(parent)
sprayRot.clearMat()
sprayScale.reparentTo(sprayRot)
sprayScale.clearMat()
sprayProp.reparentTo(sprayScale)
sprayProp.clearMat()
sprayRot.setPos(origin)
sprayRot.lookAt(Point3(target))
track.append(Func(battle.movie.needRestoreRenderProp, sprayProp))
track.append(Func(showSpray, sprayScale, sprayRot, sprayProp, origin, target, parent))
def calcTargetScale(target = target, origin = origin, horizScale = horizScale, vertScale = vertScale):
if callable(target):
target = target()
if callable(origin):
origin = origin()
distance = Vec3(target - origin).length()
yScale = distance / SPRAY_LEN
targetScale = Point3(yScale * horizScale, yScale, yScale * vertScale)
return targetScale
track.append(LerpScaleInterval(sprayScale, dScaleUp, calcTargetScale, startScale=PNT3_NEARZERO))
track.append(Wait(dHold))
def prepareToShrinkSpray(spray, sprayProp, origin, target):
if callable(target):
target = target()
if callable(origin):
origin = origin()
sprayProp.setPos(Point3(0.0, -SPRAY_LEN, 0.0))
spray.setPos(target)
track.append(Func(prepareToShrinkSpray, spray, sprayProp, origin, target))
track.append(LerpScaleInterval(sprayScale, dScaleDown, PNT3_NEARZERO))
def hideSpray(spray, sprayScale, sprayRot, sprayProp, propPool):
sprayProp.detachNode()
removeProp(sprayProp)
sprayRot.removeNode()
sprayScale.removeNode()
track.append(Func(hideSpray, spray, sprayScale, sprayRot, sprayProp, globalPropPool))
track.append(Func(battle.movie.clearRenderProp, sprayProp))
return track
T_HOLE_LEAVES_HAND = 1.708
T_TELEPORT_ANIM = 3.3
T_HOLE_CLOSES = 0.3
def getToonTeleportOutInterval(toon):
holeActors = toon.getHoleActors()
holes = [holeActors[0], holeActors[1]]
hole = holes[0]
hole2 = holes[1]
hands = toon.getRightHands()
delay = T_HOLE_LEAVES_HAND
dur = T_TELEPORT_ANIM
holeTrack = Sequence()
holeTrack.append(Func(showProps, holes, hands))
(holeTrack.append(Wait(0.5)),)
holeTrack.append(Func(base.playSfx, toon.getSoundTeleport()))
holeTrack.append(Wait(delay - 0.5))
holeTrack.append(Func(hole.reparentTo, toon))
holeTrack.append(Func(hole2.reparentTo, hidden))
holeAnimTrack = Sequence()
holeAnimTrack.append(ActorInterval(hole, 'hole', duration=dur))
holeAnimTrack.append(Func(hideProps, holes))
runTrack = Sequence(ActorInterval(toon, 'teleport', duration=dur), Wait(T_HOLE_CLOSES), Func(toon.detachNode))
return Parallel(runTrack, holeAnimTrack, holeTrack)
def getToonTeleportInInterval(toon):
hole = toon.getHoleActors()[0]
holeAnimTrack = Sequence()
holeAnimTrack.append(Func(toon.detachNode))
holeAnimTrack.append(Func(hole.reparentTo, toon))
pos = Point3(0, -2.4, 0)
holeAnimTrack.append(Func(hole.setPos, toon, pos))
holeAnimTrack.append(ActorInterval(hole, 'hole', startTime=T_TELEPORT_ANIM, endTime=T_HOLE_LEAVES_HAND))
holeAnimTrack.append(ActorInterval(hole, 'hole', startTime=T_HOLE_LEAVES_HAND, endTime=T_TELEPORT_ANIM))
holeAnimTrack.append(Func(hole.reparentTo, hidden))
delay = T_TELEPORT_ANIM - T_HOLE_LEAVES_HAND
jumpTrack = Sequence(Wait(delay), Func(toon.reparentTo, render), ActorInterval(toon, 'jump'))
return Parallel(holeAnimTrack, jumpTrack)
def getSuitRakeOffset(suit):
suitName = suit.getStyleName()
if suitName == 'gh':
return 1.4
elif suitName == 'f':
return 1.0
elif suitName == 'cc':
return 0.7
elif suitName == 'tw':
return 1.3
elif suitName == 'bf':
return 1.0
elif suitName == 'sc':
return 0.8
elif suitName == 'ym':
return 0.1
elif suitName == 'mm':
return 0.05
elif suitName == 'tm':
return 0.07
elif suitName == 'nd':
return 0.07
elif suitName == 'pp':
return 0.04
elif suitName == 'bc':
return 0.36
elif suitName == 'b':
return 0.41
elif suitName == 'dt':
return 0.31
elif suitName == 'ac':
return 0.39
elif suitName == 'ds':
return 0.41
elif suitName == 'hh':
return 0.8
elif suitName == 'cr':
return 2.1
elif suitName == 'tbc':
return 1.4
elif suitName == 'bs':
return 0.4
elif suitName == 'sd':
return 1.02
elif suitName == 'le':
return 1.3
elif suitName == 'bw':
return 1.4
elif suitName == 'nc':
return 0.6
elif suitName == 'mb':
return 1.85
elif suitName == 'ls':
return 1.4
elif suitName == 'rb':
return 1.6
elif suitName == 'ms':
return 0.7
elif suitName == 'tf':
return 0.75
elif suitName == 'm':
return 0.9
elif suitName == 'mh':
return 1.3
else:
notify.warning('getSuitRakeOffset(suit) - Unknown suit name: %s' % suitName)
return 0
def startSparksIval(tntProp):
tip = tntProp.find('**/joint_attachEmitter')
sparks = BattleParticles.createParticleEffect(file='tnt')
return Func(sparks.start, tip)
def indicateMissed(actor, duration = 1.1, scale = 0.7):
actor.showHpString(TTLocalizer.AttackMissed, duration=duration, scale=scale)
def createKapowExplosionTrack(parent, explosionPoint = None, scale = 1.0):
explosionTrack = Sequence()
explosion = loader.loadModel('phase_3.5/models/props/explosion.bam')
explosion.setBillboardPointEye()
explosion.setDepthWrite(False)
if not explosionPoint:
explosionPoint = Point3(0, 3.6, 2.1)
explosionTrack.append(Func(explosion.reparentTo, parent))
explosionTrack.append(Func(explosion.setPos, explosionPoint))
explosionTrack.append(Func(explosion.setScale, 0.4 * scale))
explosionTrack.append(Wait(0.6))
explosionTrack.append(Func(removeProp, explosion))
return explosionTrack
def createSuitStunInterval(suit, before, after):
p1 = Point3(0)
p2 = Point3(0)
stars = globalPropPool.getProp('stun')
stars.setColor(1, 1, 1, 1)
stars.adjustAllPriorities(100)
head = suit.getHeadParts()[0]
head.calcTightBounds(p1, p2)
return Sequence(Wait(before), Func(stars.reparentTo, head), Func(stars.setZ, max(0.0, p2[2] - 1.0)), Func(stars.loop, 'stun'), Wait(after), Func(stars.removeNode))
def calcAvgSuitPos(throw):
battle = throw['battle']
avgSuitPos = Point3(0, 0, 0)
numTargets = len(throw['target'])
for i in xrange(numTargets):
suit = throw['target'][i]['suit']
avgSuitPos += suit.getPos(battle)
avgSuitPos /= numTargets
return avgSuitPos