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

update examples to support H264 tracks without SPS/PPS

Browse Source
This commit is contained in:
aler9
2022-03-20 12:36:07 +01:00
parent 40af78f8bb
commit 0e8c93c5c2
4 changed files with 46 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
examples/client-read-h264-save-to-disk/mpegtsencoder.go
View File

@@ -3,7 +3,6 @@ package main
import (
"bufio"
"context"
"fmt"
"log"
"os"
"time"
@@ -27,10 +26,6 @@ type mpegtsEncoder struct {
// newMPEGTSEncoder allocates a mpegtsEncoder.
func newMPEGTSEncoder(sps []byte, pps []byte) (*mpegtsEncoder, error) {
if sps == nil || pps == nil {
return nil, fmt.Errorf("SPS or PPS not provided")
}
f, err := os.Create("mystream.ts")
if err != nil {
return nil, err
@@ -84,21 +79,32 @@ func (e *mpegtsEncoder) encode(nalus [][]byte, pts time.Duration) error {
}
for _, nalu := range nalus {
typ := h264.NALUType(nalu[0] & 0x1F)
switch typ {
case h264.NALUTypeSPS, h264.NALUTypePPS, h264.NALUTypeAccessUnitDelimiter:
// remove existing SPS, PPS, AUD
case h264.NALUTypeSPS:
e.sps = append([]byte(nil), nalu...)
case h264.NALUTypePPS:
e.pps = append([]byte(nil), nalu...)
case h264.NALUTypeAccessUnitDelimiter, h264.NALUTypeSEI:
continue
case h264.NALUTypeIDR:
// add SPS and PPS before every IDR
filteredNALUs = append(filteredNALUs, e.sps, e.pps)
if e.sps != nil && e.pps != nil {
filteredNALUs = append(filteredNALUs, e.sps, e.pps)
}
}
filteredNALUs = append(filteredNALUs, nalu)
}
// it's useless to go on if SPS or PPS have not been provided yet
if e.sps == nil || e.pps == nil {
return nil
}
// encode into Annex-B
enc, err := h264.EncodeAnnexB(filteredNALUs)
if err != nil {