#!/usr/bin/env python3
#
# Turn a PNG image into a Futhark value of type [height][width]i32 encoded in
# the binary data format.
# Usage: ./png2data image.png output.data
#
# Absolutely no error checking is done.
#
# png2data currently supports 24-bit RGB PNGs. Transparency, grayscale,
# multiple palettes and most other features of PNG is not supported.

import sys
import numpy as np
import png

if __name__ == "__main__":
    infile = sys.argv[1]
    outfile = sys.argv[2]

    r = png.Reader(infile)
    (width, height, img, _) = r.read()
    image_2d = np.vstack(list(map(np.uint32, img)))
    image_3d = np.reshape(image_2d, (height, width, 3))
    array = np.empty((height, width), dtype=np.int32)
    array = array | (image_3d[:, :, 0] << 16)
    array = array | (image_3d[:, :, 1] << 8)
    array = array | (image_3d[:, :, 2])

    with open(outfile, "wb") as f:
        f.write(b"b")
        f.write(np.int8(2))  # type: ignore
        f.write(np.int8(2))  # type: ignore
        f.write(b" i32")
        f.write(np.uint64(height))  # type: ignore
        f.write(np.uint64(width))  # type: ignore
        array.tofile(f)