#include "mcu_power_sleep.h"
#include "bsp/ml5238.h"
#include "bsp/cs1180.h"
#include "bsp/gpio.h"
#include "bsp/spi.h"
#include "bsp/uart.h"
#include "bsp/i2c.h"
#include "bsp/shark_rtc.h"
#include "bsp/AT24CXX.h"
#include "app/sox/iostate.h"
#include "bsp/gd32_adc.h"

extern void system_clock_24m_irc8m(void);
extern void system_clock_config(void);
extern void SystemCoreClockUpdate(void);

extern void current_calibrate(void);
extern void adc_init(void);
extern void adc_deinit(void);
static uint32_t _sleep_second_time = 0;
static uint32_t _sleep_second_time_now = 0;
static uint32_t _wakeup_source = 0;

static void post_deepsleep(void);

#define WDOG_TIME_FOR_SLEEP 15
#define RTC_ALARM_FOR_SLEEP 10 //rtc alarm time MUST be small than wdog time!!!!
uint32_t get_system_sleep_time(void){
	return _sleep_second_time;
}

static void enable_wakeup_irq(void){
	hall_1_detect_irq_enable(1);
	hall_2_detect_irq_enable(1);
	shark_rtc_start_alarm(RTC_ALARM_FOR_SLEEP);
}

static void disable_wakeup_irq(void){
	shark_rtc_stop_alarm();
	hall_1_detect_irq_enable(0);
	hall_2_detect_irq_enable(0);
}

static int _is_wakeup_source(void){
	return (_wakeup_source & (~WAKEUP_SOURCE_RTC)) != 0;
}

void mcu_sleep_set_wakeup_source(uint32_t source){
	_wakeup_source |= source;
}

/* 在reload命令及deepsleep/standby模式命令中间插入（3个以上）IRC40K时钟间隔 */
static void wait_for_enter_dsleep(void){
	volatile uint32_t wait = 1000 * 30;
	while(wait--> 0);
}

static int pre_deepsleep(void){
	LED_ALL_ON(0);
	systick_close();
	shark_uart_deinit(SHARK_UART0);
#if (CONFIG_BOARD_TYPE==SHARK_BOARD_SP700)
	shark_uart_deinit(SHARK_UART1);
#endif
	wdog_reload();
	ml5238_power_save(1); //call, before spi0_deinit
	cs1180_adc_shutdown();
	AT24CXX_DeInit();
	gd32_adc_deinit();
	if (AUX_VOL_IS_OPEN()) {
		AUX_VOL_OPEN(0);//we should close small power, before dcdc close
		delay_us(1000);
	}
	wdog_reload();
	io_state()->aux_lock_detect = 0;
	DCDC_VOL_OPEN(0);
	delay_us(1000); // give 1s to wait small current short, when dcdc is closed
	AUX_VOL_OPEN(1);
	delay_us(5000); //give 5s to detect if the small current is short
	if (io_state()->aux_lock_detect){
		post_deepsleep();
		return -1;
	}
	
	wdog_set_timeout(WDOG_TIME_FOR_SLEEP);
	wait_for_enter_dsleep();

	return 0;
}

static void post_deepsleep(void){
	DCDC_VOL_OPEN(1);
	AUX_VOL_OPEN(0);//must close small power, cs1180 cali need small power close
	SystemInit();
	system_clock_config();
	SystemCoreClockUpdate();
	ml5238_power_save(0);
	gd32_adc_init();
	cs1180_adc_init();
	shark_uart_init(SHARK_UART0);
#if (CONFIG_BOARD_TYPE==SHARK_BOARD_SP700)	
	shark_uart_init(SHARK_UART1);
#endif
	AT24CXX_Init();
	wdog_set_timeout(4);
	current_calibrate();
	wdog_reload();
	systick_open();
	AUX_VOL_OPEN(1);
}

void mcu_enter_deepsleep(void){
	if (pre_deepsleep()< 0){
		return;
	} 
	enable_wakeup_irq();
	_sleep_second_time_now = 0;
	_wakeup_source = 0;
	do {
		u32 start_time = shark_rtc_get_second();
		pmu_to_deepsleepmode(PMU_LDO_LOWPOWER, WFI_CMD);
		u32 end_time = shark_rtc_get_second();
		if (end_time >= start_time) {
			_sleep_second_time_now += end_time - start_time;
		}else { //rtc second wrap
			_sleep_second_time_now += 60 - start_time + end_time;
		}
		if (_wakeup_source & WAKEUP_SOURCE_RTC) {
			wdog_reload();
			while (shark_rtc_start_alarm(RTC_ALARM_FOR_SLEEP) < 0){
				wdog_reload();
				if (_is_wakeup_source() || (_sleep_second_time_now >= (24 * 60 * 60))){
					break;
				}
			}
			wait_for_enter_dsleep();
		}
	}while(!_is_wakeup_source() && (_sleep_second_time_now < (24 * 60 * 60)));
	_sleep_second_time += _sleep_second_time_now;
	disable_wakeup_irq();
	post_deepsleep();
}

uint32_t mcu_get_sleeptime(void){
	return _sleep_second_time_now;
}