Diff: neopixel.cpp

Revision:

4:c8515fbd2e44

diff -r 823f105078c7 -r c8515fbd2e44 neopixel.cpp
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/neopixel.cpp	Mon Nov 28 15:38:58 2016 +0000
@@ -0,0 +1,186 @@
+/* Lava
+ * 
+ * WS2812B Tricolor LED (neopixel) controller
+ *
+ *
+ * Example code:
+ 
+    neopixel_strip_t m_strip;
+    uint8_t dig_pin_num = 6;
+    uint8_t leds_per_strip = 24;
+    uint8_t error;
+    uint8_t led_to_enable = 10;
+    uint8_t red = 255;
+    uint8_t green = 0;
+    uint8_t blue = 159;
+
+    neopixel_init(&m_strip, dig_pin_num, leds_per_strip);
+    neopixel_clear(&m_strip);
+    error = neopixel_set_color_and_show(&m_strip, led_to_enable, red, green, blue);
+    if (error) {
+        //led_to_enable was not within number leds_per_strip
+    }
+    //clear and remove strip
+    neopixel_clear(&m_strip);
+    neopixel_destroy(&m_strip);
+ 
+ 
+ * For use with BLE stack, see information below:
+    - Include in main.c
+        #include "ble_radio_notification.h"
+    - Call (see nrf_soc.h: NRF_RADIO_NOTIFICATION_DISTANCES and NRF_APP_PRIORITIES)
+        ble_radio_notification_init(NRF_APP_PRIORITY_xxx,
+                                    NRF_RADIO_NOTIFICATION_DISTANCE_xxx,
+                                    your_radio_callback_handler);
+    - Create 
+        void your_radio_callback_handler(bool radio_active)
+        {
+            if (radio_active == false)
+            {
+                neopixel_show(&strip1);
+                neopixel_show(&strip2);
+                //...etc
+            }
+        }
+    - Do not use neopixel_set_color_and_show(...) with BLE, instead use uint8_t neopixel_set_color(...);
+ */
+
+#include "mbed.h" // remove line if not using mbed
+
+#include <stdbool.h>
+#include <stdint.h>
+#include <stdlib.h>
+#include "nrf_delay.h"
+#include "nrf_gpio.h"
+#include "neopixel.h"
+
+
+void neopixel_init(neopixel_strip_t *strip, uint8_t pin_num, uint16_t num_leds)
+{
+    strip->leds = (color_t*) malloc(sizeof(color_t) * num_leds);
+    strip->pin_num = pin_num;
+    strip->num_leds = num_leds;
+    nrf_gpio_cfg_output(pin_num);
+    NRF_GPIO->OUTCLR = (1UL << pin_num);
+    for (int i = 0; i < num_leds; i++)
+    {   
+        strip->leds[i].simple.g = 0;
+        strip->leds[i].simple.r = 0;
+        strip->leds[i].simple.b = 0;
+    }
+}
+
+void neopixel_clear(neopixel_strip_t *strip)
+{
+        for (int i = 0; i < strip->num_leds; i++)
+        {
+            strip->leds[i].simple.g = 0;
+            strip->leds[i].simple.r = 0;
+            strip->leds[i].simple.b = 0;
+        }
+            neopixel_show(strip);
+}
+
+void neopixel_show(neopixel_strip_t *strip)
+{
+    const uint8_t PIN =  strip->pin_num;
+    NRF_GPIO->OUTCLR = (1UL << PIN);
+    nrf_delay_us(50);
+            for (int i = 0; i < strip->num_leds; i++)
+            {
+                for (int j = 0; j < 3; j++)
+                {
+                    if ((strip->leds[i].grb[j] & 128) > 0)  {NEOPIXEL_SEND_ONE}
+                    else    {NEOPIXEL_SEND_ZERO}
+                    
+                    if ((strip->leds[i].grb[j] & 64) > 0)   {NEOPIXEL_SEND_ONE}
+                    else    {NEOPIXEL_SEND_ZERO}
+                    
+                    if ((strip->leds[i].grb[j] & 32) > 0)   {NEOPIXEL_SEND_ONE}
+                    else    {NEOPIXEL_SEND_ZERO}
+                    
+                    if ((strip->leds[i].grb[j] & 16) > 0)   {NEOPIXEL_SEND_ONE}
+                    else    {NEOPIXEL_SEND_ZERO}
+                    
+                    if ((strip->leds[i].grb[j] & 8) > 0)    {NEOPIXEL_SEND_ONE}
+                    else    {NEOPIXEL_SEND_ZERO}
+                    
+                    if ((strip->leds[i].grb[j] & 4) > 0)    {NEOPIXEL_SEND_ONE}
+                    else    {NEOPIXEL_SEND_ZERO}
+                    
+                    if ((strip->leds[i].grb[j] & 2) > 0)    {NEOPIXEL_SEND_ONE}
+                    else    {NEOPIXEL_SEND_ZERO}
+                    
+                    if ((strip->leds[i].grb[j] & 1) > 0)    {NEOPIXEL_SEND_ONE}
+                    else    {NEOPIXEL_SEND_ZERO}
+                }
+            }
+}
+
+uint8_t neopixel_set_color(neopixel_strip_t *strip, uint16_t index, uint8_t red, uint8_t green, uint8_t blue )
+{
+        if (index < strip->num_leds)
+        {
+            strip->leds[index].simple.r = red;
+            strip->leds[index].simple.g = green;
+            strip->leds[index].simple.b = blue;
+        }
+        else
+                return 1;
+        return 0;
+}
+
+uint8_t neopixel_set_color_and_show(neopixel_strip_t *strip, uint16_t index, uint8_t red, uint8_t green, uint8_t blue)
+{
+        if (index < strip->num_leds)
+        {
+            strip->leds[index].simple.r = red;
+            strip->leds[index].simple.g = green;
+            strip->leds[index].simple.b = blue;
+          neopixel_show(strip); 
+        }
+        else
+                return 1;
+        return 0;
+}
+
+void neopixel_destroy(neopixel_strip_t *strip)
+{
+    free(strip->leds);
+    strip->num_leds = 0;
+    strip->pin_num = 0;
+}
+/*
+int main() {
+    DigitalOut mypin(D4);
+    neopixel_strip_t m_strip;
+    uint8_t dig_pin_num = P0_21;
+    uint8_t leds_per_strip = 30;
+    uint8_t error;
+    uint8_t led_to_enable = 30;
+    uint8_t red = 0;
+    uint8_t green = 128;
+    uint8_t blue = 128;
+
+    neopixel_init(&m_strip, dig_pin_num, leds_per_strip);
+    neopixel_clear(&m_strip);
+    //error = neopixel_set_color_and_show(&m_strip, led_to_enable, red, green, blue);
+    for (int cv=0; cv < 60; cv++) {
+        
+        neopixel_set_color_and_show(&m_strip, cv, red, green, blue);
+        wait_ms(500);
+    }
+    
+    if (error) {
+        //led_to_enable was not within number leds_per_strip
+    }
+    
+    //error = neopixel_set_color_and_show(&m_strip, 1, red, green, blue);
+    while (1) {
+        wait_ms(500);
+    }
+    //clear and remove strip
+    //neopixel_clear(&m_strip);
+    //neopixel_destroy(&m_strip);
+}
+*/
\ No newline at end of file