projects/lamatrix
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added button inputs to rpyhal

Browse Source
This commit is contained in:
willem
2021-01-24 14:07:15 +01:00
parent 27c6c5c3f5
commit 11d9a58a55
1 changed files with 61 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
raspberrypihal.py
View File

@@ -1,4 +1,4 @@
# HAL for Raspberry Pi with https://github.com/rpi-ws281x/rpi-ws281x-python
q# HAL for Raspberry Pi with https://github.com/rpi-ws281x/rpi-ws281x-python
# See https://github.com/jgarff/rpi_ws281x for more details on this library.
#
# The below code assumes the LED strip is connected to GPIO 18 (PCM CLK)
@@ -14,6 +14,8 @@
#
#
from rpi_ws281x import PixelStrip, Color
from gpiozero import Button
import utime
# LED strip configuration:
LED_PIN = 18 # GPIO pin connected to the pixels (18 uses PWM!).
@@ -23,9 +25,22 @@ LED_DMA = 10 # DMA channel to use for generating signal (try 10)
LED_BRIGHTNESS = 255 # Set to 0 for darkest and 255 for brightest
LED_INVERT = False # True to invert the signal (when using NPN transistor level shift)
LED_CHANNEL = 0 # set to '1' for GPIOs 13, 19, 41, 45 or 53
BUTTON_LEFT = 17
BUTTON_RIGHT = 27
class RaspberryPiHAL:
def __init__(self, config):
self.left_button = Button(BUTTON_LEFT)
self.right_button = Button(BUTTON_RIGHT)
#set button callback
self.left_button.when_pressed = self.button_irq_press
self.right_button.when_pressed = self.button_irq_press
self.left_button.when_released = self.button_irq
self.right_button.when_released = self.button_irq
self.suspend_host_pin.when_held = self.button_irw_held
self.left_down_t = 0
self.right_down_t = 0
self.button_state = 0
self.num_pixels = config['LedMatrix']['columns'] * config['LedMatrix']['stride']
self.strip = PixelStrip(self.num_pixels, LED_PIN, LED_FREQ_HZ, LED_DMA, LED_INVERT, LED_BRIGHTNESS, LED_CHANNEL)
self.strip.begin()
@@ -44,9 +59,7 @@ class RaspberryPiHAL:
self.strip.setPixelColor(addr % self.num_pixels, Color(r, g, b))
def reset(self):
self.clear_display()
def process_input(self):
#TODO: implement
return 0
def set_rtc(self, t):
#Not relevant
pass
@@ -56,3 +69,47 @@ class RaspberryPiHAL:
def suspend_host(self, restart_timeout_seconds):
#Not relevant
pass
def button_irq_press(self):
if(button_left.is_pressed):
self.left_down_t = utime.ticks_ms()
if(button_right.is_pressed):
self.right_down_t = utime.ticks_ms()
def button_irq(self):
t = utime.ticks_ms() - self.button_down_t
shift = 0 if self.left_button.value else 4
if t > 1500:
self.button_state |= 1<<(shift+2)
elif t > 500:
self.button_state |= 1<<(shift+1)
elif t > 80:
self.button_state |= 1<<(shift+0)
self.button_down_t = 0
def process_input(self):
#TODO: implement
# Process button input
button_state = self.button_state
if button_state:
try:
if button_state & 1:
# Notify the host about the button press in a similar manner
# to what ArduinoSer2FastLED does
print('LEFTB_SHRT_PRESS\n')
elif button_state & 2:
print('LEFTB_LONG_PRESS\n')
elif button_state & 4:
print('LEFTB_HOLD_PRESS\n')
elif button_state & 16:
print('RGHTB_SHRT_PRESS\n')
elif button_state & 32:
print('RGHTB_LONG_PRESS\n')
elif button_state & 64:
print('RGHTB_HOLD_PRESS\n')
except OSError as e:
print('HAL: UART write failed: {}'.format(e.args[0]))
self.button_state = 0
return button_state