/*
 * Copyright (c) 2020-2021 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "audio_encoder.h"
#include "media_log.h"
#include "securec.h"

namespace OHOS {
namespace Audio {
using namespace OHOS::Media;

constexpr uint32_t AUDIO_READ_STREAM_TIME_OUT_MS = 1000; /* 1S */

constexpr uint32_t AUDIO_CHANNEL_MONO = 1;
constexpr uint32_t AUDIO_CHANNEL_STEREO = 2;

AudioEncoder::AudioEncoder()
    :initialized_(false),
    encHandle_(nullptr),
    started_(false)
{
    for (int i = 0; i < AUDIO_ENC_PARAM_NUM; i++) {
        encAttr_[i] = {};
    }
    CodecInit();
    MEDIA_INFO_LOG("AudioEncoder ctor");
}

AudioEncoder::~AudioEncoder()
{
    if (initialized_) {
        Release();
    }
    MEDIA_INFO_LOG("AudioEncoder dtor");
}

static bool IsAudioCodecFormatSupported(AudioCodecFormat format)
{
    if ((format < AAC_LC) || (format > G726)) {
        MEDIA_ERR_LOG("Invalid format:%d", format);
        return false;
    }
    return true;
}

static bool IsAudioSampleRateSupported(AudioCodecFormat format, uint32_t sampleRate)
{
    if (format == G711A || format == G711U || format == G726) {
        if (sampleRate != 8000) { // 8000 G711/G726 only support 8kHz
            MEDIA_ERR_LOG("Invalid sampleRate:%d, G711/G726 only support 8kHz", sampleRate);
            return false;
        }
    }
    return true;
}

static AvCodecMime GetMimeFromAudioCodecFormat(AudioCodecFormat format)
{
    switch (format) {
        case AAC_LC:
        case AAC_HE_V1:
        case AAC_HE_V2:
        case AAC_LD:
        case AAC_ELD:
            return MEDIA_MIMETYPE_AUDIO_AAC;
        case G711A:
            return MEDIA_MIMETYPE_AUDIO_G711A;
        case G711U:
            return MEDIA_MIMETYPE_AUDIO_G711U;
        case G726:
            return MEDIA_MIMETYPE_AUDIO_G726;
        default:
            MEDIA_ERR_LOG("Invalid format:0x%x", format);
            return MEDIA_MIMETYPE_INVALID;
    }
}

static Profile GetProfileFromAudioCodecFormat(AudioCodecFormat format)
{
    switch (format) {
        case AAC_LC:
            return AAC_LC_PROFILE;
        case AAC_HE_V1:
            return AAC_HE_V1_PROFILE;
        case AAC_HE_V2:
            return AAC_HE_V2_PROFILE;
        case AAC_LD:
            return AAC_LD_PROFILE;
        case AAC_ELD:
            return AAC_ELD_PROFILE;
        default:
            MEDIA_ERR_LOG("Invalid format: 0x%x", format);
            return AAC_LC_PROFILE;
    }
}

static AudioSoundMode ConvertSoundMode(uint32_t channelCount)
{
    switch (channelCount) {
        case AUDIO_CHANNEL_MONO:
            return AUD_SOUND_MODE_MONO;
        case AUDIO_CHANNEL_STEREO:
            return AUD_SOUND_MODE_STEREO;
        default:
            MEDIA_ERR_LOG("Invalid soundMode:%u", channelCount);
            return AUD_SOUND_MODE_MONO;
    }
}

void AudioEncoder::setEncAttrValue(const AudioEncodeConfig &config)
{
    uint32_t paramIndex = 0;
    domainKind_ = AUDIO_ENCODER;
    encAttr_[paramIndex].key = KEY_CODEC_TYPE;
    encAttr_[paramIndex].val = &domainKind_;
    encAttr_[paramIndex].size = sizeof(CodecType);
    paramIndex++;
    codecMime_ = GetMimeFromAudioCodecFormat(config.audioFormat);
    encAttr_[paramIndex].key = KEY_MIMETYPE;
    encAttr_[paramIndex].val = &codecMime_;
    encAttr_[paramIndex].size = sizeof(AvCodecMime);
    paramIndex++;
    profile_ = GetProfileFromAudioCodecFormat(config.audioFormat);
    encAttr_[paramIndex].key = KEY_AUDIO_PROFILE;
    encAttr_[paramIndex].val = &profile_;
    encAttr_[paramIndex].size = sizeof(Profile);
    paramIndex++;
    sampleRate_ = config.sampleRate;
    encAttr_[paramIndex].key = KEY_AUDIO_SAMPLE_RATE;
    encAttr_[paramIndex].val = &sampleRate_;
    encAttr_[paramIndex].size = sizeof(uint32_t);
    paramIndex++;
    bitRate_ = config.bitRate;
    encAttr_[paramIndex].key = KEY_BITRATE;
    encAttr_[paramIndex].val = &bitRate_;
    encAttr_[paramIndex].size = sizeof(uint32_t);
    paramIndex++;
    soundMode_ = ConvertSoundMode(config.channelCount);
    encAttr_[paramIndex].key = KEY_AUDIO_SOUND_MODE;
    encAttr_[paramIndex].val = &soundMode_;
    encAttr_[paramIndex].size = sizeof(AudioSoundMode);
    paramIndex++;
    ptNumPerFrm_ = AUDIO_POINT_NUM;
    encAttr_[paramIndex].key = KEY_AUDIO_POINTS_PER_FRAME;
    encAttr_[paramIndex].val = &ptNumPerFrm_;
    encAttr_[paramIndex].size = sizeof(uint32_t);
    paramIndex++;
    bufSize_ = AUDIO_FRAME_NUM_IN_BUF;
    encAttr_[paramIndex].key = KEY_BUFFERSIZE;
    encAttr_[paramIndex].val = &bufSize_;
    encAttr_[paramIndex].size = sizeof(uint32_t);
}

