"""Drive the three original trucks with controlled starting position/key/passengers."""from importlib.machinery import SourceFileLoaderfrom pathlib import Pathimport json,sysrunner=SourceFileLoader('rules','/workspace/analysis/rule-runner.py').load_module()from cpc import Dumproot=Path('/workspace');world=json.loads((root/'analysis/extracted/world.json').read_text())rooms={r['id']:r for r in world['rooms']+world['vehicle_interiors']}source=int(sys.argv[1]);key=int(sys.argv[2]);door={72:2,77:1,82:3}[source]inside=door+86;target={72:66,77:65,82:64}[source]e=runner.Experiment(f'truck-{source}-key-{key}',source,position_start=True)def press(name): e.events += [f'{e.frame}:{name}+',f'{e.frame+24}:{name}-'];e.frame+=220e.key=presse.pokes += [f'8500:{0x6876+(source-1)*50+6:04x}=00']e.pokes += [f'8500:{0x6876+(target-1)*50+6:04x}=00']if key:e.pokes += ['11000:66c7=30']# Alexandre is in the same cab; Sylvie in a different cab. These are controlled riders.pos=int(rooms[source]['map_address'],16)e.pokes += [f'11000:6633={pos&255:02x}',f'11000:6634={pos>>8:02x}',f'11000:6653={inside:02x}', f'11000:667b={87 if inside==89 else 89:02x}']def cell(room,obj): y,x=next((y,x) for y,row in enumerate(rooms[room]['objects_grid']) for x,v in enumerate(row) if v==obj) e.cell(y,x)e.mark('initial');e.select(8);e.key('copy');cell(source,door);e.key('copy');e.mark('open')e.select(16);e.key('copy');e.mark('inside')e.select(4);e.key('copy');cell(inside,63);e.key('copy');e.mark('contact')e.key('copy');cell(inside,71);e.key('copy');e.mark('driven',1800)e.select(17);e.key('copy');e.mark('outside',500)e.pokes += [f'{e.frame}:66ab={pos&255:02x}',f'{e.frame}:66ac={pos>>8:02x}']e.select(10);e.key('copy');e.key('copy');e.mark('source-after',500)run=runner.subprocess.runrunner.subprocess.run=lambda cmd,**kwargs:run(cmd+['-b','9d98','-b-after','11200'],**kwargs)e.run()out=root/f'analysis/runs/{e.name}';reports=[]for frame,label in e.marks: d=Dump(out/f'dump-{frame:05d}.bin') reports.append({'label':label,'rooms':{str(i):list(d.mem[0x6876+(i-1)*50:0x6876+i*50]) for i in (19,31,32,64,65,66,72,77,82,87,88,89)}, 'positions':[int.from_bytes(d.mem[a:a+2],'little')-0x6726 for a in range(0x6633,0x66d3,40)], 'interiors':[d.mem[a+32] for a in range(0x6633,0x66d3,40)]})(out/'truck-states.json').write_text(json.dumps(reports,indent=2))if key: for room,variant in ((source,'empty'),(target,'parked')): records=[] for p in out.glob('bp9d98-*.bin'): d=Dump(p);rp=int.from_bytes(d.mem[0x6617:0x6619],'little') if rp==0x6876+(room-1)*50 and d.mem[rp+9]==(0 if variant=='empty' else 77): records.append((int(p.stem.split('-')[-1]),d)) assert records,(room,variant) background=max(records,key=lambda p:p[0])[1].screen().crop((8,48,168,128)) assets=root/'analysis/extracted/trucks';assets.mkdir(exist_ok=True) background.save(assets/f'background-{room:02d}-{variant}.png')print('Truck observation',source,'key',key)