oldschool-toontown/toontown/hood/GZHoodDataAI.py
2020-01-07 21:12:08 -05:00

179 lines
7.7 KiB
Python

from direct.directnotify import DirectNotifyGlobal
from . import HoodDataAI, ZoneUtil
from toontown.toonbase import ToontownGlobals
from toontown.racing import DistributedStartingBlockAI
from pandac.PandaModules import *
from libtoontown import *
from toontown.racing.RaceGlobals import *
from toontown.classicchars import DistributedGoofySpeedwayAI
from toontown.safezone import DistributedGolfKartAI
import string
if __debug__:
import pdb
class GZHoodDataAI(HoodDataAI.HoodDataAI):
notify = DirectNotifyGlobal.directNotify.newCategory('GZHoodDataAI')
def __init__(self, air, zoneId=None):
hoodId = ToontownGlobals.GolfZone
if zoneId == None:
zoneId = hoodId
HoodDataAI.HoodDataAI.__init__(self, air, zoneId, hoodId)
return
def startup(self):
HoodDataAI.HoodDataAI.startup(self)
self.createStartingBlocks()
self.createGolfKarts()
def cleanup(self):
taskMgr.removeTasksMatching(str(self) + '_leaderBoardSwitch')
for board in self.leaderBoards:
board.delete()
del self.leaderBoards
def createLeaderBoards(self):
self.leaderBoards = []
dnaStore = DNAStorage()
dnaData = simbase.air.loadDNAFileAI(dnaStore, simbase.air.lookupDNAFileName('goofy_speedway_sz.dna'))
if isinstance(dnaData, DNAData):
self.leaderBoards = self.air.findLeaderBoards(dnaData, self.zoneId)
for distObj in self.leaderBoards:
if distObj:
if distObj.getName().count('city'):
type = 'city'
else:
if distObj.getName().count('stadium'):
type = 'stadium'
else:
if distObj.getName().count('country'):
type = 'country'
for subscription in LBSubscription[type]:
distObj.subscribeTo(subscription)
self.addDistObj(distObj)
def __cycleLeaderBoards(self, task=None):
messenger.send('GS_LeaderBoardSwap')
taskMgr.doMethodLater(self.cycleDuration, self.__cycleLeaderBoards, str(self) + '_leaderBoardSwitch')
def createStartingBlocks(self):
self.racingPads = []
self.viewingPads = []
self.viewingBlocks = []
self.startingBlocks = []
self.foundRacingPadGroups = []
self.foundViewingPadGroups = []
self.golfKartPads = []
self.golfKartPadGroups = []
for zone in self.air.zoneTable[self.canonicalHoodId]:
zoneId = ZoneUtil.getTrueZoneId(zone[0], self.zoneId)
dnaData = self.air.dnaDataMap.get(zone[0], None)
if isinstance(dnaData, DNAData):
area = ZoneUtil.getCanonicalZoneId(zoneId)
foundRacingPads, foundRacingPadGroups = self.air.findRacingPads(dnaData, zoneId, area, overrideDNAZone=True)
foundViewingPads, foundViewingPadGroups = self.air.findRacingPads(dnaData, zoneId, area, type='viewing_pad', overrideDNAZone=True)
self.racingPads += foundRacingPads
self.foundRacingPadGroups += foundRacingPadGroups
self.viewingPads += foundViewingPads
self.foundViewingPadGroups += foundViewingPadGroups
self.startingBlocks = []
for dnaGroup, distRacePad in zip(self.foundRacingPadGroups, self.racingPads):
startingBlocks = self.air.findStartingBlocks(dnaGroup, distRacePad)
self.startingBlocks += startingBlocks
for startingBlock in startingBlocks:
distRacePad.addStartingBlock(startingBlock)
for distObj in self.startingBlocks:
self.addDistObj(distObj)
for dnaGroup, distViewPad in zip(self.foundViewingPadGroups, self.viewingPads):
startingBlocks = self.air.findStartingBlocks(dnaGroup, distViewPad)
for viewingBlock in self.viewingBlocks:
distViewPad.addStartingBlock(viewingBlocks)
for distObj in self.viewingBlocks:
self.addDistObj(distObj)
for viewPad in self.viewingPads:
self.addDistObj(viewPad)
for racePad in self.racingPads:
racePad.request('WaitEmpty')
self.addDistObj(racePad)
return
def findAndCreateGolfKarts(self, dnaGroup, zoneId, area, overrideDNAZone = 0, type = 'golf_kart'):
golfKarts = []
golfKartGroups = []
if isinstance(dnaGroup, DNAGroup) and string.find(dnaGroup.getName(), type) >= 0:
golfKartGroups.append(dnaGroup)
if type == 'golf_kart':
nameInfo = dnaGroup.getName().split('_')
golfCourse = int(nameInfo[2])
pos = Point3(0, 0, 0)
hpr = Point3(0, 0, 0)
for i in range(dnaGroup.getNumChildren()):
childDnaGroup = dnaGroup.at(i)
if string.find(childDnaGroup.getName(), 'starting_block') >= 0:
padLocation = dnaGroup.getName().split('_')[2]
pos = childDnaGroup.getPos()
hpr = childDnaGroup.getHpr()
break
pos += Point3(0, 0, 0.05)
golfKart = DistributedGolfKartAI.DistributedGolfKartAI(self.air, golfCourse, pos[0], pos[1], pos[2], hpr[0], hpr[1], hpr[2])
else:
self.notify.warning('unhandled case')
golfKart.generateWithRequired(zoneId)
golfKarts.append(golfKart)
else:
if isinstance(dnaGroup, DNAVisGroup) and not overrideDNAZone:
zoneId = ZoneUtil.getTrueZoneId(int(dnaGroup.getName().split(':')[0]), zoneId)
for i in range(dnaGroup.getNumChildren()):
childGolfKarts, childGolfKartGroups = self.findAndCreateGolfKarts(dnaGroup.at(i), zoneId, area, overrideDNAZone, type)
golfKarts += childGolfKarts
golfKartGroups += childGolfKartGroups
return (golfKarts, golfKartGroups)
def createGolfKarts(self):
self.golfKarts = []
self.golfKartGroups = []
for zone in self.air.zoneTable[self.canonicalHoodId]:
zoneId = ZoneUtil.getTrueZoneId(zone[0], self.zoneId)
dnaData = self.air.dnaDataMap.get(zone[0], None)
if isinstance(dnaData, DNAData):
area = ZoneUtil.getCanonicalZoneId(zoneId)
foundKarts, foundKartGroups = self.findAndCreateGolfKarts(dnaData, zoneId, area, overrideDNAZone=True)
self.golfKarts += foundKarts
self.golfKartGroups += foundKartGroups
print(self.golfKarts, self.golfKartGroups)
for golfKart in self.golfKarts:
golfKart.start()
self.addDistObj(golfKart)
return
def findStartingBlocks(self, dnaRacingPadGroup, distRacePad):
startingBlocks = []
for i in range(dnaRacingPadGroup.getNumChildren()):
dnaGroup = dnaRacingPadGroup.at(i)
if string.find(dnaGroup.getName(), 'starting_block') >= 0:
padLocation = dnaGroup.getName().split('_')[2]
pos = dnaGroup.getPos()
hpr = dnaGroup.getHpr()
if isinstance(distRacePad, DistributedRacePadAI):
sb = DistributedStartingBlockAI(self, distRacePad, pos[0], pos[1], pos[2], hpr[0], hpr[1], hpr[2], int(padLocation))
else:
sb = DistributedViewingBlockAI(self, distRacePad, pos[0], pos[1], pos[2], hpr[0], hpr[1], hpr[2], int(padLocation))
sb.generateWithRequired(distRacePad.zoneId)
startingBlocks.append(sb)
else:
self.notify.debug('Found dnaGroup that is not a starting_block under a race pad group')
return startingBlocks