package/gortsplib
1
0
Fork 0
mirror of https://github.com/aler9/gortsplib synced 2025-10-05 07:06:58 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
2e88705875bccf11dc5ad184dbd61b57073a66f7
gortsplib/pkg/codecs/h265/sps_test.go
Alessandro Ros 2e88705875 improve coverage (#168)
2023-01-06 23:34:10 +01:00

392 lines
13 KiB
Go
Raw Blame History

package h265
import (
"testing"
"github.com/stretchr/testify/require"
)
func TestSPSUnmarshal(t *testing.T) {
for _, ca := range []struct {
name string
byts []byte
sps SPS
width int
height int
fps float64
}{
{
"1920x1080",
[]byte{
0x42, 0x01, 0x01, 0x01, 0x60, 0x00, 0x00, 0x03,
0x00, 0x90, 0x00, 0x00, 0x03, 0x00, 0x00, 0x03,
0x00, 0x78, 0xa0, 0x03, 0xc0, 0x80, 0x10, 0xe5,
0x96, 0x66, 0x69, 0x24, 0xca, 0xe0, 0x10, 0x00,
0x00, 0x03, 0x00, 0x10, 0x00, 0x00, 0x03, 0x01,
0xe0, 0x80,
},
SPS{
TemporalIDNestingFlag: true,
ProfileTierLevel: SPS_ProfileTierLevel{
GeneralProfileIdc: 1,
GeneralProfileCompatibilityFlag: [32]bool{
false, true, true, false, false, false, false, false,
false, false, false, false, false, false, false, false,
false, false, false, false, false, false, false, false,
false, false, false, false, false, false, false, false,
},
GeneralProgressiveSourceFlag: true,
GeneralFrameOnlyConstraintFlag: true,
GeneralLevelIdc: 120,
},
ChromaFormatIdc: 1,
PicWidthInLumaSamples: 1920,
PicHeightInLumaSamples: 1080,
Log2MaxPicOrderCntLsbMinus4: 4,
SubLayerOrderingInfoPresentFlag: true,
MaxDecPicBufferingMinus1: []uint32{5},
MaxNumReorderPics: []uint32{2},
MaxLatencyIncreasePlus1: []uint32{5},
Log2DiffMaxMinLumaCodingBlockSize: 3,
Log2DiffMaxMinLumaTransformBlockSize: 3,
SampleAdaptiveOffsetEnabledFlag: true,
TemporalMvpEnabledFlag: true,
StrongIntraSmoothingEnabledFlag: true,
VUI: &SPS_VUI{
TimingInfo: &SPS_TimingInfo{
NumUnitsInTick: 1,
TimeScale: 30,
},
},
},
1920,
1080,
30,
},
{
"1920x800",
[]byte{
0x42, 0x01, 0x01, 0x01, 0x60, 0x00, 0x00, 0x03,
0x00, 0x90, 0x00, 0x00, 0x03, 0x00, 0x00, 0x03,
0x00, 0x78, 0xa0, 0x03, 0xc0, 0x80, 0x32, 0x16,
0x59, 0x59, 0xa4, 0x93, 0x2b, 0xc0, 0x5a, 0x80,
0x80, 0x80, 0x82, 0x00, 0x00, 0x07, 0xd2, 0x00,
0x00, 0xbb, 0x80, 0x10,
},
SPS{
TemporalIDNestingFlag: true,
ProfileTierLevel: SPS_ProfileTierLevel{
GeneralProfileIdc: 1,
GeneralProfileCompatibilityFlag: [32]bool{
false, true, true, false, false, false, false, false,
false, false, false, false, false, false, false, false,
false, false, false, false, false, false, false, false,
false, false, false, false, false, false, false, false,
},
GeneralProgressiveSourceFlag: true,
GeneralFrameOnlyConstraintFlag: true,
GeneralLevelIdc: 120,
},
ChromaFormatIdc: 1,
PicWidthInLumaSamples: 1920,
PicHeightInLumaSamples: 800,
Log2MaxPicOrderCntLsbMinus4: 4,
SubLayerOrderingInfoPresentFlag: true,
MaxDecPicBufferingMinus1: []uint32{4},
MaxNumReorderPics: []uint32{2},
MaxLatencyIncreasePlus1: []uint32{5},
Log2DiffMaxMinLumaCodingBlockSize: 3,
Log2DiffMaxMinLumaTransformBlockSize: 3,
SampleAdaptiveOffsetEnabledFlag: true,
TemporalMvpEnabledFlag: true,
StrongIntraSmoothingEnabledFlag: true,
VUI: &SPS_VUI{
AspectRatioInfoPresentFlag: true,
AspectRatioIdc: 1,
VideoSignalTypePresentFlag: true,
VideoFormat: 5,
ColourDescriptionPresentFlag: true,
ColourPrimaries: 1,
TransferCharacteristics: 1,
MatrixCoefficients: 1,
TimingInfo: &SPS_TimingInfo{
NumUnitsInTick: 1001,
TimeScale: 24000,
},
},
},
1920,
800,
23.976023976023978,
},
{
"1280x720",
[]byte{
0x42, 0x01, 0x01, 0x04, 0x08, 0x00, 0x00, 0x03,
0x00, 0x98, 0x08, 0x00, 0x00, 0x03, 0x00, 0x00,
0x5d, 0x90, 0x00, 0x50, 0x10, 0x05, 0xa2, 0x29,
0x4b, 0x74, 0x94, 0x98, 0x5f, 0xfe, 0x00, 0x02,
0x00, 0x02, 0xd4, 0x04, 0x04, 0x04, 0x10, 0x00,
0x00, 0x03, 0x00, 0x10, 0x00, 0x00, 0x03, 0x01,
0xe0, 0x80,
},
SPS{
TemporalIDNestingFlag: true,
ProfileTierLevel: SPS_ProfileTierLevel{
GeneralProfileIdc: 4,
GeneralProfileCompatibilityFlag: [32]bool{
false, false, false, false, true, false, false, false,
false, false, false, false, false, false, false, false,
false, false, false, false, false, false, false, false,
false, false, false, false, false, false, false, false,
},
GeneralProgressiveSourceFlag: true,
GeneralFrameOnlyConstraintFlag: true,
GeneralMax12bitConstraintFlag: true,
GeneralLowerBitRateConstraintFlag: true,
GeneralLevelIdc: 93,
},
ChromaFormatIdc: 3,
PicWidthInLumaSamples: 1280,
PicHeightInLumaSamples: 720,
BitDepthLumaMinus8: 4,
BitDepthChromaMinus8: 4,
Log2MaxPicOrderCntLsbMinus4: 4,
SubLayerOrderingInfoPresentFlag: true,
MaxDecPicBufferingMinus1: []uint32{2},
MaxNumReorderPics: []uint32{0},
MaxLatencyIncreasePlus1: []uint32{1},
Log2MinLumaCodingBlockSizeMinus3: 1,
Log2DiffMaxMinLumaCodingBlockSize: 1,
Log2DiffMaxMinLumaTransformBlockSize: 3,
TemporalMvpEnabledFlag: true,
StrongIntraSmoothingEnabledFlag: true,
VUI: &SPS_VUI{
AspectRatioInfoPresentFlag: true,
AspectRatioIdc: 255,
SarWidth: 1,
SarHeight: 1,
VideoSignalTypePresentFlag: true,
VideoFormat: 5,
ColourDescriptionPresentFlag: true,
ColourPrimaries: 1,
TransferCharacteristics: 1,
MatrixCoefficients: 1,
TimingInfo: &SPS_TimingInfo{
NumUnitsInTick: 1,
TimeScale: 30,
},
},
},
1280,
720,
30,
},
{
"10 bit",
[]byte{
0x42, 0x01, 0x01, 0x22, 0x20, 0x00, 0x00, 0x03,
0x00, 0x90, 0x00, 0x00, 0x03, 0x00, 0x00, 0x03,
0x00, 0x78, 0xa0, 0x03, 0xc0, 0x80, 0x10, 0xe4,
0xd9, 0x66, 0x66, 0x92, 0x4c, 0xaf, 0x01, 0x01,
0x00, 0x00, 0x03, 0x00, 0x64, 0x00, 0x00, 0x0b,
0xb5, 0x08,
},
SPS{
TemporalIDNestingFlag: true,
ProfileTierLevel: SPS_ProfileTierLevel{
GeneralTierFlag: 1,
GeneralProfileIdc: 2,
GeneralProfileCompatibilityFlag: [32]bool{
false, false, true, false, false, false, false, false,
false, false, false, false, false, false, false, false,
false, false, false, false, false, false, false, false,
false, false, false, false, false, false, false, false,
},
GeneralProgressiveSourceFlag: true,
GeneralFrameOnlyConstraintFlag: true,
GeneralLevelIdc: 120,
},
ChromaFormatIdc: 1,
PicWidthInLumaSamples: 1920,
PicHeightInLumaSamples: 1080,
BitDepthLumaMinus8: 2,
BitDepthChromaMinus8: 2,
Log2MaxPicOrderCntLsbMinus4: 4,
SubLayerOrderingInfoPresentFlag: true,
MaxDecPicBufferingMinus1: []uint32{5},
MaxNumReorderPics: []uint32{2},
MaxLatencyIncreasePlus1: []uint32{5},
Log2DiffMaxMinLumaCodingBlockSize: 3,
Log2DiffMaxMinLumaTransformBlockSize: 3,
SampleAdaptiveOffsetEnabledFlag: true,
TemporalMvpEnabledFlag: true,
StrongIntraSmoothingEnabledFlag: true,
VUI: &SPS_VUI{
AspectRatioInfoPresentFlag: true,
AspectRatioIdc: 1,
TimingInfo: &SPS_TimingInfo{
NumUnitsInTick: 100,
TimeScale: 2997,
},
},
},
1920,
1080,
29.97,
},
{
"nvenc",
[]byte{
0x42, 0x01, 0x01, 0x01, 0x40, 0x00, 0x00, 0x03,
0x00, 0x00, 0x03, 0x00, 0x00, 0x03, 0x00, 0x00,
0x03, 0x00, 0x7b, 0xa0, 0x03, 0xc0, 0x80, 0x11,
0x07, 0xcb, 0x96, 0xb4, 0xa4, 0x25, 0x92, 0xe3,
0x01, 0x6a, 0x02, 0x02, 0x02, 0x08, 0x00, 0x00,
0x03, 0x00, 0x08, 0x00, 0x00, 0x03, 0x01, 0xe3,
0x00, 0x2e, 0xf2, 0x88, 0x00, 0x07, 0x27, 0x0c,
0x00, 0x00, 0x98, 0x96, 0x82,
},
SPS{
TemporalIDNestingFlag: true,
ProfileTierLevel: SPS_ProfileTierLevel{
GeneralProfileIdc: 1,
GeneralProfileCompatibilityFlag: [32]bool{
false, true, false, false, false, false, false, false,
false, false, false, false, false, false, false, false,
false, false, false, false, false, false, false, false,
false, false, false, false, false, false, false, false,
},
GeneralLevelIdc: 123,
},
ChromaFormatIdc: 1,
PicWidthInLumaSamples: 1920,
PicHeightInLumaSamples: 1088,
ConformanceWindow: &SPS_ConformanceWindow{
BottomOffset: 4,
},
Log2MaxPicOrderCntLsbMinus4: 4,
SubLayerOrderingInfoPresentFlag: true,
MaxDecPicBufferingMinus1: []uint32{1},
MaxNumReorderPics: []uint32{0},
MaxLatencyIncreasePlus1: []uint32{0},
Log2MinLumaCodingBlockSizeMinus3: 1,
Log2DiffMaxMinLumaCodingBlockSize: 1,
Log2DiffMaxMinLumaTransformBlockSize: 3,
MaxTransformHierarchyDepthInter: 3,
AmpEnabledFlag: true,
SampleAdaptiveOffsetEnabledFlag: true,
ShortTermRefPicSets: []*SPS_ShortTermRefPicSet{{
NumNegativePics: 1,
DeltaPocS0Minus1: []uint32{0},
UsedByCurrPicS0Flag: []bool{true},
}},
VUI: &SPS_VUI{
AspectRatioInfoPresentFlag: true,
AspectRatioIdc: 1,
VideoSignalTypePresentFlag: true,
VideoFormat: 5,
ColourDescriptionPresentFlag: true,
ColourPrimaries: 1,
TransferCharacteristics: 1,
MatrixCoefficients: 1,
TimingInfo: &SPS_TimingInfo{
NumUnitsInTick: 1,
TimeScale: 60,
},
},
},
1920,
1080,
60,
},
{
"avigilon",
[]byte{
0x42, 0x01, 0x01, 0x01, 0x60, 0x00, 0x00, 0x03,
0x00, 0x80, 0x00, 0x00, 0x03, 0x00, 0x00, 0x03,
0x00, 0x96, 0xa0, 0x01, 0x80, 0x20, 0x06, 0xc1,
0xfe, 0x36, 0xbb, 0xb5, 0x37, 0x77, 0x25, 0xd6,
0x02, 0xdc, 0x04, 0x04, 0x04, 0x10, 0x00, 0x00,
0x3e, 0x80, 0x00, 0x04, 0x26, 0x87, 0x21, 0xde,
0xe5, 0x10, 0x01, 0x6e, 0x20, 0x00, 0x66, 0xff,
0x00, 0x0b, 0x71, 0x00, 0x03, 0x37, 0xf8, 0x80,
},
SPS{
TemporalIDNestingFlag: true,
ProfileTierLevel: SPS_ProfileTierLevel{
GeneralProfileIdc: 1,
GeneralProfileCompatibilityFlag: [32]bool{
false, true, true, false, false, false, false, false,
false, false, false, false, false, false, false, false,
false, false, false, false, false, false, false, false,
false, false, false, false, false, false, false, false,
},
GeneralProgressiveSourceFlag: true,
GeneralLevelIdc: 150,
},
ChromaFormatIdc: 1,
PicWidthInLumaSamples: 3072,
PicHeightInLumaSamples: 1728,
ConformanceWindow: &SPS_ConformanceWindow{},
Log2MaxPicOrderCntLsbMinus4: 12,
SubLayerOrderingInfoPresentFlag: true,
MaxDecPicBufferingMinus1: []uint32{1},
MaxNumReorderPics: []uint32{0},
MaxLatencyIncreasePlus1: []uint32{0},
Log2DiffMaxMinLumaCodingBlockSize: 2,
SampleAdaptiveOffsetEnabledFlag: true,
PcmEnabledFlag: true,
PcmSampleBitDepthLumaMinus1: 7,
PcmSampleBitDepthChromaMinus1: 7,
Log2DiffMaxMinLumaTransformBlockSize: 2,
MaxTransformHierarchyDepthInter: 1,
Log2MinPcmLumaCodingBlockSizeMinus3: 2,
ShortTermRefPicSets: []*SPS_ShortTermRefPicSet{
{
NumNegativePics: 1,
DeltaPocS0Minus1: []uint32{0},
UsedByCurrPicS0Flag: []bool{true},
},
},
TemporalMvpEnabledFlag: true,
VUI: &SPS_VUI{
AspectRatioInfoPresentFlag: true,
AspectRatioIdc: 1,
VideoSignalTypePresentFlag: true,
VideoFormat: 5,
VideoFullRangeFlag: true,
ColourDescriptionPresentFlag: true,
ColourPrimaries: 1,
TransferCharacteristics: 1,
MatrixCoefficients: 1,
TimingInfo: &SPS_TimingInfo{
NumUnitsInTick: 1000,
TimeScale: 17000,
},
},
},
3072,
1728,
17,
},
} {
t.Run(ca.name, func(t *testing.T) {
var sps SPS
err := sps.Unmarshal(ca.byts)
require.NoError(t, err)
require.Equal(t, ca.sps, sps)
require.Equal(t, ca.width, sps.Width())
require.Equal(t, ca.height, sps.Height())
require.Equal(t, ca.fps, sps.FPS())
})
}
}
func FuzzSPSUnmarshal(f *testing.F) {
f.Fuzz(func(t *testing.T, b []byte) {
var sps SPS
sps.Unmarshal(b)
})
}
Reference in New Issue View Git Blame Copy Permalink