Seeed / color_pixels

Color pixels library using WS2812B and nRF51822

Dependents:   BLE_Color_Pixels Grove_Node BLE_HeartRate

Color pixels library using WS2812B and nRF51822 (16Hz)

http://www.seeedstudio.com/depot/bmz_cache/4/4f346dc15724a7b5a5c1383253aeefc9.image.530x397.jpg

It's for

color_pixels.cpp@4:16ef874fa57f, 2015-05-05 (annotated)

Committer:
yihui
Date:
Tue May 05 05:37:20 2015 +0000
Revision:
4:16ef874fa57f
Parent:
3:91af7b451005 
update header file name

Who changed what in which revision?

UserRevisionLine numberNew contents of line
Yihui Xiong 2:5c805323a2c0 1 /** Color pixels library using WS2812B and nRF51822 (16MHz)
yihui 0:1e1ce549ad91 2 * It's for
yihui 0:1e1ce549ad91 3 * + http://www.seeedstudio.com/depot/Digital-RGB-LED-FlexiStrip-60-LED-1-Meter-p-1666.html
yihui 0:1e1ce549ad91 4 * + http://www.seeedstudio.com/depot/WS2812B-Digital-RGB-LED-Waterproof-FlexiStrip-144-LEDmeter-2-meter-p-1869.html
yihui 0:1e1ce549ad91 5 * + http://www.seeedstudio.com/depot/WS2812B-RGB-LED-with-Integrated-Driver-Chip-10-PCs-pack-p-1675.html
yihui 0:1e1ce549ad91 6 *
yihui 0:1e1ce549ad91 7 * The MIT License (MIT)
yihui 0:1e1ce549ad91 8 *
yihui 0:1e1ce549ad91 9 * Copyright (c) 2014 Seeed Technology Inc.
yihui 0:1e1ce549ad91 10 *
yihui 0:1e1ce549ad91 11 * Permission is hereby granted, free of charge, to any person obtaining a copy
yihui 0:1e1ce549ad91 12 * of this software and associated documentation files (the "Software"), to deal
yihui 0:1e1ce549ad91 13 * in the Software without restriction, including without limitation the rights
yihui 0:1e1ce549ad91 14 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
yihui 0:1e1ce549ad91 15 * copies of the Software, and to permit persons to whom the Software is
yihui 0:1e1ce549ad91 16 * furnished to do so, subject to the following conditions:
yihui 0:1e1ce549ad91 17
yihui 0:1e1ce549ad91 18 * The above copyright notice and this permission notice shall be included in
yihui 0:1e1ce549ad91 19 * all copies or substantial portions of the Software.
yihui 0:1e1ce549ad91 20
yihui 0:1e1ce549ad91 21 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
yihui 0:1e1ce549ad91 22 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
yihui 0:1e1ce549ad91 23 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
yihui 0:1e1ce549ad91 24 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
yihui 0:1e1ce549ad91 25 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
yihui 0:1e1ce549ad91 26 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
yihui 0:1e1ce549ad91 27 * THE SOFTWARE.
yihui 0:1e1ce549ad91 28 */
yihui 0:1e1ce549ad91 29
yihui 0:1e1ce549ad91 30 #include "color_pixels.h"
yihui 4:16ef874fa57f 31 #include "nrf51.h"
yihui 4:16ef874fa57f 32 #include "nrf51_bitfields.h"
yihui 0:1e1ce549ad91 33
Yihui Xiong 2:5c805323a2c0 34 #if !defined ( __GNUC__ )
Yihui Xiong 2:5c805323a2c0 35 // Generate a high level pulse (0.81us) of WS2812B's 1 code (0.9us +- 0.15us)
yihui 0:1e1ce549ad91 36 #define COLOR_PIXELS_ONE_HIGH(mask) \
yihui 0:1e1ce549ad91 37 NRF_GPIO->OUTSET = (mask); \
yihui 0:1e1ce549ad91 38 __ASM ( \
yihui 0:1e1ce549ad91 39 " NOP\n\t" \
yihui 0:1e1ce549ad91 40 " NOP\n\t" \
yihui 0:1e1ce549ad91 41 " NOP\n\t" \
yihui 0:1e1ce549ad91 42 " NOP\n\t" \
yihui 0:1e1ce549ad91 43 " NOP\n\t" \
yihui 0:1e1ce549ad91 44 " NOP\n\t" \
yihui 0:1e1ce549ad91 45 " NOP\n\t" \
yihui 0:1e1ce549ad91 46 " NOP\n\t" \
yihui 0:1e1ce549ad91 47 " NOP\n\t" \
yihui 0:1e1ce549ad91 48 ); \
yihui 0:1e1ce549ad91 49 NRF_GPIO->OUTCLR = (mask)
yihui 0:1e1ce549ad91 50
yihui 0:1e1ce549ad91 51 // Generate a high level pulse (0.31us) of WS2812B's 0 code (0.35us +- 0.15us)
yihui 0:1e1ce549ad91 52 #define COLOR_PIXELS_ZERO_HIGH(mask) \
yihui 0:1e1ce549ad91 53 NRF_GPIO->OUTSET = (mask); \
yihui 0:1e1ce549ad91 54 __ASM ( \
yihui 0:1e1ce549ad91 55 " NOP\n\t" \
yihui 0:1e1ce549ad91 56 ); \
Yihui Xiong 2:5c805323a2c0 57 NRF_GPIO->OUTCLR = (mask); \
Yihui Xiong 2:5c805323a2c0 58 __ASM ( \
Yihui Xiong 2:5c805323a2c0 59 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 60 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 61 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 62 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 63 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 64 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 65 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 66 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 67 )
Yihui Xiong 2:5c805323a2c0 68 #else
Yihui Xiong 2:5c805323a2c0 69 // Generate a high level pulse (0.94us) of WS2812B's 1 code (0.9us +- 0.15us)
Yihui Xiong 2:5c805323a2c0 70 #define COLOR_PIXELS_ONE_HIGH(mask) \
Yihui Xiong 2:5c805323a2c0 71 NRF_GPIO->OUTSET = (mask); \
Yihui Xiong 2:5c805323a2c0 72 __ASM ( \
Yihui Xiong 2:5c805323a2c0 73 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 74 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 75 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 76 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 77 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 78 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 79 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 80 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 81 ); \
yihui 0:1e1ce549ad91 82 NRF_GPIO->OUTCLR = (mask)
yihui 0:1e1ce549ad91 83
Yihui Xiong 2:5c805323a2c0 84 // Generate a high level pulse (0.44us) of WS2812B's 0 code (0.35us +- 0.15us)
Yihui Xiong 2:5c805323a2c0 85 #define COLOR_PIXELS_ZERO_HIGH(mask) \
Yihui Xiong 2:5c805323a2c0 86 NRF_GPIO->OUTSET = (mask); \
Yihui Xiong 2:5c805323a2c0 87 __ASM ( \
Yihui Xiong 2:5c805323a2c0 88 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 89 ); \
Yihui Xiong 2:5c805323a2c0 90 NRF_GPIO->OUTCLR = (mask); \
Yihui Xiong 2:5c805323a2c0 91 __ASM ( \
Yihui Xiong 2:5c805323a2c0 92 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 93 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 94 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 95 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 96 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 97 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 98 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 99 " NOP\n\t" \
Yihui Xiong 2:5c805323a2c0 100 )
Yihui Xiong 2:5c805323a2c0 101 #endif
Yihui Xiong 2:5c805323a2c0 102
yihui 0:1e1ce549ad91 103 static void delay_us(uint32_t us)
yihui 0:1e1ce549ad91 104 {
yihui 0:1e1ce549ad91 105 do {
yihui 0:1e1ce549ad91 106 __ASM volatile (
yihui 0:1e1ce549ad91 107 "NOP\n\t"
yihui 0:1e1ce549ad91 108 "NOP\n\t"
yihui 0:1e1ce549ad91 109 "NOP\n\t"
yihui 0:1e1ce549ad91 110 "NOP\n\t"
yihui 0:1e1ce549ad91 111 "NOP\n\t"
yihui 0:1e1ce549ad91 112 "NOP\n\t"
yihui 0:1e1ce549ad91 113 "NOP\n\t"
yihui 0:1e1ce549ad91 114 "NOP\n\t"
yihui 0:1e1ce549ad91 115 "NOP\n\t"
yihui 0:1e1ce549ad91 116 "NOP\n\t"
yihui 0:1e1ce549ad91 117 "NOP\n\t"
yihui 0:1e1ce549ad91 118 "NOP\n\t"
yihui 0:1e1ce549ad91 119 );
yihui 0:1e1ce549ad91 120 } while (--us);
yihui 0:1e1ce549ad91 121 }
yihui 0:1e1ce549ad91 122
yihui 0:1e1ce549ad91 123 ColorPixels::ColorPixels(uint8_t pin, uint16_t num)
yihui 0:1e1ce549ad91 124 {
yihui 0:1e1ce549ad91 125 this->pin = pin;
yihui 0:1e1ce549ad91 126 this->num = num;
yihui 0:1e1ce549ad91 127
yihui 0:1e1ce549ad91 128 NRF_GPIO->PIN_CNF[pin] = (GPIO_PIN_CNF_SENSE_Disabled << GPIO_PIN_CNF_SENSE_Pos)
yihui 0:1e1ce549ad91 129 | (GPIO_PIN_CNF_DRIVE_S0S1 << GPIO_PIN_CNF_DRIVE_Pos)
yihui 0:1e1ce549ad91 130 | (GPIO_PIN_CNF_PULL_Disabled << GPIO_PIN_CNF_PULL_Pos)
yihui 0:1e1ce549ad91 131 | (GPIO_PIN_CNF_INPUT_Connect << GPIO_PIN_CNF_INPUT_Pos)
yihui 0:1e1ce549ad91 132 | (GPIO_PIN_CNF_DIR_Output << GPIO_PIN_CNF_DIR_Pos);
yihui 0:1e1ce549ad91 133 NRF_GPIO->OUTCLR = (1UL << pin);
yihui 0:1e1ce549ad91 134
Yihui Xiong 2:5c805323a2c0 135 colors = new grb_t[num];
yihui 0:1e1ce549ad91 136 for (int i = 0; i < num; i++) {
yihui 0:1e1ce549ad91 137 colors[i].grb = 0;
yihui 0:1e1ce549ad91 138 }
yihui 0:1e1ce549ad91 139 }
yihui 0:1e1ce549ad91 140
yihui 0:1e1ce549ad91 141 ColorPixels::~ColorPixels()
yihui 0:1e1ce549ad91 142 {
yihui 0:1e1ce549ad91 143 delete colors;
yihui 0:1e1ce549ad91 144 }
yihui 0:1e1ce549ad91 145
yihui 0:1e1ce549ad91 146
yihui 0:1e1ce549ad91 147 void ColorPixels::set_color(uint16_t index, uint8_t r, uint8_t g, uint8_t b)
yihui 0:1e1ce549ad91 148 {
yihui 0:1e1ce549ad91 149 if (index < num) {
yihui 0:1e1ce549ad91 150 colors[index].r = r;
yihui 0:1e1ce549ad91 151 colors[index].g = g;
yihui 0:1e1ce549ad91 152 colors[index].b = b;
yihui 0:1e1ce549ad91 153 }
yihui 0:1e1ce549ad91 154 }
yihui 0:1e1ce549ad91 155
Yihui Xiong 2:5c805323a2c0 156 void ColorPixels::set_color(uint16_t index, uint32_t rgb) {
Yihui Xiong 2:5c805323a2c0 157 color_t c = *(color_t *)&rgb;
Yihui Xiong 2:5c805323a2c0 158 set_color(index, c.r, c.g, c.b);
Yihui Xiong 2:5c805323a2c0 159 }
Yihui Xiong 2:5c805323a2c0 160
yihui 3:91af7b451005 161 void ColorPixels::set_all_color(uint8_t r, uint8_t g, uint8_t b)
yihui 3:91af7b451005 162 {
yihui 3:91af7b451005 163 for (int i = 0; i < num; i++) {
yihui 3:91af7b451005 164 colors[i].r = r;
yihui 3:91af7b451005 165 colors[i].g = g;
yihui 3:91af7b451005 166 colors[i].b = b;
yihui 3:91af7b451005 167 }
yihui 3:91af7b451005 168
yihui 3:91af7b451005 169 update();
yihui 3:91af7b451005 170 }
yihui 3:91af7b451005 171
yihui 0:1e1ce549ad91 172 void ColorPixels::update()
yihui 0:1e1ce549ad91 173 {
yihui 0:1e1ce549ad91 174 uint32_t mask = 1 << pin;
yihui 0:1e1ce549ad91 175 NRF_GPIO->OUTCLR = mask;
yihui 0:1e1ce549ad91 176 delay_us(50);
Yihui Xiong 2:5c805323a2c0 177
Yihui Xiong 2:5c805323a2c0 178 grb_t *pcolor = colors;
yihui 0:1e1ce549ad91 179 for (int i = 0; i < num; i++) {
Yihui Xiong 2:5c805323a2c0 180 uint32_t grb = (*pcolor).grb;
yihui 0:1e1ce549ad91 181 for (int bit = 0; bit < 24; bit++) {
Yihui Xiong 2:5c805323a2c0 182 if (grb & 1) {
yihui 0:1e1ce549ad91 183 COLOR_PIXELS_ONE_HIGH(mask);
yihui 0:1e1ce549ad91 184 } else {
yihui 0:1e1ce549ad91 185 COLOR_PIXELS_ZERO_HIGH(mask);
yihui 0:1e1ce549ad91 186 }
Yihui Xiong 2:5c805323a2c0 187 grb >>= 1;
yihui 0:1e1ce549ad91 188 }
Yihui Xiong 2:5c805323a2c0 189 pcolor++;
yihui 0:1e1ce549ad91 190 }
yihui 0:1e1ce549ad91 191 }
yihui 0:1e1ce549ad91 192
yihui 0:1e1ce549ad91 193 void ColorPixels::clear()
yihui 0:1e1ce549ad91 194 {
yihui 0:1e1ce549ad91 195 for (int i = 0; i < num; i++) {
yihui 0:1e1ce549ad91 196 colors[i].grb = 0;
yihui 0:1e1ce549ad91 197 }
yihui 0:1e1ce549ad91 198
yihui 0:1e1ce549ad91 199 update();
yihui 0:1e1ce549ad91 200 }
Yihui Xiong 2:5c805323a2c0 201
Yihui Xiong 2:5c805323a2c0 202 uint32_t toRGB(uint8_t red, uint8_t green, uint8_t blue)
Yihui Xiong 2:5c805323a2c0 203 {
Yihui Xiong 2:5c805323a2c0 204 color_t c;
Yihui Xiong 2:5c805323a2c0 205
Yihui Xiong 2:5c805323a2c0 206 c.r = red;
Yihui Xiong 2:5c805323a2c0 207 c.g = green;
Yihui Xiong 2:5c805323a2c0 208 c.b = blue;
Yihui Xiong 2:5c805323a2c0 209
Yihui Xiong 2:5c805323a2c0 210 return c.rgb;
Yihui Xiong 2:5c805323a2c0 211 };
Yihui Xiong 2:5c805323a2c0 212
Yihui Xiong 2:5c805323a2c0 213 uint32_t wheel(float position, uint8_t min = 0, uint8_t max = 255)
Yihui Xiong 2:5c805323a2c0 214 {
Yihui Xiong 2:5c805323a2c0 215 uint8_t d = max - min;
Yihui Xiong 2:5c805323a2c0 216 uint32_t c;
Yihui Xiong 2:5c805323a2c0 217
Yihui Xiong 2:5c805323a2c0 218 if(position < 0.166) {
Yihui Xiong 2:5c805323a2c0 219 c = toRGB(max, min + position * 6 * d, min);
Yihui Xiong 2:5c805323a2c0 220 } else if(position < 0.332) {
Yihui Xiong 2:5c805323a2c0 221 position -= 0.166;
Yihui Xiong 2:5c805323a2c0 222 c = toRGB(max - position * 6 * d, max, min);
Yihui Xiong 2:5c805323a2c0 223 } else if(position < 0.498) {
Yihui Xiong 2:5c805323a2c0 224 position -= 0.332;
Yihui Xiong 2:5c805323a2c0 225 c = toRGB(min, max, min + position * 6 * d);
Yihui Xiong 2:5c805323a2c0 226 } else if(position < 0.664) {
Yihui Xiong 2:5c805323a2c0 227 position -= 0.498;
Yihui Xiong 2:5c805323a2c0 228 c = toRGB(min, max - position * 6 * d, max);
Yihui Xiong 2:5c805323a2c0 229 } else if(position < 0.83) {
Yihui Xiong 2:5c805323a2c0 230 position -= 0.664;
Yihui Xiong 2:5c805323a2c0 231 c = toRGB(min + d * position * 6, min, max);
Yihui Xiong 2:5c805323a2c0 232 } else {
Yihui Xiong 2:5c805323a2c0 233 position -= 0.83;
Yihui Xiong 2:5c805323a2c0 234 c = toRGB(max, min, max - position * 6 * d);
Yihui Xiong 2:5c805323a2c0 235 }
Yihui Xiong 2:5c805323a2c0 236
Yihui Xiong 2:5c805323a2c0 237 return c;
Yihui Xiong 2:5c805323a2c0 238 }
Yihui Xiong 2:5c805323a2c0 239
Yihui Xiong 2:5c805323a2c0 240 void find_wheel_options(uint32_t rgb, float *position, uint8_t *min, uint8_t *max)
Yihui Xiong 2:5c805323a2c0 241 {
Yihui Xiong 2:5c805323a2c0 242 color_t c;
Yihui Xiong 2:5c805323a2c0 243 c.rgb = rgb;
Yihui Xiong 2:5c805323a2c0 244
Yihui Xiong 2:5c805323a2c0 245 if (c.r > c.g) {
Yihui Xiong 2:5c805323a2c0 246 if (c.g >= c.b) {
Yihui Xiong 2:5c805323a2c0 247 *max = c.r;
Yihui Xiong 2:5c805323a2c0 248 *min = c.b;
Yihui Xiong 2:5c805323a2c0 249 *position = (float)(c.g - c.b) / (c.r - c.b) / 6.0;
Yihui Xiong 2:5c805323a2c0 250 } else if (c.b >= c.r) {
Yihui Xiong 2:5c805323a2c0 251 *max = c.b;
Yihui Xiong 2:5c805323a2c0 252 *min = c.g;
Yihui Xiong 2:5c805323a2c0 253 *position = (float)(c.r - c.g) / (c.b - c.g) / 6.0 + 0.664;
Yihui Xiong 2:5c805323a2c0 254 } else {
Yihui Xiong 2:5c805323a2c0 255 *max = c.r;
Yihui Xiong 2:5c805323a2c0 256 *min = c.g;
Yihui Xiong 2:5c805323a2c0 257 *position = (float)(c.b - c.g) / (c.r - c.g) / 6.0 + 0.83;
Yihui Xiong 2:5c805323a2c0 258 }
Yihui Xiong 2:5c805323a2c0 259 } else {
Yihui Xiong 2:5c805323a2c0 260 if (c.r > c.b) {
Yihui Xiong 2:5c805323a2c0 261 *max = c.g;
Yihui Xiong 2:5c805323a2c0 262 *min = c.b;
Yihui Xiong 2:5c805323a2c0 263 *position = (float)(c.r - c.b) / (c.g - c.b) / 6.0 + 0.166;
Yihui Xiong 2:5c805323a2c0 264 } else if (c.b > c.g) {
Yihui Xiong 2:5c805323a2c0 265 *max = c.b;
Yihui Xiong 2:5c805323a2c0 266 *min = c.r;
Yihui Xiong 2:5c805323a2c0 267 *position = (float)(c.g - c.r) / (c.b - c.r) / 6.0 + 0.498;
Yihui Xiong 2:5c805323a2c0 268 } else {
Yihui Xiong 2:5c805323a2c0 269 *max = c.g;
Yihui Xiong 2:5c805323a2c0 270 *min = c.r;
Yihui Xiong 2:5c805323a2c0 271 *position = (float)(c.b - c.r) / (c.g - c.r) / 6.0 + 0.332;
Yihui Xiong 2:5c805323a2c0 272 }
Yihui Xiong 2:5c805323a2c0 273 }
Yihui Xiong 2:5c805323a2c0 274 }
Yihui Xiong 2:5c805323a2c0 275
Yihui Xiong 2:5c805323a2c0 276 void ColorPixels::rainbow(uint32_t rgb)
Yihui Xiong 2:5c805323a2c0 277 {
Yihui Xiong 2:5c805323a2c0 278 float position;
Yihui Xiong 2:5c805323a2c0 279 uint8_t min, max;
Yihui Xiong 2:5c805323a2c0 280
Yihui Xiong 2:5c805323a2c0 281 find_wheel_options(rgb, &position, &min, &max);
Yihui Xiong 2:5c805323a2c0 282 for (int i = 0; i < num; i++) {
Yihui Xiong 2:5c805323a2c0 283 float current = ((int)(i + num * position) % num) / (float)num;
Yihui Xiong 2:5c805323a2c0 284 set_color(i, wheel(current, min, max));
Yihui Xiong 2:5c805323a2c0 285 }
Yihui Xiong 2:5c805323a2c0 286
Yihui Xiong 2:5c805323a2c0 287 update();
Yihui Xiong 2:5c805323a2c0 288 }
yihui 3:91af7b451005 289
yihui 3:91af7b451005 290 void ColorPixels::rainbow(uint8_t r, uint8_t g, uint8_t b)
yihui 3:91af7b451005 291 {
yihui 3:91af7b451005 292 color_t c;
yihui 3:91af7b451005 293
yihui 3:91af7b451005 294 c.r = r;
yihui 3:91af7b451005 295 c.g = g;
yihui 3:91af7b451005 296 c.b = b;
yihui 3:91af7b451005 297
yihui 3:91af7b451005 298 rainbow(c.rgb);
yihui 3:91af7b451005 299 }
yihui 3:91af7b451005 300