initial neopixel

Author: tannewt

neopixel.py

@@ -0,0 +1,38 @@
+# NeoPixel driver for MicroPython on ESP8266
+# MIT license; Copyright (c) 2016 Damien P. George
+
+import nativeio
+from neopixel_write import neopixel_write
+
+class NeoPixel:
+    def __init__(self, pin, n):
+        self.pin = nativeio.DigitalInOut(pin)
+        self.n = n
+        self.buf = bytearray(n * 3)
+        self.pin.switch_to_output()
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exception_type, exception_value, traceback):
+        self.pin.deinit()
+
+    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)