listen to mempool.space websocket

display/display.go

@@ -1,71 +1,69 @@
 package display
 
 import (
+	"image"
+	"image/draw"
+
+	"git.dvdt.dev/david/bitcoin-ticker-pi/fonts"
 	rgbmatrix "git.dvdt.dev/david/bitcoin-ticker-pi/go-rpi-rgb-led-matrix"
 )
 
 type Display struct {
 	canvas *rgbmatrix.Canvas
 
-	tl, tr, c, bl, br *Slot
+	topleft, topright, main, bottom *Slot
 }
 
-func New(canvas *rgbmatrix.Canvas, tl, tr, c, bl, br *Slot) *Display {
-	return &display{
-		canvas: canvas,
-		tl:     tl,
-		tr:     tr,
-		c:      c,
-		bl:     bl,
-		br:     br,
+func New(canvas *rgbmatrix.Canvas, topleft, topright, main, bottom *Slot) *Display {
+	return &Display{
+		canvas:   canvas,
+		topleft:  topleft,
+		topright: topright,
+		main:     main,
+		bottom:   bottom,
 	}
 }
 
 func (d *Display) UpdateTopLeft(text []Segment) error {
-	err := d.tl.Update(text)
+	err := d.topleft.Update(text)
 	if err != nil {
 		return err
 	}
-	d.tl.Draw(d.canvas)
+	d.topleft.Draw(d.canvas)
 	return nil
 }
 
 func (d *Display) UpdateTopRight(text []Segment) error {
-	err := d.tr.Update(text)
+	err := d.topright.Update(text)
 	if err != nil {
 		return err
 	}
-	d.tr.Draw(d.canvas)
+	d.topright.Draw(d.canvas)
 	return nil
 }
 
-func (d *Display) UpdateCenter(text []Segment) error {
-	err := d.c.Update(text)
+func (d *Display) UpdateMain(text []Segment) error {
+	err := d.main.Update(text)
 	if err != nil {
 		return err
 	}
-	d.c.Draw(d.canvas)
+	d.main.Draw(d.canvas)
 	return nil
 }
 
-func (d *Display) UpdateBottomLeft(text []Segment) error {
-	err := d.bl.Update(text)
+func (d *Display) UpdateBottom(text []Segment) error {
+	err := d.bottom.Update(text)
 	if err != nil {
 		return err
 	}
-	d.bl.Draw(d.canvas)
-	return nil
-}
-
-func (d *Display) UpdateBottomRight(text []Segment) error {
-	err := d.br.Update(text)
-	if err != nil {
-		return err
-	}
-	d.bl.Draw(d.canvas)
+	d.bottom.Draw(d.canvas)
 	return nil
 }
 
 func (d *Display) Render() error {
 	return d.canvas.Render()
 }
+
+func (d *Display) Clear() {
+	draw.Draw(d.canvas, d.canvas.Bounds(), &image.Uniform{fonts.Black}, image.ZP, draw.Src)
+}