/*
 * dvr_tim.c
 *
 *  Created on: Mar 31, 2023
 *      Author: gxms0
 */

#include "drv_tim.h"
#include "Global.h"

void TIM1_UP_IRQHandler(void) __attribute__((interrupt("WCH-Interrupt-fast")));
void TIM5_IRQHandler(void) __attribute__((interrupt("WCH-Interrupt-fast")));


volatile uint32_t tick_ms = 0;
volatile uint32_t tick_us = 0;


uint32_t get_ticks(void)
{
    return get_ticks_usec();
}

uint32_t get_ticks_msec(void)
{
    return tick_ms;
}

uint32_t get_ticks_usec(void)
{
    return tick_us + (uint32_t)TIM1->CNT;
}

void delay_s(uint32_t time)
{
    uint32_t tick = get_ticks_msec();
    while(1)
    {
        if((get_ticks_msec() - tick) >= (time * 1000))
        {
            break;
        }
    }
}

void delay_ms(uint32_t time)
{
    uint32_t tick = get_ticks_msec();
    while(1)
    {
        if((get_ticks_msec() - tick) >= time)
        {
            break;
        }
    }
}

void delay_us(uint32_t time)
{
    uint32_t tick = get_ticks();
    while(1)
    {
        if((get_ticks() - tick) >= time)
        {
            break;
        }
    }
}


void TIM1_UP_IRQHandler(void)
{
    if( TIM_GetITStatus( TIM1, TIM_IT_Update ) != RESET )
    {
        tick_us += 1000;
        tick_ms++;
        TIM_ClearITPendingBit( TIM1, TIM_IT_Update);
    }
}

void TIM5_IRQHandler(void)
{
    if( TIM_GetITStatus( TIM5, TIM_IT_Update ) != RESET )
    {
        TIM_ClearITPendingBit(TIM5, TIM_IT_Update);
    }

}


void tim_init(void)
{
    TIM_BASE_Timer1();
    TIM_PWM_Init();
    //TIM_BASE_Timer5();
}

void TIM_PWM_Init()
{
    TIM_PWM_Timer2();
    TIM_PWM_Timer3();
    TIM_PWM_Timer4();
}


void TIM_BASE_Timer1(void)
{
    TIM_TimeBaseInitTypeDef TIM_TimeBaseInitStructure={0};
    NVIC_InitTypeDef        NVIC_InitStructure = {0};

    RCC_APB2PeriphClockCmd( RCC_APB2Periph_TIM1, ENABLE );

    TIM_TimeBaseInitStructure.TIM_Prescaler = SystemCoreClock/1000000 - 1;//1M
    TIM_TimeBaseInitStructure.TIM_Period = 1000000 / TIM1_FRQ - 1;//1000hz
    TIM_TimeBaseInitStructure.TIM_ClockDivision = TIM_CKD_DIV1;
    TIM_TimeBaseInitStructure.TIM_CounterMode = TIM_CounterMode_Up;
    TIM_TimeBaseInit( TIM1, &TIM_TimeBaseInitStructure);

    TIM_ARRPreloadConfig( TIM1, ENABLE );
    TIM_ITConfig(TIM1, TIM_IT_Update, ENABLE);
    TIM_Cmd( TIM1, ENABLE );

    NVIC_InitStructure.NVIC_IRQChannel = TIM1_UP_IRQn;
    NVIC_InitStructure.NVIC_IRQChannelPreemptionPriority = 0;
    NVIC_InitStructure.NVIC_IRQChannelSubPriority = 1;
    NVIC_InitStructure.NVIC_IRQChannelCmd = ENABLE;
    NVIC_Init(&NVIC_InitStructure);

}



