added button inputs to rpyhal

raspberrypihal.py

@@ -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
+
+