mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-11-01 17:11:31 +08:00
133 lines
2.9 KiB
C++
133 lines
2.9 KiB
C++
/*
|
||
* RtpPraser.h
|
||
*
|
||
* Created on: 2016年9月5日
|
||
* Author: xzl
|
||
*/
|
||
|
||
#ifndef SRC_RTP_RTPPARSER_H_
|
||
#define SRC_RTP_RTPPARSER_H_
|
||
|
||
#include <unordered_map>
|
||
#include "Rtsp/Rtsp.h"
|
||
#include "Player/Player.h"
|
||
#include "Player/PlayerBase.h"
|
||
#include "Util/TimeTicker.h"
|
||
|
||
using namespace std;
|
||
using namespace ZL::Util;
|
||
using namespace ZL::Player;
|
||
|
||
namespace ZL {
|
||
namespace Rtsp {
|
||
|
||
class RtpParser : public PlayerBase{
|
||
public:
|
||
typedef std::shared_ptr<RtpParser> Ptr;
|
||
RtpParser(const string &sdp);
|
||
virtual ~RtpParser();
|
||
//返回值：true 代表是i帧第一个rtp包
|
||
bool inputRtp(const RtpPacket &rtp);
|
||
void setOnVideoCB(const function<void(const H264Frame &frame)> &cb) override{
|
||
lock_guard<recursive_mutex> lck(m_mtxCB);
|
||
onVideo = cb;
|
||
}
|
||
void setOnAudioCB(const function<void(const AdtsFrame &frame)> &cb) override{
|
||
lock_guard<recursive_mutex> lck(m_mtxCB);
|
||
onAudio = cb;
|
||
}
|
||
|
||
int getVideoHeight() const override{
|
||
return m_iVideoHeight;
|
||
}
|
||
|
||
int getVideoWidth() const override{
|
||
return m_iVideoWidth;
|
||
}
|
||
|
||
float getVideoFps() const override{
|
||
return m_fVideoFps;
|
||
}
|
||
|
||
int getAudioSampleRate() const override{
|
||
return m_iSampleRate;
|
||
}
|
||
|
||
int getAudioSampleBit() const override{
|
||
return m_iSampleBit;
|
||
}
|
||
|
||
int getAudioChannel() const override{
|
||
return m_iChannel;
|
||
}
|
||
|
||
const string& getPps() const override{
|
||
return m_strPPS;
|
||
}
|
||
|
||
const string& getSps() const override{
|
||
return m_strSPS;
|
||
}
|
||
|
||
const string& getAudioCfg() const override{
|
||
return m_strAudioCfg;
|
||
}
|
||
bool containAudio() const override{
|
||
return m_bHaveAudio;
|
||
}
|
||
bool containVideo() const override{
|
||
return m_bHaveVideo;
|
||
}
|
||
bool isInited() const override{
|
||
if (m_bHaveAudio && !m_strAudioCfg.size()) {
|
||
return false;
|
||
}
|
||
if (m_bHaveVideo && !m_strSPS.size()) {
|
||
return false;
|
||
}
|
||
return true;
|
||
}
|
||
float getDuration() const override {
|
||
return m_fDuration;
|
||
}
|
||
private:
|
||
std::unordered_map<uint8_t, RtspTrack> m_mapTracks;
|
||
|
||
inline void onGetAudioTrack(const RtspTrack &audio);
|
||
inline void onGetVideoTrack(const RtspTrack &video);
|
||
|
||
//返回值：true 代表是i帧第一个rtp包
|
||
inline bool inputVideo(const RtpPacket &rtp, const RtspTrack &track);
|
||
inline bool inputAudio(const RtpPacket &rtp, const RtspTrack &track);
|
||
inline void _onGetH264(H264Frame &frame);
|
||
inline void onGetH264(H264Frame &frame);
|
||
inline void onGetAdts(AdtsFrame &frame);
|
||
//video
|
||
H264Frame m_h264frame;
|
||
//aduio
|
||
AdtsFrame m_adts;
|
||
|
||
int m_iSampleRate = 44100;
|
||
int m_iSampleBit = 16;
|
||
int m_iChannel = 1;
|
||
|
||
string m_strSPS;
|
||
string m_strPPS;
|
||
string m_strAudioCfg;
|
||
int m_iVideoWidth = 0;
|
||
int m_iVideoHeight = 0;
|
||
float m_fVideoFps = 0;
|
||
bool m_bHaveAudio = false;
|
||
bool m_bHaveVideo= false;
|
||
float m_fDuration = 0;
|
||
|
||
function<void(const H264Frame &frame)> onVideo;
|
||
function<void(const AdtsFrame &frame)> onAudio;
|
||
recursive_mutex m_mtxCB;
|
||
};
|
||
|
||
} /* namespace Rtsp */
|
||
} /* namespace ZL */
|
||
|
||
#endif /* SRC_RTP_RTPPARSER_H_ */
|