95 lines
5.7 KiB
Python
95 lines
5.7 KiB
Python
"""EXPERIMENTAL adapter; inherited bounded process/socket cleanup stays intact."""
|
|
import argparse
|
|
from pathlib import Path
|
|
import subprocess
|
|
import sys
|
|
from campaign import Campaign
|
|
from scenario import SCENARIO
|
|
|
|
p = argparse.ArgumentParser()
|
|
p.add_argument('lab', type=Path)
|
|
p.add_argument('mode', choices=('control', 'private_standard', 'private_mosaic'))
|
|
p.add_argument('--source-only', action='store_true',
|
|
help='generate and syntax-check the child without reserving a run')
|
|
a = p.parse_args()
|
|
lab = a.lab.resolve()
|
|
repo = Path(__file__).resolve().parents[3]
|
|
# WHY: one lock covers all modes, rotations and complete child cleanup.
|
|
# CONTRACT: repeated/failed runs cannot silently replenish campaign budgets.
|
|
# INVARIANT: no dynamic action is launched before its persistent reservation.
|
|
if not a.source_only:
|
|
campaign = Campaign(lab)
|
|
campaign.begin(a.mode)
|
|
s = (repo / 'tooling/probes/step20c-applied-state-input/run.py').read_text()
|
|
s = s.replace("f'{args.mode}-{args.kind}-asi{args.asi_mode}-{time.time_ns()}'",
|
|
"f'{args.mode}-rotated-{time.time_ns()}'")
|
|
s = s.replace('output HEADLESS-1 resolution 800x600 position -800 0',
|
|
'output HEADLESS-1 resolution 1000x750 scale 1.25 position -800 0')
|
|
s = s.replace("Path('/usr/bin/sway')", "lab / 'standard-build/sway/sway'")
|
|
s = s.replace(" snapshot('startup')", " time.sleep(1.0)\n snapshot('startup')")
|
|
s = s.replace('import argparse', 'import argparse\nfrom PIL import Image, ImageDraw\nimport sys\nsys.path.insert(0, '+repr(str(Path(__file__).parent.resolve()))+')\nfrom campaign import reserve')
|
|
s = s.replace("run.mkdir(parents=True, mode=0o700)", """run.mkdir(parents=True, mode=0o700)
|
|
(run / 'fixture-source.py').write_text(Path(__file__).read_text())
|
|
# WHY: five asymmetric markers identify whether PNG uses logical or buffer axes.
|
|
# CONTRACT: predeclared logical centers and distinct exact RGB codes.
|
|
marker_path = run / 'marker.png'
|
|
marker = Image.new('RGB', (800, 600), (30, 30, 46))
|
|
draw = ImageDraw.Draw(marker)
|
|
for name, box, color in (
|
|
('TL', (15, 15, 44, 44), (255, 32, 32)),
|
|
('TR', (755, 15, 784, 44), (32, 255, 32)),
|
|
('BL', (15, 555, 44, 584), (32, 32, 255)),
|
|
('BR', (755, 555, 784, 584), (255, 255, 32)),
|
|
('C', (385, 285, 414, 314), (192, 32, 255))):
|
|
draw.rectangle(box, fill=color)
|
|
marker.save(marker_path)
|
|
""")
|
|
s = s.replace('env = os.environ.copy()', "(run / 'config').write_text((run / 'config').read_text() + f'output * bg {marker_path} fill\\n')\nenv = os.environ.copy()", 1)
|
|
start = s.index(" expected = {'HEADLESS-1':")
|
|
end = s.index(" for output, side in", start)
|
|
s = s[:start]+''' expected = {'HEADLESS-1': (-800, 0, 800, 600), 'HEADLESS-2': (0, 0, 800, 600)}
|
|
actual = {o['name']: tuple(o['rect'][k] for k in ('x','y','width','height')) for o in outputs if o['active']}
|
|
rotated = globals().get('rotation_stage') == 'R90'
|
|
h1 = next(o for o in outputs if o['name'] == 'HEADLESS-1')
|
|
h2 = next(o for o in outputs if o['name'] == 'HEADLESS-2')
|
|
wanted_mode = (750, 1000) if rotated else (1000, 750)
|
|
if (actual != expected or h1['scale'] != 1.25 or h2['scale'] != 1 or
|
|
h2['transform'] != 'normal' or h1['transform'] != ('90' if rotated else 'normal') or
|
|
(h1['current_mode']['width'], h1['current_mode']['height']) != wanted_mode):
|
|
raise RuntimeError(f'applied rotation geometry gate failed: {actual} {h1}')
|
|
record('state', tag=tag, workspaces=ipc('-t','get_workspaces'), workspace=workspace(), owner=owner(), outputs=outputs)
|
|
'''+s[end:]
|
|
s = s.replace("['grim', '-o', output, str(path)]", "['grim', '-s', str(next(o['scale'] for o in outputs if o['name']==output)), '-o', output, str(path)]")
|
|
s = s.replace(" record('capture', tag=tag", """ out = next(o for o in outputs if o['name']==output)
|
|
dimensions = list(Image.open(path).size)
|
|
record('capture_geometry',tag=tag,output=output,dimensions=dimensions,rect=out['rect'],scale=out['scale'],transform=out['transform'],mode=out['current_mode'], physical_expected=[out['current_mode']['width'],out['current_mode']['height']])
|
|
record('capture', tag=tag""")
|
|
start=s.index('def inject(action, x, y):');end=s.index('\ndef client_command',start)
|
|
s=s[:start]+'''def inject(action, x, y):
|
|
outputs = [o for o in ipc('-t','get_outputs') if o['active']]
|
|
min_x = min(o['rect']['x'] for o in outputs)
|
|
min_y = min(o['rect']['y'] for o in outputs)
|
|
max_x = max(o['rect']['x']+o['rect']['width'] for o in outputs)
|
|
max_y = max(o['rect']['y']+o['rect']['height'] for o in outputs)
|
|
if not any(o['rect']['x']<=x<o['rect']['x']+o['rect']['width'] and o['rect']['y']<=y<o['rect']['y']+o['rect']['height'] for o in outputs):
|
|
raise ValueError('global point is outside applied outputs')
|
|
sent_x,sent_y = int(x-min_x),int(y-min_y)
|
|
width,height = max_x-min_x,max_y-min_y
|
|
marker=f'INJECT_DONE {action} {sent_x},{sent_y}'
|
|
before=(run/'pointer.log').read_text().count(marker)
|
|
record('pointer_request',action=action,global_point=[x,y],sent=[sent_x,sent_y],extent=[width,height],minimum=[min_x,min_y])
|
|
pointer.stdin.write(f'{action} {sent_x} {sent_y} {width} {height}\\n');pointer.stdin.flush()
|
|
wait(lambda: (run/'pointer.log').read_text().count(marker)>before)
|
|
|
|
'''+s[end:]
|
|
start=s.index(" if args.kind == 'cycles':\n");end=s.index('except Exception as exc:',start)
|
|
scenario=SCENARIO
|
|
s=s[:start]+scenario+'\n'+s[end:]
|
|
generated=lab/f'rotated-{a.mode}.py'
|
|
generated.write_text(s)
|
|
if a.source_only:
|
|
import ast
|
|
ast.parse(s, filename=str(generated))
|
|
print(f'SOURCE_ONLY_PASS={generated}')
|
|
raise SystemExit(0)
|
|
raise SystemExit(subprocess.call([sys.executable,str(generated),str(lab),a.mode,'cycles','--max-seconds','360','--max-captures','120','--max-bytes','70000000'], pass_fds=(campaign.lock.fileno(),)))
|