package/gortsplib
1
0
Fork 0
mirror of https://github.com/aler9/gortsplib synced 2025-10-10 01:20:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

add MPEG-1/2 video decoder and encoder (#415)

Browse Source
This commit is contained in:
Alessandro Ros
2023-09-13 13:11:49 +02:00
committed by GitHub
parent 73ba46591e
commit 3cdd680f06
16 changed files with 694 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

179
pkg/format/rtpmpeg1video/encoder_test.go Normal file
View File

@@ -0,0 +1,179 @@
package rtpmpeg1video
import (
"bytes"
"testing"
"github.com/pion/rtp"
"github.com/stretchr/testify/require"
)
func uint16Ptr(v uint16) *uint16 {
return &v
}
func uint32Ptr(v uint32) *uint32 {
return &v
}
func mergeBytes(vals ...[]byte) []byte {
size := 0
for _, v := range vals {
size += len(v)
}
res := make([]byte, size)
pos := 0
for _, v := range vals {
n := copy(res[pos:], v)
pos += n
}
return res
}
var cases = []struct {
name string
frame []byte
pkts []*rtp.Packet
}{
{
"single",
bytes.Repeat([]byte{1, 2, 3, 4}, 240/4),
[]*rtp.Packet{{
Header: rtp.Header{
Version: 2,
Marker: true,
PayloadType: 32,
SequenceNumber: 17645,
SSRC: 0x9dbb7812,
},
Payload: mergeBytes(
[]byte{0, 0, 0x18, 0},
bytes.Repeat([]byte{1, 2, 3, 4}, 240/4),
),
}},
},
{
"aggregated",
mergeBytes(
[]byte{0, 0, 1},
bytes.Repeat([]byte{1, 2, 3, 4}, 128/4),
[]byte{0, 0, 1},
bytes.Repeat([]byte{5, 6, 7, 8}, 128/4),
),
[]*rtp.Packet{{
Header: rtp.Header{
Version: 2,
Marker: true,
PayloadType: 32,
SequenceNumber: 17645,
SSRC: 0x9dbb7812,
},
Payload: mergeBytes(
[]byte{0, 0, 0x18, 0},
[]byte{0, 0, 1},
bytes.Repeat([]byte{1, 2, 3, 4}, 128/4),
[]byte{0, 0, 1},
bytes.Repeat([]byte{5, 6, 7, 8}, 128/4),
),
}},
},
{
"fragmented",
mergeBytes(
[]byte{0, 0, 1},
bytes.Repeat([]byte{1}, 2000),
),
[]*rtp.Packet{
{
Header: rtp.Header{
Version: 2,
Marker: false,
PayloadType: 32,
SequenceNumber: 17645,
SSRC: 0x9dbb7812,
},
Payload: mergeBytes(
[]byte{0, 0, 0x10, 0},
[]byte{0, 0, 1},
bytes.Repeat([]byte{1}, 1453),
),
},
{
Header: rtp.Header{
Version: 2,
Marker: true,
PayloadType: 32,
SequenceNumber: 17646,
SSRC: 0x9dbb7812,
},
Payload: mergeBytes(
[]byte{0, 0, 0x08, 0},
bytes.Repeat([]byte{1}, 547),
),
},
},
},
{
"fragmented to the limit",
mergeBytes(
[]byte{0, 0, 1},
bytes.Repeat([]byte{1}, 2909),
),
[]*rtp.Packet{
{
Header: rtp.Header{
Version: 2,
Marker: false,
PayloadType: 32,
SequenceNumber: 17645,
SSRC: 2646308882,
},
Payload: mergeBytes(
[]byte{0, 0, 0x10, 0},
[]byte{0, 0, 1},
bytes.Repeat([]byte{1}, 1453),
),
},
{
Header: rtp.Header{
Version: 2,
Marker: true,
PayloadType: 32,
SequenceNumber: 17646,
SSRC: 2646308882,
},
Payload: mergeBytes(
[]byte{0, 0, 0x08, 0},
bytes.Repeat([]byte{1}, 1456),
),
},
},
},
}
func TestEncode(t *testing.T) {
for _, ca := range cases {
t.Run(ca.name, func(t *testing.T) {
e := &Encoder{
SSRC: uint32Ptr(0x9dbb7812),
InitialSequenceNumber: uint16Ptr(0x44ed),
}
err := e.Init()
require.NoError(t, err)
pkts, err := e.Encode(ca.frame)
require.NoError(t, err)
require.Equal(t, ca.pkts, pkts)
})
}
}
func TestEncodeRandomInitialState(t *testing.T) {
e := &Encoder{}
err := e.Init()
require.NoError(t, err)
require.NotEqual(t, nil, e.SSRC)
require.NotEqual(t, nil, e.InitialSequenceNumber)
}