190 lines
6.9 KiB
Python
190 lines
6.9 KiB
Python
from direct.interval.IntervalGlobal import *
|
|
from pandac.PandaModules import *
|
|
from toontown.toonbase import TTLocalizer
|
|
from toontown.toonbase import ToontownBattleGlobals
|
|
from toontown.battle import BattleParticles
|
|
from direct.particles import Particles
|
|
import random
|
|
TTBG = ToontownBattleGlobals
|
|
TTL = TTLocalizer
|
|
EFFECT_RADIUS = 30
|
|
RESISTANCE_TOONUP = 0
|
|
RESISTANCE_RESTOCK = 1
|
|
RESISTANCE_MONEY = 2
|
|
resistanceMenu = [RESISTANCE_TOONUP, RESISTANCE_RESTOCK, RESISTANCE_MONEY]
|
|
resistanceDict = {RESISTANCE_TOONUP: {'menuName': TTL.ResistanceToonupMenu,
|
|
'itemText': TTL.ResistanceToonupItem,
|
|
'chatText': TTL.ResistanceToonupChat,
|
|
'values': [10,
|
|
20,
|
|
40,
|
|
80,
|
|
-1],
|
|
'items': [0,
|
|
1,
|
|
2,
|
|
3,
|
|
4]},
|
|
RESISTANCE_MONEY: {'menuName': TTL.ResistanceMoneyMenu,
|
|
'itemText': TTL.ResistanceMoneyItem,
|
|
'chatText': TTL.ResistanceMoneyChat,
|
|
'values': [100,
|
|
200,
|
|
350,
|
|
600],
|
|
'items': [0,
|
|
1,
|
|
2,
|
|
3]},
|
|
RESISTANCE_RESTOCK: {'menuName': TTL.ResistanceRestockMenu,
|
|
'itemText': TTL.ResistanceRestockItem,
|
|
'chatText': TTL.ResistanceRestockChat,
|
|
'values': [TTBG.HEAL_TRACK,
|
|
TTBG.TRAP_TRACK,
|
|
TTBG.LURE_TRACK,
|
|
TTBG.SOUND_TRACK,
|
|
TTBG.THROW_TRACK,
|
|
TTBG.SQUIRT_TRACK,
|
|
TTBG.DROP_TRACK,
|
|
-1],
|
|
'extra': [TTL.MovieNPCSOSHeal,
|
|
TTL.MovieNPCSOSTrap,
|
|
TTL.MovieNPCSOSLure,
|
|
TTL.MovieNPCSOSSound,
|
|
TTL.MovieNPCSOSThrow,
|
|
TTL.MovieNPCSOSSquirt,
|
|
TTL.MovieNPCSOSDrop,
|
|
TTL.MovieNPCSOSAll],
|
|
'items': [0,
|
|
1,
|
|
2,
|
|
3,
|
|
4,
|
|
5,
|
|
6,
|
|
7]}}
|
|
|
|
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, itemIndex = decodeId(textId)
|
|
return resistanceDict[menuIndex]['menuName']
|
|
|
|
|
|
def getItemText(textId):
|
|
menuIndex, itemIndex = decodeId(textId)
|
|
value = resistanceDict[menuIndex]['values'][itemIndex]
|
|
text = resistanceDict[menuIndex]['itemText']
|
|
if menuIndex is RESISTANCE_TOONUP:
|
|
if value is -1:
|
|
value = TTL.ResistanceToonupItemMax
|
|
elif menuIndex is RESISTANCE_RESTOCK:
|
|
value = resistanceDict[menuIndex]['extra'][itemIndex]
|
|
return text % str(value)
|
|
|
|
|
|
def getChatText(textId):
|
|
menuIndex, itemIndex = 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, itemIndex = 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 list(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 range(6):
|
|
iconName = ToontownBattleGlobals.AvPropsNew[itemValue][item]
|
|
icons.append(invModel.find('**/%s' % iconName))
|
|
|
|
else:
|
|
tracks = list(range(7))
|
|
random.shuffle(tracks)
|
|
for i in range(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 list(iconDict.items()):
|
|
p = effect.getParticlesNamed(name)
|
|
p.renderer.setFromNode(icon)
|
|
|
|
fadeColor = VBase4(0, 0, 1, 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()
|