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

implement rtcp sender reports

Browse Source
This commit is contained in:
aler9
2020-11-21 23:54:32 +01:00
parent a21bced1dd
commit 2eebf48fca
8 changed files with 233 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
pkg/rtph264/encoder.go
View File

@@ -32,18 +32,18 @@ func NewEncoder(relativeType uint8) (*Encoder, error) {
}
// Write encodes NALUs into RTP/H264 packets.
func (e *Encoder) Write(nalus [][]byte, timestamp time.Duration) ([][]byte, error) {
func (e *Encoder) Write(nalus [][]byte, ts time.Duration) ([][]byte, error) {
if e.started == 0 {
e.started = timestamp
e.started = ts
}
// rtp/h264 uses a 90khz clock
rtpTs := e.initialTs + uint32((timestamp-e.started).Seconds()*90000)
rtpTime := e.initialTs + uint32((ts-e.started).Seconds()*90000)
var frames [][]byte
for i, nalu := range nalus {
naluFrames, err := e.writeNalu(nalu, rtpTs, (i == len(nalus)-1))
naluFrames, err := e.writeNALU(nalu, rtpTime, (i == len(nalus)-1))
if err != nil {
return nil, err
}
@@ -53,23 +53,23 @@ func (e *Encoder) Write(nalus [][]byte, timestamp time.Duration) ([][]byte, erro
return frames, nil
}
func (e *Encoder) writeNalu(nalu []byte, rtpTs uint32, isFinal bool) ([][]byte, error) {
func (e *Encoder) writeNALU(nalu []byte, rtpTime uint32, isFinal bool) ([][]byte, error) {
// if the NALU fits into a single RTP packet, use a single NALU payload
if len(nalu) < rtpPayloadMaxSize {
return e.writeSingle(nalu, rtpTs, isFinal)
return e.writeSingle(nalu, rtpTime, isFinal)
}
// otherwise, split the NALU into multiple fragmentation payloads
return e.writeFragmented(nalu, rtpTs, isFinal)
return e.writeFragmented(nalu, rtpTime, isFinal)
}
func (e *Encoder) writeSingle(nalu []byte, rtpTs uint32, isFinal bool) ([][]byte, error) {
func (e *Encoder) writeSingle(nalu []byte, rtpTime uint32, isFinal bool) ([][]byte, error) {
rpkt := rtp.Packet{
Header: rtp.Header{
Version: rtpVersion,
PayloadType: e.payloadType,
SequenceNumber: e.sequenceNumber,
Timestamp: rtpTs,
Timestamp: rtpTime,
SSRC: e.ssrc,
},
Payload: nalu,
@@ -88,7 +88,7 @@ func (e *Encoder) writeSingle(nalu []byte, rtpTs uint32, isFinal bool) ([][]byte
return [][]byte{frame}, nil
}
func (e *Encoder) writeFragmented(nalu []byte, rtpTs uint32, isFinal bool) ([][]byte, error) {
func (e *Encoder) writeFragmented(nalu []byte, rtpTime uint32, isFinal bool) ([][]byte, error) {
// use only FU-A, not FU-B, since we always use non-interleaved mode
// (packetization-mode=1)
frameCount := (len(nalu) - 1) / (rtpPayloadMaxSize - 2)
@@ -125,7 +125,7 @@ func (e *Encoder) writeFragmented(nalu []byte, rtpTs uint32, isFinal bool) ([][]
Version: rtpVersion,
PayloadType: e.payloadType,
SequenceNumber: e.sequenceNumber,
Timestamp: rtpTs,
Timestamp: rtpTime,
SSRC: e.ssrc,
},
Payload: data,