from displayio import Bitmap
import bitmapfilter
import ulab
from dump_bitmap import dump_bitmap_rgb_swapped
from blinka_image import decode_resource


def test_pattern():
    return decode_resource("testpattern", 2)


def make_quadrant_bitmap():
    b = Bitmap(17, 17, 1)
    for i in range(b.height):
        for j in range(b.width):
            b[i, j] = (i < 8) ^ (j < 8)
    return b


q = make_quadrant_bitmap()
b = test_pattern()
dump_bitmap_rgb_swapped(b)

sepia_weights = bitmapfilter.ChannelMixer(
    0.393, 0.769, 0.189, 0.349, 0.686, 0.168, 0.272, 0.534, 0.131
)

print("sepia")
bitmapfilter.mix(b, sepia_weights)
dump_bitmap_rgb_swapped(b)

# Red channel only
print("red channel only (note: masked)")
b = test_pattern()
bitmapfilter.mix(b, bitmapfilter.ChannelScale(1, 0, 0), mask=q)
dump_bitmap_rgb_swapped(b)

# Scale green channel
print("scale green channel (note: masked)")
b = test_pattern()
bitmapfilter.mix(b, bitmapfilter.ChannelScale(1, 2, 0), mask=q)
dump_bitmap_rgb_swapped(b)

# Swap R & G channels
print("swap R&G")
b = test_pattern()
bitmapfilter.mix(b, bitmapfilter.ChannelMixerOffset(0, 1, 0, 0, 1, 0, 0, 0, 0, 0, 1, 0))
dump_bitmap_rgb_swapped(b)

# invert B
print("invert B")
b = test_pattern()
bitmapfilter.mix(b, bitmapfilter.ChannelScaleOffset(1, 0, 1, 0, -1, 1))
dump_bitmap_rgb_swapped(b)