void TIM_PWM_Timer2()
{
    GPIO_InitTypeDef GPIO_InitStructure={0};
    TIM_OCInitTypeDef TIM_OCInitStructure={0};
    TIM_TimeBaseInitTypeDef TIM_TimeBaseInitStructure={0};

    RCC_APB2PeriphClockCmd( RCC_APB2Periph_GPIOA, ENABLE );
    RCC_APB1PeriphClockCmd( RCC_APB1Periph_TIM2, ENABLE );

    GPIO_InitStructure.GPIO_Pin = GPIO_Pin_0 | GPIO_Pin_1 | GPIO_Pin_2 | GPIO_Pin_3;
    GPIO_InitStructure.GPIO_Mode = GPIO_Mode_AF_PP;
    GPIO_InitStructure.GPIO_Speed = GPIO_Speed_50MHz;
    GPIO_Init( GPIOA, &GPIO_InitStructure );

    TIM_TimeBaseInitStructure.TIM_Prescaler = SystemCoreClock/1000000 - 1;//1M
    TIM_TimeBaseInitStructure.TIM_Period = 1000000 / TIM2_FRQ - 1;//50hz
    TIM_TimeBaseInitStructure.TIM_ClockDivision = TIM_CKD_DIV1;
    TIM_TimeBaseInitStructure.TIM_CounterMode = TIM_CounterMode_Up;
    TIM_TimeBaseInit( TIM2, &TIM_TimeBaseInitStructure);


    TIM_OCInitStructure.TIM_OCMode = TIM_OCMode_PWM1;
    TIM_OCInitStructure.TIM_OutputState = TIM_OutputState_Enable;
    TIM_OCInitStructure.TIM_OCPolarity = TIM_OCPolarity_High;

    TIM_OCInitStructure.TIM_Pulse = PWM_CH3;
    TIM_OC1Init( TIM2, &TIM_OCInitStructure );

    TIM_OCInitStructure.TIM_Pulse = PWM_CH4;
    TIM_OC2Init( TIM2, &TIM_OCInitStructure );

    TIM_OCInitStructure.TIM_Pulse = PWM_CH5;
    TIM_OC3Init( TIM2, &TIM_OCInitStructure );

    TIM_OCInitStructure.TIM_Pulse = PWM_CH6;
    TIM_OC4Init( TIM2, &TIM_OCInitStructure );


    TIM_OC1PreloadConfig( TIM2, TIM_OCPreload_Enable );
    TIM_OC2PreloadConfig( TIM2, TIM_OCPreload_Enable );
    TIM_OC3PreloadConfig( TIM2, TIM_OCPreload_Enable );
    TIM_OC4PreloadConfig( TIM2, TIM_OCPreload_Enable );

    TIM_ARRPreloadConfig( TIM2, ENABLE );

    TIM_CtrlPWMOutputs(TIM2, ENABLE );
    TIM_Cmd( TIM2, ENABLE );

}


void TIM_PWM_Timer3()
{
    GPIO_InitTypeDef GPIO_InitStructure={0};
    TIM_OCInitTypeDef TIM_OCInitStructure={0};
    TIM_TimeBaseInitTypeDef TIM_TimeBaseInitStructure={0};

    RCC_APB2PeriphClockCmd( RCC_APB2Periph_GPIOC | RCC_APB2Periph_AFIO, ENABLE );
    RCC_APB1PeriphClockCmd( RCC_APB1Periph_TIM3, ENABLE );

    GPIO_PinRemapConfig(GPIO_FullRemap_TIM3,ENABLE);


    GPIO_InitStructure.GPIO_Pin = GPIO_Pin_6 | GPIO_Pin_7 | GPIO_Pin_8 | GPIO_Pin_9;
    GPIO_InitStructure.GPIO_Mode = GPIO_Mode_AF_PP;
    GPIO_InitStructure.GPIO_Speed = GPIO_Speed_50MHz;
    GPIO_Init( GPIOC, &GPIO_InitStructure );

    TIM_TimeBaseInitStructure.TIM_Prescaler = SystemCoreClock/1000000 - 1;//1M
    TIM_TimeBaseInitStructure.TIM_Period = 1000000 / TIM3_FRQ - 1;//1khz
    TIM_TimeBaseInitStructure.TIM_ClockDivision = TIM_CKD_DIV1;
    TIM_TimeBaseInitStructure.TIM_CounterMode = TIM_CounterMode_Up;
    TIM_TimeBaseInit( TIM3, &TIM_TimeBaseInitStructure);


    TIM_OCInitStructure.TIM_OCMode = TIM_OCMode_PWM1;
    TIM_OCInitStructure.TIM_OutputState = TIM_OutputState_Enable;
    TIM_OCInitStructure.TIM_OCPolarity = TIM_OCPolarity_High;

    TIM_OCInitStructure.TIM_Pulse = MOT_CH1;
    TIM_OC1Init( TIM3, &TIM_OCInitStructure );

    TIM_OCInitStructure.TIM_Pulse = MOT_CH2;
    TIM_OC2Init( TIM3, &TIM_OCInitStructure );

    TIM_OCInitStructure.TIM_Pulse = MOT_CH3;
    TIM_OC3Init( TIM3, &TIM_OCInitStructure );

    TIM_OCInitStructure.TIM_Pulse = MOT_CH4;
    TIM_OC4Init( TIM3, &TIM_OCInitStructure );


    TIM_OC1PreloadConfig( TIM3, TIM_OCPreload_Enable );
    TIM_OC2PreloadConfig( TIM3, TIM_OCPreload_Enable );
    TIM_OC3PreloadConfig( TIM3, TIM_OCPreload_Enable );
    TIM_OC4PreloadConfig( TIM3, TIM_OCPreload_Enable );

    TIM_ARRPreloadConfig( TIM3, ENABLE );

    TIM_CtrlPWMOutputs(TIM3, ENABLE );
    TIM_Cmd( TIM3, ENABLE );
}

