增加rtmp/rtsp对音频G711A,G711U的支持

src/Extension/Factory.cpp

@@ -169,7 +169,7 @@ RtpCodec::Ptr Factory::getRtpDecoderByTrack(const Track::Ptr &track) {
 
 /////////////////////////////rtmp相关///////////////////////////////////////////
 
-Track::Ptr Factory::getTrackByAmf(const AMFValue &amf) {
+Track::Ptr Factory::getVideoTrackByAmf(const AMFValue &amf) {
     CodecId codecId = getCodecIdByAmf(amf);
     if(codecId == CodecInvalid){
         return nullptr;
@@ -178,6 +178,15 @@ Track::Ptr Factory::getTrackByAmf(const AMFValue &amf) {
 }
 
 
+mediakit::Track::Ptr Factory::getAudioTrackByAmf(const AMFValue& amf)
+{
+    CodecId codecId = getAudioCodecIdByAmf(amf);
+    if (codecId == CodecInvalid) {
+        return nullptr;
+    }
+    return getTrackByCodecId(codecId);
+}
+
 CodecId Factory::getCodecIdByAmf(const AMFValue &val){
     if (val.type() == AMF_STRING){
         auto str = val.as_string();
@@ -212,6 +221,36 @@ CodecId Factory::getCodecIdByAmf(const AMFValue &val){
 }
 
 
+CodecId Factory::getAudioCodecIdByAmf(const AMFValue& val)
+{
+    if (val.type() == AMF_STRING) {
+        auto str = val.as_string();
+        if (str == "mp4a") {
+            return CodecAAC;
+        }
+        WarnL << "暂不支持该Amf:" << str;
+        return CodecInvalid;
+    }
+
+    if (val.type() != AMF_NULL) {
+        auto type_id = val.as_integer();
+        switch (type_id) {
+        case FLV_CODEC_AAC: return CodecAAC;
+        case FLV_CODEC_G711A: return CodecG711A;
+        case FLV_CODEC_G711U: return CodecG711U;
+
+        default:
+            WarnL << "暂不支持该Amf:" << type_id;
+            return CodecInvalid;
+        }
+    }
+    else {
+        WarnL << "Metadata不存在相应的Track";
+    }
+
+    return CodecInvalid;
+}
+
 RtmpCodec::Ptr Factory::getRtmpCodecByTrack(const Track::Ptr &track) {
     switch (track->getCodecId()){
         case CodecH264:
@@ -234,8 +273,8 @@ AMFValue Factory::getAmfByCodecId(CodecId codecId) {
         case CodecAAC: return AMFValue("mp4a");
         case CodecH264: return AMFValue("avc1");
         case CodecH265: return AMFValue(FLV_CODEC_H265);
-        case CodecG711A: return AMFValue(7);
-        case CodecG711U: return AMFValue(8);
+        case CodecG711A: return AMFValue(FLV_CODEC_G711A);
+        case CodecG711U: return AMFValue(FLV_CODEC_G711U);
         default: return AMFValue(AMF_NULL);
     }
 }

src/Extension/Factory.h

@@ -56,11 +56,18 @@ public:
     ////////////////////////////////rtmp相关//////////////////////////////////
 
     /**
-     * 根据amf对象获取响应的Track
-     * @param amf rtmp metadata中的videocodecid或audiocodecid的值
+     * 根据amf对象获取视频相应的Track
+     * @param amf rtmp metadata中的videocodecid的值
      * @return
      */
-    static Track::Ptr getTrackByAmf(const AMFValue &amf);
+    static Track::Ptr getVideoTrackByAmf(const AMFValue &amf);
+
+    /**
+     * 根据amf对象获取音频相应的Track
+     * @param amf rtmp metadata中的audiocodecid的值
+     * @return
+     */
+    static Track::Ptr getAudioTrackByAmf(const AMFValue& amf);
 
     /**
      * 根据amf对象获取相应的CodecId
@@ -69,6 +76,13 @@ public:
      */
     static CodecId getCodecIdByAmf(const AMFValue &val);
 
+    /**
+     * 根据amf对象获取音频相应的CodecId
+     * @param val rtmp metadata中的audiocodecid的值
+     * @return
+     */
+    static CodecId getAudioCodecIdByAmf(const AMFValue& val);
+
     /**
      * 根据Track获取Rtmp的编解码器
      * @param track 媒体描述对象

src/Extension/Frame.cpp

@@ -28,6 +28,8 @@ const char *CodecInfo::getCodecName() {
         SWITCH_CASE(CodecH264);
         SWITCH_CASE(CodecH265);
         SWITCH_CASE(CodecAAC);
+        SWITCH_CASE(CodecG711A);
+        SWITCH_CASE(CodecG711U);
         default:
             return "unknown codec";
     }

src/Extension/G711.cpp

@@ -15,7 +15,7 @@ namespace mediakit{
 
 Sdp::Ptr G711Track::getSdp() {
     if(!ready()){
-        WarnL << "AAC Track未准备好";
+        WarnL << getCodecName() << " Track未准备好";
         return nullptr;
     }
     return std::make_shared<G711Sdp>(getCodecId(), getAudioSampleRate(), getCodecId() == CodecG711A ? 8 : 0, getAudioSampleBit());

src/Extension/G711.h

@@ -57,7 +57,7 @@ public:
     }
 
     CodecId getCodecId() const override{
-        return CodecAAC;
+        return _codecId;
     }
 
     bool keyFrame() const override {
@@ -68,6 +68,7 @@ public:
         return false;
     }
 public:
+    CodecId _codecId = CodecG711A;
     unsigned int frameLength; // 一个帧的长度包括 raw data block
     unsigned char buffer[2 * 1024 + 7];
     uint32_t timeStamp;

src/Extension/G711Rtmp.h

@@ -51,7 +51,7 @@ protected:
     G711Frame::Ptr obtainFrame();
 protected:
     G711Frame::Ptr _adts;
-    CodecId _codecid = CodecG711A;
+    CodecId _codecid = CodecInvalid;
 
 };
 

src/Extension/G711Rtp.cpp

@@ -51,12 +51,11 @@ void G711RtpEncoder::makeG711Rtp(const void *data, unsigned int len, bool mark,
 /////////////////////////////////////////////////////////////////////////////////////
 
 G711RtpDecoder::G711RtpDecoder(const Track::Ptr &track){
-    auto aacTrack = dynamic_pointer_cast<G711Track>(track);
-    _codecid = aacTrack->getCodecId();
-    if(!aacTrack || !aacTrack->ready()){
+    auto g711Track = dynamic_pointer_cast<G711Track>(track);
+    _codecid = g711Track->getCodecId();
+    if(!g711Track || !g711Track->ready()){
         WarnL << "该g711 track无效!";
     }else{
-        //_aac_cfg = aacTrack->getAacCfg();
     }
     _adts = obtainFrame();
 }
@@ -81,6 +80,7 @@ bool G711RtpDecoder::inputRtp(const RtpPacket::Ptr &rtppack, bool key_pos) {
 
     _adts->frameLength = length;
     memcpy(_adts->buffer, rtp_packet_buf, length);
+    _adts->_codecId = _codecid;
     if (rtppack->mark == true) {
         _adts->timeStamp = rtppack->timeStamp;
         onGetG711(_adts);

src/Extension/G711Rtp.h

@@ -44,7 +44,7 @@ private:
     G711Frame::Ptr obtainFrame();
 private:
     G711Frame::Ptr _adts;
-    CodecId _codecid = CodecG711A;
+    CodecId _codecid = CodecInvalid;
 };