初始提交

src/Rtmp/RtmpParser.h

@@ -0,0 +1,130 @@
+/*
+ * RtmpParser.h
+ *
+ *  Created on: 2016年12月2日
+ *      Author: xzl
+ */
+
+#ifndef SRC_RTMP_RTMPPARSER_H_
+#define SRC_RTMP_RTMPPARSER_H_
+
+#include "Rtmp.h"
+#include "amf.h"
+#include "Player/Player.h"
+#include <unordered_map>
+#include "Util/TimeTicker.h"
+#include <functional>
+#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_ */