#include "pdctrl.h"
#include "main.h"
// 模式转换
#define DAC_MODE() GPIO_RESET(PDCTRL_GPIO_Port, PDCTRL_Pin)
#define PWM_MODE() GPIO_SET(PDCTRL_GPIO_Port, PDCTRL_Pin)

__IO static pdctrl_mode_e out_mode = PDCTRL_DAC;
__IO uint16_t last_out;
static uint16_t arr_default = 0;

static void dac_dinit(void)
{
    DAC_STOP(DAC, LL_DAC_CHANNEL_2); // DAC通道禁用
}
static void dac_init(void)
{
    last_out = 0xffff;
    DAC_MODE();

    LL_DAC_InitTypeDef DAC_InitStruct = {0};
    LL_GPIO_InitTypeDef GPIO_InitStruct = {0};
    LL_APB1_GRP1_EnableClock(LL_APB1_GRP1_PERIPH_DAC1);
    LL_IOP_GRP1_EnableClock(LL_IOP_GRP1_PERIPH_GPIOA);
    GPIO_InitStruct.Pin = LL_GPIO_PIN_5;
    GPIO_InitStruct.Mode = LL_GPIO_MODE_ANALOG;
    GPIO_InitStruct.Pull = LL_GPIO_PULL_NO;
    LL_GPIO_Init(GPIOA, &GPIO_InitStruct);
    NVIC_SetPriority(TIM6_DAC_IRQn, 0);
    NVIC_EnableIRQ(TIM6_DAC_IRQn);
    DAC_InitStruct.TriggerSource = LL_DAC_TRIG_SOFTWARE;
    DAC_InitStruct.WaveAutoGeneration = LL_DAC_WAVE_AUTO_GENERATION_NONE;
    DAC_InitStruct.OutputBuffer = LL_DAC_OUTPUT_BUFFER_ENABLE;
    LL_DAC_Init(DAC, LL_DAC_CHANNEL_2, &DAC_InitStruct);

    DAC_START(DAC, LL_DAC_CHANNEL_2); // DAC通道使能
}
static void pwm_dinit(void)
{
    PWM_STOP(TIM2, LL_TIM_CHANNEL_CH1); // PWM通道禁用
}
static void pwm_init(uint8_t pwm_frequency)
{
    last_out = 0xffff;
    PWM_MODE();
    arr_default = PDCTRL_SYS_FREQUENCY / (pwm_frequency - 1);

    LL_TIM_InitTypeDef TIM_InitStruct = {0};
    LL_TIM_OC_InitTypeDef TIM_OC_InitStruct = {0};

    LL_GPIO_InitTypeDef GPIO_InitStruct = {0};
    LL_APB1_GRP1_EnableClock(LL_APB1_GRP1_PERIPH_TIM2);
    NVIC_SetPriority(TIM2_IRQn, 0);
    NVIC_EnableIRQ(TIM2_IRQn);

    TIM_InitStruct.Prescaler = 0;
    TIM_InitStruct.CounterMode = LL_TIM_COUNTERMODE_UP;
    TIM_InitStruct.Autoreload = arr_default - 1;
    TIM_InitStruct.ClockDivision = LL_TIM_CLOCKDIVISION_DIV1;
    LL_TIM_Init(TIM2, &TIM_InitStruct);
    LL_TIM_EnableARRPreload(TIM2);
    LL_TIM_OC_EnablePreload(TIM2, LL_TIM_CHANNEL_CH1);
    TIM_OC_InitStruct.OCMode = LL_TIM_OCMODE_PWM1;
    TIM_OC_InitStruct.OCState = LL_TIM_OCSTATE_DISABLE;
    TIM_OC_InitStruct.CompareValue = 0;
    TIM_OC_InitStruct.OCPolarity = LL_TIM_OCPOLARITY_HIGH;
    LL_TIM_OC_Init(TIM2, LL_TIM_CHANNEL_CH1, &TIM_OC_InitStruct);
    LL_TIM_OC_DisableFast(TIM2, LL_TIM_CHANNEL_CH1);

    LL_TIM_SetTriggerOutput(TIM2, LL_TIM_TRGO_RESET);
    LL_TIM_DisableMasterSlaveMode(TIM2);

    LL_IOP_GRP1_EnableClock(LL_IOP_GRP1_PERIPH_GPIOA);

    GPIO_InitStruct.Pin = LL_GPIO_PIN_5;
    GPIO_InitStruct.Mode = LL_GPIO_MODE_ALTERNATE;
    GPIO_InitStruct.Speed = LL_GPIO_SPEED_FREQ_LOW;
    GPIO_InitStruct.OutputType = LL_GPIO_OUTPUT_PUSHPULL;
    GPIO_InitStruct.Pull = LL_GPIO_PULL_NO;
    GPIO_InitStruct.Alternate = LL_GPIO_AF_5;
    LL_GPIO_Init(GPIOA, &GPIO_InitStruct);

    PWM_START(TIM2, LL_TIM_CHANNEL_CH1); // PWM通道使能
}
/**
 * @brief 控制初始化
 * @param {pdctrl_mode_e} mode
 * @param {uint8_t} pwm_frequency
 * @return {*}
 * @note
 */
void pdctrl_init(pdctrl_mode_e mode, uint8_t pwm_frequency)
{
    out_mode = mode;
    if (out_mode == PDCTRL_DAC)
    {
        pwm_dinit();
        dac_init();
    }
    else if (out_mode == PDCTRL_PWM)
    {
        dac_dinit();
        pwm_init(pwm_frequency);
    }
}

/**
 * @brief 控制输出
 * @param {uint16_t} out
 * @return {*}
 * @note
 */
void pdctrl_out(uint16_t out)
{
    if (last_out == out)
    {
        return;
    }
    last_out = out;
    if (out_mode == PDCTRL_DAC)
    {
        DAC_OUT(DAC, LL_DAC_CHANNEL_2, out);
    }
    else if (out_mode == PDCTRL_PWM)
    {
        PWM_SET_DUTY(TIM2, 1, out);
    }
}

/**
 * @brief 占空比转换为CCR值
 * @param {float32} out
 * @return {*}
 * @note
 */
uint16_t pdctrl_pwm_duty_convert_ccr(float32 pwm_duty)
{
    return (pwm_duty * arr_default) / 100;
}