class ToontownAIRepository(ToontownInternalRepository):
def __init__(self, baseChannel, serverId, districtName):
ToontownInternalRepository.__init__(self, baseChannel, serverId, dcSuffix='AI')
self.dnaSpawner = DNASpawnerAI(self)
self.districtName = districtName
self.zoneAllocator = UniqueIdAllocator(ToontownGlobals.DynamicZonesBegin,
ToontownGlobals.DynamicZonesEnd)
self.zoneId2owner = {}
NPCToons.generateZone2NpcDict()
self.hoods = []
self.zoneDataStore = AIZoneDataStore()
self.useAllMinigames = self.config.GetBool('want-all-minigames', False)
self.doLiveUpdates = self.config.GetBool('want-live-updates', True)
self.holidayManager = HolidayManagerAI(self)
self.fishManager = FishManagerAI()
self.questManager = QuestManagerAI(self)
self.cogPageManager = CogPageManagerAI()
self.factoryMgr = FactoryManagerAI(self)
self.mintMgr = MintManagerAI(self)
self.lawOfficeMgr = LawOfficeManagerAI(self)
self.countryClubMgr = CountryClubManagerAI(self)
self.promotionMgr = PromotionManagerAI(self)
self.cogSuitMgr = CogSuitManagerAI(self)
self.suitInvasionManager = SuitInvasionManagerAI(self)
self.statusSender = ShardStatusSender(self)
self.dnaStoreMap = {}
self.buildingManagers = {}
self.suitPlanners = {}
def getTrackClsends(self):
return False
def handleConnected(self):
ToontownInternalRepository.handleConnected(self)
self.districtId = self.allocateChannel()
self.distributedDistrict = ToontownDistrictAI(self)
self.distributedDistrict.setName(self.districtName)
self.distributedDistrict.generateWithRequiredAndId(self.districtId,
self.getGameDoId(), 2)
# Claim ownership of that district...
dg = PyDatagram()
dg.addServerHeader(self.districtId, self.ourChannel, STATESERVER_OBJECT_SET_AI)
dg.addChannel(self.ourChannel)
self.send(dg)
self.createGlobals()
self.createZones()
self.statusSender.start()
self.distributedDistrict.b_setAvailable(1)
self.notify.info('District is now ready.')
def incrementPopulation(self):
self.districtStats.b_setAvatarCount(self.districtStats.getAvatarCount() + 1)
self.statusSender.sendStatus()
def decrementPopulation(self):
self.districtStats.b_setAvatarCount(self.districtStats.getAvatarCount() - 1)
self.statusSender.sendStatus()
def allocateZone(self, owner=None):
zoneId = self.zoneAllocator.allocate()
if owner:
self.zoneId2owner[zoneId] = owner
return zoneId
def deallocateZone(self, zone):
if self.zoneId2owner.get(zone):
del self.zoneId2owner[zone]
self.zoneAllocator.free(zone)
def getZoneDataStore(self):
return self.zoneDataStore
def getAvatarExitEvent(self, avId):
return 'distObjDelete-%d' % avId
def createGlobals(self):
"""
Create "global" objects, e.g. TimeManager et al.
"""
self.districtStats = ToontownDistrictStatsAI(self)
self.districtStats.settoontownDistrictId(self.districtId)
self.districtStats.generateWithRequiredAndId(self.allocateChannel(),
self.getGameDoId(), 3)
self.timeManager = TimeManagerAI(self)
self.timeManager.generateWithRequired(2)
self.newsManager = NewsManagerAI(self)
self.newsManager.generateWithRequired(2)
self.magicWordManager = MagicWordManagerAI(self)
self.magicWordManager.generateWithRequired(2)
self.friendManager = FriendManagerAI(self)
self.friendManager.generateWithRequired(2)
if config.GetBool('want-parties', True):
self.partyManager = DistributedPartyManagerAI(self)
self.partyManager.generateWithRequired(2)
self.globalPartyMgr = self.generateGlobalObject(OTP_DO_ID_GLOBAL_PARTY_MANAGER, 'GlobalPartyManager')
self.estateManager = EstateManagerAI(self)
self.estateManager.generateWithRequired(2)
self.trophyMgr = DistributedTrophyMgrAI(self)
self.trophyMgr.generateWithRequired(2)
self.tutorialManager = TutorialManagerAI(self)
self.tutorialManager.generateWithRequired(2)
self.catalogManager = CatalogManagerAI(self)
self.catalogManager.generateWithRequired(2)
self.codeRedemptionManager = TTCodeRedemptionMgrAI(self)
self.codeRedemptionManager.generateWithRequired(2)
def createZones(self):
"""
Spawn safezone objects, streets, doors, NPCs, etc.
"""
start = time.clock()
def clearQueue():
'''So the TCP window doesn't fill up and we get the axe'''
while self.readerPollOnce():
pass
self.hoods.append(TTHoodAI.TTHoodAI(self))
clearQueue()
self.hoods.append(DDHoodAI.DDHoodAI(self))
clearQueue()
self.hoods.append(DGHoodAI.DGHoodAI(self))
clearQueue()
self.hoods.append(BRHoodAI.BRHoodAI(self))
clearQueue()
self.hoods.append(MMHoodAI.MMHoodAI(self))
clearQueue()
self.hoods.append(DLHoodAI.DLHoodAI(self))
clearQueue()
self.hoods.append(GSHoodAI.GSHoodAI(self))
clearQueue()
self.hoods.append(OZHoodAI.OZHoodAI(self))
clearQueue()
self.hoods.append(GZHoodAI.GZHoodAI(self))
clearQueue()
self.hoods.append(SellbotHQAI.SellbotHQAI(self))
clearQueue()
self.hoods.append(CashbotHQAI.CashbotHQAI(self))
clearQueue()
self.hoods.append(LawbotHQAI.LawbotHQAI(self))
clearQueue()
self.hoods.append(BossbotHQAI.BossbotHQAI(self))
clearQueue()
print('Done creating Zones!')
for sp in self.suitPlanners.values():
sp.assignInitialSuitBuildings()
def genDNAFileName(self, zoneId):
zoneId = ZoneUtil.getCanonicalZoneId(zoneId)
hoodId = ZoneUtil.getCanonicalHoodId(zoneId)
hood = ToontownGlobals.dnaMap[hoodId]
if hoodId == zoneId:
zoneId = 'sz'
phase = ToontownGlobals.phaseMap[hoodId]
else:
phase = ToontownGlobals.streetPhaseMap[hoodId]
return 'phase_%s/dna/%s_%s.xml' % (phase, hood, zoneId)
def loadDNA(self, filename):
with open('/' + filename) as f:
tree = DNAParser.parse(f)
return tree
class ToontownAIRepository(ToontownInternalRepository):
def __init__(self, baseChannel, serverId, districtName):
ToontownInternalRepository.__init__(self,
baseChannel,
serverId,
dcSuffix='AI')
self.districtName = districtName
self.zoneAllocator = UniqueIdAllocator(
ToontownGlobals.DynamicZonesBegin, ToontownGlobals.DynamicZonesEnd)
self.zoneId2owner = {}
NPCToons.generateZone2NpcDict()
self.hoods = []
self.zoneDataStore = AIZoneDataStore()
self.dnaStoreMap = {}
self.dnaDataMap = {}
self.currentHour = 0
self.isRaining = False
self.useAllMinigames = self.config.GetBool('want-all-minigames', False)
self.doLiveUpdates = self.config.GetBool('want-live-updates', True)
self.wantFishing = self.config.GetBool('want-fishing', True)
self.wantHousing = self.config.GetBool('want-housing', True)
self.wantPets = self.config.GetBool('want-pets', True)
self.wantParties = self.config.GetBool('want-parties', True)
self.wantCogbuildings = self.config.GetBool('want-cogbuildings', True)
self.wantCogdominiums = self.config.GetBool('want-cogdominiums', True)
self.doLiveUpdates = self.config.GetBool('want-live-updates', False)
self.wantTrackClsends = self.config.GetBool('want-track-clsends',
False)
self.baseXpMultiplier = self.config.GetFloat('base-xp-multiplier', 1.0)
self.wantHalloween = self.config.GetBool('want-halloween', False)
self.wantChristmas = self.config.GetBool('want-christmas', False)
self.holidayManager = HolidayManagerAI(self)
self.fishManager = FishManagerAI()
self.questManager = QuestManagerAI(self)
self.cogPageManager = CogPageManagerAI()
self.factoryMgr = FactoryManagerAI(self)
self.mintMgr = MintManagerAI(self)
self.lawOfficeMgr = LawOfficeManagerAI(self)
self.countryClubMgr = CountryClubManagerAI(self)
self.promotionMgr = PromotionManagerAI(self)
self.cogSuitMgr = CogSuitManagerAI(self)
self.suitInvasionManager = SuitInvasionManagerAI(self)
self.wantCogdominiums = self.config.GetBool('want-cogdominums', False)
self.temperatureManager = TemperatureManagerAI(self)
self.statusSender = ShardStatusSender(self)
self.dnaStoreMap = {}
self.buildingManagers = {}
self.suitPlanners = {}
def getTrackClsends(self):
return False
def handleConnected(self):
ToontownInternalRepository.handleConnected(self)
self.districtId = self.allocateChannel()
self.notify.info('Creating new district (%d)...' % self.districtId)
self.distributedDistrict = ToontownDistrictAI(self)
self.distributedDistrict.setName(self.districtName)
self.distributedDistrict.generateWithRequiredAndId(
self.districtId, self.getGameDoId(), 2)
# Claim ownership of that district...
self.notify.info('Claiming ownership of district (%d)...' %
self.districtId)
dg = PyDatagram()
dg.addServerHeader(self.districtId, self.ourChannel,
STATESERVER_OBJECT_SET_AI)
dg.addChannel(self.ourChannel)
self.send(dg)
self.notify.info('Creating global objects...')
self.createGlobals()
self.notify.info('Creating zones (Playgrounds and Cog HQs)...')
self.createZones()
self.statusSender.start()
self.notify.info('Making district available to enter...')
self.distributedDistrict.b_setAvailable(1)
self.notify.info('District is now ready. Have fun in Toontown!')
def incrementPopulation(self):
self.districtStats.b_setAvatarCount(
self.districtStats.getAvatarCount() + 1)
self.statusSender.sendStatus()
def decrementPopulation(self):
self.districtStats.b_setAvatarCount(
self.districtStats.getAvatarCount() - 1)
self.statusSender.sendStatus()
def setHour(self, hour):
self.districtStats.b_setHour(hour)
self.statusSender.sendStatus()
def allocateZone(self, owner=None):
zoneId = self.zoneAllocator.allocate()
if owner:
self.zoneId2owner[zoneId] = owner
return zoneId
def deallocateZone(self, zone):
if self.zoneId2owner.get(zone):
del self.zoneId2owner[zone]
self.zoneAllocator.free(zone)
def getZoneDataStore(self):
return self.zoneDataStore
def getAvatarExitEvent(self, avId):
return 'distObjDelete-%d' % avId
def createGlobals(self):
"""
Create "global" objects, e.g. TimeManager et al.
"""
self.districtStats = ToontownDistrictStatsAI(self)
self.districtStats.settoontownDistrictId(self.districtId)
self.districtStats.generateWithRequiredAndId(self.allocateChannel(),
self.getGameDoId(), 3)
self.notify.info('Created district stats AI (%d).' %
self.districtStats.doId)
self.timeManager = TimeManagerAI(self)
self.timeManager.generateWithRequired(2)
self.newsManager = NewsManagerAI(self)
self.newsManager.generateWithRequired(2)
self.staffManager = StaffManagerAI(self)
self.staffManager.generateWithRequired(2)
self.banManager = BanManagerAI(self)
self.banManager.generateWithRequired(2)
self.magicWordManager = MagicWordManagerAI(self)
self.magicWordManager.generateWithRequired(2)
self.friendManager = FriendManagerAI(self)
self.friendManager.generateWithRequired(2)
if config.GetBool('want-parties', True):
self.partyManager = DistributedPartyManagerAI(self)
self.partyManager.generateWithRequired(2)
# setup our view of the global party manager ud
self.globalPartyMgr = self.generateGlobalObject(
OTP_DO_ID_GLOBAL_PARTY_MANAGER, 'GlobalPartyManager')
self.estateManager = EstateManagerAI(self)
self.estateManager.generateWithRequired(2)
self.trophyMgr = DistributedTrophyMgrAI(self)
self.trophyMgr.generateWithRequired(2)
if config.GetBool('want-pets', True):
self.petMgr = PetManagerAI(self)
self.tutorialManager = TutorialManagerAI(self)
self.tutorialManager.generateWithRequired(2)
self.catalogManager = CatalogManagerAI(self)
self.catalogManager.generateWithRequired(2)
self.codeRedemptionManager = TTCodeRedemptionMgrAI(self)
self.codeRedemptionManager.generateWithRequired(2)
def createZones(self):
"""
Spawn safezone objects, streets, doors, NPCs, etc.
"""
start = time.clock()
def clearQueue():
'''So the TCP window doesn't fill up and we get the axe'''
while self.readerPollOnce():
pass
self.hoods.append(TTHoodAI.TTHoodAI(self))
clearQueue()
self.hoods.append(DDHoodAI.DDHoodAI(self))
clearQueue()
self.hoods.append(DGHoodAI.DGHoodAI(self))
clearQueue()
self.hoods.append(BRHoodAI.BRHoodAI(self))
clearQueue()
self.hoods.append(MMHoodAI.MMHoodAI(self))
clearQueue()
self.hoods.append(DLHoodAI.DLHoodAI(self))
clearQueue()
self.hoods.append(GSHoodAI.GSHoodAI(self))
clearQueue()
self.hoods.append(OZHoodAI.OZHoodAI(self))
clearQueue()
self.hoods.append(GZHoodAI.GZHoodAI(self))
clearQueue()
if config.GetBool('want-sbhq', True):
self.hoods.append(SellbotHQAI.SellbotHQAI(self))
clearQueue()
if config.GetBool('want-cbhq', True):
self.hoods.append(CashbotHQAI.CashbotHQAI(self))
clearQueue()
if config.GetBool('want-lbhq', True):
self.hoods.append(LawbotHQAI.LawbotHQAI(self))
clearQueue()
if config.GetBool('want-bbhq', True):
self.hoods.append(BossbotHQAI.BossbotHQAI(self))
clearQueue()
for sp in self.suitPlanners.values():
sp.assignInitialSuitBuildings()
def lookupDNAFileName(self, zoneId):
zoneId = ZoneUtil.getCanonicalZoneId(zoneId)
hoodId = ZoneUtil.getCanonicalHoodId(zoneId)
hood = ToontownGlobals.dnaMap[hoodId]
if hoodId == zoneId:
zoneId = 'sz'
phase = ToontownGlobals.phaseMap[hoodId]
else:
phase = ToontownGlobals.streetPhaseMap[hoodId]
return 'phase_%s/dna/%s_%s.pdna' % (phase, hood, zoneId)
def loadDNAFileAI(self, dnastore, filename):
return loadDNAFileAI(dnastore, filename)
class ToontownAIRepository(ToontownInternalRepository):
def __init__(self, baseChannel, serverId, districtName):
ToontownInternalRepository.__init__(self,
baseChannel,
serverId,
dcSuffix='AI')
self.dnaSpawner = DNASpawnerAI(self)
self.districtName = districtName
self.zoneAllocator = UniqueIdAllocator(
ToontownGlobals.DynamicZonesBegin, ToontownGlobals.DynamicZonesEnd)
self.zoneId2owner = {}
NPCToons.generateZone2NpcDict()
self.hoods = []
self.zoneDataStore = AIZoneDataStore()
self.useAllMinigames = self.config.GetBool('want-all-minigames', False)
self.doLiveUpdates = self.config.GetBool('want-live-updates', True)
self.baseXpMultiplier = self.config.GetFloat('base-xp-multiplier', 1.0)
self.wantHalloween = self.config.GetBool('want-halloween', False)
self.wantChristmas = self.config.GetBool('want-christmas', False)
self.holidayManager = HolidayManagerAI(self)
if config.GetBool('want-pets', True):
self.PetManager = PetManagerAI(self)
self.fishManager = FishManagerAI()
self.questManager = QuestManagerAI(self)
self.cogPageManager = CogPageManagerAI()
self.factoryMgr = FactoryManagerAI(self)
self.mintMgr = MintManagerAI(self)
self.lawOfficeMgr = LawOfficeManagerAI(self)
self.countryClubMgr = CountryClubManagerAI(self)
self.promotionMgr = PromotionManagerAI(self)
self.cogSuitMgr = CogSuitManagerAI(self)
self.suitInvasionManager = SuitInvasionManagerAI(self)
self.statusSender = ShardStatusSender(self)
self.dnaStoreMap = {}
self.buildingManagers = {}
self.suitPlanners = {}
def getTrackClsends(self):
return False
def handleConnected(self):
self.notify.info('Yarn. Waking up (This may take a while!).')
ToontownInternalRepository.handleConnected(self)
self.districtId = self.allocateChannel()
self.distributedDistrict = ToontownDistrictAI(self)
self.distributedDistrict.setName(self.districtName)
self.distributedDistrict.generateWithRequiredAndId(
self.districtId, self.getGameDoId(), 2)
# Claim ownership of that district...
dg = PyDatagram()
dg.addServerHeader(self.districtId, self.ourChannel,
STATESERVER_OBJECT_SET_AI)
dg.addChannel(self.ourChannel)
self.send(dg)
self.notify.info('Creating Global Managers')
self.createGlobals()
self.notify.info('Creating Toontown')
self.createZones()
self.statusSender.start()
self.distributedDistrict.b_setAvailable(1)
self.notify.info('District is now ready.')
def incrementPopulation(self):
self.districtStats.b_setAvatarCount(
self.districtStats.getAvatarCount() + 1)
self.statusSender.sendStatus()
def decrementPopulation(self):
self.districtStats.b_setAvatarCount(
self.districtStats.getAvatarCount() - 1)
self.statusSender.sendStatus()
def allocateZone(self, owner=None):
zoneId = self.zoneAllocator.allocate()
if owner:
self.zoneId2owner[zoneId] = owner
return zoneId
def deallocateZone(self, zone):
if self.zoneId2owner.get(zone):
del self.zoneId2owner[zone]
self.zoneAllocator.free(zone)
def getZoneDataStore(self):
return self.zoneDataStore
def getAvatarExitEvent(self, avId):
return 'distObjDelete-%d' % avId
def createGlobals(self):
"""
Create "global" objects, e.g. TimeManager et al.
"""
self.notify.info('Creating District Stats')
self.districtStats = ToontownDistrictStatsAI(self)
self.districtStats.settoontownDistrictId(self.districtId)
self.districtStats.generateWithRequiredAndId(self.allocateChannel(),
self.getGameDoId(), 3)
self.notify.info('Creating Time Manager')
self.timeManager = TimeManagerAI(self)
self.timeManager.generateWithRequired(2)
self.notify.info('Creating News Manager')
self.newsManager = NewsManagerAI(self)
self.newsManager.generateWithRequired(2)
if config.GetBool('want-mw-manager', True):
self.notify.info('Creating Magic Words Manager')
self.magicWordManager = MagicWordManagerAI(self)
self.magicWordManager.generateWithRequired(2)
if config.GetBool('want-friends-manager', True):
self.notify.info('Creating Friends Manager')
self.friendManager = FriendManagerAI(self)
self.friendManager.generateWithRequired(2)
if config.GetBool('want-parties', True):
self.notify.info('Creating Parties Manager')
self.partyManager = DistributedPartyManagerAI(self)
self.partyManager.generateWithRequired(2)
# Setup our view of the global party manager ud
self.globalPartyMgr = self.generateGlobalObject(
OTP_DO_ID_GLOBAL_PARTY_MANAGER, 'GlobalPartyManager')
if config.GetBool('want-estates-manager', True):
self.notify.info('Creating Estates')
self.estateManager = EstateManagerAI(self)
self.estateManager.generateWithRequired(2)
self.notify.info('Creating TrophyMgr')
self.trophyMgr = DistributedTrophyMgrAI(self)
self.trophyMgr.generateWithRequired(2)
self.notify.info('Creating Toontorial Manager')
self.tutorialManager = TutorialManagerAI(self)
self.tutorialManager.generateWithRequired(2)
self.notify.info('Creating Catalog Manager')
self.catalogManager = CatalogManagerAI(self)
self.catalogManager.generateWithRequired(2)
if config.GetBool('want-pets', True):
self.notify.info('Creating Pets Manager')
self.PetManager = PetManagerAI(self)
self.notify.info('Creating Code Redemption Manager')
self.codeRedemptionManager = TTCodeRedemptionMgrAI(self)
self.codeRedemptionManager.generateWithRequired(2)
def createZones(self):
"""
Spawn safezone objects, streets, doors, NPCs, etc.
"""
start = time.clock()
def clearQueue():
'''So the TCP window doesn't fill up and we get the axe'''
while self.readerPollOnce():
pass
self.notify.info('Creating TTC (Toontown Central) ')
self.hoods.append(TTHoodAI.TTHoodAI(self))
clearQueue()
self.notify.info('Creating DD (Donalds Dock)')
self.hoods.append(DDHoodAI.DDHoodAI(self))
clearQueue()
self.notify.info('Creating DG (Daisy Gardens) ')
self.hoods.append(DGHoodAI.DGHoodAI(self))
clearQueue()
self.notify.info('Creating BR (The Brrrgh) ')
self.hoods.append(BRHoodAI.BRHoodAI(self))
clearQueue()
self.notify.info('Creating MML (Minnie Melody Land) ')
self.hoods.append(MMHoodAI.MMHoodAI(self))
clearQueue()
self.notify.info('Creating DDL (Donalds Dream Land) ')
self.hoods.append(DLHoodAI.DLHoodAI(self))
clearQueue()
self.notify.info('Creating GS (SpeedWay) ')
self.hoods.append(GSHoodAI.GSHoodAI(self))
clearQueue()
self.notify.info('Creating OZ (Outdoor Zone)')
self.hoods.append(OZHoodAI.OZHoodAI(self))
clearQueue()
self.notify.info('Creating GZ (Golf Zone) ')
self.hoods.append(GZHoodAI.GZHoodAI(self))
clearQueue()
self.notify.info('Creating TF (Toonfest) ')
self.hoods.append(TFHoodAI.TFHoodAI(self))
clearQueue()
if config.GetBool('want-sbhq', True):
self.notify.info('Creating SBHQ (Sellbot HQ) ')
self.hoods.append(SellbotHQAI.SellbotHQAI(self))
clearQueue()
if config.GetBool('want-cbhq', True):
self.notify.info('Creating CBHQ (Cashbot HQ) ')
self.hoods.append(CashbotHQAI.CashbotHQAI(self))
clearQueue()
if config.GetBool('want-lbhq', True):
self.notify.info('Creating LBHQ (Lawbot HQ) ')
self.hoods.append(LawbotHQAI.LawbotHQAI(self))
clearQueue()
if config.GetBool('want-bbhq', True):
self.notify.info('Creating BBHQ (Bossbot HQ) ')
self.hoods.append(BossbotHQAI.BossbotHQAI(self))
clearQueue()
for sp in self.suitPlanners.values():
sp.assignInitialSuitBuildings()
def genDNAFileName(self, zoneId):
zoneId = ZoneUtil.getCanonicalZoneId(zoneId)
hoodId = ZoneUtil.getCanonicalHoodId(zoneId)
hood = ToontownGlobals.dnaMap[hoodId]
if hoodId == zoneId:
zoneId = 'sz'
phase = ToontownGlobals.phaseMap[hoodId]
else:
phase = ToontownGlobals.streetPhaseMap[hoodId]
return 'phase_%s/dna/%s_%s.xml' % (phase, hood, zoneId)
def loadDNA(self, filename):
with open('/' + filename) as f:
tree = DNAParser.parse(f)
return tree