#include <stdio.h>
#include <string.h>
#include <pthread.h>

#include "./cnTTS.h"
#include "../Megaphone_TTS.h"
#include "../../Megaphone.h"

#include "AudioDeal/AudioDeal.h"

#include "JZsdkLib.h"

#include "version_choose.h"
#include "qtts.h"
#include "msp_cmn.h"
#include "msp_errors.h"
#include "msp_types.h"

#define TTS_SAVE 0

static T_JZsdkReturnCode TTS_Synthesis(const char* src_text, const char* params)
{
	int          ret          = -1;
	const char*  sessionID    = NULL;
	unsigned int audio_len    = 0;
	int          synth_status = MSP_TTS_FLAG_STILL_HAVE_DATA;

	unsigned char read_str[3*1276 + 3];

#if TTS_SAVE
	FILE *fp = fopen("/root/sdcard/tts_save.pcm", "wb+");
#endif

	if (src_text == NULL)
	{
		JZSDK_LOG_ERROR("params is error!\n");
		return JZ_ERROR_SYSTEM_MODULE_CODE_FAILURE;
	}

	/* 开始合成 */	
	sessionID = QTTSSessionBegin(params, &ret);

	JZSDK_LOG_INFO("%s",src_text);

	//将文本数据放入tts库
	ret = QTTSTextPut(sessionID, src_text, (unsigned int)strlen(src_text), NULL);
	if (MSP_SUCCESS != ret)
	{
		printf("QTTSTextPut failed, error code: %d.\n",ret);
		QTTSSessionEnd(sessionID, "TextPutError");
		return JZ_ERROR_SYSTEM_MODULE_CODE_FAILURE;
	}

	printf("正在合成 ...\n");

	while (Megaphone_MegDataGenFlag(JZ_FLAGCODE_GET, 0) == JZ_FLAGCODE_ON)
	{
		/* 获取合成音频 */
		//const void* data = QTTSAudioGet(sessionID, &audio_len, &synth_status, &ret);
		unsigned char *data = (unsigned char *)QTTSAudioGet(sessionID, &audio_len, &synth_status, &ret);
		if (ret != MSP_SUCCESS)
		{
			break;
		}


		if (data != NULL)
		{
#if TTS_SAVE
			fwrite((unsigned char *)data, sizeof(unsigned char), audio_len, fp); // 写入数据  
#endif
			// //降噪
			// if (1)
			// {
			// 	//降噪处理
			// 	//把数据组成一个数组
			// 	short TempPcm;     
			// 	for (int i = 0; i < audio_len; i+=2) 
			// 	{
			// 		TempPcm = ( data[i] & 0xFF) | ((short)data[i+1] << 8);
			// 		TempPcm = PcmNoiseReduction(TempPcm);
			// 		data[i] = TempPcm & 0xFF;
			// 		data[i+1] = (TempPcm >> 8) & 0xFF;

			// 		// TempPcm = (data[i+1] & 0xFF) | ((short)data[i] << 8);
			// 		// TempPcm = PcmNoiseReduction(TempPcm);
			// 		// data[i+1] = TempPcm & 0xFF;
			// 		// data[i] = (TempPcm >> 8) & 0xFF;
			// 	}
			// }

			//没将数据放入PCM通道
			//printf("产生了%d 的数据\n",audio_len);
			AudioDeal_PcmDataInput(16000, (unsigned char *)data, audio_len, 0);
		}

		if (synth_status == MSP_TTS_FLAG_DATA_END)
		{
			break;
		}	
	}
	
	if (MSP_SUCCESS != ret)
	{
		printf("QTTSAudioGet failed, error code: %d.\n",ret);
		QTTSSessionEnd(sessionID, "AudioGetError");
		return ret;
	}
	ret = QTTSSessionEnd(sessionID, "Normal");
	if (MSP_SUCCESS != ret)
	{
		printf("QTTSSessionEnd failed, error code: %d.\n",ret);
	}

#if TTS_SAVE
	fclose(fp);
#endif

}

/*************************
 * 
 *  获取音源
 * 
 * ********/
