better color support
This commit is contained in:
parent
e3a95c47d3
commit
890bc1f848
1 changed files with 19 additions and 4 deletions
|
|
@ -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)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue