scan_bench_test.go RawHistoryFolder

1package hints
2
3import (
4	"context"
5	"os"
6	"os/exec"
7	"runtime"
8	"testing"
9	"time"
10
11	"github.com/bnema/neferwl/internal/ports"
12)
13
14// BenchmarkScan runs whole scans of a 5120x2160 frame with a noisy window
15// (the cache never hits) and reports the Go heap bytes allocated per scan.
16func BenchmarkScan(b *testing.B) {
17	if _, err := exec.LookPath("tesseract"); err != nil {
18		b.Skip("tesseract not on PATH")
19	}
20	const w, h = 5120, 2160
21	ctx := context.Background()
22	w0 := &worker{}
23	requests := make(chan ports.HintScanRequest)
24	captures := make(chan ports.CaptureRequest, 1)
25	captured := make(chan ports.CaptureDone, 1)
26	results := make(chan ports.HintScanResult, 1)
27	req := ports.HintScanRequest{Output: "o", Width: w, Height: h, Scale: 1, Windows: []ports.HintWindow{{ID: 1, Rect: ports.Rect{W: w, H: h}}}}
28	if os.Getenv("NEFERWL_HINT_BENCH_PGM") == "0" {
29		rawPGM = false
30	}
31	b.ReportAllocs()
32	var peak uint64
33	for i := range b.N {
34		go func() {
35			creq := <-captures
36			creq.Mapped[0] = byte(i) // each scan differs: no cache hit
37			creq.Mapped[1] = byte(i >> 8)
38			captured <- ports.CaptureDone{ID: creq.ID, Output: creq.Output}
39		}()
40		res := w0.serve(ctx, req, requests, captures, captured, results, 5*time.Second)
41		if res.Err != nil {
42			b.Fatal(res.Err)
43		}
44		var m runtime.MemStats
45		runtime.ReadMemStats(&m)
46		peak = max(peak, m.HeapSys-m.HeapReleased)
47	}
48	b.ReportMetric(float64(peak)/1e6, "heapMB")
49}