New effect collision

Signed-off-by: Paulchen-Panther <paulchen-panter@protonmail.com>

effects/collision.py

@@ -0,0 +1,60 @@
+# 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, math
+
+# 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)