mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-11-23 19:20:53 +08:00
108 lines
3.2 KiB
C++
108 lines
3.2 KiB
C++
//
|
|
// Created by xzl on 2018/10/24.
|
|
//
|
|
|
|
#include "Factory.h"
|
|
|
|
namespace mediakit{
|
|
|
|
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;
|
|
}
|
|
}
|
|
|
|
}//namespace mediakit
|
|
|