LEDController/cmd/ledcontroller-server/main.go

package main

import (
	"fmt"
	"log"
	"math"
	"os"
	"strconv"
	"time"

	patterns "git.martyn.berlin/martyn/LEDController/internal/patterns"
	queue "git.martyn.berlin/martyn/LEDController/internal/queue"
	remapping "git.martyn.berlin/martyn/LEDController/internal/remapping"
	webserver "git.martyn.berlin/martyn/LEDController/internal/webserver"

	"github.com/Hundemeier/go-sacn/sacn"
)

type RGBcolor = [3]byte

var currentFrame [99][512]byte //TODO make dynamic
var channels [99]chan<- [512]byte
var sema = make(chan struct{}, 1) // a binary semaphore guarding currentFrame

var currentEffect queue.QueueItem
var globalEffectChannel = make(chan queue.QueueItem, 1024)
var universeCount int = 0

var overrideEffect queue.QueueItem
var overrideFlag = make(chan queue.QueueItem, 1)
var previousEffect queue.QueueItem

var msDelay = 0

func foreverLoop() {
	for /*ever*/ {
		time.Sleep(time.Duration(msDelay) * time.Millisecond) //25fps
		for u := 0; u < universeCount; u++ {
			sema <- struct{}{} // acquire token
			channels[u] <- currentFrame[u]
			<-sema
		}
	}
}

func reduceBrightness(universe [512]byte, percentage int) [512]byte {
	for i := range universe {
		universe[i] = byte(float64(universe[i]) * (float64(percentage) / float64(100)))
	}
	return universe
}

