|
|
// 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"
|
|
|
"github.com/q191201771/naza/pkg/nazalog"
|
|
|
)
|
|
|
|
|
|
// 传入RTP包，合成帧数据，并回调返回
|
|
|
// 一路音频或一路视频各对应一个对象
|
|
|
|
|
|
var (
|
|
|
_ IRTPUnpacker = &RTPUnpackContainer{}
|
|
|
_ IRTPUnpackContainer = &RTPUnpackContainer{}
|
|
|
_ IRTPUnpackerProtocol = &RTPUnpackerAAC{}
|
|
|
_ IRTPUnpackerProtocol = &RTPUnpackerAVCHEVC{}
|
|
|
)
|
|
|
|
|
|
type IRTPUnpacker interface {
|
|
|
IRTPUnpackContainer
|
|
|
}
|
|
|
|
|
|
type IRTPUnpackContainer interface {
|
|
|
Feed(pkt RTPPacket)
|
|
|
}
|
|
|
|
|
|
type IRTPUnpackerProtocol interface {
|
|
|
// 计算rtp包处于帧中的位置
|
|
|
CalcPositionIfNeeded(pkt *RTPPacket)
|
|
|
|
|
|
// 尝试合成一个完整帧
|
|
|
//
|
|
|
// 从当前队列的第一个包开始合成
|
|
|
// 如果一个rtp包对应一个完整帧，则合成一帧
|
|
|
// 如果一个rtp包对应多个完整帧，则合成多帧
|
|
|
// 如果多个rtp包对应一个完整帧，则尝试合成一帧
|
|
|
//
|
|
|
// @return unpackedFlag 本次调用是否成功合成
|
|
|
// @return unpackedSeq 如果成功合成，合成使用的最后一个seq号；如果失败，则为0
|
|
|
TryUnpackOne(list *RTPPacketList) (unpackedFlag bool, unpackedSeq uint16)
|
|
|
}
|
|
|
|
|
|
// @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)
|
|
|
|
|
|
// 目前支持AVC，HEVC和AAC MPEG4-GENERIC/44100/2，业务方也可以自己实现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:
|
|
|
nazalog.Fatalf("payload type not support yet. payloadType=%d", payloadType)
|
|
|
}
|
|
|
return NewRTPUnpackContainer(maxSize, protocol)
|
|
|
}
|