includes go-rpi-rgb-led-matrix as a dep

go-rpi-rgb-led-matrix/canvas.go

@@ -0,0 +1,77 @@
+package rgbmatrix
+
+import (
+	"image"
+	"image/color"
+	"image/draw"
+)
+
+// Canvas is a image.Image representation of a WS281x matrix, it implements
+// image.Image interface and can be used with draw.Draw for example
+type Canvas struct {
+	w, h   int
+	m      Matrix
+	closed bool
+}
+
+// NewCanvas returns a new Canvas using the given width and height and creates
+// a new WS281x matrix using the given config
+func NewCanvas(m Matrix) *Canvas {
+	w, h := m.Geometry()
+	return &Canvas{
+		w: w,
+		h: h,
+		m: m,
+	}
+}
+
+// Render update the display with the data from the LED buffer
+func (c *Canvas) Render() error {
+	return c.m.Render()
+}
+
+// ColorModel returns the canvas' color model, always color.RGBAModel
+func (c *Canvas) ColorModel() color.Model {
+	return color.RGBAModel
+}
+
+// Bounds return the topology of the Canvas
+func (c *Canvas) Bounds() image.Rectangle {
+	return image.Rect(0, 0, c.w, c.h)
+}
+
+// At returns the color of the pixel at (x, y)
+func (c *Canvas) At(x, y int) color.Color {
+	return c.m.At(c.position(x, y))
+}
+
+// Set set LED at position x,y to the provided 24-bit color value
+func (c *Canvas) Set(x, y int, color color.Color) {
+	c.m.Set(c.position(x, y), color)
+}
+
+func (c *Canvas) position(x, y int) int {
+	return x + (y * c.w)
+}
+
+// Clear set all the leds on the matrix with color.Black
+func (c *Canvas) Clear() error {
+	draw.Draw(c, c.Bounds(), &image.Uniform{color.Black}, image.ZP, draw.Src)
+	return c.m.Render()
+}
+
+// Close clears the matrix and close the matrix
+func (c *Canvas) Close() error {
+	c.Clear()
+	return c.m.Close()
+}
+
+// Matrix is an interface that represent any RGB matrix, very useful for testing
+type Matrix interface {
+	Geometry() (width, height int)
+	At(position int) color.Color
+	Set(position int, c color.Color)
+	Apply([]color.Color) error
+	Render() error
+	Close() error
+}