Ludovic Marie
8f13b2be8c
Rockets list their flight duration and each run of equal explosions (shape, colours, fade colours, trail, twinkle); firework stars list their explosion. Both honour TooltipDisplay. New harness: tools/test_firework_tooltip_parity.py. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
73 lines
3.3 KiB
Python
73 lines
3.3 KiB
Python
#!/usr/bin/env python3
|
|
"""Compare firework rocket and firework star tooltips with the pinned vanilla client.
|
|
|
|
Random rockets (flight durations 0 to 3 and higher, up to six explosions, runs of equal ones) and stars, with dye and
|
|
custom colours, fade colours, trails and twinkles, some without the component (a rocket's default) and some whose
|
|
tooltip_display hides it. Vanilla prints what `Fireworks.addToTooltip` and `FireworkExplosion.addToTooltip` add
|
|
(VanillaFireworkTooltipOracle); Horizon what `item_tooltip::firework_lines` gives (examples/firework_tooltip_fixture).
|
|
"""
|
|
import random
|
|
try:
|
|
from . import lab, oracle_harness as harness
|
|
except ImportError:
|
|
import lab, oracle_harness as harness
|
|
|
|
SHAPES = ['small_ball', 'large_ball', 'star', 'creeper', 'burst']
|
|
DYES = [0xF0F0F0, 0xEB8844, 0xC354CD, 0x6689D3, 0xDECF2A, 0x41CD34, 0xD88198, 0x434343, 0xABABAB, 0x287697,
|
|
0x7B2FBE, 0x253192, 0x51301A, 0x3B511A, 0xB3312C, 0x1E1B1B]
|
|
COUNT = 3000
|
|
|
|
|
|
def explosion(rng):
|
|
def colors():
|
|
return [rng.choice(DYES) if rng.random() < 0.8 else rng.randrange(1 << 24) for _ in range(rng.choice([0, 1, 1, 2, 3]))]
|
|
fields = [f'shape:"{rng.choice(SHAPES)}"']
|
|
for key, values in (('colors', colors()), ('fade_colors', colors() if rng.random() < 0.4 else [])):
|
|
if values:
|
|
fields.append(f'{key}:[I;{",".join(map(str, values))}]')
|
|
if rng.random() < 0.4:
|
|
fields.append('has_trail:1b')
|
|
if rng.random() < 0.4:
|
|
fields.append('has_twinkle:1b')
|
|
return '{' + ','.join(fields) + '}'
|
|
|
|
|
|
def stack(rng):
|
|
components = []
|
|
if rng.random() < 0.7:
|
|
item = 'firework_rocket'
|
|
if rng.random() < 0.9:
|
|
explosions = []
|
|
for _ in range(rng.choice([0, 0, 1, 1, 2, 3, 6])):
|
|
# Runs of equal explosions are counted.
|
|
explosions.append(explosions[-1] if explosions and rng.random() < 0.4 else explosion(rng))
|
|
flight = rng.choice([0, 1, 1, 2, 3, 3, 7, 255])
|
|
components.append(f'"minecraft:fireworks":{{flight_duration:{flight},explosions:[{",".join(explosions)}]}}')
|
|
else:
|
|
item = 'firework_star'
|
|
if rng.random() < 0.9:
|
|
components.append(f'"minecraft:firework_explosion":{explosion(rng)}')
|
|
if rng.random() < 0.1:
|
|
hidden = rng.choice(['minecraft:fireworks', 'minecraft:firework_explosion'])
|
|
components.append(f'"minecraft:tooltip_display":{{hidden_components:["{hidden}"]}}')
|
|
body = f',components:{{{",".join(components)}}}' if components else ''
|
|
return f'{{id:"minecraft:{item}",count:1{body}}}'
|
|
|
|
|
|
def cases(build):
|
|
rng = random.Random(0x46495245)
|
|
path = build / 'cases.txt'
|
|
resources = lab.LOCAL / 'horizon-client' / '26.3'
|
|
path.write_text(f'#resources {resources}\n' + ''.join(stack(rng) + '\n' for _ in range(COUNT)))
|
|
return path, COUNT
|
|
|
|
|
|
if __name__ == '__main__':
|
|
harness.differential(
|
|
'firework-tooltip-parity', 'VanillaFireworkTooltipOracle', 'firework_tooltip_fixture', cases,
|
|
'Firework rocket and star tooltip lines: flight duration, explosions (counted runs), colours, fade, trail, '
|
|
'twinkle, defaults and tooltip_display',
|
|
sources=[lab.ROOT / 'crates/horizon-client/src/item_tooltip.rs', lab.ROOT / 'crates/horizon-client/src/item_data.rs',
|
|
lab.ROOT / 'crates/horizon-client/src/particles.rs'],
|
|
expected_lines=COUNT)
|