整理优化代码

2026-06-13 11:41:44 +08:00 · 2019-06-28 17:37:11 +08:00
parent 0b08831f5d
commit 7aa0c2ae33
11 changed files with 10 additions and 10 deletions
--- a/src/Extension/AACRtmp.cpp
+++ b/src/Extension/AACRtmp.cpp
@@ -0,0 +1,162 @@
+/*
+ * MIT License
+ *
+ * Copyright (c) 2016-2019 xiongziliang <771730766@qq.com>
+ *
+ * This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit).
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+#include "AACRtmp.h"
+
+namespace mediakit{
+
+AACRtmpDecoder::AACRtmpDecoder() {
+    _adts = obtainFrame();
+}
+
+AACFrame::Ptr AACRtmpDecoder::obtainFrame() {
+    //从缓存池重新申请对象，防止覆盖已经写入环形缓存的对象
+    auto frame = ResourcePoolHelper<AACFrame>::obtainObj();
+    frame->aac_frame_length = 7;
+    frame->iPrefixSize = 7;
+    return frame;
+}
+
+bool AACRtmpDecoder::inputRtmp(const RtmpPacket::Ptr &pkt, bool key_pos) {
+    RtmpCodec::inputRtmp(pkt, false);
+
+    if (pkt->isCfgFrame()) {
+        _aac_cfg = pkt->getAacCfg();
+        return false;
+    }
+    if (!_aac_cfg.empty()) {
+        onGetAAC(pkt->strBuf.data() + 2, pkt->strBuf.size() - 2, pkt->timeStamp);
+    }
+    return false;
+}
+
+void AACRtmpDecoder::onGetAAC(const char* pcData, int iLen, uint32_t ui32TimeStamp) {
+    if(iLen + 7 > sizeof(_adts->buffer)){
+        WarnL << "Illegal adts data, exceeding the length limit.";
+        return;
+    }
+    //写adts结构头
+    makeAdtsHeader(_aac_cfg,*_adts);
+
+    //拷贝aac负载
+    memcpy(_adts->buffer + 7, pcData, iLen);
+    _adts->aac_frame_length = 7 + iLen;
+    _adts->timeStamp = ui32TimeStamp;
+
+    //adts结构头转成头7个字节
+    writeAdtsHeader(*_adts, _adts->buffer);
+
+    //写入环形缓存
+    RtmpCodec::inputFrame(_adts);
+    _adts = obtainFrame();
+}
+/////////////////////////////////////////////////////////////////////////////////////
+
+AACRtmpEncoder::AACRtmpEncoder(const Track::Ptr &track) {
+    _track = dynamic_pointer_cast<AACTrack>(track);
+}
+
+void AACRtmpEncoder::inputFrame(const Frame::Ptr &frame) {
+    RtmpCodec::inputFrame(frame);
+
+    if(_aac_cfg.empty()){
+        if(frame->prefixSize() >= 7){
+            //包含adts头,从adts头获取aac配置信息
+            _aac_cfg = makeAdtsConfig(reinterpret_cast<const uint8_t *>(frame->data()));
+            makeAudioConfigPkt();
+        } else if(_track && _track->ready()){
+            //从track中和获取aac配置信息
+            _aac_cfg = _track->getAacCfg();
+            makeAudioConfigPkt();
+        }
+    }
+
+    if(!_aac_cfg.empty()){
+        RtmpPacket::Ptr rtmpPkt = ResourcePoolHelper<RtmpPacket>::obtainObj();
+        rtmpPkt->strBuf.clear();
+
+        //////////header
+        uint8_t is_config = false;
+        rtmpPkt->strBuf.push_back(_ui8AudioFlags);
+        rtmpPkt->strBuf.push_back(!is_config);
+        rtmpPkt->strBuf.append(frame->data() + frame->prefixSize(), frame->size() - frame->prefixSize());
+
+        rtmpPkt->bodySize = rtmpPkt->strBuf.size();
+        rtmpPkt->chunkId = CHUNK_AUDIO;
+        rtmpPkt->streamId = STREAM_MEDIA;
+        rtmpPkt->timeStamp = frame->stamp();
+        rtmpPkt->typeId = MSG_AUDIO;
+        RtmpCodec::inputRtmp(rtmpPkt, false);
+    }
+
+}
+
+void AACRtmpEncoder::makeAudioConfigPkt() {
+    makeAdtsHeader(_aac_cfg,*_adts);
+    int iSampleRate , iChannel , iSampleBit = 16;
+    getAACInfo(*_adts,iSampleRate,iChannel);
+
+    uint8_t flvStereoOrMono = (iChannel > 1);
+    uint8_t flvSampleRate;
+    switch (iSampleRate) {
+        case 48000:
+        case 44100:
+            flvSampleRate = 3;
+            break;
+        case 24000:
+        case 22050:
+            flvSampleRate = 2;
+            break;
+        case 12000:
+        case 11025:
+            flvSampleRate = 1;
+            break;
+        default:
+            flvSampleRate = 0;
+            break;
+    }
+    uint8_t flvSampleBit = iSampleBit == 16;
+    uint8_t flvAudioType = 10; //aac
+
+    _ui8AudioFlags = (flvAudioType << 4) | (flvSampleRate << 2) | (flvSampleBit << 1) | flvStereoOrMono;
+
+    RtmpPacket::Ptr rtmpPkt = ResourcePoolHelper<RtmpPacket>::obtainObj();
+    rtmpPkt->strBuf.clear();
+
+    //////////header
+    uint8_t is_config = true;
+    rtmpPkt->strBuf.push_back(_ui8AudioFlags);
+    rtmpPkt->strBuf.push_back(!is_config);
+    rtmpPkt->strBuf.append(_aac_cfg);
+
+    rtmpPkt->bodySize = rtmpPkt->strBuf.size();
+    rtmpPkt->chunkId = CHUNK_AUDIO;
+    rtmpPkt->streamId = STREAM_MEDIA;
+    rtmpPkt->timeStamp = 0;
+    rtmpPkt->typeId = MSG_AUDIO;
+    RtmpCodec::inputRtmp(rtmpPkt, false);
+}
+
+}//namespace mediakit