m.giller/E-Paper-Calendar
1
0
Fork 0
Code Issues Pull requests Projects Releases 2 Packages Wiki Activity

Fine tuned the color interpretation on the Epd7in5b

Browse source
This commit is contained in:
Maximilian Giller 2019-03-01 18:53:58 +01:00
parent 83ac33043b
commit b9091c2dc7
1 changed files with 25 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
Calendar/Epd7in5bAdapter.py
View file

@ -1,6 +1,7 @@
from EpdAdapter import EpdAdapter, DISPLAY_REFRESH, DATA_START_TRANSMISSION_1
from settings import display_colours
from PIL import Image, ImageDraw
from math import sqrt, pow
class Epd7in5bAdapter (EpdAdapter):
def __init__ (self):
@ -50,33 +51,50 @@ class Epd7in5bAdapter (EpdAdapter):
# Set the bits for the column of pixels at the current
# position.
pixel = image_rgb.getpixel((x, y))
if self.__brightness__(pixel) < 75: #was 64 # black
color = self.__get_color__(pixel)
if color is 'white':
buf[int((x + y * self.height) / 4)] |= 0xC0 >> (x % 4 * 2)
elif color is 'black':
buf[int((x + y * self.height) / 4)] &= ~(0xC0 >> (x % 4 * 2))
elif pixel[0] > 150 and (pixel[2] + pixel[1]) < 50: #was 192
if color is 'red':
buf[int((x + y * self.height) / 4)] &= ~(0xC0 >> (x % 4 * 2))
buf[int((x + y * self.height) / 4)] |= 0x40 >> (x % 4 * 2)
else: # white
buf[int((x + y * self.height) / 4)] |= 0xC0 >> (x % 4 * 2)
return buf #due to python2 -> python3, int had to be added in 'get_frame
#_buffer
def __get_color__ (self, pixel):
color_percent = self.__get_color_percentage__(pixel)
brightness = self.__brightness__(pixel)
if brightness > 220 or (brightness > 150 and color_percent[0] > 35):
return 'white'
elif color_percent[0] > 50:
return 'red'
else:
return 'black'
def __get_color_percentage__ (self, pixel):
sum = pixel[0] + pixel[1] + pixel[2]
if sum is 0:
return (0,0,0)
return (pixel[0] / sum * 100, pixel[1] / sum * 100, pixel[2] / sum * 100)
def __brightness__ (self, pixel):
return (pixel[0] + pixel[1] + pixel[2]) / 3
def calibrate (self):
for _ in range(2):
self.init_render()
black = Image.new('1', (self.height, self.width), 'black')
black = Image.new('RGB', (self.height, self.width), 'black')
print('calibrating black...')
ImageDraw.Draw(black)
self.display_frame(self.get_frame_buffer(black))
red = Image.new('L', (self.height, self.width), 'red')
red = Image.new('RGB', (self.height, self.width), 'red')
ImageDraw.Draw(red)
print('calibrating red...')
self.display_frame(self.get_frame_buffer(red))
white = Image.new('1', (self.height, self.width), 'white')
white = Image.new('RGB', (self.height, self.width), 'white')
ImageDraw.Draw(white)
print('calibrating white...')
self.display_frame(self.get_frame_buffer(white))