toontown-just-works/toontown/minigame/DistributedMazeGameAI.py

130 lines
4.8 KiB
Python
Raw Permalink Normal View History

2024-07-07 18:08:39 -05:00
from DistributedMinigameAI import *
from direct.fsm import ClassicFSM, State
from direct.fsm import State
import PatternGameGlobals
from direct.task.Task import Task
import MazeGameGlobals
import MazeData
class DistributedMazeGameAI(DistributedMinigameAI):
def __init__(self, air, minigameId):
try:
self.DistributedMinigameTemplateAI_initialized
except:
self.DistributedMinigameTemplateAI_initialized = 1
DistributedMinigameAI.__init__(self, air, minigameId)
self.gameFSM = ClassicFSM.ClassicFSM('DistributedMazeGameAI', [State.State('inactive', self.enterInactive, self.exitInactive, ['play']),
State.State('play', self.enterPlay, self.exitPlay, ['waitShowScores', 'cleanup']),
State.State('waitShowScores', self.enterWaitShowScores, self.exitWaitShowScores, ['cleanup']),
State.State('cleanup', self.enterCleanup, self.exitCleanup, ['inactive'])], 'inactive', 'inactive')
self.addChildGameFSM(self.gameFSM)
def delete(self):
self.notify.debug('delete')
del self.gameFSM
DistributedMinigameAI.delete(self)
def setGameReady(self):
self.notify.debug('setGameReady')
DistributedMinigameAI.setGameReady(self)
mazeName = MazeGameGlobals.getMazeName(self.doId, self.numPlayers, MazeData.mazeNames)
mData = MazeData.mazeData[mazeName]
self.numTreasures = len(mData['treasurePosList'])
self.numTreasuresTaken = 0
self.takenTable = [0] * self.numTreasures
for avId in self.scoreDict.keys():
self.scoreDict[avId] = 0
def setGameStart(self, timestamp):
self.notify.debug('setGameStart')
DistributedMinigameAI.setGameStart(self, timestamp)
self.gameFSM.request('play')
def setGameAbort(self):
self.notify.debug('setGameAbort')
if self.gameFSM.getCurrentState():
self.gameFSM.request('cleanup')
DistributedMinigameAI.setGameAbort(self)
def gameOver(self):
self.notify.debug('gameOver')
self.gameFSM.request('cleanup')
DistributedMinigameAI.gameOver(self)
def enterInactive(self):
self.notify.debug('enterInactive')
def exitInactive(self):
pass
def getTimeBase(self):
return self.__timeBase
def enterPlay(self):
self.notify.debug('enterPlay')
taskMgr.doMethodLater(MazeGameGlobals.GAME_DURATION, self.timerExpired, self.taskName('gameTimer'))
def exitPlay(self):
taskMgr.remove(self.taskName('gameTimer'))
def claimTreasure(self, treasureNum):
if self.gameFSM.getCurrentState() is None or self.gameFSM.getCurrentState().getName() != 'play':
return
avId = self.air.getAvatarIdFromSender()
if avId not in self.scoreDict:
self.notify.warning('PROBLEM: avatar %s called claimTreasure(%s) but he is not in the scoreDict: %s. avIdList is: %s' % (avId,
treasureNum,
self.scoreDict,
self.avIdList))
return
if treasureNum < 0 or treasureNum >= self.numTreasures:
self.air.writeServerEvent('warning', treasureNum, 'MazeGameAI.claimTreasure treasureNum out of range')
return
if self.takenTable[treasureNum]:
return
self.takenTable[treasureNum] = 1
avId = self.air.getAvatarIdFromSender()
self.sendUpdate('setTreasureGrabbed', [avId, treasureNum])
self.scoreDict[avId] += 1
self.numTreasuresTaken += 1
if self.numTreasuresTaken >= self.numTreasures:
self.logAllPerfect()
self.sendUpdate('allTreasuresTaken', [])
if not MazeGameGlobals.ENDLESS_GAME:
self.gameFSM.request('waitShowScores')
return
def timerExpired(self, task):
self.notify.debug('timer expired')
if not MazeGameGlobals.ENDLESS_GAME:
self.gameFSM.request('waitShowScores')
return Task.done
def enterWaitShowScores(self):
self.notify.debug('enterWaitShowScores')
taskMgr.doMethodLater(MazeGameGlobals.SHOWSCORES_DURATION, self.__doneShowingScores, self.taskName('waitShowScores'))
def __doneShowingScores(self, task):
self.notify.debug('doneShowingScores')
for key in self.scoreDict.keys():
self.scoreDict[key] = max(1, self.scoreDict[key] / 12)
if self.numTreasuresTaken >= self.numTreasures:
for key in self.scoreDict.keys():
self.scoreDict[key] += 8
self.gameOver()
return Task.done
def exitWaitShowScores(self):
taskMgr.remove(self.taskName('waitShowScores'))
def enterCleanup(self):
self.notify.debug('enterCleanup')
del self.takenTable
self.gameFSM.request('inactive')
def exitCleanup(self):
pass