mirror of
https://github.com/Sneed-Group/Poodletooth-iLand
synced 2024-10-31 16:57:54 +00:00
39 lines
1.6 KiB
Python
39 lines
1.6 KiB
Python
|
from direct.directnotify import DirectNotifyGlobal
|
||
|
|
from direct.fsm import ClassicFSM, State
|
||
|
|
from direct.fsm import State
|
||
|
|
from pandac.PandaModules import *
|
||
|
|
from toontown.battle import BattlePlace
|
||
|
|
from toontown.building import Elevator
|
||
|
|
from toontown.coghq import CogHQExterior
|
||
|
|
from toontown.dna.DNAParser import loadDNAFileAI, DNAStorage
|
||
|
|
from toontown.hood import ZoneUtil
|
||
|
|
from toontown.toonbase import ToontownGlobals
|
||
|
|
|
||
|
|
|
||
|
|
class LawbotHQExterior(CogHQExterior.CogHQExterior):
|
||
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('LawbotHQExterior')
|
||
|
|
|
||
|
|
def enter(self, requestStatus):
|
||
|
|
CogHQExterior.CogHQExterior.enter(self, requestStatus)
|
||
|
|
|
||
|
|
# Load the CogHQ DNA file:
|
||
|
|
dnaStore = DNAStorage()
|
||
|
|
dnaFileName = self.genDNAFileName(self.zoneId)
|
||
|
|
loadDNAFileAI(dnaStore, dnaFileName)
|
||
|
|
|
||
|
|
# Collect all of the vis group zone IDs:
|
||
|
|
self.zoneVisDict = {}
|
||
|
|
for i in xrange(dnaStore.getNumDNAVisGroupsAI()):
|
||
|
|
groupFullName = dnaStore.getDNAVisGroupName(i)
|
||
|
|
visGroup = dnaStore.getDNAVisGroupAI(i)
|
||
|
|
visZoneId = int(base.cr.hoodMgr.extractGroupName(groupFullName))
|
||
|
|
visZoneId = ZoneUtil.getTrueZoneId(visZoneId, self.zoneId)
|
||
|
|
visibles = []
|
||
|
|
for i in xrange(visGroup.getNumVisibles()):
|
||
|
|
visibles.append(int(visGroup.visibles[i]))
|
||
|
|
visibles.append(ZoneUtil.getBranchZone(visZoneId))
|
||
|
|
self.zoneVisDict[visZoneId] = visibles
|
||
|
|
|
||
|
|
# Next, we want interest in all vis groups due to this being a Cog HQ:
|
||
|
|
base.cr.sendSetZoneMsg(self.zoneId, self.zoneVisDict.values()[0])
|