mirror
/
av
mirror of https://bitbucket.org/ausocean/av.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

codec/h264/parse.go: skip over access unit delimiters when getting nalType

This commit is contained in:
Saxon 2019-09-11 09:47:37 +09:30
parent 54fb4a551e
commit ecdaab175b
2 changed files with 25 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codec/h264/h264dec/frame.go
View File

@ -11,7 +11,7 @@ const (
naluTypeSEI naluTypeSEI
NALTypeSPS NALTypeSPS
naluTypePPS naluTypePPS
naluTypeAccessUnitDelimiter NALTypeAccessUnitDelimiter
naluTypeEndOfSequence naluTypeEndOfSequence
naluTypeEndOfStream naluTypeEndOfStream
naluTypeFillerData naluTypeFillerData

15
codec/h264/parse.go
View File

@ -26,14 +26,20 @@ LICENSE
package h264 package h264
import "errors" import (
"errors"
"bitbucket.org/ausocean/av/codec/h264/h264dec"
)
var errNotEnoughBytes = errors.New("not enough bytes to read") var errNotEnoughBytes = errors.New("not enough bytes to read")
// NALType returns the NAL type of the given NAL unit bytes. The given NAL unit // NALType returns the NAL type of the given NAL unit bytes. The given NAL unit
// may be in byte stream or packet format. // may be in byte stream or packet format.
// NB: access unit delimiters are skipped.
func NALType(n []byte) (int, error) { func NALType(n []byte) (int, error) {
sc := frameScanner{buf: n} sc := frameScanner{buf: n}
for {
b, ok := sc.readByte() b, ok := sc.readByte()
if !ok { if !ok {
return 0, errNotEnoughBytes return 0, errNotEnoughBytes
@ -51,9 +57,12 @@ func NALType(n []byte) (int, error) {
if !ok { if !ok {
return 0, errNotEnoughBytes return 0, errNotEnoughBytes
} }
return int(b & 0x1f), nil nalType := int(b & 0x1f)
if nalType != h264dec.NALTypeAccessUnitDelimiter {
return nalType, nil
}
}
} }
return int(b & 0x1f), nil
} }
type frameScanner struct { type frameScanner struct {