From 2709d98125dc5d06dcfd3be2b40245ed37b7cece Mon Sep 17 00:00:00 2001
From: Martyn <m@rtyn.berlin>
Date: Sun, 3 Jan 2021 10:46:39 +0000
Subject: [PATCH] moarpatterns (#1)

Make the comparison work

Allow color setting

types.... yeah... I know about them.

single-colour plasma, I hope

Reviewed-on: https://git.martyn.berlin/martyn/LEDController/pulls/1
Co-Authored-By: Martyn <m@rtyn.berlin>
Co-Committed-By: Martyn <m@rtyn.berlin>
---
 cmd/ledcontroller-server/main.go | 11 +++++++-
 internal/patterns/plasma.go      | 43 ++++++++++++++++++++++++++++++++
 internal/webserver/webserver.go  | 15 +++++++++++
 3 files changed, 68 insertions(+), 1 deletion(-)

diff --git a/cmd/ledcontroller-server/main.go b/cmd/ledcontroller-server/main.go
index a6fe692..284af68 100755
--- a/cmd/ledcontroller-server/main.go
+++ b/cmd/ledcontroller-server/main.go
@@ -32,6 +32,11 @@ var previousEffect queue.QueueItem
 
 var msDelay = 0
 
+func isColorSet(c RGBcolor) bool {
+	// Any color so long as it's NOT black!
+	return c[0] != 0 || c[1] != 0 || c[2] != 0
+}
+
 func foreverLoop() {
 	for /*ever*/ {
 		time.Sleep(time.Duration(msDelay) * time.Millisecond) //25fps
@@ -169,7 +174,11 @@ func main() {
 			case "line":
 				rearranged = remapping.SliceRearrange(PanelWidth, PanelHeight, true, remapping.XYGridToLinear(PanelWidth, PanelHeight, patterns.ZigZag(PanelWidth, PanelHeight)))
 			case "plasma":
-				rearranged = remapping.SliceRearrange(PanelWidth, PanelHeight, true, remapping.XYGridToLinear(PanelWidth, PanelHeight, patterns.PlasmaPanel(PanelWidth, PanelHeight, currentEffect.Speed)))
+				if isColorSet(currentEffect.SeedColour) {
+					rearranged = remapping.SliceRearrange(PanelWidth, PanelHeight, true, remapping.XYGridToLinear(PanelWidth, PanelHeight, patterns.PlasmaPanelSingleColor(PanelWidth, PanelHeight, currentEffect.Speed, currentEffect.SeedColour)))
+				} else {
+					rearranged = remapping.SliceRearrange(PanelWidth, PanelHeight, true, remapping.XYGridToLinear(PanelWidth, PanelHeight, patterns.PlasmaPanel(PanelWidth, PanelHeight, currentEffect.Speed)))
+				}
 			case "red":
 				rearranged = remapping.SliceRearrange(PanelWidth, PanelHeight, false, remapping.XYGridToLinear(PanelWidth, PanelHeight, patterns.RedPanel(PanelWidth, PanelHeight)))
 			case "random":
diff --git a/internal/patterns/plasma.go b/internal/patterns/plasma.go
index 4684d97..01d5896 100755
--- a/internal/patterns/plasma.go
+++ b/internal/patterns/plasma.go
@@ -112,3 +112,46 @@ func PlasmaPanel(w int, h int, speed uint16) [][]RGBcolor {
 	}
 	return grid
 }
+
+func singleColorPlasmaFromFloatVal(val float64, baseColor RGBcolor) (byte, byte, byte) {
+	var r byte
+	var g byte
+	var b byte
+	val = positiveModF(val)
+	r = byte(math.Trunc(float64(baseColor[0]) * val))
+	g = byte(math.Trunc(float64(baseColor[1]) * val))
+	b = byte(math.Trunc(float64(baseColor[2]) * val))
+	return r, g, b
+}
+
+func PlasmaPanelSingleColor(w int, h int, speed uint16, baseColor RGBcolor) [][]RGBcolor {
+	grid := make([][]RGBcolor, w)
+	for i := 0; i < w; i++ {
+		grid[i] = make([]RGBcolor, h)
+	}
+
+	palletteOffset -= 0.05 / 1.0
+	if palletteOffset < 0 {
+		palletteOffset += 1.0
+	}
+
+	offset := 0.5 // center offset
+	scaleW := math.Pi * 2.0 / float64(w)
+	scaleH := math.Pi * 2.0 / float64(h)
+
+	for y := 0; y < h; y++ {
+		for x := 0; x < w; x++ {
+			u := math.Cos((float64(x) + offset) * scaleW)
+			v := math.Cos((float64(y) + offset) * scaleH)
+			j := math.Cos(offset * scaleW) // 2D - No Z
+			e := (u + v + j + 3.0) / 6.0
+			r, g, b := singleColorPlasmaFromFloatVal(palletteOffset+e, baseColor)
+			var rgb [3]byte
+			rgb[0] = r
+			rgb[1] = g
+			rgb[2] = b
+			grid[x][y] = rgb
+		}
+	}
+	return grid
+}
diff --git a/internal/webserver/webserver.go b/internal/webserver/webserver.go
index 84d664c..6759639 100755
--- a/internal/webserver/webserver.go
+++ b/internal/webserver/webserver.go
@@ -55,6 +55,17 @@ func PatternHandler(response http.ResponseWriter, request *http.Request) {
 	} else {
 		e.Speed = 40
 	}
+	if vars["color"] != "" {
+		var c queue.RGBcolor
+		for cn, cv := range colornames.Map {
+			if cn == strings.ToLower(vars["color"]) {
+				c[0] = cv.R
+				c[1] = cv.G
+				c[2] = cv.B
+			}
+		}
+		e.SeedColour = c
+	}
 	if vars["override"] == "true" {
 		overrideFlag <- e
 	} else {
@@ -159,6 +170,10 @@ func HandleHTTP(queueChannel chan queue.QueueItem, Port int, overrideChannel cha
 	r.HandleFunc("/pattern/{pattern}/", PatternHandler)
 	r.HandleFunc("/pattern/{pattern}/{duration}", PatternHandler)
 	r.HandleFunc("/pattern/{pattern}/{duration}/{speed}", PatternHandler)
+	r.HandleFunc("/colorpattern/{color}/{pattern}/override={override}", PatternHandler)
+	r.HandleFunc("/colorpattern/{color}/{pattern}/", PatternHandler)
+	r.HandleFunc("/colorpattern/{color}/{pattern}/{duration}", PatternHandler)
+	r.HandleFunc("/colorpattern/{color}/{pattern}/{duration}/{speed}", PatternHandler)
 	r.HandleFunc("/colour/{name}", ColourHandler)
 	r.HandleFunc("/colour/{name}/override={override}", ColourHandler)
 	r.HandleFunc("/colour/{name}/{duration}", ColourHandler)