// 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)
}