44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
from toontown.hood import HoodAI
|
|
from toontown.safezone import ButterflyGlobals
|
|
from toontown.safezone import DistributedButterflyAI
|
|
from toontown.safezone import DistributedTrolleyAI
|
|
from toontown.toonbase import ToontownGlobals
|
|
from toontown.ai import DistributedEffectMgrAI
|
|
|
|
class TTHoodAI(HoodAI.HoodAI):
|
|
def __init__(self, air):
|
|
HoodAI.HoodAI.__init__(self, air,
|
|
ToontownGlobals.ToontownCentral,
|
|
ToontownGlobals.ToontownCentral)
|
|
|
|
self.trolley = None
|
|
|
|
self.startup()
|
|
|
|
def startup(self):
|
|
HoodAI.HoodAI.startup(self)
|
|
|
|
if simbase.config.GetBool('want-minigames', True):
|
|
self.createTrolley()
|
|
if simbase.config.GetBool('want-butterflies', True):
|
|
self.createButterflies()
|
|
|
|
|
|
def shutdown(self):
|
|
HoodAI.HoodAI.shutdown(self)
|
|
ButterflyGlobals.clearIndexes(self.zoneId)
|
|
|
|
def createTrolley(self):
|
|
self.trolley = DistributedTrolleyAI.DistributedTrolleyAI(self.air)
|
|
self.trolley.generateWithRequired(self.zoneId)
|
|
self.trolley.start()
|
|
|
|
def createButterflies(self):
|
|
playground = ButterflyGlobals.TTC
|
|
ButterflyGlobals.generateIndexes(self.zoneId, ButterflyGlobals.TTC)
|
|
|
|
for i in xrange(0, ButterflyGlobals.NUM_BUTTERFLY_AREAS[ButterflyGlobals.TTC]):
|
|
for _ in xrange(0, ButterflyGlobals.NUM_BUTTERFLIES[ButterflyGlobals.TTC]):
|
|
butterfly = DistributedButterflyAI.DistributedButterflyAI(self.air, playground, i, self.zoneId)
|
|
butterfly.generateWithRequired(self.zoneId)
|
|
butterfly.start()
|