ssd1306: refactor tests -- show fps and heap usage

Author: ysoldak

examples/ssd1306/common/common.go

@@ -0,0 +1,52 @@
+package common
+
+import (
+	"runtime"
+
+	"image/color"
+	"time"
+
+	"tinygo.org/x/drivers/ssd1306"
+)
+
+var ms = runtime.MemStats{}
+
+func Loop(display ssd1306.Device) {
+	display.ClearDisplay()
+	w, h := display.Size()
+	x := int16(0)
+	y := int16(0)
+	deltaX := int16(1)
+	deltaY := int16(1)
+	trace := time.Now().UnixMilli() + 1000
+	frames := 0
+	for {
+		pixel := display.GetPixel(x, y)
+		c := color.RGBA{255, 255, 255, 255}
+		if pixel {
+			c = color.RGBA{0, 0, 0, 255}
+		}
+		display.SetPixel(x, y, c)
+		display.Display()
+
+		x += deltaX
+		y += deltaY
+
+		if x == 0 || x == w-1 {
+			deltaX = -deltaX
+		}
+
+		if y == 0 || y == h-1 {
+			deltaY = -deltaY
+		}
+
+		frames++
+		now := time.Now().UnixMilli()
+		if now >= trace {
+			runtime.ReadMemStats(&ms)
+			println("TS", now, "| FPS", frames, "| HeapInuse", ms.HeapInuse)
+			trace = now + 1000
+			frames = 0
+		}
+	}
+}

examples/ssd1306/i2c_128x32/main.go

@@ -3,15 +3,15 @@ package main
 import (
 	"machine"
 
-	"image/color"
-	"time"
-
+	"tinygo.org/x/drivers/examples/ssd1306/common"
 	"tinygo.org/x/drivers/ssd1306"
 )
 
 func main() {
 	machine.I2C0.Configure(machine.I2CConfig{
-		Frequency: machine.TWI_FREQ_400KHZ,
+		Frequency: 400 * machine.KHz,
+		SDA:       machine.SDA0_PIN,
+		SCL:       machine.SCL0_PIN,
 	})
 
 	display := ssd1306.NewI2C(machine.I2C0)
@@ -21,31 +21,5 @@ func main() {
 		Height:  32,
 	})
 
-	display.ClearDisplay()
-
-	x := int16(0)
-	y := int16(0)
-	deltaX := int16(1)
-	deltaY := int16(1)
-	for {
-		pixel := display.GetPixel(x, y)
-		c := color.RGBA{255, 255, 255, 255}
-		if pixel {
-			c = color.RGBA{0, 0, 0, 255}
-		}
-		display.SetPixel(x, y, c)
-		display.Display()
-
-		x += deltaX
-		y += deltaY
-
-		if x == 0 || x == 127 {
-			deltaX = -deltaX
-		}
-
-		if y == 0 || y == 31 {
-			deltaY = -deltaY
-		}
-		time.Sleep(1 * time.Millisecond)
-	}
+	common.Loop(display)
 }

examples/ssd1306/i2c_128x64/main.go

@@ -12,15 +12,14 @@ package main
 import (
 	"machine"
 
-	"image/color"
-	"time"
+	"tinygo.org/x/drivers/examples/ssd1306/common"
 
 	"tinygo.org/x/drivers/ssd1306"
 )
 
 func main() {
 	machine.I2C0.Configure(machine.I2CConfig{
-		Frequency: machine.TWI_FREQ_400KHZ,
+		Frequency: 400 * machine.KHz,
 	})
 
 	display := ssd1306.NewI2C(machine.I2C0)
@@ -30,31 +29,5 @@ func main() {
 		Height:  64,
 	})
 
-	display.ClearDisplay()
-
-	x := int16(0)
-	y := int16(0)
-	deltaX := int16(1)
-	deltaY := int16(1)
-	for {
-		pixel := display.GetPixel(x, y)
-		c := color.RGBA{255, 255, 255, 255}
-		if pixel {
-			c = color.RGBA{0, 0, 0, 255}
-		}
-		display.SetPixel(x, y, c)
-		display.Display()
-
-		x += deltaX
-		y += deltaY
-
-		if x == 0 || x == 127 {
-			deltaX = -deltaX
-		}
-
-		if y == 0 || y == 63 {
-			deltaY = -deltaY
-		}
-		time.Sleep(1 * time.Millisecond)
-	}
+	common.Loop(display)
 }

examples/ssd1306/spi_128x64/main.go

@@ -1,10 +1,9 @@
 package main
 
 import (
-	"image/color"
 	"machine"
-	"time"
 
+	"tinygo.org/x/drivers/examples/ssd1306/common"
 	"tinygo.org/x/drivers/ssd1306"
 )
 
@@ -18,31 +17,5 @@ func main() {
 		Height: 64,
 	})
 
-	display.ClearDisplay()
-
-	x := int16(64)
-	y := int16(32)
-	deltaX := int16(1)
-	deltaY := int16(1)
-	for {
-		pixel := display.GetPixel(x, y)
-		c := color.RGBA{255, 255, 255, 255}
-		if pixel {
-			c = color.RGBA{0, 0, 0, 255}
-		}
-		display.SetPixel(x, y, c)
-		display.Display()
-
-		x += deltaX
-		y += deltaY
-
-		if x == 0 || x == 127 {
-			deltaX = -deltaX
-		}
-
-		if y == 0 || y == 63 {
-			deltaY = -deltaY
-		}
-		time.Sleep(1 * time.Millisecond)
-	}
+	common.Loop(display)
 }

examples/ssd1306/spi_thumby/main.go

@@ -3,10 +3,9 @@
 package main
 
 import (
-	"image/color"
 	"machine"
-	"time"
 
+	"tinygo.org/x/drivers/examples/ssd1306/common"
 	"tinygo.org/x/drivers/ssd1306"
 )
 
@@ -20,31 +19,5 @@ func main() {
 		ResetPage: ssd1306.ResetValue{0, 5},
 	})
 
-	display.ClearDisplay()
-
-	x := int16(36)
-	y := int16(20)
-	deltaX := int16(1)
-	deltaY := int16(1)
-	for {
-		pixel := display.GetPixel(x, y)
-		c := color.RGBA{255, 255, 255, 255}
-		if pixel {
-			c = color.RGBA{0, 0, 0, 255}
-		}
-		display.SetPixel(x, y, c)
-		display.Display()
-
-		x += deltaX
-		y += deltaY
-
-		if x == 0 || x == 71 {
-			deltaX = -deltaX
-		}
-
-		if y == 0 || y == 39 {
-			deltaY = -deltaY
-		}
-		time.Sleep(1 * time.Millisecond)
-	}
+	common.Loop(display)
 }