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

container/mts/mpegts.go: GetPTSRange now generalised for any PID

This commit is contained in:
Saxon 2019-05-11 13:04:30 +09:30
parent e2d9853264
commit ebb69ea6aa
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
container/mts/mpegts.go
View File

@ -318,9 +318,9 @@ func DiscontinuityIndicator(f bool) Option {
}
// GetPTSRange retreives the first and last PTS of an MPEGTS clip.
func GetPTSRange(clip []byte) (pts [2]uint64, err error) {
// Find the first video packet.
pkt, _, err := FindPid(clip, videoPid)
func GetPTSRange(clip []byte, pidType int) (pts [2]uint64, err error) {
// Find the first packet with PID pidType.
pkt, _, err := FindPid(clip, uint16(pidType))
if err != nil {
return [2]uint64{}, err
}
@ -344,7 +344,7 @@ func GetPTSRange(clip []byte) (pts [2]uint64, err error) {
// Get the final PTS searching from end of clip for access unit start.
for i := len(clip) - PacketSize; i >= 0; i -= PacketSize {
copy(_pkt[:], clip[i:i+PacketSize])
if packet.PayloadUnitStartIndicator(&_pkt) {
if packet.PayloadUnitStartIndicator(&_pkt) && _pkt.PID() == pidType {
payload, err = packet.Payload(&_pkt)
if err != nil {
return [2]uint64{}, err

2
container/mts/mpegts_test.go
View File

@ -51,7 +51,7 @@ func TestGetPTSRange(t *testing.T) {
curTime += interval
}
got, err := GetPTSRange(clip.Bytes())
got, err := GetPTSRange(clip.Bytes(), videoPid)
if err != nil {
t.Fatalf("did not expect error getting PTS range: %v", err)
}