SashegDev
874b6f20c2
Основа — Horizon (MIT). История апстрима не переносится: этот репозиторий ведёт свою.
80 lines
4.0 KiB
Python
80 lines
4.0 KiB
Python
#!/usr/bin/env python3
|
|
"""Compare enchanting-table offers (costs, clues, applied enchantments) with vanilla.
|
|
|
|
Java loads the vanilla data pack's enchantment registry and runs
|
|
EnchantmentHelper as EnchantmentMenu does; Rust runs horizon_client::stations
|
|
on the prepared resources. Every enchantable item and the book are tested
|
|
across seeds and bookshelf counts.
|
|
"""
|
|
import argparse
|
|
import json
|
|
import os
|
|
from pathlib import Path
|
|
import platform
|
|
import random
|
|
import subprocess
|
|
try:
|
|
from . import lab, test_rust_parity as base
|
|
except ImportError:
|
|
import lab
|
|
import test_rust_parity as base
|
|
|
|
|
|
def corpus(items):
|
|
rng = random.Random(0x454E4348414E54)
|
|
seeds = [-2147483648, -1, 0, 1, 2147483647, *[rng.randrange(-2**31, 2**31) for _ in range(27)]]
|
|
for item in items:
|
|
for seed in seeds:
|
|
for bookcases in [0, 1, 4, 8, 12, 15, 30]:
|
|
yield f"{seed} {bookcases} {item}"
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument('--prepare', action='store_true')
|
|
args = parser.parse_args()
|
|
build = lab.LOCAL / 'agents/stations/enchanting-parity'
|
|
build.mkdir(parents=True, exist_ok=True)
|
|
if args.prepare:
|
|
lab.prepare()
|
|
lock, jars, libraries = base.verify_artifacts()
|
|
resources = lab.RESOURCES
|
|
catalog = json.loads((resources / 'items.json').read_text())
|
|
items = sorted(k for k, v in catalog.items() if 'minecraft:enchantable' in v)
|
|
oracle = lab.ROOT / 'mods/horizon/src/test/java/dev/horizonlab/parity/VanillaEnchantingOracle.java'
|
|
sources = [oracle, Path(__file__), lab.ROOT / 'crates/horizon-client/examples/probe/enchanting_fixture.rs',
|
|
*sorted((lab.ROOT / 'crates/horizon-client/src').glob('*.rs'))]
|
|
tested = base.source_hashes(sources)
|
|
inputs = build / 'inputs.txt'
|
|
fixtures = list(corpus(items))
|
|
inputs.write_text(''.join(line + '\n' for line in fixtures), encoding='utf-8')
|
|
classes = build / 'classes'
|
|
classes.mkdir(exist_ok=True)
|
|
deps = os.pathsep.join(map(str, libraries))
|
|
|
|
def run(command, name):
|
|
with (build / (name + '.log')).open('w', encoding='utf-8') as log:
|
|
base.run(list(map(str, command)), stdout=log, stderr=subprocess.STDOUT)
|
|
run(['javac', '--release', lock['minecraft']['java_major'], '-proc:none','-sourcepath',str(lab.ROOT/'mods/horizon/src/test/java'), '-cp', str(jars['server']) + os.pathsep + deps, '-d', classes, oracle], 'compile-java')
|
|
run(['cargo', 'build', '--profile', 'fast', '-p', 'horizon-client', '--example', 'client_probe'], 'compile-rust')
|
|
metadata = json.loads(subprocess.check_output(['cargo', 'metadata', '--no-deps', '--format-version', '1'], cwd=lab.ROOT, text=True))
|
|
binary = Path(metadata['target_directory']) / 'fast/examples' / ('client_probe.exe' if os.name=='nt' else 'client_probe')
|
|
run([binary, 'enchanting_fixture',resources, inputs, build / 'rust.txt'], 'rust')
|
|
references = []
|
|
for side in ['server']:
|
|
run(['java', '-Xmx1G', '-cp', os.pathsep.join([str(classes), str(jars[side]), deps]), 'dev.horizonlab.parity.VanillaEnchantingOracle', inputs, build / f'{side}.txt'], side)
|
|
matched = base.compare(build / f'{side}.txt', build / 'rust.txt')
|
|
references.append({'side': side, 'jar_sha256': lab.digest(jars[side], 'sha256'), 'matched_output_lines': matched})
|
|
print(f'{side}: {matched:,} enchanting offer lines match Rust', flush=True)
|
|
base.ensure_sources_unchanged(sources, tested)
|
|
report = {'schema': 1, 'minecraft': lock['minecraft']['version'], 'platform': platform.platform(), 'fixtures': len(fixtures),
|
|
'items': len(items), 'references': references, 'sources_sha256': tested,
|
|
'scope': 'EnchantmentHelper.getEnchantmentCost / selectEnchantment on the data-driven enchantment registry and '
|
|
'#in_enchanting_table, EnchantmentMenu clue draws and book removal, for every enchantable item and the book.'}
|
|
lab.write_json(build / 'report.json', report)
|
|
print(f'Enchanting parity passed: {build / "report.json"}')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|