ZLMediaKit/ext-codec/H265Rtp.h

112 lines
3.0 KiB
C++
Raw Normal View History

2018-10-30 14:59:42 +08:00
/*
2023-12-09 16:23:51 +08:00
* Copyright (c) 2016-present The ZLMediaKit project authors. All Rights Reserved.
2018-10-30 14:59:42 +08:00
*
2023-12-09 16:23:51 +08:00
* This file is part of ZLMediaKit(https://github.com/ZLMediaKit/ZLMediaKit).
2018-10-30 14:59:42 +08:00
*
2023-12-09 16:23:51 +08:00
* Use of this source code is governed by MIT-like license that can be found in the
2020-04-04 20:30:09 +08:00
* 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.
2018-10-30 14:59:42 +08:00
*/
#ifndef ZLMEDIAKIT_H265RTPCODEC_H
#define ZLMEDIAKIT_H265RTPCODEC_H
2023-12-10 10:21:40 +08:00
#include "H265.h"
2019-06-28 17:30:13 +08:00
#include "Rtsp/RtpCodec.h"
// for DtsGenerator
#include "Common/Stamp.h"
2018-10-30 14:59:42 +08:00
2023-12-10 10:21:40 +08:00
namespace mediakit {
2018-10-30 14:59:42 +08:00
/**
* h265 rtp解码类
2019-08-30 11:17:27 +08:00
* h265 over rtsp-rtp h265-Frame
* H265-over-RTPdraft-ietf-payload-rtp-h265-07.pdf
* h265 rtp decoder class
* Demultiplex h265-Frame from h265 over rtsp-rtp
* Draft (H265-over-RTP) draft-ietf-payload-rtp-h265-07.pdf
* [AUTO-TRANSLATED:24e7e278]
2018-10-30 14:59:42 +08:00
*/
2021-02-05 11:51:16 +08:00
class H265RtpDecoder : public RtpCodec {
2018-10-30 14:59:42 +08:00
public:
2022-12-02 14:43:06 +08:00
using Ptr = std::shared_ptr<H265RtpDecoder>;
2018-10-30 14:59:42 +08:00
H265RtpDecoder();
/**
* 265 rtp包
* @param rtp rtp包
* @param key_pos
* Input 265 rtp packet
* @param rtp rtp packet
* @param key_pos This parameter is ignored
* [AUTO-TRANSLATED:35e8fa1d]
2018-10-30 14:59:42 +08:00
*/
bool inputRtp(const RtpPacket::Ptr &rtp, bool key_pos = true) override;
private:
2022-08-08 17:13:39 +08:00
bool unpackAp(const RtpPacket::Ptr &rtp, const uint8_t *ptr, ssize_t size, uint64_t stamp);
bool mergeFu(const RtpPacket::Ptr &rtp, const uint8_t *ptr, ssize_t size, uint64_t stamp, uint16_t seq);
bool singleFrame(const RtpPacket::Ptr &rtp, const uint8_t *ptr, ssize_t size, uint64_t stamp);
2021-01-23 09:43:16 +08:00
bool decodeRtp(const RtpPacket::Ptr &rtp);
2018-10-30 14:59:42 +08:00
H265Frame::Ptr obtainFrame();
void outputFrame(const RtpPacket::Ptr &rtp, const H265Frame::Ptr &frame);
2018-10-30 14:59:42 +08:00
private:
bool _is_gop = false;
2021-01-23 09:43:16 +08:00
bool _using_donl_field = false;
bool _gop_dropped = false;
bool _fu_dropped = true;
2021-01-23 09:43:16 +08:00
uint16_t _last_seq = 0;
H265Frame::Ptr _frame;
DtsGenerator _dts_generator;
2018-10-30 14:59:42 +08:00
};
/**
* 265 rtp打包类
* 265 rtp packer class
* [AUTO-TRANSLATED:4b3f96fe]
2018-10-30 14:59:42 +08:00
*/
2023-12-09 16:23:51 +08:00
class H265RtpEncoder : public RtpCodec {
2018-10-30 14:59:42 +08:00
public:
2022-12-02 14:43:06 +08:00
using Ptr = std::shared_ptr<H265RtpEncoder>;
2018-10-30 14:59:42 +08:00
/**
* 265
* @param frame
* Input 265 frame
* @param frame Frame data, required
* [AUTO-TRANSLATED:48454707]
2018-10-30 14:59:42 +08:00
*/
bool inputFrame(const Frame::Ptr &frame) override;
2023-04-06 19:54:59 +08:00
/**
* frame缓存
* Flush all frame cache in output
* [AUTO-TRANSLATED:adaea568]
2023-04-06 19:54:59 +08:00
*/
void flush() override;
private:
void packRtp(const char *ptr, size_t len, uint64_t pts, bool is_mark, bool gop_pos);
void packRtpFu(const char *ptr, size_t len, uint64_t pts, bool is_mark, bool gop_pos);
void insertConfigFrame(uint64_t pts);
bool inputFrame_l(const Frame::Ptr &frame, bool is_mark);
private:
Frame::Ptr _sps;
Frame::Ptr _pps;
Frame::Ptr _vps;
Frame::Ptr _last_frame;
2018-10-30 14:59:42 +08:00
};
}//namespace mediakit
2018-10-30 14:59:42 +08:00
#endif //ZLMEDIAKIT_H265RTPCODEC_H