circuitpython/tests/circuitpython-manual/bitmaptools/_bmp16.py

54 lines
1.6 KiB
Python

import ulab.numpy as np
import displayio
import bitmaptools
try:
import struct
except:
import ustruct as struct
base_header = b"BMFX\x02\x00\x00\x00\x00\x00F\x00\x00\x008\x00\x00\x00@\x01\x00\x00\xf0\x00\x00\x00\x01\x00\x10\x00\x03\x00\x00\x00\x00X\x02\x00\xd7\r\x00\x00\xd7\r\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf8\x00\x00\xe0\x07\x00\x00\x1f\x00\x00\x00\x00\x00\x00\x00"
def writebmp16(filename, bitmap):
header = bytearray(base_header)
header[18:26] = struct.pack("<II", bitmap.width, bitmap.height)
with open(filename, "wb") as f:
f.write(header)
b = np.frombuffer(bitmap, dtype=np.uint16)
for i in range(bitmap.height):
j = (bitmap.height - i - 1) * bitmap.width
f.write(b[j : j + bitmap.width])
def loadbmp16(filename, width=320, height=240):
"""This specialized routine loads 16bpp uncompressed bmp files with a
70-byte header. It is not appropriate for generic bmp files."""
bitmap = displayio.Bitmap(width, height, 65536)
with open(filename, "rb") as f:
f.seek(70)
bitmaptools.readinto(
bitmap,
f,
bits_per_pixel=16,
element_size=2,
reverse_rows=True,
)
return bitmap
if __name__ == "__main__":
if "/" in __file__:
here = __file__.rsplit("/", 1)[0]
else:
here = "."
b = loadbmp16(here + "/minerva16.bmp")
print(b[0, 0])
print(b[160, 160])
for i, p in enumerate(sorted(set(memoryview(b)))):
print("%04x" % p, end="\n" if (i % 8) == 7 else " ")
if i % 8 != 7:
print()