#include <stdio.h>
#include <pthread.h>
#include <alsa/asoundlib.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "FF_Statement.h"
#include "./AudioDeal.h"
#include "JZsdkLib.h"
#include "version_choose.h"


#include "./AudioDealThread/AudioDealThread.h"
#include "./Alsa/pcm_AlsaPlay.h"

#include "Resample/pcm_Resample.h"
#include "AudioStreamDeal/AudioStreamDeal.h"
#include "Filter/FF_Filter.h"

#ifdef SPEEX_STATUS_ON
    #include "Speex/Speex.h"
#endif

#define MP3_DIR "/root/sdcard/1.mp3"
#define PCM_16000_DIR "/root/sdcard/16000_test.pcm"
#define READ_SIZE (2000)

//音频库索引值，用于管理所有的音频库信息
struct AudioDealInfo *AudioDeakInfo_index = NULL; 
T_JZsdkReturnCode g_AudioDealPauseFlag = JZ_FLAGCODE_OFF;
static int Audiodeal_status = JZ_FLAGCODE_OFF;

T_JZsdkReturnCode AudioDeal_Init()
{
    T_JZsdkReturnCode ret;
    
    ret = JZsdk_Malloc((void **)&AudioDeakInfo_index, sizeof(struct AudioDealInfo));

    //初始化参数
    AudioDeakInfo_index->AudioDeal_ResampleAndFilter_Execute_Flag = JZ_FLAGCODE_OFF;
    AudioDeakInfo_index->AudioDeal_ResampleAndFilterAndPlay_Finish_Flag = JZ_FLAGCODE_OFF;

    AudioDeakInfo_index->Target_SampleRate = 44100;                  //cd采样率
    AudioDeakInfo_index->Target_SampleFormat = AV_SAMPLE_FMT_S16;    //16位深

    av_channel_layout_copy(&(AudioDeakInfo_index->Target_ChannelLayout), &(AVChannelLayout)AV_CHANNEL_LAYOUT_STEREO);
    //AudioDeakInfo_index->Target_ChannelLayout = AV_CH_LAYOUT_STEREO; //立体声道
    //AudioDeakInfo_index->Target_ChannelLayout = AV_CH_LAYOUT_MONO; //dan声道

    AudioDeakInfo_index->Raw_SampleRate = 0;

    AudioDeakInfo_index->FilterMode = JZ_FLAGCODE_ON;  //滤波模式

    //JZSDK_LOG_DEBUG("thread_init开始");

    //初始化thread线程管理
    AudioDealThread_Init(AudioDeakInfo_index);

    AudioDeakInfo_index->AlsaInfo = NULL;
    AlsaPlay_Init(AudioDeakInfo_index);

    JZSDK_LOG_DEBUG("初始化重采样器开始");

    //初始化重采样器, 以下为默认文本的参数
    AudioDeakInfo_index->ResampleInfo = NULL;

    FF_Resample_Init(AudioDeakInfo_index, 16000, (AVChannelLayout)AV_CHANNEL_LAYOUT_STEREO, AV_SAMPLE_FMT_S16);

    JZSDK_LOG_DEBUG("初始化滤波器开始");

    //初始化滤波器
    AudioDeakInfo_index->FilterInfo = NULL;
    FF_Filter_Init(AudioDeakInfo_index, 0x00);

    //初始化实时流
    File_Stream_deal_Init(AV_CODEC_ID_MP2);

    Audiodeal_status = JZ_FLAGCODE_ON;

    JZSDK_LOG_INFO("MODULE_AUDIODEL_INIT_COMPLETE");
}

/*

    设置滤波模式
    0x00 默认滤波 0x01 M30滤波


*/
T_JZsdkReturnCode AudioDeal_SetFilterMode(int mode)
{
    FF_Filter_Set_FilterMode(mode);
}

/*
    获取音频库的滤波模式
    0x00 默认滤波 0x01 M30滤波
*/
T_JZsdkReturnCode AudioDeal_GetFilterMode()
{
    return FF_Filter_Get_FilterMode();
}


