Add off() and let SetColorXX() handle slices of colors too.

Signed-off-by: David Greaves <[email protected]>

Author: lbt

library/rpi_ws281x/rpi_ws281x.py

@@ -224,8 +224,14 @@ def __len__(self):
 
         def setPixelColor(self, n, color):
             """Set LED at position n to the provided 24-bit color value (in RGB order).
+            If n is a slice then color can be a value which is repeated for all leds
+            or a slice of values which are applied to the leds.
             """
-            self.strip[self.first + n] = color
+            if isinstance(n, slice):
+                n = slice(n.start + self.first, n.stop + self.first, n.step)
+            else:
+                n = self.first + n
+            self.strip[n] = color
 
         def setPixelColorRGB(self, n, red, green, blue, white=0):
             """Set LED at position n to the provided red, green, and blue color.
@@ -269,6 +275,10 @@ def getPixelColorRGBW(self, n):
         def show(self):
             self.strip.show()
 
+    def off(self):
+        self.strip[self.first:self.last] = 0
+        self.strip.show()
+
 # Shim for back-compatibility
 class Adafruit_NeoPixel(PixelStrip):
     pass