/* Copyright 2023 Dual Tachyon
 * https://github.com/DualTachyon
 *
 * 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.
 */

#ifdef ENABLE_FMRADIO
#include "app/fm.h"
#endif
#include "audio.h"
#include "bsp/dp32g030/gpio.h"
#ifdef ENABLE_FMRADIO
#include "driver/bk1080.h"
#endif
#include "driver/bk4819.h"
#include "driver/gpio.h"
#include "driver/system.h"
#include "driver/systick.h"
#include "functions.h"
#include "misc.h"
#include "settings.h"
#include "ui/ui.h"


BEEP_Type_t gBeepToPlay = BEEP_NONE;

void AUDIO_PlayBeep(BEEP_Type_t Beep)
{

    if (Beep != BEEP_880HZ_60MS_TRIPLE_BEEP &&
        Beep != BEEP_500HZ_60MS_DOUBLE_BEEP &&
        Beep != BEEP_440HZ_500MS &&
        Beep != BEEP_880HZ_200MS &&
        Beep != BEEP_880HZ_500MS &&
        !gEeprom.BEEP_CONTROL)
        return;

#ifdef ENABLE_AIRCOPY
    if (gScreenToDisplay == DISPLAY_AIRCOPY)
		return;
#endif

    if (gCurrentFunction == FUNCTION_RECEIVE)
        return;

    if (gCurrentFunction == FUNCTION_MONITOR)
        return;

#ifdef ENABLE_FMRADIO
    if (gFmRadioMode)
		BK1080_Mute(true);
#endif

    AUDIO_AudioPathOff();

    if (gCurrentFunction == FUNCTION_POWER_SAVE && gRxIdleMode)
        BK4819_RX_TurnOn();

    SYSTEM_DelayMs(20);

    uint16_t ToneConfig = BK4819_ReadRegister(BK4819_REG_71);

    uint16_t ToneFrequency;
    switch (Beep)
    {
        default:
        case BEEP_NONE:
            ToneFrequency = 220;
            break;
        case BEEP_1KHZ_60MS_OPTIONAL:
            ToneFrequency = 1000;
            break;
        case BEEP_500HZ_60MS_DOUBLE_BEEP_OPTIONAL:
        case BEEP_500HZ_60MS_DOUBLE_BEEP:
            ToneFrequency = 500;
            break;
        case BEEP_440HZ_40MS_OPTIONAL:
        case BEEP_440HZ_500MS:
            ToneFrequency = 440;
            break;
        case BEEP_880HZ_40MS_OPTIONAL:
        case BEEP_880HZ_60MS_TRIPLE_BEEP:
        case BEEP_880HZ_200MS:
        case BEEP_880HZ_500MS:
            ToneFrequency = 880;
            break;
    }

    BK4819_PlayTone(ToneFrequency, true);

    SYSTEM_DelayMs(2);

    AUDIO_AudioPathOn();

    SYSTEM_DelayMs(60);

    uint16_t Duration;
    switch (Beep)
    {
        case BEEP_880HZ_60MS_TRIPLE_BEEP:
            BK4819_ExitTxMute();
            SYSTEM_DelayMs(60);
            BK4819_EnterTxMute();
            SYSTEM_DelayMs(20);
            [[fallthrough]];
        case BEEP_500HZ_60MS_DOUBLE_BEEP_OPTIONAL:
        case BEEP_500HZ_60MS_DOUBLE_BEEP:
            BK4819_ExitTxMute();
            SYSTEM_DelayMs(60);
            BK4819_EnterTxMute();
            SYSTEM_DelayMs(20);
            [[fallthrough]];
        case BEEP_1KHZ_60MS_OPTIONAL:
            BK4819_ExitTxMute();
            Duration = 60;
            break;
        case BEEP_880HZ_40MS_OPTIONAL:
        case BEEP_440HZ_40MS_OPTIONAL:
            BK4819_ExitTxMute();
            Duration = 40;
            break;
        case BEEP_880HZ_200MS:
            BK4819_ExitTxMute();
            Duration = 200;
            break;
        case BEEP_440HZ_500MS:
        case BEEP_880HZ_500MS:
        default:
            BK4819_ExitTxMute();
            Duration = 500;
            break;
    }

    SYSTEM_DelayMs(Duration);
    BK4819_EnterTxMute();
    SYSTEM_DelayMs(20);

    AUDIO_AudioPathOff();

    SYSTEM_DelayMs(5);
    BK4819_TurnsOffTones_TurnsOnRX();
    SYSTEM_DelayMs(5);
    BK4819_WriteRegister(BK4819_REG_71, ToneConfig);

    if (gEnableSpeaker)
        AUDIO_AudioPathOn();

#ifdef ENABLE_FMRADIO
    if (gFmRadioMode)
		BK1080_Mute(false);
#endif

    if (gCurrentFunction == FUNCTION_POWER_SAVE && gRxIdleMode)
        BK4819_Sleep();

#ifdef ENABLE_VOX
    gVoxResumeCountdown = 80;
#endif

}