int32_t AudioEncoder::InitAudioEncoderAttr(const AudioEncodeConfig &config)
{
    if (!IsAudioCodecFormatSupported(config.audioFormat)) {
        MEDIA_ERR_LOG("audioFormat:0x%x is not support", config.audioFormat);
        return ERR_INVALID_PARAM;
    }
    if (!IsAudioSampleRateSupported(config.audioFormat, config.sampleRate)) {
        MEDIA_ERR_LOG("audioFormat:%d is not support sampleRate:%d", config.audioFormat, config.sampleRate);
        return ERR_INVALID_PARAM;
    }
    setEncAttrValue(config);

    return SUCCESS;
}

int32_t AudioEncoder::Initialize(const AudioEncodeConfig &config)
{
    int32_t ret = InitAudioEncoderAttr(config);
    if (ret != SUCCESS) {
        MEDIA_ERR_LOG("InitAudioEncoderAttr failed:%d", ret);
        return ret;
    }
    ret = CodecCreateByType(domainKind_, codecMime_, &encHandle_);
    if (ret != SUCCESS) {
        MEDIA_ERR_LOG("CodecCreateByType failed:0x%x", ret);
        return ret;
    }
    ret = CodecSetParameter(encHandle_, encAttr_, AUDIO_ENC_PARAM_NUM);
    if (ret != SUCCESS) {
        CodecDestroy(encHandle_);
        MEDIA_ERR_LOG("CodecSetParameter failed:0x%x", ret);
        return ret;
    }
    initialized_ = true;
    return SUCCESS;
}

int32_t AudioEncoder::BindSource(uint32_t deviceId)
{
    Param params[1];
    (void)memset_s(params, sizeof(params), 0x00, sizeof(params));
    params[0].key = KEY_DEVICE_ID;
    params[0].val = reinterpret_cast<void *>(&deviceId);
    params[0].size = sizeof(uint32_t);
    int32_t ret = CodecSetParameter(encHandle_, params, sizeof(params) / sizeof(params[0]));
    if (ret != SUCCESS) {
        MEDIA_ERR_LOG("CodecSetDevice:0x%x", ret);
        return ret;
    }
    return SUCCESS;
}

int32_t AudioEncoder::GetMute(bool &muted)
{
    (void)muted;
    return ERR_UNKNOWN;
}

int32_t AudioEncoder::SetMute(bool muted)
{
    (void)muted;
    return ERR_UNKNOWN;
}

int32_t AudioEncoder::Start()
{
    if (!initialized_) {
        MEDIA_ERR_LOG("not initialized");
        return ERR_ILLEGAL_STATE;
    }
    int32_t ret = CodecStart(encHandle_);
    if (ret != SUCCESS) {
        MEDIA_ERR_LOG("CodecStart failed:0x%x", ret);
        return ret;
    }
    started_ = true;
    return ret;
}

int32_t AudioEncoder::ReadStream(AudioStream &stream, bool isBlockingRead)
{
    if (!started_) {
        MEDIA_ERR_LOG("Codec not Started");
        return ERR_INVALID_READ;
    }
    if (stream.buffer == nullptr || stream.bufferLen == 0) {
        MEDIA_ERR_LOG("stream.buffer is nullptr");
        return ERR_INVALID_READ;
    }
    uint32_t timeoutMs;
    if (isBlockingRead) {
        timeoutMs = AUDIO_READ_STREAM_TIME_OUT_MS;
    } else {
        timeoutMs = 0;
    }
    AudioBufferInfo outInfo;
    int32_t ret = CodecDequeueOutput(encHandle_, timeoutMs, nullptr, (CodecBuffer *)&outInfo);
    if (ret != SUCCESS && outInfo.info.buffer[0].buf == 0) {
        MEDIA_ERR_LOG("CodecDequeueOutput failed:0x%x", ret);
        return ERR_INVALID_READ;
    }
    int32_t readLen = 0;
    errno_t retCopy = memcpy_s(stream.buffer, stream.bufferLen, (void *)outInfo.info.buffer[0].buf,
                               outInfo.info.buffer[0].length);
    if (retCopy != EOK) {
        MEDIA_ERR_LOG("memcpy_s failed, timeStamp:%lld, retCopy:0x%x", outInfo.info.timeStamp, retCopy);
        return ERR_INVALID_OPERATION;
    } else {
        readLen = outInfo.info.buffer[0].length;
    }
    stream.timeStamp = outInfo.info.timeStamp;
    (void)CodecQueueOutput(encHandle_, (CodecBuffer *)&outInfo, timeoutMs, -1);
    return readLen;
}

int32_t AudioEncoder::Stop()
{
    MEDIA_DEBUG_LOG("AudioEncoder::Stop");
    if (!started_) {
        MEDIA_ERR_LOG("Codec not Start");
    }
    return CodecStop(encHandle_);
}

int32_t AudioEncoder::Release()
{
    MEDIA_DEBUG_LOG("AudioEncoder::Release");
    if (!initialized_) {
        MEDIA_ERR_LOG("Codec not initialized");
    }
    if (encHandle_ != nullptr) {
        CodecDestroy(encHandle_);
        encHandle_ = nullptr;
    }
    initialized_ = false;
    return SUCCESS;
}
}  // namespace Audio
}  // namespace OHOS