// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (C) 2018-2020 Oplus. All rights reserved.
 */

#ifndef _OPLUS_SWITCHING_H_
#define _OPLUS_SWITCHING_H_

#include <linux/i2c.h>
#include <linux/power_supply.h>
#include "../oplus_chg_track.h"

#define SWITCH_ERROR_OVER_VOLTAGE      1
#define SWITCH_ERROR_I2C_ERROR         1

#define REASON_I2C_ERROR	       1
#define REASON_MOS_OPEN_ERROR	       2
#define REASON_SUB_BATT_FULL	       4
#define REASON_VBAT_GAP_BIG	       8
#define REASON_SOC_NOT_FULL	       16
#define REASON_CURRENT_UNBALANCE       32
#define REASON_RECORD_SOC	       48
#define REASON_SOC_GAP_TOO_BIG	       64 /* low 4 bit use to MOS error */

#define BATT_OPEN_RETRY_COUNT	       12
#define BATT_FGI2C_RETRY_COUNT	       2
#define TRACK_SOC_GAP_BIG	       15

enum {
	MAIN_BATT_OVER_CURR = 1,
	SUB_BATT_OVER_CURR,
};

struct batt_spec {
	int volt;
	int main_curr;
	int sub_curr;
};

struct parallel_bat_table {
	struct batt_spec * batt_table;
	int length;
};

struct oplus_switch_chip {
	struct i2c_client *client;
	struct device *dev;
	struct oplus_switch_operations *switch_ops;
	int error_status;
	int ctrl_type;
	bool normal_chg_check;

	int parallel_vbat_gap_abnormal;
	int parallel_vbat_gap_full;
	int parallel_vbat_gap_recov;
	int parallel_mos_abnormal_litter_curr;
	int parallel_mos_abnormal_gap_curr;

	int main_hot_bat_decidegc;
	int main_warm_bat_decidegc;
	int main_normal_bat_decidegc;
	int main_little_cool_bat_decidegc;
	int main_cool_bat_decidegc;
	int main_little_cold_bat_decidegc;
	int main_cold_bat_decidegc;
	int main_removed_bat_decidegc;
	int sub_hot_bat_decidegc;
	int sub_warm_bat_decidegc;
	int sub_normal_bat_decidegc;
	int sub_little_cool_bat_decidegc;
	int sub_cool_bat_decidegc;
	int sub_little_cold_bat_decidegc;
	int sub_cold_bat_decidegc;
	int sub_removed_bat_decidegc;
	int main_tbatt_status;
	int main_pre_shake;
	int sub_tbatt_status;
	int sub_pre_shake;

	struct parallel_bat_table *parallel_bat_data;
	int pre_spec_index;
	int pre_sub_spec_index;
	bool mos_err_uploading;
	int track_unbalance_high;
	int track_unbalance_low;
	bool parallel_mos_debug;
	struct mutex track_mos_err_lock;
	oplus_chg_track_trigger *mos_err_load_trigger;
	struct delayed_work parallel_mos_trigger_work;
	int debug_force_mos_err;
};

struct oplus_switch_operations {
	int (*switching_hw_enable)(int en);
	int (*switching_set_fastcharge_current)(int curr_ma);
	int (*switching_set_discharge_current)(int curr_ma);
	int (*switching_enable_charge)(int en);
	int (*switching_get_fastcharge_current)(void);
	int (*switching_get_hw_enable)(void);
	int (*switching_get_discharge_current)(void);
	int (*switching_get_charge_enable)(void);
};

enum {
	PARALLEL_NOT_NEED_BALANCE_BAT__START_CHARGE,
	PARALLEL_NEED_BALANCE_BAT_STATUS1__STOP_CHARGE,           /*VBAT0 - VBAT1 >= 800mv*/
	PARALLEL_NEED_BALANCE_BAT_STATUS2__STOP_CHARGE,           /*800mv > VBAT0 - VBAT1 >= 600mv*/
	PARALLEL_NEED_BALANCE_BAT_STATUS3__STOP_CHARGE,           /*600mv > VBAT0 - VBAT1 >= 50mv*/
	PARALLEL_NEED_BALANCE_BAT_STATUS4__STOP_CHARGE,           /*400mv > VBAT1 - VBAT0 > 50mv*/
	PARALLEL_NEED_BALANCE_BAT_STATUS5__STOP_CHARGE,           /*VBAT0 > 3400mv && VBAT1 < 3400 && VBAT0 - VBAT1 <= 1000mv*/
	PARALLEL_BAT_BALANCE_ERROR_STATUS6__STOP_CHARGE,          /*VBAT0 > 3400mv && VBAT1 < 3100 && VBAT0 - VBAT1 > 1000mv*/
	PARALLEL_NEED_BALANCE_BAT_STATUS7__START_CHARGE,          /*VBAT1 - VBAT0 > 400mv */
	PARALLEL_BAT_BALANCE_ERROR_STATUS8,                       /*bat error*/
	PARALLEL_BAT_BALANCE_ERROR_STATUS9,			  /* bat error but need to check recovery */
};

enum {
	NO_PARALLEL_TYPE = 0,
	PARALLEL_SWITCH_IC,
	PARALLEL_MOS_CTRL,
	INVALID_PARALLEL,
};

void oplus_switching_init(struct oplus_switch_chip *chip, int type);
int oplus_switching_hw_enable(int en);
int oplus_switching_set_fastcharge_current(int curr_ma);
int oplus_switching_get_if_need_balance_bat(int vbat0_mv, int vbat1_mv);
int oplus_switching_set_balance_bat_status(int status);
int oplus_switching_enable_charge(int en);
int oplus_switching_support_parallel_chg(void);
bool oplus_support_normal_batt_spec_check(void);
int oplus_switching_set_current(int current_ma);
void oplus_init_parallel_temp_threshold(void);
int oplus_chg_is_parellel_ibat_over_spec(int main_temp, int sub_temp,
					 int *target_curr);
void oplus_chg_parellel_variables_reset(void);
#endif /* _OPLUS_GAUGE_H */