|
|
// Copyright 2020, Chef. All rights reserved.
|
|
|
// https://github.com/q191201771/lal
|
|
|
//
|
|
|
// Use of this source code is governed by a MIT-style license
|
|
|
// that can be found in the License file.
|
|
|
//
|
|
|
// Author: Chef (191201771@qq.com)
|
|
|
|
|
|
package rtprtcp
|
|
|
|
|
|
import (
|
|
|
"github.com/q191201771/lal/pkg/base"
|
|
|
)
|
|
|
|
|
|
// 传入RTP包，合成帧数据，并回调返回
|
|
|
// 一路音频或一路视频各对应一个对象
|
|
|
|
|
|
var (
|
|
|
_ IRtpUnpacker = &RtpUnpackContainer{}
|
|
|
_ IRtpUnpackContainer = &RtpUnpackContainer{}
|
|
|
_ IRtpUnpackerProtocol = &RtpUnpackerAac{}
|
|
|
_ IRtpUnpackerProtocol = &RtpUnpackerAvcHevc{}
|
|
|
)
|
|
|
|
|
|
type IRtpUnpacker interface {
|
|
|
IRtpUnpackContainer
|
|
|
}
|
|
|
|
|
|
type IRtpUnpackContainer interface {
|
|
|
Feed(pkt RtpPacket)
|
|
|
}
|
|
|
|
|
|
type IRtpUnpackerProtocol interface {
|
|
|
// CalcPositionIfNeeded 计算rtp包处于帧中的位置
|
|
|
CalcPositionIfNeeded(pkt *RtpPacket)
|
|
|
|
|
|
// TryUnpackOne 尝试合成一个完整帧
|
|
|
//
|
|
|
// 从当前队列的第一个包开始合成
|
|
|
// 如果一个rtp包对应一个完整帧，则合成一帧
|
|
|
// 如果一个rtp包对应多个完整帧，则合成多帧
|
|
|
// 如果多个rtp包对应一个完整帧，则尝试合成一帧
|
|
|
//
|
|
|
// @return unpackedFlag 本次调用是否成功合成
|
|
|
// @return unpackedSeq 如果成功合成，合成使用的最后一个seq号；如果失败，则为0
|
|
|
TryUnpackOne(list *RtpPacketList) (unpackedFlag bool, unpackedSeq uint16)
|
|
|
}
|
|
|
|
|
|
// OnAvPacket @param pkt: pkt.Timestamp RTP包头中的时间戳(pts)经过clockrate换算后的时间戳，单位毫秒
|
|
|
// 注意，不支持带B帧的视频流，pts和dts永远相同
|
|
|
// pkt.PayloadType base.AvPacketPTXXX
|
|
|
// pkt.Payload AAC:
|
|
|
// 返回的是raw frame，一个AvPacket只包含一帧
|
|
|
// 引用的是接收到的RTP包中的内存块
|
|
|
// AVC或HEVC:
|
|
|
// AVCC格式，每个NAL前包含4字节NAL的长度
|
|
|
// 新申请的内存块，回调结束后，内部不再使用该内存块
|
|
|
// 注意，这一层只做RTP包的合并，假如sps和pps是两个RTP single包，则合并结果为两个AvPacket，
|
|
|
// 假如sps和pps是一个stapA包，则合并结果为一个AvPacket
|
|
|
type OnAvPacket func(pkt base.AvPacket)
|
|
|
|
|
|
// DefaultRtpUnpackerFactory 目前支持AVC，HEVC和AAC MPEG4-GENERIC，业务方也可以自己实现IRtpUnpackerProtocol，甚至是IRtpUnpackContainer
|
|
|
func DefaultRtpUnpackerFactory(payloadType base.AvPacketPt, clockRate int, maxSize int, onAvPacket OnAvPacket) IRtpUnpacker {
|
|
|
var protocol IRtpUnpackerProtocol
|
|
|
switch payloadType {
|
|
|
case base.AvPacketPtAac:
|
|
|
protocol = NewRtpUnpackerAac(payloadType, clockRate, onAvPacket)
|
|
|
case base.AvPacketPtAvc:
|
|
|
fallthrough
|
|
|
case base.AvPacketPtHevc:
|
|
|
protocol = NewRtpUnpackerAvcHevc(payloadType, clockRate, onAvPacket)
|
|
|
default:
|
|
|
Log.Fatalf("payload type not support yet. payloadType=%d", payloadType)
|
|
|
}
|
|
|
return NewRtpUnpackContainer(maxSize, protocol)
|
|
|
}
|