ZLMediaKit/src/RtspMuxer/AACRtpCodec.cpp

//
// Created by xzl on 2018/10/18.
//

#include "AACRtpCodec.h"

namespace mediakit{

AACRtpEncoder::AACRtpEncoder(uint32_t ui32Ssrc,
                             uint32_t ui32MtuSize,
                             uint32_t ui32SampleRate,
                             uint8_t ui8PlayloadType,
                             uint8_t ui8Interleaved) :
        RtpInfo(ui32Ssrc,
                ui32MtuSize,
                ui32SampleRate,
                ui8PlayloadType,
                ui8Interleaved),
        AACRtpDecoder(ui32SampleRate){
}

void AACRtpEncoder::inputFrame(const Frame::Ptr &frame) {
    RtpCodec::inputFrame(frame);

    GET_CONFIG_AND_REGISTER(uint32_t, cycleMS, Rtp::kCycleMS);
    auto uiStamp = frame->stamp();
    auto pcData = frame->data() + frame->prefixSize();
    auto iLen = frame->size() - frame->prefixSize();

    uiStamp %= cycleMS;
    char *ptr = (char *) pcData;
    int iSize = iLen;
    while (iSize > 0) {
        if (iSize <= _ui32MtuSize - 20) {
            _aucSectionBuf[0] = 0;
            _aucSectionBuf[1] = 16;
            _aucSectionBuf[2] = iLen >> 5;
            _aucSectionBuf[3] = (iLen & 0x1F) << 3;
            memcpy(_aucSectionBuf + 4, ptr, iSize);
            makeAACRtp(_aucSectionBuf, iSize + 4, true, uiStamp);
            break;
        }
        _aucSectionBuf[0] = 0;
        _aucSectionBuf[1] = 16;
        _aucSectionBuf[2] = (iLen) >> 5;
        _aucSectionBuf[3] = (iLen & 0x1F) << 3;
        memcpy(_aucSectionBuf + 4, ptr, _ui32MtuSize - 20);
        makeAACRtp(_aucSectionBuf, _ui32MtuSize - 16, false, uiStamp);
        ptr += (_ui32MtuSize - 20);
        iSize -= (_ui32MtuSize - 20);
    }
}

void AACRtpEncoder::makeAACRtp(const void *pData, unsigned int uiLen, bool bMark, uint32_t uiStamp) {
    uint16_t u16RtpLen = uiLen + 12;
    _ui32TimeStamp = (_ui32SampleRate / 1000) * uiStamp;
    uint32_t ts = htonl(_ui32TimeStamp);
    uint16_t sq = htons(_ui16Sequence);
    uint32_t sc = htonl(_ui32Ssrc);
    auto pRtppkt = ResourcePoolHelper<RtpPacket>::obtainObj();
    auto &rtppkt = *pRtppkt;
    unsigned char *pucRtp = rtppkt.payload;
    pucRtp[0] = '$';
    pucRtp[1] = _ui8Interleaved;
    pucRtp[2] = u16RtpLen >> 8;
    pucRtp[3] = u16RtpLen & 0x00FF;
    pucRtp[4] = 0x80;
    pucRtp[5] = (bMark << 7) | _ui8PlayloadType;
    memcpy(&pucRtp[6], &sq, 2);
    memcpy(&pucRtp[8], &ts, 4);
    //ssrc
    memcpy(&pucRtp[12], &sc, 4);
    //playload
    memcpy(&pucRtp[16], pData, uiLen);

    rtppkt.PT = _ui8PlayloadType;
    rtppkt.interleaved = _ui8Interleaved;
    rtppkt.mark = bMark;
    rtppkt.length = uiLen + 16;
    rtppkt.sequence = _ui16Sequence;
    rtppkt.timeStamp = _ui32TimeStamp;
    rtppkt.ssrc = _ui32Ssrc;
    rtppkt.type = TrackAudio;
    rtppkt.offset = 16;

    RtpCodec::inputRtp(pRtppkt, false);
    _ui16Sequence++;
}

/////////////////////////////////////////////////////////////////////////////////////
AACRtpDecoder::AACRtpDecoder(uint32_t ui32SampleRate) {
    _adts = obtainFrame();
    _sampleRate = ui32SampleRate;
}

AACFrame::Ptr AACRtpDecoder::obtainFrame() {
    //从缓存池重新申请对象，防止覆盖已经写入环形缓存的对象
    auto frame = ResourcePoolHelper<AACFrame>::obtainObj();
    frame->aac_frame_length = 7;
    frame->iPrefixSize = 7;
    return frame;
}

bool AACRtpDecoder::inputRtp(const RtpPacket::Ptr &rtppack, bool key_pos) {
    RtpCodec::inputRtp(rtppack, false);

    int length = rtppack->length - rtppack->offset;
    if (_adts->aac_frame_length + length - 4 > sizeof(AACFrame::buffer)) {
        _adts->aac_frame_length = 7;
        WarnL << "aac负载数据太长";
        return false;
    }
    memcpy(_adts->buffer + _adts->aac_frame_length, rtppack->payload + rtppack->offset + 4, length - 4);
    _adts->aac_frame_length += (length - 4);
    if (rtppack->mark == true) {
        _adts->sequence = rtppack->sequence;
        _adts->timeStamp = rtppack->timeStamp * (1000.0 / _sampleRate);
        writeAdtsHeader(*_adts, _adts->buffer);
        onGetAAC(_adts);
    }
    return false;
}

void AACRtpDecoder::onGetAAC(const AACFrame::Ptr &frame) {
    //写入环形缓存
    RtpCodec::inputFrame(frame);
    _adts = obtainFrame();
}

}//namespace mediakit