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@36:9ee1ec2135d5, 2015-06-12 (annotated)

Committer:
Ned Konz
Date:
Fri Jun 12 18:23:46 2015 -0700
Revision:
36:9ee1ec2135d5
Parent:
34:cd56c00ed910
Child:
37:e25d212ee3fe
Refactored self-test; getting wait_for_dma_done() to work

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 36:9ee1ec2135d5 16 // PTA5 (D5) servomotor (20 msec period; 1.0-2.0msec ON)
Ned Konz 34:cd56c00ed910 17
Ned Konz 36:9ee1ec2135d5 18 const unsigned DATA_OUT_PIN1 = 2; // PTD2
Ned Konz 36:9ee1ec2135d5 19 const unsigned DATA_OUT_PIN2 = 3; // PTD3
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 36:9ee1ec2135d5 25 const unsigned 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 36:9ee1ec2135d5 37 PwmOut rled(LED_RED); // max = 0.0
Ned Konz 36:9ee1ec2135d5 38 PwmOut gled(LED_GREEN); // max = 0.0
Ned Konz 34:cd56c00ed910 39 // LED_BLUE is on PTD1
Ned Konz 36:9ee1ec2135d5 40 PwmOut eyes(D3); // also redLED1; max = 1.0
Ned Konz 36:9ee1ec2135d5 41 PwmOut servo(D5);
Ned Konz 36:9ee1ec2135d5 42 DigitalOut greenLED2(D4); // max = 1.0
Ned Konz 36:9ee1ec2135d5 43 DigitalIn button1(D6); // low=ON, debounced
Ned Konz 36:9ee1ec2135d5 44 DigitalIn button2(D7); // low=ON, debounced
Ned Konz 34:cd56c00ed910 45
Ned Konz 36:9ee1ec2135d5 46 // Limits
Ned Konz 36:9ee1ec2135d5 47 const float maxBrite = 0.5;
Ned Konz 34:cd56c00ed910 48
Ned Konz 36:9ee1ec2135d5 49 const float minServo = -0.7; // -1.0 = -60°
Ned Konz 36:9ee1ec2135d5 50 const float maxServo = 0.6; // 1.0 = +60°
Ned Konz 34:cd56c00ed910 51
Ned Konz 34:cd56c00ed910 52 // @brief sets different colors in each of the LEDs of a strip
Ned Konz 34:cd56c00ed910 53 // @param strip the light strip
Ned Konz 34:cd56c00ed910 54 // @param sat saturation, 0.0 - 1.0
Ned Konz 34:cd56c00ed910 55 // @param brite brightness, 0.0 - 1.0
Ned Konz 34:cd56c00ed910 56 // @param hueShift shift, 0.0 - 1.0 is equivalent to 0 - 360 degrees
Ned Konz 34:cd56c00ed910 57 static void showRainbow(MyWS2811 &strip, float sat, float brite, float hueShift)
Ned Konz 34:cd56c00ed910 58 {
Ned Konz 34:cd56c00ed910 59 unsigned nLEDs = strip.numPixels();
Ned Konz 34:cd56c00ed910 60 for (unsigned i = 0; i < nLEDs; i++) {
Ned Konz 34:cd56c00ed910 61 uint8_t r, g, b;
Ned Konz 34:cd56c00ed910 62 float hue = ((float)i / (float)nLEDs) + hueShift;
Ned Konz 34:cd56c00ed910 63 HSBtoRGB(hue, sat, brite, &r, &g, &b);
Ned Konz 34:cd56c00ed910 64 strip.setPixelColor(i, r, g, b);
Ned Konz 34:cd56c00ed910 65 }
Ned Konz 34:cd56c00ed910 66 strip.show();
Ned Konz 34:cd56c00ed910 67 }
Ned Konz 34:cd56c00ed910 68
Ned Konz 34:cd56c00ed910 69 static void showSolidColor(MyWS2811 &strip, uint8_t r, uint8_t g, uint8_t b)
Ned Konz 34:cd56c00ed910 70 {
Ned Konz 34:cd56c00ed910 71 unsigned nLEDs = strip.numPixels();
Ned Konz 34:cd56c00ed910 72 for (unsigned i = 0; i < nLEDs; i++) {
Ned Konz 34:cd56c00ed910 73 strip.setPixelColor(i, r, g, b);
Ned Konz 34:cd56c00ed910 74 }
Ned Konz 34:cd56c00ed910 75 strip.show();
Ned Konz 34:cd56c00ed910 76 }
Ned Konz 34:cd56c00ed910 77
Ned Konz 36:9ee1ec2135d5 78 // range is -1.0 (full CCW) to +1.0 (full CW)
Ned Konz 36:9ee1ec2135d5 79 static void positionServo(float pos)
Ned Konz 36:9ee1ec2135d5 80 {
Ned Konz 36:9ee1ec2135d5 81 if (pos < minServo) pos = minServo;
Ned Konz 36:9ee1ec2135d5 82 else if (pos > maxServo) pos = maxServo;
Ned Konz 36:9ee1ec2135d5 83
Ned Konz 36:9ee1ec2135d5 84 servo.pulsewidth_us((1.5 + (pos / 2.0)) * 1000.0);
Ned Konz 36:9ee1ec2135d5 85 }
Ned Konz 36:9ee1ec2135d5 86
Ned Konz 36:9ee1ec2135d5 87 static void selfTestServo()
Ned Konz 34:cd56c00ed910 88 {
Ned Konz 36:9ee1ec2135d5 89 pc.printf("Servo:\r\n");
Ned Konz 36:9ee1ec2135d5 90 pc.printf("CCW, ");
Ned Konz 36:9ee1ec2135d5 91 positionServo(-1.0);
Ned Konz 36:9ee1ec2135d5 92 wait(1.0);
Ned Konz 36:9ee1ec2135d5 93 pc.printf("CW, ");
Ned Konz 36:9ee1ec2135d5 94 positionServo(+1.0);
Ned Konz 36:9ee1ec2135d5 95 wait(1.0);
Ned Konz 36:9ee1ec2135d5 96 pc.printf("center.\r\n");
Ned Konz 36:9ee1ec2135d5 97 positionServo(0.0);
Ned Konz 36:9ee1ec2135d5 98 }
Ned Konz 34:cd56c00ed910 99
Ned Konz 36:9ee1ec2135d5 100 static void selfTestLEDs()
Ned Konz 36:9ee1ec2135d5 101 {
Ned Konz 34:cd56c00ed910 102 pc.printf("LEDs .");
Ned Konz 34:cd56c00ed910 103 rled = 0.0; // red LED on
Ned Konz 34:cd56c00ed910 104 wait(1.0);
Ned Konz 34:cd56c00ed910 105 pc.printf(".");
Ned Konz 34:cd56c00ed910 106 rled = 1.0; // red LED off, green LED on
Ned Konz 34:cd56c00ed910 107 gled = 0.0;
Ned Konz 34:cd56c00ed910 108 wait(1.0);
Ned Konz 34:cd56c00ed910 109 pc.printf(".");
Ned Konz 34:cd56c00ed910 110 gled = 1.0; // green LED off, eyes on
Ned Konz 34:cd56c00ed910 111 eyes = 1.0;
Ned Konz 34:cd56c00ed910 112 wait(1.0);
Ned Konz 34:cd56c00ed910 113 pc.printf(".");
Ned Konz 34:cd56c00ed910 114 eyes = 0.0;
Ned Konz 36:9ee1ec2135d5 115 pc.printf("\r\n");
Ned Konz 36:9ee1ec2135d5 116 }
Ned Konz 34:cd56c00ed910 117
Ned Konz 36:9ee1ec2135d5 118 static void selfTestLightStrips()
Ned Konz 36:9ee1ec2135d5 119 {
Ned Konz 34:cd56c00ed910 120 pc.printf("light strips");
Ned Konz 36:9ee1ec2135d5 121 uint8_t rgb[4] = { 255 * maxBrite, 0, 0, 0 };
Ned Konz 34:cd56c00ed910 122 for (int i = 0; i < 3; i++) {
Ned Konz 34:cd56c00ed910 123 showSolidColor(lightStrip1, rgb[0], rgb[1], rgb[2]);
Ned Konz 34:cd56c00ed910 124 showSolidColor(lightStrip2, rgb[1], rgb[2], rgb[0]);
Ned Konz 34:cd56c00ed910 125 MyWS2811::startDMA();
Ned Konz 34:cd56c00ed910 126 if (i == 2) break;
Ned Konz 34:cd56c00ed910 127 wait(1.0);
Ned Konz 34:cd56c00ed910 128 rgb[3] = rgb[2];
Ned Konz 34:cd56c00ed910 129 rgb[2] = rgb[1];
Ned Konz 34:cd56c00ed910 130 rgb[1] = rgb[0];
Ned Konz 34:cd56c00ed910 131 rgb[0] = rgb[3];
Ned Konz 34:cd56c00ed910 132 pc.printf(".");
Ned Konz 34:cd56c00ed910 133 }
Ned Konz 34:cd56c00ed910 134
Ned Konz 34:cd56c00ed910 135 pc.printf("\r\n");
Ned Konz 34:cd56c00ed910 136 }
Ned Konz 34:cd56c00ed910 137
Ned Konz 36:9ee1ec2135d5 138 static void selfTest()
Ned Konz 36:9ee1ec2135d5 139 {
Ned Konz 36:9ee1ec2135d5 140 pc.printf("self test: ");
Ned Konz 36:9ee1ec2135d5 141
Ned Konz 36:9ee1ec2135d5 142 selfTestServo();
Ned Konz 36:9ee1ec2135d5 143 selfTestLEDs();
Ned Konz 36:9ee1ec2135d5 144 selfTestLightStrips();
Ned Konz 36:9ee1ec2135d5 145 }
Ned Konz 36:9ee1ec2135d5 146
Ned Konz 34:cd56c00ed910 147 int main(void)
Ned Konz 34:cd56c00ed910 148 {
Ned Konz 34:cd56c00ed910 149 pc.baud(115200);
Ned Konz 36:9ee1ec2135d5 150 pc.printf("\r\n\r\nNevermore's Revenge!\r\ncompiled " __DATE__ ", " __TIME__ "\r\n");
Ned Konz 34:cd56c00ed910 151
Ned Konz 34:cd56c00ed910 152 lightStrip1.begin();
Ned Konz 34:cd56c00ed910 153 lightStrip2.begin();
Ned Konz 34:cd56c00ed910 154
Ned Konz 34:cd56c00ed910 155 rled = 1.0;
Ned Konz 34:cd56c00ed910 156 gled = 1.0;
Ned Konz 34:cd56c00ed910 157 eyes = 0.0;
Ned Konz 36:9ee1ec2135d5 158 greenLED2 = 0.0;
Ned Konz 36:9ee1ec2135d5 159 servo.period_ms(20);
Ned Konz 34:cd56c00ed910 160
Ned Konz 34:cd56c00ed910 161 selfTest();
Ned Konz 34:cd56c00ed910 162
Ned Konz 34:cd56c00ed910 163 float xyz[3];
Ned Konz 34:cd56c00ed910 164
Ned Konz 34:cd56c00ed910 165 for (;;) {
Ned Konz 34:cd56c00ed910 166 acc.getAccAllAxis(xyz);
Ned Konz 34:cd56c00ed910 167 pc.printf("x: %f y: %f z: %f\r\n", xyz[0], xyz[1], xyz[2]);
Ned Konz 34:cd56c00ed910 168 rled = 1.0 - fabs(xyz[0]);
Ned Konz 34:cd56c00ed910 169 gled = 1.0 - fabs(xyz[1]);
Ned Konz 34:cd56c00ed910 170 eyes = fabs(xyz[2]);
Ned Konz 34:cd56c00ed910 171 showRainbow(lightStrip1, 1.0, maxBrite, fabs(xyz[0]));
Ned Konz 34:cd56c00ed910 172 showRainbow(lightStrip2, 1.0, maxBrite, fabs(xyz[1]));
Ned Konz 34:cd56c00ed910 173 MyWS2811::startDMA();
Ned Konz 36:9ee1ec2135d5 174 MyWS2811::wait_for_dma_done();
Ned Konz 36:9ee1ec2135d5 175 wait(0.000055);
Ned Konz 34:cd56c00ed910 176 }
Ned Konz 34:cd56c00ed910 177 }
Ned Konz 34:cd56c00ed910 178