"""Controlled entrance states, with the native hourly algorithm left intact."""from importlib.machinery import SourceFileLoaderimport json,sysrunner=SourceFileLoader('rules','/workspace/analysis/rule-runner.py').load_module()from cpc import Dumpcase=sys.argv[1];e=runner.Experiment('enemy-'+case,62,position_start=True)for room,field in ((19,0),(31,1),(32,3)): base=0x6876+(room-1)*50 if case=='full': for i in range(6):e.pokes.append(f'11000:{base+i:04x}=1e') e.pokes.append(f'11000:{base+6:04x}=05') else:e.pokes.append(f'11000:{base+field:04x}={10 if case=="blocked" else 30:02x}')e.mark('before',0);e.mark('after',15500);e.run(watch='6620-7941')out=runner.ROOT/f'analysis/runs/enemy-{case}';states=[]for frame,label in e.marks: d=Dump(out/f'dump-{frame:05d}.bin') rooms={str(i):list(d.mem[0x6876+(i-1)*50:0x6876+i*50]) for i in range(1,91)} states.append({'label':label,'clock':list(d.mem[0x661e:0x6620]),'target':d.mem[0x6621],'timer':int.from_bytes(d.mem[0x6720:0x6722],'little'),'rooms':rooms})(out/'enemy-states.json').write_text(json.dumps(states,indent=2))print(case,'quota',[s['target'] for s in states],'entry counts',[[s['rooms'][str(i)][6] for i in (19,31,32)] for s in states])