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

main.cpp@34:cd56c00ed910, 2015-06-12 (annotated)

Committer:
Ned Konz
Date:
Fri Jun 12 11:44:26 2015 -0700
Revision:
34:cd56c00ed910
Parent:
33:43b504e417e3
Child:
36:9ee1ec2135d5
fixed up IO; DMA not yet ending

Who changed what in which revision?

UserRevisionLine numberNew contents of line
Ned Konz 34:cd56c00ed910 1 #include "mbed.h"
Ned Konz 34:cd56c00ed910 2 #include "Colors.h"
Ned Konz 34:cd56c00ed910 3 #include "MMA8451Q.h"
Ned Konz 34:cd56c00ed910 4
Ned Konz 34:cd56c00ed910 5 #define MMA8451_I2C_ADDRESS (0x1d<<1)
Ned Konz 34:cd56c00ed910 6
Ned Konz 34:cd56c00ed910 7 #define INSTANTIATE_TEMPLATES 1
Ned Konz 34:cd56c00ed910 8 #include "WS2811.h"
Ned Konz 34:cd56c00ed910 9
Ned Konz 34:cd56c00ed910 10 // I/O pin usage
Ned Konz 34:cd56c00ed910 11 // PTD0 TPM0 CH0 monitor
Ned Konz 34:cd56c00ed910 12 // PTD1 TPM0 CH1 monitor
Ned Konz 34:cd56c00ed910 13 // PTD2 (D11) data output for strip# 1
Ned Konz 34:cd56c00ed910 14 // PTD3 (D12) data output for strip# 2
Ned Konz 34:cd56c00ed910 15 // PTA2 (D3) blinking eyes output (HI = ON)
Ned Konz 34:cd56c00ed910 16
Ned Konz 34:cd56c00ed910 17 unsigned const DATA_OUT_PIN1 = 2; // PTD2
Ned Konz 34:cd56c00ed910 18 unsigned const DATA_OUT_PIN2 = 3; // PTD3
Ned Konz 34:cd56c00ed910 19
Ned Konz 34:cd56c00ed910 20 const unsigned MAX_LEDS_PER_STRIP = 32;
Ned Konz 34:cd56c00ed910 21
Ned Konz 34:cd56c00ed910 22 // per LED: 3 * 20 mA = 60mA max
Ned Konz 34:cd56c00ed910 23 // 60 LEDs: 60 * 60mA = 3600 mA max
Ned Konz 34:cd56c00ed910 24 // 120 LEDs: 7200 mA max
Ned Konz 34:cd56c00ed910 25 unsigned const nLEDs = MAX_LEDS_PER_STRIP;
Ned Konz 34:cd56c00ed910 26
Ned Konz 34:cd56c00ed910 27 template class WS2811<MAX_LEDS_PER_STRIP>;
Ned Konz 34:cd56c00ed910 28
Ned Konz 34:cd56c00ed910 29 typedef WS2811<MAX_LEDS_PER_STRIP> MyWS2811;
Ned Konz 34:cd56c00ed910 30
Ned Konz 34:cd56c00ed910 31 MyWS2811 lightStrip1(nLEDs, DATA_OUT_PIN1);
Ned Konz 34:cd56c00ed910 32 MyWS2811 lightStrip2(nLEDs, DATA_OUT_PIN2);
Ned Konz 34:cd56c00ed910 33
Ned Konz 34:cd56c00ed910 34 Serial pc(USBTX, USBRX);
Ned Konz 34:cd56c00ed910 35
Ned Konz 34:cd56c00ed910 36 MMA8451Q acc(PTE25, PTE24, MMA8451_I2C_ADDRESS);
Ned Konz 34:cd56c00ed910 37 PwmOut rled(LED_RED); // max = 0.0
Ned Konz 34:cd56c00ed910 38 PwmOut gled(LED_GREEN); // max = 0.0
Ned Konz 34:cd56c00ed910 39 // LED_BLUE is on PTD1
Ned Konz 34:cd56c00ed910 40 PwmOut eyes(D3); // max = 1.0
Ned Konz 34:cd56c00ed910 41
Ned Konz 34:cd56c00ed910 42 float touchPercentage;
Ned Konz 34:cd56c00ed910 43
Ned Konz 34:cd56c00ed910 44 float const minBrite = 0.2;
Ned Konz 34:cd56c00ed910 45 float const maxBrite = 0.5;
Ned Konz 34:cd56c00ed910 46 float brite;
Ned Konz 34:cd56c00ed910 47
Ned Konz 34:cd56c00ed910 48 // @brief sets different colors in each of the LEDs of a strip
Ned Konz 34:cd56c00ed910 49 // @param strip the light strip
Ned Konz 34:cd56c00ed910 50 // @param sat saturation, 0.0 - 1.0
Ned Konz 34:cd56c00ed910 51 // @param brite brightness, 0.0 - 1.0
Ned Konz 34:cd56c00ed910 52 // @param hueShift shift, 0.0 - 1.0 is equivalent to 0 - 360 degrees
Ned Konz 34:cd56c00ed910 53 static void showRainbow(MyWS2811 &strip, float sat, float brite, float hueShift)
Ned Konz 34:cd56c00ed910 54 {
Ned Konz 34:cd56c00ed910 55 unsigned nLEDs = strip.numPixels();
Ned Konz 34:cd56c00ed910 56 for (unsigned i = 0; i < nLEDs; i++) {
Ned Konz 34:cd56c00ed910 57 uint8_t r, g, b;
Ned Konz 34:cd56c00ed910 58 float hue = ((float)i / (float)nLEDs) + hueShift;
Ned Konz 34:cd56c00ed910 59 HSBtoRGB(hue, sat, brite, &r, &g, &b);
Ned Konz 34:cd56c00ed910 60 strip.setPixelColor(i, r, g, b);
Ned Konz 34:cd56c00ed910 61 }
Ned Konz 34:cd56c00ed910 62 strip.show();
Ned Konz 34:cd56c00ed910 63 }
Ned Konz 34:cd56c00ed910 64
Ned Konz 34:cd56c00ed910 65 static void showSolidColor(MyWS2811 &strip, uint8_t r, uint8_t g, uint8_t b)
Ned Konz 34:cd56c00ed910 66 {
Ned Konz 34:cd56c00ed910 67 unsigned nLEDs = strip.numPixels();
Ned Konz 34:cd56c00ed910 68 for (unsigned i = 0; i < nLEDs; i++) {
Ned Konz 34:cd56c00ed910 69 strip.setPixelColor(i, r, g, b);
Ned Konz 34:cd56c00ed910 70 }
Ned Konz 34:cd56c00ed910 71 strip.show();
Ned Konz 34:cd56c00ed910 72 }
Ned Konz 34:cd56c00ed910 73
Ned Konz 34:cd56c00ed910 74 static void selfTest()
Ned Konz 34:cd56c00ed910 75 {
Ned Konz 34:cd56c00ed910 76 pc.printf("self test: ");
Ned Konz 34:cd56c00ed910 77
Ned Konz 34:cd56c00ed910 78 pc.printf("LEDs .");
Ned Konz 34:cd56c00ed910 79 rled = 0.0; // red LED on
Ned Konz 34:cd56c00ed910 80 wait(1.0);
Ned Konz 34:cd56c00ed910 81 pc.printf(".");
Ned Konz 34:cd56c00ed910 82 rled = 1.0; // red LED off, green LED on
Ned Konz 34:cd56c00ed910 83 gled = 0.0;
Ned Konz 34:cd56c00ed910 84 wait(1.0);
Ned Konz 34:cd56c00ed910 85 pc.printf(".");
Ned Konz 34:cd56c00ed910 86 gled = 1.0; // green LED off, eyes on
Ned Konz 34:cd56c00ed910 87 eyes = 1.0;
Ned Konz 34:cd56c00ed910 88 wait(1.0);
Ned Konz 34:cd56c00ed910 89 pc.printf(".");
Ned Konz 34:cd56c00ed910 90 eyes = 0.0;
Ned Konz 34:cd56c00ed910 91 pc.printf(", ");
Ned Konz 34:cd56c00ed910 92
Ned Konz 34:cd56c00ed910 93 pc.printf("light strips");
Ned Konz 34:cd56c00ed910 94 float rgb[4] = { 1.0, 0.0, 0.0, 0.0 };
Ned Konz 34:cd56c00ed910 95 for (int i = 0; i < 3; i++) {
Ned Konz 34:cd56c00ed910 96 showSolidColor(lightStrip1, rgb[0], rgb[1], rgb[2]);
Ned Konz 34:cd56c00ed910 97 showSolidColor(lightStrip2, rgb[1], rgb[2], rgb[0]);
Ned Konz 34:cd56c00ed910 98 MyWS2811::startDMA();
Ned Konz 34:cd56c00ed910 99 if (i == 2) break;
Ned Konz 34:cd56c00ed910 100 wait(1.0);
Ned Konz 34:cd56c00ed910 101 rgb[3] = rgb[2];
Ned Konz 34:cd56c00ed910 102 rgb[2] = rgb[1];
Ned Konz 34:cd56c00ed910 103 rgb[1] = rgb[0];
Ned Konz 34:cd56c00ed910 104 rgb[0] = rgb[3];
Ned Konz 34:cd56c00ed910 105 pc.printf(".");
Ned Konz 34:cd56c00ed910 106 }
Ned Konz 34:cd56c00ed910 107
Ned Konz 34:cd56c00ed910 108 pc.printf("\r\n");
Ned Konz 34:cd56c00ed910 109 }
Ned Konz 34:cd56c00ed910 110
Ned Konz 34:cd56c00ed910 111 int main(void)
Ned Konz 34:cd56c00ed910 112 {
Ned Konz 34:cd56c00ed910 113 pc.baud(115200);
Ned Konz 34:cd56c00ed910 114 pc.printf("\r\nNevermore's Revenge!\r\ncompiled " __DATE__ ", " __TIME__ "\r\n");
Ned Konz 34:cd56c00ed910 115
Ned Konz 34:cd56c00ed910 116 lightStrip1.begin();
Ned Konz 34:cd56c00ed910 117 lightStrip2.begin();
Ned Konz 34:cd56c00ed910 118
Ned Konz 34:cd56c00ed910 119 rled = 1.0;
Ned Konz 34:cd56c00ed910 120 gled = 1.0;
Ned Konz 34:cd56c00ed910 121 eyes = 0.0;
Ned Konz 34:cd56c00ed910 122
Ned Konz 34:cd56c00ed910 123 selfTest();
Ned Konz 34:cd56c00ed910 124
Ned Konz 34:cd56c00ed910 125 float xyz[3];
Ned Konz 34:cd56c00ed910 126
Ned Konz 34:cd56c00ed910 127 for (;;) {
Ned Konz 34:cd56c00ed910 128 acc.getAccAllAxis(xyz);
Ned Konz 34:cd56c00ed910 129 pc.printf("x: %f y: %f z: %f\r\n", xyz[0], xyz[1], xyz[2]);
Ned Konz 34:cd56c00ed910 130 rled = 1.0 - fabs(xyz[0]);
Ned Konz 34:cd56c00ed910 131 gled = 1.0 - fabs(xyz[1]);
Ned Konz 34:cd56c00ed910 132 eyes = fabs(xyz[2]);
Ned Konz 34:cd56c00ed910 133 showRainbow(lightStrip1, 1.0, maxBrite, fabs(xyz[0]));
Ned Konz 34:cd56c00ed910 134 showRainbow(lightStrip2, 1.0, maxBrite, fabs(xyz[1]));
Ned Konz 34:cd56c00ed910 135 MyWS2811::startDMA();
Ned Konz 34:cd56c00ed910 136 }
Ned Konz 34:cd56c00ed910 137 }
Ned Konz 34:cd56c00ed910 138