from direct.directnotify import DirectNotifyGlobal from pandac.PandaModules import * from toontown.toonbase import TTLocalizer from toontown.toonbase import ToontownGlobals from direct.interval.IntervalGlobal import * from direct.distributed.PyDatagram import PyDatagram from direct.distributed.PyDatagramIterator import PyDatagramIterator import types import sys CatalogReverseType = None CatalogItemVersion = 8 CatalogBackorderMarkup = 1.2 CatalogSaleMarkdown = 0.75 Customization = 1 DeliveryDate = 2 Location = 4 WindowPlacement = 8 GiftTag = 16 CatalogTypeUnspecified = 0 CatalogTypeWeekly = 1 CatalogTypeBackorder = 2 CatalogTypeMonthly = 3 CatalogTypeLoyalty = 4 class CatalogItem: notify = DirectNotifyGlobal.directNotify.newCategory('CatalogItem') def __init__(self, *args, **kw): self.saleItem = 0 self.deliveryDate = None self.posHpr = None self.giftTag = None self.giftCode = 0 self.hasPicture = False self.volume = 0 self.specialEventId = 0 if len(args) >= 1 and isinstance(args[0], DatagramIterator): self.decodeDatagram(*args, **kw) else: self.makeNewItem(*args, **kw) return def isAward(self): result = self.specialEventId != 0 return result def makeNewItem(self): pass def needsCustomize(self): return 0 def saveHistory(self): return 0 def getBackSticky(self): itemType = 0 numSticky = 0 return (itemType, numSticky) def putInBackCatalog(self, backCatalog, lastBackCatalog): if self.saveHistory() and not self.isSaleItem(): if self not in backCatalog: if self in lastBackCatalog: lastBackCatalog.remove(self) backCatalog.append(self) def replacesExisting(self): return 0 def hasExisting(self): return 0 def getYourOldDesc(self): return None def storedInCloset(self): return 0 def storedInTrunk(self): return 0 def storedInAttic(self): return 0 def notOfferedTo(self, avatar): return 0 def getPurchaseLimit(self): return 0 def reachedPurchaseLimit(self, avatar): return 0 def hasBeenGifted(self, avatar): if avatar.onGiftOrder.count(self) != 0: return 1 return 0 def getTypeName(self): return 'Unknown Type Item' def getName(self): return 'Unnamed Item' def getDisplayName(self): return self.getName() def recordPurchase(self, avatar, optional): self.notify.warning('%s has no purchase method.' % self) return ToontownGlobals.P_NoPurchaseMethod def isSaleItem(self): return self.saleItem def isGift(self): if self.getEmblemPrices(): return 0 return 1 def isRental(self): return 0 def forBoysOnly(self): return 0 def forGirlsOnly(self): return 0 def setLoyaltyRequirement(self, days): self.loyaltyDays = days def loyaltyRequirement(self): if not hasattr(self, 'loyaltyDays'): return 0 else: return self.loyaltyDays def getPrice(self, catalogType): if catalogType == CatalogTypeBackorder: return self.getBackPrice() elif self.isSaleItem(): return self.getSalePrice() else: return self.getCurrentPrice() def getCurrentPrice(self): return int(self.getBasePrice()) def getBackPrice(self): return int(self.getBasePrice() * CatalogBackorderMarkup) def getSalePrice(self): return int(self.getBasePrice() * CatalogSaleMarkdown) def getDeliveryTime(self): return 0 def getPicture(self, avatar): self.hasPicture = True return (None, None) def cleanupPicture(self): self.hasPicture = False def requestPurchase(self, phone, callback, optional = -1): phone.requestPurchase(self, callback, optional) def requestGiftPurchase(self, phone, targetDoID, callback, optional = -1): phone.requestGiftPurchase(self, targetDoID, callback, optional) def requestPurchaseCleanup(self): pass def getRequestPurchaseErrorText(self, retcode): if retcode == ToontownGlobals.P_ItemAvailable: return TTLocalizer.CatalogPurchaseItemAvailable elif retcode == ToontownGlobals.P_ItemOnOrder: return TTLocalizer.CatalogPurchaseItemOnOrder elif retcode == ToontownGlobals.P_MailboxFull: return TTLocalizer.CatalogPurchaseMailboxFull elif retcode == ToontownGlobals.P_OnOrderListFull: return TTLocalizer.CatalogPurchaseOnOrderListFull else: return TTLocalizer.CatalogPurchaseGeneralError % retcode def getRequestGiftPurchaseErrorText(self, retcode): if retcode == ToontownGlobals.P_ItemAvailable: return TTLocalizer.CatalogPurchaseGiftItemAvailable elif retcode == ToontownGlobals.P_ItemOnOrder: return TTLocalizer.CatalogPurchaseGiftItemOnOrder elif retcode == ToontownGlobals.P_MailboxFull: return TTLocalizer.CatalogPurchaseGiftMailboxFull elif retcode == ToontownGlobals.P_OnOrderListFull: return TTLocalizer.CatalogPurchaseGiftOnOrderListFull elif retcode == ToontownGlobals.P_NotAGift: return TTLocalizer.CatalogPurchaseGiftNotAGift elif retcode == ToontownGlobals.P_WillNotFit: return TTLocalizer.CatalogPurchaseGiftWillNotFit elif retcode == ToontownGlobals.P_ReachedPurchaseLimit: return TTLocalizer.CatalogPurchaseGiftLimitReached elif retcode == ToontownGlobals.P_NotEnoughMoney: return TTLocalizer.CatalogPurchaseGiftNotEnoughMoney else: return TTLocalizer.CatalogPurchaseGiftGeneralError % {'friend': '%s', 'error': retcode} def acceptItem(self, mailbox, index, callback): mailbox.acceptItem(self, index, callback) def discardItem(self, mailbox, index, callback): print('Item discardItem') mailbox.discardItem(self, index, callback) def acceptItemCleanup(self): pass def getAcceptItemErrorText(self, retcode): if retcode == ToontownGlobals.P_NoRoomForItem: return TTLocalizer.CatalogAcceptRoomError elif retcode == ToontownGlobals.P_ReachedPurchaseLimit: return TTLocalizer.CatalogAcceptLimitError elif retcode == ToontownGlobals.P_WillNotFit: return TTLocalizer.CatalogAcceptFitError elif retcode == ToontownGlobals.P_InvalidIndex: return TTLocalizer.CatalogAcceptInvalidError else: return TTLocalizer.CatalogAcceptGeneralError % retcode def output(self, store = -1): return 'CatalogItem' def getFilename(self): return '' def getColor(self): return None def formatOptionalData(self, store = -1): result = '' if store & Location and self.posHpr != None: result += ', posHpr = (%s, %s, %s, %s, %s, %s)' % self.posHpr return result def __str__(self): return self.output() def __repr__(self): return self.output() def compareTo(self, other): return 0 def getHashContents(self): return None def __cmp__(self, other): c = cmp(self.__class__, other.__class__) if c != 0: return c return self.compareTo(other) def __hash__(self): return hash((self.__class__, self.getHashContents())) def getBasePrice(self): return 0 def getEmblemPrices(self): return () def loadModel(self): return None def decodeDatagram(self, di, versionNumber, store): if store & DeliveryDate: self.deliveryDate = di.getUint32() if store & Location: x = di.getArg(STInt16, 10) y = di.getArg(STInt16, 10) z = di.getArg(STInt16, 100) if versionNumber < 2: h = di.getArg(STInt16, 10) p = 0.0 r = 0.0 elif versionNumber < 5: h = di.getArg(STInt8, 256.0 / 360.0) p = di.getArg(STInt8, 256.0 / 360.0) r = di.getArg(STInt8, 256.0 / 360.0) hpr = oldToNewHpr(VBase3(h, p, r)) h = hpr[0] p = hpr[1] r = hpr[2] else: h = di.getArg(STInt8, 256.0 / 360.0) p = di.getArg(STInt8, 256.0 / 360.0) r = di.getArg(STInt8, 256.0 / 360.0) self.posHpr = (x, y, z, h, p, r) if store & GiftTag: self.giftTag = di.getString() if versionNumber >= 8: self.specialEventId = di.getUint8() else: self.specialEventId = 0 def encodeDatagram(self, dg, store): if store & DeliveryDate: dg.addUint32(self.deliveryDate) if store & Location: dg.putArg(self.posHpr[0], STInt16, 10) dg.putArg(self.posHpr[1], STInt16, 10) dg.putArg(self.posHpr[2], STInt16, 100) dg.putArg(self.posHpr[3], STInt8, 256.0 / 360.0) dg.putArg(self.posHpr[4], STInt8, 256.0 / 360.0) dg.putArg(self.posHpr[5], STInt8, 256.0 / 360.0) if store & GiftTag: dg.addString(self.giftTag) dg.addUint8(self.specialEventId) def getTypeCode(self): from . import CatalogItemTypes return CatalogItemTypes.CatalogItemTypes[self.__class__] def applyColor(self, model, colorDesc): if model == None or colorDesc == None: return for partName, color in colorDesc: matches = model.findAllMatches(partName) if color == None: matches.hide() elif isinstance(color, bytes): tex = loader.loadTexture(color) tex.setMinfilter(Texture.FTLinearMipmapLinear) tex.setMagfilter(Texture.FTLinear) for i in range(matches.getNumPaths()): matches.getPath(i).setTexture(tex, 1) else: needsAlpha = color[3] != 1 color = VBase4(color[0], color[1], color[2], color[3]) for i in range(matches.getNumPaths()): matches.getPath(i).setColorScale(color, 1) if needsAlpha: matches.getPath(i).setTransparency(1) return def makeFrame(self): from direct.gui.DirectGui import DirectFrame frame = DirectFrame(parent=hidden, frameSize=(-1.0, 1.0, -1.0, 1.0), relief=None) return frame def makeFrameModel(self, model, spin = 1): frame = self.makeFrame() ival = None if model: model.setDepthTest(1) model.setDepthWrite(1) if spin: pitch = frame.attachNewNode('pitch') rotate = pitch.attachNewNode('rotate') scale = rotate.attachNewNode('scale') model.reparentTo(scale) bMin, bMax = model.getTightBounds() center = (bMin + bMax) / 2.0 model.setPos(-center[0], -center[1], -center[2]) pitch.setP(20) bMin, bMax = pitch.getTightBounds() center = (bMin + bMax) / 2.0 corner = Vec3(bMax - center) scale.setScale(1.0 / max(corner[0], corner[1], corner[2])) pitch.setY(2) ival = LerpHprInterval(rotate, 10, VBase3(-270, 0, 0), startHpr=VBase3(90, 0, 0)) else: scale = frame.attachNewNode('scale') model.reparentTo(scale) bMin, bMax = model.getTightBounds() center = (bMin + bMax) / 2.0 model.setPos(-center[0], 2, -center[2]) corner = Vec3(bMax - center) scale.setScale(1.0 / max(corner[0], corner[1], corner[2])) return (frame, ival) def getBlob(self, store = 0): dg = PyDatagram() dg.addUint8(CatalogItemVersion) encodeCatalogItem(dg, self, store) return dg.getMessage() def getRequestPurchaseErrorTextTimeout(self): return 6 def getDaysToGo(self, avatar): accountDays = avatar.getAccountDays() daysToGo = self.loyaltyRequirement() - accountDays if daysToGo < 0: daysToGo = 0 return int(daysToGo) def encodeCatalogItem(dg, item, store): from . import CatalogItemTypes flags = item.getTypeCode() if item.isSaleItem(): flags |= CatalogItemTypes.CatalogItemSaleFlag if item.giftTag != None: flags |= CatalogItemTypes.CatalogItemGiftTag dg.addUint8(flags) if item.giftTag != None: dg.addUint32(item.giftTag) if not item.giftCode: item.giftCode = 0 dg.addUint8(item.giftCode) item.encodeDatagram(dg, store) return def decodeCatalogItem(di, versionNumber, store): global CatalogReverseType from . import CatalogItemTypes if CatalogReverseType == None: CatalogReverseType = {} for itemClass, index in list(CatalogItemTypes.CatalogItemTypes.items()): CatalogReverseType[index] = itemClass startIndex = di.getCurrentIndex() try: flags = di.getUint8() typeIndex = flags & CatalogItemTypes.CatalogItemTypeMask gift = None code = None if flags & CatalogItemTypes.CatalogItemGiftTag: gift = di.getUint32() code = di.getUint8() itemClass = CatalogReverseType[typeIndex] item = itemClass(di, versionNumber, store=store) except Exception as e: CatalogItem.notify.warning('Invalid catalog item in stream: %s, %s' % (sys.exc_info()[0], e)) d = Datagram(di.getDatagram().getMessage()[startIndex:]) d.dumpHex(Notify.out()) from . import CatalogInvalidItem return CatalogInvalidItem.CatalogInvalidItem() if flags & CatalogItemTypes.CatalogItemSaleFlag: item.saleItem = 1 item.giftTag = gift item.giftCode = code return item def getItem(blob, store = 0): dg = PyDatagram(blob) di = PyDatagramIterator(dg) try: versionNumber = di.getUint8() return decodeCatalogItem(di, versionNumber, store) except Exception as e: CatalogItem.notify.warning('Invalid catalog item: %s, %s' % (sys.exc_info()[0], e)) dg.dumpHex(Notify.out()) from . import CatalogInvalidItem return CatalogInvalidItem.CatalogInvalidItem()