class HoodAI:
"""
AI-side representation of everything in a single neighborhood.
One subclass of this class exists for every neighborhood in the game.
HoodAIs are responsible for spawning all TreasurePlanners, SuitPlanners,
ponds, and other safezone objects, etc.
"""
HOOD = None
def __init__(self, air):
self.air = air
self.spawnNpcsIn = [2000]
self.safezone = self.HOOD
self.streets = {}
self.trolley = None
self.pond = None
self.createSafeZone()
self.createStreets()
def createTreasurePlanner(self):
treasureType, healAmount, spawnPoints, spawnRate, maxTreasures = TreasureGlobals.SafeZoneTreasureSpawns[
self.HOOD]
self.treasurePlanner = SZTreasurePlannerAI(self.safezone, treasureType,
healAmount, spawnPoints,
spawnRate, maxTreasures)
self.treasurePlanner.start()
def createTrolley(self):
self.trolley = DistributedTrolleyAI(self.air)
self.trolley.generateWithRequired(self.safezone)
def createGate(self):
self.gate = DistributedPartyGateAI(self.air)
self.gate.generateWithRequired(self.safezone)
def createSafeZone(self):
self.createTrolley()
self.createTreasurePlanner()
self.createGate()
def createStreets(self):
branchIds = ToontownGlobals.HoodHierarchy.get(self.HOOD, [])
for branch in branchIds:
street = StreetAI(self.air, branch)
self.streets[branch] = street
def spawnObjects(self):
filename = self.air.genDNAFileName(self.safezone)
DNASpawnerAI().spawnObjects(filename, self.safezone)
class HoodAI:
"""
AI-side representation of everything in a single neighborhood.
One subclass of this class exists for every neighborhood in the game.
HoodAIs are responsible for spawning all TreasurePlanners, SuitPlanners,
ponds, and other safezone objects, etc.
"""
HOOD = None
def __init__(self, air):
self.air = air
self.spawnNpcsIn = [2000]
self.safezone = self.HOOD
self.streets = {}
self.trolley = None
self.pond = None
self.createSafeZone()
self.createStreets()
def createTreasurePlanner(self):
treasureType, healAmount, spawnPoints, spawnRate, maxTreasures = TreasureGlobals.SafeZoneTreasureSpawns[self.HOOD]
self.treasurePlanner = SZTreasurePlannerAI(self.safezone, treasureType, healAmount, spawnPoints, spawnRate, maxTreasures)
self.treasurePlanner.start()
def createTrolley(self):
self.trolley = DistributedTrolleyAI(self.air)
self.trolley.generateWithRequired(self.safezone)
def createGate(self):
self.gate = DistributedPartyGateAI(self.air)
self.gate.generateWithRequired(self.safezone)
def createSafeZone(self):
self.createTrolley()
self.createTreasurePlanner()
self.createGate()
def createStreets(self):
branchIds = ToontownGlobals.HoodHierarchy.get(self.HOOD, [])
for branch in branchIds:
street = StreetAI(self.air, branch)
self.streets[branch] = street
def spawnObjects(self):
filename = self.air.genDNAFileName(self.safezone)
DNASpawnerAI().spawnObjects(filename, self.safezone)
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 _createObjects(self, group, zone):
if group.getName()[:13] == 'fishing_pond_':
visGroup = group.getVisGroup()
pondZone = 0
if visGroup is None:
pondZone = zone
else:
pondZone = int(visGroup.getName().split(':')[0])
pondIndex = int(group.getName()[13:])
pond = DistributedFishingPondAI(simbase.air)
pond.setArea(zone)
pond.generateWithRequired(pondZone)
bingoManager = DistributedPondBingoManagerAI(simbase.air)
bingoManager.setPondDoId(pond.getDoId())
bingoManager.generateWithRequired(pondZone)
#temporary, until we have scheduled stuff
bingoManager.createGame()
pond.bingoMgr = bingoManager
simbase.air.fishManager.ponds[zone] = pond
for i in range(FishingTargetGlobals.getNumTargets(zone)):
target = DistributedFishingTargetAI(simbase.air)
target.setPondDoId(pond.getDoId())
target.generateWithRequired(pondZone)
for i in range(group.getNumChildren()):
posSpot = group.at(i)
if posSpot.getName()[:13] == 'fishing_spot_':
posSpot = group.atAsNode(i)
spot = DistributedFishingSpotAI(simbase.air)
spot.setPondDoId(pond.getDoId())
x, y, z = posSpot.getPos()
h, p, r = posSpot.getHpr()
spot.setPosHpr(x, y, z, h, p, r)
spot.generateWithRequired(pondZone)
NPCToons.createNpcsInZone(simbase.air, pondZone)
elif group.getName()[:10] == 'racing_pad':
index, dest = group.getName()[11:].split('_', 2)
index = int(index)
pad = DistributedRacePadAI(simbase.air)
pad.setArea(zone)
pad.nameType = dest
pad.index = index
nri = RaceGlobals.getNextRaceInfo(-1, dest, index)
pad.setTrackInfo([nri[0], nri[1]])
pad.generateWithRequired(zone)
for i in range(group.getNumChildren()):
posSpot = group.at(i)
if posSpot.getName()[:14] == 'starting_block':
spotIndex = int(posSpot.getName()[15:])
posSpot = group.atAsNode(i)
x, y, z = posSpot.getPos()
h, p, r = posSpot.getHpr()
startingBlock = DistributedStartingBlockAI(simbase.air)
startingBlock.setPosHpr(x, y, z, h, p, r)
startingBlock.setPadDoId(pad.getDoId())
startingBlock.setPadLocationId(index)
startingBlock.generateWithRequired(zone)
pad.addStartingBlock(startingBlock)
elif group.getName()[:11] == 'viewing_pad':
pad = DistributedViewPadAI(simbase.air)
pad.setArea(zone)
pad.generateWithRequired(zone)
for i in range(group.getNumChildren()):
posSpot = group.at(i)
if posSpot.getName()[:14] == 'starting_block':
spotIndex = int(posSpot.getName()[15:])
posSpot = group.atAsNode(i)
x, y, z = posSpot.getPos()
h, p, r = posSpot.getHpr()
startingBlock = DistributedViewingBlockAI(simbase.air)
startingBlock.setPosHpr(x, y, z, h, p, r)
startingBlock.setPadDoId(pad.getDoId())
startingBlock.setPadLocationId(0)
startingBlock.generateWithRequired(zone)
pad.addStartingBlock(startingBlock)
elif group.getName()[:13] == 'picnic_table_' and zone != 7000:
pos = group.getPos()
hpr = group.getHpr()
nameInfo = group.getName().split('_')
picnicTable = DistributedPicnicBasketAI.DistributedPicnicBasketAI(
simbase.air, nameInfo[2], pos[0], pos[1], pos[2], hpr[0],
hpr[1], hpr[2])
picnicTable.generateWithRequired(zone)
picnicTable.start()
elif group.getName() == 'prop_game_table_DNARoot' and config.GetBool(
'want-oz-game-tables', True):
pos = group.getPos()
hpr = group.getHpr()
nameInfo = group.getName().split('_')
tableIndex = int(group.parent.getName().split('_')[-1])
picnicTable = DistributedPicnicTableAI.DistributedPicnicTableAI(
simbase.air, zone, nameInfo[2], pos[0], pos[1], pos[2], hpr[0],
hpr[1], hpr[2])
picnicTable.setTableIndex(tableIndex)
picnicTable.generateOtpObject(
simbase.air.districtId, zone,
['setX', 'setY', 'setZ', 'setH', 'setP', 'setR'])
elif group.getName()[:9] == 'golf_kart' and config.GetBool(
'want-golf-karts', False):
info = group.getName()[10:].split('_')
golfCourse = int(info[0])
kartId = info[1]
for i in range(group.getNumChildren()):
prop = group.at(i)
if prop.getName()[:15] == 'starting_block_':
pos, hpr = (prop.getPos(), prop.getHpr())
kart = DistributedGolfKartAI.DistributedGolfKartAI(
simbase.air, golfCourse, pos[0], pos[1], pos[2], hpr[0],
hpr[1], hpr[2])
kart.generateWithRequired(zone)
kart.sendUpdate('setGolfCourse', [golfCourse])
kart.sendUpdate('setPosHpr',
[pos[0], pos[1], pos[2], hpr[0], hpr[1], hpr[2]])
color = kart.getColor()
kart.sendUpdate('setColor', [color[0], color[1], color[2]])
kart.start()
if group.getName()[:15] == 'prop_party_gate':
gate = DistributedPartyGateAI(simbase.air)
gate.setArea(zone)
gate.generateWithRequired(zone)
for i in range(group.getNumChildren()):
child = group.at(i)
self._createObjects(child, zone)
def createGate(self):
self.gate = DistributedPartyGateAI(self.air)
self.gate.generateWithRequired(self.safezone)
def _createObjects(self, group, zone):
if group.getName()[:13] == 'fishing_pond_':
visGroup = group.getVisGroup()
pondZone = 0
if visGroup is None:
pondZone = zone
else:
pondZone = int(visGroup.getName().split(':')[0])
pondIndex = int(group.getName()[13:])
pond = DistributedFishingPondAI(simbase.air)
pond.setArea(zone)
pond.generateWithRequired(pondZone)
#self.ponds[pondIndex] = pond
bingoManager = DistributedPondBingoManagerAI(simbase.air)
bingoManager.setPondDoId(pond.getDoId())
bingoManager.generateWithRequired(pondZone)
#temporary, until we have scheduled stuff
bingoManager.createGame()
pond.bingoMgr = bingoManager
simbase.air.fishManager.ponds[zone] = pond
for i in range(FishingTargetGlobals.getNumTargets(zone)):
target = DistributedFishingTargetAI(simbase.air)
target.setPondDoId(pond.getDoId())
target.generateWithRequired(pondZone)
for i in range(group.getNumChildren()):
posSpot = group.at(i)
if posSpot.getName()[:13] == 'fishing_spot_':
spot = DistributedFishingSpotAI(simbase.air)
spot.setPondDoId(pond.getDoId())
x, y, z = posSpot.getPos()
h, p, r = posSpot.getHpr()
spot.setPosHpr(x, y, z, h, p, r)
spot.generateWithRequired(pondZone)
NPCToons.createNpcsInZone(simbase.air, pondZone)
elif group.getName()[:10] == 'racing_pad':
index, dest = group.getName()[11:].split('_', 2)
index = int(index)
pad = DistributedRacePadAI(simbase.air)
pad.setArea(zone)
pad.nameType = dest
pad.index = index
nri = RaceGlobals.getNextRaceInfo(-1, dest, index)
pad.setTrackInfo([nri[0], nri[1]])
pad.generateWithRequired(zone)
for i in range(group.getNumChildren()):
posSpot = group.at(i)
if posSpot.getName()[:14] == 'starting_block':
spotIndex = int(posSpot.getName()[15:])
x, y, z = posSpot.getPos()
h, p, r = posSpot.getHpr()
startingBlock = DistributedStartingBlockAI(simbase.air)
startingBlock.setPosHpr(x, y, z, h, p, r)
startingBlock.setPadDoId(pad.getDoId())
startingBlock.setPadLocationId(index)
startingBlock.generateWithRequired(zone)
pad.addStartingBlock(startingBlock)
elif group.getName()[:11] == 'viewing_pad':
pad = DistributedViewPadAI(simbase.air)
pad.setArea(zone)
pad.generateWithRequired(zone)
for i in range(group.getNumChildren()):
posSpot = group.at(i)
if posSpot.getName()[:14] == 'starting_block':
spotIndex = int(posSpot.getName()[15:])
x, y, z = posSpot.getPos()
h, p, r = posSpot.getHpr()
startingBlock = DistributedViewingBlockAI(simbase.air)
startingBlock.setPosHpr(x, y, z, h, p, r)
startingBlock.setPadDoId(pad.getDoId())
startingBlock.setPadLocationId(0)
startingBlock.generateWithRequired(zone)
pad.addStartingBlock(startingBlock)
if group.getName()[:15] == 'prop_party_gate':
gate = DistributedPartyGateAI(simbase.air)
gate.setArea(zone)
gate.generateWithRequired(zone)
for i in range(group.getNumChildren()):
self._createObjects(group.at(i), zone)
def _createObjects(self, group, zone):
if group.getName()[:13] == "fishing_pond_":
visGroup = group.getVisGroup()
pondZone = 0
if visGroup is None:
pondZone = zone
else:
pondZone = int(visGroup.getName().split(":")[0])
pondIndex = int(group.getName()[13:])
pond = DistributedFishingPondAI(simbase.air)
pond.setArea(zone)
pond.generateWithRequired(pondZone)
bingoManager = DistributedPondBingoManagerAI(simbase.air)
bingoManager.setPondDoId(pond.getDoId())
bingoManager.generateWithRequired(pondZone)
# temporary, until we have scheduled stuff
bingoManager.createGame()
pond.bingoMgr = bingoManager
simbase.air.fishManager.ponds[zone] = pond
for i in range(FishingTargetGlobals.getNumTargets(zone)):
target = DistributedFishingTargetAI(simbase.air)
target.setPondDoId(pond.getDoId())
target.generateWithRequired(pondZone)
for i in range(group.getNumChildren()):
posSpot = group.at(i)
if posSpot.getName()[:13] == "fishing_spot_":
posSpot = group.atAsNode(i)
spot = DistributedFishingSpotAI(simbase.air)
spot.setPondDoId(pond.getDoId())
x, y, z = posSpot.getPos()
h, p, r = posSpot.getHpr()
spot.setPosHpr(x, y, z, h, p, r)
spot.generateWithRequired(pondZone)
NPCToons.createNpcsInZone(simbase.air, pondZone)
elif group.getName()[:10] == "racing_pad":
index, dest = group.getName()[11:].split("_", 2)
index = int(index)
pad = DistributedRacePadAI(simbase.air)
pad.setArea(zone)
pad.nameType = dest
pad.index = index
nri = RaceGlobals.getNextRaceInfo(-1, dest, index)
pad.setTrackInfo([nri[0], nri[1]])
pad.generateWithRequired(zone)
for i in range(group.getNumChildren()):
posSpot = group.at(i)
if posSpot.getName()[:14] == "starting_block":
spotIndex = int(posSpot.getName()[15:])
posSpot = group.atAsNode(i)
x, y, z = posSpot.getPos()
h, p, r = posSpot.getHpr()
startingBlock = DistributedStartingBlockAI(simbase.air)
startingBlock.setPosHpr(x, y, z, h, p, r)
startingBlock.setPadDoId(pad.getDoId())
startingBlock.setPadLocationId(index)
startingBlock.generateWithRequired(zone)
pad.addStartingBlock(startingBlock)
elif group.getName()[:11] == "viewing_pad":
pad = DistributedViewPadAI(simbase.air)
pad.setArea(zone)
pad.generateWithRequired(zone)
for i in range(group.getNumChildren()):
posSpot = group.at(i)
if posSpot.getName()[:14] == "starting_block":
spotIndex = int(posSpot.getName()[15:])
posSpot = group.atAsNode(i)
x, y, z = posSpot.getPos()
h, p, r = posSpot.getHpr()
startingBlock = DistributedViewingBlockAI(simbase.air)
startingBlock.setPosHpr(x, y, z, h, p, r)
startingBlock.setPadDoId(pad.getDoId())
startingBlock.setPadLocationId(0)
startingBlock.generateWithRequired(zone)
pad.addStartingBlock(startingBlock)
elif group.getName()[:13] == "picnic_table_" and zone != 7000:
pos = group.getPos()
hpr = group.getHpr()
nameInfo = group.getName().split("_")
picnicTable = DistributedPicnicBasketAI.DistributedPicnicBasketAI(
simbase.air, nameInfo[2], pos[0], pos[1], pos[2], hpr[0], hpr[1], hpr[2]
)
picnicTable.generateWithRequired(zone)
picnicTable.start()
elif group.getName() == "prop_game_table_DNARoot" and config.GetBool("want-oz-game-tables", True):
pos = group.getPos()
hpr = group.getHpr()
nameInfo = group.getName().split("_")
tableIndex = int(group.parent.getName().split("_")[-1])
picnicTable = DistributedPicnicTableAI.DistributedPicnicTableAI(
simbase.air, zone, nameInfo[2], pos[0], pos[1], pos[2], hpr[0], hpr[1], hpr[2]
)
picnicTable.setTableIndex(tableIndex)
picnicTable.generateOtpObject(
simbase.air.districtId, zone, ["setX", "setY", "setZ", "setH", "setP", "setR"]
)
elif group.getName()[:9] == "golf_kart" and config.GetBool("want-golf-karts", True):
info = group.getName()[10:].split("_")
golfCourse = int(info[0])
kartId = info[1]
for i in range(group.getNumChildren()):
prop = group.at(i)
if prop.getName()[:15] == "starting_block_":
pos, hpr = (prop.getPos(), prop.getHpr())
kart = DistributedGolfKartAI.DistributedGolfKartAI(
simbase.air, golfCourse, pos[0], pos[1], pos[2], hpr[0], hpr[1], hpr[2]
)
kart.generateWithRequired(zone)
kart.sendUpdate("setGolfCourse", [golfCourse])
kart.sendUpdate("setPosHpr", [pos[0], pos[1], pos[2], hpr[0], hpr[1], hpr[2]])
color = kart.getColor()
kart.sendUpdate("setColor", [color[0], color[1], color[2]])
kart.start()
if group.getName()[:15] == "prop_party_gate" and simbase.air.wantParties:
gate = DistributedPartyGateAI(simbase.air)
gate.setArea(zone)
gate.generateWithRequired(zone)
for i in range(group.getNumChildren()):
child = group.at(i)
self._createObjects(child, zone)
def createGate(self):
self.gate = DistributedPartyGateAI(self.air)
self.gate.generateWithRequired(self.safezone)