src/rainbowio.py

   1 """
   2 `rainbowio` - Provides the `colorwheel()` function
   3 ===========================================================
   4 See `CircuitPython:rainbowio` in CircuitPython for more details.
   5 Not supported by all boards.
   6
   7 * Author(s): Kattni Rembor
   8 """
   9
  10
  11 def colorwheel(color_value):
  12     """
  13     A colorwheel. ``0`` and ``255`` are red, ``85`` is green, and ``170`` is blue, with the values
  14     between being the rest of the rainbow.
  15
  16     :param int color_value: 0-255 of color value to return
  17     :return: tuple of RGB values
  18     """
  19     if color_value < 0 or color_value > 255:
  20         return 0, 0, 0
  21     if color_value < 85:
  22         return 255 - color_value * 3, color_value * 3, 0
  23     if color_value < 170:
  24         color_value -= 85
  25         return 0, 255 - color_value * 3, color_value * 3
  26     color_value -= 170
  27     return color_value * 3, 0, 255 - color_value * 3