/* 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 "frequencies.h"
#include "misc.h"
#include "settings.h"

const uint32_t LowerLimitFrequencyBandTable[7] = {
	 5000000,
	10800000,
	13600000,
	17400000,
	35000000,
	40000000,
	47000000,
};

const uint32_t MiddleFrequencyBandTable[7] = {
	 6500000,
	12200000,
	15000000,
	26000000,
	37000000,
	43500000,
	55000000,
};

const uint32_t UpperLimitFrequencyBandTable[7] = {
	 7600000,
	13599990,
	17399990,
	34999990,
	39999990,
	46999990,
	60000000,
};

const uint32_t NoaaFrequencyTable[10] = {
	16255000,
	16240000,
	16247500,
	16242500,
	16245000,
	16250000,
	16252500,
	16152500,
	16177500,
	16327500,
};

const uint16_t StepFrequencyTable[6] = {
	250,
	500,
	625,
	1000,
	1250,
	2500
};

FREQUENCY_Band_t FREQUENCY_GetBand(uint32_t Frequency)
{
	if ((Frequency - 5000000) < 2600001) {
		return BAND1_50MHz;
	}
	if ((Frequency - 10800000) < 2799991) {
		return BAND2_108MHz;
	}
	if ((Frequency - 13600000) < 3799991) {
		return BAND3_136MHz;
	}
	if ((Frequency - 17400000) < 17599991) {
		return BAND4_174MHz;
	}
	if ((Frequency - 35000000) < 4999991) {
		return BAND5_350MHz;
	}
	if ((Frequency - 40000000) < 6999991) {
		return BAND6_400MHz;
	}
	if ((Frequency - 47000000) < 13000001) {
		return BAND7_470MHz;
	}

	// TODO: Double check the assembly
	return BAND6_400MHz;
}

uint32_t FREQUENCY_CalculateOutputPower(uint8_t TxpLow, uint8_t TxpMid, uint8_t TxpHigh, uint32_t LowerLimit, uint32_t Middle, uint32_t UpperLimit, uint32_t Frequency)
{
	if (Frequency <= LowerLimit) {
		return TxpLow;
	}
	if (UpperLimit <= Frequency) {
		return TxpHigh;
	}
	if (Frequency <= Middle) {
		return TxpMid + ((TxpMid - TxpLow) * (Frequency - LowerLimit)) / (Middle - LowerLimit);
	}

	return TxpMid + ((TxpHigh - TxpMid) * (Frequency - Middle)) / (UpperLimit - Middle);
}

uint32_t FREQUENCY_FloorToStep(uint32_t Frequency, uint32_t Step, uint32_t Base)
{
	uint32_t Index;

	Index = (Frequency - Base) / Step;
	return Base + (Step * Index);
}

int FREQUENCY_Check(RADIO_Info_t *pInfo)
{
	uint32_t Frequency;

	if (pInfo->CHANNEL_SAVE >= 207) {
		return -1;
	}
	Frequency = pInfo->pDCS_Reverse->Frequency;
	if (gSetting_F_LOCK == F_LOCK_FCC) {
		if ((Frequency + 14400000) < 399991) {
			return 0;
		}
		if (2999990 < (Frequency - 42000000)) {
			return -1;
		}
		return 0;
	}

	if (gSetting_F_LOCK == F_LOCK_CE) {
		if ((Frequency - 14400000) < 199991) {
			return 0;
		}
	}

	if (gSetting_F_LOCK != F_LOCK_GB) {
		if (gSetting_F_LOCK != F_LOCK_430) {
			if (gSetting_F_LOCK == F_LOCK_438) {
				if ((Frequency - 13600000) < 3799991) {
					return 0;
				}
				if ((Frequency - 40000000) < 3799991) {
					return 0;
				}
				return -1;
			}
			if ((Frequency - 13600000) < 3799991) {
				return 0;
			}
			if (((Frequency - 35000000) < 4999991 && gSetting_350TX && gSetting_350EN)) {
				return 0;
			}
			if ((Frequency - 40000000) < 6999991) {
				return 0;
			}
			if ((Frequency - 17400000) < 17599991 && gSetting_200TX) {
				return 0;
			}
			if ((Frequency - 47000000) < 13000001 && gSetting_500TX) {
				return 0;
			}
			return -1;
		}
		if ((Frequency - 13600000) < 3799991) {
			return 0;
		}
		if (2999990 < (Frequency - 40000000)) {
			return -1;
		}
	}
	if ((Frequency - 14400000) < 399991) {
		return 0;
	}

	if (999990 < (Frequency - 43000000)) {
		return -1;
	}

	return 0;
}