API追加ver
Fork of NeoStrip by
NeoStrip.cpp@0:9f237b11f0a8, 2014-03-12 (annotated)
- Committer:
- aswild
- Date:
- Wed Mar 12 18:36:58 2014 +0000
- Revision:
- 0:9f237b11f0a8
- Child:
- 1:f531a2be180d
Initial version
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
aswild | 0:9f237b11f0a8 | 1 | /********************************************** |
aswild | 0:9f237b11f0a8 | 2 | * NeoStrip.cpp |
aswild | 0:9f237b11f0a8 | 3 | * |
aswild | 0:9f237b11f0a8 | 4 | * Allen Wild |
aswild | 0:9f237b11f0a8 | 5 | * March 2014 |
aswild | 0:9f237b11f0a8 | 6 | * |
aswild | 0:9f237b11f0a8 | 7 | * Controls a strip of Adafruit NeoPixels, addressable RGB LEDs |
aswild | 0:9f237b11f0a8 | 8 | * Currently, because of the global nature of the IO register and bitmask variables, |
aswild | 0:9f237b11f0a8 | 9 | * it is only possible to use one NeoStrip instance at a time. |
aswild | 0:9f237b11f0a8 | 10 | * |
aswild | 0:9f237b11f0a8 | 11 | * This library supports only the NXP LPC1768! |
aswild | 0:9f237b11f0a8 | 12 | */ |
aswild | 0:9f237b11f0a8 | 13 | |
aswild | 0:9f237b11f0a8 | 14 | #include "mbed.h" |
aswild | 0:9f237b11f0a8 | 15 | #include "NeoStrip.h" |
aswild | 0:9f237b11f0a8 | 16 | |
aswild | 0:9f237b11f0a8 | 17 | // function to write to the strip, implemented in ARM assembly |
aswild | 0:9f237b11f0a8 | 18 | extern "C" void neo_out(NeoColor*, int); |
aswild | 0:9f237b11f0a8 | 19 | |
aswild | 0:9f237b11f0a8 | 20 | // FastIO register address and bitmask for the GPIO pin |
aswild | 0:9f237b11f0a8 | 21 | // because these are imported in the assembly |
aswild | 0:9f237b11f0a8 | 22 | uint32_t neo_fio_reg; |
aswild | 0:9f237b11f0a8 | 23 | uint32_t neo_bitmask; |
aswild | 0:9f237b11f0a8 | 24 | |
aswild | 0:9f237b11f0a8 | 25 | NeoStrip::NeoStrip(PinName pin, int N) : N(N) |
aswild | 0:9f237b11f0a8 | 26 | { |
aswild | 0:9f237b11f0a8 | 27 | bright = 0.5; |
aswild | 0:9f237b11f0a8 | 28 | Nbytes = N * 3; |
aswild | 0:9f237b11f0a8 | 29 | strip = (NeoColor*)malloc(N * sizeof(NeoColor)); |
aswild | 0:9f237b11f0a8 | 30 | if (strip == NULL) |
aswild | 0:9f237b11f0a8 | 31 | { |
aswild | 0:9f237b11f0a8 | 32 | printf("NeoStrip: ERROR unable to malloc strip data"); |
aswild | 0:9f237b11f0a8 | 33 | N = 0; |
aswild | 0:9f237b11f0a8 | 34 | } |
aswild | 0:9f237b11f0a8 | 35 | |
aswild | 0:9f237b11f0a8 | 36 | gpio_init(&gpio, pin, PIN_OUTPUT); // initialize GPIO registers |
aswild | 0:9f237b11f0a8 | 37 | neo_fio_reg = (uint32_t)gpio.reg_dir; // set registers and bitmask for |
aswild | 0:9f237b11f0a8 | 38 | neo_bitmask = 1 << ((int)pin & 0x1F); // the assembly to use |
aswild | 0:9f237b11f0a8 | 39 | } |
aswild | 0:9f237b11f0a8 | 40 | |
aswild | 0:9f237b11f0a8 | 41 | void NeoStrip::setBrightness(float bright) |
aswild | 0:9f237b11f0a8 | 42 | { |
aswild | 0:9f237b11f0a8 | 43 | this->bright = bright; |
aswild | 0:9f237b11f0a8 | 44 | } |
aswild | 0:9f237b11f0a8 | 45 | |
aswild | 0:9f237b11f0a8 | 46 | void NeoStrip::setPixel(int p, int color) |
aswild | 0:9f237b11f0a8 | 47 | { |
aswild | 0:9f237b11f0a8 | 48 | int red = (color & 0xFF0000) >> 16; |
aswild | 0:9f237b11f0a8 | 49 | int green = (color & 0x00FF00) >> 8; |
aswild | 0:9f237b11f0a8 | 50 | int blue = (color & 0x0000FF); |
aswild | 0:9f237b11f0a8 | 51 | setPixel(p, red, green, blue); |
aswild | 0:9f237b11f0a8 | 52 | } |
aswild | 0:9f237b11f0a8 | 53 | |
aswild | 0:9f237b11f0a8 | 54 | void NeoStrip::setPixel(int p, uint8_t red, uint8_t green, uint8_t blue) |
aswild | 0:9f237b11f0a8 | 55 | { |
aswild | 0:9f237b11f0a8 | 56 | // set the given pixel's RGB values |
aswild | 0:9f237b11f0a8 | 57 | // the array is indexed modulo N to avoid overflow |
aswild | 0:9f237b11f0a8 | 58 | strip[p % N].red = (uint8_t)(red * bright); |
aswild | 0:9f237b11f0a8 | 59 | strip[p % N].green = (uint8_t)(green * bright); |
aswild | 0:9f237b11f0a8 | 60 | strip[p % N].blue = (uint8_t)(blue * bright); |
aswild | 0:9f237b11f0a8 | 61 | } |
aswild | 0:9f237b11f0a8 | 62 | |
aswild | 0:9f237b11f0a8 | 63 | void NeoStrip::setPixels(int p, int n, const int *colors) |
aswild | 0:9f237b11f0a8 | 64 | { |
aswild | 0:9f237b11f0a8 | 65 | int r, g, b; |
aswild | 0:9f237b11f0a8 | 66 | for (int i = 0; i < n; i++) |
aswild | 0:9f237b11f0a8 | 67 | { |
aswild | 0:9f237b11f0a8 | 68 | r = (colors[i] & 0xFF0000) >> 16; |
aswild | 0:9f237b11f0a8 | 69 | g = (colors[i] & 0x00FF00) >>8; |
aswild | 0:9f237b11f0a8 | 70 | b = colors[i] & 0x0000FF; |
aswild | 0:9f237b11f0a8 | 71 | setPixel(p+i, r, g, b); |
aswild | 0:9f237b11f0a8 | 72 | } |
aswild | 0:9f237b11f0a8 | 73 | } |
aswild | 0:9f237b11f0a8 | 74 | |
aswild | 0:9f237b11f0a8 | 75 | void NeoStrip::clear() |
aswild | 0:9f237b11f0a8 | 76 | { |
aswild | 0:9f237b11f0a8 | 77 | for (int i = 0; i < N; i++) |
aswild | 0:9f237b11f0a8 | 78 | { |
aswild | 0:9f237b11f0a8 | 79 | strip[i].red = 0; |
aswild | 0:9f237b11f0a8 | 80 | strip[i].green = 0; |
aswild | 0:9f237b11f0a8 | 81 | strip[i].blue = 0; |
aswild | 0:9f237b11f0a8 | 82 | } |
aswild | 0:9f237b11f0a8 | 83 | } |
aswild | 0:9f237b11f0a8 | 84 | |
aswild | 0:9f237b11f0a8 | 85 | void NeoStrip::write() |
aswild | 0:9f237b11f0a8 | 86 | { |
aswild | 0:9f237b11f0a8 | 87 | __disable_irq(); // disable interrupts |
aswild | 0:9f237b11f0a8 | 88 | neo_out(strip, Nbytes); // output to the strip |
aswild | 0:9f237b11f0a8 | 89 | __enable_irq(); // enable interrupts |
aswild | 0:9f237b11f0a8 | 90 | wait_us(50); // wait 50us for the reset pulse |
aswild | 0:9f237b11f0a8 | 91 | } |
aswild | 0:9f237b11f0a8 | 92 | |
aswild | 0:9f237b11f0a8 | 93 |