#!/usr/bin/env python3

import argparse
import struct
import numpy as np
from PIL import Image
from skimage import color

def run(args):
    total_bytes = 0
    with open(args.dst, 'wb') as f:
        for img_path in args.src:
            print(f'Processing {img_path}...')
            img = np.asarray(Image.open(img_path))
            if img.ndim == 2:
                img = np.tile(img[:,:,None], 3)
            img = img[:,:,:3]
            img = np.asarray(Image.fromarray(img).resize((256, 256), resample=3))
            img = color.rgb2lab(img)[:,:,0]
            img = img.flatten()
            data = struct.pack(f'{img.size}f', *img)
            f.write(data)
            total_bytes += len(data)
    print(f'{total_bytes} bytes written to {args.dst}')

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('src', nargs='+', help='Input images. (e.g., imgs/*)')
    parser.add_argument('dst', help='Output binary name. (e.g., input.bin)')
    args = parser.parse_args()
    run(args)

if __name__ == '__main__':
    main()