def __fireworksTick(self, task):
# The next tick will occur in exactly an hour.
task.delayTime = 3600
showName = config.GetString('hourly-fireworks-type', 'july4')
if showName == 'july4':
showType = ToontownGlobals.JULY4_FIREWORKS
elif showName == 'newyears':
showType = ToontownGlobals.NEWYEARS_FIREWORKS
elif showName == 'summer':
showType = PartyGlobals.FireworkShows.Summer
elif showName == 'random':
shows = [ToontownGlobals.JULY4_FIREWORKS, ToontownGlobals.NEWYEARS_FIREWORKS, PartyGlobals.FireworkShows.Summer]
showType = random.choice(shows)
else:
raise AttributeError('%s is an invalid firework type' % showName)
return
numShows = len(FireworkShows.shows.get(showType, []))
showIndex = random.randint(0, numShows - 1)
for hood in self.air.hoods:
if hood.HOOD == ToontownGlobals.GolfZone:
continue
fireworksShow = DistributedFireworkShowAI(self.air)
fireworksShow.generateWithRequired(hood.HOOD)
fireworksShow.b_startShow(showType, showIndex, globalClockDelta.getRealNetworkTime())
return task.again
def startFireworks(self, type, task=None):
maxShow = len(FireworkShows.shows.get(type, [])) - 1
for hood in self.air.hoods:
if hood.zoneId == ToontownGlobals.GolfZone:
continue
fireworkShow = DistributedFireworkShowAI(self.air)
fireworkShow.generateWithRequired(hood.zoneId)
fireworkShow.b_startShow(type, random.randint(0, maxShow), globalClockDelta.getRealNetworkTime())
return Task.again
def startFireworks(self, task):
allFwTypes = [PartyGlobals.FireworkShows.Summer, ToontownGlobals.JULY4_FIREWORKS]
fwType = allFwTypes[random.randint(0, len(allFwTypes)-1)]
numShows = len(FireworkShows.shows.get(fwType, []))
showIndex = random.randint(0, numShows-1)
for hood in self.hoods:
if hood.safezone == ToontownGlobals.GolfZone:
continue
fwShow = DistributedFireworkShowAI(self)
fwShow.generateWithRequired(hood.safezone)
fwShow.b_startShow(fwType, showIndex, globalClockDelta.getRealNetworkTime())
task.delayTime = 3600
return task.again
def startFireworks(self, type, task=None):
maxShow = len(FireworkShows.shows.get(type, [])) - 1
for hood in self.air.hoods:
if hood.zoneId == ToontownGlobals.GolfZone:
continue
fireworkShow = DistributedFireworkShowAI(self.air)
fireworkShow.generateWithRequired(hood.zoneId)
fireworkShow.b_startShow(type, random.randint(0, maxShow),
globalClockDelta.getRealNetworkTime())
return Task.again
def __fireworksTick(self, task):
task.delayTime = 3600
showName = config.GetString('hourly-fireworks-type', 'july4')
if self.isHolidayRunning(ToontownGlobals.JULY4_FIREWORKS):
showType = ToontownGlobals.JULY4_FIREWORKS
else:
if self.isHolidayRunning(ToontownGlobals.NEWYEARS_FIREWORKS):
showType = ToontownGlobals.NEWYEARS_FIREWORKS
else:
if self.isHolidayRunning(ToontownGlobals.JULY14_FIREWORKS) or self.isHolidayRunning(ToontownGlobals.OCTOBER31_FIREWORKS) or self.isHolidayRunning(ToontownGlobals.NOVEMBER19_FIREWORKS) or self.isHolidayRunning(ToontownGlobals.FEBRUARY14_FIREWORKS):
showType = PartyGlobals.FireworkShows.Summer
else:
if self.isHolidayRunning(ToontownGlobals.COMBO_FIREWORKS):
shows = [
ToontownGlobals.JULY4_FIREWORKS, ToontownGlobals.NEWYEARS_FIREWORKS, PartyGlobals.FireworkShows.Summer]
showType = random.choice(shows)
else:
if showName == 'july4':
showType = ToontownGlobals.JULY4_FIREWORKS
else:
if showName == 'newyears':
showType = ToontownGlobals.NEWYEARS_FIREWORKS
else:
if showName == 'summer':
showType = PartyGlobals.FireworkShows.Summer
else:
if showName == 'random':
shows = [
ToontownGlobals.JULY4_FIREWORKS, ToontownGlobals.NEWYEARS_FIREWORKS, PartyGlobals.FireworkShows.Summer]
showType = random.choice(shows)
else:
raise AttributeError('%s is an invalid firework type' % showName)
return
numShows = len(FireworkShows.shows.get(showType, []))
showIndex = random.randint(0, numShows - 1)
for hood in self.air.hoods:
if hood.HOOD == ToontownGlobals.GolfZone:
continue
fireworksShow = DistributedFireworkShowAI(self.air)
fireworksShow.generateWithRequired(hood.HOOD)
fireworksShow.b_startShow(showType, showIndex, globalClockDelta.getRealNetworkTime())
return task.again
def fireworksTick(self, task):
# The next tick will occur in exactly an hour.
task.delayTime = 3600
#showName = config.GetString('hourly-fireworks-type', 'july4')
#if showName == 'july4':
# showType = ToontownGlobals.JULY4_FIREWORKS
show = HolidayGlobals.IsItFireworks()
if show == 'Nyear':
showType = ToontownGlobals.NEWYEARS_FIREWORKS
elif show == 'Summer':
showType = PartyGlobals.FireworkShows.Summer
elif show == 'Release':
showType = ToontownGlobals.VICTORY_RELEASE_FIREWORKS
elif show == 'None':
return
# elif showName == 'random':
# shows = [
# ToontownGlobals.JULY4_FIREWORKS,
# ToontownGlobals.NEWYEARS_FIREWORKS,
# PartyGlobals.FireworkShows.Summer,
# ToontownGlobals.VICTORY_RELEASE_FIREWORKS]
# showType = random.choice(shows)
numShows = len(FireworkShows.shows.get(showType, []))
showIndex = random.randint(0, numShows - 1)
for hood in self.air.hoods:
if hood.HOOD == ToontownGlobals.GolfZone:
continue
fireworksShow = DistributedFireworkShowAI(self.air)
fireworksShow.generateWithRequired(hood.HOOD)
fireworksShow.b_startShow(
showType, showIndex, globalClockDelta.getRealNetworkTime())
self.notify.info(
'Oh! A fireworks show has started in this District - next one in exactly 1 hours time!')
return task.again
def generateFireworkShow(self):
self.fireworkShow = DistributedFireworkShowAI(self.air)
self.fireworkShow.generateWithRequired(self.zoneId)
class HoodAI:
notify = directNotify.newCategory('HoodAI')
notify.setInfo(True)
def __init__(self, air, zoneId, canonicalHoodId):
self.air = air
self.zoneId = zoneId
self.canonicalHoodId = canonicalHoodId
self.fishingPonds = []
self.partyGates = []
self.treasurePlanner = None
self.buildingManagers = []
self.suitPlanners = []
for zoneId in self.getZoneTable():
self.notify.info('Creating objects... ' + self.getLocationName(zoneId))
dnaFileName = self.air.lookupDNAFileName(zoneId)
dnaStore = DNAStorage()
dnaData = simbase.air.loadDNAFileAI(dnaStore, dnaFileName)
self.air.dnaStoreMap[zoneId] = dnaStore
self.air.dnaDataMap[zoneId] = dnaData
def getZoneTable(self):
zoneTable = [self.zoneId]
zoneTable.extend(ToontownGlobals.HoodHierarchy.get(self.canonicalHoodId, []))
return zoneTable
def getLocationName(self, zoneId):
lookupTable = ToontownGlobals.hoodNameMap
isStreet = (zoneId%1000) != 0
if isStreet:
lookupTable = TTLocalizer.GlobalStreetNames
name = lookupTable.get(zoneId, '')
if isStreet:
return '%s, %s' % (self.getLocationName(self.zoneId), name[2])
return name[2]
def startup(self):
if self.air.wantFishing:
self.createFishingPonds()
if self.air.wantParties:
self.createPartyPeople()
if simbase.config.GetBool('want-treasure-planners', True):
self.createTreasurePlanner()
self.createBuildingManagers()
if simbase.config.GetBool('want-suit-planners', True):
self.createSuitPlanners()
if simbase.air.wantHalloween or simbase.air.holidayManager.isHolidayRunning(ToontownGlobals.TRICK_OR_TREAT):
self.startupTrickOrTreat()
if simbase.air.wantChristmas or simbase.air.holidayManager.isHolidayRunning(ToontownGlobals.WINTER_CAROLING):
self.startupWinterCaroling()
if simbase.air.wantFireworks:
self.generateFireworkShow()
def shutdown(self):
if self.treasurePlanner:
self.treasurePlanner.stop()
self.treasurePlanner.deleteAllTreasuresNow()
self.treasurePlanner = None
for suitPlanner in self.suitPlanners:
suitPlanner.requestDelete()
del self.air.suitPlanners[suitPlanner.zoneId]
self.suitPlanners = []
for buildingManager in self.buildingManagers:
buildingManager.cleanup()
del self.air.buildingManagers[buildingManager.branchId]
self.buildingManagers = []
del self.fishingPonds
for distObj in self.doId2do.values():
distObj.requestDelete()
def findFishingPonds(self, dnaGroup, zoneId, area):
fishingPonds = []
fishingPondGroups = []
if isinstance(dnaGroup, DNAGroup) and ('fishing_pond' in dnaGroup.getName()):
fishingPondGroups.append(dnaGroup)
fishingPond = DistributedFishingPondAI(simbase.air)
fishingPond.setArea(area)
fishingPond.generateWithRequired(zoneId)
fishingPond.start()
fishingPonds.append(fishingPond)
elif isinstance(dnaGroup, DNAVisGroup):
zoneId = ZoneUtil.getTrueZoneId(int(dnaGroup.getName().split(':')[0]), zoneId)
for i in xrange(dnaGroup.getNumChildren()):
(foundFishingPonds, foundFishingPondGroups) = self.findFishingPonds(dnaGroup.at(i), zoneId, area)
fishingPonds.extend(foundFishingPonds)
fishingPondGroups.extend(foundFishingPondGroups)
return (fishingPonds, fishingPondGroups)
def findFishingSpots(self, dnaGroup, fishingPond):
fishingSpots = []
if isinstance(dnaGroup, DNAGroup) and ('fishing_spot' in dnaGroup.getName()):
fishingSpot = DistributedFishingSpotAI(simbase.air)
fishingSpot.setPondDoId(fishingPond.doId)
x, y, z = dnaGroup.getPos()
h, p, r = dnaGroup.getHpr()
fishingSpot.setPosHpr(x, y, z, h, p, r)
fishingSpot.generateWithRequired(fishingPond.zoneId)
fishingSpots.append(fishingSpot)
for i in xrange(dnaGroup.getNumChildren()):
foundFishingSpots = self.findFishingSpots(dnaGroup.at(i), fishingPond)
fishingSpots.extend(foundFishingSpots)
return fishingSpots
def createFishingPonds(self):
self.fishingPonds = []
fishingPondGroups = []
for zoneId in self.getZoneTable():
dnaData = self.air.dnaDataMap.get(zoneId, None)
zoneId = ZoneUtil.getTrueZoneId(zoneId, self.zoneId)
if dnaData.getName() == 'root':
area = ZoneUtil.getCanonicalZoneId(zoneId)
(foundFishingPonds, foundFishingPondGroups) = self.findFishingPonds(dnaData, zoneId, area)
self.fishingPonds.extend(foundFishingPonds)
fishingPondGroups.extend(foundFishingPondGroups)
for fishingPond in self.fishingPonds:
NPCToons.createNpcsInZone(self.air, fishingPond.zoneId)
fishingSpots = []
for (dnaGroup, fishingPond) in zip(fishingPondGroups, self.fishingPonds):
fishingSpots.extend(self.findFishingSpots(dnaGroup, fishingPond))
for fishingPond in self.fishingPonds:
fishingPond.bingoMgr = DistributedPondBingoManagerAI(self.air)
fishingPond.bingoMgr.setPondDoId(fishingPond.doId)
fishingPond.bingoMgr.generateWithRequired(self.zoneId)
def findPartyGates(self, dnaGroup, zoneId):
partyGates = []
if isinstance(dnaGroup, DNAGroup) and ('prop_party_gate' in dnaGroup.getName()):
partyGate = DistributedPartyGateAI(simbase.air)
partyGate.setArea(zoneId)
partyGate.generateWithRequired(zoneId)
partyGates.append(partyGates)
for i in xrange(dnaGroup.getNumChildren()):
foundPartyGates = self.findPartyGates(dnaGroup.at(i), zoneId)
partyGates.extend(foundPartyGates)
return partyGates
def createPartyPeople(self):
self.partyGates = []
for zoneId in self.getZoneTable():
dnaData = self.air.dnaDataMap.get(zoneId, None)
zoneId = ZoneUtil.getTrueZoneId(zoneId, self.zoneId)
if dnaData.getName() == 'root':
foundPartyGates = self.findPartyGates(dnaData, zoneId)
self.partyGates.extend(foundPartyGates)
def createTreasurePlanner(self):
spawnInfo = TreasureGlobals.SafeZoneTreasureSpawns.get(self.canonicalHoodId)
if not spawnInfo:
return
treasureType, healAmount, spawnPoints, spawnRate, maxTreasures = spawnInfo
self.treasurePlanner = SZTreasurePlannerAI(
self.canonicalHoodId, treasureType, healAmount, spawnPoints,
spawnRate, maxTreasures)
self.treasurePlanner.start()
def createBuildingManagers(self):
for zoneId in self.getZoneTable():
dnaStore = self.air.dnaStoreMap[zoneId]
zoneId = ZoneUtil.getTrueZoneId(zoneId, self.zoneId)
buildingManager = DistributedBuildingMgrAI.DistributedBuildingMgrAI(
self.air, zoneId, dnaStore, self.air.trophyMgr)
self.buildingManagers.append(buildingManager)
self.air.buildingManagers[zoneId] = buildingManager
def createSuitPlanners(self):
for zoneId in self.getZoneTable():
if zoneId == self.zoneId:
continue
zoneId = ZoneUtil.getTrueZoneId(zoneId, self.zoneId)
suitPlanner = DistributedSuitPlannerAI.DistributedSuitPlannerAI(self.air, zoneId)
suitPlanner.generateWithRequired(zoneId)
suitPlanner.d_setZoneId(zoneId)
suitPlanner.initTasks()
self.suitPlanners.append(suitPlanner)
self.air.suitPlanners[zoneId] = suitPlanner
def startupTrickOrTreat(self):
if hasattr(self, 'TrickOrTreatManager'):
return
if self.canonicalHoodId in HOLIDAY_SHOPKEEPER_ZONES[ToontownGlobals.TRICK_OR_TREAT]:
self.TrickOrTreatManager = DistributedTrickOrTreatTargetAI.DistributedTrickOrTreatTargetAI(self.air)
self.TrickOrTreatManager.generateWithRequired(
HOLIDAY_SHOPKEEPER_ZONES[ToontownGlobals.TRICK_OR_TREAT][self.canonicalHoodId]
)
def endTrickOrTreat(self):
if hasattr(self, 'TrickOrTreatManager'):
self.TrickOrTreatManager.requestDelete()
del self.TrickOrTreatManager
def startupWinterCaroling(self):
if hasattr(self, 'WinterCarolingManager'):
return
if self.canonicalHoodId in HOLIDAY_SHOPKEEPER_ZONES[ToontownGlobals.WINTER_CAROLING]:
self.WinterCarolingManager = DistributedWinterCarolingTargetAI.DistributedWinterCarolingTargetAI(self.air)
self.WinterCarolingManager.generateWithRequired(
HOLIDAY_SHOPKEEPER_ZONES[ToontownGlobals.WINTER_CAROLING][self.canonicalHoodId]
)
def endWinterCaroling(self):
if hasattr(self, 'WinterCarolingManager'):
self.WinterCarolingManager.requestDelete()
del self.WinterCarolingManager
def generateFireworkShow(self):
self.fireworkShow = DistributedFireworkShowAI(self.air)
self.fireworkShow.generateWithRequired(self.zoneId)
def startFireworks(self, showType, showIndex):
self.fireworkShow.b_startShow(showType, showIndex, globalClockDelta.getRealNetworkTime())