#ifdef ENABLE_VOICE

static const uint8_t VoiceClipLengthChinese[58] =
{
	0x32, 0x32, 0x32, 0x37, 0x37, 0x32, 0x32, 0x32,
	0x32, 0x37, 0x37, 0x32, 0x64, 0x64, 0x64, 0x64,
	0x64, 0x69, 0x64, 0x69, 0x5A, 0x5F, 0x5F, 0x64,
	0x64, 0x69, 0x64, 0x64, 0x69, 0x69, 0x69, 0x64,
	0x64, 0x6E, 0x69, 0x5F, 0x64, 0x64, 0x64, 0x69,
	0x69, 0x69, 0x64, 0x69, 0x64, 0x64, 0x55, 0x5F,
	0x5A, 0x4B, 0x4B, 0x46, 0x46, 0x69, 0x64, 0x6E,
	0x5A, 0x64,
};

static const uint8_t VoiceClipLengthEnglish[76] =
{
	0x50, 0x32, 0x2D, 0x2D, 0x2D, 0x37, 0x37, 0x37,
	0x32, 0x32, 0x3C, 0x37, 0x46, 0x46, 0x4B, 0x82,
	0x82, 0x6E, 0x82, 0x46, 0x96, 0x64, 0x46, 0x6E,
	0x78, 0x6E, 0x87, 0x64, 0x96, 0x96, 0x46, 0x9B,
	0x91, 0x82, 0x82, 0x73, 0x78, 0x64, 0x82, 0x6E,
	0x78, 0x82, 0x87, 0x6E, 0x55, 0x78, 0x64, 0x69,
	0x9B, 0x5A, 0x50, 0x3C, 0x32, 0x55, 0x64, 0x64,
	0x50, 0x46, 0x46, 0x46, 0x4B, 0x4B, 0x50, 0x50,
	0x55, 0x4B, 0x4B, 0x32, 0x32, 0x32, 0x32, 0x37,
	0x41, 0x32, 0x3C, 0x37,
};

VOICE_ID_t        gVoiceID[8];
uint8_t           gVoiceReadIndex;
uint8_t           gVoiceWriteIndex;
volatile uint16_t gCountdownToPlayNextVoice_10ms;
volatile bool     gFlagPlayQueuedVoice;
VOICE_ID_t        gAnotherVoiceID = VOICE_ID_INVALID;


static void AUDIO_PlayVoice(uint8_t VoiceID)
{
	unsigned int i;

	GPIO_SetBit(&GPIOA->DATA, GPIOA_PIN_VOICE_0);
	SYSTEM_DelayMs(20);
	GPIO_ClearBit(&GPIOA->DATA, GPIOA_PIN_VOICE_0);

	for (i = 0; i < 8; i++)
	{
		if ((VoiceID & 0x80U) == 0)
			GPIO_ClearBit(&GPIOA->DATA, GPIOA_PIN_VOICE_1);
		else
			GPIO_SetBit(&GPIOA->DATA, GPIOA_PIN_VOICE_1);

		SYSTICK_DelayUs(1000);
		GPIO_SetBit(&GPIOA->DATA, GPIOA_PIN_VOICE_0);
		SYSTICK_DelayUs(1200);
		GPIO_ClearBit(&GPIOA->DATA, GPIOA_PIN_VOICE_0);
		VoiceID <<= 1;
		SYSTICK_DelayUs(200);
	}
}