static T_JZsdkReturnCode TTS_GetSoundSource(int tone,unsigned char *name_tts, unsigned char *name_path, unsigned char *rule_path)
{

	/*
	* rdn:           合成音频数字发音方式
	* volume:        合成音频的音量
	* pitch:         合成音频的音调
	* speed:         合成音频对应的语速
	* voice_name:    合成发音人
	* sample_rate:   合成音频采样率
	* text_encoding: 合成文本编码格式
	*
	*/
	switch(tone){
		case 0x01://中文女生
			snprintf(name_tts,16, "xiaoyan");
			snprintf(name_path,64, "fo|/root/tts/xiaoyan.jet");
			break;
		case 0x02://中文男生
			snprintf(name_tts,16, "xiaofeng");
			snprintf(name_path,64, "fo|/root/tts/xiaofeng.jet");
			break;
		case 0x11://英文女生
			snprintf(name_tts,16, "catherine");
			snprintf(name_path,64, "fo|/root/tts/catherine.jet");
			break;
		case 0x12://英文男生
			snprintf(name_tts,16, "john");
			snprintf(name_path,64, "fo|/root/tts/john.jet");
			break;
		case 0x31://东北
			snprintf(name_tts,16, "xiaoqian");
			snprintf(name_path,64, "fo|/root/tts/xiaoqian.jet");
			break;
		case 0x32://四川
			snprintf(name_tts,16, "xiaorong");
			snprintf(name_path,64, "fo|/root/tts/xiaorong.jet");
			break;
		case 0x33://河南
			snprintf(name_tts,16, "xiaokun");
			snprintf(name_path,64, "fo|/root/tts/xiaokun.jet");
			break;
		case 0x34://湖南
			snprintf(name_tts,16, "xiaoqiang");
			snprintf(name_path,64, "fo|/root/tts/xiaoqiang.jet");
			break;	
		case 0x35://陕西
			snprintf(name_tts,16, "xiaoying");
			snprintf(name_path,64, "fo|/root/tts/xiaoying.jet");
			break;	
		case 0x36://广东
			snprintf(name_tts,16, "xiaomei");
			snprintf(name_path,64, "fo|/root/tts/xiaomei.jet");
			break;				
		default://默认中文女生
			snprintf(name_tts,16, "xiaoyan");
			snprintf(name_path,64, "fo|/root/tts/xiaoyan.jet");
			break;
	}

	//写入发音规则
	memset(rule_path,0,128);
	snprintf(rule_path,128,"/root/tts/common.jet");

	//不是英文就直接返回
	if ( tone != 0x11 && tone != 0x12)
	{
		return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
	}

	//由于有讯飞更新，让英文有了新的发音规则，所以要检测本地是否有新的发音规则
	if(JZsdk_check_file_exists("/root/tts/common_en.jet") == JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS)
	{
	    memset(rule_path,0,128);
		snprintf(rule_path,128,"/root/tts/common_en.jet");
	}

	return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
}

static void remove_letters(char *str) 
{
    if (str == NULL) return;

    int i = 0, j = 0;
    while (str[i] != '\0') {
        // 如果是多字节字符（如中文），直接保留（UTF-8 的高位为 1）
        if ((str[i] & 0x80) != 0) {
            // 复制完整的 UTF-8 字符（可能占 2~4 字节）
            do {
                str[j++] = str[i++];
            } while ((str[i] & 0xC0) == 0x80); // 继续复制后续字节
        }
        // 如果是 ASCII 字符且不是字母，则保留
        else if (!isalpha(str[i])) {
            str[j++] = str[i++];
        }
        // 如果是字母，跳过
        else {
            i++;
        }
    }
    str[j] = '\0'; // 终止字符串
}

