添加工厂类 统一管理

src/Common/Factory.cpp

@@ -0,0 +1,103 @@
+//
+// Created by xzl on 2018/10/24.
+//
+
+#include "Factory.h"
+
+Sdp::Ptr Factory::getSdpByTrack(const Track::Ptr &track) {
+    switch (track->getCodecId()){
+        case CodecH264:{
+            H264Track::Ptr h264Track = dynamic_pointer_cast<H264Track>(track);
+            if(!h264Track){
+                return nullptr;
+            }
+            return std::make_shared<H264Sdp>(h264Track->getSps(),h264Track->getPps());
+        }
+
+        case CodecAAC:{
+            AACTrack::Ptr aacTrack = dynamic_pointer_cast<AACTrack>(track);
+            if(!aacTrack){
+                return nullptr;
+            }
+            return std::make_shared<AACSdp>(aacTrack->getAacCfg(),aacTrack->getAudioSampleRate());
+        }
+
+        default:
+            return nullptr;
+    }
+}
+
+
+Track::Ptr Factory::getTrackBySdp(const string &sdp) {
+    if (strcasestr(sdp.data(), "mpeg4-generic") != nullptr) {
+        string aac_cfg_str = FindField(sdp.c_str(), "config=", "\r\n");
+        if (aac_cfg_str.size() != 4) {
+            aac_cfg_str = FindField(sdp.c_str(), "config=", ";");
+        }
+        if (aac_cfg_str.size() != 4) {
+            return nullptr;
+        }
+        string aac_cfg;
+
+        unsigned int cfg1;
+        sscanf(aac_cfg_str.substr(0, 2).c_str(), "%02X", &cfg1);
+        cfg1 &= 0x00FF;
+        aac_cfg.push_back(cfg1);
+
+        unsigned int cfg2;
+        sscanf(aac_cfg_str.substr(2, 2).c_str(), "%02X", &cfg2);
+        cfg2 &= 0x00FF;
+        aac_cfg.push_back(cfg2);
+
+        return std::make_shared<AACTrack>(aac_cfg);
+    }
+
+    if (strcasestr(sdp.data(), "h264") != nullptr) {
+        string sps_pps = FindField(sdp.c_str(), "sprop-parameter-sets=", "\r\n");
+        if(sps_pps.empty()){
+            return std::make_shared<H264Track>();
+        }
+        string base64_SPS = FindField(sps_pps.c_str(), NULL, ",");
+        string base64_PPS = FindField(sps_pps.c_str(), ",", NULL);
+        if(base64_PPS.back() == ';'){
+            base64_PPS.pop_back();
+        }
+
+        auto sps = decodeBase64(base64_SPS);
+        auto pps = decodeBase64(base64_PPS);
+        return std::make_shared<H264Track>(sps,pps,0,0);
+    }
+
+
+    return nullptr;
+}
+
+
+
+RtpCodec::Ptr Factory::getRtpEncoderById(CodecId codecId,
+                                          uint32_t ui32Ssrc,
+                                          uint32_t ui32MtuSize,
+                                          uint32_t ui32SampleRate,
+                                          uint8_t ui8PlayloadType,
+                                          uint8_t ui8Interleaved) {
+    switch (codecId){
+        case CodecH264:
+            return std::make_shared<H264RtpEncoder>(ui32Ssrc,ui32MtuSize,ui32SampleRate,ui8PlayloadType,ui8Interleaved);
+        case CodecAAC:
+            return std::make_shared<AACRtpEncoder>(ui32Ssrc,ui32MtuSize,ui32SampleRate,ui8PlayloadType,ui8Interleaved);
+        default:
+            return nullptr;
+    }
+}
+
+RtpCodec::Ptr Factory::getRtpDecoderById(CodecId codecId, uint32_t ui32SampleRate) {
+    switch (codecId){
+        case CodecH264:
+            return std::make_shared<H264RtpDecoder>();
+        case CodecAAC:
+            return std::make_shared<AACRtpDecoder>(ui32SampleRate);
+        default:
+            return nullptr;
+    }
+}
+

src/Common/Factory.h

@@ -0,0 +1,57 @@
+//
+// Created by xzl on 2018/10/24.
+//
+
+#ifndef ZLMEDIAKIT_FACTORY_H
+#define ZLMEDIAKIT_FACTORY_H
+
+#include <string>
+#include "Player/Track.h"
+#include "Rtsp/RtspSdp.h"
+
+using namespace std;
+using namespace ZL::Rtsp;
+
+class Factory {
+public:
+    /**
+     * 根据sdp生成Track对象
+     */
+    static Track::Ptr getTrackBySdp(const string &sdp);
+
+    /**
+    * 根据Track生成SDP对象
+    * @param track 媒体信息
+    * @return 返回sdp对象
+    */
+    static Sdp::Ptr getSdpByTrack(const Track::Ptr &track);
+
+
+    /**
+     * 根据CodecId生成Rtp打包器
+     * @param codecId
+     * @param ui32Ssrc
+     * @param ui32MtuSize
+     * @param ui32SampleRate
+     * @param ui8PlayloadType
+     * @param ui8Interleaved
+     * @return
+     */
+    static RtpCodec::Ptr getRtpEncoderById(CodecId codecId,
+                                           uint32_t ui32Ssrc,
+                                           uint32_t ui32MtuSize,
+                                           uint32_t ui32SampleRate,
+                                           uint8_t ui8PlayloadType,
+                                           uint8_t ui8Interleaved);
+
+    /**
+     * 根据CodecId生成Rtp解包器
+     * @param codecId
+     * @param ui32SampleRate
+     * @return
+     */
+    static RtpCodec::Ptr getRtpDecoderById(CodecId codecId, uint32_t ui32SampleRate);
+};
+
+
+#endif //ZLMEDIAKIT_FACTORY_H