diff options
Diffstat (limited to 'quantum/light_ws2812.c')
| -rwxr-xr-x | quantum/light_ws2812.c | 342 |
1 files changed, 0 insertions, 342 deletions
diff --git a/quantum/light_ws2812.c b/quantum/light_ws2812.c deleted file mode 100755 index 2506e3d8e..000000000 --- a/quantum/light_ws2812.c +++ /dev/null | |||
| @@ -1,342 +0,0 @@ | |||
| 1 | /* | ||
| 2 | * light weight WS2812 lib V2.0b | ||
| 3 | * | ||
| 4 | * Controls WS2811/WS2812/WS2812B RGB-LEDs | ||
| 5 | * Author: Tim (cpldcpu@gmail.com) | ||
| 6 | * | ||
| 7 | * Jan 18th, 2014 v2.0b Initial Version | ||
| 8 | * Nov 29th, 2015 v2.3 Added SK6812RGBW support | ||
| 9 | * | ||
| 10 | * This program is free software: you can redistribute it and/or modify | ||
| 11 | * it under the terms of the GNU General Public License as published by | ||
| 12 | * the Free Software Foundation, either version 2 of the License, or | ||
| 13 | * (at your option) any later version. | ||
| 14 | * | ||
| 15 | * This program is distributed in the hope that it will be useful, | ||
| 16 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| 17 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
| 18 | * GNU General Public License for more details. | ||
| 19 | * | ||
| 20 | * You should have received a copy of the GNU General Public License | ||
| 21 | * along with this program. If not, see <http://www.gnu.org/licenses/>. | ||
| 22 | */ | ||
| 23 | |||
| 24 | #include "light_ws2812.h" | ||
| 25 | #include <avr/interrupt.h> | ||
| 26 | #include <avr/io.h> | ||
| 27 | #include <util/delay.h> | ||
| 28 | #include "debug.h" | ||
| 29 | |||
| 30 | #ifdef RGBW_BB_TWI | ||
| 31 | |||
| 32 | // Port for the I2C | ||
| 33 | #define I2C_DDR DDRD | ||
| 34 | #define I2C_PIN PIND | ||
| 35 | #define I2C_PORT PORTD | ||
| 36 | |||
| 37 | // Pins to be used in the bit banging | ||
| 38 | #define I2C_CLK 0 | ||
| 39 | #define I2C_DAT 1 | ||
| 40 | |||
| 41 | #define I2C_DATA_HI()\ | ||
| 42 | I2C_DDR &= ~ (1 << I2C_DAT);\ | ||
| 43 | I2C_PORT |= (1 << I2C_DAT); | ||
| 44 | #define I2C_DATA_LO()\ | ||
| 45 | I2C_DDR |= (1 << I2C_DAT);\ | ||
| 46 | I2C_PORT &= ~ (1 << I2C_DAT); | ||
| 47 | |||
| 48 | #define I2C_CLOCK_HI()\ | ||
| 49 | I2C_DDR &= ~ (1 << I2C_CLK);\ | ||
| 50 | I2C_PORT |= (1 << I2C_CLK); | ||
| 51 | #define I2C_CLOCK_LO()\ | ||
| 52 | I2C_DDR |= (1 << I2C_CLK);\ | ||
| 53 | I2C_PORT &= ~ (1 << I2C_CLK); | ||
| 54 | |||
| 55 | #define I2C_DELAY 1 | ||
| 56 | |||
| 57 | void I2C_WriteBit(unsigned char c) | ||
| 58 | { | ||
| 59 | if (c > 0) | ||
| 60 | { | ||
| 61 | I2C_DATA_HI(); | ||
| 62 | } | ||
| 63 | else | ||
| 64 | { | ||
| 65 | I2C_DATA_LO(); | ||
| 66 | } | ||
| 67 | |||
| 68 | I2C_CLOCK_HI(); | ||
| 69 | _delay_us(I2C_DELAY); | ||
| 70 | |||
| 71 | I2C_CLOCK_LO(); | ||
| 72 | _delay_us(I2C_DELAY); | ||
| 73 | |||
| 74 | if (c > 0) | ||
| 75 | { | ||
| 76 | I2C_DATA_LO(); | ||
| 77 | } | ||
| 78 | |||
| 79 | _delay_us(I2C_DELAY); | ||
| 80 | } | ||
| 81 | |||
| 82 | // Inits bitbanging port, must be called before using the functions below | ||
| 83 | // | ||
| 84 | void I2C_Init(void) | ||
| 85 | { | ||
| 86 | I2C_PORT &= ~ ((1 << I2C_DAT) | (1 << I2C_CLK)); | ||
| 87 | |||
| 88 | I2C_CLOCK_HI(); | ||
| 89 | I2C_DATA_HI(); | ||
| 90 | |||
| 91 | _delay_us(I2C_DELAY); | ||
| 92 | } | ||
| 93 | |||
| 94 | // Send a START Condition | ||
| 95 | // | ||
| 96 | void I2C_Start(void) | ||
| 97 | { | ||
| 98 | // set both to high at the same time | ||
| 99 | I2C_DDR &= ~ ((1 << I2C_DAT) | (1 << I2C_CLK)); | ||
| 100 | _delay_us(I2C_DELAY); | ||
| 101 | |||
| 102 | I2C_DATA_LO(); | ||
| 103 | _delay_us(I2C_DELAY); | ||
| 104 | |||
| 105 | I2C_CLOCK_LO(); | ||
| 106 | _delay_us(I2C_DELAY); | ||
| 107 | } | ||
| 108 | |||
| 109 | // Send a STOP Condition | ||
| 110 | // | ||
| 111 | void I2C_Stop(void) | ||
| 112 | { | ||
| 113 | I2C_CLOCK_HI(); | ||
| 114 | _delay_us(I2C_DELAY); | ||
| 115 | |||
| 116 | I2C_DATA_HI(); | ||
| 117 | _delay_us(I2C_DELAY); | ||
| 118 | } | ||
| 119 | |||
| 120 | // write a byte to the I2C slave device | ||
| 121 | // | ||
| 122 | unsigned char I2C_Write(unsigned char c) | ||
| 123 | { | ||
| 124 | for (char i = 0; i < 8; i++) | ||
| 125 | { | ||
| 126 | I2C_WriteBit(c & 128); | ||
| 127 | |||
| 128 | c <<= 1; | ||
| 129 | } | ||
| 130 | |||
| 131 | |||
| 132 | I2C_WriteBit(0); | ||
| 133 | _delay_us(I2C_DELAY); | ||
| 134 | _delay_us(I2C_DELAY); | ||
| 135 | |||
| 136 | // _delay_us(I2C_DELAY); | ||
| 137 | //return I2C_ReadBit(); | ||
| 138 | return 0; | ||
| 139 | } | ||
| 140 | |||
| 141 | |||
| 142 | #endif | ||
| 143 | |||
| 144 | // Setleds for standard RGB | ||
| 145 | void inline ws2812_setleds(LED_TYPE *ledarray, uint16_t leds) | ||
| 146 | { | ||
| 147 | // ws2812_setleds_pin(ledarray,leds, _BV(ws2812_pin)); | ||
| 148 | ws2812_setleds_pin(ledarray,leds, _BV(RGB_DI_PIN & 0xF)); | ||
| 149 | } | ||
| 150 | |||
| 151 | void inline ws2812_setleds_pin(LED_TYPE *ledarray, uint16_t leds, uint8_t pinmask) | ||
| 152 | { | ||
| 153 | // ws2812_DDRREG |= pinmask; // Enable DDR | ||
| 154 | // new universal format (DDR) | ||
| 155 | _SFR_IO8((RGB_DI_PIN >> 4) + 1) |= pinmask; | ||
| 156 | |||
| 157 | ws2812_sendarray_mask((uint8_t*)ledarray,leds+leds+leds,pinmask); | ||
| 158 | _delay_us(50); | ||
| 159 | } | ||
| 160 | |||
| 161 | // Setleds for SK6812RGBW | ||
| 162 | void inline ws2812_setleds_rgbw(LED_TYPE *ledarray, uint16_t leds) | ||
| 163 | { | ||
| 164 | |||
| 165 | #ifdef RGBW_BB_TWI | ||
| 166 | uint8_t sreg_prev, twcr_prev; | ||
| 167 | sreg_prev=SREG; | ||
| 168 | twcr_prev=TWCR; | ||
| 169 | cli(); | ||
| 170 | TWCR &= ~(1<<TWEN); | ||
| 171 | I2C_Init(); | ||
| 172 | I2C_Start(); | ||
| 173 | I2C_Write(0x84); | ||
| 174 | uint16_t datlen = leds<<2; | ||
| 175 | uint8_t curbyte; | ||
| 176 | uint8_t * data = (uint8_t*)ledarray; | ||
| 177 | while (datlen--) { | ||
| 178 | curbyte=*data++; | ||
| 179 | I2C_Write(curbyte); | ||
| 180 | } | ||
| 181 | I2C_Stop(); | ||
| 182 | SREG=sreg_prev; | ||
| 183 | TWCR=twcr_prev; | ||
| 184 | #endif | ||
| 185 | |||
| 186 | |||
| 187 | // ws2812_DDRREG |= _BV(ws2812_pin); // Enable DDR | ||
| 188 | // new universal format (DDR) | ||
| 189 | _SFR_IO8((RGB_DI_PIN >> 4) + 1) |= _BV(RGB_DI_PIN & 0xF); | ||
| 190 | |||
| 191 | ws2812_sendarray_mask((uint8_t*)ledarray,leds<<2,_BV(RGB_DI_PIN & 0xF)); | ||
| 192 | |||
| 193 | |||
| 194 | #ifndef RGBW_BB_TWI | ||
| 195 | _delay_us(80); | ||
| 196 | #endif | ||
| 197 | } | ||
| 198 | |||
| 199 | void ws2812_sendarray(uint8_t *data,uint16_t datlen) | ||
| 200 | { | ||
| 201 | ws2812_sendarray_mask(data,datlen,_BV(RGB_DI_PIN & 0xF)); | ||
| 202 | } | ||
| 203 | |||
| 204 | /* | ||
| 205 | This routine writes an array of bytes with RGB values to the Dataout pin | ||
| 206 | using the fast 800kHz clockless WS2811/2812 protocol. | ||
| 207 | */ | ||
| 208 | |||
| 209 | // Timing in ns | ||
| 210 | #define w_zeropulse 350 | ||
| 211 | #define w_onepulse 900 | ||
| 212 | #define w_totalperiod 1250 | ||
| 213 | |||
| 214 | // Fixed cycles used by the inner loop | ||
| 215 | #define w_fixedlow 2 | ||
| 216 | #define w_fixedhigh 4 | ||
| 217 | #define w_fixedtotal 8 | ||
| 218 | |||
| 219 | // Insert NOPs to match the timing, if possible | ||
| 220 | #define w_zerocycles (((F_CPU/1000)*w_zeropulse )/1000000) | ||
| 221 | #define w_onecycles (((F_CPU/1000)*w_onepulse +500000)/1000000) | ||
| 222 | #define w_totalcycles (((F_CPU/1000)*w_totalperiod +500000)/1000000) | ||
| 223 | |||
| 224 | // w1 - nops between rising edge and falling edge - low | ||
| 225 | #define w1 (w_zerocycles-w_fixedlow) | ||
| 226 | // w2 nops between fe low and fe high | ||
| 227 | #define w2 (w_onecycles-w_fixedhigh-w1) | ||
| 228 | // w3 nops to complete loop | ||
| 229 | #define w3 (w_totalcycles-w_fixedtotal-w1-w2) | ||
| 230 | |||
| 231 | #if w1>0 | ||
| 232 | #define w1_nops w1 | ||
| 233 | #else | ||
| 234 | #define w1_nops 0 | ||
| 235 | #endif | ||
| 236 | |||
| 237 | // The only critical timing parameter is the minimum pulse length of the "0" | ||
| 238 | // Warn or throw error if this timing can not be met with current F_CPU settings. | ||
| 239 | #define w_lowtime ((w1_nops+w_fixedlow)*1000000)/(F_CPU/1000) | ||
| 240 | #if w_lowtime>550 | ||
| 241 | #error "Light_ws2812: Sorry, the clock speed is too low. Did you set F_CPU correctly?" | ||
| 242 | #elif w_lowtime>450 | ||
| 243 | #warning "Light_ws2812: The timing is critical and may only work on WS2812B, not on WS2812(S)." | ||
| 244 | #warning "Please consider a higher clockspeed, if possible" | ||
| 245 | #endif | ||
| 246 | |||
| 247 | #if w2>0 | ||
| 248 | #define w2_nops w2 | ||
| 249 | #else | ||
| 250 | #define w2_nops 0 | ||
| 251 | #endif | ||
| 252 | |||
| 253 | #if w3>0 | ||
| 254 | #define w3_nops w3 | ||
| 255 | #else | ||
| 256 | #define w3_nops 0 | ||
| 257 | #endif | ||
| 258 | |||
| 259 | #define w_nop1 "nop \n\t" | ||
| 260 | #define w_nop2 "rjmp .+0 \n\t" | ||
| 261 | #define w_nop4 w_nop2 w_nop2 | ||
| 262 | #define w_nop8 w_nop4 w_nop4 | ||
| 263 | #define w_nop16 w_nop8 w_nop8 | ||
| 264 | |||
| 265 | void inline ws2812_sendarray_mask(uint8_t *data,uint16_t datlen,uint8_t maskhi) | ||
| 266 | { | ||
| 267 | uint8_t curbyte,ctr,masklo; | ||
| 268 | uint8_t sreg_prev; | ||
| 269 | |||
| 270 | // masklo =~maskhi&ws2812_PORTREG; | ||
| 271 | // maskhi |= ws2812_PORTREG; | ||
| 272 | masklo =~maskhi&_SFR_IO8((RGB_DI_PIN >> 4) + 2); | ||
| 273 | maskhi |= _SFR_IO8((RGB_DI_PIN >> 4) + 2); | ||
| 274 | sreg_prev=SREG; | ||
| 275 | cli(); | ||
| 276 | |||
| 277 | while (datlen--) { | ||
| 278 | curbyte=(*data++); | ||
| 279 | |||
| 280 | asm volatile( | ||
| 281 | " ldi %0,8 \n\t" | ||
| 282 | "loop%=: \n\t" | ||
| 283 | " out %2,%3 \n\t" // '1' [01] '0' [01] - re | ||
| 284 | #if (w1_nops&1) | ||
| 285 | w_nop1 | ||
| 286 | #endif | ||
| 287 | #if (w1_nops&2) | ||
| 288 | w_nop2 | ||
| 289 | #endif | ||
| 290 | #if (w1_nops&4) | ||
| 291 | w_nop4 | ||
| 292 | #endif | ||
| 293 | #if (w1_nops&8) | ||
| 294 | w_nop8 | ||
| 295 | #endif | ||
| 296 | #if (w1_nops&16) | ||
| 297 | w_nop16 | ||
| 298 | #endif | ||
| 299 | " sbrs %1,7 \n\t" // '1' [03] '0' [02] | ||
| 300 | " out %2,%4 \n\t" // '1' [--] '0' [03] - fe-low | ||
| 301 | " lsl %1 \n\t" // '1' [04] '0' [04] | ||
| 302 | #if (w2_nops&1) | ||
| 303 | w_nop1 | ||
| 304 | #endif | ||
| 305 | #if (w2_nops&2) | ||
| 306 | w_nop2 | ||
| 307 | #endif | ||
| 308 | #if (w2_nops&4) | ||
| 309 | w_nop4 | ||
| 310 | #endif | ||
| 311 | #if (w2_nops&8) | ||
| 312 | w_nop8 | ||
| 313 | #endif | ||
| 314 | #if (w2_nops&16) | ||
| 315 | w_nop16 | ||
| 316 | #endif | ||
| 317 | " out %2,%4 \n\t" // '1' [+1] '0' [+1] - fe-high | ||
| 318 | #if (w3_nops&1) | ||
| 319 | w_nop1 | ||
| 320 | #endif | ||
| 321 | #if (w3_nops&2) | ||
| 322 | w_nop2 | ||
| 323 | #endif | ||
| 324 | #if (w3_nops&4) | ||
| 325 | w_nop4 | ||
| 326 | #endif | ||
| 327 | #if (w3_nops&8) | ||
| 328 | w_nop8 | ||
| 329 | #endif | ||
| 330 | #if (w3_nops&16) | ||
| 331 | w_nop16 | ||
| 332 | #endif | ||
| 333 | |||
| 334 | " dec %0 \n\t" // '1' [+2] '0' [+2] | ||
| 335 | " brne loop%=\n\t" // '1' [+3] '0' [+4] | ||
| 336 | : "=&d" (ctr) | ||
| 337 | : "r" (curbyte), "I" (_SFR_IO_ADDR(_SFR_IO8((RGB_DI_PIN >> 4) + 2))), "r" (maskhi), "r" (masklo) | ||
| 338 | ); | ||
| 339 | } | ||
| 340 | |||
| 341 | SREG=sreg_prev; | ||
| 342 | } | ||
