40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
|
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)
|