348 lines
14 KiB
Python
348 lines
14 KiB
Python
import random
|
|
from direct.distributed.ClockDelta import globalClockDelta
|
|
from DistCogdoGameAI import DistCogdoGameAI
|
|
import CogdoFlyingGameGlobals as Globals
|
|
|
|
class DistCogdoFlyingGameAI(DistCogdoGameAI):
|
|
notify = directNotify.newCategory('DistCogdoFlyingGameAI')
|
|
EagleExitCooldownTaskName = 'CFG_EagleExitCooldownTask-%s'
|
|
InvulBuffRemoveTaskName = 'CFG_InvulBuffRemoveTask-%s'
|
|
AnnounceGameDoneTimerTaskName = 'CFG_AnnounceGameDoneTimerTask'
|
|
|
|
def __init__(self, air, id):
|
|
DistCogdoGameAI.__init__(self, air, id)
|
|
self.toonsOnEndPlatform = []
|
|
self.toonsInWinState = []
|
|
self.eagleId2targetIds = {}
|
|
self.pickups = []
|
|
self.memoCount = 0
|
|
self.toonId2buffList = {}
|
|
self.broadcastedGotoWinState = False
|
|
self.broadcastedGameFinished = False
|
|
self._gameState = False
|
|
if __debug__ and simbase.config.GetBool('schellgames-dev', True):
|
|
self.accept('onCodeReload', self._DistCogdoFlyingGameAI__sgOnCodeReload)
|
|
|
|
def getLegalEagleAttackRoundTime(self, fromCooldown = False):
|
|
time = 0.0
|
|
time += Globals.LegalEagle.LiftOffTime
|
|
time += Globals.LegalEagle.LockOnTime
|
|
time += Globals.LegalEagle.ChargeUpTime
|
|
time += Globals.LegalEagle.PreAttackTime
|
|
time += Globals.LegalEagle.PostAttackTime
|
|
time += Globals.LegalEagle.RetreatToSkyTime
|
|
time += Globals.LegalEagle.CooldownTime
|
|
if fromCooldown:
|
|
time += Globals.LegalEagle.ExtraPostCooldownTime
|
|
|
|
return time
|
|
|
|
def delete(self):
|
|
self.ignoreAll()
|
|
DistCogdoGameAI.delete(self)
|
|
|
|
def areAllToonsOnPlatform(self):
|
|
if self.getNumPlayers() == 0:
|
|
return False
|
|
|
|
return self.getNumPlayers() == len(self.toonsOnEndPlatform)
|
|
|
|
def areAllToonsInWinState(self):
|
|
if self.getNumPlayers() == 0:
|
|
return False
|
|
|
|
return self.getNumPlayers() == len(self.toonsInWinState)
|
|
|
|
def getCurrentNetworkTime(self):
|
|
return globalClockDelta.localToNetworkTime(globalClock.getRealTime())
|
|
|
|
def addBuff(self, toonId, pickupType):
|
|
if toonId in self.toonId2buffList:
|
|
buffList = self.toonId2buffList[toonId]
|
|
if pickupType in buffList:
|
|
return 0
|
|
else:
|
|
buffList.append(pickupType)
|
|
else:
|
|
self.toonId2buffList[toonId] = [
|
|
pickupType]
|
|
return 1
|
|
|
|
def removeBuff(self, toonId, pickupType):
|
|
if toonId in self.toonId2buffList:
|
|
buffList = self.toonId2buffList[toonId]
|
|
if pickupType in buffList:
|
|
buffList.remove(pickupType)
|
|
return 1
|
|
|
|
return 0
|
|
|
|
def isBuffOnToon(self, toonId, pickupType):
|
|
if toonId in self.toonId2buffList:
|
|
buffList = self.toonId2buffList[toonId]
|
|
if pickupType in buffList:
|
|
return 1
|
|
|
|
return 0
|
|
|
|
def updateGotoWinState(self):
|
|
if not (self.broadcastedGotoWinState) and self.areAllToonsOnPlatform():
|
|
self.forceDebuffAll()
|
|
self.broadcastedGotoWinState = True
|
|
self.d_broadcastDoAction(Globals.AI.GameActions.GotoWinState)
|
|
|
|
def updateGameFinished(self):
|
|
if not (self.broadcastedGameFinished) and self.areAllToonsInWinState():
|
|
self.broadcastedGameFinished = True
|
|
self._handleGameFinished()
|
|
|
|
def _isPlayingGame(self, senderId):
|
|
if self._gameState:
|
|
return True
|
|
elif self.fsm.getCurrentState() not in ('Finish',):
|
|
self._reportSuspiciousEvent(senderId, 'Client %s has made a game action request from illegal state: %s.' % (senderId, self.fsm.getCurrentState()))
|
|
|
|
return False
|
|
|
|
def requestAction(self, action, data):
|
|
senderId = self.air.getAvatarIdFromSender()
|
|
if not self._validateSenderId(senderId) or not self._isPlayingGame(senderId):
|
|
return False
|
|
|
|
av = simbase.air.doId2do.get(senderId)
|
|
if action == Globals.AI.GameActions.LandOnWinPlatform:
|
|
if senderId not in self.toonsOnEndPlatform:
|
|
self.toonsOnEndPlatform.append(senderId)
|
|
self.d_broadcastDoAction(action, senderId)
|
|
self.updateGotoWinState()
|
|
else:
|
|
self._reportSuspiciousEvent(senderId, 'Client %s has landed on the win platform twice.' % senderId)
|
|
elif action == Globals.AI.GameActions.WinStateFinished:
|
|
if senderId not in self.toonsInWinState:
|
|
self.toonsInWinState.append(senderId)
|
|
self.d_broadcastDoAction(action, senderId)
|
|
self.updateGameFinished()
|
|
else:
|
|
self._reportSuspiciousEvent(senderId, 'Client %s wants to exit from the win state multiple times.' % senderId)
|
|
elif action == Globals.AI.GameActions.HitWhirlwind:
|
|
if Globals.Dev.Invincibility != True:
|
|
self.damageToon(av, Globals.AI.SafezoneId2WhirlwindDamage)
|
|
|
|
elif action == Globals.AI.GameActions.HitLegalEagle:
|
|
if Globals.Dev.Invincibility != True:
|
|
self.damageToon(av, Globals.AI.SafezoneId2LegalEagleDamage)
|
|
|
|
elif action == Globals.AI.GameActions.HitMinion:
|
|
if Globals.Dev.Invincibility != True:
|
|
self.damageToon(av, Globals.AI.SafezoneId2MinionDamage)
|
|
|
|
elif action == Globals.AI.GameActions.RequestEnterEagleInterest:
|
|
self.b_toonSetAsEagleTarget(senderId, data, self.getCurrentNetworkTime())
|
|
elif action == Globals.AI.GameActions.RequestExitEagleInterest:
|
|
self.b_toonClearAsEagleTarget(senderId, data, self.getCurrentNetworkTime())
|
|
elif action == Globals.AI.GameActions.RanOutOfTimePenalty:
|
|
self.memoCount = 0
|
|
elif action == Globals.AI.GameActions.Died:
|
|
self.toonDied(av)
|
|
self.sendUpdate('toonDied', [
|
|
senderId,
|
|
self.getCurrentNetworkTime()])
|
|
elif action == Globals.AI.GameActions.Spawn:
|
|
self.sendUpdate('toonSpawn', [
|
|
senderId,
|
|
self.getCurrentNetworkTime()])
|
|
elif action == Globals.AI.GameActions.SetBlades:
|
|
if data in Globals.Gameplay.FuelStates:
|
|
self.sendUpdate('toonSetBlades', [
|
|
senderId,
|
|
data])
|
|
else:
|
|
self._reportSuspiciousEvent(senderId, "Client %s has requested a fuel state that doesn't exist:%s." % (senderId, data))
|
|
elif action == Globals.AI.GameActions.BladeLost:
|
|
self.sendUpdate('toonBladeLost', [
|
|
senderId])
|
|
else:
|
|
self._reportSuspiciousEvent(senderId, 'Client %s has made an illegal game action request: %s.' % (senderId, action))
|
|
|
|
def requestPickUp(self, pickupNum, pickupType):
|
|
senderId = self.air.getAvatarIdFromSender()
|
|
if not self._validateSenderId(senderId) or not self._isPlayingGame(senderId):
|
|
return False
|
|
|
|
if pickupType not in Globals.Level.GatherableTypes:
|
|
self._reportSuspiciousEvent(senderId, 'Client %s has requested an illegal pickup type: %s.' % (senderId, pickupType))
|
|
return False
|
|
|
|
if pickupType == Globals.Level.GatherableTypes.Memo:
|
|
if pickupNum not in self.pickups:
|
|
self.pickups.append(pickupNum)
|
|
self.memoCount += 1
|
|
self.d_broadcastPickup(senderId, pickupNum, self.getCurrentNetworkTime())
|
|
|
|
else:
|
|
self.d_broadcastPickup(senderId, pickupNum, self.getCurrentNetworkTime())
|
|
if pickupType == Globals.Level.GatherableTypes.LaffPowerup:
|
|
self.pickedUpLaffPowerup(senderId)
|
|
elif pickupType == Globals.Level.GatherableTypes.InvulPowerup:
|
|
self.pickedUpInvulPowerup(senderId)
|
|
|
|
def damageToon(self, av, damageDict):
|
|
safezoneId = self.getSafezoneId()
|
|
if safezoneId in damageDict:
|
|
av.takeDamage(damageDict[safezoneId], quietly = 0)
|
|
else:
|
|
self.notify.warning('Safezone Id: %s is not in flying game damage dictionary' % safezoneId)
|
|
|
|
def toonDied(self, av):
|
|
if Globals.Dev.Invincibility != True:
|
|
self.damageToon(av, Globals.AI.SafezoneId2DeathDamage)
|
|
|
|
def d_broadcastDoAction(self, action, data = 0):
|
|
self.sendUpdate('doAction', [
|
|
action,
|
|
data])
|
|
|
|
def b_broadcastDebuffPowerup(self, senderId, pickupType):
|
|
if self.removeBuff(senderId, pickupType):
|
|
self.d_broadcastDebuffPowerup(senderId, pickupType)
|
|
|
|
def d_broadcastDebuffPowerup(self, senderId, pickupType):
|
|
self.sendUpdate('debuffPowerup', [
|
|
senderId,
|
|
pickupType,
|
|
self.getCurrentNetworkTime()])
|
|
|
|
def pickedUpLaffPowerup(self, senderId):
|
|
av = simbase.air.doId2do.get(senderId)
|
|
if av:
|
|
if av.hp > 0 and av.hp < av.maxHp:
|
|
safezoneId = self.getSafezoneId()
|
|
if safezoneId in Globals.Gameplay.SafezoneId2LaffPickupHealAmount:
|
|
av.toonUp(Globals.Gameplay.SafezoneId2LaffPickupHealAmount[safezoneId])
|
|
else:
|
|
self.notify.warning('Safezone Id: %s is not in SafezoneId2LaffPickupHealAmount' % safezoneId)
|
|
|
|
def pickedUpInvulPowerup(self, senderId):
|
|
taskMgr.remove(self.uniqueName(DistCogdoFlyingGameAI.InvulBuffRemoveTaskName % senderId))
|
|
self.addBuff(senderId, Globals.Level.GatherableTypes.InvulPowerup)
|
|
taskMgr.doMethodLater(Globals.Gameplay.InvulBuffTime, self.b_broadcastDebuffPowerup, self.uniqueName(DistCogdoFlyingGameAI.InvulBuffRemoveTaskName % senderId), extraArgs = [
|
|
senderId,
|
|
Globals.Level.GatherableTypes.InvulPowerup])
|
|
|
|
def forceDebuffAll(self):
|
|
tasks = taskMgr.getTasksMatching(DistCogdoFlyingGameAI.InvulBuffRemoveTaskName % '*')
|
|
for task in tasks:
|
|
taskMgr.remove(task.getName())
|
|
|
|
for toonId in self.toonId2buffList:
|
|
buffList = self.toonId2buffList[toonId][:]
|
|
for buff in buffList:
|
|
self.b_broadcastDebuffPowerup(toonId, buff)
|
|
|
|
def d_broadcastPickup(self, senderId, pickupNum, networkTime):
|
|
self.sendUpdate('pickUp', [
|
|
senderId,
|
|
pickupNum,
|
|
networkTime])
|
|
|
|
def b_toonSetAsEagleTarget(self, toonId, eagleId, networkTime):
|
|
if eagleId not in self.eagleId2targetIds:
|
|
self.eagleId2targetIds[eagleId] = [
|
|
toonId]
|
|
if not taskMgr.hasTaskNamed(DistCogdoFlyingGameAI.EagleExitCooldownTaskName % eagleId):
|
|
taskMgr.doMethodLater(self.getLegalEagleAttackRoundTime(), self.d_broadcastEagleExitCooldown, self.uniqueName(DistCogdoFlyingGameAI.EagleExitCooldownTaskName % eagleId), extraArgs = [
|
|
eagleId])
|
|
|
|
self.d_broadcastToonSetAsEagleTarget(toonId, eagleId, networkTime)
|
|
else:
|
|
self.eagleId2targetIds[eagleId].append(toonId)
|
|
|
|
def b_toonClearAsEagleTarget(self, toonId, eagleId, networkTime):
|
|
if eagleId in self.eagleId2targetIds:
|
|
toonIds = self.eagleId2targetIds[eagleId]
|
|
toonIds.remove(toonId)
|
|
self.d_broadcastToonClearAsEagleTarget(toonId, eagleId, networkTime)
|
|
if len(toonIds) == 0:
|
|
del self.eagleId2targetIds[eagleId]
|
|
else:
|
|
index = random.randint(0, len(toonIds) - 1)
|
|
self.d_broadcastToonSetAsEagleTarget(toonIds[index], eagleId, networkTime)
|
|
|
|
def d_broadcastEagleExitCooldown(self, eagleId):
|
|
self.sendUpdate('eagleExitCooldown', [
|
|
eagleId,
|
|
self.getCurrentNetworkTime()])
|
|
if eagleId in self.eagleId2targetIds:
|
|
taskMgr.remove(self.uniqueName(DistCogdoFlyingGameAI.EagleExitCooldownTaskName % eagleId))
|
|
if eagleId in self.eagleId2targetIds:
|
|
taskMgr.doMethodLater(self.getLegalEagleAttackRoundTime(fromCooldown = True), self.d_broadcastEagleExitCooldown, self.uniqueName(DistCogdoFlyingGameAI.EagleExitCooldownTaskName % eagleId), extraArgs = [
|
|
eagleId])
|
|
|
|
def d_broadcastToonSetAsEagleTarget(self, toonId, eagleId, networkTime):
|
|
self.sendUpdate('toonSetAsEagleTarget', [
|
|
toonId,
|
|
eagleId,
|
|
networkTime])
|
|
|
|
def d_broadcastToonClearAsEagleTarget(self, toonId, eagleId, networkTime):
|
|
self.sendUpdate('toonClearAsEagleTarget', [
|
|
toonId,
|
|
eagleId,
|
|
networkTime])
|
|
|
|
def enterGame(self):
|
|
DistCogdoGameAI.enterGame(self)
|
|
self._gameState = True
|
|
|
|
def handleGameTimerExpired(self):
|
|
self._handleGameFinished()
|
|
|
|
def handleToonDisconnected(self, toonId):
|
|
DistCogdoGameAI.handleToonDisconnected(self, toonId)
|
|
if toonId in self.toonsOnEndPlatform:
|
|
self.toonsOnEndPlatform.remove(toonId)
|
|
|
|
if toonId in self.toonsInWinState:
|
|
self.toonsInWinState.remove(toonId)
|
|
|
|
self.updateGotoWinState()
|
|
self.updateGameFinished()
|
|
|
|
def handleToonWentSad(self, toonId):
|
|
DistCogdoGameAI.handleToonWentSad(self, toonId)
|
|
if toonId in self.toonsOnEndPlatform:
|
|
self.toonsOnEndPlatform.remove(toonId)
|
|
|
|
if toonId in self.toonsInWinState:
|
|
self.toonsInWinState.remove(toonId)
|
|
|
|
self.updateGotoWinState()
|
|
self.updateGameFinished()
|
|
|
|
def exitGame(self):
|
|
self._gameState = False
|
|
DistCogdoGameAI.exitGame(self)
|
|
if hasattr(self, '_gameTimerExpiredTask'):
|
|
taskMgr.remove(self._gameTimerExpiredTask)
|
|
del self._gameTimerExpiredTask
|
|
|
|
taskMgr.removeTasksMatching(DistCogdoFlyingGameAI.InvulBuffRemoveTaskName % '*')
|
|
taskMgr.removeTasksMatching(DistCogdoFlyingGameAI.EagleExitCooldownTaskName % '*')
|
|
for toonId in self.toonId2buffList:
|
|
del self.toonId2buffList[toonId][:]
|
|
|
|
self.toonId2buffList.clear()
|
|
self.eagleId2targetIds.clear()
|
|
del self.pickups[:]
|
|
del self.toonsOnEndPlatform[:]
|
|
del self.toonsInWinState[:]
|
|
|
|
def enterFinish(self):
|
|
DistCogdoGameAI.enterFinish(self)
|
|
self.ignoreAll()
|
|
self._announceGameDoneTask = taskMgr.doMethodLater(Globals.Gameplay.FinishDurationSeconds, self.announceGameDone, self.taskName(DistCogdoFlyingGameAI.AnnounceGameDoneTimerTaskName), [])
|
|
|
|
def exitFinish(self):
|
|
DistCogdoGameAI.exitFinish(self)
|
|
taskMgr.remove(self._announceGameDoneTask)
|
|
del self._announceGameDoneTask
|