/******************************
 * 
 *  采样器重初始化
 * 
 * 
 * *********************************************/
T_JZsdkReturnCode AudioDeal_ResampleRest(unsigned int in_sampleRate, AVChannelLayout in_ch_layout, enum AVSampleFormat in_format)
{

    FF_Resample_Reset(AudioDeakInfo_index ,in_sampleRate, in_ch_layout, in_format);

}

/***************************************************************************************************************************************************
 * 
 *  各音频播放接口
 *  
 * 
 * ***************************************************************************************************************************************************/

/*************
 * 
 *  文本类型的pcm数据导入接口
 *  
 *  区别
 *      a 使用文本专用的滤波
 * 
 * **********************/
T_JZsdkReturnCode AudioDeal_PcmDataInput_TextSteam(int In_Bitrate, unsigned char *buffer, int bytesRead)
{
    if (AudioDeakInfo_index == NULL)
    {
        JZSDK_LOG_ERROR("音频处理器未注册");
        return JZ_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
    }

    while (Audiodeal_status != JZ_FLAGCODE_ON)
    {
        delayMs(1);
    }

    int DateType = 0x00;

    //1、打开音频库的处理标志位，该标志可以重复打开，只有强制关闭音频时，需要关闭该标志
    AudioDeakInfo_index->AudioDeal_ResampleAndFilter_Execute_Flag = JZ_FLAGCODE_ON;

    //默认数据pcm接入口都是 单声道，16位
    PCM_PooL_Interface_PcmData(AudioDeakInfo_index, In_Bitrate, (AVChannelLayout)AV_CHANNEL_LAYOUT_MONO, AV_SAMPLE_FMT_S16, buffer, bytesRead, DateType);
    //printf("Read %zu bytes from the PCM file.\n", bytesRead);

    return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
}


/*************
 * 
 *  录音语音类型的pcm数据导入接口
 *  
 *  区别
 *      a 使用语音专用的滤波
 * 
 * **********************/
T_JZsdkReturnCode AudioDeal_PcmDataInput_RecordSteam(int In_Bitrate, unsigned char *buffer, int bytesRead)
{
    if (AudioDeakInfo_index == NULL)
    {
        JZSDK_LOG_ERROR("音频处理器未注册");
        return JZ_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
    }

    while (Audiodeal_status != JZ_FLAGCODE_ON)
    {
        delayMs(1);
    }

    int DateType = 0x01;

    //1、打开音频库的处理标志位，该标志可以重复打开，只有强制关闭音频时，需要关闭该标志
    AudioDeakInfo_index->AudioDeal_ResampleAndFilter_Execute_Flag = JZ_FLAGCODE_ON;

    //默认数据pcm接入口都是 单声道，16位
    PCM_PooL_Interface_PcmData(AudioDeakInfo_index, In_Bitrate, (AVChannelLayout)AV_CHANNEL_LAYOUT_MONO, AV_SAMPLE_FMT_S16, buffer, bytesRead, DateType);
    
    return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
}

/*************
 * 
 *  实时语音语音类型的pcm数据导入接口
 *  
 *  区别
 *      a 使用语音专用的滤波
 *      b 需要进行回音消除
 * 
 * **********************/
