添加工厂类 统一管理

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;
+    }
+}
+