historical/toontown-classic.git/toontown/coghq/DistributedLaserFieldAI.py
2024-01-16 11:20:27 -06:00

211 lines
7.3 KiB
Python

from otp.ai.AIBase import *
from direct.interval.IntervalGlobal import *
from direct.directnotify import DirectNotifyGlobal
from direct.distributed import ClockDelta
from direct.task import Task
from otp.level import DistributedEntityAI
from otp.level import BasicEntities
from direct.directnotify import DirectNotifyGlobal
from toontown.coghq import BattleBlockerAI
from toontown.coghq import LaserGameMineSweeper
from toontown.coghq import LaserGameRoll
from toontown.coghq import LaserGameAvoid
from toontown.coghq import LaserGameDrag
import random
class DistributedLaserFieldAI(BattleBlockerAI.BattleBlockerAI, NodePath, BasicEntities.NodePathAttribs):
def __init__(self, level, entId):
BattleBlockerAI.BattleBlockerAI.__init__(self, level, entId)
node = hidden.attachNewNode('DistributedLaserFieldAI')
NodePath.__init__(self, node)
if not hasattr(self, 'switchId'):
self.switchId = 0
self.gridScale = 1
self.game = LaserGameRoll.LaserGameRoll(self.trapDisable, self.trapFire, self.sendField, self.setGrid)
if not hasattr(self, 'gridGame'):
self.gridGame = 'Roll'
self.enabled = 1
self.hasShownSuits = 0
self.healReady = 1
self.playedSound = 0
self.canButton = 1
self.title = 'MemTag: This is a laserField %s' % random.random()
def setGridGame(self, gameName):
if gameName == 'Random':
gameName = random.choice(['MineSweeper',
'Roll',
'Avoid',
'Drag'])
self.gridGame = gameName
if hasattr(self, 'game'):
self.game.delete()
self.game = None
if gameName == 'Drag':
self.game = LaserGameDrag.LaserGameDrag(self.trapDisable, self.trapFire, self.sendField, self.setGrid)
elif gameName == 'MineSweeper':
self.game = LaserGameMineSweeper.LaserGameMineSweeper(self.trapDisable, self.trapFire, self.sendField, self.setGrid)
elif gameName == 'Roll':
self.game = LaserGameRoll.LaserGameRoll(self.trapDisable, self.trapFire, self.sendField, self.setGrid)
elif gameName == 'Avoid':
self.game = LaserGameAvoid.LaserGameAvoid(self.trapDisable, self.trapFire, self.sendField, self.setGrid)
else:
self.game = LaserGameMineSweeper.LaserGameMineSweeper(self.trapDisable, self.trapFire, self.sendField, self.setGrid)
self.game.startGrid()
self.sendField()
self.sendUpdate('setGridGame', [gameName])
return
def generate(self):
BattleBlockerAI.BattleBlockerAI.generate(self)
if self.switchId != 0:
self.accept(self.getOutputEventName(self.switchId), self.reactToSwitch)
self.detectName = 'laserField %s' % self.doId
taskMgr.doMethodLater(1.0, self.__detect, self.detectName)
self.setPos(self.pos)
self.setHpr(self.hpr)
self.setGridGame(self.gridGame)
def registerBlocker(self):
BattleBlockerAI.BattleBlockerAI.registerBlocker(self)
self.hideSuits()
def delete(self):
taskMgr.remove(self.detectName)
self.ignoreAll()
self.game.delete()
self.game = None
BattleBlockerAI.BattleBlockerAI.delete(self)
return
def destroy(self):
self.notify.info('destroy entity(laserField) %s' % self.entId)
BattleBlockerAI.BattleBlockerAI.destroy(self)
def setGrid(self, gridNumX, gridNumY):
self.game.setGridSize(gridNumX, gridNumY)
def getGrid(self):
return (self.game.gridNumX, self.game.gridNumY)
def getField(self):
fieldData = []
fieldData.append(self.game.gridNumX)
fieldData.append(self.game.gridNumY)
for column in xrange(0, self.game.gridNumX):
for row in xrange(0, self.game.gridNumY):
fieldData.append(self.game.gridData[column][row])
return fieldData
def sendField(self):
self.sendUpdate('setField', [self.getField()])
def __detect(self, task):
isThereAnyToons = False
if hasattr(self, 'level'):
toonInRange = 0
for avId in self.level.presentAvIds:
if avId in self.air.doId2do:
av = self.air.doId2do[avId]
isThereAnyToons = True
distance = self.getDistance(av)
if isThereAnyToons:
taskMgr.doMethodLater(1.0, self.__detect, self.detectName)
self.__run()
return Task.done
def hit(self, hitX, hitY, oldX, oldY):
if self.enabled:
self.game.hit(hitX, hitY, oldX, oldY)
def __run(self):
pass
def __toonHit(self):
self.gridNumX = random.randint(1, 4)
self.gridNumY = random.randint(1, 4)
self.gridScale = random.randint(1, 4)
self.sendUpdate('setGrid', [self.gridNumX, self.gridNumY, self.gridScale])
def reactToSwitch(self, on):
if on and self.canButton:
self.trapDisable()
self.game.win()
def trapFire(self):
self.game.lose()
self.showSuits()
stage = self.air.getDo(self.level.stageDoId)
stage.resetPuzzelReward()
self.healReady = 0
if not __dev__ and 1:
self.canButton = 0
self.sendUpdate('setActiveLF', [0])
if not self.playedSound:
self.sendUpdate('setSuccess', [0])
self.playedSound = 1
def setBattleFinished(self):
print 'battle Finished'
BattleBlockerAI.BattleBlockerAI.setBattleFinished(self)
messenger.send(self.getOutputEventName(), [1])
self.switchFire()
def switchFire(self):
print 'switchFire'
if self.switchId != 0:
switch = self.level.getEntity(self.switchId)
if switch:
switch.setIsOn(1)
def trapDisable(self):
self.enabled = 0
suits = self.level.planner.battleCellId2suits.get(self.cellId)
messenger.send(self.getOutputEventName(), [1])
if self.hasShownSuits == 0:
for suit in suits:
suit.requestRemoval()
self.sendUpdate('setActiveLF', [0])
stage = self.air.getDo(self.level.stageDoId)
reward = stage.getPuzzelReward()
if self.healReady:
for avId in self.level.presentAvIds:
av = self.air.doId2do.get(avId)
if av:
av.toonUp(reward)
stage.increasePuzzelReward()
self.healReady = 0
if not self.playedSound:
self.sendUpdate('setSuccess', [1])
self.playedSound = 1
self.switchFire()
def hideSuits(self):
suits = self.level.planner.battleCellId2suits.get(self.cellId)
suitArray = []
for suit in suits:
suitArray.append(suit.doId)
if suitArray:
self.sendUpdate('hideSuit', [suitArray])
def showSuits(self):
if self.hasShownSuits == 0:
suits = self.level.planner.battleCellId2suits.get(self.cellId)
suitArray = []
for suit in suits:
suit.setVirtual()
suitArray.append(suit.doId)
if suitArray:
self.sendUpdate('showSuit', [suitArray])
self.hasShownSuits = 1
def addSuit(self, suit):
print 'Adding Suit %s' % suit.doId
BattleBlockerAI.BattleBlockerAI.addSuit(self, suit)