mirror of
https://github.com/digistump/DigistumpArduino.git
synced 2025-09-17 09:22:28 -07:00
switch to setup for Arduino Boards Manager
This commit is contained in:
Erik Tylek Kettenburg
154
digistump-avr/libraries/WS2811/WS2811.h
Normal file
154
digistump-avr/libraries/WS2811/WS2811.h
Normal file
@@ -0,0 +1,154 @@
|
||||
/*
|
||||
* Copyright 2012 Alan Burlison, alan@bleaklow.com. All rights reserved.
|
||||
* Use is subject to license terms.
|
||||
*/
|
||||
|
||||
/*
|
||||
* WS2811 RGB LED driver.
|
||||
*/
|
||||
|
||||
#include <Arduino.h>
|
||||
|
||||
#ifndef WS2811_h
|
||||
#define WS2811_h
|
||||
|
||||
// RGB value structure.
|
||||
typedef struct __attribute__ ((__packed__)) {
|
||||
uint8_t r;
|
||||
uint8_t g;
|
||||
uint8_t b;
|
||||
} RGB_t;
|
||||
|
||||
#ifndef ARRAYLEN
|
||||
#define ARRAYLEN(A) (sizeof(A) / sizeof(A[0]))
|
||||
#endif
|
||||
|
||||
/*
|
||||
* Inline asm macro to output 24-bit RGB value in (G,R,B) order, MSBit first.
|
||||
* 0 bits are 250ns hi, 1000ns lo, 1 bits are 1000ns hi, 250ns lo.
|
||||
* r18 = red byte to be output
|
||||
* r19 = green byte to be output
|
||||
* r20 = blue byte to be output
|
||||
* r26 = saved SREG
|
||||
* r27 = inner loop counter
|
||||
*/
|
||||
#define WS2811(PORT, PIN, RGB, LEN) \
|
||||
asm volatile( \
|
||||
/* initialise */ \
|
||||
" cp %A[len], r1 ; check len > 0, return immediately if it is\n" \
|
||||
" cpc %B[len], r1\n" \
|
||||
" brne 1f\n" \
|
||||
" rjmp 16f\n" \
|
||||
"1: ld r18, Z+ ; load in first red byte to be output\n" \
|
||||
" ld r19, Z+ ; load in first green byte to be output\n" \
|
||||
" ld r20, Z+ ; load in first blue byte to be output\n" \
|
||||
" ldi r27, 8 ; load inner loop counter\n" \
|
||||
" in r26, __SREG__ ; timing-critical, so no interrupts\n" \
|
||||
" cli\n" \
|
||||
/* green - loop over 8 bits */ \
|
||||
"2: sbi %[port], %[pin] ; pin lo -> hi\n" \
|
||||
" sbrc r19, 7 ; test hi bit clear\n" \
|
||||
" rjmp 3f ; true, skip pin hi -> lo\n" \
|
||||
" cbi %[port], %[pin] ; false, pin hi -> lo\n" \
|
||||
"3: sbrc r19, 7 ; equalise delay of both code paths\n" \
|
||||
" rjmp 4f\n" \
|
||||
"4: nop ; pulse timing delay\n" \
|
||||
" nop\n" \
|
||||
" nop\n" \
|
||||
" nop\n" \
|
||||
" nop\n" \
|
||||
" nop\n" \
|
||||
" lsl r19 ; shift to next bit\n" \
|
||||
" dec r27 ; decrement loop counter\n" \
|
||||
" cbi %[port], %[pin] ; pin hi -> lo\n" \
|
||||
" brne 2b\n ; loop if required\n" \
|
||||
" ldi r27, 7 ; reload inner loop counter\n" \
|
||||
/* red - loop over first 7 bits */ \
|
||||
"5: sbi %[port], %[pin] ; pin lo -> hi\n" \
|
||||
" sbrc r18, 7 ; test hi bit clear\n" \
|
||||
" rjmp 6f ; true, skip pin hi -> lo\n" \
|
||||
" cbi %[port], %[pin] ; false, pin hi -> lo\n" \
|
||||
"6: sbrc r18, 7 ; equalise delay of both code paths\n" \
|
||||
" rjmp 7f\n" \
|
||||
"7: nop ; pulse timing delay\n" \
|
||||
" nop\n" \
|
||||
" nop\n" \
|
||||
" nop\n" \
|
||||
" nop\n" \
|
||||
" nop\n" \
|
||||
" lsl r18 ; shift to next bit\n" \
|
||||
" dec r27 ; decrement inner loop counter\n" \
|
||||
" cbi %[port], %[pin] ; pin hi -> lo\n" \
|
||||
" brne 5b ; inner loop, if required\n" \
|
||||
" nop ; equalise delay of both code paths\n" \
|
||||
/* red, 8th bit - output & fetch next values */ \
|
||||
" sbi %[port], %[pin] ; pin lo -> hi\n" \
|
||||
" sbrc r18, 7 ; test hi bit clear\n" \
|
||||
" rjmp 8f ; true, skip pin hi -> lo\n" \
|
||||
" cbi %[port], %[pin] ; false, pin hi -> lo\n" \
|
||||
"8: sbrc r18, 7 ; equalise delay of both code paths\n" \
|
||||
" rjmp 9f\n" \
|
||||
"9: nop ; pulse timing delay\n" \
|
||||
" nop\n" \
|
||||
" nop\n" \
|
||||
" ld r18, Z+ ; load next red byte\n" \
|
||||
" ld r19, Z+ ; load next green byte\n" \
|
||||
" ldi r27, 7 ; reload inner loop counter\n" \
|
||||
" cbi %[port], %[pin] ; pin hi -> lo\n" \
|
||||
" nop ; pulse timing delay\n" \
|
||||
" nop\n" \
|
||||
/* blue - loop over first 7 bits */ \
|
||||
"10: sbi %[port], %[pin] ; pin lo -> hi\n" \
|
||||
" sbrc r20, 7 ; test hi bit clear\n" \
|
||||
" rjmp 11f ; true, skip pin hi -> lo\n" \
|
||||
" cbi %[port], %[pin] ; false, pin hi -> lo\n" \
|
||||
"11: sbrc r20, 7 ; equalise delay of both code paths\n" \
|
||||
" rjmp 12f\n" \
|
||||
"12: nop ; pulse timing delay\n" \
|
||||
" nop\n" \
|
||||
" nop\n" \
|
||||
" nop\n" \
|
||||
" nop\n" \
|
||||
" nop\n" \
|
||||
" lsl r20 ; shift to next bit\n" \
|
||||
" dec r27 ; decrement inner loop counter\n" \
|
||||
" cbi %[port], %[pin] ; pin hi -> lo\n" \
|
||||
" brne 10b ; inner loop, if required\n" \
|
||||
" nop ; equalise delay of both code paths\n" \
|
||||
/* blue, 8th bit - output & handle outer loop */ \
|
||||
" sbi %[port], %[pin] ; pin lo -> hi\n" \
|
||||
" sbrc r20, 7 ; test hi bit clear\n" \
|
||||
" rjmp 13f ; true, skip pin hi -> lo\n" \
|
||||
" cbi %[port], %[pin] ; false, pin hi -> lo\n" \
|
||||
"13: sbrc r20, 7 ; equalise delay of both code paths\n" \
|
||||
" rjmp 14f\n" \
|
||||
"14: nop ; pulse timing delay\n" \
|
||||
" nop\n" \
|
||||
" ldi r27, 8 ; reload inner loop counter\n" \
|
||||
" sbiw %A[len], 1 ; decrement outer loop counter\n" \
|
||||
" breq 15f ; exit outer loop if zero\n" \
|
||||
" ld r20, Z+ ; load in next blue byte\n" \
|
||||
" cbi %[port], %[pin] ; pin hi -> lo\n" \
|
||||
" rjmp 2b ; outer loop, if required\n" \
|
||||
"15: nop ; pulse timing delay\n" \
|
||||
" cbi %[port], %[pin] ; pin hi -> lo\n" \
|
||||
" nop ; pulse timing delay\n" \
|
||||
" nop\n" \
|
||||
" out __SREG__, r26 ; reenable interrupts\n" \
|
||||
"16:\n" \
|
||||
: \
|
||||
: [rgb] "z" (RGB), \
|
||||
[len] "w" (LEN), \
|
||||
[port] "I" (_SFR_IO_ADDR(PORT)), \
|
||||
[pin] "I" (PIN) \
|
||||
: "r18", "r19", "r20", "r26", "r27", "cc", "memory" \
|
||||
)
|
||||
|
||||
/*
|
||||
* Define a C function to wrap the inline WS2811 macro for a given port and pin.
|
||||
*/
|
||||
#define DEFINE_WS2811_FN(NAME, PORT, PIN) \
|
||||
extern void NAME(const RGB_t *rgb, uint16_t len) __attribute__((noinline)); \
|
||||
void NAME(const RGB_t *rgb, uint16_t len) { WS2811(PORT, PIN, rgb, len); }
|
||||
|
||||
#endif /* WS2811_h */
|
||||
@@ -0,0 +1,24 @@
|
||||
#include <WS2811.h>
|
||||
DEFINE_WS2811_FN(WS2811RGB, PORTB, 1)
|
||||
RGB_t rgb[1]; //1 for 1 pixel
|
||||
|
||||
void setup() {
|
||||
|
||||
pinMode(1,OUTPUT);
|
||||
|
||||
}
|
||||
|
||||
void loop() {
|
||||
setPixel(0,255,0,0); //set first pixel (zero indexed) to red
|
||||
updatePixels(); //show the change
|
||||
}
|
||||
|
||||
void setPixel(i,r,g,b){
|
||||
rgb[r].r=r;
|
||||
rgb[g].g=g;
|
||||
rgb[b].b=b;
|
||||
}
|
||||
|
||||
void updatePixels(){
|
||||
WS2811RGB(rgb, ARRAYLEN(rgb));
|
||||
}
|
||||
Reference in New Issue
Block a user