1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
#!/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)
|
