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

shark_bool shark_charger_detected;
shark_bool shark_charger_enabled;

shark_u16 shark_charge_times;
shark_bool shark_battery_full;

static shark_u16 shark_charger_remove;
static shark_u16 shark_charger_insert;

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_bit_reset(GPIO_PORT_CHG_EN, GPIO_PIN_CHG_EN);
	gpio_init(GPIO_PORT_CHG_EN, GPIO_MODE_OUT_PP, GPIO_OSPEED_50MHZ, GPIO_PIN_CHG_EN);
}

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(u16 times)
{
	if (shark_charge_times > times) {
		shark_charger_set_enable(shark_false);
	}
}

void shark_charge_tick(void)
{
	if (gpio_input_bit_get(GPIO_PORT_CHG_DET, GPIO_PIN_CHG_DET) == RESET) {
		shark_charger_remove = 0;

		if (shark_charger_insert < 50) {
			shark_charger_insert++;
		} else if (shark_xl_check() == shark_false) {
			shark_charger_detected = shark_true;

			if (shark_charger_enabled && shark_charge_times < SHARK_CHG_TIME_MAX) {
				shark_charge_times++;
			}
		}
	} else {
		shark_charger_insert = 0;

		if (shark_charger_remove < 2000) {
			shark_charger_remove++;
		} else {
			shark_charger_detected = shark_false;
		}
	}
}

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(SHARK_CHG_TIME_MIN);
				}
				break;

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

			case CB_BAT1_BAT2_PARRALLEL:
				if (sub_bms_info_1.packet_common.charge_flag == 0) {
					if (sub_bms_info_2.packet_common.charge_flag == 0) {
						shark_charger_disable(SHARK_CHG_TIME_MIN);
					}

					if (shark_battery_get_current12() < 0) {
						shark_charger_disable(SHARK_CHG_TIME_MIN);
					}
				} else if (sub_bms_info_2.packet_common.charge_flag == 0 && shark_battery_get_current12() < 0) {
					shark_charger_disable(SHARK_CHG_TIME_MIN);
				}
				break;
			}
		} else {
			shark_charger_set_enable(shark_true);
		}
	}
}