void TIM_PWM_Timer4()
{
    GPIO_InitTypeDef GPIO_InitStructure={0};
    TIM_OCInitTypeDef TIM_OCInitStructure={0};
    TIM_TimeBaseInitTypeDef TIM_TimeBaseInitStructure={0};

    RCC_APB2PeriphClockCmd( RCC_APB2Periph_GPIOB, ENABLE );
    RCC_APB1PeriphClockCmd( RCC_APB1Periph_TIM4, ENABLE );

    GPIO_InitStructure.GPIO_Pin = GPIO_Pin_8 | GPIO_Pin_9;
    GPIO_InitStructure.GPIO_Mode = GPIO_Mode_AF_PP;
    GPIO_InitStructure.GPIO_Speed = GPIO_Speed_50MHz;
    GPIO_Init( GPIOB, &GPIO_InitStructure );

    TIM_TimeBaseInitStructure.TIM_Prescaler = SystemCoreClock/1000000 - 1;//1M
    TIM_TimeBaseInitStructure.TIM_Period = 1000000 / TIM4_FRQ - 1;//50hz
    TIM_TimeBaseInitStructure.TIM_ClockDivision = TIM_CKD_DIV1;
    TIM_TimeBaseInitStructure.TIM_CounterMode = TIM_CounterMode_Up;
    TIM_TimeBaseInit( TIM4, &TIM_TimeBaseInitStructure);

    TIM_OCInitStructure.TIM_OCMode = TIM_OCMode_PWM1;
    TIM_OCInitStructure.TIM_OutputState = TIM_OutputState_Enable;
    TIM_OCInitStructure.TIM_OCPolarity = TIM_OCPolarity_High;

    TIM_OCInitStructure.TIM_Pulse = PWM_CH1;
    TIM_OC3Init( TIM4, &TIM_OCInitStructure );

    TIM_OCInitStructure.TIM_Pulse = PWM_CH2;
    TIM_OC4Init( TIM4, &TIM_OCInitStructure );

    TIM_OC1PreloadConfig( TIM4, TIM_OCPreload_Enable );
    TIM_OC2PreloadConfig( TIM4, TIM_OCPreload_Enable );
    TIM_OC3PreloadConfig( TIM4, TIM_OCPreload_Enable );
    TIM_OC4PreloadConfig( TIM4, TIM_OCPreload_Enable );

    TIM_ARRPreloadConfig( TIM4, ENABLE );

    TIM_CtrlPWMOutputs(TIM4, ENABLE );
    TIM_Cmd( TIM4, ENABLE );

}

void TIM_BASE_Timer5()
{
    TIM_TimeBaseInitTypeDef TIM_TimeBaseInitStructure={0};
    NVIC_InitTypeDef        NVIC_InitStructure = {0};

    RCC_APB1PeriphClockCmd( RCC_APB1Periph_TIM5, ENABLE );

    RCC_PCLK1Config(RCC_HCLK_Div1);

    TIM_TimeBaseInitStructure.TIM_Prescaler = SystemCoreClock/1000000 - 1;//1M
    TIM_TimeBaseInitStructure.TIM_Period = 1000000 / TIM5_FRQ - 1;//1000hz
    TIM_TimeBaseInitStructure.TIM_ClockDivision = TIM_CKD_DIV1;
    TIM_TimeBaseInitStructure.TIM_CounterMode = TIM_CounterMode_Up;
    TIM_TimeBaseInit( TIM5, &TIM_TimeBaseInitStructure);

    TIM_ARRPreloadConfig( TIM5, ENABLE );
    TIM_ITConfig(TIM5, TIM_IT_Update, ENABLE);
    TIM_Cmd( TIM5, ENABLE );

    NVIC_InitStructure.NVIC_IRQChannel = TIM5_IRQn;
    NVIC_InitStructure.NVIC_IRQChannelPreemptionPriority = 0;
    NVIC_InitStructure.NVIC_IRQChannelSubPriority = 1;
    NVIC_InitStructure.NVIC_IRQChannelCmd = ENABLE;
    NVIC_Init(&NVIC_InitStructure);


}

void tim_setPWM(uint8_t channel, uint16_t width)
{
    switch(channel)
    {
        case 0:TIM_SetCompare3(TIM4, width);break;
        case 1:TIM_SetCompare4(TIM4, width);break;
        case 2:TIM_SetCompare1(TIM2, width);break;
        case 3:TIM_SetCompare2(TIM2, width);break;
        case 4:TIM_SetCompare3(TIM2, width);break;
        case 5:TIM_SetCompare4(TIM2, width);break;
    }
}

void tim_setMOT(uint8_t channel, uint16_t width)
{
    switch(channel)
    {
        case 0:TIM_SetCompare1(TIM3, width);break;
        case 1:TIM_SetCompare2(TIM3, width);break;
        case 2:TIM_SetCompare3(TIM3, width);break;
        case 3:TIM_SetCompare4(TIM3, width);break;
    }
}