mirror of
https://github.com/Sneed-Group/Poodletooth-iLand
synced 2024-12-25 12:42:41 -06:00
53 lines
2.3 KiB
Python
53 lines
2.3 KiB
Python
|
from otp.ai.AIBaseGlobal import *
|
||
|
|
from direct.distributed.ClockDelta import *
|
||
|
|
from direct.directnotify import DirectNotifyGlobal
|
||
|
|
from direct.fsm import ClassicFSM
|
||
|
|
from toontown.building import DistributedDoorAI
|
||
|
|
from direct.fsm import State
|
||
|
|
from toontown.toonbase import ToontownGlobals
|
||
|
|
import CogDisguiseGlobals
|
||
|
|
from toontown.building import FADoorCodes
|
||
|
|
from toontown.building import DoorTypes
|
||
|
|
|
||
|
|
class DistributedCogHQDoorAI(DistributedDoorAI.DistributedDoorAI):
|
||
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('DistributedCogHQDoorAI')
|
||
|
|
|
||
|
|
def __init__(self, air, blockNumber, doorType, destinationZone, doorIndex = 0, lockValue = FADoorCodes.SB_DISGUISE_INCOMPLETE, swing = 3):
|
||
|
|
DistributedDoorAI.DistributedDoorAI.__init__(self, air, blockNumber, doorType, doorIndex, lockValue, swing)
|
||
|
|
self.destinationZone = destinationZone
|
||
|
|
|
||
|
|
def requestEnter(self):
|
||
|
|
avatarID = self.air.getAvatarIdFromSender()
|
||
|
|
dept = ToontownGlobals.cogHQZoneId2deptIndex(self.destinationZone)
|
||
|
|
av = self.air.doId2do.get(avatarID)
|
||
|
|
if av:
|
||
|
|
if self.doorType == DoorTypes.EXT_COGHQ and self.isLockedDoor():
|
||
|
|
parts = av.getCogParts()
|
||
|
|
if CogDisguiseGlobals.isSuitComplete(parts, dept):
|
||
|
|
allowed = 1
|
||
|
|
else:
|
||
|
|
allowed = 0
|
||
|
|
else:
|
||
|
|
allowed = 1
|
||
|
|
if not allowed:
|
||
|
|
self.sendReject(avatarID, self.isLockedDoor())
|
||
|
|
else:
|
||
|
|
self.enqueueAvatarIdEnter(avatarID)
|
||
|
|
self.sendUpdateToAvatarId(avatarID, 'setOtherZoneIdAndDoId', [self.destinationZone, self.otherDoor.getDoId()])
|
||
|
|
|
||
|
|
def requestExit(self):
|
||
|
|
avatarID = self.air.getAvatarIdFromSender()
|
||
|
|
if avatarID in self.avatarsWhoAreEntering:
|
||
|
|
del self.avatarsWhoAreEntering[avatarID]
|
||
|
|
if avatarID not in self.avatarsWhoAreExiting:
|
||
|
|
dept = ToontownGlobals.cogHQZoneId2deptIndex(self.destinationZone)
|
||
|
|
self.avatarsWhoAreExiting[avatarID] = 1
|
||
|
|
self.sendUpdate('avatarExit', [avatarID])
|
||
|
|
self.openDoor(self.exitDoorFSM)
|
||
|
|
if self.lockedDoor:
|
||
|
|
av = self.air.doId2do[avatarID]
|
||
|
|
if self.doorType == DoorTypes.EXT_COGHQ:
|
||
|
|
av.b_setCogIndex(-1)
|
||
|
|
else:
|
||
|
|
av.b_setCogIndex(dept)
|