75 lines
1.6 KiB
Python
75 lines
1.6 KiB
Python
from pimoroni import ShiftRegister
|
|
from machine import Pin
|
|
from wakeup import get_shift_state, reset_shift_state
|
|
import time
|
|
|
|
|
|
SR_CLOCK = 8
|
|
SR_LATCH = 9
|
|
SR_OUT = 10
|
|
|
|
LED_A = 11
|
|
LED_B = 12
|
|
LED_C = 13
|
|
LED_D = 14
|
|
LED_E = 15
|
|
|
|
LED_BUSY = 6
|
|
LED_WIFI = 7
|
|
|
|
SHIFT_STATE = get_shift_state()
|
|
|
|
reset_shift_state()
|
|
|
|
|
|
class Button:
|
|
def __init__(self, sr, idx, led, debounce=50):
|
|
self.sr = sr
|
|
self.startup_state = bool(SHIFT_STATE & (1 << idx))
|
|
self.led = Pin(led, Pin.OUT) # LEDs are just regular IOs
|
|
self.led.off()
|
|
self._idx = idx
|
|
self._debounce_time = debounce
|
|
self._changed = time.ticks_ms()
|
|
self._last_value = None
|
|
|
|
def led_on(self):
|
|
self.led.on()
|
|
|
|
def led_off(self):
|
|
self.led.off()
|
|
|
|
def read(self):
|
|
if self.startup_state:
|
|
self.startup_state = False
|
|
return True
|
|
|
|
value = self.raw()
|
|
if value != self._last_value and time.ticks_ms() - self._changed > self._debounce_time:
|
|
self._last_value = value
|
|
self._changed = time.ticks_ms()
|
|
return value
|
|
return False
|
|
|
|
def raw(self):
|
|
if self.startup_state:
|
|
self.startup_state = False
|
|
return True
|
|
return self.sr[self._idx] == 1
|
|
|
|
@property
|
|
def is_pressed(self):
|
|
return self.raw()
|
|
|
|
|
|
sr = ShiftRegister(SR_CLOCK, SR_LATCH, SR_OUT)
|
|
|
|
button_a = Button(sr, 7, LED_A)
|
|
button_b = Button(sr, 6, LED_B)
|
|
button_c = Button(sr, 5, LED_C)
|
|
button_d = Button(sr, 4, LED_D)
|
|
button_e = Button(sr, 3, LED_E)
|
|
|
|
led_busy = Pin(LED_BUSY, Pin.OUT)
|
|
led_wifi = Pin(LED_WIFI, Pin.OUT)
|