adafruit-mirror/Adafruit_CircuitPython_LED_Animation
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add more sparkles

Browse source
This commit is contained in:
Roy Hooper 2019-12-16 14:49:54 -05:00
parent f1b1fc76a4
commit 437c4476d0
1 changed files with 11 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
adafruit_led_animation/animation.py
View file

@ -329,12 +329,15 @@ class Sparkle(Animation):
:param pixel_object: The initialised LED object. :param pixel_object: The initialised LED object.
:param int speed: Animation speed in seconds, e.g. ``0.1``. :param int speed: Animation speed in seconds, e.g. ``0.1``.
:param color: Animation color in ``(r, g, b)`` tuple, or ``0x000000`` hex format. :param color: Animation color in ``(r, g, b)`` tuple, or ``0x000000`` hex format.
:param num_sparkles: Number of sparkles to generate per animation cycle.
""" """
def __init__(self, pixel_object, speed, color, name=None): # pylint: disable=too-many-arguments
def __init__(self, pixel_object, speed, color, num_sparkles=1, name=None):
if len(pixel_object) < 2: if len(pixel_object) < 2:
raise ValueError("Sparkle needs at least 2 pixels") raise ValueError("Sparkle needs at least 2 pixels")
self._half_color = None self._half_color = None
self._dim_color = None self._dim_color = None
self._num_sparkles = num_sparkles
super(Sparkle, self).__init__(pixel_object, speed, color, name=name) super(Sparkle, self).__init__(pixel_object, speed, color, name=name)
def _recompute_color(self, color): def _recompute_color(self, color):
@ -349,11 +352,14 @@ class Sparkle(Animation):
self._dim_color = dim_color self._dim_color = dim_color
def draw(self): def draw(self):
pixel = random.randint(0, (len(self.pixel_object) - 2)) pixels = [random.randint(0, (len(self.pixel_object) - 2))
self.pixel_object[pixel] = self._color for n in range(self._num_sparkles)]
for pixel in pixels:
self.pixel_object[pixel] = self._color
self.show() self.show()
self.pixel_object[pixel] = self._half_color for pixel in pixels:
self.pixel_object[pixel + 1] = self._dim_color self.pixel_object[pixel] = self._half_color
self.pixel_object[pixel + 1] = self._dim_color
self.show() self.show()