from pathlib import Path
from PIL import Image
import struct,json,hashlib
R=Path(__file__).resolve().parent; S=R/'extracted'; O=R/'web/x68000/buster/images/files';O.mkdir(parents=True,exist_ok=True)
def palette(p):
    out=[]
    for (v,) in struct.iter_unpack('>I',p.read_bytes()):
        i=v&1
        out.append(tuple((((v>>s)&31)*2+i)*255//63 for s in (6,11,1)))
    return out
def rawimg(src,pal,w,tiled=False):
    b=src.read_bytes();h=((len(b)+w*16-1)//(w*16)*16) if tiled else (len(b)+w-1)//w; im=Image.new('RGB',(w,h))
    for n,v in enumerate(b):
        if tiled:
            t=n//256;y=(t//(w//16))*16+n%256//16;x=(t%(w//16))*16+n%16
        else:x=n%w;y=n//w
        if y<h:im.putpixel((x,y),pal[v])
    return im
items=[]
for src in S.rglob('*'):
    if src.suffix not in ('.spi','.gri') and src.name!='title.dat':continue
    if src.suffix=='.spi':pp=src.with_suffix('.prt');tiled=True
    elif src.name=='bos.gri':pp=S/'data/mysp.prt';tiled=True
    elif src.name=='title.dat':pp=S/'data/title.prt';tiled=False
    else:pp=src.parent/'prt.prt';tiled=True
    rel=src.relative_to(S).as_posix();name=rel.replace('/','-')+'.png'; pal=palette(pp)
    if src.name=='bos.gri':
        # GAME.X _ending $1022-$108c passes RGB(2*i,int(1.8*i),int(1.5*i))
        # to the game's GRBI conversion routine _rgb $a9a0, for i=0..15.
        pal=[tuple(v*2*255//63 for v in (i*2,int(i*1.8),int(i*1.5))) for i in range(16)]+[(0,0,0)]*240
    im=rawimg(src,pal,256,tiled);im.save(O/name)
    kind='sprite' if src.suffix=='.spi' or src.name=='bos.gri' else 'title' if src.name=='title.dat' else 'background'
    banks=[]
    if src.suffix=='.spi':
        seen=set()
        for bank in range(15):
            colors=pal[bank*16:bank*16+16]
            if tuple(colors) in seen or not any(any(v) for v in colors):continue
            seen.add(tuple(colors));bn=rel.replace('/','-')+f'-palette-{bank+1}.png'
            rawimg(src,colors+[(0,0,0)]*240,256,True).save(O/bn)
            banks.append(dict(bank=bank+1,file='files/'+bn))
    items.append(dict(id=rel,file='files/'+name,width=im.width,height=im.height,palette='GAME.X _ending $1022-$108c, generated 16-color RGB ramp' if src.name=='bos.gri' else pp.relative_to(S).as_posix(),layout='16 × 16 tiles' if tiled else '256-pixel raw rows',kind=kind,banks=banks,sha256=hashlib.sha256(src.read_bytes()).hexdigest()))
for n in range(1,7):
    folder=S/f'r{n}'; pal=palette(folder/'prt.prt'); tiles=rawimg(folder/'chr.gri',pal,256,True)
    for mn,w,h in [('map.map',408,20),('maph.map',16,16)]:
        raw=(folder/mn).read_bytes();assert len(raw)==w*h
        im=Image.new('RGB',(w*16,h*16))
        for i,v in enumerate(raw):im.paste(tiles.crop(((v%16)*16,(v//16)*16,(v%16+1)*16,(v//16+1)*16)),((i%w)*16,(i//w)*16))
        name=f'r{n}-{mn}.png';im.save(O/name)
        thumb=im.copy();thumb.thumbnail((1024,256));thumb.save(O/(name+'.thumb.png'))
        items.append(dict(id=f'r{n}/{mn}',file='files/'+name,thumbnail='files/'+name+'.thumb.png',width=im.width,height=im.height,palette=f'r{n}/prt.prt',layout=f'{w} × {h} cells of 16 × 16 pixels; original tile order',kind='map',sha256=hashlib.sha256(raw).hexdigest()))
(R/'graphics-inventory.json').write_text(json.dumps(items,indent=2)+'\n')
print(f'Recovered {len(items)} image records with original palette variants.')