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

improve example

Browse Source
This commit is contained in:
aler9
2021-12-03 11:39:12 +01:00
parent 2fbd177b05
commit 03fcdc7297
1 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -102,6 +102,19 @@ func (e *mpegtsEncoder) encode(nalus [][]byte, pts time.Duration) error {
pts -= e.startPTS
dts := e.dtsEst.Feed(pts)
oh := &astits.PESOptionalHeader{
MarkerBits: 2,
}
if dts == pts {
oh.PTSDTSIndicator = astits.PTSDTSIndicatorOnlyPTS
oh.PTS = &astits.ClockReference{Base: int64(pts.Seconds() * 90000)}
} else {
oh.PTSDTSIndicator = astits.PTSDTSIndicatorBothPresent
oh.DTS = &astits.ClockReference{Base: int64(dts.Seconds() * 90000)}
oh.PTS = &astits.ClockReference{Base: int64(pts.Seconds() * 90000)}
}
// write TS packet
_, err = e.mux.WriteData(&astits.MuxerData{
PID: 256,
@@ -110,13 +123,8 @@ func (e *mpegtsEncoder) encode(nalus [][]byte, pts time.Duration) error {
},
PES: &astits.PESData{
Header: &astits.PESHeader{
OptionalHeader: &astits.PESOptionalHeader{
MarkerBits: 2,
PTSDTSIndicator: astits.PTSDTSIndicatorBothPresent,
DTS: &astits.ClockReference{Base: int64(dts.Seconds() * 90000)},
PTS: &astits.ClockReference{Base: int64(pts.Seconds() * 90000)},
},
StreamID: 224, // video
OptionalHeader: oh,
StreamID: 224, // video
},
Data: enc,
},