code.py

import alarm
import microcontroller
import time
import board
import busio
import displayio
from digitalio import DigitalInOut, Direction, Pull
from adafruit_display_text import label
from adafruit_bitmap_font import bitmap_font
from fourwire import FourWire
import adafruit_ssd1680

displayio.release_displays()

#vars
counterLock = False
needsRefresh = True
pressed = False
locking = False
resetting = False

#init time
pressStart = float(0)

#init button
btn = DigitalInOut(board.BUTTON)
btn.direction = Direction.INPUT
btn.pull = Pull.UP

#set display vars
spi = busio.SPI(board.EPD_SCK, MOSI=board.EPD_MOSI, MISO=None)
epdCs = board.EPD_CS
epdDc = board.EPD_DC
epdReset = board.EPD_RESET
epdBusy = board.EPD_BUSY

displayBus = FourWire(spi, command=epdDc, chip_select=epdCs, reset=epdReset, baudrate=1000000)

display = adafruit_ssd1680.SSD1680(
    displayBus,
    width=250,
    height=122,
    busy_pin=epdBusy,
    rotation=270,
    colstart = 8,
    seconds_per_frame=2.0
)

group = displayio.Group()

#init fonts
big = bitmap_font.load_font("fonts/AtkinsonHyperlegibleMono-Medium-35.bdf")
small = bitmap_font.load_font("fonts/AtkinsonHyperlegibleMono-MediumItalic-15.bdf")

#init background
backgroundBitmap = displayio.Bitmap(250,122,1)
backgroundPalette = displayio.Palette(1)
backgroundPalette[0] = 0xFFFFFF
background = displayio.TileGrid(backgroundBitmap, pixel_shader=backgroundPalette, x =0, y=0)
group.append(background)

#init text
dispText = "days since fire alarm"
botText = label.Label(font = small, text = dispText, scale = 1, color = 0x000000)
botText.anchor_point = (0,0)
botText.anchored_position = (10, 105)
group.append(botText)

def saveLock(lock):
    microcontroller.nvm[3] = lock

def loadLock():
    lock = microcontroller.nvm[3]
    return lock

def saveCount(count):
    microcontroller.nvm[0:3] = count.to_bytes(3, 'big')

def loadCount():
    val = int.from_bytes(microcontroller.nvm[0:3], 'big')
    if val > 99999:
        return 0
    return val

print(loadCount())
#init counter
count = loadCount()
countText = str(count)
counter = label.Label(font= big, text = countText, scale = 2, color = 0x000000)
counter.anchor_point = (0,0)
counter.anchored_position =(10,30)
group.append(counter)

#init lock
lockText = ""
lockDisp = label.Label(font = small, text = lockText, scale = 1, color = 0x000000)
lockDisp.anchor_point = (0,0)
lockDisp.anchored_position = (240, 105)
group.append(lockDisp)

#init sleeping
sleepText = ""
sleepDisp = label.Label(font = small, text = sleepText, scale = 1, color = 0x000000)
sleepDisp.anchor_point = (0,0)
sleepDisp.anchored_position = (0, 0)
group.append(sleepDisp)

#display group
display.root_group = group
display.refresh()

#init sleep feature
btn.deinit()
pin_alarm = alarm.pin.PinAlarm(pin=board.BUTTON, value=False, pull=True)

def sleep():
    if (time.monotonic() - pressStart) > 60:
        sleepDisp.text = "zzz...sleeping...huhmgh...zzz"
        time.sleep(display.time_to_refresh)
        while display.busy:
            time.sleep(0.01)
        display.refresh()
        print("sleeping")
        alarm.exit_and_deep_sleep_until_alarms(pin_alarm)

if alarm.wake_alarm:
    sleepDisp.text = ""
    pressStart = time.monotonic()
    counterLock = loadLock()
    needsRefresh = True

#init button again
btn = DigitalInOut(board.BUTTON)
btn.direction = Direction.INPUT
btn.pull = Pull.UP

def counterIncrement():
    global count
    global needsRefresh
    if not counterLock:
        count += 1
        counter.text = str(count)
        needsRefresh = True
        saveCount(count)
        print(count)

def counterDecrement():
    global count
    global needsRefresh
    if not counterLock:
        count -= 1
        counter.text = str(count)
        needsRefresh = True
        saveCount(count)
        print(count)

def reset():
    global resetting
    global count
    if (time.monotonic() - pressStart) > 10 and pressed and not resetting and not counterLock:
        resetting = True
        print("resetting")
        count = 0
        counter.text = str(count)
        saveCount(count)
        print("Reset!")
        refresh()

def refresh():
    global needsRefresh
    time.sleep(display.time_to_refresh)
    while display.busy:
        time.sleep(0.01)
    display.refresh()
    print("refreshed count")
    needsRefresh = False

def lockUnlock():
    global locking
    global counterLock
    counterLock = loadLock()
    if counterLock:
        lockDisp.text = "μ"
    if (time.monotonic() - pressStart) > 4.5 and pressed and not locking:
        locking = True
        if counterLock:
            print("unlocking")
            counterLock = False
            saveLock(counterLock)
            lockDisp.text = ""
            refresh()
        else:
            print("locking")
            counterDecrement()
            counterLock = True
            saveLock(counterLock)
            lockDisp.text = "μ"
            refresh()

while True:
    if not btn.value and not pressed:
        pressed = True
        counterIncrement()
        pressStart = time.monotonic()
    
    if btn.value:
        pressed = False
        locking = False
        resetting = False

    lockUnlock()

    reset()
    if ((time.monotonic() - pressStart) > 1.0 and needsRefresh and not pressed):
        refresh()

    sleep()

    time.sleep(0.1)