133 lines
2.8 KiB
C++
133 lines
2.8 KiB
C++
/*
|
||
* RtmpParser.h
|
||
*
|
||
* Created on: 2016年12月2日
|
||
* Author: xzl
|
||
*/
|
||
|
||
#ifndef SRC_RTMP_RTMPPARSER_H_
|
||
#define SRC_RTMP_RTMPPARSER_H_
|
||
|
||
#include <functional>
|
||
#include <unordered_map>
|
||
#include "amf.h"
|
||
#include "Rtmp.h"
|
||
#include "Player/Player.h"
|
||
#include "Util/TimeTicker.h"
|
||
#include "Player/PlayerBase.h"
|
||
|
||
using namespace std;
|
||
using namespace ZL::Util;
|
||
using namespace ZL::Player;
|
||
|
||
namespace ZL {
|
||
namespace Rtmp {
|
||
|
||
class RtmpParser : public PlayerBase{
|
||
public:
|
||
typedef std::shared_ptr<RtmpParser> Ptr;
|
||
RtmpParser(const AMFValue &val);
|
||
virtual ~RtmpParser();
|
||
|
||
bool inputRtmp(const RtmpPacket &pkt);
|
||
|
||
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:
|
||
inline void onCheckMedia(const AMFValue &obj);
|
||
|
||
//返回值：true 代表是i帧第一个rtp包
|
||
inline bool inputVideo(const RtmpPacket &pkt);
|
||
inline bool inputAudio(const RtmpPacket &pkt);
|
||
inline void _onGetH264(const char *pcData, int iLen, uint32_t ui32TimeStamp);
|
||
inline void onGetH264(const char *pcData, int iLen, uint32_t ui32TimeStamp);
|
||
inline void onGetAAC(const char *pcData, int iLen, uint32_t ui32TimeStamp);
|
||
//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 Rtmp */
|
||
} /* namespace ZL */
|
||
|
||
#endif /* SRC_RTMP_RTMPPARSER_H_ */
|