352 lines
12 KiB
Python
352 lines
12 KiB
Python
|
from direct.directnotify import DirectNotifyGlobal
|
||
|
|
from direct.distributed.DistributedObjectAI import DistributedObjectAI
|
||
|
|
|
||
|
|
from toontown.building import DoorTypes
|
||
|
|
from toontown.catalog.CatalogFurnitureItem import *
|
||
|
|
from toontown.catalog.CatalogItemList import CatalogItemList
|
||
|
|
from toontown.estate.DistributedHouseDoorAI import DistributedHouseDoorAI
|
||
|
|
from toontown.estate.DistributedHouseInteriorAI import DistributedHouseInteriorAI
|
||
|
|
from toontown.estate.DistributedMailboxAI import DistributedMailboxAI
|
||
|
|
from toontown.estate.GardenManagerAI import GardenManagerAI
|
||
|
|
|
||
|
|
|
||
|
|
class DistributedHouseAI(DistributedObjectAI):
|
||
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('DistributedHouseAI')
|
||
|
|
|
||
|
|
def __init__(self, air):
|
||
|
|
DistributedObjectAI.__init__(self, air)
|
||
|
|
self.estate = None
|
||
|
|
self.housePos = 0
|
||
|
|
self.houseType = 0
|
||
|
|
self.gardenPos = 0
|
||
|
|
self.avatarId = 0
|
||
|
|
self.name = ''
|
||
|
|
self.color = 0
|
||
|
|
self.atticItems = CatalogItemList(store=CatalogItem.Customization)
|
||
|
|
self.interiorItems = CatalogItemList(store=CatalogItem.Customization)
|
||
|
|
self.atticWallpaper = CatalogItemList(store=CatalogItem.Customization)
|
||
|
|
self.interiorWallpaper = CatalogItemList(store=CatalogItem.Customization)
|
||
|
|
self.atticWindows = CatalogItemList(store=CatalogItem.Customization)
|
||
|
|
self.interiorWindows = CatalogItemList(store=CatalogItem.Customization)
|
||
|
|
self.deletedItems = CatalogItemList(store=CatalogItem.Customization)
|
||
|
|
self.cannonEnabled = 0
|
||
|
|
self.interiorZone = None
|
||
|
|
self.exteriorDoor = None
|
||
|
|
self.interiorDoor = None
|
||
|
|
self.interior = None
|
||
|
|
self.mailbox = None
|
||
|
|
self.gardenManager = None
|
||
|
|
|
||
|
|
def announceGenerate(self):
|
||
|
|
DistributedObjectAI.announceGenerate(self)
|
||
|
|
|
||
|
|
# Allocate a zone for this house's interior:
|
||
|
|
self.interiorZone = self.air.allocateZone()
|
||
|
|
|
||
|
|
# Setup interior & exterior doors:
|
||
|
|
self.exteriorDoor = DistributedHouseDoorAI(self.air, self.getDoId(), DoorTypes.EXT_STANDARD)
|
||
|
|
self.exteriorDoor.setSwing(3)
|
||
|
|
self.exteriorDoor.generateWithRequired(self.zoneId)
|
||
|
|
self.interiorDoor = DistributedHouseDoorAI(self.air, self.getDoId(), DoorTypes.INT_STANDARD)
|
||
|
|
self.interiorDoor.setSwing(3)
|
||
|
|
self.interiorDoor.setOtherDoor(self.exteriorDoor)
|
||
|
|
self.interiorDoor.generateWithRequired(self.interiorZone)
|
||
|
|
self.exteriorDoor.setOtherDoor(self.interiorDoor)
|
||
|
|
|
||
|
|
# Setup interior:
|
||
|
|
self.interior = DistributedHouseInteriorAI(self.air, self)
|
||
|
|
self.interior.setHouseIndex(self.housePos)
|
||
|
|
self.interior.setHouseId(self.getDoId())
|
||
|
|
self.interior.generateWithRequired(self.interiorZone)
|
||
|
|
|
||
|
|
# Generate our mailbox:
|
||
|
|
if self.avatarId:
|
||
|
|
self.mailbox = DistributedMailboxAI(self.air, self)
|
||
|
|
self.mailbox.generateWithRequired(self.zoneId)
|
||
|
|
|
||
|
|
# Send the house ready update:
|
||
|
|
self.d_setHouseReady()
|
||
|
|
|
||
|
|
def delete(self):
|
||
|
|
self.exteriorDoor.requestDelete()
|
||
|
|
self.interiorDoor.requestDelete()
|
||
|
|
if self.interior.furnitureManager:
|
||
|
|
self.interior.furnitureManager.requestDelete()
|
||
|
|
|
||
|
|
self.interior.requestDelete()
|
||
|
|
if self.avatarId:
|
||
|
|
self.mailbox.requestDelete()
|
||
|
|
|
||
|
|
if self.gardenManager:
|
||
|
|
self.gardenManager.destroy()
|
||
|
|
|
||
|
|
self.air.deallocateZone(self.interiorZone)
|
||
|
|
self.estate = None
|
||
|
|
DistributedObjectAI.delete(self)
|
||
|
|
|
||
|
|
def setHousePos(self, housePos):
|
||
|
|
self.housePos = housePos
|
||
|
|
|
||
|
|
def d_setHousePos(self, housePos):
|
||
|
|
self.sendUpdate('setHousePos', [housePos])
|
||
|
|
|
||
|
|
def b_setHousePos(self, housePos):
|
||
|
|
self.setHousePos(housePos)
|
||
|
|
self.d_setHousePos(housePos)
|
||
|
|
|
||
|
|
def getHousePos(self):
|
||
|
|
return self.housePos
|
||
|
|
|
||
|
|
def setHouseType(self, houseType):
|
||
|
|
self.houseType = houseType
|
||
|
|
|
||
|
|
def d_setHouseType(self, houseType):
|
||
|
|
self.sendUpdate('setHouseType', [houseType])
|
||
|
|
|
||
|
|
def b_setHouseType(self, houseType):
|
||
|
|
self.setHouseType(houseType)
|
||
|
|
self.d_setHouseType(houseType)
|
||
|
|
|
||
|
|
def getHouseType(self):
|
||
|
|
return self.houseType
|
||
|
|
|
||
|
|
def setGardenPos(self, gardenPos):
|
||
|
|
self.gardenPos = gardenPos
|
||
|
|
|
||
|
|
def d_setGardenPos(self, gardenPos):
|
||
|
|
self.sendUpdate('setGardenPos', [gardenPos])
|
||
|
|
|
||
|
|
def b_setGardenPos(self, gardenPos):
|
||
|
|
self.setGardenPos(gardenPos)
|
||
|
|
self.d_setGardenPos(gardenPos)
|
||
|
|
|
||
|
|
def getGardenPos(self):
|
||
|
|
return self.gardenPos
|
||
|
|
|
||
|
|
def setAvatarId(self, avatarId):
|
||
|
|
self.avatarId = avatarId
|
||
|
|
|
||
|
|
def d_setAvatarId(self, avatarId):
|
||
|
|
self.sendUpdate('setAvatarId', [avatarId])
|
||
|
|
|
||
|
|
def b_setAvatarId(self, avatarId):
|
||
|
|
self.setAvatarId(avatarId)
|
||
|
|
self.d_setAvatarId(avatarId)
|
||
|
|
|
||
|
|
def getAvatarId(self):
|
||
|
|
return self.avatarId
|
||
|
|
|
||
|
|
def setName(self, name):
|
||
|
|
self.name = name
|
||
|
|
|
||
|
|
def d_setName(self, name):
|
||
|
|
self.sendUpdate('setName', [name])
|
||
|
|
|
||
|
|
def b_setName(self, name):
|
||
|
|
self.setName(name)
|
||
|
|
self.d_setName(name)
|
||
|
|
|
||
|
|
def getName(self):
|
||
|
|
return self.name
|
||
|
|
|
||
|
|
def setColor(self, color):
|
||
|
|
self.color = color
|
||
|
|
|
||
|
|
def d_setColor(self, color):
|
||
|
|
self.sendUpdate('setColor', [color])
|
||
|
|
|
||
|
|
def b_setColor(self, color):
|
||
|
|
self.setColor(color)
|
||
|
|
self.d_setColor(color)
|
||
|
|
|
||
|
|
def getColor(self):
|
||
|
|
return self.color
|
||
|
|
|
||
|
|
def setAtticItems(self, atticItems):
|
||
|
|
self.atticItems = CatalogItemList(atticItems, store=CatalogItem.Customization)
|
||
|
|
|
||
|
|
def d_setAtticItems(self, atticItems):
|
||
|
|
self.sendUpdate('setAtticItems', [atticItems])
|
||
|
|
|
||
|
|
def b_setAtticItems(self, atticItems):
|
||
|
|
self.setAtticItems(atticItems)
|
||
|
|
self.d_setAtticItems(atticItems)
|
||
|
|
|
||
|
|
def getAtticItems(self):
|
||
|
|
return self.atticItems.getBlob()
|
||
|
|
|
||
|
|
def setInteriorItems(self, interiorItems):
|
||
|
|
self.interiorItems = CatalogItemList(interiorItems, store=CatalogItem.Customization | CatalogItem.Location)
|
||
|
|
|
||
|
|
def d_setInteriorItems(self, interiorItems):
|
||
|
|
self.sendUpdate('setInteriorItems', [interiorItems])
|
||
|
|
|
||
|
|
def b_setInteriorItems(self, interiorItems):
|
||
|
|
self.setInteriorItems(interiorItems)
|
||
|
|
self.d_setInteriorItems(interiorItems)
|
||
|
|
|
||
|
|
def getInteriorItems(self):
|
||
|
|
return self.interiorItems.getBlob()
|
||
|
|
|
||
|
|
def setAtticWallpaper(self, atticWallpaper):
|
||
|
|
self.atticWallpaper = CatalogItemList(atticWallpaper, store=CatalogItem.Customization)
|
||
|
|
|
||
|
|
def d_setAtticWallpaper(self, atticWallpaper):
|
||
|
|
self.sendUpdate('setAtticWallpaper', [atticWallpaper])
|
||
|
|
|
||
|
|
def b_setAtticWallpaper(self, atticWallpaper):
|
||
|
|
self.setAtticWallpaper(atticWallpaper)
|
||
|
|
self.d_setAtticWallpaper(atticWallpaper)
|
||
|
|
|
||
|
|
def getAtticWallpaper(self):
|
||
|
|
return self.atticWallpaper.getBlob()
|
||
|
|
|
||
|
|
def setInteriorWallpaper(self, interiorWallpaper):
|
||
|
|
self.interiorWallpaper = CatalogItemList(interiorWallpaper, store=CatalogItem.Customization)
|
||
|
|
|
||
|
|
def d_setInteriorWallpaper(self, interiorWallpaper):
|
||
|
|
self.sendUpdate('setInteriorWallpaper', [interiorWallpaper])
|
||
|
|
|
||
|
|
def b_setInteriorWallpaper(self, interiorWallpaper):
|
||
|
|
self.setInteriorWallpaper(interiorWallpaper)
|
||
|
|
self.d_setInteriorWallpaper(interiorWallpaper)
|
||
|
|
|
||
|
|
def getInteriorWallpaper(self):
|
||
|
|
return self.interiorWallpaper.getBlob()
|
||
|
|
|
||
|
|
def setAtticWindows(self, atticWindows):
|
||
|
|
self.atticWindows = CatalogItemList(atticWindows, store=CatalogItem.Customization)
|
||
|
|
|
||
|
|
def d_setAtticWindows(self, atticWindows):
|
||
|
|
self.sendUpdate('setAtticWindows', [atticWindows])
|
||
|
|
|
||
|
|
def b_setAtticWindows(self, atticWindows):
|
||
|
|
self.setAtticWindows(atticWindows)
|
||
|
|
self.d_setAtticWindows(atticWindows)
|
||
|
|
|
||
|
|
def getAtticWindows(self):
|
||
|
|
return self.atticWindows.getBlob()
|
||
|
|
|
||
|
|
def setInteriorWindows(self, interiorWindows):
|
||
|
|
self.interiorWindows = CatalogItemList(interiorWindows,
|
||
|
|
store=CatalogItem.Customization | CatalogItem.WindowPlacement)
|
||
|
|
|
||
|
|
def d_setInteriorWindows(self, interiorWindows):
|
||
|
|
self.sendUpdate('setInteriorWindows', [interiorWindows])
|
||
|
|
|
||
|
|
def b_setInteriorWindows(self, interiorWindows):
|
||
|
|
self.setInteriorWindows(interiorWindows)
|
||
|
|
self.d_setInteriorWindows(interiorWindows)
|
||
|
|
|
||
|
|
def getInteriorWindows(self):
|
||
|
|
return self.interiorWindows.getBlob()
|
||
|
|
|
||
|
|
def setDeletedItems(self, deletedItems):
|
||
|
|
self.deletedItems = CatalogItemList(deletedItems, store=CatalogItem.Customization)
|
||
|
|
|
||
|
|
def d_setDeletedItems(self, deletedItems):
|
||
|
|
self.sendUpdate('setDeletedItems', [deletedItems])
|
||
|
|
|
||
|
|
def b_setDeletedItems(self, deletedItems):
|
||
|
|
self.setDeletedItems(deletedItems)
|
||
|
|
self.d_setDeletedItems(deletedItems)
|
||
|
|
|
||
|
|
def getDeletedItems(self):
|
||
|
|
return self.deletedItems.getBlob()
|
||
|
|
|
||
|
|
def setCannonEnabled(self, cannonEnabled):
|
||
|
|
self.cannonEnabled = cannonEnabled
|
||
|
|
|
||
|
|
def d_setCannonEnabled(self, cannonEnabled):
|
||
|
|
self.sendUpdate('setCannonEnabled', [cannonEnabled])
|
||
|
|
|
||
|
|
def b_setCannonEnabled(self, cannonEnabled):
|
||
|
|
self.setCannonEnabled(cannonEnabled)
|
||
|
|
self.d_setCannonEnabled(cannonEnabled)
|
||
|
|
|
||
|
|
def getCannonEnabled(self):
|
||
|
|
return self.cannonEnabled
|
||
|
|
|
||
|
|
def d_setHouseReady(self):
|
||
|
|
self.sendUpdate('setHouseReady')
|
||
|
|
|
||
|
|
def addAtticItem(self, item):
|
||
|
|
if item.replacesExisting and item.hasExisting():
|
||
|
|
if item.getFlags() & FLCloset:
|
||
|
|
closet = ClosetToClothes.keys()
|
||
|
|
for atticItem in self.atticItems:
|
||
|
|
if atticItem.furnitureType in closet:
|
||
|
|
self.interior.furnitureManager.atticItems.remove(atticItem)
|
||
|
|
break
|
||
|
|
|
||
|
|
for interiorItem in self.interior.furnitureManager.items[:]:
|
||
|
|
if interiorItem.catalogItem.furnitureType in closet:
|
||
|
|
interiorItem.requestDelete()
|
||
|
|
self.interior.furnitureManager.items.remove(interiorItem)
|
||
|
|
item.posHpr = interiorItem.catalogItem.posHpr
|
||
|
|
self.interior.furnitureManager.generateItem(item)
|
||
|
|
self.interior.furnitureManager.saveFurniture()
|
||
|
|
return
|
||
|
|
|
||
|
|
self.interior.furnitureManager.atticItems.append(item)
|
||
|
|
self.interior.furnitureManager.saveFurniture()
|
||
|
|
|
||
|
|
def addWallpaper(self, item):
|
||
|
|
self.interior.furnitureManager.atticWallpaper.append(item)
|
||
|
|
self.interior.furnitureManager.saveFurniture()
|
||
|
|
|
||
|
|
def addWindow(self, item):
|
||
|
|
self.interior.furnitureManager.atticWindows.append(item)
|
||
|
|
self.interior.furnitureManager.saveFurniture()
|
||
|
|
|
||
|
|
def placeStarterGarden(self):
|
||
|
|
if not config.GetBool('want-gardening', False):
|
||
|
|
return
|
||
|
|
|
||
|
|
if not self.estate:
|
||
|
|
return
|
||
|
|
|
||
|
|
av = self.air.doId2do.get(self.getAvatarId())
|
||
|
|
if av is None:
|
||
|
|
return
|
||
|
|
|
||
|
|
if av.getGardenStarted():
|
||
|
|
self.notify.warning('Avatar %s tried to start their garden twice!' % self.getAvatarId())
|
||
|
|
return
|
||
|
|
|
||
|
|
# Set the avatar's garden to started:
|
||
|
|
av.b_setGardenStarted(1)
|
||
|
|
|
||
|
|
# Create the GardenManagerAI:
|
||
|
|
self.gardenManager = GardenManagerAI(self.air, self.estate)
|
||
|
|
self.gardenManager.loadGarden(av.doId)
|
||
|
|
|
||
|
|
def createGardenManager(self):
|
||
|
|
if not config.GetBool('want-gardening', False):
|
||
|
|
return
|
||
|
|
|
||
|
|
if not self.estate:
|
||
|
|
return
|
||
|
|
|
||
|
|
if not self.getAvatarId():
|
||
|
|
return
|
||
|
|
|
||
|
|
av = self.air.doId2do.get(self.getAvatarId())
|
||
|
|
if av is not None:
|
||
|
|
if av.getGardenStarted():
|
||
|
|
self.gardenManager = GardenManagerAI(self.air, self.estate)
|
||
|
|
self.gardenManager.loadGarden(av.doId)
|
||
|
|
|
||
|
|
return
|
||
|
|
|
||
|
|
def __gotOwner(dclass, fields, self=self):
|
||
|
|
if dclass != self.air.dclassesByName['DistributedToonAI']:
|
||
|
|
return
|
||
|
|
|
||
|
|
gardenStarted = fields['setGardenStarted'][0]
|
||
|
|
if gardenStarted:
|
||
|
|
self.gardenManager = GardenManagerAI(self.air, self.estate)
|
||
|
|
self.gardenManager.loadGarden(self.getAvatarId())
|
||
|
|
|
||
|
|
self.air.dbInterface.queryObject(self.air.dbId, self.getAvatarId(), __gotOwner)
|