#include "soc.h"
#include "app/sox/measure.h"
#include "app/sox/measure_task.h"
#include "app/nv_storage.h"
#include "libs/logger.h"
#include "health.h"
#include "state.h"

static soc_t _soc;
static uint8_t chargering = 0;
static u64     time_ms = 0;
static float _charger_coefficient = 1.0f;
static float _discharger_coefficient = 1.0f;
uint32_t charger_remain_time = 0;
#define DEFALUT_MAX_COULOMB (MAX_HA * 3600.0f)

static void calibrate_soc_by_ocv(void);


void soc_init(void){
	set_log_level(MOD_SOC, L_debug);
	time_ms = shark_get_mseconds();
	if (nv_restore_soc() != 0){
		soc_warning("SOC: nv storage is not inited, use default value!!\n");
		_soc.coulomb_min = 0;
		_soc.coulomb_max = DEFALUT_MAX_COULOMB; //30HA,这个值最总需要soh模块给
		_soc.flags = 0;
		_soc.charger_coulomb = 0;
		_soc.pre_charger_coulomb = 0;
		_soc.dischrger_coulomb = 0;
		_soc.pre_discharger_coulomb = 0;
		_soc.total_coulomb = 0;
	}
	if ((_soc.flags & SOC_FLAG_CALIBRATED) == 0){
		calibrate_soc_by_ocv();
		nv_save_soc();
	}
}

//初始上电或者nv出问题后，通过开路电压对soc做一次初略校准
static void calibrate_soc_by_ocv(void){
	uint16_t pack_vol = 0;
	for (int i = 0; i < CELLS_NUM; i++){
		pack_vol += measure_value()->cell_vol[i];
	}
	if (pack_vol < (2700 * CELLS_NUM)){
		_soc.capacity = 0;
	}else if (pack_vol < (2950 * CELLS_NUM)){
		_soc.capacity = 5;
	}else if (pack_vol < (3200 * CELLS_NUM)){
		_soc.capacity = 15;
	}else if (pack_vol < (3400 * CELLS_NUM)){
		_soc.capacity = 25;
	}else if (pack_vol < (3500 * CELLS_NUM)){
		_soc.capacity = 85;
	}else if (pack_vol < (3550 * CELLS_NUM)){
		_soc.capacity = 95;
	}else {
		_soc.capacity = 100;
	}
	_soc.coulomb_now = (_soc.coulomb_max - _soc.coulomb_min) * _soc.capacity / 100.0f + _soc.coulomb_min;
	soc_warning("SOC: calibrate_soc_by_ocv -> capacity = %d, pack_voltage = %d\n", _soc.capacity, pack_vol);
}

static __inline__ float _delta_time(void){
	u32 delta = shark_get_mseconds() - time_ms;
	time_ms = shark_get_mseconds();
	return  (float)delta / (1000.0f); //秒
}

void soc_update_by_ocv(void){
	if (_soc.flags & SOC_FLAG_CALIBRATED){
		if (!chargering && bms_health()->powerdown_lower_voltage){
			_soc.coulomb_min = _soc.coulomb_now; //已经校准过了，而且电池进入powerdown，最小容量修正为当前容量
			_soc.capacity = 0;
			soc_warning("current coulomb %f\n", _soc.coulomb_now);
			return;
		}
		if (chargering){
			if (bms_state_get()->pack_voltage >= (54000)){
				_soc.capacity = 100;
			}
		}
	}else {
		if (chargering){//用ocv进行严格校准
			if (measure_value()->load_current <= 300.0f){				
				//判断总电压
				if (bms_state_get()->pack_voltage >= 54000){
					_soc.capacity = 100;
				}
			}
		}
	}
}

static void soc_update_charger_remain_time(void){
	if (!chargering) {
		return;
	}
	float delta_c = _soc.coulomb_max - _soc.coulomb_now;
	float current = measure_value()->load_current / 1000.0f; //A
	uint32_t remain = delta_c / current / 60; //分钟
	if (charger_remain_time == 0){
		charger_remain_time = remain;
	}else if (remain < charger_remain_time){
		charger_remain_time = remain;
	}
	if (_soc.capacity == 100) {
		charger_remain_time = 0;
	}
}

uint32_t soc_get_cycle(void){
	return _soc.total_coulomb/MAX_HA;
}

uint32_t soc_get_charger_remain_time(void){
	return charger_remain_time;
}

void soc_update(void){
	if (!chargering && bms_state_get()->charging){
		_soc.pre_charger_coulomb = _soc.charger_coulomb;
		_soc.charger_coulomb = 0;//clear charing
		_soc.total_coulomb += _soc.pre_charger_coulomb / 3600.0f;
		chargering = 1;
	}else if (chargering && !bms_state_get()->charging){
		_soc.pre_discharger_coulomb = _soc.dischrger_coulomb;
		_soc.dischrger_coulomb = 0; //clear discharger
		_soc.total_coulomb += _soc.pre_discharger_coulomb / 3600.0f;
		chargering = 0;
	}
	float current = measure_value()->load_current / 1000.0f; //A
	float delta_q = current * _delta_time();
	if (chargering){
		delta_q = delta_q * _charger_coefficient;
		_soc.charger_coulomb += abs(delta_q);
	}else {
		delta_q = delta_q * _discharger_coefficient;
		_soc.dischrger_coulomb += abs(delta_q); //转为正数
	}
	_soc.coulomb_now += delta_q; //充电加, 放电减

	if (_soc.coulomb_now > _soc.coulomb_max){
		_soc.coulomb_now = _soc.coulomb_max;
	}else if (_soc.coulomb_now < _soc.coulomb_min){
		_soc.coulomb_now = _soc.coulomb_min;
	}
	uint8_t old_cap = _soc.capacity;
	_soc.capacity = (_soc.coulomb_now - _soc.coulomb_min)/(_soc.coulomb_max - _soc.coulomb_min) * 100;
	if (chargering && (_soc.capacity == 100)){
		_soc.capacity = 99;//充电的时候必须通过ovc才能把电量校准到100
	}else if (!chargering && (_soc.capacity == 0)){
		_soc.capacity = 1;
	}
	//通过电压校准SOC，只能在电压范围的两端校准
	soc_update_by_ocv();

	//如果没有校准过，充电过程中，电量100%后，设置校准标志位
	if (chargering && (_soc.capacity == 100)){
		if ((_soc.flags & SOC_FLAG_CALIBRATED) == 0){
			_soc.coulomb_now = _soc.coulomb_max;
			_soc.flags |= SOC_FLAG_CALIBRATED;
			nv_save_soc();
			soc_warning("calibrate OK, charging coulomb: %f\n", _soc.charger_coulomb);
		}else { //如果校准过，单电芯过压，100%的容量，设置最大容量为当前容量
			if (bms_health()->sigle_cell_over_voltage){
				_soc.coulomb_max = _soc.coulomb_now;
				soc_warning("signal cell over vol, cap full, reset coul max to coul now: %f\n", _soc.coulomb_max);
			}
		}
	}
	_soc.energy = bms_state_get()->pack_voltage/1000.f * (_soc.coulomb_now - _soc.coulomb_min) * _soc.capacity/100.0f;
	if (old_cap != _soc.capacity) {
		nv_save_soc();
	}

	soc_update_charger_remain_time();
	
}

soc_t *get_soc(void){
	return &_soc;
}