"""Prepare active equipment and one pressure peak; native health/damage/random code is untouched."""from importlib.machinery import SourceFileLoaderimport json,re,sysrunner=SourceFileLoader('rules','/workspace/analysis/rule-runner.py').load_module()from cpc import Dumpweapon=int(sys.argv[1]);e=runner.Experiment(f'melee-damage-{weapon}',41,position_start=True)e.pokes += [f'8500:66c7={weapon:02x}',f'8500:66d2={weapon:02x}']e.pc_pokes += ['880d:6703=14'];e.mark('after',0)run=runner.subprocess.run;runner.subprocess.run=lambda cmd,**kwargs:run(cmd+['-b','880d','-b-after','10000'],**kwargs)e.run(watch='670c-670d');out=runner.ROOT/f'analysis/runs/{e.name}'paths=sorted(out.glob('bp880d-*.bin'),key=lambda p:int(p.stem.split('-')[-1]));assert pathsd=Dump(paths[0]);before={'F':d.mem[0x66bd],'H':d.mem[0x66be],'hp':d.mem[0x670c], 'threshold':d.mem[0x6704],'bonus':d.mem[0x6706],'peak':d.mem[0x6703]}writes=[]for line in (out/'log.txt').read_text().splitlines(): m=re.fullmatch(r'write 670c=([0-9a-f]+) pc=888e f=(\d+)',line) if m and int(m[2])>=10000:writes.append(int(m[1],16))assert writesresult={'weapon':weapon,'before':before,'afterHP':writes[0],'damage':before['hp']-writes[0]}(out/'melee.json').write_text(json.dumps(result,indent=2));print(result)