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.7 KiB
Go
80 lines
1.7 KiB
Go
package rpc
|
|
|
|
import (
|
|
"encoding/gob"
|
|
"image/color"
|
|
"net/rpc"
|
|
|
|
"github.com/mcuadros/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))
|
|
}
|