223 lines
7.1 KiB
Python
223 lines
7.1 KiB
Python
#!/usr/bin/python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
####################
|
|
# Inspiration / Further Reading
|
|
####################
|
|
# https://circuitpython.readthedocs.io/projects/bh1750/en/latest/
|
|
# https://raspi.tv/2013/how-to-use-interrupts-with-python-on-the-raspberry-pi-and-rpi-gpio-part-3
|
|
# https://stackoverflow.com/questions/16148735/how-to-implement-a-watchdog-timer-in-python
|
|
# https://learn.adafruit.com/adafruit-bh1750-ambient-light-sensor
|
|
# https://learn.adafruit.com/adafruit-mini-pitft-135x240-color-tft-add-on-for-raspberry-pi/overview
|
|
|
|
import board
|
|
import busio
|
|
import dbus
|
|
import digitalio
|
|
import os
|
|
import time
|
|
from threading import Timer
|
|
from PIL import Image, ImageDraw, ImageFont
|
|
import RPi.GPIO as GPIO
|
|
import adafruit_rgb_display.st7789 as st7789
|
|
import adafruit_bh1750
|
|
|
|
# Generic watchdog timer used for backlight controls
|
|
class Watchdog:
|
|
def __init__(self, timeout, userHandler=None): # timeout in seconds
|
|
self.timeout = timeout
|
|
self.handler = userHandler if userHandler is not None else self.defaultHandler
|
|
self.timer = Timer(self.timeout, self.handler)
|
|
# Do NOT start timer by default ; we want this triggered by reset events that come through
|
|
# via the inotify filesystem watcher
|
|
#self.timer.start()
|
|
|
|
def reset(self):
|
|
self.timer.cancel()
|
|
self.timer = Timer(self.timeout, self.handler)
|
|
self.timer.start()
|
|
|
|
def stop(self):
|
|
self.timer.cancel()
|
|
|
|
def defaultHandler(self):
|
|
raise self
|
|
|
|
# Cheat for walking only one level deep via os.walk (faster than other options)
|
|
def walklevel(some_dir, level=1):
|
|
some_dir = some_dir.rstrip(os.path.sep)
|
|
assert os.path.isdir(some_dir)
|
|
num_sep = some_dir.count(os.path.sep)
|
|
for root, dirs, files in os.walk(some_dir):
|
|
yield root, dirs, files
|
|
num_sep_this = root.count(os.path.sep)
|
|
if num_sep + level <= num_sep_this:
|
|
del dirs[:]
|
|
|
|
# bh1740 lux sensor
|
|
i2c = busio.I2C(board.SCL, board.SDA)
|
|
sensor = adafruit_bh1750.BH1750(i2c)
|
|
|
|
# Configuration for CS and DC pins (these are FeatherWing defaults on M0/M4):
|
|
cs_pin = digitalio.DigitalInOut(board.CE0)
|
|
dc_pin = digitalio.DigitalInOut(board.D25)
|
|
reset_pin = None
|
|
|
|
# Setup button pins as inputs
|
|
GPIO.setup(23, GPIO.IN)
|
|
GPIO.setup(24, GPIO.IN)
|
|
|
|
# Config for display baudrate (default max is 24mhz):
|
|
BAUDRATE = 64000000
|
|
|
|
# Setup SPI bus using hardware SPI:
|
|
spi = board.SPI()
|
|
|
|
# Create the ST7789 display:
|
|
disp = st7789.ST7789(
|
|
spi,
|
|
cs=cs_pin,
|
|
dc=dc_pin,
|
|
rst=reset_pin,
|
|
baudrate=BAUDRATE,
|
|
width=135,
|
|
height=240,
|
|
x_offset=53,
|
|
y_offset=40,
|
|
)
|
|
|
|
# Create blank image for drawing.
|
|
# Make sure to create image with mode 'RGB' for full color.
|
|
height = disp.width # we swap height/width to rotate it to landscape!
|
|
width = disp.height
|
|
image = Image.new('RGB', (width, height))
|
|
rotation = 90
|
|
|
|
# Get drawing object to draw on image.
|
|
draw = ImageDraw.Draw(image)
|
|
|
|
# Draw a black filled box to clear the image.
|
|
draw.rectangle((0, 0, width, height), outline=0, fill=(0, 0, 0))
|
|
disp.image(image, rotation)
|
|
# Draw some shapes.
|
|
# First define some constants to allow easy resizing of shapes.
|
|
padding = -2
|
|
top = padding
|
|
bottom = height - padding
|
|
# Move left to right keeping track of the current x position for drawing shapes.
|
|
x = 0
|
|
|
|
# Alternatively load a TTF font. Make sure the .ttf font file is in the
|
|
# same directory as the python script!
|
|
# Some other nice fonts to try: http://www.dafont.com/bitmap.php
|
|
#font = ImageFont.truetype('/usr/share/fonts/truetype/dejavu/DejaVuSansMono-Bold.ttf', 24)
|
|
font_small = font = ImageFont.truetype('/usr/share/fonts/truetype/dejavu/DejaVuSansMono-Bold.ttf', 24)
|
|
font = ImageFont.truetype('/usr/share/fonts/truetype/dejavu/DejaVuSansMono-Bold.ttf', 48)
|
|
|
|
# Turn on the backlight
|
|
backlight = digitalio.DigitalInOut(board.D22)
|
|
backlight.switch_to_output()
|
|
backlight.value = True
|
|
|
|
# Watchdog timer for screen on/off control based on button presses
|
|
# Turns off screen 10s after last button press
|
|
# Turns on screen for 10s when starting up
|
|
# Restart fim slideshow when the screen blanks so there is an indicator when it's going to change the selection
|
|
def backlight_callback():
|
|
global backlight
|
|
backlight.value = False
|
|
sysbus = dbus.SystemBus()
|
|
systemd1 = sysbus.get_object('org.freedesktop.systemd1', '/org/freedesktop/systemd1')
|
|
manager = dbus.Interface(systemd1, 'org.freedesktop.systemd1.Manager')
|
|
job = manager.RestartUnit('fim.service', 'fail')
|
|
|
|
watchdog = Watchdog(10.0, backlight_callback)
|
|
|
|
# Figure out which albums are present as well as the special 'All' album
|
|
albums = [ 'All' ]
|
|
selected_album = 0
|
|
try:
|
|
with open('/etc/default/fim_album_index', 'r') as f:
|
|
selected_album = int(f.readline())
|
|
except IOError:
|
|
pass
|
|
except ValueError:
|
|
pass
|
|
for root, dirs, files in walklevel('/tank/pictures'):
|
|
for folder in dirs:
|
|
if folder == '.stfolder':
|
|
continue
|
|
albums.append(folder)
|
|
albums.sort()
|
|
|
|
# Method for drawing on the screen
|
|
# Called by the button handlers since we don't need anything more than a basic while(1) main for this setup
|
|
def refresh_screen():
|
|
global width, height, x, y, top, draw, selected_album, albums, font, font_small
|
|
|
|
# Draw a black filled box to clear the image.
|
|
draw.rectangle((0, 0, width, height), outline=0, fill=0)
|
|
|
|
# Split the line on spaces and then add a blank line if needed to avoid crashes
|
|
text_lines = albums[selected_album].split(' ')
|
|
if len(text_lines) < 2:
|
|
text_lines.append('')
|
|
|
|
# Write some text
|
|
y = top
|
|
draw.text((x, y), text_lines[0], font=font, fill='#DCDCDC')
|
|
y += font.getsize('0')[1]
|
|
draw.text((x, y), text_lines[1], font=font, fill='#DCDCDC')
|
|
lux = 'Light: %.2f Lux' % sensor.lux
|
|
y = height - font_small.getsize(lux)[1]
|
|
draw.text((x, y), lux, font=font_small, fill='#DCDCDC')
|
|
|
|
# Display image.
|
|
disp.image(image, rotation)
|
|
|
|
# Write current album when restarting the slideshow
|
|
def write_selected_album():
|
|
with open('/etc/default/fim_album', 'w') as f:
|
|
f.write(str(selected_album))
|
|
with open('/etc/default/fim_album_path', 'w') as f:
|
|
if not selected_album == 0:
|
|
f.write('/tank/pictures/' + albums[selected_album])
|
|
else:
|
|
f.write('/tank/pictures')
|
|
|
|
# Interrupt callback for changing albums (up)
|
|
def up_button(channel):
|
|
global selected_album, albums, backlight
|
|
backlight.value = True
|
|
selected_album += 1
|
|
if selected_album >= len(albums):
|
|
selected_album = 0
|
|
refresh_screen()
|
|
write_selected_album()
|
|
watchdog.reset()
|
|
|
|
# Interrupt callback for changing albums (dn)
|
|
def dn_button(channel):
|
|
global selected_album, albums, backlight
|
|
backlight.value = True
|
|
selected_album -= 1
|
|
if selected_album < 0:
|
|
selected_album = len(albums) - 1
|
|
refresh_screen()
|
|
write_selected_album()
|
|
watchdog.reset()
|
|
|
|
# Setup button interrupts
|
|
GPIO.add_event_detect(23, GPIO.FALLING, callback=up_button, bouncetime=300)
|
|
GPIO.add_event_detect(24, GPIO.FALLING, callback=dn_button, bouncetime=300)
|
|
|
|
# Draw initial screen state
|
|
refresh_screen()
|
|
|
|
# Kick off watchdog to start the 10s initial 'on' status
|
|
watchdog.reset()
|
|
|
|
# Work with the LCD / Buttons in an infinate loop
|
|
while True:
|
|
time.sleep(300)
|