#include "bsp/adc.h"
#include "libs/utils.h"
#include "os/co_task.h"
#include "libs/logger.h"
static void _gpio_init(void);

void adc_init(void){
	rcu_adc_clock_config(RCU_AF);
	/* init adc input gpio */
	_gpio_init();
    /* config ADC clock */
    rcu_adc_clock_config(RCU_CKADC_CKAPB2_DIV4); //APB2 clk 120M, adc clk 30M

	rcu_periph_clock_enable(RCU_ADC0);
	rcu_periph_clock_enable(RCU_ADC1);
#if SHUNT_NUM==THREE_SHUNTS_SAMPLE
	adc_mode_config(ADC_DAUL_INSERTED_PARALLEL);
#else
	adc_mode_config(ADC_MODE_FREE);
#endif
    /* ADC scan mode function enable */
    adc_special_function_config(ADC0, ADC_SCAN_MODE, ENABLE);
    adc_special_function_config(ADC1, ADC_SCAN_MODE, ENABLE);

    /* configure ADC data alignment */
    adc_data_alignment_config(ADC0, ADC_DATAALIGN_RIGHT);
    adc_data_alignment_config(ADC1, ADC_DATAALIGN_RIGHT);
	
    /* configure ADC inserted channel trigger */
    adc_external_trigger_source_config(ADC0, ADC_INSERTED_CHANNEL, ADC_TRIGGER_PHASE);
    /* configure ADC inserted channel length */
    adc_channel_length_config(ADC0, ADC_INSERTED_CHANNEL, 1);
    /* configure ADC inserted channel trigger */
#if SHUNT_NUM==THREE_SHUNTS_SAMPLE	
    adc_external_trigger_source_config(ADC1, ADC_INSERTED_CHANNEL, ADC_TRIGGER_NONE);
#else
	adc_external_trigger_source_config(ADC1, ADC_INSERTED_CHANNEL, ADC_TRIGGER_PHASE2);
#endif
    /* configure ADC inserted channel length */
    adc_channel_length_config(ADC1, ADC_INSERTED_CHANNEL, 1);
    /* configure ADC inserted channel */
#if SHUNT_NUM==THREE_SHUNTS_SAMPLE		
    adc_inserted_channel_config(ADC0, 0, U_PHASE_I_CHAN, ADC_SAMPLE_TIME);
    adc_inserted_channel_config(ADC1, 0, V_PHASE_I_CHAN, ADC_SAMPLE_TIME);

	adc_update_insert_sample_time(ADC0, U_PHASE_I_CHAN, ADC_SAMPLE_TIME);
	adc_update_insert_sample_time(ADC0, V_PHASE_I_CHAN, ADC_SAMPLE_TIME);
	adc_update_insert_sample_time(ADC0, W_PHASE_I_CHAN, ADC_SAMPLE_TIME);

	adc_update_insert_sample_time(ADC1, U_PHASE_I_CHAN, ADC_SAMPLE_TIME);
	adc_update_insert_sample_time(ADC1, V_PHASE_I_CHAN, ADC_SAMPLE_TIME);
	adc_update_insert_sample_time(ADC1, W_PHASE_I_CHAN, ADC_SAMPLE_TIME);
#else
	adc_inserted_channel_config(ADC0, 0, VBUS_I_CHAN, ADC_SAMPLE_TIME);
	adc_inserted_channel_config(ADC1, 0, VBUS_I_CHAN, ADC_SAMPLE_TIME);
#endif
    /* ADC external trigger enable */
    adc_external_trigger_config(ADC0, ADC_INSERTED_CHANNEL, ENABLE);
    adc_external_trigger_config(ADC1, ADC_INSERTED_CHANNEL, ENABLE);
#if SHUNT_NUM==THREE_SHUNTS_SAMPLE
    adc_interrupt_disable(ADC0, ADC_INT_EOIC);
	adc_interrupt_disable(ADC1, ADC_INT_EOIC);
#else
	adc_interrupt_disable(ADC0, ADC_INT_EOIC);
	adc_interrupt_disable(ADC1, ADC_INT_EOIC);
#endif
    /* configure ADC regular channel trigger */
    adc_external_trigger_source_config(ADC0, ADC_REGULAR_CHANNEL, ADC0_1_2_EXTTRIG_REGULAR_NONE);
    adc_external_trigger_config(ADC0, ADC_REGULAR_CHANNEL, ENABLE);

    /* enable ADC interface */
    adc_enable(ADC0);
    adc_enable(ADC1);

    /* ADC calibration and reset calibration */
    adc_calibration_enable(ADC0);
    adc_calibration_enable(ADC1);

	nvic_irq_enable(ADC0_1_IRQn, ADC_IRQ_PRIORITY, 0);

	adc_disable_ext_trigger();
}