T_JZsdkReturnCode AudioDeal_PcmDataInput_RealTimeSteam(int In_Bitrate, unsigned char *buffer, int bytesRead)
{
    if (AudioDeakInfo_index == NULL)
    {
        JZSDK_LOG_ERROR("音频处理器未注册");
        return JZ_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
    }

    while (Audiodeal_status != JZ_FLAGCODE_ON)
    {
            delayMs(1);
    }

    int DateType = 0x02;

    //1、打开音频库的处理标志位，该标志可以重复打开，只有强制关闭音频时，需要关闭该标志
    AudioDeakInfo_index->AudioDeal_ResampleAndFilter_Execute_Flag = JZ_FLAGCODE_ON;

    //将数据拿去进行回音消除
    //将数据合并为u16型

#ifdef SPEEX_STATUS_ON

    short TempPcm[bytesRead / 2];
    short OutPcm[bytesRead / 2];
    for (int i = 0; i < bytesRead / 2; i++)
    {
        TempPcm[i] = (buffer[2*i] & 0xFF) | (((short)buffer[2*i + 1]) << 8);
    }

    //送去回音消除
    T_JZsdkReturnCode ret = Speex_DealData(TempPcm, OutPcm, bytesRead / 2);
    // if (ret == JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS)
    // {
    //     //将输出的pcm还原回u8型到数据缓冲区
    //     for (int i = 0; i < bytesRead / 2; i++)
    //     {
    //         buffer[2*i] = (OutPcm[i] & 0xFF);
    //         buffer[2*i + 1] = ((OutPcm[i] >> 8) & 0xFF);
    //     } 
    // }

#endif

    //默认数据pcm接入口都是 单声道，16位
    PCM_PooL_Interface_PcmData(AudioDeakInfo_index, In_Bitrate, (AVChannelLayout)AV_CHANNEL_LAYOUT_MONO, AV_SAMPLE_FMT_S16, buffer, bytesRead, DateType);

    return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
}



// /***************************
//  * 
//  *  无回复接口
//  * 
//  * ************/
// T_JZsdkReturnCode AudioDeal_PcmDataInput_WithoutReply(int In_Bitrate, unsigned char *buffer, int bytesRead)
// {
//     if (AudioDeakInfo_index == NULL)
//     {
//         JZSDK_LOG_ERROR("音频处理器未注册");
//         return JZ_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
//     }

//     while (Audiodeal_status != JZ_FLAGCODE_ON)
//     {
//         delayMs(1);
//     }

//     //1、打开音频库的处理标志位，该标志可以重复打开，只有强制关闭音频时，需要关闭该标志
//     AudioDeakInfo_index->AudioDeal_ResampleAndFilter_Execute_Flag = JZ_FLAGCODE_ON;
    
//     //默认数据pcm接入口都是 单声道，16位
//     PCM_PooL_Interface_PcmData(AudioDeakInfo_index, In_Bitrate, (AVChannelLayout)AV_CHANNEL_LAYOUT_MONO, AV_SAMPLE_FMT_S16, buffer, bytesRead);

//     return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
// }



/***************************
 * 
 *  mp3数据接入接口
 * 
 * ************/
T_JZsdkReturnCode AudioDeal_Mp3DataInput(int In_Bitrate, unsigned char *buffer, int bytesRead)
{
    if (AudioDeakInfo_index == NULL)
    {
        JZSDK_LOG_ERROR("音频处理器未注册");
        return JZ_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
    }
    
    while (Audiodeal_status != JZ_FLAGCODE_ON)
    {
        delayMs(1);
    }

    //1、打开音频库的处理标志位，该标志可以重复打开，只有强制关闭音频时，需要关闭该标志
    AudioDeakInfo_index->AudioDeal_ResampleAndFilter_Execute_Flag = JZ_FLAGCODE_ON;

    T_JZsdkReturnCode ret =  mp3_Stream_Interface_Mp3Data(AudioDeakInfo_index, In_Bitrate, buffer, bytesRead);
    if (ret == JZ_ERROR_SYSTEM_MODULE_CODE_HARDLY_FAILURE)
    {
        return JZ_ERROR_SYSTEM_MODULE_CODE_HARDLY_FAILURE;
    }

    return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
}

T_JZsdkReturnCode AudioDeal_Mp3DataStop()
{
    mp3_Stream_Interface_Mp3Data_Stop(AudioDeakInfo_index);
}

/*******************
 * 
 *  暂停播放
 *  jzon 为暂停播放
 *  jzoff 为继续播放
 * *********************/
T_JZsdkReturnCode AudioDeal_PauseAndContinuePlay(int status)
{
    //如果处于实时喊话，禁止使用该功能
    g_AudioDealPauseFlag = status;
}

/*******************
 * 
 *  获取音频暂停状态
 *  jzon 为暂停播放
 *  jzoff 为继续播放
 * 
 * *********************/
int AudioDeal_GetPauseAndContinuePlayStatus()
{
    return g_AudioDealPauseFlag;
}

