32 lines
840 B
Python
32 lines
840 B
Python
# NeoPixel driver for MicroPython on ESP8266
|
|
# MIT license; Copyright (c) 2016 Damien P. George
|
|
|
|
from esp import neopixel_write
|
|
|
|
class NeoPixel:
|
|
def __init__(self, pin, n):
|
|
self.pin = pin
|
|
self.n = n
|
|
self.buf = bytearray(n * 3)
|
|
self.pin.init(pin.OUT)
|
|
|
|
def __setitem__(self, index, val):
|
|
r, g, b = val
|
|
self.buf[index * 3] = g
|
|
self.buf[index * 3 + 1] = r
|
|
self.buf[index * 3 + 2] = b
|
|
|
|
def __getitem__(self, index):
|
|
i = index * 3
|
|
return self.buf[i + 1], self.buf[i], self.buf[i + 2]
|
|
|
|
def fill(self, color):
|
|
r, g, b = color
|
|
for i in range(len(self.buf) / 3):
|
|
self.buf[i * 3] = g
|
|
self.buf[i * 3 + 1] = r
|
|
self.buf[i * 3 + 2] = b
|
|
|
|
def write(self):
|
|
neopixel_write(self.pin, self.buf, True)
|