Tidy a bit

This commit is contained in:
kris 2021-11-02 12:30:43 +00:00
parent 5675fac40d
commit d442baf1f1
2 changed files with 12 additions and 9 deletions

View File

@ -1,6 +1,5 @@
"""Precomputes all possible colours available via NTSC emulation."""
import colour
import numpy as np
from PIL import Image
import screen
@ -14,6 +13,7 @@ def main():
print("import numpy as np")
print()
print("# Indexed by (trailing 8-bit dot pattern, x % 4)")
print("SRGB = {")
# For each sequence of 8 pixels, compute the RGB colour of the right-most
# pixel, using NTSC emulation.
@ -30,12 +30,14 @@ def main():
bitmap[0, j:j + 8] = bits8
# bitmap_to_ntsc produces 3 output pixels for each DHGR input
ntsc = s.bitmap_to_ntsc(bitmap)
last_colour = ntsc[0, 3 * (j + 8) - 1, :]
colours[(bits, j - ntsc_shift)] = last_colour
unique.add(tuple(last_colour))
print(" (%d, %d): np.array((%d, %d, %d))," % (
bits, j-ntsc_shift, last_colour[0], last_colour[1], last_colour[2]))
bits, j - ntsc_shift, last_colour[0], last_colour[1],
last_colour[2]))
print("}")
print("# %d unique colours" % len(unique))

View File

@ -1,5 +1,6 @@
import numpy as np
# Indexed by (trailing 8-bit dot pattern, x % 4)
SRGB = {
(0, 0): np.array((0, 0, 0)),
(1, 0): np.array((0, 37, 0)),