toontown-just-works/toontown/estate/DistributedHouseAI.py
2024-07-07 18:08:39 -05:00

308 lines
No EOL
10 KiB
Python

from direct.distributed.DistributedObjectAI import DistributedObjectAI
from toontown.estate.DistributedHouseInteriorAI import DistributedHouseInteriorAI
from toontown.estate.DistributedHouseDoorAI import DistributedHouseDoorAI
from toontown.estate.DistributedMailboxAI import DistributedMailboxAI
from toontown.building import DoorTypes
from toontown.catalog.CatalogItemList import CatalogItemList
from otp.ai.MagicWordGlobal import *
from toontown.catalog.CatalogFurnitureItem import *
from toontown.catalog.CatalogItem import Customization, WindowPlacement, Location
class DistributedHouseAI(DistributedObjectAI):
notify = directNotify.newCategory("DistributedHouseAI")
def __init__(self, air):
DistributedObjectAI.__init__(self, air)
self.houseType = 0
self.gardenPos = 0
self.avatarId = 0
self.name = ''
self.color = 0
self.housePos = 0
self.gender = 0
self.isInteriorInitialized = 1
self.atticItems = CatalogItemList(store=Customization)
self.interiorItems = CatalogItemList(store=Customization)
self.interiorWallpaper = CatalogItemList(store=Customization)
self.atticWallpaper = CatalogItemList(store=Customization)
self.interiorWindows = CatalogItemList(store=Customization)
self.atticWindows = CatalogItemList(store=Customization)
self.deletedItems = CatalogItemList(store=Customization)
def announceGenerate(self):
DistributedObjectAI.announceGenerate(self)
self.interiorZone = self.air.allocateZone()
self.door = DistributedHouseDoorAI(self.air, self.getDoId(), DoorTypes.EXT_STANDARD)
self.door.setSwing(3)
self.door.generateWithRequired(self.zoneId)
self.interiorDoor = DistributedHouseDoorAI(self.air, self.getDoId(), DoorTypes.INT_STANDARD)
self.interiorDoor.setSwing(3)
self.interiorDoor.setOtherDoor(self.door)
self.interiorDoor.generateWithRequired(self.interiorZone)
self.door.setOtherDoor(self.interiorDoor)
self.interior = DistributedHouseInteriorAI(self.air, self)
self.interior.setHouseIndex(self.housePos)
self.interior.setHouseId(self.getDoId())
self.interior.generateWithRequired(self.interiorZone)
if self.avatarId:
self.mailbox = DistributedMailboxAI(self.air, self)
self.mailbox.generateWithRequired(self.zoneId)
self.sendUpdate('setHouseReady', [])
def delete(self):
self.door.requestDelete()
self.interiorDoor.requestDelete()
self.interior.requestDelete()
if self.avatarId:
self.mailbox.requestDelete()
self.air.deallocateZone(self.interiorZone)
DistributedObjectAI.delete(self)
def setHousePos(self, pos):
self.housePos = pos
def d_setHousePos(self, pos):
self.sendUpdate('setHousePos', [pos])
def b_setHousePos(self, pos):
self.setHousePos(pos)
self.d_setHousePos(pos)
def getHousePos(self):
return self.housePos
def setHouseType(self, type):
self.houseType = type
def d_setHouseType(self, type):
self.sendUpdate('setHouseType', [type])
def b_setHouseType(self, type):
self.setHouseType(type)
self.d_setHouseType(type)
def getHouseType(self):
return self.houseType
def setGardenPos(self, pos):
self.gardenPos = pos
def d_setGardenPos(self, pos):
self.sendUpdate('setGardenPos', [pos])
def b_setGardenPos(self, pos):
self.setGardenPow(pos)
self.d_setGardenPos(pos)
def getGardenPos(self):
return self.gardenPos
def setAvatarId(self, avId):
self.avatarId = avId
def d_setAvatarId(self, avId):
self.sendUpdate('setAvatarId', [avId])
def b_setAvatarId(self, avId):
self.setAvatarId(avId)
self.d_setAvatarId(avId)
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 setGender(self, genderIndex):
self.gender = genderIndex
def getGender(self):
return self.gender
def setAtticItems(self, atticItems):
self.atticItems = CatalogItemList(atticItems, store=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=Customization | 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=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=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=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=Customization | 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=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 setInteriorInitialized(self, initialized):
self.isInteriorInitialized = initialized
def d_setInteriorInitialized(self, initialized):
self.sendUpdate('setInteriorInitialized', [initialized])
def b_setInteriorInitialized(self, initialized):
self.setInteriorInitialized(initialized)
self.d_setInteriorInitialized(initialized)
def getInteriorInitialized(self):
return self.isInteriorInitialized
def setCannonEnabled(self, todo0):
pass
def getCannonEnabled(self):
return 0
def setHouseReady(self):
pass
def addAtticItem(self, item):
self.interior.furnitureManager.saveToHouse()
if item.getFlags() & FLTrunk:
self.atticItems.append(item)
elif item.replacesExisting() and item.hasExisting():
if item.getFlags() & FLCloset:
items = ClosetToClothes.keys() if item.getFlags() & FLCloset else BankToMoney.keys()
for itItem in self.interiorItems:
if itItem.furnitureType in items:
posHpr = itItem.posHpr
self.interiorItems.remove(itItem)
item.posHpr = posHpr
self.interiorItems.append(item)
break
for itItem in self.atticItems:
if itItem.furnitureType in items:
self.atticItems.remove(itItem)
self.atticItems.append(item)
break
else:
self.atticItems.append(item)
self.d_setAtticItems(self.atticItems.getBlob())
self.d_setInteriorItems(self.interiorItems.getBlob())
self.interior.furnitureManager.loadFromHouse()
def addWindow(self, item):
self.interior.furnitureManager.saveToHouse()
self.atticWindows.append(item)
self.d_setAtticWindows(self.atticWindows.getBlob())
self.interior.furnitureManager.loadFromHouse()
def addWallpaper(self, item):
self.interior.furnitureManager.saveToHouse()
self.atticWallpaper.append(item)
self.d_setAtticWallpaper(self.atticWallpaper.getBlob())
self.interior.furnitureManager.loadFromHouse()
def initializeInterior(self):
if (not self.isInteriorInitialized):
self.notify.info('Initializing interior...')
self.interior.initialize()
self.b_setInteriorInitialized(1)