217 lines
7.5 KiB
Python
217 lines
7.5 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)
|
|
else:
|
|
if gameName == 'MineSweeper':
|
|
self.game = LaserGameMineSweeper.LaserGameMineSweeper(self.trapDisable, self.trapFire, self.sendField, self.setGrid)
|
|
else:
|
|
if gameName == 'Roll':
|
|
self.game = LaserGameRoll.LaserGameRoll(self.trapDisable, self.trapFire, self.sendField, self.setGrid)
|
|
else:
|
|
if 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 range(0, self.game.gridNumX):
|
|
for row in range(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):
|
|
if not self.enabled:
|
|
return
|
|
self.enabled = 0
|
|
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):
|
|
if not self.enabled:
|
|
return
|
|
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)
|