From a7d8e53dac2bbcf8306e0dafe3878644c760d368 Mon Sep 17 00:00:00 2001
From: Sergey Stepanov <sergystepanov@gmail.com>
Date: Sat, 17 Apr 2021 12:55:07 +0300
Subject: [PATCH] Add video encoder tests

---
 pkg/encoder/h264/x264.go      |  3 --
 pkg/encoder/pipe.go           |  3 +-
 pkg/worker/room/media_test.go | 53 ++++++++++++++++++++++++-----------
 3 files changed, 39 insertions(+), 20 deletions(-)

diff --git a/pkg/encoder/h264/x264.go b/pkg/encoder/h264/x264.go
index 0b60d33a..6b14e71a 100644
--- a/pkg/encoder/h264/x264.go
+++ b/pkg/encoder/h264/x264.go
@@ -84,9 +84,6 @@ func NewEncoder(width, height int, options ...Option) (encoder *H264, err error)
 		width:      int32(width),
 	}
 
-	var picIn Picture
-	PictureInit(&picIn)
-
 	if encoder.ref = EncoderOpen(&param); encoder.ref == nil {
 		err = fmt.Errorf("x264: cannot open the encoder")
 		return
diff --git a/pkg/encoder/pipe.go b/pkg/encoder/pipe.go
index aab993e6..4fc0b81c 100644
--- a/pkg/encoder/pipe.go
+++ b/pkg/encoder/pipe.go
@@ -46,7 +46,8 @@ func (vp *VideoPipe) Start() {
 
 	yuvProc := yuv.NewYuvImgProcessor(vp.w, vp.h)
 	for img := range vp.Input {
-		frame := vp.encoder.Encode(yuvProc.Process(img.Image).Get())
+		yCbCr := yuvProc.Process(img.Image).Get()
+		frame := vp.encoder.Encode(yCbCr)
 		if len(frame) > 0 {
 			vp.Output <- OutFrame{Data: frame, Timestamp: img.Timestamp}
 		}
diff --git a/pkg/worker/room/media_test.go b/pkg/worker/room/media_test.go
index 8b7c2644..4adc6836 100644
--- a/pkg/worker/room/media_test.go
+++ b/pkg/worker/room/media_test.go
@@ -2,7 +2,6 @@ package room
 
 import (
 	"image"
-	"image/color"
 	"math/rand"
 	"testing"
 	"time"
@@ -13,17 +12,31 @@ import (
 	"github.com/giongto35/cloud-game/v2/pkg/encoder/vpx"
 )
 
-func BenchmarkH264(b *testing.B) {
-	benchmarkEncoder(1920, 1080, codec.H264, b)
+func TestEncoders(t *testing.T) {
+	tests := []struct {
+		n      int
+		w, h   int
+		codec  codec.VideoCodec
+		frames int
+	}{
+		{n: 3, w: 1920, h: 1080, codec: codec.H264, frames: 60 * 2},
+		{n: 3, w: 1920, h: 1080, codec: codec.VPX, frames: 60 * 2},
+	}
+
+	for _, test := range tests {
+		a := genTestImage(test.w, test.h, rand.New(rand.NewSource(int64(1))).Float32())
+		b := genTestImage(test.w, test.h, rand.New(rand.NewSource(int64(2))).Float32())
+		for i := 0; i < test.n; i++ {
+			run(test.w, test.h, test.codec, test.frames, a, b, t)
+		}
+	}
 }
 
-func BenchmarkVP8(b *testing.B) {
-	benchmarkEncoder(1920, 1080, codec.VPX, b)
-}
+func BenchmarkH264(b *testing.B) { run(1920, 1080, codec.H264, b.N, nil, nil, b) }
+func BenchmarkVP8(b *testing.B)  { run(1920, 1080, codec.VPX, b.N, nil, nil, b) }
 
-func benchmarkEncoder(w, h int, cod codec.VideoCodec, b *testing.B) {
+func run(w, h int, cod codec.VideoCodec, count int, a *image.RGBA, b *image.RGBA, backend testing.TB) {
 	var enc encoder.Encoder
-
 	if cod == codec.H264 {
 		enc, _ = h264.NewEncoder(w, h)
 	} else {
@@ -34,19 +47,23 @@ func benchmarkEncoder(w, h int, cod codec.VideoCodec, b *testing.B) {
 	go pipe.Start()
 	defer pipe.Stop()
 
-	image1 := genTestImage(w, h, rand.New(rand.NewSource(int64(1))).Float32())
-	image2 := genTestImage(w, h, rand.New(rand.NewSource(int64(2))).Float32())
+	if a == nil {
+		a = genTestImage(w, h, rand.New(rand.NewSource(int64(1))).Float32())
+	}
+	if b == nil {
+		b = genTestImage(w, h, rand.New(rand.NewSource(int64(2))).Float32())
+	}
 
-	for i := 0; i < b.N; i++ {
-		im := image1
+	for i := 0; i < count; i++ {
+		im := a
 		if i%2 == 0 {
-			im = image2
+			im = b
 		}
 		pipe.Input <- encoder.InFrame{Image: im}
 		select {
 		case <-pipe.Output:
 		case <-time.After(5 * time.Second):
-			b.Fatalf("encoder didn't produce an image")
+			backend.Fatalf("encoder didn't produce an image")
 		}
 	}
 }
@@ -55,8 +72,12 @@ func genTestImage(w, h int, seed float32) *image.RGBA {
 	img := image.NewRGBA(image.Rectangle{Max: image.Point{X: w, Y: h}})
 	for x := 0; x < w; x++ {
 		for y := 0; y < h; y++ {
-			col := color.RGBA{R: uint8(seed * 255), G: uint8(seed * 255), B: uint8(seed * 255), A: 0xff}
-			img.Set(x, y, col)
+			i := img.PixOffset(x, y)
+			s := img.Pix[i : i+4 : i+4]
+			s[0] = uint8(seed * 255)
+			s[1] = uint8(seed * 255)
+			s[2] = uint8(seed * 255)
+			s[3] = 0xff
 		}
 	}
 	return img