From 890bc1f848476df3be6ab0b661f928cb079c371f Mon Sep 17 00:00:00 2001 From: Nitzan Zada Date: Fri, 24 Jul 2020 22:04:01 +0300 Subject: [PATCH] better color support --- .../python/examples/fucking_retarded.py | 23 +++++++++++++++---- 1 file changed, 19 insertions(+), 4 deletions(-) diff --git a/RaspberryPi&JetsonNano/python/examples/fucking_retarded.py b/RaspberryPi&JetsonNano/python/examples/fucking_retarded.py index 705022a..bed1e8c 100644 --- a/RaspberryPi&JetsonNano/python/examples/fucking_retarded.py +++ b/RaspberryPi&JetsonNano/python/examples/fucking_retarded.py @@ -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)