hyperion.ng/effects/collision.py

# Two projectiles are sent from random positions and collide with each other
# Template from https://github.com/nickpesce/lit/blob/master/lit/effects/collision.py
import hyperion, time, colorsys, random

# Get parameters
sleepTime     = max(0.02, float(hyperion.args.get('speed', 100))/1000.0)
trailLength   = max(3, int(hyperion.args.get('trailLength', 5)))
explodeRadius = int(hyperion.args.get('explodeRadius', 8))

# Create additional variables
increment = None
projectiles = []

# Initialize the led data
ledData = bytearray()
for i in range(hyperion.ledCount):
	ledData += bytearray((0,0,0))

# Start the write data loop
while not hyperion.abort():
	if (len(projectiles) != 2):
		projectiles = [ [0, 1, random.uniform(0.0, 1.0)], [hyperion.ledCount-1, -1, random.uniform(0.0, 1.0)] ]
		increment = -random.randint(0, hyperion.ledCount-1) if random.choice([True, False]) else random.randint(0, hyperion.ledCount-1)

	ledDataBuf = ledData[:]
	for i, v in enumerate(projectiles):
		projectiles[i][0] = projectiles[i][0]+projectiles[i][1]
		for t in range(0, trailLength):
			pixel = v[0] - v[1]*t
			if pixel + 2 < 0:
				pixel += hyperion.ledCount
			if pixel + 2 > hyperion.ledCount-1:
				pixel -= hyperion.ledCount-1
			rgb = colorsys.hsv_to_rgb(v[2], 1, (trailLength - 1.0*t)/trailLength)
			ledDataBuf[3*pixel    ] = int(255*rgb[0])
			ledDataBuf[3*pixel + 1] = int(255*rgb[1])
			ledDataBuf[3*pixel + 2] = int(255*rgb[2])

	hyperion.setColor(ledDataBuf[-increment:] + ledDataBuf[:-increment])

	for i1, p1 in enumerate(projectiles):
		for i2, p2 in enumerate(projectiles):
			if (p1 is not p2):
				prev1 = p1[0] - p1[1]
				prev2 = p2[0] - p2[1]
				if (prev1 - prev2 < 0) != (p1[0] - p2[0] < 0):
					for d in range(0, explodeRadius):
						for pixel in range(p1[0] - d, p1[0] + d):
							rgb = colorsys.hsv_to_rgb(random.choice([p1[2], p2[2]]), 1, (1.0 * explodeRadius - d) / explodeRadius)
							ledDataBuf[3*pixel    ] = int(255*rgb[0])
							ledDataBuf[3*pixel + 1] = int(255*rgb[1])
							ledDataBuf[3*pixel + 2] = int(255*rgb[2])

						hyperion.setColor(ledDataBuf[-increment:] + ledDataBuf[:-increment])
						time.sleep(sleepTime)

					projectiles.remove(p1)
					projectiles.remove(p2)

	time.sleep(sleepTime)