You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

80 lines
1.8 KiB
Go

package rpc
import (
"encoding/gob"
"image/color"
"net/rpc"
"gitea.wagshome.duckdns.org/publicWagsHome/go-rpi-rgb-led-matrix"
)
func init() {
gob.Register(color.RGBA{})
}
// RGBLedMatrix matrix representation for ws281x
type Client struct {
network string
addr string
client *rpc.Client
leds []color.Color
}
// NewRGBLedMatrix returns a new matrix using the given size and config
func NewClient(network, addr string) (rgbmatrix.Matrix, error) {
client, err := rpc.DialHTTP(network, addr)
if err != nil {
return nil, err
}
return &Client{
network: network,
addr: addr,
client: client,
leds: make([]color.Color, 2048),
}, nil
}
// Geometry returns the width and the height of the matrix
func (c *Client) Geometry() (width, height int) {
var reply *GeometryReply
err := c.client.Call("RPCMatrix.Geometry", &GeometryArgs{}, &reply)
if err != nil {
panic(err)
}
return reply.Width, reply.Height
}
func (c *Client) Apply(leds []color.Color) error {
defer func() { c.leds = make([]color.Color, 2048) }()
var reply *ApplyReply
return c.client.Call("RPCMatrix.Apply", &ApplyArgs{Colors: leds}, &reply)
}
// Render update the display with the data from the LED buffer
func (c *Client) Render() error {
return c.Apply(c.leds)
}
// At return an Color which allows access to the LED display data as
// if it were a sequence of 24-bit RGB values.
func (c *Client) At(position int) color.Color {
if c.leds[position] == nil {
return color.Black
}
return c.leds[position]
}
// Set set LED at position x,y to the provided 24-bit color value.
func (m *Client) Set(position int, c color.Color) {
m.leds[position] = color.RGBAModel.Convert(c)
}
// Close finalizes the ws281x interface
func (c *Client) Close() error {
return c.Apply(make([]color.Color, 2048))
}