#include "shark_charge.h"
#include "app_rs485_1.h"
#include "app.h"

shark_bool shark_charger_detected;
shark_bool shark_charger_enabled;
shark_u16 shark_charger_remove;

shark_u16 shark_charge_times;
shark_bool shark_battery_full;

void shark_charge_init(void)
{
	rcu_periph_clock_enable(GPIO_RCU_CHG_DET);
	gpio_init(GPIO_PORT_CHG_DET, GPIO_MODE_IN_FLOATING, GPIO_OSPEED_50MHZ, GPIO_PIN_CHG_DET);

	rcu_periph_clock_enable(GPIO_RCU_CHG_EN);
	gpio_init(GPIO_PORT_CHG_EN, GPIO_MODE_OUT_PP, GPIO_OSPEED_50MHZ, GPIO_PIN_CHG_EN);

	gpio_exti_source_select(EXTI_PORT_CHG_DET, EXTI_PIN_CHG_DET);
	exti_init(EXTI_CHG_DET, EXTI_INTERRUPT, EXTI_TRIG_FALLING);
	exti_interrupt_flag_clear(EXTI_CHG_DET);
	nvic_irq_enable(EXTI4_IRQn, 1, 0);
}

void shark_charger_set_enable(shark_bool enable)
{
	if (sub_bms_info_1.packet_common.m_percent > 99 && sub_bms_info_2.packet_common.m_percent > 99) {
		shark_battery_full = shark_true;
	}

	if (sub_bms_info_1.packet_common.m_percent < 97 || sub_bms_info_2.packet_common.m_percent < 97) {
		shark_battery_full = shark_false;
	}

	if (shark_battery_full || cb_operate_state == CB_BAT1_BAT2_SERIES || shark_charger_remove) {
		enable = shark_false;
	}

	if (shark_charger_enabled != enable) {
		shark_charger_enabled = enable;
		shark_charge_times = 0;

		if (enable) {
			gpio_bit_set(GPIO_PORT_CHG_EN, GPIO_PIN_CHG_EN);
			println("charger enabled");
		} else {
			gpio_bit_reset(GPIO_PORT_CHG_EN, GPIO_PIN_CHG_EN);
			shark_charger_remove = SHARK_CHG_DEBOUNCE;
			println("charger disabled");
		}
	}
}

void shark_charger_disable(void)
{
	if (shark_charge_times > SHARK_CHG_TIME_MIN) {
		shark_charger_set_enable(shark_false);
	}
}

void shark_charge_tick(void)
{
	if (shark_charger_remove > 0) {
		if (shark_charger_remove > 1) {
			shark_charger_remove--;
		} else {
			shark_charger_remove = 0;
			shark_charger_detected = shark_false;
		}
	}

	if (shark_charger_enabled && shark_charge_times < SHARK_CHG_TIME_MAX) {
		shark_charge_times++;
	}
}

void shark_charge_poll(void)
{
	if (shark_charger_detected) {
		if (shark_charger_enabled) {
			switch (cb_operate_state) {
			case CB_BAT1:
				if (sub_bms_info_1.packet_common.charge_flag == 0) {
					shark_charger_disable();
				}
				break;

			case CB_BAT2:
				if (sub_bms_info_2.packet_common.charge_flag == 0) {
					shark_charger_disable();
				}
				break;

			case CB_BAT1_BAT2_PARRALLEL:
				if (sub_bms_info_1.packet_common.charge_flag == 0 && sub_bms_info_2.packet_common.charge_flag == 0) {
					shark_charger_disable();
				}
				break;
			}
		} else {
			shark_charger_set_enable(shark_true);
		}
	}
}

void EXTI4_IRQHandler(void)
{
	if (exti_flag_get(EXTI_4) != RESET) {
		exti_flag_clear(EXTI_4);
		shark_charger_remove = 0;
		shark_charger_detected = shark_true;
	}
}