mediadevices/pkg/wave/decoder_bench_test.go

package wave

import (
	"encoding/binary"
	"fmt"
	"testing"
)

func BenchmarkDecoder(b *testing.B) {
	formats := []Format{
		FormatInt16Interleaved,
		FormatInt16NonInterleaved,
		FormatFloat32Interleaved,
		FormatFloat32NonInterleaved,
	}

	var nonHostEndian binary.ByteOrder
	if hostEndian == binary.BigEndian {
		nonHostEndian = binary.LittleEndian
	} else {
		nonHostEndian = binary.BigEndian
	}

	for _, format := range formats {
		format := format
		decoder, err := NewDecoder(format)
		if err != nil {
			b.Fatal(err)
		}

		b.Run(fmt.Sprintf("%sHostEndian", format), func(b *testing.B) {
			for i := 0; i < b.N; i++ {
				_, err := decoder.Decode(hostEndian, make([]byte, 800), 2)
				if err != nil {
					b.Fatal(err)
				}
			}
		})

		b.Run(fmt.Sprintf("%sNonHostEndian", format), func(b *testing.B) {
			for i := 0; i < b.N; i++ {
				_, err := decoder.Decode(nonHostEndian, make([]byte, 800), 2)
				if err != nil {
					b.Fatal(err)
				}
			}
		})
	}
}