static T_JZsdkReturnCode TTS_FILE_GenerateAndPlay(struct t_Megaphone_tts_param *Megaphone_tts_param)
{
	int tone = Megaphone_tts_param->tts_tone;
	int volume = Megaphone_tts_param->tts_volume;
	int speed = Megaphone_tts_param->tts_speed;
	char *argv = Megaphone_tts_param->ttsdata;
	int tts_add_mode = Megaphone_tts_param->tts_add;
	int ReplyFlag = Megaphone_tts_param->ReplyFlag;

	int  ret = MSP_SUCCESS;
	char name_tts[16];
	char name_path[64];

	char session_begin_params[512];
	char tts_add_params[512];
	char rule[128];

	//获取音源，以及规则
	TTS_GetSoundSource(tone, name_tts, name_path, rule);

	//如果是方言和男生，则去掉字母 或者追加音色里有男生
	if (tone == 0x31 || tone == 0x32 || tone == 0x33 || tone == 0x34 || tone == 0x35 || tone == 0x36 || tone == 0x02
		|| tts_add_mode == 2)
	{
		remove_letters(argv);
	}

	//配置合成参数
	snprintf(session_begin_params,512,
			"engine_type = local,voice_name=%s, text_encoding = UTF8, tts_res_path = %s;fo|%s, sample_rate = 16000, speed = %d, volume = %d, pitch = 50, rdn = 2"
			,name_tts , name_path, rule, speed, volume);

	printf("session_begin_params=%s\n",session_begin_params);
	printf("argv=%s\n",argv);

	//如果开启了语音追加
	if (tts_add_mode == 1)
	{
		snprintf(tts_add_params,512,
			"engine_type = local,voice_name=xiaoyan, text_encoding = UTF8, tts_res_path = fo|/root/tts/xiaoyan.jet;fo|/root/tts/common.jet, sample_rate = 16000, speed = %d, volume = 100, pitch = 50, rdn = 2"
			,speed);
	}
	else if (tts_add_mode == 2)
	{
		snprintf(tts_add_params,512,
			"engine_type = local,voice_name=xiaofeng, text_encoding = UTF8, tts_res_path = fo|/root/tts/xiaofeng.jet;fo|/root/tts/common.jet, sample_rate = 16000, speed = %d, volume = 100, pitch = 50, rdn = 2"
			,speed);
	}

	/* 文本合成 */
	printf("开始合成 ...\n");

	//合成文本音乐
	int loop;
	int loop_interval;
	do
	{
		
		TTS_Synthesis(argv, session_begin_params);
		if (tts_add_mode != 0)
		{
			TTS_Synthesis(argv, tts_add_params);
		}

		Megaphone_param(JZ_FLAGCODE_GET, MEGAPHONE_LOOP, &loop);
		Megaphone_param(JZ_FLAGCODE_GET, MEGAPHONE_LOOP_INTERVAL, &loop_interval);

		//循环播放延时
		while (loop_interval > 0 && (loop == JZ_FLAGCODE_ON) && (Megaphone_MegDataGenFlag(JZ_FLAGCODE_GET, 0) == JZ_FLAGCODE_ON))
		{
			delayMs(50);
			loop_interval -= 50;
		}
		
	} while ( (loop == JZ_FLAGCODE_ON) && (Megaphone_MegDataGenFlag(JZ_FLAGCODE_GET, 0) == JZ_FLAGCODE_ON));
	
	return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
}



