mirror of https://github.com/q191201771/lal.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
76 lines
2.6 KiB
Go
76 lines
2.6 KiB
Go
4 years ago
|
// Copyright 2020, Chef. All rights reserved.
|
||
|
|
// https://github.com/q191201771/lal
|
||
|
|
//
|
||
|
|
// Use of this source code is governed by a MIT-style license
|
||
|
|
// that can be found in the License file.
|
||
|
|
//
|
||
|
|
// Author: Chef (191201771@qq.com)
|
||
|
|
|
||
|
|
package hevc_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/q191201771/lal/pkg/hevc"
|
||
|
|
"github.com/q191201771/naza/pkg/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
// https://github.com/ksvc/FFmpeg/blob/release/3.3/libavformat/hevc.c#L936
|
||
|
|
var goldenSH = []byte{
|
||
|
|
0x1c, 0x00, 0x00, 0x00, 0x00,
|
||
|
|
0x01, // configurationVersion
|
||
|
|
0x01,
|
||
|
|
0x60, 0x00, 0x00, 0x00,
|
||
|
|
0x90, 0x00, 0x00, 0x00, 0x00, 0x00,
|
||
|
|
0x5d,
|
||
|
|
0xf0, 0x00,
|
||
|
|
0xfc,
|
||
|
|
0xfd,
|
||
|
|
0xf8,
|
||
|
|
0xf8,
|
||
|
|
0x00, 0x00, // favgFrameRate
|
||
|
|
0x0f,
|
||
|
|
0x03, // numOfArrarys
|
||
|
|
0x20, 0x00, 0x01, 0x00, 0x17,
|
||
|
|
0x40, 0x01, 0x0c, 0x01, 0xff, 0xff, 0x01, 0x60, 0x00, 0x00, 0x03, 0x00, 0xb0, 0x00, 0x00, 0x03, 0x00, 0x00, 0x03, 0x00, 0x5d, 0xac, 0x09,
|
||
|
|
0x21, 0x00, 0x01, 0x00, 0x25,
|
||
|
|
0x42, 0x01, 0x01, 0x01, 0x60, 0x00, 0x00, 0x03, 0x00, 0xb0, 0x00, 0x00, 0x03, 0x00, 0x00, 0x03, 0x00, 0x5d, 0xa0, 0x04, 0x82, 0x00, 0x40, 0x16, 0x5a, 0xee, 0x4c, 0x92, 0xea, 0x52, 0x0a, 0x0c, 0x0c, 0x05, 0xda, 0x14, 0x25,
|
||
|
|
0x22, 0x00, 0x01, 0x00, 0x08,
|
||
|
|
0x44, 0x01, 0xc0, 0xe3, 0x0f, 0x03, 0xb0, 0x84,
|
||
|
|
}
|
||
|
|
|
||
|
|
var goldenVPS = []byte{
|
||
|
|
0x40, 0x01, 0x0c, 0x01, 0xff, 0xff, 0x01, 0x60, 0x00, 0x00, 0x03, 0x00, 0xb0, 0x00, 0x00, 0x03, 0x00, 0x00, 0x03, 0x00, 0x5d, 0xac, 0x09,
|
||
|
|
}
|
||
|
|
|
||
|
|
var goldenSPS = []byte{
|
||
|
|
0x42, 0x01, 0x01, 0x01, 0x60, 0x00, 0x00, 0x03, 0x00, 0xb0, 0x00, 0x00, 0x03, 0x00, 0x00, 0x03, 0x00, 0x5d, 0xa0, 0x04, 0x82, 0x00, 0x40, 0x16, 0x5a, 0xee, 0x4c, 0x92, 0xea, 0x52, 0x0a, 0x0c, 0x0c, 0x05, 0xda, 0x14, 0x25,
|
||
|
|
}
|
||
|
|
|
||
|
|
var goldenPPS = []byte{
|
||
|
|
0x44, 0x01, 0xc0, 0xe3, 0x0f, 0x03, 0xb0, 0x84,
|
||
|
|
}
|
||
|
|
|
||
|
|
var goldenVPSSPSPPSAnnexB = []byte{
|
||
|
|
0x00, 0x00, 0x00, 0x01,
|
||
|
|
0x40, 0x01, 0x0c, 0x01, 0xff, 0xff, 0x01, 0x60, 0x00, 0x00, 0x03, 0x00, 0xb0, 0x00, 0x00, 0x03, 0x00, 0x00, 0x03, 0x00, 0x5d, 0xac, 0x09,
|
||
|
|
0x00, 0x00, 0x00, 0x01,
|
||
|
|
0x42, 0x01, 0x01, 0x01, 0x60, 0x00, 0x00, 0x03, 0x00, 0xb0, 0x00, 0x00, 0x03, 0x00, 0x00, 0x03, 0x00, 0x5d, 0xa0, 0x04, 0x82, 0x00, 0x40, 0x16, 0x5a, 0xee, 0x4c, 0x92, 0xea, 0x52, 0x0a, 0x0c, 0x0c, 0x05, 0xda, 0x14, 0x25,
|
||
|
|
0x00, 0x00, 0x00, 0x01,
|
||
|
|
0x44, 0x01, 0xc0, 0xe3, 0x0f, 0x03, 0xb0, 0x84,
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestParseVPSSPSPPSFromSeqHeader(t *testing.T) {
|
||
|
|
vps, sps, pps, err := hevc.ParseVPSSPSPPSFromSeqHeader(goldenSH)
|
||
|
|
assert.Equal(t, nil, err)
|
||
|
|
assert.Equal(t, goldenVPS, vps)
|
||
|
|
assert.Equal(t, goldenSPS, sps)
|
||
|
|
assert.Equal(t, goldenPPS, pps)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestVPSSPSPPSSeqHeader2AnnexB(t *testing.T) {
|
||
|
|
out, err := hevc.VPSSPSPPSSeqHeader2AnnexB(goldenSH)
|
||
|
|
assert.Equal(t, nil, err)
|
||
|
|
assert.Equal(t, goldenVPSSPSPPSAnnexB, out)
|
||
|
|
}
|