initial commit, working code

diff --git a/requirements.txt b/requirements.txt
index 8bd5314643164b5ce2881e4e8c9da4741c30ba36..089cf9b8b18e3a40b6a972251154ccb2654d81c3 100755 (executable)
--- a/requirements.txt
+++ b/requirements.txt
@@ -1 +1,2 @@
 Adafruit-GPIO
 Adafruit-GPIO

+rpi_ws281x>=4.0.0

diff --git a/src/adafruit_blinka/microcontroller/raspi_23/neopixel.py b/src/adafruit_blinka/microcontroller/raspi_23/neopixel.py
new file mode 100644 (file)
index 0000000..5e99cf2
--- /dev/null
+++ b/src/adafruit_blinka/microcontroller/raspi_23/neopixel.py
@@ -0,0 +1,82 @@
+import time
+import math
+import _rpi_ws281x as ws
+import atexit
+
+# LED configuration.
+LED_CHANNEL    = 0
+LED_FREQ_HZ    = 800000     # Frequency of the LED signal.  We only support 800KHz
+LED_DMA_NUM    = 10         # DMA channel to use, can be 0-14.
+LED_BRIGHTNESS = 255        # We manage the brightness in the neopixel library
+LED_INVERT     = 0          # We don't support inverted logic
+LED_STRIP      = ws.WS2811_STRIP_RGB # We manage the color order within the neopixel library
+
+# a 'static' object that we will use to manage our PWM DMA channel
+# we only support one LED strip per raspi
+_led_strip = None
+
+def neopixel_write(gpio, buf):
+    global _led_strip # we'll have one strip we init if its not at first
+
+    if _led_strip is None:
+        # Create a ws2811_t structure from the LED configuration.
+        # Note that this structure will be created on the heap so you
+        # need to be careful that you delete its memory by calling
+        # delete_ws2811_t when it's not needed.
+        _led_strip = ws.new_ws2811_t()
+
+        # Initialize all channels to off
+        for channum in range(2):
+            channel = ws.ws2811_channel_get(_led_strip, channum)
+            ws.ws2811_channel_t_count_set(channel, 0)
+            ws.ws2811_channel_t_gpionum_set(channel, 0)
+            ws.ws2811_channel_t_invert_set(channel, 0)
+            ws.ws2811_channel_t_brightness_set(channel, 0)
+
+        channel = ws.ws2811_channel_get(_led_strip, LED_CHANNEL)
+
+        # Initialize the channel in use
+        ws.ws2811_channel_t_count_set(channel, math.ceil(len(buf)/3)) # we manage 4 vs 3 bytes in the library
+        ws.ws2811_channel_t_gpionum_set(channel, gpio._pin.id)
+        ws.ws2811_channel_t_invert_set(channel, LED_INVERT)
+        ws.ws2811_channel_t_brightness_set(channel, LED_BRIGHTNESS)
+        ws.ws2811_channel_t_strip_type_set(channel, LED_STRIP)
+
+        # Initialize the controller
+        ws.ws2811_t_freq_set(_led_strip, LED_FREQ_HZ)
+        ws.ws2811_t_dmanum_set(_led_strip, LED_DMA_NUM)
+    
+        resp = ws.ws2811_init(_led_strip)
+        if resp != ws.WS2811_SUCCESS:
+            message = ws.ws2811_get_return_t_str(resp)
+            raise RuntimeError('ws2811_init failed with code {0} ({1})'.format(resp, message))
+        atexit.register(neopixel_cleanup)
+
+    channel = ws.ws2811_channel_get(_led_strip, LED_CHANNEL)
+    if gpio._pin.id != ws.ws2811_channel_t_gpionum_get(channel):
+        raise RuntimeError("Raspberry Pi neopixel support is for one strip only!")
+
+    # assign all colors!
+    # bad way
+    pixels = zip(*[iter(buf)]*3)
+    for i, pixel in enumerate(pixels):
+        #print(pixel)
+        ws.ws2811_led_set(channel, i, pixel[0]<<16 | pixel[1] << 8 | pixel[2])
+    
+    resp = ws.ws2811_render(_led_strip)
+    if resp != ws.WS2811_SUCCESS:
+        message = ws.ws2811_get_return_t_str(resp)
+        raise RuntimeError('ws2811_render failed with code {0} ({1})'.format(resp, message))
+    time.sleep(0.001 * ((len(buf)//100)+1))  # about 1ms per 100 bytes
+
+
+def neopixel_cleanup():
+    global _led_strip
+
+    if _led_strip is not None:
+        # Ensure ws2811_fini is called before the program quits.
+        ws.ws2811_fini(_led_strip)
+        # Example of calling delete function to clean up structure memory.  Isn't
+        # strictly necessary at the end of the program execution here, but is good practice.
+        ws.delete_ws2811_t(_led_strip)
+        _led_strip = None

diff --git a/src/neopixel_write.py b/src/neopixel_write.py
new file mode 100644 (file)
index 0000000..d5cb7e0
--- /dev/null
+++ b/src/neopixel_write.py
@@ -0,0 +1,24 @@
+"""
+`neopixel_write` - NeoPixel precision timed writing support
+=================================================
+
+See `CircuitPython:neopixel_write` in CircuitPython for more details.
+
+* Author(s): ladyada
+"""
+
+import sys
+
+from adafruit_blinka.agnostic import board_id
+
+if board_id == "raspi_2" or board_id == "raspi_3":
+    from adafruit_blinka.microcontroller.raspi_23 import neopixel as _neopixel
+elif "sphinx" in sys.modules:
+    pass
+else:
+    raise NotImplementedError("Board not supported")
+
+
+def neopixel_write(gpio, buf):
+    """Write buf out on the given DigitalInOut."""
+    return _neopixel.neopixel_write(gpio, buf)