抽象部分rtsp相关功能

src/Player/PlayerBase.h

@@ -44,7 +44,28 @@ using namespace ZL::Network;
 namespace ZL {
 namespace Player {
 
-class PlayerBase : public mINI{
+class MediaFormat {
+public:
+	virtual ~MediaFormat(){};
+	virtual int getVideoHeight() const { return 0; };
+	virtual int getVideoWidth() const { return 0; };
+	virtual float getVideoFps() const { return 0; };
+
+	virtual int getAudioSampleRate() const { return 0; };
+	virtual int getAudioSampleBit() const { return 0; };
+	virtual int getAudioChannel() const { return 0; };
+
+	virtual const string& getPps() const { static string null;return null; };
+	virtual const string& getSps() const { static string null;return null; };
+	virtual const string& getAudioCfg() const { static string null;return null; };
+
+	virtual bool containAudio() const { return false; };
+	virtual bool containVideo() const { return false; };
+
+	virtual float getDuration() const { return 0;};
+};
+
+class PlayerBase : public MediaFormat,public mINI{
 public:
 	typedef std::shared_ptr<PlayerBase> Ptr;
 	typedef enum {
@@ -77,24 +98,13 @@ public:
 	virtual void setOnVideoCB( const function<void(const H264Frame &frame)> &cb) {};
 	virtual void setOnAudioCB( const function<void(const AdtsFrame &frame)> &cb) {};
     
-	virtual int getVideoHeight() const { return 0; };
-	virtual int getVideoWidth() const { return 0; };
-	virtual float getVideoFps() const { return 0; };
-	virtual int getAudioSampleRate() const { return 0; };
-	virtual int getAudioSampleBit() const { return 0; };
-	virtual int getAudioChannel() const { return 0; };
-	//TrackVideo = 0, TrackAudio = 1
-    virtual float getRtpLossRate(int trackType) const {return 0; };
-	virtual const string& getPps() const { static string null;return null; };
-	virtual const string& getSps() const { static string null;return null; };
-	virtual const string& getAudioCfg() const { static string null;return null; };
-	virtual bool containAudio() const { return false; };
-    virtual bool containVideo() const { return false; };
-    virtual bool isInited() const { return true; };
-    virtual float getDuration() const { return 0;};
     virtual float getProgress() const { return 0;};
     virtual void seekTo(float fProgress) {};
     virtual void setMediaSouce(const MediaSource::Ptr & src) {};
+
+	virtual bool isInited() const { return true; };
+	//TrackVideo = 0, TrackAudio = 1
+	virtual float getRtpLossRate(int trackType) const {return 0; };
 protected:
     virtual void onShutdown(const SockException &ex) {};
     virtual void onPlayResult(const SockException &ex) {};

src/RTP/RtpMaker.h

@@ -44,6 +44,8 @@ namespace Rtsp {
 class RtpMaker {
 public:
 	typedef function<void(const RtpPacket::Ptr &pPkt, bool bKeyPos)> onGetRTP;
+	typedef std::shared_ptr<RtpMaker> Ptr;
+
 	RtpMaker(const onGetRTP &cb, uint32_t ui32Ssrc, int iMtuSize,int iSampleRate,
 			uint8_t ui8PlayloadType, uint8_t ui8Interleaved) {
 		callBack = cb;

src/Rtsp/Rtsp.h

@@ -37,7 +37,9 @@ using namespace std;
 using namespace ZL::Util;
 
 typedef enum {
-	TrackVideo = 0, TrackAudio
+	TrackInvalid = -1,
+	TrackVideo = 0,
+	TrackAudio
 } TrackType;
 
 class RtspTrack{

src/Rtsp/RtspMaker.h

+//
+// Created by xzl on 2018/9/18.
+//
+
+#ifndef ZLMEDIAKIT_RTSPMAKER_H
+#define ZLMEDIAKIT_RTSPMAKER_H
+
+#include "Device/base64.h"
+#include "Rtsp.h"
+#include "RTP/RtpMaker.h"
+#include "RTP/RtpMakerH264.h"
+#include "RTP/RtpMakerAAC.h"
+#include "Player/PlayerBase.h"
+
+using namespace ZL::Player;
+using namespace ZL::Rtsp;
+
+namespace ZL{
+namespace Rtsp{
+
+class Sdp {
+public:
+    typedef std::shared_ptr<Sdp> Ptr;
+    virtual ~Sdp(){}
+    virtual string getSdp() { return "";};
+    virtual TrackType getTrackType() { return TrackInvalid;};
+    virtual RtpMaker::Ptr createRtpMaker(const RtpMaker::onGetRTP &cb,uint32_t ui32Ssrc, int iMtuSize) { return nullptr;};
+};
+
+class SdpTitle : public Sdp{
+public:
+    SdpTitle(float dur_sec = 0,
+             const map<string,string> &header = map<string,string>(),
+             int version = 0){
+        _printer << "v=" << version << "\r\n";
+
+        if(!header.empty()){
+            for (auto &pr : header){
+                _printer << pr.first << "=" << pr.second << "\r\n";
+            }
+        } else {
+            _printer << "o=- 1383190487994921 1 IN IP4 0.0.0.0\r\n";
+            _printer << "s=RTSP Session, streamed by the ZL\r\n";
+            _printer << "i=ZL Live Stream\r\n";
+            _printer << "c=IN IP4 0.0.0.0\r\n";
+            _printer << "t=0 0\r\n";
+        }
+
+        if(dur_sec <= 0){
+            _printer << "a=range:npt=0-\r\n";
+        }else{
+            _printer << "a=range:npt=0-" << dur_sec  << "\r\n";
+        }
+        _printer << "a=control:*\r\n";
+    }
+    string getSdp() override {
+        return _printer;
+    }
+
+private:
+    _StrPrinter _printer;
+};
+
+class SdpH264 : public Sdp {
+public:
+    SdpH264(const string &sps,
+            const string &pps,
+            int sample_rate = 90000,
+            int playload_type = 96,
+            int track_id = TrackVideo,
+            int bitrate = 4000){
+
+        _playload_type = playload_type;
+        _sample_rate = sample_rate;
+        _track_id = track_id;
+        
+        //视频通道
+        _printer << "m=video 0 RTP/AVP " << playload_type << "\r\n";
+        _printer << "b=AS:" << bitrate << "\r\n";
+        _printer << "a=rtpmap:" << playload_type << " H264/" << sample_rate << "\r\n";
+        _printer << "a=fmtp:" << playload_type << " packetization-mode=1;profile-level-id=";
+
+        char strTemp[100];
+        int profile_level_id = 0;
+        string strSPS = sps.substr(4);
+        string strPPS = pps.substr(4);
+        if (strSPS.length() >= 4) { // sanity check
+            profile_level_id = (strSPS[1] << 16) | (strSPS[2] << 8) | strSPS[3]; // profile_idc|constraint_setN_flag|level_idc
+        }
+        memset(strTemp, 0, 100);
+        sprintf(strTemp, "%06X", profile_level_id);
+        _printer << strTemp;
+        _printer << ";sprop-parameter-sets=";
+        memset(strTemp, 0, 100);
+        av_base64_encode(strTemp, 100, (uint8_t *) strSPS.data(), strSPS.size());
+        _printer << strTemp << ",";
+        memset(strTemp, 0, 100);
+        av_base64_encode(strTemp, 100, (uint8_t *) strPPS.data(), strPPS.size());
+        _printer << strTemp << "\r\n";
+        _printer << "a=control:trackID=" << track_id << "\r\n";
+    }
+
+    string getSdp() override {
+        return _printer;
+    }
+
+    TrackType getTrackType() override {
+        return TrackVideo;
+    };
+
+    RtpMaker::Ptr createRtpMaker(const RtpMaker::onGetRTP &cb,uint32_t ui32Ssrc, int iMtuSize) override{
+        return std::make_shared<RtpMaker_H264>(cb,ui32Ssrc,iMtuSize,_sample_rate,_playload_type,_track_id * 2);
+    };
+
+private:
+    _StrPrinter _printer;
+    int _playload_type;
+    int _sample_rate;
+    int _track_id;
+
+};
+
+
+class SdpAAC : public Sdp {
+public:
+    SdpAAC(const string &aac_cfg,
+           int sample_rate,
+           int playload_type = 98,
+           int track_id = TrackAudio,
+           int bitrate = 128){
+
+        _playload_type = playload_type;
+        _sample_rate = sample_rate;
+        _track_id = track_id;
+
+        _printer << "m=audio 0 RTP/AVP " << playload_type << "\r\n";
+        _printer << "b=AS:" << bitrate << "\r\n";
+        _printer << "a=rtpmap:" << playload_type << " MPEG4-GENERIC/" << sample_rate << "\r\n";
+
+        char configStr[32] = {0};
+        snprintf(configStr, sizeof(configStr), "%02X%02x", aac_cfg[0], aac_cfg[1]);
+        _printer << "a=fmtp:" << playload_type << " streamtype=5;profile-level-id=1;mode=AAC-hbr;"
+                 << "sizelength=13;indexlength=3;indexdeltalength=3;config="
+                 << configStr << "\r\n";
+        _printer << "a=control:trackID=" << track_id << "\r\n";
+    }
+
+    string getSdp() override {
+        return _printer;
+    }
+
+    TrackType getTrackType() override {
+        return TrackAudio;
+    };
+
+    RtpMaker::Ptr createRtpMaker(const RtpMaker::onGetRTP &cb,uint32_t ui32Ssrc, int iMtuSize) override{
+        return std::make_shared<RtpMaker_AAC>(cb,ui32Ssrc,iMtuSize,_sample_rate,_playload_type,_track_id * 2);
+    };
+private:
+    _StrPrinter _printer;
+    int _playload_type;
+    int _sample_rate;
+    int _track_id;
+};
+
+class RtspMaker{
+public:
+    RtspMaker(const RtpMaker::onGetRTP &callback) : _vec_rtp_maker(8){
+        _callback = callback;
+    }
+    
+    template<typename First,typename ...Others>
+    void addTrack(First && first,Others &&...others){
+        addTrack(std::forward<First>(first));
+        addTrack(std::forward<Others>(others)...);
+    }
+    template<typename First>
+    void addTrack(First && first){
+        _printer << first->getSdp();
+        if(first->getTrackType() >= 0){
+            auto rtpMaker = first-> createRtpMaker(_callback, 0,1400);
+            _vec_rtp_maker[first->getTrackType()] = rtpMaker;
+        }
+    }
+
+    virtual ~RtspMaker() {};
+    
+    string getSdp() {
+        return _printer;
+    }
+
+    bool makeRtp(TrackType type,const char *pcData, int iDataLen, uint32_t uiStamp){
+        if(type < 0 || type > _vec_rtp_maker.size()){
+            return false;
+        }
+        auto track = _vec_rtp_maker[type];
+        if(!track){
+            return false;
+        }
+        track->makeRtp(pcData,iDataLen,uiStamp);
+        return true;
+    }
+
+private:
+    vector<RtpMaker::Ptr> _vec_rtp_maker;
+    _StrPrinter _printer;
+    RtpMaker::onGetRTP _callback;
+
+};
+
+
+
+
+
+
+}
+}
+
+
+
+#endif //ZLMEDIAKIT_RTSPMAKER_H