90 lines
2.7 KiB
Python
90 lines
2.7 KiB
Python
|
from otp.ai.AIBase import *
|
||
|
|
from toontown.toonbase.ToontownGlobals import *
|
||
|
|
from direct.distributed.ClockDelta import *
|
||
|
|
from direct.distributed import DistributedObjectAI
|
||
|
|
from direct.task import Task
|
||
|
|
PianoSpeeds = [1.0,
|
||
|
|
2.0,
|
||
|
|
3.0,
|
||
|
|
4.0,
|
||
|
|
5.0,
|
||
|
|
6.0,
|
||
|
|
8.0,
|
||
|
|
10.0,
|
||
|
|
12.0,
|
||
|
|
14.0,
|
||
|
|
16.0,
|
||
|
|
18.0]
|
||
|
|
PianoMaxSpeed = PianoSpeeds[len(PianoSpeeds) - 1]
|
||
|
|
PianoSlowDownFactor = 0.7
|
||
|
|
PianoSlowDownInterval = 10.0
|
||
|
|
PianoSlowDownMinimum = 0.1
|
||
|
|
|
||
|
|
class DistributedMMPianoAI(DistributedObjectAI.DistributedObjectAI):
|
||
|
|
|
||
|
|
def __init__(self, air):
|
||
|
|
DistributedObjectAI.DistributedObjectAI.__init__(self, air)
|
||
|
|
self.spinStartTime = 0.0
|
||
|
|
self.rpm = 0.0
|
||
|
|
self.degreesPerSecond = self.rpm / 60.0 * 360.0
|
||
|
|
self.offset = 0.0
|
||
|
|
self.direction = 1
|
||
|
|
return None
|
||
|
|
|
||
|
|
def delete(self):
|
||
|
|
DistributedObjectAI.DistributedObjectAI.delete(self)
|
||
|
|
|
||
|
|
def requestSpeedUp(self):
|
||
|
|
if self.rpm < PianoMaxSpeed:
|
||
|
|
for speed in PianoSpeeds:
|
||
|
|
if speed > self.rpm:
|
||
|
|
break
|
||
|
|
|
||
|
|
self.updateSpeed(speed, self.direction)
|
||
|
|
self.d_playSpeedUp(self.air.getAvatarIdFromSender())
|
||
|
|
self.__slowDownLater()
|
||
|
|
|
||
|
|
def requestChangeDirection(self):
|
||
|
|
rpm = self.rpm
|
||
|
|
if rpm == 0.0:
|
||
|
|
rpm = PianoSpeeds[0]
|
||
|
|
self.updateSpeed(rpm, -self.direction)
|
||
|
|
self.__slowDownLater()
|
||
|
|
self.d_playChangeDirection(self.air.getAvatarIdFromSender())
|
||
|
|
|
||
|
|
def d_setSpeed(self, rpm, offset, startTime):
|
||
|
|
self.sendUpdate('setSpeed', [rpm, offset, globalClockDelta.localToNetworkTime(startTime)])
|
||
|
|
|
||
|
|
def d_playSpeedUp(self, avId):
|
||
|
|
self.sendUpdate('playSpeedUp', [avId])
|
||
|
|
|
||
|
|
def d_playChangeDirection(self, avId):
|
||
|
|
self.sendUpdate('playChangeDirection', [avId])
|
||
|
|
|
||
|
|
def updateSpeed(self, rpm, direction):
|
||
|
|
now = globalClock.getRealTime()
|
||
|
|
heading = self.degreesPerSecond * (now - self.spinStartTime) + self.offset
|
||
|
|
self.rpm = rpm
|
||
|
|
self.direction = direction
|
||
|
|
self.degreesPerSecond = rpm / 60.0 * 360.0 * direction
|
||
|
|
self.offset = heading % 360.0
|
||
|
|
self.spinStartTime = now
|
||
|
|
self.d_setSpeed(self.rpm * self.direction, self.offset, self.spinStartTime)
|
||
|
|
|
||
|
|
def start(self):
|
||
|
|
return None
|
||
|
|
|
||
|
|
def __slowDownLater(self):
|
||
|
|
taskName = self.uniqueName('slowDown')
|
||
|
|
taskMgr.remove(taskName)
|
||
|
|
taskMgr.doMethodLater(PianoSlowDownInterval, self.__slowDown, taskName)
|
||
|
|
|
||
|
|
def __slowDown(self, task):
|
||
|
|
rpm = self.rpm * PianoSlowDownFactor
|
||
|
|
if rpm < PianoSlowDownMinimum:
|
||
|
|
self.updateSpeed(0.0, self.direction)
|
||
|
|
else:
|
||
|
|
self.updateSpeed(rpm, self.direction)
|
||
|
|
self.__slowDownLater()
|
||
|
|
return Task.done
|