Something went wrong. Try again.
This repository has no description
Something went wrong. Try again.
936 B · 32 lines
Go
123456789101112131415161718192021222324252627282930313233package media
import ( "testing"
"github.com/stretchr/testify/require")
func TestAUHasIDR(t *testing.T) { annexB := func(types ...byte) []byte { var b []byte for _, ty := range types { b = append(b, 0, 0, 0, 1, ty, 0xaa, 0xbb) } return b } avc := func(types ...byte) []byte { var b []byte for _, ty := range types { b = append(b, 0, 0, 0, 3, ty, 0xaa, 0xbb) } return b } require.True(t, auHasIDR(annexB(9, 7, 8, 5)), "IDR with headers, Annex B") require.True(t, auHasIDR(avc(7, 8, 5)), "IDR with headers, length-prefixed") require.False(t, auHasIDR(annexB(9, 1)), "a non-IDR I- or P-slice") require.False(t, auHasIDR(avc(1)), "length-prefixed non-IDR") require.False(t, auHasIDR(avc(6, 1)), "SEI then non-IDR slice") require.False(t, auHasIDR(nil)) require.False(t, auHasIDR([]byte{1, 2, 3, 4, 5, 6}), "garbage is not an IDR") require.Equal(t, []int{7, 8, 5, 1}, h264NALTypes(avc(7, 8, 5, 1)))}