From 2709d98125dc5d06dcfd3be2b40245ed37b7cece Mon Sep 17 00:00:00 2001 From: Martyn 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 Co-Committed-By: Martyn --- 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)