#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 "bsp/cht8305.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 int enable_wakeup_irq(void){
	//hall_1_detect_irq_enable(1);
	//hall_2_detect_irq_enable(1);
	return shark_rtc_start_alarm(2);
}

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 = 10000 * 30;
	while(wait--> 0);
}

static void wait_dcdc_good(void) {
	volatile uint32_t wait_cycle = 2000;
	while(--wait_cycle > 0 && (!IS_DCDC_POWER_GOOD()));
}
static int pre_deepsleep(void){
	LED_ALL_ON(0);
	DIANOCTIC_LED(0);
	_wakeup_source = 0;
	systick_close();
	shark_rtc_set_backup(0xF0);
	shark_uart_deinit(SHARK_UART0);
#if (CONFIG_BOARD_TYPE==SHARK_BOARD_SP700)
	shark_uart_deinit(SHARK_UART1);
#endif
	wdog_reload();
	shark_rtc_set_backup(0xF1);
	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);
	}
	shark_rtc_set_backup(0xF2);
	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){
		shark_rtc_set_backup(0xF3);
		post_deepsleep();
		return -1;
	}
	shark_rtc_set_backup(0xF4);
	if (wdog_set_timeout(WDOG_TIME_FOR_SLEEP) < 0){
		post_deepsleep();
		return -1;		
	}
	shark_rtc_set_backup(0xF5);
	wait_for_enter_dsleep();
	shark_rtc_set_backup(0xF6);
	if (_is_wakeup_source()) {
		_wakeup_source = 0;
		post_deepsleep();
		return -1;
	}
	shark_rtc_set_backup(0xF7);
	return 0;
}

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

void mcu_enter_deepsleep(void){
	_sleep_second_time_now = 0;
	if (pre_deepsleep()< 0){
		return;
	}
	_wakeup_source = 0;
	if (enable_wakeup_irq()< 0){
		disable_wakeup_irq();
		post_deepsleep();
		return;
	}
	shark_rtc_set_backup(0xFF0);
	do {
		u32 start_time = shark_rtc_get_second();
		pmu_to_deepsleepmode(PMU_LDO_LOWPOWER, WFI_CMD);
		shark_rtc_set_backup(0xFF1);
		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 (IS_HALL1_DETECTED()|| IS_HALL2_DETECTED()) {
			mcu_sleep_set_wakeup_source(WAKEUP_SOURCE_HALL1);
		}
		if (_wakeup_source & WAKEUP_SOURCE_RTC) {
			wdog_reload();
			shark_rtc_set_backup(0xFF2);
			while (shark_rtc_start_alarm(RTC_ALARM_FOR_SLEEP) < 0){
				wdog_reload();
				shark_rtc_set_backup(0xFF3);
				if (_is_wakeup_source() || (_sleep_second_time_now >= (60))){
					break;
				}
			}
			shark_rtc_set_backup(0xFF4);
			wait_for_enter_dsleep();
			shark_rtc_set_backup(0xFF5);
		}
	}while(!_is_wakeup_source() && (_sleep_second_time_now < (60)));
	_sleep_second_time += _sleep_second_time_now;
	disable_wakeup_irq();
	shark_rtc_set_backup(0xFF6);
	post_deepsleep();
}

uint32_t mcu_get_sleeptime(void){
	return _sleep_second_time_now;
}