better color support

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'))
+    epd.display(epd.getbuffer(image_black), epd.getbuffer(image_red))
-    redimage = Image.open(os.path.join(picdir, 'fucking_retarded.bmp'))
-
-    epd.display(epd.getbuffer(blackimage), epd.getbuffer(redimage))
 
     time.sleep(1)