mirror of
https://github.com/jayofelony/pwnagotchi.git
synced 2025-07-01 18:37:27 -04:00
521 lines
18 KiB
Python
521 lines
18 KiB
Python
# *****************************************************************************
|
|
# * | File : epd2in7_V2.py
|
|
# * | Author : Waveshare team
|
|
# * | Function : Electronic paper driver
|
|
# * | Info :
|
|
# *----------------
|
|
# * | This version: V1.0
|
|
# * | Date : 2022-09-17
|
|
# # | Info : python demo
|
|
# -----------------------------------------------------------------------------
|
|
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
# of this software and associated documnetation files (the "Software"), to deal
|
|
# in the Software without restriction, including without limitation the rights
|
|
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
# copies of the Software, and to permit persons to whom the Software is
|
|
# furished to do so, subject to the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included in
|
|
# all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
# FITNESS OR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
# LIABILITY WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
# THE SOFTWARE.
|
|
#
|
|
|
|
import logging
|
|
from .. import epdconfig
|
|
|
|
# Display resolution
|
|
EPD_WIDTH = 176
|
|
EPD_HEIGHT = 264
|
|
|
|
GRAY1 = 0xff # white
|
|
GRAY2 = 0xC0
|
|
GRAY3 = 0x80 # gray
|
|
GRAY4 = 0x00 # Blackest
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class EPD:
|
|
def __init__(self):
|
|
self.reset_pin = epdconfig.RST_PIN
|
|
self.dc_pin = epdconfig.DC_PIN
|
|
self.busy_pin = epdconfig.BUSY_PIN
|
|
self.cs_pin = epdconfig.CS_PIN
|
|
self.width = EPD_WIDTH
|
|
self.height = EPD_HEIGHT
|
|
self.GRAY1 = GRAY1 # white
|
|
self.GRAY2 = GRAY2
|
|
self.GRAY3 = GRAY3 # gray
|
|
self.GRAY4 = GRAY4 # Blackest
|
|
|
|
LUT_DATA_4Gray = [
|
|
0x40, 0x48, 0x80, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
|
|
0x8, 0x48, 0x10, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
|
|
0x2, 0x48, 0x4, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
|
|
0x20, 0x48, 0x1, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
|
|
0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
|
|
0xA, 0x19, 0x0, 0x3, 0x8, 0x0, 0x0,
|
|
0x14, 0x1, 0x0, 0x14, 0x1, 0x0, 0x3,
|
|
0xA, 0x3, 0x0, 0x8, 0x19, 0x0, 0x0,
|
|
0x1, 0x0, 0x0, 0x0, 0x0, 0x0, 0x1,
|
|
0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
|
|
0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
|
|
0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
|
|
0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
|
|
0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
|
|
0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
|
|
0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
|
|
0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
|
|
0x22, 0x22, 0x22, 0x22, 0x22, 0x22, 0x0, 0x0, 0x0,
|
|
0x22, 0x17, 0x41, 0x0, 0x32, 0x1C,
|
|
]
|
|
|
|
# Hardware reset
|
|
def reset(self):
|
|
epdconfig.digital_write(self.reset_pin, 1)
|
|
epdconfig.delay_ms(200)
|
|
epdconfig.digital_write(self.reset_pin, 0)
|
|
epdconfig.delay_ms(2)
|
|
epdconfig.digital_write(self.reset_pin, 1)
|
|
epdconfig.delay_ms(200)
|
|
|
|
def send_command(self, command):
|
|
epdconfig.digital_write(self.dc_pin, 0)
|
|
epdconfig.digital_write(self.cs_pin, 0)
|
|
epdconfig.spi_writebyte([command])
|
|
epdconfig.digital_write(self.cs_pin, 1)
|
|
|
|
def send_data(self, data):
|
|
epdconfig.digital_write(self.dc_pin, 1)
|
|
epdconfig.digital_write(self.cs_pin, 0)
|
|
epdconfig.spi_writebyte([data])
|
|
epdconfig.digital_write(self.cs_pin, 1)
|
|
|
|
def ReadBusy(self):
|
|
logger.debug("e-Paper busy")
|
|
while (epdconfig.digital_read(self.busy_pin) == 1): # 1: idle, 0: busy
|
|
epdconfig.delay_ms(20)
|
|
logger.debug("e-Paper busy release")
|
|
|
|
def TurnOnDisplay(self):
|
|
self.send_command(0x22) # Display Update Control
|
|
self.send_data(0xF7)
|
|
self.send_command(0x20) # Activate Display Update Sequence
|
|
self.ReadBusy()
|
|
|
|
def TurnOnDisplay_Fast(self):
|
|
self.send_command(0x22) # Display Update Control
|
|
self.send_data(0xC7)
|
|
self.send_command(0x20) # Activate Display Update Sequence
|
|
self.ReadBusy()
|
|
|
|
def TurnOnDisplay_Partial(self):
|
|
self.send_command(0x22) # Display Update Control
|
|
self.send_data(0xFF)
|
|
self.send_command(0x20) # Activate Display Update Sequence
|
|
self.ReadBusy()
|
|
|
|
def TurnOnDisplay_4GRAY(self):
|
|
self.send_command(0x22) # Display Update Control
|
|
self.send_data(0xC7)
|
|
self.send_command(0x20) # Activate Display Update Sequence
|
|
self.ReadBusy()
|
|
|
|
def Lut(self):
|
|
self.send_command(0x32)
|
|
for i in range(159):
|
|
self.send_data(self.LUT_DATA_4Gray[i])
|
|
|
|
def init(self):
|
|
if (epdconfig.module_init() != 0):
|
|
return -1
|
|
|
|
# EPD hardware init start
|
|
self.reset()
|
|
self.ReadBusy()
|
|
|
|
self.send_command(0x12) # SWRESET
|
|
self.ReadBusy()
|
|
|
|
self.send_command(0x45) # set Ram-Y address start/end position
|
|
self.send_data(0x00)
|
|
self.send_data(0x00)
|
|
self.send_data(0x07) # 0x0107-->(263+1)=264
|
|
self.send_data(0x01)
|
|
|
|
self.send_command(0x4F) # set RAM y address count to 0;
|
|
self.send_data(0x00)
|
|
self.send_data(0x00)
|
|
|
|
self.send_command(0x11) # data entry mode
|
|
self.send_data(0x03)
|
|
return 0
|
|
|
|
def init_Fast(self):
|
|
if (epdconfig.module_init() != 0):
|
|
return -1
|
|
|
|
# EPD hardware init start
|
|
self.reset()
|
|
self.ReadBusy()
|
|
|
|
self.send_command(0x12) # SWRESET
|
|
self.ReadBusy()
|
|
|
|
self.send_command(0x12) # SWRESET
|
|
self.ReadBusy()
|
|
|
|
self.send_command(0x18) # Read built-in temperature sensor
|
|
self.send_data(0x80)
|
|
|
|
self.send_command(0x22) # Load temperature value
|
|
self.send_data(0xB1)
|
|
self.send_command(0x20)
|
|
self.ReadBusy()
|
|
|
|
self.send_command(0x1A) # Write to temperature register
|
|
self.send_data(0x64)
|
|
self.send_data(0x00)
|
|
|
|
self.send_command(0x45) # set Ram-Y address start/end position
|
|
self.send_data(0x00)
|
|
self.send_data(0x00)
|
|
self.send_data(0x07) # 0x0107-->(263+1)=264
|
|
self.send_data(0x01)
|
|
|
|
self.send_command(0x4F) # set RAM y address count to 0;
|
|
self.send_data(0x00)
|
|
self.send_data(0x00)
|
|
|
|
self.send_command(0x11) # data entry mode
|
|
self.send_data(0x03)
|
|
|
|
self.send_command(0x22) # Load temperature value
|
|
self.send_data(0x91)
|
|
self.send_command(0x20)
|
|
self.ReadBusy()
|
|
return 0
|
|
|
|
def Init_4Gray(self):
|
|
if (epdconfig.module_init() != 0):
|
|
return -1
|
|
self.reset()
|
|
|
|
self.send_command(0x12) # soft reset
|
|
self.ReadBusy();
|
|
|
|
self.send_command(0x74) # set analog block control
|
|
self.send_data(0x54)
|
|
self.send_command(0x7E) # set digital block control
|
|
self.send_data(0x3B)
|
|
|
|
self.send_command(0x01) # Driver output control
|
|
self.send_data(0x07)
|
|
self.send_data(0x01)
|
|
self.send_data(0x00)
|
|
|
|
self.send_command(0x11) # data entry mode
|
|
self.send_data(0x03)
|
|
|
|
self.send_command(0x44) # set Ram-X address start/end position
|
|
self.send_data(0x00)
|
|
self.send_data(0x15) # 0x15-->(21+1)*8=176
|
|
|
|
self.send_command(0x45) # set Ram-Y address start/end position
|
|
self.send_data(0x00)
|
|
self.send_data(0x00)
|
|
self.send_data(0x07) # 0x0107-->(263+1)=264
|
|
self.send_data(0x01)
|
|
|
|
self.send_command(0x3C) # BorderWavefrom
|
|
self.send_data(0x00)
|
|
|
|
self.send_command(0x2C) # VCOM Voltage
|
|
self.send_data(self.LUT_DATA_4Gray[158]) # 0x1C
|
|
|
|
self.send_command(0x3F) # EOPQ
|
|
self.send_data(self.LUT_DATA_4Gray[153])
|
|
|
|
self.send_command(0x03) # VGH
|
|
self.send_data(self.LUT_DATA_4Gray[154])
|
|
|
|
self.send_command(0x04) #
|
|
self.send_data(self.LUT_DATA_4Gray[155]) # VSH1
|
|
self.send_data(self.LUT_DATA_4Gray[156]) # VSH2
|
|
self.send_data(self.LUT_DATA_4Gray[157]) # VSL
|
|
|
|
self.Lut() # LUT
|
|
|
|
self.send_command(0x4E) # set RAM x address count to 0;
|
|
self.send_data(0x00)
|
|
self.send_command(0x4F) # set RAM y address count to 0X199;
|
|
self.send_data(0x00)
|
|
self.send_data(0x00)
|
|
self.ReadBusy()
|
|
return 0
|
|
|
|
def getbuffer(self, image):
|
|
# logger.debug("bufsiz = ",int(self.width/8) * self.height)
|
|
buf = [0xFF] * (int(self.width / 8) * self.height)
|
|
image_monocolor = image.convert('1')
|
|
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):
|
|
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] == 0:
|
|
buf[int((x + y * self.width) / 8)] &= ~(0x80 >> (x % 8))
|
|
elif (imwidth == self.height and imheight == self.width):
|
|
logger.debug("Horizontal")
|
|
for y in range(imheight):
|
|
for x in range(imwidth):
|
|
newx = y
|
|
newy = self.height - x - 1
|
|
if pixels[x, y] == 0:
|
|
buf[int((newx + newy * self.width) / 8)] &= ~(0x80 >> (y % 8))
|
|
return buf
|
|
|
|
def getbuffer_4Gray(self, image):
|
|
# logger.debug("bufsiz = ",int(self.width/8) * self.height)
|
|
buf = [0xFF] * (int(self.width / 4) * self.height)
|
|
image_monocolor = image.convert('L')
|
|
imwidth, imheight = image_monocolor.size
|
|
pixels = image_monocolor.load()
|
|
i = 0
|
|
# logger.debug("imwidth = %d, imheight = %d",imwidth,imheight)
|
|
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):
|
|
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)
|
|
|
|
elif (imwidth == self.height and imheight == self.width):
|
|
logger.debug("Horizontal")
|
|
for x in range(imwidth):
|
|
for y in range(imheight):
|
|
newx = y
|
|
newy = self.height - x - 1
|
|
if (pixels[x, y] == 0xC0):
|
|
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)
|
|
return buf
|
|
|
|
def Clear(self):
|
|
if (self.width % 8 == 0):
|
|
Width = self.width // 8
|
|
else:
|
|
Width = self.width // 8 + 1
|
|
Height = self.height
|
|
self.send_command(0x24)
|
|
for j in range(Height):
|
|
for i in range(Width):
|
|
self.send_data(0XFF)
|
|
self.TurnOnDisplay()
|
|
|
|
def display(self, image):
|
|
if (self.width % 8 == 0):
|
|
Width = self.width // 8
|
|
else:
|
|
Width = self.width // 8 + 1
|
|
Height = self.height
|
|
self.send_command(0x24)
|
|
for j in range(Height):
|
|
for i in range(Width):
|
|
self.send_data(image[i + j * Width])
|
|
self.TurnOnDisplay()
|
|
|
|
def display_Fast(self, image):
|
|
if (self.width % 8 == 0):
|
|
Width = self.width // 8
|
|
else:
|
|
Width = self.width // 8 + 1
|
|
Height = self.height
|
|
self.send_command(0x24)
|
|
for j in range(Height):
|
|
for i in range(Width):
|
|
self.send_data(image[i + j * Width])
|
|
self.TurnOnDisplay_Fast()
|
|
|
|
def display_Base(self, image):
|
|
if (self.width % 8 == 0):
|
|
Width = self.width // 8
|
|
else:
|
|
Width = self.width // 8 + 1
|
|
Height = self.height
|
|
self.send_command(0x24) # Write Black and White image to RAM
|
|
for j in range(Height):
|
|
for i in range(Width):
|
|
self.send_data(image[i + j * Width])
|
|
|
|
self.send_command(0x26) # Write Black and White image to RAM
|
|
for j in range(Height):
|
|
for i in range(Width):
|
|
self.send_data(image[i + j * Width])
|
|
self.TurnOnDisplay()
|
|
|
|
def display_Base_color(self, color):
|
|
if (self.width % 8 == 0):
|
|
Width = self.width // 8
|
|
else:
|
|
Width = self.width // 8 + 1
|
|
Height = self.height
|
|
self.send_command(0x24) # Write Black and White image to RAM
|
|
for j in range(Height):
|
|
for i in range(Width):
|
|
self.send_data(color)
|
|
|
|
self.send_command(0x26) # Write Black and White image to RAM
|
|
for j in range(Height):
|
|
for i in range(Width):
|
|
self.send_data(color)
|
|
# self.TurnOnDisplay()
|
|
|
|
def display_Partial(self, Image, Xstart, Ystart, Xend, Yend):
|
|
if ((Xstart % 8 + Xend % 8 == 8 & Xstart % 8 > Xend % 8) | Xstart % 8 + Xend % 8 == 0 | (
|
|
Xend - Xstart) % 8 == 0):
|
|
Xstart = Xstart // 8
|
|
Xend = Xend // 8
|
|
else:
|
|
Xstart = Xstart // 8
|
|
if Xend % 8 == 0:
|
|
Xend = Xend // 8
|
|
else:
|
|
Xend = Xend // 8 + 1
|
|
|
|
if (self.width % 8 == 0):
|
|
Width = self.width // 8
|
|
else:
|
|
Width = self.width // 8 + 1
|
|
Height = self.height
|
|
|
|
Xend -= 1
|
|
Yend -= 1
|
|
|
|
# Reset
|
|
self.reset()
|
|
|
|
self.send_command(0x3C) # BorderWavefrom
|
|
self.send_data(0x80)
|
|
|
|
self.send_command(0x44) # set RAM x address start/end, in page 35
|
|
self.send_data(Xstart & 0xff) # RAM x address start at 00h;
|
|
self.send_data(Xend & 0xff) # RAM x address end at 0fh(15+1)*8->128
|
|
self.send_command(0x45) # set RAM y address start/end, in page 35
|
|
self.send_data(Ystart & 0xff) # RAM y address start at 0127h;
|
|
self.send_data((Ystart >> 8) & 0x01) # RAM y address start at 0127h;
|
|
self.send_data(Yend & 0xff) # RAM y address end at 00h;
|
|
self.send_data((Yend >> 8) & 0x01)
|
|
|
|
self.send_command(0x4E) # set RAM x address count to 0;
|
|
self.send_data(Xstart & 0xff)
|
|
self.send_command(0x4F) # set RAM y address count to 0X127;
|
|
self.send_data(Ystart & 0xff)
|
|
self.send_data((Ystart >> 8) & 0x01)
|
|
|
|
self.send_command(0x24) # Write Black and White image to RAM
|
|
for j in range(Height):
|
|
for i in range(Width):
|
|
if ((j > Ystart - 1) & (j < (Yend + 1)) & (i > Xstart - 1) & (i < (Xend + 1))):
|
|
self.send_data(Image[i + j * Width])
|
|
self.TurnOnDisplay_Partial()
|
|
|
|
def display_4Gray(self, image):
|
|
self.send_command(0x24)
|
|
for i in range(0, 48000): # 5808*4 46464
|
|
temp3 = 0
|
|
for j in range(0, 2):
|
|
temp1 = image[i * 2 + j]
|
|
for k in range(0, 2):
|
|
temp2 = temp1 & 0xC0
|
|
if (temp2 == 0xC0):
|
|
temp3 |= 0x00
|
|
elif (temp2 == 0x00):
|
|
temp3 |= 0x01
|
|
elif (temp2 == 0x80):
|
|
temp3 |= 0x01
|
|
else: # 0x40
|
|
temp3 |= 0x00
|
|
temp3 <<= 1
|
|
|
|
temp1 <<= 2
|
|
temp2 = temp1 & 0xC0
|
|
if (temp2 == 0xC0):
|
|
temp3 |= 0x00
|
|
elif (temp2 == 0x00):
|
|
temp3 |= 0x01
|
|
elif (temp2 == 0x80):
|
|
temp3 |= 0x01
|
|
else: # 0x40
|
|
temp3 |= 0x00
|
|
if (j != 1 or k != 1):
|
|
temp3 <<= 1
|
|
temp1 <<= 2
|
|
self.send_data(temp3)
|
|
|
|
self.send_command(0x26)
|
|
for i in range(0, 48000): # 5808*4 46464
|
|
temp3 = 0
|
|
for j in range(0, 2):
|
|
temp1 = image[i * 2 + j]
|
|
for k in range(0, 2):
|
|
temp2 = temp1 & 0xC0
|
|
if (temp2 == 0xC0):
|
|
temp3 |= 0x00
|
|
elif (temp2 == 0x00):
|
|
temp3 |= 0x01
|
|
elif (temp2 == 0x80):
|
|
temp3 |= 0x00
|
|
else: # 0x40
|
|
temp3 |= 0x01
|
|
temp3 <<= 1
|
|
|
|
temp1 <<= 2
|
|
temp2 = temp1 & 0xC0
|
|
if (temp2 == 0xC0):
|
|
temp3 |= 0x00
|
|
elif (temp2 == 0x00):
|
|
temp3 |= 0x01
|
|
elif (temp2 == 0x80):
|
|
temp3 |= 0x00
|
|
else: # 0x40
|
|
temp3 |= 0x01
|
|
if (j != 1 or k != 1):
|
|
temp3 <<= 1
|
|
temp1 <<= 2
|
|
self.send_data(temp3)
|
|
|
|
self.TurnOnDisplay_4GRAY()
|
|
|
|
def sleep(self):
|
|
self.send_command(0X10)
|
|
self.send_data(0x01)
|
|
|
|
epdconfig.delay_ms(2000)
|
|
epdconfig.module_exit()
|
|
|
|
### END OF FILE ### |