diff --git a/effects/snake.json b/effects/snake.json new file mode 100644 index 00000000..2c7ba395 --- /dev/null +++ b/effects/snake.json @@ -0,0 +1,10 @@ +{ + "name" : "Snake", + "script" : "snake.py", + "args" : + { + "rotation-time" : 10.0, + "color" : [255, 0, 0], + "percentage" : 25 + } +} diff --git a/effects/snake.py b/effects/snake.py new file mode 100644 index 00000000..0ee622a6 --- /dev/null +++ b/effects/snake.py @@ -0,0 +1,39 @@ +import hyperion +import time +import colorsys + +# Get the parameters +rotationTime = float(hyperion.args.get('rotation-time', 10.0)) +color = hyperion.args.get('color', (255,0,0)) +percentage = int(hyperion.args.get('percentage', 10)) + +# Check parameters +rotationTime = max(0.1, rotationTime) +percentage = max(1, min(percentage, 100)) + +# Initialize the led data +factor = float(percentage)/100.0 +snakeLeds = int(hyperion.ledCount*factor) +ledData = bytearray() + +for i in range(hyperion.ledCount-snakeLeds): + ledData += bytearray((0, 0, 0)) + +for i in range(1, snakeLeds+1): + hsv = colorsys.rgb_to_hsv(float(color[0])/float(255), float(color[1])/float(255), float(color[2])/float(255)) + rgb = colorsys.hsv_to_rgb(hsv[0], hsv[1], hsv[2]/float(snakeLeds+1-i)) + ledData += bytearray((int(rgb[0]*255), int(rgb[1]*255), int(rgb[2]*255))) + +# Calculate the sleep time and rotation increment +increment = 3 +sleepTime = rotationTime / hyperion.ledCount +while sleepTime < 0.05: + increment *= 2 + sleepTime *= 2 +increment %= hyperion.ledCount + +# Start the write data loop +while not hyperion.abort(): + hyperion.setColor(ledData) + ledData = ledData[-increment:] + ledData[:-increment] + time.sleep(sleepTime)