"""Prepare deprivation counters and one real CPC periodic pass, or a sleep wake boundary."""from importlib.machinery import SourceFileLoaderimport sysr=SourceFileLoader('rules','/workspace/analysis/rule-runner.py').load_module()case=sys.argv[1];e=r.Experiment('deprivation-'+case,62,position_start=True)food=int(case[4:]) if case.startswith('food') else (5 if case.startswith('wake') else 0)awake=int(case[5:]) if case.startswith('awake') else (26 if case in ('forced-sleep','forced-dead','forced-gone','repeat','wake-period') else (20 if case.startswith('wake') else 0))state={'forced-sleep':16,'forced-dead':64,'forced-gone':255,'wake':16,'wake-cap':16}.get(case,16 if case.startswith('wake') else 0)s=3 if case in ('starve','combined-ill','combined-corpse') else (50 if case in ('wake','wake-period') else (77 if case=='wake-cap' else 80))if case in ('starve','combined-ill','combined-corpse'):food=6ill=1 if case=='combined-ill' else 0e.pokes += [f'11000:6650={food:02x}',f'11000:6651={awake:02x}',f'11000:6654={state:02x}',f'11000:6644={s:02x}',f'11000:6655={ill:02x}']if case=='combined-corpse':e.pokes+=['11000:7468=01']for t in range(10050,12500,300):e.pokes += [f'{t}:6717=00',f'{t}:6718=00']e.mark('before',0)if case=='wake-period':e.pokes += [f'{e.frame}:6720=ff',f'{e.frame}:6721=3a']if case.startswith('wake'): e.pokes += [f'{e.frame}:6656=ff',f'{e.frame}:6657=d3',f'{e.frame}:6658=01'];e.mark('after',250)else: e.pokes += [f'{e.frame}:6720=ff',f'{e.frame}:6721=3a'];e.mark('after',250)if case=='repeat': e.pokes += [f'{e.frame}:6720=ff',f'{e.frame}:6721=3a'];e.mark('again',250)e.run(watch='6644-665a')from cpc import Dumpout=r.ROOT/f'analysis/runs/{e.name}';vals=[]for f,label in e.marks: d=Dump(out/f'dump-{f:05d}.bin');vals.append({'label':label,'S':d.mem[0x6644],'state':d.mem[0x6654],'food':d.mem[0x6650],'awake':d.mem[0x6651],'sleepTicks':int.from_bytes(d.mem[0x6656:0x6659],'little')})(out/'deprivation.json').write_text(r.json.dumps(vals,indent=2))