Update epd4in2.py

Fixed code formatting
Removed ";" from lines
Removed parens that shouldn't be there
This commit is contained in:
kmanc 2022-10-29 09:39:59 -07:00 committed by GitHub
commit b2cc986f8b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -4,8 +4,8 @@
# * | Function : Electronic paper driver
# * | Info :
# *----------------
# * | This version: V4.1
# * | Date : 2022-08-9
# * | This version: V4.2
# * | Date : 2022-10-29
# # | Info : python demo
# -----------------------------------------------------------------------------
# Permission is hereby granted, free of charge, to any person obtaining a copy
@ -44,6 +44,7 @@ GRAY4 = 0x00 #Blackest
logger = logging.getLogger(__name__)
class EPD:
def __init__(self):
self.reset_pin = epdconfig.RST_PIN
@ -251,7 +252,7 @@ class EPD:
def ReadBusy(self):
self.send_command(0x71)
while(epdconfig.digital_read(self.busy_pin) == 0): # 0: idle, 1: busy
while epdconfig.digital_read(self.busy_pin) == 0: # 0: idle, 1: busy
self.send_command(0x71)
epdconfig.delay_ms(100)
@ -271,7 +272,6 @@ class EPD:
self.send_command(0x24) # bb b
self.send_data2(self.lut_wb)
def Partial_SetLut(self):
self.send_command(0x20)
self.send_data2(self.EPD_4IN2_Partial_lut_vcom1)
@ -288,8 +288,6 @@ class EPD:
self.send_command(0x24)
self.send_data2(self.EPD_4IN2_Partial_lut_bb1)
def Gray_SetLut(self):
self.send_command(0x20) # vcom
self.send_data2(self.EPD_4IN2_4Gray_lut_vcom)
@ -309,9 +307,8 @@ class EPD:
self.send_command(0x25) # vcom
self.send_data2(self.EPD_4IN2_4Gray_lut_ww)
def init(self):
if (epdconfig.module_init() != 0):
if epdconfig.module_init() != 0:
return -1
# EPD hardware init start
self.reset()
@ -346,14 +343,15 @@ class EPD:
self.send_data(0x12)
self.send_command(0X50) # VCOM AND DATA INTERVAL SETTING
self.send_data(0x97) # 97white border 77black border VBDF 17|D7 VBDW 97 VBDB 57 VBDF F7 VBDW 77 VBDB 37 VBDR B7
self.send_data(
0x97) # 97white border 77black border VBDF 17|D7 VBDW 97 VBDB 57 VBDF F7 VBDW 77 VBDB 37 VBDR B7
self.set_lut()
# EPD hardware init end
return 0
def init_Partial(self):
if (epdconfig.module_init() != 0):
if epdconfig.module_init() != 0:
return -1
# EPD hardware init start
self.reset()
@ -388,14 +386,15 @@ class EPD:
self.send_data(0x12)
self.send_command(0X50) # VCOM AND DATA INTERVAL SETTING
self.send_data(0x07) # 97white border 77black border VBDF 17|D7 VBDW 97 VBDB 57 VBDF F7 VBDW 77 VBDB 37 VBDR B7
self.send_data(
0x07) # 97white border 77black border VBDF 17|D7 VBDW 97 VBDB 57 VBDF F7 VBDW 77 VBDB 37 VBDR B7
self.Partial_SetLut();
self.Partial_SetLut()
# EPD hardware init end
return 0
def Init_4Gray(self):
if (epdconfig.module_init() != 0):
if epdconfig.module_init() != 0:
return -1
# EPD hardware init start
self.reset()
@ -440,14 +439,14 @@ class EPD:
imwidth, imheight = image_monocolor.size
pixels = image_monocolor.load()
# logger.debug("imwidth = %d, imheight = %d",imwidth,imheight)
if(imwidth == self.width and imheight == self.height):
if imwidth == self.width and imheight == self.height:
logger.debug("Horizontal")
for y in range(imheight):
for x in range(imwidth):
# Set the bits for the column of pixels at the current position.
if pixels[x, y] == 0:
buf[int((x + y * self.width) / 8)] &= ~(0x80 >> (x % 8))
elif(imwidth == self.height and imheight == self.width):
elif imwidth == self.height and imheight == self.width:
logger.debug("Vertical")
for y in range(imheight):
for x in range(imwidth):
@ -465,32 +464,36 @@ class EPD:
pixels = image_monocolor.load()
i = 0
# logger.debug("imwidth = %d, imheight = %d",imwidth,imheight)
if(imwidth == self.width and imheight == self.height):
if imwidth == self.width and imheight == self.height:
logger.debug("Vertical")
for y in range(imheight):
for x in range(imwidth):
# Set the bits for the column of pixels at the current position.
if(pixels[x, y] == 0xC0):
if pixels[x, y] == 0xC0:
pixels[x, y] = 0x80
elif (pixels[x, y] == 0x80):
elif pixels[x, y] == 0x80:
pixels[x, y] = 0x40
i = i + 1
if(i%4 == 0):
buf[int((x + (y * self.width))/4)] = ((pixels[x-3, y]&0xc0) | (pixels[x-2, y]&0xc0)>>2 | (pixels[x-1, y]&0xc0)>>4 | (pixels[x, y]&0xc0)>>6)
if i % 4 == 0:
buf[int((x + (y * self.width)) / 4)] = (
(pixels[x - 3, y] & 0xc0) | (pixels[x - 2, y] & 0xc0) >> 2 | (
pixels[x - 1, y] & 0xc0) >> 4 | (pixels[x, y] & 0xc0) >> 6)
elif(imwidth == self.height and imheight == self.width):
elif imwidth == self.height and imheight == self.width:
logger.debug("Horizontal")
for x in range(imwidth):
for y in range(imheight):
newx = y
newy = x
if(pixels[x, y] == 0xC0):
if pixels[x, y] == 0xC0:
pixels[x, y] = 0x80
elif (pixels[x, y] == 0x80):
elif pixels[x, y] == 0x80:
pixels[x, y] = 0x40
i = i + 1
if(i%4 == 0):
buf[int((newx + (newy * self.width))/4)] = ((pixels[x, y-3]&0xc0) | (pixels[x, y-2]&0xc0)>>2 | (pixels[x, y-1]&0xc0)>>4 | (pixels[x, y]&0xc0)>>6)
if i % 4 == 0:
buf[int((newx + (newy * self.width)) / 4)] = (
(pixels[x, y - 3] & 0xc0) | (pixels[x, y - 2] & 0xc0) >> 2 | (
pixels[x, y - 1] & 0xc0) >> 4 | (pixels[x, y] & 0xc0) >> 6)
return buf
@ -500,7 +503,7 @@ class EPD:
else:
linewidth = int(self.width / 8) + 1
self.send_command(0x92);
self.send_command(0x92)
self.set_lut()
self.send_command(0x10)
self.send_data2([0xFF] * int(self.width * linewidth))
@ -515,24 +518,23 @@ class EPD:
# EPD_WIDTH = 400
# EPD_HEIGHT = 300
if(EPD_WIDTH % 8 != 0):
if EPD_WIDTH % 8 != 0:
Width = int(EPD_WIDTH / 8) + 1
else:
Width = int(EPD_WIDTH / 8)
Height = EPD_HEIGHT
if(X_start % 8 != 0):
if X_start % 8 != 0:
X_start = int(X_start / 8) + 1
else:
X_start = int(X_start / 8)
if(X_end % 8 != 0):
if X_end % 8 != 0:
X_end = int(X_end / 8) + 1
else:
X_end = int(X_end / 8)
buf = [0x00] * (Y_end - Y_start) * (X_end - X_start)
self.send_command(0x91) # This command makes the display enter partial mode
self.send_command(0x90) # resolution setting
self.send_data(int(X_start * 8 / 256))
@ -544,19 +546,17 @@ class EPD:
self.send_data(int(Y_start / 256))
self.send_data(int(Y_start % 256)) # y-start
self.send_data(int(Y_end / 256))
self.send_data(int(Y_end % 256) - 1) # y-end
self.send_data(0x28)
self.send_command(0x10); #writes Old data to SRAM for programming
self.send_command(0x10) # writes Old data to SRAM for programming
for j in range(0, Y_end - Y_start):
for i in range(0, X_end - X_start):
buf[j * (X_end - X_start) + i] = self.DATA[(Y_start + j) * Width + X_start + i]
self.send_data2(buf)
self.send_command(0x13); #writes New data to SRAM.
self.send_command(0x13) # writes New data to SRAM.
for j in range(0, Y_end - Y_start):
for i in range(0, X_end - X_start):
buf[j * (X_end - X_start) + i] = ~Image[(Y_start + j) * Width + X_start + i]
@ -567,9 +567,8 @@ class EPD:
epdconfig.delay_ms(200) # The delay here is necessary, 200uS at least!!!
self.ReadBusy()
def display_4Gray(self, image):
self.send_command(0x92);
self.send_command(0x92)
self.set_lut()
self.send_command(0x10)
@ -586,11 +585,11 @@ class EPD:
temp1 = image[i * 2 + j]
for k in range(0, 2):
temp2 = temp1 & 0xC0
if(temp2 == 0xC0):
if temp2 == 0xC0:
temp3 |= 0x01 # white
elif(temp2 == 0x00):
elif temp2 == 0x00:
temp3 |= 0x00 # black
elif(temp2 == 0x80):
elif temp2 == 0x80:
temp3 |= 0x01 # gray1
else: # 0x40
temp3 |= 0x00 # gray2
@ -598,15 +597,15 @@ class EPD:
temp1 <<= 2
temp2 = temp1 & 0xC0
if(temp2 == 0xC0): #white
if temp2 == 0xC0: # white
temp3 |= 0x01
elif(temp2 == 0x00): #black
elif temp2 == 0x00: # black
temp3 |= 0x00
elif(temp2 == 0x80):
elif temp2 == 0x80:
temp3 |= 0x01 # gray1
else: # 0x40
temp3 |= 0x00 # gray2
if(j!=1 or k!=1):
if j != 1 or k != 1:
temp3 <<= 1
temp1 <<= 2
buf[i] = temp3
@ -620,11 +619,11 @@ class EPD:
temp1 = image[i * 2 + j]
for k in range(0, 2):
temp2 = temp1 & 0xC0
if(temp2 == 0xC0):
if temp2 == 0xC0:
temp3 |= 0x01 # white
elif(temp2 == 0x00):
elif temp2 == 0x00:
temp3 |= 0x00 # black
elif(temp2 == 0x80):
elif temp2 == 0x80:
temp3 |= 0x00 # gray1
else: # 0x40
temp3 |= 0x01 # gray2
@ -632,15 +631,15 @@ class EPD:
temp1 <<= 2
temp2 = temp1 & 0xC0
if(temp2 == 0xC0): #white
if temp2 == 0xC0: # white
temp3 |= 0x01
elif(temp2 == 0x00): #black
elif temp2 == 0x00: # black
temp3 |= 0x00
elif(temp2 == 0x80):
elif temp2 == 0x80:
temp3 |= 0x00 # gray1
else: # 0x40
temp3 |= 0x01 # gray2
if(j!=1 or k!=1):
if j != 1 or k != 1:
temp3 <<= 1
temp1 <<= 2
buf[i] = temp3
@ -677,4 +676,3 @@ class EPD:
epdconfig.module_exit()
### END OF FILE ###