/* Original work Copyright 2023 Dual Tachyon
 * https://github.com/DualTachyon
 *
 * Modified work Copyright 2024 kamilsss655
 * https://github.com/kamilsss655
 *
 * 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 <string.h>

#include "app/dtmf.h"
#if defined(ENABLE_FMRADIO)
	#include "app/fm.h"
#endif
#include "audio.h"
#include "bsp/dp32g030/gpio.h"
#include "dcs.h"
#include "driver/backlight.h"
#if defined(ENABLE_FMRADIO)
	#include "driver/bk1080.h"
#endif
#include "driver/bk4819.h"
#include "driver/gpio.h"
#include "driver/system.h"
#include "driver/st7565.h"
#include "frequencies.h"
#include "functions.h"
#include "helper/battery.h"
#include "misc.h"
#include "radio.h"
#include "settings.h"
#include "ui/status.h"
#include "ui/ui.h"
#ifdef ENABLE_MESSENGER
	#include "app/messenger.h"
#endif

FUNCTION_Type_t gCurrentFunction;

void FUNCTION_Init(void)
{
#ifdef ENABLE_NOAA
	if (!IS_NOAA_CHANNEL(gRxVfo->CHANNEL_SAVE))
#endif
	{
		gCurrentCodeType = (gRxVfo->Modulation != MODULATION_FM) ? CODE_TYPE_OFF : gRxVfo->pRX->CodeType;
	}
#ifdef ENABLE_NOAA
	else
		gCurrentCodeType = CODE_TYPE_CONTINUOUS_TONE;
#endif

#ifdef ENABLE_DTMF_CALLING
	DTMF_clear_RX();
#endif

	g_CxCSS_TAIL_Found = false;
	g_CDCSS_Lost       = false;
	g_CTCSS_Lost       = false;

	#ifdef ENABLE_VOX
		g_VOX_Lost     = false;
	#endif

	g_SquelchLost      = false;

	gFlagTailNoteEliminationComplete   = false;
	gTailNoteEliminationCountdown_10ms = 0;
	gFoundCTCSS                        = false;
	gFoundCDCSS                        = false;
	gFoundCTCSSCountdown_10ms          = 0;
	gFoundCDCSSCountdown_10ms          = 0;
	gEndOfRxDetectedMaybe              = false;

	#ifdef ENABLE_NOAA
		gNOAACountdown_10ms = 0;
	#endif

	gUpdateStatus = true;
}

void FUNCTION_Select(FUNCTION_Type_t Function)
{
	const FUNCTION_Type_t PreviousFunction = gCurrentFunction;
	const bool            bWasPowerSave    = (PreviousFunction == FUNCTION_POWER_SAVE);

	gCurrentFunction = Function;

	if (bWasPowerSave && Function != FUNCTION_POWER_SAVE)
	{
		BK4819_Conditional_RX_TurnOn_and_GPIO6_Enable();
		gRxIdleMode = false;
		UI_DisplayStatus();
	}

	switch (Function)
	{
		case FUNCTION_FOREGROUND:
#ifdef ENABLE_DTMF_CALLING
			if (gDTMF_ReplyState != DTMF_REPLY_NONE)
				RADIO_PrepareCssTX();
#endif
			if (PreviousFunction == FUNCTION_TRANSMIT)
			{
				ST7565_FixInterfGlitch();
				gVFO_RSSI_bar_level[0] = 0;
				gVFO_RSSI_bar_level[1] = 0;
			}
			else
			if (PreviousFunction != FUNCTION_RECEIVE)
				break;

			#if defined(ENABLE_FMRADIO)
				if (gFmRadioMode)
					gFM_RestoreCountdown_10ms = fm_restore_countdown_10ms;
			#endif
			
#ifdef ENABLE_DTMF_CALLING
			if (gDTMF_CallState == DTMF_CALL_STATE_CALL_OUT ||
			    gDTMF_CallState == DTMF_CALL_STATE_RECEIVED ||
				gDTMF_CallState == DTMF_CALL_STATE_RECEIVED_STAY)
			{
				gDTMF_auto_reset_time_500ms = gEeprom.DTMF_auto_reset_time * 2;
			}
#endif
			gUpdateStatus = true;
			return;

		case FUNCTION_MONITOR:
			gMonitor = true;
			break;

		case FUNCTION_INCOMING:
		case FUNCTION_RECEIVE:
			break;

		case FUNCTION_POWER_SAVE:
			gPowerSave_10ms            = gEeprom.BATTERY_SAVE * 10;
			gPowerSaveCountdownExpired = false;

			gRxIdleMode = true;

			gMonitor = false;

			BK4819_DisableVox();			
			BK4819_Sleep();

			BK4819_ToggleGpioOut(BK4819_GPIO0_PIN28_RX_ENABLE, false);

			gUpdateStatus = true;

			if (gScreenToDisplay != DISPLAY_MENU)     // 1of11 .. don't close the menu
				GUI_SelectNextDisplay(DISPLAY_MAIN);

			return;

		case FUNCTION_TRANSMIT:
			#ifdef ENABLE_MESSENGER
				MSG_EnableRX(false);	
			#endif	

			// if DTMF is enabled when TX'ing, it changes the TX audio filtering !! .. 1of11
			BK4819_DisableDTMF();

#ifdef ENABLE_DTMF_CALLING
			// clear the DTMF RX buffer
			DTMF_clear_RX();
#endif

			// clear the DTMF RX live decoder buffer
			gDTMF_RX_live_timeout = 0;
			gDTMF_RX_live_timeout = 0;
			memset(gDTMF_RX_live, 0, sizeof(gDTMF_RX_live));

			#if defined(ENABLE_FMRADIO)
				if (gFmRadioMode)
					BK1080_Init(0, false);
			#endif

			#ifdef ENABLE_ALARM
				if (gAlarmState == ALARM_STATE_TXALARM && gEeprom.ALARM_MODE != ALARM_MODE_TONE)
				{
					gAlarmState = ALARM_STATE_ALARM;

					GUI_DisplayScreen();

					AUDIO_AudioPathOff();

					SYSTEM_DelayMs(20);
					BK4819_PlayTone(500, 0);
					SYSTEM_DelayMs(2);

					AUDIO_AudioPathOn();

					gEnableSpeaker = true;

					SYSTEM_DelayMs(60);
					BK4819_ExitTxMute();

					gAlarmToneCounter = 0;
					break;
				}
			#endif

			gUpdateStatus = true;

			GUI_DisplayScreen();

			RADIO_SetTxParameters();

			// turn the RED LED on
			BK4819_ToggleGpioOut(BK4819_GPIO5_PIN1_RED, true);

			DTMF_Reply();

			if (gCurrentVfo->DTMF_PTT_ID_TX_MODE == PTT_ID_APOLLO)
				BK4819_PlaySingleTone(2525, 250, 0, gEeprom.DTMF_SIDE_TONE);

			#if defined(ENABLE_ALARM) || defined(ENABLE_TX1750)
				if (gAlarmState != ALARM_STATE_OFF)
				{
					#ifdef ENABLE_TX1750
						if (gAlarmState == ALARM_STATE_TX1750)
							BK4819_TransmitTone(true, 1750);
					#endif
					#ifdef ENABLE_ALARM
						if (gAlarmState == ALARM_STATE_TXALARM)
							BK4819_TransmitTone(true, 500);
					#endif
					SYSTEM_DelayMs(2);
					AUDIO_AudioPathOn();
					#ifdef ENABLE_ALARM
						gAlarmToneCounter = 0;
					#endif
					gEnableSpeaker = true;
					break;
				}
			#endif

			if (gCurrentVfo->SCRAMBLING_TYPE > 0 && gSetting_ScrambleEnable)
				BK4819_EnableScramble(gCurrentVfo->SCRAMBLING_TYPE - 1);
			else
				BK4819_DisableScramble();

			if (gSetting_backlight_on_tx_rx == BACKLIGHT_ON_TR_TX || 
			    gSetting_backlight_on_tx_rx == BACKLIGHT_ON_TR_TXRX)
				BACKLIGHT_TurnOn();

			break;

		case FUNCTION_BAND_SCOPE:
			break;
	}

	gBatterySaveCountdown_10ms = battery_save_count_10ms;
	gSchedulePowerSave         = false;

	#if defined(ENABLE_FMRADIO)
		gFM_RestoreCountdown_10ms = 0;
	#endif
}