//TTS合成线程
static void *Play_cntts_pthread(void *arg)
{
	struct t_Megaphone_tts_param *Megaphone_tts_param = (struct t_Megaphone_tts_param *) arg;

	int ReplyFlag = Megaphone_tts_param->ReplyFlag;

	//设置标志位
	Megaphone_MegDataGenFlag(JZ_FLAGCODE_SET, JZ_FLAGCODE_ON);
	Megaphone_MegDataGenFinshFlag(JZ_FLAGCODE_SET, JZ_FLAGCODE_ON);

	//广播状态开始
	if (ReplyFlag == JZ_FLAGCODE_ON)
	{
		Megaphone_status_Broadcast(AUDIO_PLAY_STATUS_IFLAY_TTS);
	}
	
	int amplifier = JZ_FLAGCODE_ON;
    Megaphone_Amplifier_param(JZ_FLAGCODE_SET, &amplifier);

	//修改tts前的音量
	Megaphone_SetVolume(0, 1);

	//生成tts播放文件
	TTS_FILE_GenerateAndPlay(Megaphone_tts_param);

	//释放掉结构体和标志位
	JZsdk_Free(Megaphone_tts_param);

	//恢复音量
	int audio_volume = Megaphone_GetUiVolume();
	Megaphone_SetVolume(audio_volume,0);

	amplifier = JZ_FLAGCODE_OFF;
    Megaphone_Amplifier_param(JZ_FLAGCODE_SET, &amplifier);

	Megaphone_MegDataGenFlag(JZ_FLAGCODE_SET, JZ_FLAGCODE_OFF);
	Megaphone_MegDataGenFinshFlag(JZ_FLAGCODE_SET, JZ_FLAGCODE_OFF);

	//广播关闭
	if (ReplyFlag == JZ_FLAGCODE_ON)
	{
		Megaphone_status_Broadcast(JZ_FLAGCODE_OFF);
	}

	JZSDK_LOG_INFO("TTS线程结束");
}


/*********
 * 
 *  tts注册
 * 
 * 
**********/
static void login_tts()
{
	int         ret                  = MSP_SUCCESS;
	int error_num=0;
	const char* login_params         = "appid = 03857dfd, work_dir = .";//登录参数,appid与msc库绑定,请勿随意改动
relogin:
	/* 用户登录 */
	ret = MSPLogin(NULL, NULL, login_params); //第一个参数是用户名，第二个参数是密码，第三个参数是登录参数，用户名和密码可在http://www.xfyun.cn注册获取
	if (MSP_SUCCESS != ret&&error_num<10)
	{
		error_num+=1;
		printf("TTS Login failed, error code: %d num %d.\n", ret,error_num);
		goto relogin ;//登录失败，重新登录
	}
	else{
		printf("TTS login\n");
	}
	return;	
}

/*********
 * 
 *  tts初始化
 * 
**********/
T_JZsdkReturnCode Megaphone_cnTTS_Init()
{
	login_tts();
}



/*********
 * 
 *  讯飞TTS_播放函数
 * 
**********/
T_JZsdkReturnCode Megaphone_cnTTS_Play(unsigned char *str, unsigned int str_len, int ReplyFlag)
{
	//1、制造一个数据产生函数
	pthread_t TTS_flie_task;
	pthread_attr_t task_attribute; //线程属性
	pthread_attr_init(&task_attribute);  //初始化线程属性
	pthread_attr_setdetachstate(&task_attribute, PTHREAD_CREATE_DETACHED);      //设置线程属性

	//tts参数结构体
	struct t_Megaphone_tts_param *Megaphone_tts_param = NULL;
	if(JZsdk_Malloc( (void **)&Megaphone_tts_param, sizeof(struct t_Megaphone_tts_param)) != JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS)//如果申请内存失败，malloc会返回一个NULl
	{
		printf("malloc use failure");
		return JZ_ERROR_SYSTEM_MODULE_CODE_FAILURE;//申请内存失败，错误退出
	}

	//清空参数
	memset (Megaphone_tts_param,0,sizeof(struct t_Megaphone_tts_param));

	//写入参数
	Megaphone_tts_param->tts_speed = TTS_Get_speed();
	Megaphone_tts_param->tts_tone =  TTS_Get_tone();
	Megaphone_tts_param->tts_volume = TTS_Get_NormalVolume();
	Megaphone_tts_param->tts_add = TTS_Get_TTS_AppendTone();
	Megaphone_tts_param->ReplyFlag = ReplyFlag;
	memset(Megaphone_tts_param->ttsdata,0,sizeof(Megaphone_tts_param->ttsdata));
	memcpy(Megaphone_tts_param->ttsdata, str, str_len);

	int tts_ret = pthread_create(&TTS_flie_task,&task_attribute,Play_cntts_pthread,(void *)Megaphone_tts_param);		//TTS线程
	if(tts_ret != 0)
	{
		printf("创建TTS线程失败\n");
		return JZ_ERROR_SYSTEM_MODULE_CODE_FAILURE;
	}

	return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
}