oldschool-toontown/toontown/suit/DistributedSellbotBossAI.py

454 lines
17 KiB
Python
Raw Normal View History

2019-11-02 17:27:54 -05:00
from otp.ai.AIBaseGlobal import *
from direct.distributed.ClockDelta import *
import DistributedBossCogAI
from direct.directnotify import DirectNotifyGlobal
from otp.avatar import DistributedAvatarAI
import DistributedSuitAI
from toontown.battle import BattleExperienceAI
from direct.fsm import FSM
from toontown.toonbase import ToontownGlobals
from toontown.toon import InventoryBase
from toontown.toonbase import TTLocalizer
from toontown.battle import BattleBase
from toontown.toon import NPCToons
from toontown.suit import SellbotBossGlobals
import SuitDNA, random
class DistributedSellbotBossAI(DistributedBossCogAI.DistributedBossCogAI, FSM.FSM):
notify = DirectNotifyGlobal.directNotify.newCategory('DistributedSellbotBossAI')
limitHitCount = 6
numPies = ToontownGlobals.FullPies
def __init__(self, air):
DistributedBossCogAI.DistributedBossCogAI.__init__(self, air, 's')
FSM.FSM.__init__(self, 'DistributedSellbotBossAI')
self.doobers = []
self.nerfed = ToontownGlobals.SELLBOT_NERF_HOLIDAY in self.air.holidayManager.currentHolidays
if self.nerfed:
self.bossMaxDamage = ToontownGlobals.SellbotBossMaxDamageNerfed
self.pieHitToonup = SellbotBossGlobals.PieToonupNerfed
self.pieDamageMult = SellbotBossGlobals.PieDamageMultNerfed
self.hitCountDamage = SellbotBossGlobals.HitCountDamageNerfed
else:
self.bossMaxDamage = ToontownGlobals.SellbotBossMaxDamage
self.pieHitToonup = SellbotBossGlobals.PieToonup
self.pieDamageMult = SellbotBossGlobals.PieDamageMult
self.hitCountDamage = SellbotBossGlobals.HitCountDamage
self.recoverRate = 0
self.recoverStartTime = 0
def generateWithRequired(self, zoneId):
self.numRentalDiguises, self.numNormalDiguises = self.countDisguises()
self.__setCagedToonNpcId()
DistributedBossCogAI.DistributedBossCogAI.generateWithRequired(self, zoneId)
def delete(self):
self.destroyEasyModeBarrels()
return DistributedBossCogAI.DistributedBossCogAI.delete(self)
def getHoodId(self):
return ToontownGlobals.SellbotHQ
def getCagedToonNpcId(self):
return self.cagedToonNpcId
def __setCagedToonNpcId(self):
def npcFriendsMaxStars(stars):
return [ id for id in NPCToons.npcFriends.keys() if NPCToons.getNPCTrackLevelHpRarity(id)[3] <= stars ]
if self.numRentalDiguises >= 4:
self.cagedToonNpcId = random.choice(NPCToons.npcFriendsMinMaxStars(3, 3))
else:
if 1 <= self.numRentalDiguises <= 3:
self.cagedToonNpcId = random.choice(NPCToons.npcFriendsMinMaxStars(3, 4))
else:
self.cagedToonNpcId = random.choice(NPCToons.npcFriendsMinMaxStars(3, 5))
def magicWordHit(self, damage, avId):
if self.attackCode != ToontownGlobals.BossCogDizzyNow:
self.hitBossInsides()
self.hitBoss(damage)
def hitBoss(self, bossDamage):
avId = self.air.getAvatarIdFromSender()
if not self.validate(avId, avId in self.involvedToons, 'DistributedSellbotBossAI.hitBoss from unknown avatar'):
return
self.validate(avId, bossDamage == 1, 'invalid bossDamage %s' % bossDamage)
bossDamage = int(round(bossDamage * self.pieDamageMult))
if bossDamage < 1:
return
currState = self.getCurrentOrNextState()
if currState != 'BattleThree':
return
if self.attackCode != ToontownGlobals.BossCogDizzyNow:
return
bossDamage = min(self.getBossDamage() + bossDamage, self.bossMaxDamage)
self.b_setBossDamage(bossDamage, 0, 0)
if self.bossDamage >= self.bossMaxDamage:
self.setState('NearVictory')
else:
self.__recordHit()
def hitBossInsides(self):
avId = self.air.getAvatarIdFromSender()
if not self.validate(avId, avId in self.involvedToons, 'hitBossInsides from unknown avatar'):
return
currState = self.getCurrentOrNextState()
if currState != 'BattleThree':
return
self.b_setAttackCode(ToontownGlobals.BossCogDizzyNow)
self.b_setBossDamage(self.getBossDamage(), 0, 0)
def hitToon(self, toonId):
avId = self.air.getAvatarIdFromSender()
if not self.validate(avId, avId != toonId, 'hitToon on self'):
return
if avId not in self.involvedToons or toonId not in self.involvedToons:
return
toon = self.air.doId2do.get(toonId)
if toon:
self.healToon(toon, self.pieHitToonup)
def getDamageMultiplier(self):
if self.nerfed:
return SellbotBossGlobals.AttackMultNerfed
else:
return SellbotBossGlobals.AttackMult
def touchCage(self):
avId = self.air.getAvatarIdFromSender()
currState = self.getCurrentOrNextState()
if currState != 'BattleThree' and currState != 'NearVictory':
return
if not self.validate(avId, avId in self.involvedToons, 'touchCage from unknown avatar'):
return
toon = simbase.air.doId2do.get(avId)
if toon:
toon.b_setNumPies(self.numPies)
toon.__touchedCage = 1
self.__goodJump(avId)
def finalPieSplat(self):
if self.state != 'NearVictory':
return
self.b_setState('Victory')
def doNextAttack(self, task):
if self.attackCode == ToontownGlobals.BossCogDizzyNow:
attackCode = ToontownGlobals.BossCogRecoverDizzyAttack
else:
attackCode = random.choice([ToontownGlobals.BossCogAreaAttack, ToontownGlobals.BossCogFrontAttack, ToontownGlobals.BossCogDirectedAttack, ToontownGlobals.BossCogDirectedAttack, ToontownGlobals.BossCogDirectedAttack, ToontownGlobals.BossCogDirectedAttack])
if attackCode == ToontownGlobals.BossCogAreaAttack:
self.__doAreaAttack()
else:
if attackCode == ToontownGlobals.BossCogDirectedAttack:
self.__doDirectedAttack()
else:
self.b_setAttackCode(attackCode)
def __doAreaAttack(self):
self.b_setAttackCode(ToontownGlobals.BossCogAreaAttack)
if self.recoverRate:
newRecoverRate = min(200, self.recoverRate * 1.2)
else:
newRecoverRate = 2
now = globalClock.getFrameTime()
self.b_setBossDamage(self.getBossDamage(), newRecoverRate, now)
def __doDirectedAttack(self):
if self.nearToons:
toonId = random.choice(self.nearToons)
self.b_setAttackCode(ToontownGlobals.BossCogDirectedAttack, toonId)
else:
self.__doAreaAttack()
def b_setBossDamage(self, bossDamage, recoverRate, recoverStartTime):
self.d_setBossDamage(bossDamage, recoverRate, recoverStartTime)
self.setBossDamage(bossDamage, recoverRate, recoverStartTime)
def setBossDamage(self, bossDamage, recoverRate, recoverStartTime):
self.bossDamage = bossDamage
self.recoverRate = recoverRate
self.recoverStartTime = recoverStartTime
def getBossDamage(self):
now = globalClock.getFrameTime()
elapsed = now - self.recoverStartTime
return int(max(self.bossDamage - self.recoverRate * elapsed / 60.0, 0))
def d_setBossDamage(self, bossDamage, recoverRate, recoverStartTime):
timestamp = globalClockDelta.localToNetworkTime(recoverStartTime)
self.sendUpdate('setBossDamage', [bossDamage, recoverRate, timestamp])
def waitForNextStrafe(self, delayTime):
currState = self.getCurrentOrNextState()
if currState == 'BattleThree':
taskName = self.uniqueName('NextStrafe')
taskMgr.remove(taskName)
taskMgr.doMethodLater(delayTime, self.doNextStrafe, taskName)
def stopStrafes(self):
taskName = self.uniqueName('NextStrafe')
taskMgr.remove(taskName)
def doNextStrafe(self, task):
if self.attackCode != ToontownGlobals.BossCogDizzyNow:
side = random.choice([0, 1])
direction = random.choice([0, 1])
self.sendUpdate('doStrafe', [side, direction])
delayTime = 9
self.waitForNextStrafe(delayTime)
def __sendDooberIds(self):
dooberIds = []
for suit in self.doobers:
dooberIds.append(suit.doId)
self.sendUpdate('setDooberIds', [dooberIds])
def d_cagedToonBattleThree(self, index, avId):
self.sendUpdate('cagedToonBattleThree', [index, avId])
def formatReward(self):
return str(self.cagedToonNpcId)
def makeBattleOneBattles(self):
self.postBattleState = 'RollToBattleTwo'
self.initializeBattles(1, ToontownGlobals.SellbotBossBattleOnePosHpr)
def generateSuits(self, battleNumber):
if self.nerfed:
if battleNumber == 1:
return self.invokeSuitPlanner(15, 0)
else:
return self.invokeSuitPlanner(16, 1)
else:
if battleNumber == 1:
return self.invokeSuitPlanner(9, 0)
else:
return self.invokeSuitPlanner(10, 1)
def removeToon(self, avId):
toon = simbase.air.doId2do.get(avId)
if toon:
toon.b_setNumPies(0)
DistributedBossCogAI.DistributedBossCogAI.removeToon(self, avId)
def enterOff(self):
DistributedBossCogAI.DistributedBossCogAI.enterOff(self)
self.__resetDoobers()
def enterElevator(self):
DistributedBossCogAI.DistributedBossCogAI.enterElevator(self)
self.b_setBossDamage(0, 0, 0)
if self.nerfed:
self.createEasyModeBarrels()
def enterIntroduction(self):
DistributedBossCogAI.DistributedBossCogAI.enterIntroduction(self)
self.__makeDoobers()
self.b_setBossDamage(0, 0, 0)
def exitIntroduction(self):
DistributedBossCogAI.DistributedBossCogAI.exitIntroduction(self)
self.__resetDoobers()
def enterRollToBattleTwo(self):
self.divideToons()
self.barrier = self.beginBarrier('RollToBattleTwo', self.involvedToons, 45, self.__doneRollToBattleTwo)
def __doneRollToBattleTwo(self, avIds):
self.b_setState('PrepareBattleTwo')
def exitRollToBattleTwo(self):
self.ignoreBarrier(self.barrier)
def enterPrepareBattleTwo(self):
self.barrier = self.beginBarrier('PrepareBattleTwo', self.involvedToons, 30, self.__donePrepareBattleTwo)
self.makeBattleTwoBattles()
def __donePrepareBattleTwo(self, avIds):
self.b_setState('BattleTwo')
def exitPrepareBattleTwo(self):
self.ignoreBarrier(self.barrier)
def makeBattleTwoBattles(self):
self.postBattleState = 'PrepareBattleThree'
self.initializeBattles(2, ToontownGlobals.SellbotBossBattleTwoPosHpr)
def enterBattleTwo(self):
if self.battleA:
self.battleA.startBattle(self.toonsA, self.suitsA)
if self.battleB:
self.battleB.startBattle(self.toonsB, self.suitsB)
def exitBattleTwo(self):
self.resetBattles()
def enterPrepareBattleThree(self):
self.barrier = self.beginBarrier('PrepareBattleThree', self.involvedToons, 30, self.__donePrepareBattleThree)
def __donePrepareBattleThree(self, avIds):
self.b_setState('BattleThree')
def exitPrepareBattleThree(self):
self.ignoreBarrier(self.barrier)
def enterBattleThree(self):
self.resetBattles()
self.setPieType()
self.b_setBossDamage(0, 0, 0)
self.battleThreeStart = globalClock.getFrameTime()
for toonId in self.involvedToons:
toon = simbase.air.doId2do.get(toonId)
if toon:
toon.__touchedCage = 0
self.waitForNextAttack(5)
self.waitForNextStrafe(9)
self.cagedToonDialogIndex = 100
self.__saySomethingLater()
def __saySomething(self, task=None):
index = None
avId = 0
if len(self.involvedToons) == 0:
return
avId = random.choice(self.involvedToons)
toon = simbase.air.doId2do.get(avId)
if toon.__touchedCage:
if self.cagedToonDialogIndex <= TTLocalizer.CagedToonBattleThreeMaxAdvice:
index = self.cagedToonDialogIndex
self.cagedToonDialogIndex += 1
elif random.random() < 0.2:
index = random.randrange(100, TTLocalizer.CagedToonBattleThreeMaxAdvice + 1)
else:
index = random.randrange(20, TTLocalizer.CagedToonBattleThreeMaxTouchCage + 1)
if index:
self.d_cagedToonBattleThree(index, avId)
self.__saySomethingLater()
return
def __saySomethingLater(self, delayTime=15):
taskName = self.uniqueName('CagedToonSaySomething')
taskMgr.remove(taskName)
taskMgr.doMethodLater(delayTime, self.__saySomething, taskName)
def __goodJump(self, avId):
currState = self.getCurrentOrNextState()
if currState != 'BattleThree':
return
index = random.randrange(10, TTLocalizer.CagedToonBattleThreeMaxGivePies + 1)
self.d_cagedToonBattleThree(index, avId)
self.__saySomethingLater()
def exitBattleThree(self):
self.stopAttacks()
self.stopStrafes()
taskName = self.uniqueName('CagedToonSaySomething')
taskMgr.remove(taskName)
def enterNearVictory(self):
self.resetBattles()
def exitNearVictory(self):
pass
def enterVictory(self):
self.resetBattles()
self.suitsKilled.append({'type': None, 'level': None, 'track': self.dna.dept, 'isSkelecog': 0, 'isForeman': 0, 'isVP': 1, 'isCFO': 0, 'isSupervisor': 0, 'isVirtual': 0, 'activeToons': self.involvedToons[:]})
self.barrier = self.beginBarrier('Victory', self.involvedToons, 10, self.__doneVictory)
return
def __doneVictory(self, avIds):
self.d_setBattleExperience()
self.b_setState('Reward')
BattleExperienceAI.assignRewards(self.involvedToons, self.toonSkillPtsGained, self.suitsKilled, ToontownGlobals.dept2cogHQ(self.dept), self.helpfulToons)
for toonId in self.involvedToons:
toon = self.air.doId2do.get(toonId)
if toon:
configMax = simbase.config.GetInt('max-sos-cards', 16)
if configMax == 8:
maxNumCalls = 1
else:
maxNumCalls = 2
if not toon.attemptAddNPCFriend(self.cagedToonNpcId, numCalls=maxNumCalls):
self.notify.info('%s.unable to add NPCFriend %s to %s.' % (self.doId, self.cagedToonNpcId, toonId))
if self.__shouldPromoteToon(toon):
toon.b_promote(self.deptIndex)
self.sendUpdateToAvatarId(toonId, 'toonPromoted', [1])
else:
self.sendUpdateToAvatarId(toonId, 'toonPromoted', [0])
def __shouldPromoteToon(self, toon):
if not toon.readyForPromotion(self.deptIndex):
return False
else:
if self.isToonWearingRentalSuit(toon.doId):
return False
return True
def exitVictory(self):
self.takeAwayPies()
def enterFrolic(self):
DistributedBossCogAI.DistributedBossCogAI.enterFrolic(self)
self.b_setBossDamage(0, 0, 0)
def __resetDoobers(self):
for suit in self.doobers:
suit.requestDelete()
self.doobers = []
def __makeDoobers(self):
self.__resetDoobers()
for i in range(8):
suit = DistributedSuitAI.DistributedSuitAI(self.air, None)
level = random.randrange(len(SuitDNA.suitsPerLevel))
suit.dna = SuitDNA.SuitDNA()
suit.dna.newSuitRandom(level=level, dept=self.dna.dept)
suit.setLevel(level)
suit.generateWithRequired(self.zoneId)
self.doobers.append(suit)
self.__sendDooberIds()
return
def setPieType(self):
for toonId in self.involvedToons:
toon = simbase.air.doId2do.get(toonId)
if toon:
toon.d_setPieType(4)
def takeAwayPies(self):
for toonId in self.involvedToons:
toon = simbase.air.doId2do.get(toonId)
if toon:
toon.b_setNumPies(0)
def __recordHit(self):
now = globalClock.getFrameTime()
self.hitCount += 1
if self.hitCount < self.limitHitCount or self.bossDamage < self.hitCountDamage:
return
self.b_setAttackCode(ToontownGlobals.BossCogRecoverDizzyAttack)
def createEasyModeBarrels(self):
self.barrels = []
for entId, entDef in SellbotBossGlobals.BarrelDefs.iteritems():
barrelType = entDef['type']
barrel = barrelType(self.air, entId)
SellbotBossGlobals.setBarrelAttr(barrel, entId)
barrel.generateWithRequired(self.zoneId)
self.barrels.append(barrel)
def destroyEasyModeBarrels(self):
if hasattr(self, 'barrels') and self.barrels:
for barrel in self.barrels:
barrel.requestDelete()
self.barrels = []