Files
sheogorath/sheogorath.py
T
2024-01-12 20:56:34 +01:00

109 lines
3.5 KiB
Python

#!/usr/bin/python
# -*- coding:utf-8 -*-
import sys
import os
import logging
import time
import traceback
import argparse
from random import randint, choice
from PIL import Image,ImageDraw,ImageFont,ImageShow
parser = argparse.ArgumentParser(description='e-Ink Sheogorath')
parser.add_argument('--debug', action='store_true', help='If true, does not initialize e-Ink, shows images in window')
args = parser.parse_args()
logging.basicConfig(level=logging.DEBUG)
class EPDummy:
def __init__(self):
self.width = 122
self.height = 250
FULL_UPDATE = 0
PART_UPDATE = 1
def display(self, image):
ImageShow.show(image)
def displayPartial(self, image):
ImageShow.show(image)
def displayPartBaseImage(self, image):
ImageShow.show(image)
def init(self, a):
pass
def Clear(self, a):
pass
def getbuffer(self, image):
return image
epd = EPDummy()
if not args.debug:
from lib import epd2in13_V2
epd = epd2in13_V2.EPD()
try:
logging.info("Sheogorath.")
epd.init(epd.FULL_UPDATE)
epd.Clear(0xFF)
font15 = ImageFont.truetype('res/Font.ttc', 15)
font24 = ImageFont.truetype('res/Font.ttc', 24)
image = Image.new('1', (epd.height, epd.width), 255)
draw = ImageDraw.Draw(image)
draw.rectangle([(0,0),(50,50)],outline = 0)
draw.rectangle([(55,0),(100,50)],fill = 0)
draw.line([(0,0),(50,50)], fill = 0,width = 1)
draw.line([(0,50),(50,0)], fill = 0,width = 1)
draw.chord((10, 60, 50, 100), 0, 360, fill = 0)
draw.ellipse((55, 60, 95, 100), outline = 0)
draw.pieslice((55, 60, 95, 100), 90, 180, outline = 0)
draw.pieslice((55, 60, 95, 100), 270, 360, fill = 0)
draw.polygon([(110,0),(110,50),(150,25)],outline = 0)
draw.polygon([(190,0),(190,50),(150,25)],fill = 0)
draw.text((120, 60), 'e-Paper demo', font = font15, fill = 0)
draw.text((110, 90), u'', font = font24, fill = 0)
# read bmp file on window
logging.info("3.read bmp file on window...")
epd.Clear(0xFF)
#image1 = Image.new('1', (epd.height, epd.width), 255) # 255: clear the frame
bmp = Image.open('res/sheo1.bmp')
#epd.init(epd.FULL_UPDATE)
#epd.displayPartBaseImage(epd.getbuffer(time_image))
epd.init(epd.PART_UPDATE)
lyrics = open("lyrics.txt", "r").readlines()
def get_lyric():
LENGTH = 17
lyric = choice(lyrics)
out_lyric = ""
counter = 0
for word in lyric.split(" "):
if counter + len(word) < LENGTH:
counter += len(word) + 1
out_lyric += word + " "
else:
counter = len(word) + 1
out_lyric += "\n" + word + " "
return out_lyric
while (True):
lyric = get_lyric()
print(lyric)
time_image = Image.new('1', (epd.height, epd.width), 255)
time_draw = ImageDraw.Draw(time_image)
x = randint(0, 40)
time_image.paste(bmp, (x,randint(0, 10)))
time_draw.rectangle((160, 90, 250, 122), fill = 255)
time_draw.text((180, 90), time.strftime('%H:%M'), font = font24, fill = 0)
time_draw.text((100+x, 5), lyric, font = font15, fill = 0)
epd.displayPartial(epd.getbuffer(time_image))
epd.displayPartial(epd.getbuffer(time_image))
epd.displayPartial(epd.getbuffer(time_image))
time.sleep(10)
except IOError as e:
logging.info(e)
except KeyboardInterrupt:
logging.info("ctrl + c:")
epd2in13_V2.epdconfig.module_exit(cleanup=True)
exit()