merge format and formatdecenc into formats (#222)

This commit is contained in:
Alessandro Ros
2023-04-01 13:23:16 +02:00
committed by GitHub
parent 03b96abf84
commit af3ed2bd83
161 changed files with 217 additions and 219 deletions

View File

@@ -1,95 +0,0 @@
package format
import (
"testing"
"github.com/pion/rtp"
"github.com/stretchr/testify/require"
)
func TestH264Attributes(t *testing.T) {
format := &H264{
PayloadTyp: 96,
SPS: []byte{0x01, 0x02},
PPS: []byte{0x03, 0x04},
PacketizationMode: 1,
}
require.Equal(t, "H264", format.String())
require.Equal(t, 90000, format.ClockRate())
require.Equal(t, uint8(96), format.PayloadType())
require.Equal(t, []byte{0x01, 0x02}, format.SafeSPS())
require.Equal(t, []byte{0x03, 0x04}, format.SafePPS())
format.SafeSetSPS([]byte{0x07, 0x08})
format.SafeSetPPS([]byte{0x09, 0x0A})
require.Equal(t, []byte{0x07, 0x08}, format.SafeSPS())
require.Equal(t, []byte{0x09, 0x0A}, format.SafePPS())
}
func TestH264PTSEqualsDTS(t *testing.T) {
format := &H264{
PayloadTyp: 96,
SPS: []byte{0x01, 0x02},
PPS: []byte{0x03, 0x04},
PacketizationMode: 1,
}
require.Equal(t, true, format.PTSEqualsDTS(&rtp.Packet{
Payload: []byte{0x05},
}))
require.Equal(t, false, format.PTSEqualsDTS(&rtp.Packet{
Payload: []byte{0x01},
}))
}
func TestH264MediaDescription(t *testing.T) {
t.Run("standard", func(t *testing.T) {
format := &H264{
PayloadTyp: 96,
SPS: []byte{
0x67, 0x64, 0x00, 0x0c, 0xac, 0x3b, 0x50, 0xb0,
0x4b, 0x42, 0x00, 0x00, 0x03, 0x00, 0x02, 0x00,
0x00, 0x03, 0x00, 0x3d, 0x08,
},
PPS: []byte{
0x68, 0xee, 0x3c, 0x80,
},
PacketizationMode: 1,
}
rtpmap, fmtp := format.Marshal()
require.Equal(t, "H264/90000", rtpmap)
require.Equal(t, map[string]string{
"packetization-mode": "1",
"sprop-parameter-sets": "Z2QADKw7ULBLQgAAAwACAAADAD0I,aO48gA==",
"profile-level-id": "64000C",
}, fmtp)
})
t.Run("no sps/pps", func(t *testing.T) {
format := &H264{
PayloadTyp: 96,
PacketizationMode: 1,
}
rtpmap, fmtp := format.Marshal()
require.Equal(t, "H264/90000", rtpmap)
require.Equal(t, map[string]string{
"packetization-mode": "1",
}, fmtp)
})
}
func TestH264DecEncoder(t *testing.T) {
format := &H264{}
enc := format.CreateEncoder()
pkts, err := enc.Encode([][]byte{{0x01, 0x02, 0x03, 0x04}}, 0)
require.NoError(t, err)
require.Equal(t, format.PayloadType(), pkts[0].PayloadType)
dec := format.CreateDecoder()
byts, _, err := dec.Decode(pkts[0])
require.NoError(t, err)
require.Equal(t, [][]byte{{0x01, 0x02, 0x03, 0x04}}, byts)
}