96 lines
3.3 KiB
C++
96 lines
3.3 KiB
C++
|
/*
|
|||
|
|
* Copyright (c) 2016-present The ZLMediaKit project authors. All Rights Reserved.
|
|||
|
|
*
|
|||
|
|
* This file is part of ZLMediaKit(https://github.com/ZLMediaKit/ZLMediaKit).
|
|||
|
|
*
|
|||
|
|
* Use of this source code is governed by MIT-like license that can be found in the
|
|||
|
|
* LICENSE file in the root of the source tree. All contributing project authors
|
|||
|
|
* may be found in the AUTHORS file in the root of the source tree.
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
#include "CommonRtp.h"
|
|||
|
|
|
|||
|
|
using namespace mediakit;
|
|||
|
|
|
|||
|
|
CommonRtpDecoder::CommonRtpDecoder(CodecId codec, size_t max_frame_size ){
|
|||
|
|
_codec = codec;
|
|||
|
|
_max_frame_size = max_frame_size;
|
|||
|
|
obtainFrame();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void CommonRtpDecoder::obtainFrame() {
|
|||
|
|
_frame = FrameImp::create();
|
|||
|
|
_frame->_codec_id = _codec;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
bool CommonRtpDecoder::inputRtp(const RtpPacket::Ptr &rtp, bool){
|
|||
|
|
auto payload_size = rtp->getPayloadSize();
|
|||
|
|
if (payload_size <= 0) {
|
|||
|
|
// 无实际负载 [AUTO-TRANSLATED:305af48f]
|
|||
|
|
// No actual load
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
auto payload = rtp->getPayload();
|
|||
|
|
auto stamp = rtp->getStamp();
|
|||
|
|
auto seq = rtp->getSeq();
|
|||
|
|
|
|||
|
|
if (_last_stamp != stamp || _frame->_buffer.size() > _max_frame_size) {
|
|||
|
|
// 时间戳发生变化或者缓存超过MAX_FRAME_SIZE，则清空上帧数据 [AUTO-TRANSLATED:96f15576]
|
|||
|
|
// If the timestamp changes or the cache exceeds MAX_FRAME_SIZE, clear the previous frame data
|
|||
|
|
if (!_frame->_buffer.empty()) {
|
|||
|
|
// 有有效帧，则输出 [AUTO-TRANSLATED:f3ff1bda]
|
|||
|
|
// If there is a valid frame, output it
|
|||
|
|
RtpCodec::inputFrame(_frame);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 新的一帧数据 [AUTO-TRANSLATED:5b5f3a35]
|
|||
|
|
// New frame data
|
|||
|
|
obtainFrame();
|
|||
|
|
_frame->_dts = rtp->getStampMS();
|
|||
|
|
_last_stamp = stamp;
|
|||
|
|
_drop_flag = false;
|
|||
|
|
} else if (_last_seq != 0 && (uint16_t)(_last_seq + 1) != seq) {
|
|||
|
|
// 时间戳未发生变化，但是seq却不连续，说明中间rtp丢包了，那么整帧应该废弃 [AUTO-TRANSLATED:577bf835]
|
|||
|
|
// If the timestamp does not change, but the seq is not continuous, it means that the RTP packet has been lost in the middle, so the entire frame should be discarded
|
|||
|
|
WarnL << "rtp丢包:" << _last_seq << " -> " << seq;
|
|||
|
|
_drop_flag = true;
|
|||
|
|
_frame->_buffer.clear();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (!_drop_flag) {
|
|||
|
|
_frame->_buffer.append((char *)payload, payload_size);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
_last_seq = seq;
|
|||
|
|
|
|||
|
|
if (_drop_flag && rtp->getHeader()->mark) {
|
|||
|
|
_drop_flag = false;
|
|||
|
|
}
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
////////////////////////////////////////////////////////////////
|
|||
|
|
|
|||
|
|
bool CommonRtpEncoder::inputFrame(const Frame::Ptr &frame){
|
|||
|
|
auto stamp = frame->pts();
|
|||
|
|
auto ptr = frame->data() + frame->prefixSize();
|
|||
|
|
auto len = frame->size() - frame->prefixSize();
|
|||
|
|
auto remain_size = len;
|
|||
|
|
auto max_size = getRtpInfo().getMaxSize();
|
|||
|
|
bool is_key = frame->keyFrame();
|
|||
|
|
bool mark = false;
|
|||
|
|
while (remain_size > 0) {
|
|||
|
|
size_t rtp_size;
|
|||
|
|
if (remain_size > max_size) {
|
|||
|
|
rtp_size = max_size;
|
|||
|
|
} else {
|
|||
|
|
rtp_size = remain_size;
|
|||
|
|
mark = true;
|
|||
|
|
}
|
|||
|
|
RtpCodec::inputRtp(getRtpInfo().makeRtp(frame->getTrackType(), ptr, rtp_size, mark, stamp), is_key);
|
|||
|
|
ptr += rtp_size;
|
|||
|
|
remain_size -= rtp_size;
|
|||
|
|
is_key = false;
|
|||
|
|
}
|
|||
|
|
return len > 0;
|
|||
|
|
}
|