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

shark_charger_state_t shark_charger_state;
shark_bool shark_charger_enabled;
shark_u16 shark_charge_times;

static shark_u16 shark_charger_remove;
static shark_u16 shark_charger_insert;
static 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_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) {
		enable = shark_false;
	}

	if (enable) {
		gpio_bit_set(GPIO_PORT_CHG_EN, GPIO_PIN_CHG_EN);
	} else {
		gpio_bit_reset(GPIO_PORT_CHG_EN, GPIO_PIN_CHG_EN);
	}

	shark_charger_enabled = enable;
	shark_charge_times = 0;
}

void shark_charger_disable(void)
{
	if (shark_charge_times > SHARK_CHG_TIME_DETECT) {
		shark_charger_set_enable(shark_false);
		shark_charger_insert = 0;
		shark_charger_state = SHARK_CHG_DETECT;
	}
}

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 > SHARK_CHG_INSERT_MIN) {
			if (shark_xl_check_with_qd() == shark_false) {
				shark_charger_state = SHARK_CHG_INSERT;
			}

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

		if (shark_charger_remove > SHARK_CHG_REMOVE_MIN) {
			shark_charger_state = SHARK_CHG_REMOVE;
		} else {
			shark_charger_remove++;
		}
	}
}

void shark_charge_poll(void)
{
	if (shark_charger_state == SHARK_CHG_INSERT) {
		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) {
					if (sub_bms_info_2.packet_common.charge_flag == 0) {
						shark_charger_disable();
					}

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