splitting up some code responsibilitiy
build rgb-board / build (push) Successful in 8m23s
Details
build rgb-board / build (push) Successful in 8m23s
Details
extending the mq Message to have a type fieldmain
parent
94cfad1190
commit
1a0ea1833b
@ -0,0 +1,148 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
b64 "encoding/base64"
|
||||||
|
"encoding/json"
|
||||||
|
"image"
|
||||||
|
"image/color"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
rgbmatrix "gitea.wagshome.duckdns.org/publicWagsHome/go-rpi-rgb-led-matrix"
|
||||||
|
"github.com/disintegration/imaging"
|
||||||
|
mqtt "github.com/eclipse/paho.mqtt.golang"
|
||||||
|
"github.com/fogleman/gg"
|
||||||
|
)
|
||||||
|
|
||||||
|
// contents of struct mostly don't matter for toolkit.
|
||||||
|
type incomingMessage struct {
|
||||||
|
Type string `json:"type"`
|
||||||
|
Image string `json:"image"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Animation struct {
|
||||||
|
ctx *gg.Context
|
||||||
|
position image.Point
|
||||||
|
dir image.Point
|
||||||
|
height int
|
||||||
|
width int
|
||||||
|
stroke int
|
||||||
|
image []image.Image
|
||||||
|
images map[string]image.Image
|
||||||
|
updown string
|
||||||
|
mqmsg chan mqtt.Message
|
||||||
|
msg string
|
||||||
|
|
||||||
|
countDown int
|
||||||
|
}
|
||||||
|
|
||||||
|
// animator is a wrapping function for go routine that can receive an mq channel
|
||||||
|
func animator(tk *rgbmatrix.ToolKit, mqMessages chan mqtt.Message) {
|
||||||
|
//Playanimation comes from the toolkit, all it takes is an animation struct
|
||||||
|
tk.PlayAnimation(NewAnimation(image.Point{127, 64}, mqMessages))
|
||||||
|
}
|
||||||
|
|
||||||
|
// initializes the struct for the an play animation function, this could all be dumped into function that's wrapping go routine if I wanted
|
||||||
|
func NewAnimation(sz image.Point, mqMessages chan mqtt.Message) *Animation {
|
||||||
|
imageMap := make(map[string]image.Image)
|
||||||
|
reader, err := os.Open("marioUp.png")
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
rawMario, _, err := image.Decode(reader)
|
||||||
|
|
||||||
|
//marioUp := imaging.FlipH(imaging.Resize(rawMario, 16, 16, imaging.Lanczos))
|
||||||
|
marioUp := imaging.Resize(rawMario, 16, 16, imaging.Lanczos)
|
||||||
|
imageMap["marioUp"] = marioUp
|
||||||
|
reader, err = os.Open("marioDown.png")
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
rawMario, _, err = image.Decode(reader)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
//marioDown := imaging.FlipH(imaging.Resize(rawMario, 16, 16, imaging.Lanczos))
|
||||||
|
marioDown := imaging.Resize(rawMario, 16, 16, imaging.Lanczos)
|
||||||
|
imageMap["marioDown"] = marioDown
|
||||||
|
return &Animation{
|
||||||
|
ctx: gg.NewContext(sz.X, sz.Y),
|
||||||
|
dir: image.Point{1, 1},
|
||||||
|
height: 8,
|
||||||
|
width: 8,
|
||||||
|
stroke: 8,
|
||||||
|
images: imageMap,
|
||||||
|
updown: "marioUp",
|
||||||
|
mqmsg: mqMessages,
|
||||||
|
countDown: 5000,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func appendImage(img string, a *Animation) {
|
||||||
|
baseImage, _ := b64.StdEncoding.DecodeString(img)
|
||||||
|
bigImage, _, _ := image.Decode(bytes.NewReader(baseImage))
|
||||||
|
a.images["doorbell"] = imaging.Resize(bigImage, 64, 64, imaging.Lanczos)
|
||||||
|
}
|
||||||
|
|
||||||
|
// what happens each frame, at an interval of 50 milliseconds
|
||||||
|
func (a *Animation) Next() (image.Image, <-chan time.Time, error) {
|
||||||
|
var incoming incomingMessage
|
||||||
|
|
||||||
|
a.animateMario()
|
||||||
|
if a.images["doorbell"] != nil {
|
||||||
|
if a.countDown > 0 {
|
||||||
|
a.ctx.DrawImageAnchored(a.images["doorbell"], 0, 0, 0, 0)
|
||||||
|
a.countDown -= 50
|
||||||
|
} else {
|
||||||
|
//a.image = a.image[:len(a.image)-1]
|
||||||
|
delete(a.images, "doorbell")
|
||||||
|
a.countDown = 5000
|
||||||
|
}
|
||||||
|
}
|
||||||
|
a.ctx.SetColor(color.White)
|
||||||
|
select {
|
||||||
|
case msg := <-a.mqmsg:
|
||||||
|
json.Unmarshal([]byte(string(msg.Payload())), &incoming)
|
||||||
|
if incoming.Type == "doorbell" {
|
||||||
|
a.msg = string(msg.Payload())
|
||||||
|
a.ctx.DrawString(a.msg, 5, 9)
|
||||||
|
} else {
|
||||||
|
go appendImage(incoming.Image, a)
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
}
|
||||||
|
return a.ctx.Image(), time.After(time.Millisecond * 50), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *Animation) animateMario() {
|
||||||
|
defer a.updateMarioPosition()
|
||||||
|
a.ctx.SetColor(color.Black)
|
||||||
|
a.ctx.Clear()
|
||||||
|
if a.dir.X == 1 {
|
||||||
|
a.ctx.DrawImageAnchored(a.images[a.updown], a.position.X, a.position.Y, 0.5, 0.5)
|
||||||
|
} else {
|
||||||
|
a.ctx.DrawImageAnchored(imaging.FlipH(a.images[a.updown]), a.position.X, a.position.Y, 0.5, 0.5)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// what mario does every frame
|
||||||
|
func (a *Animation) updateMarioPosition() {
|
||||||
|
a.position.X += 1 * a.dir.X
|
||||||
|
a.position.Y += 1 * a.dir.Y
|
||||||
|
|
||||||
|
if a.position.Y+a.height > a.ctx.Height() {
|
||||||
|
a.dir.Y = -1
|
||||||
|
a.updown = "marioUp"
|
||||||
|
} else if a.position.Y-a.height < 0 {
|
||||||
|
a.updown = "marioDown"
|
||||||
|
a.dir.Y = 1
|
||||||
|
}
|
||||||
|
|
||||||
|
if a.position.X+a.width > a.ctx.Width() {
|
||||||
|
a.dir.X = -1
|
||||||
|
} else if a.position.X-a.width < 0 {
|
||||||
|
a.dir.X = 1
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,44 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
MQTT "github.com/eclipse/paho.mqtt.golang"
|
||||||
|
mqtt "github.com/eclipse/paho.mqtt.golang"
|
||||||
|
)
|
||||||
|
|
||||||
|
// listens on topic for messages
|
||||||
|
func listener(mqMessages chan mqtt.Message) {
|
||||||
|
opts := setupMQTT()
|
||||||
|
client := MQTT.NewClient(opts)
|
||||||
|
topic := "home/rgbboard"
|
||||||
|
if token := client.Connect(); token.Wait() && token.Error() != nil {
|
||||||
|
log.Println(fmt.Sprintf("failed to connect to mq: %s", token.Error().Error()))
|
||||||
|
panic(token.Error())
|
||||||
|
}
|
||||||
|
log.Println("connected")
|
||||||
|
client.Subscribe(topic, 0, func(client mqtt.Client, msg mqtt.Message) {
|
||||||
|
log.Println("Receiving ", string(msg.Payload()), " on topic: ", msg.Topic())
|
||||||
|
mqMessages <- msg
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// connection lost management
|
||||||
|
func onConnectionLostHandler(c MQTT.Client, reason error) {
|
||||||
|
log.Fatalf(reason.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
// setup connection to mqtt, topic to listen to, qos
|
||||||
|
func setupMQTT() *mqtt.ClientOptions {
|
||||||
|
opts := MQTT.NewClientOptions()
|
||||||
|
opts.AddBroker(fmt.Sprintf("tcp://%s:%s", os.Getenv("MQTTBROKER"), os.Getenv("MQTTPORT")))
|
||||||
|
opts.SetUsername(os.Getenv("MQTTUSER"))
|
||||||
|
opts.SetPassword(os.Getenv("MQTTPASSWORD"))
|
||||||
|
opts.SetClientID("rgbboard")
|
||||||
|
opts.SetAutoReconnect(true)
|
||||||
|
opts.SetConnectionLostHandler(onConnectionLostHandler)
|
||||||
|
|
||||||
|
return opts
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue