mirror of
https://github.com/digistump/DigistumpArduino.git
synced 2025-04-28 15:49:02 -07:00
155 lines
5.5 KiB
C
155 lines
5.5 KiB
C
/*
|
|
* 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 */
|