share for RS

Fork of rbVectorIQ by Mark @RRVA

neopixel.cpp

Committer:
MarkSPA
Date:
2016-11-28
Revision:
4:c8515fbd2e44

File content as of revision 4:c8515fbd2e44:

/* 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);
}
*/