/* 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.
 */

#include <string.h>

#include "app/dtmf.h"
#if defined(ENABLE_FMRADIO)
	#include "app/fm.h"
#endif
#include "bsp/dp32g030/gpio.h"
#include "dcs.h"
#if defined(ENABLE_FMRADIO)
	#include "driver/bk1080.h"
#endif
#include "driver/bk4819.h"
#include "driver/gpio.h"
#include "driver/system.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"

FUNCTION_Type_t gCurrentFunction;

void FUNCTION_Init(void)
{
	#ifdef ENABLE_NOAA
		if (IS_NOT_NOAA_CHANNEL(gRxVfo->CHANNEL_SAVE))
	#endif
	{
		gCurrentCodeType = gSelectedCodeType;
		if (gCssScanMode == CSS_SCAN_MODE_OFF)
			gCurrentCodeType = gRxVfo->IsAM ? CODE_TYPE_OFF : gRxVfo->pRX->CodeType;
	}
	#ifdef ENABLE_NOAA
		else
			gCurrentCodeType = CODE_TYPE_CONTINUOUS_TONE;
	#endif

	gDTMF_RequestPending          = false;
	gDTMF_WriteIndex              = 0;

	memset(gDTMF_Received, 0, sizeof(gDTMF_Received));

	g_CxCSS_TAIL_Found            = false;
	g_CDCSS_Lost                  = false;
	g_CTCSS_Lost                  = false;
	g_VOX_Lost                    = false;
	g_SquelchLost                 = false;
	gTailNoteEliminationCountdown = 0;
	gFlagTteComplete              = false;
	gFoundCTCSS                   = false;
	gFoundCDCSS                   = false;
	gFoundCTCSSCountdown          = 0;
	gFoundCDCSSCountdown          = 0;
	gEndOfRxDetectedMaybe         = false;
	gSystickCountdown2            = 0;
}

void FUNCTION_Select(FUNCTION_Type_t Function)
{
	FUNCTION_Type_t PreviousFunction;
	bool            bWasPowerSave;

	PreviousFunction = gCurrentFunction;
	bWasPowerSave    = (PreviousFunction == FUNCTION_POWER_SAVE);
	gCurrentFunction = Function;

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

	switch (Function)
	{
		case FUNCTION_FOREGROUND:
			if (gDTMF_ReplyState != DTMF_REPLY_NONE)
				RADIO_PrepareCssTX();

			if (PreviousFunction == FUNCTION_TRANSMIT)
			{
				gVFO_RSSI_Level[0] = 0;
				gVFO_RSSI_Level[1] = 0;
			}
			else
			if (PreviousFunction != FUNCTION_RECEIVE)
				break;

			#if defined(ENABLE_FMRADIO)
				if (gFmRadioMode)
					gFM_RestoreCountdown = 500;
			#endif

			if (gDTMF_CallState == DTMF_CALL_STATE_CALL_OUT || gDTMF_CallState == DTMF_CALL_STATE_RECEIVED)
				gDTMF_AUTO_RESET_TIME = 1 + (gEeprom.DTMF_AUTO_RESET_TIME * 2);

			return;
	
		case FUNCTION_MONITOR:
		case FUNCTION_INCOMING:
		case FUNCTION_RECEIVE:
			break;
	
		case FUNCTION_POWER_SAVE:
			gBatterySave = gEeprom.BATTERY_SAVE * 10;
			gRxIdleMode  = true;
			BK4819_DisableVox();
			BK4819_Sleep();
			BK4819_ToggleGpioOut(BK4819_GPIO6_PIN2, false);
			gBatterySaveCountdownExpired = false;
			gUpdateStatus = true;
			GUI_SelectNextDisplay(DISPLAY_MAIN);
			return;
	
		case FUNCTION_TRANSMIT:
			#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();
					GPIO_ClearBit(&GPIOC->DATA, GPIOC_PIN_AUDIO_PATH);
					SYSTEM_DelayMs(20);
					BK4819_PlayTone(500, 0);
					SYSTEM_DelayMs(2);
					GPIO_SetBit(&GPIOC->DATA, GPIOC_PIN_AUDIO_PATH);
					gEnableSpeaker = true;
					SYSTEM_DelayMs(60);
					BK4819_ExitTxMute();
					gAlarmToneCounter = 0;
					break;
				}
			#endif
			
			GUI_DisplayScreen();
			RADIO_SetTxParameters();
			BK4819_ToggleGpioOut(BK4819_GPIO1_PIN29_RED, true);
	
			DTMF_Reply();
	
			#ifdef ENABLE_ALARM
				if (gAlarmState != ALARM_STATE_OFF)
				{
					BK4819_TransmitTone(true, (gAlarmState == ALARM_STATE_TX1750) ? 1750 : 500);
					SYSTEM_DelayMs(2);
					GPIO_SetBit(&GPIOC->DATA, GPIOC_PIN_AUDIO_PATH);
					gAlarmToneCounter = 0;
					gEnableSpeaker    = true;
					break;
				}
			#endif
			
			if (gCurrentVfo->SCRAMBLING_TYPE && gSetting_ScrambleEnable)
				BK4819_EnableScramble(gCurrentVfo->SCRAMBLING_TYPE - 1U);
			else
				BK4819_DisableScramble();

			break;
	}

	gBatterySaveCountdown = battery_save_count;
	gSchedulePowerSave    = false;
	#if defined(ENABLE_FMRADIO)
		gFM_RestoreCountdown = 0;
	#endif
}