From c9542e456738abb6d38802f3b23d1a57d903b916 Mon Sep 17 00:00:00 2001
From: Ben Nuttall <ben@bennuttall.com>
Date: Fri, 25 Nov 2016 21:29:15 +0000
Subject: [PATCH] Add XmasTree Board

---
 docs/api_boards.rst  |  8 +++++-
 gpiozero/__init__.py |  1 +
 gpiozero/boards.py   | 64 +++++++++++++++++++++++++++++++++++++++++++-
 3 files changed, 71 insertions(+), 2 deletions(-)

diff --git a/docs/api_boards.rst b/docs/api_boards.rst
index cec9e7e..98475e7 100644
--- a/docs/api_boards.rst
+++ b/docs/api_boards.rst
@@ -43,6 +43,13 @@ TrafficLights
     :inherited-members:
     :members:
 
+Xmas Tree
+=========
+
+.. autoclass:: XmasTree
+    :inherited-members:
+    :members:
+
 LedBorg
 =======
 
@@ -168,4 +175,3 @@ CompositeDevice
 
 .. autoclass:: CompositeDevice(\*args, _order=None, \*\*kwargs)
     :members:
-
diff --git a/gpiozero/__init__.py b/gpiozero/__init__.py
index e147943..e8bcd40 100644
--- a/gpiozero/__init__.py
+++ b/gpiozero/__init__.py
@@ -112,6 +112,7 @@ from .boards import (
     LEDBoard,
     LEDBarGraph,
     LedBorg,
+    XmasTree,
     PiLiter,
     PiLiterBarGraph,
     TrafficLights,
diff --git a/gpiozero/boards.py b/gpiozero/boards.py
index 7809fea..a962e64 100644
--- a/gpiozero/boards.py
+++ b/gpiozero/boards.py
@@ -550,6 +550,69 @@ class LEDBarGraph(LEDCollection):
             led.value = calc_value(index)
 
 
+class XmasTree(LEDBoard):
+    """
+    Extends :class:`LEDBoard` for `The Pi Hut's Xmas board`_: a Christmas tree
+    board with 3 red LEDs on each of 8 faces, and a white LED as a star on
+    top.
+
+    The Xmas Tree board pins are fixed and therefore there's no need to specify
+    them when constructing this class. The following example turns all the LEDs
+    on one at a time, ending in the star::
+
+        from gpiozero import XmasTree
+        from time import sleep
+
+        tree = XmasTree()
+
+        for light in tree:
+            light.on()
+            sleep(1)
+
+    The following example turns the star LED on and sets all the red LEDs to
+    flicker randomly::
+
+        from gpiozero import XmasTree
+        from gpiozero.tools import random_values
+        from signal import pause
+
+        tree = XmasTree(pwm=True)
+
+        tree.star.on()
+
+        for bauble in tree.baubles:
+            bauble.source_delay = 0.1
+            bauble.source = random_values()
+
+        pause()
+
+    :param bool pwm:
+        If ``True``, construct :class:`PWMLED` instances for each pin. If
+        ``False`` (the default), construct regular :class:`LED` instances.
+
+    :param bool initial_value:
+        If ``False`` (the default), all LEDs will be off initially. If
+        ``None``, each device will be left in whatever state the pin is found
+        in when configured for output (warning: this can be on). If ``True``,
+        the device will be switched on initially.
+
+    .. _The Pi Hut's Xmas board: https://thepihut.com/xmas
+    """
+
+    def __init__(self, pwm=False, initial_value=False):
+        pins = (4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 2)
+        super(XmasTree, self).__init__(*pins, pwm=pwm, initial_value=initial_value)
+
+    @property
+    def star(self):
+        return self[-1]
+
+    @property
+    def baubles(self):
+        baubles = self[:-1]
+        return {i+1: led for i, led in enumerate(baubles)}
+
+
 class LedBorg(RGBLED):
     """
     Extends :class:`RGBLED` for the `PiBorg LedBorg`_: an add-on board
@@ -1184,4 +1247,3 @@ class Energenie(SourceMixin, Device):
 
     def off(self):
         self.value = False
-