void AUDIO_PlaySingleVoice(bool bFlag)
{
	uint8_t VoiceID;
	uint8_t Delay;

	VoiceID = gVoiceID[0];

	if (gEeprom.VOICE_PROMPT != VOICE_PROMPT_OFF && gVoiceWriteIndex > 0)
	{
		if (gEeprom.VOICE_PROMPT == VOICE_PROMPT_CHINESE)
		{	// Chinese
			if (VoiceID >= ARRAY_SIZE(VoiceClipLengthChinese))
				goto Bailout;

			Delay    = VoiceClipLengthChinese[VoiceID];
			VoiceID += VOICE_ID_CHI_BASE;
		}
		else
		{	// English
			if (VoiceID >= ARRAY_SIZE(VoiceClipLengthEnglish))
				goto Bailout;

			Delay    = VoiceClipLengthEnglish[VoiceID];
			VoiceID += VOICE_ID_ENG_BASE;
		}

		if (FUNCTION_IsRx())   // 1of11
			BK4819_SetAF(BK4819_AF_MUTE);

		#ifdef ENABLE_FMRADIO
			if (gFmRadioMode)
				BK1080_Mute(true);
		#endif

		AUDIO_AudioPathOn();

		#ifdef ENABLE_VOX
			gVoxResumeCountdown = 2000;
		#endif

		SYSTEM_DelayMs(5);
		AUDIO_PlayVoice(VoiceID);

		if (gVoiceWriteIndex == 1)
			Delay += 3;

		if (bFlag)
		{
			SYSTEM_DelayMs(Delay * 10);

			if (FUNCTION_IsRx())	// 1of11
				RADIO_SetModulation(gRxVfo->Modulation);

			#ifdef ENABLE_FMRADIO
				if (gFmRadioMode)
					BK1080_Mute(false);
			#endif

			if (!gEnableSpeaker)
				AUDIO_AudioPathOff();

			gVoiceWriteIndex    = 0;
			gVoiceReadIndex     = 0;

			#ifdef ENABLE_VOX
				gVoxResumeCountdown = 80;
			#endif

			return;
		}

		gVoiceReadIndex                = 1;
		gCountdownToPlayNextVoice_10ms = Delay;
		gFlagPlayQueuedVoice           = false;

		return;
	}

Bailout:
	gVoiceReadIndex  = 0;
	gVoiceWriteIndex = 0;
}

void AUDIO_SetVoiceID(uint8_t Index, VOICE_ID_t VoiceID)
{
	if (Index >= ARRAY_SIZE(gVoiceID))
		return;

	if (Index == 0)
	{
		gVoiceWriteIndex = 0;
		gVoiceReadIndex  = 0;
	}

	gVoiceID[Index] = VoiceID;

	gVoiceWriteIndex++;
}

uint8_t AUDIO_SetDigitVoice(uint8_t Index, uint16_t Value)
{
	uint16_t Remainder;
	uint8_t  Result;
	uint8_t  Count;

	if (Index == 0)
	{
		gVoiceWriteIndex = 0;
		gVoiceReadIndex  = 0;
	}

	Count     = 0;
	Result    = Value / 1000U;
	Remainder = Value % 1000U;
	if (Remainder < 100U)
	{
		if (Remainder < 10U)
			goto Skip;
	}
	else
	{
		Result = Remainder / 100U;
		gVoiceID[gVoiceWriteIndex++] = (VOICE_ID_t)Result;
		Count++;
		Remainder -= Result * 100U;
	}
	Result = Remainder / 10U;
	gVoiceID[gVoiceWriteIndex++] = (VOICE_ID_t)Result;
	Count++;
	Remainder -= Result * 10U;

Skip:
	gVoiceID[gVoiceWriteIndex++] = (VOICE_ID_t)Remainder;

	return Count + 1U;
}

void AUDIO_PlayQueuedVoice(void)
{
	uint8_t VoiceID;
	uint8_t Delay;
	bool    Skip;

	Skip = false;

	if (gVoiceReadIndex != gVoiceWriteIndex && gEeprom.VOICE_PROMPT != VOICE_PROMPT_OFF)
	{
		VoiceID = gVoiceID[gVoiceReadIndex];
		if (gEeprom.VOICE_PROMPT == VOICE_PROMPT_CHINESE)
		{
			if (VoiceID < ARRAY_SIZE(VoiceClipLengthChinese))
			{
				Delay = VoiceClipLengthChinese[VoiceID];
				VoiceID += VOICE_ID_CHI_BASE;
			}
			else
				Skip = true;
		}
		else
		{
			if (VoiceID < ARRAY_SIZE(VoiceClipLengthEnglish))
			{
				Delay = VoiceClipLengthEnglish[VoiceID];
				VoiceID += VOICE_ID_ENG_BASE;
			}
			else
				Skip = true;
		}

		gVoiceReadIndex++;

		if (!Skip)
		{
			if (gVoiceReadIndex == gVoiceWriteIndex)
				Delay += 3;

			AUDIO_PlayVoice(VoiceID);

			gCountdownToPlayNextVoice_10ms = Delay;
			gFlagPlayQueuedVoice           = false;

			#ifdef ENABLE_VOX
				gVoxResumeCountdown = 2000;
			#endif

			return;
		}
	}

	if (FUNCTION_IsRx())
	{
		RADIO_SetModulation(gRxVfo->Modulation); // 1of11
	}

	#ifdef ENABLE_FMRADIO
		if (gFmRadioMode)
			BK1080_Mute(false);
	#endif

	if (!gEnableSpeaker)
		AUDIO_AudioPathOff();

	#ifdef ENABLE_VOX
		gVoxResumeCountdown = 80;
	#endif

	gVoiceWriteIndex    = 0;
	gVoiceReadIndex     = 0;
}

#endif