from direct.interval.IntervalGlobal import * from pandac.PandaModules import * import random, copy from toontown.toonbase import TTLocalizer from toontown.toonbase import ToontownBattleGlobals from toontown.suit import SuitDNA if process == 'client': from toontown.battle import BattleParticles try: config = base.config except: config = simbase.config EFFECT_RADIUS = 30 RESISTANCE_TOONUP = 0 RESISTANCE_RESTOCK = 1 RESISTANCE_MONEY = 2 RESISTANCE_TICKETS = 3 RESISTANCE_MERITS = 4 resistanceMenu = [RESISTANCE_TOONUP, RESISTANCE_RESTOCK, RESISTANCE_MONEY, RESISTANCE_TICKETS, RESISTANCE_MERITS] resistanceDict = { RESISTANCE_TOONUP: { 'menuName': TTLocalizer.ResistanceToonupMenu, 'itemText': TTLocalizer.ResistanceToonupItem, 'chatText': TTLocalizer.ResistanceToonupChat, 'values': [10, 20, 40, 80, -1], 'items': [0, 1, 2, 3, 4] }, RESISTANCE_MONEY: { 'menuName': TTLocalizer.ResistanceMoneyMenu, 'itemText': TTLocalizer.ResistanceMoneyItem, 'chatText': TTLocalizer.ResistanceMoneyChat, 'values': [100, 200, 350, 600, 1200, 2400], 'items': [0, 1, 2, 3, 4, 5] }, RESISTANCE_RESTOCK: { 'menuName': TTLocalizer.ResistanceRestockMenu, 'itemText': TTLocalizer.ResistanceRestockItem, 'chatText': TTLocalizer.ResistanceRestockChat, 'values': [ ToontownBattleGlobals.HEAL_TRACK, ToontownBattleGlobals.TRAP_TRACK, ToontownBattleGlobals.LURE_TRACK, ToontownBattleGlobals.SOUND_TRACK, ToontownBattleGlobals.THROW_TRACK, ToontownBattleGlobals.SQUIRT_TRACK, ToontownBattleGlobals.DROP_TRACK, -1 ], 'extra': [ TTLocalizer.MovieNPCSOSHeal, TTLocalizer.MovieNPCSOSTrap, TTLocalizer.MovieNPCSOSLure, TTLocalizer.MovieNPCSOSSound, TTLocalizer.MovieNPCSOSThrow, TTLocalizer.MovieNPCSOSSquirt, TTLocalizer.MovieNPCSOSDrop, TTLocalizer.MovieNPCSOSAll ], 'items': [0, 1, 2, 3, 4, 5, 6, 7] }, RESISTANCE_MERITS: { 'menuName': TTLocalizer.ResistanceMeritsMenu, 'itemText': TTLocalizer.ResistanceMeritsItem, 'chatText': TTLocalizer.ResistanceMeritsChat, 'values': range(len(SuitDNA.suitDepts)) + [-1], 'extra': TTLocalizer.RewardPanelMeritBarLabels + [TTLocalizer.MovieNPCSOSAll], 'items': range(len(SuitDNA.suitDepts) + 1) }, RESISTANCE_TICKETS: { 'menuName': TTLocalizer.ResistanceTicketsMenu, 'itemText': TTLocalizer.ResistanceTicketsItem, 'chatText': TTLocalizer.ResistanceTicketsChat, 'values': [200, 400, 600, 800, 1200], 'items': [0, 1, 2, 3, 4] } } def encodeId(menuIndex, itemIndex): textId = menuIndex * 100 textId += resistanceDict[menuIndex]['items'][itemIndex] return textId def decodeId(textId): menuIndex = int(textId / 100) itemIndex = textId % 100 return (menuIndex, itemIndex) def validateId(textId): if textId < 0: return 0 menuIndex, itemIndex = decodeId(textId) if menuIndex not in resistanceDict: return 0 if itemIndex >= len(resistanceDict[menuIndex]['values']): return 0 return 1 def getItems(menuIndex): return resistanceDict[menuIndex]['items'] def getMenuName(textId): menuIndex, _ = decodeId(textId) return resistanceDict[menuIndex]['menuName'] def getItemText(textId): menuIndex, itemIndex = decodeId(textId) resistance = resistanceDict[menuIndex] value = resistance['values'][itemIndex] text = resistance['itemText'] if menuIndex is RESISTANCE_TOONUP: if value is -1: value = TTLocalizer.ResistanceToonupItemMax elif 'extra' in resistance: value = resistance['extra'][itemIndex] return text % str(value) def getChatText(textId): menuIndex, _ = decodeId(textId) return resistanceDict[menuIndex]['chatText'] def getItemValue(textId): menuIndex, itemIndex = decodeId(textId) return resistanceDict[menuIndex]['values'][itemIndex] def getRandomId(): menuIndex = random.choice(resistanceMenu) itemIndex = random.choice(getItems(menuIndex)) return encodeId(menuIndex, itemIndex) def doEffect(textId, speakingToon, nearbyToons): menuIndex, _ = decodeId(textId) itemValue = getItemValue(textId) if menuIndex == RESISTANCE_TOONUP: effect = BattleParticles.loadParticleFile('resistanceEffectSparkle.ptf') fadeColor = VBase4(1, 0.5, 1, 1) elif menuIndex == RESISTANCE_MONEY: effect = BattleParticles.loadParticleFile('resistanceEffectBean.ptf') bean = loader.loadModel('phase_4/models/props/jellybean4.bam') bean = bean.find('**/jellybean') colors = { 'particles-1': (1, 1, 0, 1), 'particles-2': (1, 0, 0, 1), 'particles-3': (0, 1, 0, 1), 'particles-4': (0, 0, 1, 1), 'particles-5': (1, 0, 1, 1) } for name, color in colors.items(): node = bean.copyTo(NodePath()) node.setColorScale(*color) p = effect.getParticlesNamed(name) p.renderer.setGeomNode(node.node()) fadeColor = VBase4(0, 1, 0, 1) elif menuIndex == RESISTANCE_RESTOCK: effect = BattleParticles.loadParticleFile('resistanceEffectSprite.ptf') invModel = loader.loadModel('phase_3.5/models/gui/inventory_icons') invModel.setScale(4) invModel.flattenLight() icons = [] if itemValue != -1: for item in xrange(6): iconName = ToontownBattleGlobals.AvPropsNew[itemValue][item] icons.append(invModel.find('**/%s' % iconName)) else: tracks = range(7) random.shuffle(tracks) for i in xrange(6): track = tracks[i] item = random.randint(0, 5) iconName = ToontownBattleGlobals.AvPropsNew[track][item] icons.append(invModel.find('**/%s' % iconName)) iconDict = { 'particles-1': icons[0], 'particles-2': icons[1], 'particles-3': icons[2], 'particles-4': icons[3], 'particles-5': icons[4], 'particles-6': icons[5] } for name, icon in iconDict.items(): p = effect.getParticlesNamed(name) p.renderer.setFromNode(icon) fadeColor = VBase4(0, 0, 1, 1) elif menuIndex == RESISTANCE_MERITS: effect = BattleParticles.loadParticleFile('resistanceEffectSprite.ptf') cogModel = loader.loadModel('phase_3/models/gui/cog_icons') cogModel.setScale(0.75) cogModel.flattenLight() if itemValue != -1: iconDict = {'particles-1': cogModel.find(SuitDNA.suitDeptModelPaths[itemValue])} else: iconDict = {} for i in xrange(len(SuitDNA.suitDepts)): iconDict['particles-%s' % (i + 1)] = cogModel.find(SuitDNA.suitDeptModelPaths[i]) for name, icon in iconDict.items(): p = effect.getParticlesNamed(name) p.renderer.setFromNode(icon) fadeColor = VBase4(0.7, 0.7, 0.7, 1.0) cogModel.removeNode() elif menuIndex == RESISTANCE_TICKETS: effect = BattleParticles.loadParticleFile('resistanceEffectSprite.ptf') model = loader.loadModel('phase_6/models/karting/tickets') model.flattenLight() iconDict = {'particles-1': model} for name, icon in iconDict.items(): p = effect.getParticlesNamed(name) p.renderer.setFromNode(icon) fadeColor = VBase4(1, 1, 0, 1) else: return recolorToons = Parallel() for toonId in nearbyToons: toon = base.cr.doId2do.get(toonId) if toon and (not toon.ghostMode): i = Sequence( toon.doToonColorScale(fadeColor, 0.3), toon.doToonColorScale(toon.defaultColorScale, 0.3), Func(toon.restoreDefaultColorScale) ) recolorToons.append(i) i = Parallel( ParticleInterval(effect, speakingToon, worldRelative=0, duration=3, cleanup=True), Sequence(Wait(0.2), recolorToons), autoFinish=1 ) i.start()