/***************************
 * 
 *  音频文件数据接入接口
 * 
 * ************/
T_JZsdkReturnCode AudioFile_Stream_DataInput(AVFrame *frame)
{
    if (AudioDeakInfo_index == NULL)
    {
        JZSDK_LOG_ERROR("音频处理器未注册");
        return JZ_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
    }

    while (Audiodeal_status != JZ_FLAGCODE_ON)
    {
        delayMs(1);
    }
    
    //1、打开音频库的处理标志位，该标志可以重复打开，只有强制关闭音频时，需要关闭该标志
    AudioDeakInfo_index->AudioDeal_ResampleAndFilter_Execute_Flag = JZ_FLAGCODE_ON;

    AudioFile_Stream_Interface_PcmData(AudioDeakInfo_index, frame);

    return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
}

/*********************
 * 
 *  音频库停止处理函数
 *  
 * *********************/
T_JZsdkReturnCode AudioDeal_StopDeal()
{
    if (AudioDeakInfo_index == NULL)
    {
        JZSDK_LOG_ERROR("音频处理器未注册");
        return JZ_ERROR_SYSTEM_MODULE_CODE_INVALID_PARAMETER;
    }

    //1、关闭数据处理和alsa部分
    AudioDeakInfo_index->AudioDeal_ResampleAndFilter_Execute_Flag = JZ_FLAGCODE_OFF;
    AudioDeakInfo_index->AudioDeal_Alsa_Execute_Flag = JZ_FLAGCODE_OFF;

    //2、等候完成
    while (AudioDeakInfo_index->AudioDeal_ResampleAndFilterAndPlay_Finish_Flag != JZ_FLAGCODE_OFF)
    {
        delayMs(1);
    }

    //清空alsa里的缓冲区
    Alsa_DropPcm(AudioDeakInfo_index);

    while (AudioDeakInfo_index->AudioDeal_Alsa_Finish_Flag != JZ_FLAGCODE_OFF)
    {
        delayMs(1);
    }

    return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
}

/**************
 * 
 *  获取文件的时长信息
 * 
 * ***********/
T_JZsdkReturnCode AudioDeal_GetAudio_DurationInfo(JZ_U8 *FilePath, JZ_U32 *Duration) 
{
    int ret;
    AVFormatContext *fmt_ctx = NULL;

    fmt_ctx = avformat_alloc_context();
    if (!fmt_ctx) {
        JZSDK_LOG_ERROR("无法分配AVFormatContext");
        return JZ_ERROR_SYSTEM_MODULE_CODE_FAILURE;
    }

    ret = avformat_open_input(&fmt_ctx, FilePath, NULL, NULL);
    if (ret < 0) {
        JZSDK_LOG_ERROR("无法打开音频文件");
        goto cleanUp;
    }

    // if (avformat_find_stream_info(fmt_ctx, NULL) < 0) {
    //     JZSDK_LOG_ERROR("检索音频流信息失败");
    //     goto cleanUp;
    // }

    // if (fmt_ctx->duration <= 0) {
    //     JZSDK_LOG_ERROR("无效的音频时长");
    //     goto cleanUp;
    // }

    // *Duration = (uint32_t)(fmt_ctx->duration / AV_TIME_BASE);
    

    avformat_close_input(&fmt_ctx);
    
    return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;

cleanUp:
    if (fmt_ctx) {
        avformat_close_input(&fmt_ctx);
    }
    *Duration = 0;  // 确保失败时Duration被清零
    return JZ_ERROR_SYSTEM_MODULE_CODE_FAILURE;
}

/**************
 * 
 *  获取音频库的处理状态
 * 
 * ***********/
T_JZsdkReturnCode Get_AudioDealStatus()
{
    return Get_AudioDealThreadStatus();
}

/**************
 * 
 *  回音抑制初始化
 * 
 * ***********/
T_JZsdkReturnCode AudioDeal_EchoCancellation_Init(int SampleRate)
{
#ifdef SPEEX_STATUS_ON

    Speex_Init(SampleRate);
#endif

    return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
}