116 lines
3.6 KiB
Python
116 lines
3.6 KiB
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)
|
|
# (see https://pinout.xyz) and that you've installed the rpi_ws281x library.
|
|
#
|
|
# For Python 2.x:
|
|
#
|
|
# sudo apt install -y python-pip; sudo pip install rpi_ws281x
|
|
#
|
|
# For Python 3.x:
|
|
#
|
|
# sudo apt install -y python3-pip; sudo pip3 install rpi_ws281x
|
|
#
|
|
#
|
|
from rpi_ws281x import PixelStrip, Color
|
|
from gpiozero import Button
|
|
import os
|
|
|
|
# LED strip configuration:
|
|
LED_PIN = 18 # GPIO pin connected to the pixels (18 uses PWM!).
|
|
# LED_PIN = 10 # GPIO pin connected to the pixels (10 uses SPI /dev/spidev0.0).
|
|
LED_FREQ_HZ = 800000 # LED signal frequency in hertz (usually 800khz)
|
|
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()
|
|
def init_display(self, num_pixels=64):
|
|
self.clear_display()
|
|
def clear_display(self):
|
|
c = Color(0, 0, 0)
|
|
for i in range(self.num_pixels):
|
|
self.strip.setPixelColor(i, c)
|
|
self.strip.show()
|
|
def update_display(self, num_modified_pixels):
|
|
if not num_modified_pixels:
|
|
return
|
|
self.strip.show()
|
|
def put_pixel(self, addr, r, g, b):
|
|
self.strip.setPixelColor(addr % self.num_pixels, Color(r, g, b))
|
|
def reset(self):
|
|
self.clear_display()
|
|
|
|
def set_rtc(self, t):
|
|
#Not relevant
|
|
pass
|
|
def set_auto_time(self, enable=True):
|
|
#Not relevant
|
|
pass
|
|
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
|
|
|
|
|