52 lines
2.1 KiB
Python
52 lines
2.1 KiB
Python
import pickle
|
|
|
|
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):
|
|
trackId, period = record
|
|
if trackId not in self.records:
|
|
return
|
|
|
|
self.records[trackId][period] = [(x[0], x[3]) for x in self.air.raceMgr.getRecords(trackId, period)]
|
|
|
|
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', [pickle.dumps((trackName, periodName, leaderList))])
|