Skip to content

Commit f46401f

Browse files
committed
micropython/drivers: Move "neopixel" led driver from main repo.
Signed-off-by: Jim Mussared <[email protected]>
1 parent 75d129b commit f46401f

File tree

2 files changed

+51
-0
lines changed

2 files changed

+51
-0
lines changed
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
module("neopixel.py", opt=3)
Lines changed: 50 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,50 @@
1+
# NeoPixel driver for MicroPython
2+
# MIT license; Copyright (c) 2016 Damien P. George, 2021 Jim Mussared
3+
4+
from machine import bitstream
5+
6+
7+
class NeoPixel:
8+
# G R B W
9+
ORDER = (1, 0, 2, 3)
10+
11+
def __init__(self, pin, n, bpp=3, timing=1):
12+
self.pin = pin
13+
self.n = n
14+
self.bpp = bpp
15+
self.buf = bytearray(n * bpp)
16+
self.pin.init(pin.OUT)
17+
# Timing arg can either be 1 for 800kHz or 0 for 400kHz,
18+
# or a user-specified timing ns tuple (high_0, low_0, high_1, low_1).
19+
self.timing = (
20+
((400, 850, 800, 450) if timing else (800, 1700, 1600, 900))
21+
if isinstance(timing, int)
22+
else timing
23+
)
24+
25+
def __len__(self):
26+
return self.n
27+
28+
def __setitem__(self, i, v):
29+
offset = i * self.bpp
30+
for i in range(self.bpp):
31+
self.buf[offset + self.ORDER[i]] = v[i]
32+
33+
def __getitem__(self, i):
34+
offset = i * self.bpp
35+
return tuple(self.buf[offset + self.ORDER[i]] for i in range(self.bpp))
36+
37+
def fill(self, v):
38+
b = self.buf
39+
l = len(self.buf)
40+
bpp = self.bpp
41+
for i in range(bpp):
42+
c = v[i]
43+
j = self.ORDER[i]
44+
while j < l:
45+
b[j] = c
46+
j += bpp
47+
48+
def write(self):
49+
# BITSTREAM_TYPE_HIGH_LOW = 0
50+
bitstream(self.pin, 0, self.timing, self.buf)

0 commit comments

Comments
 (0)