static void _gpio_init(void) {
	rcu_periph_clock_enable(RCU_GPIOA);
	rcu_periph_clock_enable(RCU_GPIOB);
	gpio_init(GPIOA, GPIO_MODE_AIN, GPIO_OSPEED_50MHZ, GPIO_PIN_0|GPIO_PIN_1|GPIO_PIN_2|GPIO_PIN_3|GPIO_PIN_4|GPIO_PIN_5|GPIO_PIN_6|GPIO_PIN_7);
	gpio_init(GPIOB, GPIO_MODE_AIN, GPIO_OSPEED_50MHZ, GPIO_PIN_0|GPIO_PIN_1);
}


void adc_start_insert_convert(void) {
	int drop = 2;
    /* clear the ADC flag */
    adc_flag_clear(ADC0, ADC_FLAG_EOIC);
    adc_flag_clear(ADC1, ADC_FLAG_EOIC);

	//adc_enable_ext_trigger();
	while(drop-- > 0) {
		while ((adc_flag_get(ADC0, ADC_FLAG_EOIC) == RESET) || (adc_flag_get(ADC1, ADC_FLAG_EOIC) == RESET));
		adc_flag_clear(ADC0, ADC_FLAG_EOIC);
		adc_flag_clear(ADC1, ADC_FLAG_EOIC);
	}
    /* enable ADC interrupt */
#if SHUNT_NUM==THREE_SHUNTS_SAMPLE
	adc_interrupt_enable(ADC0, ADC_INT_EOIC);
#else
	adc_interrupt_enable(ADC1, ADC_INT_EOIC);
#endif

}

void adc_stop_insert_convert(void) {
	adc_disable_ext_trigger();
    /* disable ADC interrupt */
#if SHUNT_NUM==THREE_SHUNTS_SAMPLE
	adc_interrupt_disable(ADC0, ADC_INT_EOIC);
#else
	adc_interrupt_disable(ADC1, ADC_INT_EOIC);
#endif
    /* clear the ADC flag */
    adc_flag_clear(ADC0, ADC_FLAG_EOIC);
	adc_flag_clear(ADC1, ADC_FLAG_EOIC);
}


s32 adc_sample_regular_channel(int channel, int times) {
#if 1
	u32 adc_device = ADC0;
	int value = 0;
	int count = 0;
	int min = 0xFFFFF;
	int max = -0xFFFFF;
	u64 start_time;
	adc_channel_length_config(adc_device, ADC_REGULAR_CHANNEL, 1);
	adc_regular_channel_config(adc_device, 0, channel, ADC_SAMPLETIME_55POINT5);
	while(count < times){
restart:		
		start_time = co_task_sys64_ticks();
		adc_software_trigger_enable(adc_device, ADC_REGULAR_CHANNEL);
    	while(SET != adc_flag_get(adc_device, ADC_FLAG_EOC)){
			if (co_task_sys64_ticks() - start_time >= 2){
				goto restart;
			}
		};
    	int one = adc_regular_data_read(adc_device);
		adc_flag_clear(adc_device, ADC_FLAG_EOC);		
		value += (one & 0xFFF);
		count ++;
		if (one > max){
			max = one;
		}
		if (one < min) {
			min = one;
		}
	}
	if (times <= 2) {
		return value/times;
	}
	return (value - min - max)/(times-2);
#else
	return 0;
#endif
}