| |
| """Build the Ink Bench reference layers (data/bench/) from the page's own data. |
| |
| Each winding gets <id>_ink.png (published ink preview) and <id>_ct.png (X-ray layer |
| cropped from the atlas), resampled to the same grid, long side LONG px. Windings with our |
| ink_9um run also get <id>_gpu.png on that grid. Lossless 8-bit gray PNG, so the browser |
| and tools/inkbench.py read bit-identical pixels. |
| """ |
| import hashlib, json, os |
| from PIL import Image |
| LONG = 640 |
| root = os.path.join(os.path.dirname(__file__), '..') |
| d = json.load(open(os.path.join(root, 'data/scroll.json'))) |
| gpu = json.load(open(os.path.join(root, 'data/gpu.json')))['windings'] |
| ct = Image.open(os.path.join(root, 'data/atlas_ct.jpg')).convert('L') |
| out = {'schema': 'unroll-bench-refs/1', 'scroll': d['scroll'], 'long': LONG, 'segments': {}} |
| def save(im, name): |
| p = os.path.join(root, 'data/bench', name); im.save(p, optimize=True) |
| return {'file': 'data/bench/' + name, 'sha256': hashlib.sha256(open(p, 'rb').read()).hexdigest()} |
| for w in d['windings']: |
| hi = Image.open(os.path.join(root, 'data', w['hi'])).convert('L') |
| s = LONG / max(hi.size); W, H = round(hi.size[0] * s), round(hi.size[1] * s) |
| e = {'width': W, 'height': H, 'widthMM': w['widthMM'], 'heightMM': w['heightMM'], |
| 'pxPerMM': round(W / w['widthMM'], 4), 'layers': {}} |
| e['layers']['ink'] = save(hi.resize((W, H), Image.LANCZOS), f"{w['id']}_ink.png") |
| u0, v0, u1, v1 = w['atlas'] |
| c = ct.crop((round(u0 * 4096), round(v0 * 4096), round(u1 * 4096), round(v1 * 4096))) |
| e['layers']['ct'] = save(c.resize((W, H), Image.LANCZOS), f"{w['id']}_ct.png") |
| if w['id'] in gpu: |
| g = Image.open(os.path.join(root, 'data', gpu[w['id']]['hi'])).convert('L') |
| e['layers']['gpu'] = save(g.resize((W, H), Image.LANCZOS), f"{w['id']}_gpu.png") |
| out['segments'][w['id']] = e |
| json.dump(out, open(os.path.join(root, 'data/bench/index.json'), 'w'), indent=1) |
| print(len(out['segments']), 'segments') |
|
|
