Ned Konz / Mbed 2 deprecated Multi_WS2811_test

Test program for my Multi_WS2811 library that started out as a fork of heroic/WS2811. My library uses hardware DMA on the FRDM-KL25Z to drive up to 16 strings of WS2811 or WS2812 LEDs in parallel.

Dependencies:   Multi_WS2811 mbed MMA8451Q

Fork of WS2811 by Heroic Robotics

NOTE: I have accidentally pushed changes for another fork of this program that I used in the recent Georgetown Carnival Power Tool Races. When I get some time, I will restore the test program to its original glory.

You can see my power tool racer (Nevermore's Revenge) here

/media/uploads/bikeNomad/img_0482.jpg

This tests my FRDM-KL25Z multi-string WS2811/WS2812 library. It uses the accelerometer to change the rainbow phase on two strings of LEDs as well as the touch sense to change brightness.

A video of this program in operation is here.

Here is the library that I developed to run the LEDs:

Import libraryMulti_WS2811

Library allowing up to 16 strings of 60 WS2811 or WS2812 LEDs to be driven from a single FRDM-KL25Z board. Uses hardware DMA to do a full 800 KHz rate without much CPU burden.

Last commit 13 Jun 2015 by Ned Konz

WS2811.cpp@16:5f75ea93f15d, 2013-10-10 (annotated)

Committer:
heroic
Date:
Thu Oct 10 21:52:59 2013 +0000
Revision:
16:5f75ea93f15d
Parent:
14:c97261a9a282
Child:
17:b4e9d8f4baa9
Bump version, add register debug.

Who changed what in which revision?

UserRevisionLine numberNew contents of line
heroic 10:62368b801d16 1 // 800 KHz WS2811 driver, kinda.
heroic 8:e3249c2b7607 2 //
heroic 8:e3249c2b7607 3 // Parameterized and modified to use soft SPI.
heroic 8:e3249c2b7607 4 // Jas Strong <jasmine@electronpusher.org>
heroic 8:e3249c2b7607 5 /*****************************************************************************/
heroic 8:e3249c2b7607 6
heroic 8:e3249c2b7607 7 #include "LedStrip.h"
heroic 8:e3249c2b7607 8 #include "WS2811.h"
heroic 16:5f75ea93f15d 9 #include "bitband.h"
heroic 8:e3249c2b7607 10
heroic 10:62368b801d16 11 extern int onewire_speed_multiplier; // speed setting for ws2811 et al.
heroic 10:62368b801d16 12
heroic 10:62368b801d16 13
heroic 8:e3249c2b7607 14 WS2811::WS2811(PinName dataPin, PinName clockPin, int n) :
heroic 8:e3249c2b7607 15 dat(dataPin),
heroic 8:e3249c2b7607 16 clk(clockPin) {
heroic 8:e3249c2b7607 17 // Allocate 3 bytes per pixel:
heroic 8:e3249c2b7607 18 numLEDs = n;
heroic 14:c97261a9a282 19 pixels = (uint8_t *)malloc(numLEDs * 3);
heroic 14:c97261a9a282 20 if (pixels) {
heroic 8:e3249c2b7607 21 memset(pixels, 0x00, numLEDs * 3); // Init to RGB 'off' state
heroic 8:e3249c2b7607 22 }
heroic 8:e3249c2b7607 23 // calibrate delay loops for NRZ
heroic 8:e3249c2b7607 24 int i;
heroic 8:e3249c2b7607 25 guardtime.start();
heroic 8:e3249c2b7607 26 for (i=0; i<1000; i++)
heroic 8:e3249c2b7607 27 /* do nothing */;
heroic 8:e3249c2b7607 28 i=guardtime.read_us();
heroic 8:e3249c2b7607 29 printf("ws2811: 1000 iters took %d usec.\n", i);
heroic 11:d127f93be182 30 bogocal = (1000 / (onewire_speed_multiplier * i * 4.2)); // iterations per bitcell (417 nsec)
heroic 8:e3249c2b7607 31 printf("ws2811: calibrating to %d bogojiffies.\n", bogocal);
heroic 13:1f65330abe92 32
heroic 13:1f65330abe92 33 data_mask = dat.get_mask();
heroic 13:1f65330abe92 34 clock_mask = clk.get_mask();
heroic 13:1f65330abe92 35 data_set = dat.get_set();
heroic 13:1f65330abe92 36 data_clr = dat.get_clr();
heroic 13:1f65330abe92 37 clock_set = clk.get_set();
heroic 13:1f65330abe92 38 clock_clr = clk.get_clr();
heroic 13:1f65330abe92 39
heroic 16:5f75ea93f15d 40 printf("ws2811: data mask 0x%x, data set reg 0x%x, data clear reg 0x%x\r\n", data_mask, data_set, data_clr);
heroic 16:5f75ea93f15d 41 printf("ws2811: (unused) clock mask 0x%x, clock set reg 0x%x, clock clear reg 0x%x\r\n", clock_mask, clock_set, clock_clr);
heroic 16:5f75ea93f15d 42
heroic 8:e3249c2b7607 43 }
heroic 8:e3249c2b7607 44
heroic 8:e3249c2b7607 45 /*
heroic 8:e3249c2b7607 46 * These chips use a one-wire protocol based on a sort of NRZ signalling- jas.
heroic 8:e3249c2b7607 47 */
heroic 8:e3249c2b7607 48
heroic 8:e3249c2b7607 49 void WS2811::write(uint8_t byte) {
heroic 13:1f65330abe92 50
heroic 8:e3249c2b7607 51 for (int i=0; i<8; i++) {
heroic 8:e3249c2b7607 52 if (byte & 0x80)
heroic 8:e3249c2b7607 53 writebit(1);
heroic 8:e3249c2b7607 54 else
heroic 8:e3249c2b7607 55 writebit(0);
heroic 8:e3249c2b7607 56 byte <<= 1;
heroic 8:e3249c2b7607 57 }
heroic 8:e3249c2b7607 58 }
heroic 8:e3249c2b7607 59
heroic 8:e3249c2b7607 60 inline void WS2811::celldelay(void) {
heroic 13:1f65330abe92 61 for (volatile int i = 0; i<bogocal; i++)
heroic 8:e3249c2b7607 62 /* do nothing */ ;
heroic 8:e3249c2b7607 63 }
heroic 8:e3249c2b7607 64
heroic 13:1f65330abe92 65 inline void WS2811::writebit(bool bit) {
heroic 8:e3249c2b7607 66 // first cell is always 1
heroic 13:1f65330abe92 67 (*data_set) = data_mask;
heroic 8:e3249c2b7607 68 celldelay();
heroic 8:e3249c2b7607 69 if (bit) {
heroic 13:1f65330abe92 70 (*clock_set) = data_mask; // dummy, we don't care but must take constant time
heroic 8:e3249c2b7607 71 celldelay();
heroic 8:e3249c2b7607 72 } else {
heroic 13:1f65330abe92 73 (*data_clr) = data_mask;
heroic 8:e3249c2b7607 74 celldelay();
heroic 8:e3249c2b7607 75 }
heroic 8:e3249c2b7607 76 // last cell is always 0
heroic 13:1f65330abe92 77 (*data_clr) = data_mask;
heroic 8:e3249c2b7607 78 celldelay();
heroic 8:e3249c2b7607 79 }
heroic 8:e3249c2b7607 80
heroic 8:e3249c2b7607 81 void WS2811::begin(void) {
heroic 8:e3249c2b7607 82 blank();
heroic 8:e3249c2b7607 83 show();
heroic 8:e3249c2b7607 84 }
heroic 8:e3249c2b7607 85
heroic 8:e3249c2b7607 86 uint16_t WS2811::numPixels(void) {
heroic 8:e3249c2b7607 87 return numLEDs;
heroic 8:e3249c2b7607 88 }
heroic 8:e3249c2b7607 89
heroic 8:e3249c2b7607 90 void WS2811::blank(void) {
heroic 8:e3249c2b7607 91 memset(pixels, 0x00, numLEDs * 3);
heroic 8:e3249c2b7607 92 }
heroic 8:e3249c2b7607 93
heroic 8:e3249c2b7607 94 void WS2811::show(void) {
heroic 8:e3249c2b7607 95 uint16_t i, nl3 = numLEDs * 3; // 3 bytes per LED
heroic 8:e3249c2b7607 96 while (guardtime.read_us() < 50)
heroic 8:e3249c2b7607 97 /* spin */;
heroic 13:1f65330abe92 98 __disable_irq();
heroic 8:e3249c2b7607 99 for (i=0; i<nl3; i++ ) {
heroic 8:e3249c2b7607 100 write(pixels[i]);
heroic 8:e3249c2b7607 101 }
heroic 13:1f65330abe92 102 __enable_irq();
heroic 8:e3249c2b7607 103 guardtime.reset();
heroic 8:e3249c2b7607 104 }
heroic 8:e3249c2b7607 105
heroic 8:e3249c2b7607 106
heroic 8:e3249c2b7607 107 uint32_t WS2811::total_luminance(void) {
heroic 8:e3249c2b7607 108 uint32_t running_total;
heroic 8:e3249c2b7607 109 running_total = 0;
heroic 8:e3249c2b7607 110 for (int i=0; i<numLEDs*3; i++)
heroic 8:e3249c2b7607 111 running_total += pixels[i];
heroic 8:e3249c2b7607 112 return running_total;
heroic 8:e3249c2b7607 113 }
heroic 8:e3249c2b7607 114
heroic 8:e3249c2b7607 115 // Convert R,G,B to combined 32-bit color
heroic 8:e3249c2b7607 116 uint32_t WS2811::Color(uint8_t r, uint8_t g, uint8_t b) {
heroic 8:e3249c2b7607 117 // Take the lowest 7 bits of each value and append them end to end
heroic 8:e3249c2b7607 118 // We have the top bit set high (its a 'parity-like' bit in the protocol
heroic 8:e3249c2b7607 119 // and must be set!)
heroic 8:e3249c2b7607 120 return ((uint32_t)g << 16) | ((uint32_t)r << 8) | (uint32_t)b;
heroic 8:e3249c2b7607 121 }
heroic 8:e3249c2b7607 122
heroic 8:e3249c2b7607 123 // store the rgb component in our array
heroic 8:e3249c2b7607 124 void WS2811::setPixelColor(uint16_t n, uint8_t r, uint8_t g, uint8_t b) {
heroic 8:e3249c2b7607 125 if (n >= numLEDs) return; // '>=' because arrays are 0-indexed
heroic 8:e3249c2b7607 126
heroic 8:e3249c2b7607 127 pixels[n*3 ] = g;
heroic 8:e3249c2b7607 128 pixels[n*3+1] = r;
heroic 8:e3249c2b7607 129 pixels[n*3+2] = b;
heroic 8:e3249c2b7607 130 }
heroic 8:e3249c2b7607 131
heroic 8:e3249c2b7607 132 void WS2811::setPixelR(uint16_t n, uint8_t r) {
heroic 8:e3249c2b7607 133 if (n >= numLEDs) return; // '>=' because arrays are 0-indexed
heroic 8:e3249c2b7607 134
heroic 8:e3249c2b7607 135 pixels[n*3+1] = r;
heroic 8:e3249c2b7607 136 }
heroic 8:e3249c2b7607 137
heroic 8:e3249c2b7607 138 void WS2811::setPixelG(uint16_t n, uint8_t g) {
heroic 8:e3249c2b7607 139 if (n >= numLEDs) return; // '>=' because arrays are 0-indexed
heroic 8:e3249c2b7607 140
heroic 8:e3249c2b7607 141 pixels[n*3] = g;
heroic 8:e3249c2b7607 142 }
heroic 8:e3249c2b7607 143
heroic 8:e3249c2b7607 144 void WS2811::setPixelB(uint16_t n, uint8_t b) {
heroic 8:e3249c2b7607 145 if (n >= numLEDs) return; // '>=' because arrays are 0-indexed
heroic 8:e3249c2b7607 146
heroic 8:e3249c2b7607 147 pixels[n*3+2] = b;
heroic 8:e3249c2b7607 148 }
heroic 8:e3249c2b7607 149
heroic 12:7ebd51549c04 150 void WS2811::setPackedPixels(uint8_t * buffer, uint32_t n) {
heroic 12:7ebd51549c04 151 if (n >= numLEDs) return;
heroic 12:7ebd51549c04 152 memcpy(pixels, buffer, (size_t) (n*3));
heroic 12:7ebd51549c04 153 }
heroic 12:7ebd51549c04 154
heroic 8:e3249c2b7607 155 void WS2811::setPixelColor(uint16_t n, uint32_t c) {
heroic 8:e3249c2b7607 156 if (n >= numLEDs) return; // '>=' because arrays are 0-indexed
heroic 8:e3249c2b7607 157
heroic 8:e3249c2b7607 158 pixels[n*3 ] = (c >> 16);
heroic 8:e3249c2b7607 159 pixels[n*3+1] = (c >> 8);
heroic 8:e3249c2b7607 160 pixels[n*3+2] = c;
heroic 8:e3249c2b7607 161 }