Import the library and examples

Library/tree.py

@@ -0,0 +1,108 @@
+from gpiozero import SPIDevice, SourceMixin
+from colorzero import Color, Hue
+from statistics import mean
+from time import sleep
+
+
+class Pixel:
+    def __init__(self, parent, index):
+        self.parent = parent
+        self.index = index
+
+    @property
+    def value(self):
+        return self.parent.value[self.index]
+
+    @value.setter
+    def value(self, value):
+        new_parent_value = list(self.parent.value)
+        new_parent_value[self.index] = value
+        self.parent.value = tuple(new_parent_value)
+
+    @property
+    def color(self):
+        return Color(*self.value)
+
+    @color.setter
+    def color(self, c):
+        r, g, b = c
+        self.value = (r, g, b)
+
+    def on(self):
+        self.value = (1, 1, 1)
+
+    def off(self):
+        self.value = (0, 0, 0)
+
+
+class RGBXmasTree(SourceMixin, SPIDevice):
+    def __init__(self, pixels=25, brightness=0.5, mosi_pin=12, clock_pin=25, *args, **kwargs):
+        super(RGBXmasTree, self).__init__(mosi_pin=mosi_pin, clock_pin=clock_pin, *args, **kwargs)
+        self._all = [Pixel(parent=self, index=i) for i in range(pixels)]
+        self._value = [(0, 0, 0)] * pixels
+        self.brightness = brightness
+        self.off()
+
+    def __len__(self):
+        return len(self._all)
+
+    def __getitem__(self, index):
+        return self._all[index]
+
+    def __iter__(self):
+        return iter(self._all)
+
+    @property
+    def color(self):
+        average_r = mean(pixel.color[0] for pixel in self)
+        average_g = mean(pixel.color[1] for pixel in self)
+        average_b = mean(pixel.color[2] for pixel in self)
+        return Color(average_r, average_g, average_b)
+
+    @color.setter
+    def color(self, c):
+        r, g, b = c
+        self.value = ((r, g, b),) * len(self)
+
+    @property
+    def brightness(self):
+        return self._brightness
+
+    @brightness.setter
+    def brightness(self, brightness):
+        max_brightness = 31
+        self._brightness_bits = int(brightness * max_brightness)
+        self._brightness = brightness
+        self.value = self.value
+
+    @property
+    def value(self):
+        return self._value
+
+    @value.setter
+    def value(self, value):
+        start_of_frame = [0]*4
+        end_of_frame = [0]*5
+                     # SSSBBBBB (start, brightness)
+        brightness = 0b11100000 | self._brightness_bits
+        pixels = [[int(255*v) for v in p] for p in value]
+        pixels = [[brightness, b, g, r] for r, g, b in pixels]
+        pixels = [i for p in pixels for i in p]
+        data = start_of_frame + pixels + end_of_frame
+        self._spi.transfer(data)
+        self._value = value
+
+    def on(self):
+        self.value = ((1, 1, 1),) * len(self)
+
+    def off(self):
+        self.value = ((0, 0, 0),) * len(self)
+
+    def close(self):
+        super(RGBXmasTree, self).close()
+
+
+if __name__ == '__main__':
+    tree = RGBXmasTree()
+
+    tree.on()