/*
 * Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
 *
 * This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit).
 *
 * Use of this source code is governed by MIT license that can be found in the
 * LICENSE file in the root of the source tree. All contributing project authors
 * may be found in the AUTHORS file in the root of the source tree.
 */

#include "Device.h"
#include "Util/logger.h"
#include "Util/base64.h"
#include "Extension/AAC.h"
#include "Extension/G711.h"
#include "Extension/H264.h"
#include "Extension/H265.h"

using namespace toolkit;

namespace mediakit {

DevChannel::DevChannel(const string &strVhost,
                       const string &strApp,
                       const string &strId,
                       float fDuration,
                       bool bEanbleRtsp,
                       bool bEanbleRtmp,
                       bool bEanbleHls,
                       bool bEnableMp4) :
        MultiMediaSourceMuxer(strVhost, strApp, strId, fDuration, bEanbleRtsp, bEanbleRtmp, bEanbleHls, bEnableMp4) {}

DevChannel::~DevChannel() {}

#ifdef ENABLE_X264
void DevChannel::inputYUV(char* apcYuv[3], int aiYuvLen[3], uint32_t uiStamp) {
    //TimeTicker1(50);
    if (!_pH264Enc) {
        _pH264Enc.reset(new H264Encoder());
        if (!_pH264Enc->init(_video->iWidth, _video->iHeight, _video->iFrameRate)) {
            _pH264Enc.reset();
            WarnL << "H264Encoder init failed!";
        }
    }
    if (_pH264Enc) {
        H264Encoder::H264Frame *pOut;
        int iFrames = _pH264Enc->inputData(apcYuv, aiYuvLen, uiStamp, &pOut);
        for (int i = 0; i < iFrames; i++) {
            inputH264((char *) pOut[i].pucData, pOut[i].iLength, uiStamp);
        }
    }
}
#endif //ENABLE_X264

#ifdef ENABLE_FAAC
void DevChannel::inputPCM(char* pcData, int iDataLen, uint32_t uiStamp) {
    if (!_pAacEnc) {
        _pAacEnc.reset(new AACEncoder());
        if (!_pAacEnc->init(_audio->iSampleRate, _audio->iChannel, _audio->iSampleBit)) {
            _pAacEnc.reset();
            WarnL << "AACEncoder init failed!";
        }
    }
    if (_pAacEnc) {
        unsigned char *pucOut;
        int iRet = _pAacEnc->inputData(pcData, iDataLen, &pucOut);
        if (iRet > 0) {
            inputAAC((char *) pucOut, iRet, uiStamp);
        }
    }
}
#endif //ENABLE_FAAC

void DevChannel::inputH264(const char* pcData, int iDataLen, uint32_t dts,uint32_t pts) {
    if(dts == 0){
        dts = (uint32_t)_aTicker[0].elapsedTime();
    }
    if(pts == 0){
        pts = dts;
    }
    int prefixeSize;
    if (memcmp("\x00\x00\x00\x01", pcData, 4) == 0) {
        prefixeSize = 4;
    } else if (memcmp("\x00\x00\x01", pcData, 3) == 0) {
        prefixeSize = 3;
    } else {
        prefixeSize = 0;
    }

    H264Frame::Ptr frame = std::make_shared<H264Frame>();
    frame->_dts = dts;
    frame->_pts = pts;
    frame->_buffer.assign("\x00\x00\x00\x01",4);
    frame->_buffer.append(pcData + prefixeSize, iDataLen - prefixeSize);
    frame->_prefix_size = 4;
    inputFrame(frame);
}

void DevChannel::inputH265(const char* pcData, int iDataLen, uint32_t dts,uint32_t pts) {
    if(dts == 0){
        dts = (uint32_t)_aTicker[0].elapsedTime();
    }
    if(pts == 0){
        pts = dts;
    }
    int prefixeSize;
    if (memcmp("\x00\x00\x00\x01", pcData, 4) == 0) {
        prefixeSize = 4;
    } else if (memcmp("\x00\x00\x01", pcData, 3) == 0) {
        prefixeSize = 3;
    } else {
        prefixeSize = 0;
    }

    H265Frame::Ptr frame = std::make_shared<H265Frame>();
    frame->_dts = dts;
    frame->_pts = pts;
    frame->_buffer.assign("\x00\x00\x00\x01",4);
    frame->_buffer.append(pcData + prefixeSize, iDataLen - prefixeSize);
    frame->_prefix_size = 4;
    inputFrame(frame);
}

void DevChannel::inputAAC(const char* pcData, int iDataLen, uint32_t uiStamp,bool withAdtsHeader) {
    if(withAdtsHeader){
        inputAAC(pcData+7,iDataLen-7,uiStamp,pcData);
    } else if(_audio) {
        inputAAC(pcData,iDataLen,uiStamp,(char *)_adtsHeader);
    }
}

void DevChannel::inputAAC(const char *pcDataWithoutAdts,int iDataLen, uint32_t uiStamp,const char *pcAdtsHeader){
    if(uiStamp == 0){
        uiStamp = (uint32_t)_aTicker[1].elapsedTime();
    }
    if(pcAdtsHeader + 7 == pcDataWithoutAdts){
        inputFrame(std::make_shared<AACFrameNoCacheAble>((char *)pcDataWithoutAdts - 7,iDataLen + 7,uiStamp,0,7));
    } else {
        char *dataWithAdts = new char[iDataLen + 7];
        memcpy(dataWithAdts,pcAdtsHeader,7);
        memcpy(dataWithAdts + 7 , pcDataWithoutAdts , iDataLen);
        inputFrame(std::make_shared<AACFrameNoCacheAble>(dataWithAdts,iDataLen + 7,uiStamp,0,7));
        delete [] dataWithAdts;
    }
}

void DevChannel::inputG711(const char* pcData, int iDataLen, uint32_t uiStamp){
    if (uiStamp == 0) {
        uiStamp = (uint32_t)_aTicker[1].elapsedTime();
    }
    inputFrame(std::make_shared<G711FrameNoCacheAble>(_audio->codecId, (char*)pcData, iDataLen, uiStamp, 0));
}

void DevChannel::initVideo(const VideoInfo& info) {
    _video = std::make_shared<VideoInfo>(info);
    switch (info.codecId){
        case CodecH265 : addTrack(std::make_shared<H265Track>()); break;
        case CodecH264 : addTrack(std::make_shared<H264Track>()); break;
        default: WarnL << "不支持该类型的视频编码类型:" << info.codecId; break;
    }
}

static void makeAdtsHeader(int profile, int sampleRate, int channels,uint8_t *out){
    AACFrame adtsHeader;
    adtsHeader.syncword = 0x0FFF;
    adtsHeader.id = 0;
    adtsHeader.layer = 0;
    adtsHeader.protection_absent = 1;
    adtsHeader.profile = profile;//audioObjectType - 1;
    int i = 0;
    for (auto rate : samplingFrequencyTable) {
        if (rate == sampleRate) {
            adtsHeader.sf_index = i;
        };
        ++i;
    }
    adtsHeader.private_bit = 0;
    adtsHeader.channel_configuration = channels;
    adtsHeader.original = 0;
    adtsHeader.home = 0;
    adtsHeader.copyright_identification_bit = 0;
    adtsHeader.copyright_identification_start = 0;
    adtsHeader.aac_frame_length = 7;
    adtsHeader.adts_buffer_fullness = 2047;
    adtsHeader.no_raw_data_blocks_in_frame = 0;
    writeAdtsHeader(adtsHeader, out);
}

void DevChannel::initAudio(const AudioInfo& info) {
    _audio = std::make_shared<AudioInfo>(info);
    switch (info.codecId) {
        case CodecAAC : {
            addTrack(std::make_shared<AACTrack>());
            makeAdtsHeader(info.iProfile, info.iSampleBit, info.iChannel, _adtsHeader);
            break;
        }

        case CodecG711A :
        case CodecG711U : addTrack(std::make_shared<G711Track>(info.codecId)); break;
        default: WarnL << "不支持该类型的音频编码类型:" << info.codecId; break;
    }
}

} /* namespace mediakit */