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)