48 lines
1.9 KiB
Python
48 lines
1.9 KiB
Python
import cPickle
|
|
|
|
from direct.directnotify import DirectNotifyGlobal
|
|
from direct.distributed.DistributedObjectAI import DistributedObjectAI
|
|
|
|
from toontown.toonbase import TTLocalizer
|
|
|
|
|
|
class DistributedLeaderBoardAI(DistributedObjectAI):
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('DistributedLeaderBoardAI')
|
|
|
|
def __init__(self, air, name, x, y, z, h, p, r):
|
|
DistributedObjectAI.__init__(self, air)
|
|
self.name = name
|
|
self.posHpr = (x, y, z, h, p, r)
|
|
self.records = {}
|
|
self.subscriptions = []
|
|
self.currentIndex = -1
|
|
|
|
def announceGenerate(self):
|
|
DistributedObjectAI.announceGenerate(self)
|
|
self.accept('UpdateRaceRecord', self.handleUpdateRaceRecord)
|
|
self.accept('GS_LeaderBoardSwap' + str(self.zoneId), self.updateDisplay)
|
|
|
|
def getName(self):
|
|
return self.name
|
|
|
|
def getPosHpr(self):
|
|
return self.posHpr
|
|
|
|
def subscribeTo(self, subscription):
|
|
self.records.setdefault(subscription[0], {})[subscription[1]] = [(x[0], x[3]) for x in
|
|
self.air.raceMgr.getRecords(subscription[0],
|
|
subscription[1])]
|
|
self.subscriptions.append(subscription)
|
|
|
|
def handleUpdateRaceRecord(self, record):
|
|
self.notify.info('handleUpdateRaceRecord TODO')
|
|
|
|
def updateDisplay(self):
|
|
self.currentIndex += 1
|
|
if self.currentIndex >= len(self.subscriptions):
|
|
self.currentIndex = 0
|
|
|
|
trackName = TTLocalizer.KartRace_TrackNames[self.subscriptions[self.currentIndex][0]]
|
|
periodName = TTLocalizer.RecordPeriodStrings[self.subscriptions[self.currentIndex][1]]
|
|
leaderList = self.records[self.subscriptions[self.currentIndex][0]][self.subscriptions[self.currentIndex][1]]
|
|
self.sendUpdate('setDisplay', [cPickle.dumps((trackName, periodName, leaderList))])
|