/* 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 "bsp/dp32g030/gpio.h"
#include "bk1080.h"
#include "driver/gpio.h"
#include "driver/i2c.h"
#include "driver/system.h"
#include "../settings.h"
#include "misc.h"

#ifndef ARRAY_SIZE
    #define ARRAY_SIZE(a) (sizeof(a) / sizeof(a[0]))
#endif

static const uint16_t BK1080_RegisterTable[] =
{
    0x0008, 0x1080, 0x0201, 0x0000, 0x40C0, 0x0A1F, 0x002E, 0x02FF,
    0x5B11, 0x0000, 0x411E, 0x0000, 0xCE00, 0x0000, 0x0000, 0x1000,
    0x3197, 0x0000, 0x13FF, 0x9852, 0x0000, 0x0000, 0x0008, 0x0000,
    0x51E1, 0xA8BC, 0x2645, 0x00E4, 0x1CD8, 0x3A50, 0xEAE0, 0x3000,
    0x0200, 0x0000,
};

static bool gIsInitBK1080;

uint16_t BK1080_BaseFrequency;
uint16_t BK1080_FrequencyDeviation;

void BK1080_Init0(void)
{
    BK1080_Init(0,0/*,0*/);
}

void BK1080_Init(uint16_t freq, uint8_t band/*, uint8_t space*/)
{
    unsigned int i;

    if (freq) {
        GPIO_ClearBit(&GPIOB->DATA, GPIOB_PIN_BK1080);

        if (!gIsInitBK1080) {
            for (i = 0; i < ARRAY_SIZE(BK1080_RegisterTable); i++)
                BK1080_WriteRegister(i, BK1080_RegisterTable[i]);

            SYSTEM_DelayMs(250);

            BK1080_WriteRegister(BK1080_REG_25_INTERNAL, 0xA83C);
            BK1080_WriteRegister(BK1080_REG_25_INTERNAL, 0xA8BC);

            SYSTEM_DelayMs(60);

            gIsInitBK1080 = true;
        }
        else {
            BK1080_WriteRegister(BK1080_REG_02_POWER_CONFIGURATION, 0x0201);
        }

        #ifdef ENABLE_FEAT_F4HWN
            BK1080_WriteRegister(BK1080_REG_05_SYSTEM_CONFIGURATION2, 0x0A10 | (gEeprom.VOLUME_GAIN >> 2));
        #else
            BK1080_WriteRegister(BK1080_REG_05_SYSTEM_CONFIGURATION2, 0x0A1F);
        #endif
        BK1080_UpdateSysconf();
        BK1080_SetFrequency(freq, band/*, space*/);
    }
    else {
        BK1080_WriteRegister(BK1080_REG_02_POWER_CONFIGURATION, 0x0241);
        GPIO_SetBit(&GPIOB->DATA, GPIOB_PIN_BK1080);
    }
}

uint16_t BK1080_ReadRegister(BK1080_Register_t Register)
{
    uint8_t Value[2];

    I2C_Start();
    I2C_Write(0x80);
    I2C_Write((Register << 1) | I2C_READ);
    I2C_ReadBuffer(Value, sizeof(Value));
    I2C_Stop();

    return (Value[0] << 8) | Value[1];
}

void BK1080_WriteRegister(BK1080_Register_t Register, uint16_t Value)
{
    I2C_Start();
    I2C_Write(0x80);
    I2C_Write((Register << 1) | I2C_WRITE);
    Value = ((Value >> 8) & 0xFF) | ((Value & 0xFF) << 8);
    I2C_WriteBuffer(&Value, sizeof(Value));
    I2C_Stop();
}

void BK1080_UpdateSysconf(void) {
    BK1080_WriteRegister(BK1080_REG_04_SYSTEM_CONFIGURATION1,
                         (gEeprom.BK1080_DEEMPHASIS_CONFIG == DEEMPHASIS_OFF ? BK1080_DEEMPHASIS_OFF : BK1080_DEEMPHASIS_ON) |
                         (gEeprom.BK1080_DEEMPHASIS_CONFIG == DEEMPHASIS_EUR ? BK1080_DEEMPHASIS_EUR : BK1080_DEEMPHASIS_USA) |
                                 (gEeprom.BK1080_AGC_ENABLED ? BK1080_AGC_ON : BK1080_AGC_OFF) |
                                 (gEeprom.BK1080_BLEND_CONFIG << 6)
                         );
}

void BK1080_Mute(bool Mute)
{
    BK1080_WriteRegister(BK1080_REG_02_POWER_CONFIGURATION, Mute ? 0x4201 : 0x0201);
}

void BK1080_SetFrequency(uint16_t frequency, uint8_t band/*, uint8_t space*/)
{
    //uint8_t spacings[] = {20,10,5};
    //space %= 3;

    uint16_t channel = (frequency - BK1080_GetFreqLoLimit(band))/* * 10 / spacings[space]*/;

    uint16_t regval = BK1080_ReadRegister(BK1080_REG_05_SYSTEM_CONFIGURATION2);
    regval = (regval & ~(0b11 << 6)) | ((band & 0b11) << 6);
    //regval = (regval & ~(0b11 << 4)) | ((space & 0b11) << 4);

    BK1080_WriteRegister(BK1080_REG_05_SYSTEM_CONFIGURATION2, regval);

    BK1080_WriteRegister(BK1080_REG_03_CHANNEL, channel);
    SYSTEM_DelayMs(10);
    BK1080_WriteRegister(BK1080_REG_03_CHANNEL, channel | 0x8000);
}

void BK1080_GetFrequencyDeviation(uint16_t Frequency)
{
    BK1080_BaseFrequency      = Frequency;
    BK1080_FrequencyDeviation = BK1080_ReadRegister(BK1080_REG_07_TEST1) / 16;
}

uint16_t BK1080_GetFreqLoLimit(uint8_t band)
{
    uint16_t lim[] = {875, 760, 760, 640};
    return lim[band % 4];
}

uint16_t BK1080_GetFreqHiLimit(uint8_t band)
{
    band %= 4;
    uint16_t lim[] = {1080, 1080, 900, 760};
    return lim[band % 4];
}