better color support

This commit is contained in:
Nitzan Zada 2020-07-24 22:04:01 +03:00
commit 890bc1f848

View file

@ -3,6 +3,7 @@ import sys
import os
import time
import traceback
import numpy as np
from waveshare_epd import epd7in5b_HD
from PIL import Image, ImageDraw, ImageFont
@ -12,6 +13,23 @@ logging.basicConfig(level=logging.DEBUG)
picdir = os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))), 'pic')
libdir = os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))), 'lib')
image_full: Image = Image.open(os.path.join(picdir, 'fucking_retarded.bmp'))
buffer = np.array(image_full.convert('RGB'))
r, g, b = buffer[:, :, 0], buffer[:, :, 1], buffer[:, :, 2]
buffer_black = np.full(buffer.shape, 255, 'uint8')
buffer_black[np.logical_and(r == 0, g == 0, b == 0)] = [0, 0, 0]
buffer_red = np.full(buffer.shape, 255, 'uint8')
buffer_red[r == 255] = [0, 0, 0]
image_black = Image.fromarray(buffer_black)
image_red = Image.fromarray(buffer_red)
if os.path.exists(libdir):
sys.path.append(libdir)
@ -28,10 +46,7 @@ try:
logging.info("3.read bmp file...")
blackimage = Image.open(os.path.join(picdir, 'fucking_retarded.bmp'))
redimage = Image.open(os.path.join(picdir, 'fucking_retarded.bmp'))
epd.display(epd.getbuffer(blackimage), epd.getbuffer(redimage))
epd.display(epd.getbuffer(image_black), epd.getbuffer(image_red))
time.sleep(1)