func main() {
	go func() {
		var err error
		listenPort := 5353
		if os.Getenv("LISTEN_PORT") != "" {
			listenPort, err = strconv.Atoi(os.Getenv("LISTEN_PORT"))
			if err != nil {
				listenPort = 5353
			}
		}
		fmt.Printf("Starting webserver on port %d\n", listenPort)
		webserver.HandleHTTP(globalEffectChannel, listenPort, overrideFlag)
	}()

	PanelIP := os.Getenv("PANEL_IP")
	if PanelIP == "" {
		PanelIP = "127.0.0.1"
	}

	PanelWidth, err := strconv.Atoi(os.Getenv("PANEL_WIDTH"))
	if err != nil {
		PanelWidth = 60
	}
	PanelHeight, err := strconv.Atoi(os.Getenv("PANEL_HEIGHT"))
	if err != nil {
		PanelHeight = 15
	}
	msDelay, err = strconv.Atoi(os.Getenv("MS_DELAY"))
	if err != nil {
		msDelay = 40
	}

	universeCount = int(math.Ceil(float64(PanelHeight*PanelWidth*3) / 510))
	fmt.Printf("Universe count is %d\n", universeCount)

	PanelBrightness, err := strconv.Atoi(os.Getenv("PANEL_BRIGHTNESS"))
	if err != nil {
		PanelBrightness = 100
	}

	//instead of "" you could provide an ip-address that the socket should bind to
	trans, err := sacn.NewTransmitter("", [16]byte{1, 2}, "test")
	if err != nil {
		log.Fatal(err)
	}

	//activates the universes
	for i := 0; i < universeCount; i++ {
		channels[i], err = trans.Activate(uint16(i + 1))
		//deactivate the channel on exit
		defer close(channels[i])
		trans.SetMulticast(uint16(i+1), false) //this specific setup will not multicast on windows,
		trans.SetDestinations(uint16(i+1), []string{PanelIP})
	}

	// Plasma for start frame
	/*rearranged := reduceBrightness(remapping.SliceRearrange(PanelWidth, PanelHeight, true, remapping.XYGridToLinear(PanelWidth, PanelHeight, patterns.PlasmaPanel(PanelWidth, PanelHeight, 40))), PanelBrightness)
	for i := 0; i < universeCount; i++ {
		currentFrame[i] = remapping.Slice512(rearranged[i])
	}*/
	var e queue.QueueItem
	e.Effect = "red"
	e.Duration = 40 * 50
	globalEffectChannel <- e
	e.Effect = "colour"
	e.Duration = 40 * 10
	e.SeedColour[0] = 0
	e.SeedColour[1] = 255
	e.SeedColour[2] = 0
	globalEffectChannel <- e
	e.Effect = "plasma"
	e.Speed = 40
	globalEffectChannel <- e
	e.Effect = "queue"
	overrideEffect = e
	previousEffect.Effect = "queue"
	go func() {
		for /*ever*/ {
			if len(overrideFlag) > 0 {
				e = <-overrideFlag
				if e.Effect == "queue" {
					if previousEffect.Effect != "queue" {
						fmt.Printf("Returning to queue : %s\n", previousEffect.Effect)
						overrideEffect = e
						currentEffect = previousEffect
						previousEffect.Effect = "queue"
						currentEffect = <-globalEffectChannel
					}
				} else {
					fmt.Printf("Overriding with : %s\n", e.Effect)
					if previousEffect.Effect == "queue" {
						previousEffect = currentEffect
					}
					overrideEffect = e
					currentEffect = e
				}
			}
			if overrideEffect.Effect == "queue" {
				if currentEffect.Duration > 0 {
					if currentEffect.Duration%uint64(msDelay) != 0 {
						currentEffect.Duration = uint64(currentEffect.Duration/uint64(msDelay)) * uint64(msDelay)
					}
					currentEffect.Duration -= uint64(msDelay)
				} else {
					if len(globalEffectChannel) > 0 {
						previousEffect.Effect = "queue"
						lastEffect := currentEffect
						currentEffect = <-globalEffectChannel
						if lastEffect != currentEffect {
							fmt.Printf("New effect selected : %s\n", currentEffect.Effect)
						}
					}
				}
			}
			var rearranged [][]byte
			switch currentEffect.Effect {
			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)))
			case "red":
				rearranged = remapping.SliceRearrange(PanelWidth, PanelHeight, false, remapping.XYGridToLinear(PanelWidth, PanelHeight, patterns.RedPanel(PanelWidth, PanelHeight)))
			case "random":
				rearranged = remapping.SliceRearrange(PanelWidth, PanelHeight, false, remapping.XYGridToLinear(PanelWidth, PanelHeight, patterns.RandomColourPanel(PanelWidth, PanelHeight, currentEffect.Speed)))
			case "linearplasma":
				rearranged = remapping.SliceRearrange(PanelWidth, PanelHeight, false, patterns.LinearPlasma(PanelWidth*PanelHeight))
			case "gradientred":
				rearranged = remapping.SliceRearrange(PanelWidth, PanelHeight, false, patterns.Gradient(255, 0, 0, 0, 0, 255, 0, PanelWidth*PanelHeight))
			case "sine":
				var black patterns.RGBcolor
				var red patterns.RGBcolor
				red[0] = 255
				rearranged = remapping.SliceRearrange(PanelWidth, PanelHeight, true, remapping.XYGridToLinear(PanelWidth, PanelHeight, patterns.Sinewave(PanelWidth, PanelHeight, black, red, currentEffect.Speed)))
			case "sinechase":
				var black patterns.RGBcolor
				var red patterns.RGBcolor
				red[0] = 255
				rearranged = remapping.SliceRearrange(PanelWidth, PanelHeight, true, remapping.XYGridToLinear(PanelWidth, PanelHeight, patterns.SineChase(PanelWidth, PanelHeight, black, red, currentEffect.Speed)))
			case "plasmapulse":
				rearranged = remapping.SliceRearrange(PanelWidth, PanelHeight, false, remapping.XYGridToLinear(PanelWidth, PanelHeight, patterns.PlasmaColourPanel(PanelWidth, PanelHeight, currentEffect.Speed)))
			case "colour":
				rearranged = remapping.SliceRearrange(PanelWidth, PanelHeight, false, remapping.XYGridToLinear(PanelWidth, PanelHeight, patterns.FillPanel(PanelWidth, PanelHeight, currentEffect.SeedColour[0], currentEffect.SeedColour[1], currentEffect.SeedColour[2])))
			case "fade":
				rearranged = remapping.SliceRearrange(PanelWidth, PanelHeight, true, remapping.XYGridToLinear(PanelWidth, PanelHeight, patterns.GradientPanel(PanelWidth, PanelHeight, currentEffect.SeedColour, currentEffect.SecondColour)))
			default:
				rearranged = remapping.SliceRearrange(PanelWidth, PanelHeight, false, remapping.XYGridToLinear(PanelWidth, PanelHeight, patterns.FillPanel(PanelWidth, PanelHeight, 128, 0, 128)))
			}
			//rearranged := remapping.SliceRearrange(PanelWidth,PanelHeight,true,linearPlasma(PanelWidth*PanelHeight))
			sema <- struct{}{} // acquire token
			for i := 0; i < universeCount; i++ {
				currentFrame[i] = reduceBrightness(remapping.Slice512(rearranged[i]), PanelBrightness)
			}
			<-sema
			time.Sleep(40 * time.Millisecond)
		}
	}()
	foreverLoop()
}