From 2093641e5d7b5fa6a5ac83d57a937d686375b09e Mon Sep 17 00:00:00 2001
From: yanyh2 <yanyh2@motorola.com>
Date: Fri, 11 Feb 2022 20:02:29 +0800
Subject: [PATCH] Support a main battery power supply

Add main battery power supply to support main battery
property read in multiple batteries products.

Change-Id: I73c89fc4795ad0ae299460ed616b7d07d5a66ca1
Signed-off-by: yanyh2 <yanyh2@motorola.com>
Reviewed-on: https://gerrit.mot.com/2189094
SME-Granted: SME Approvals Granted
SLTApproved: Slta Waiver
Tested-by: Jira Key
Reviewed-by: Jianqi Yang <yangj@motorola.com>
Submit-Approved: Jira Key
---
 .../qti_glink_charger/qti_glink_charger.c     | 101 ++++++++++++++++++
 1 file changed, 101 insertions(+)

diff --git a/drivers/power/qti_glink_charger/qti_glink_charger.c b/drivers/power/qti_glink_charger/qti_glink_charger.c
index 0dd8361d4626..d79fa5ee823d 100644
--- a/drivers/power/qti_glink_charger/qti_glink_charger.c
+++ b/drivers/power/qti_glink_charger/qti_glink_charger.c
@@ -199,6 +199,7 @@ struct qti_charger {
 	u32				rx_connected;
 	struct notifier_block		wls_nb;
 	struct dentry		*debug_root;
+	struct power_supply		*batt_psy;
 };
 
 static struct qti_charger *this_chip = NULL;
@@ -1744,6 +1745,90 @@ static void wireless_psy_deinit(struct qti_charger *chg)
 	chg->wls_psy = NULL;
 }
 
+/* Battery presence detection threshold on battery temperature */
+#define BPD_TEMP_THRE -3000
+static int battery_psy_get_prop(struct power_supply *psy,
+		enum power_supply_property prop,
+		union power_supply_propval *pval)
+{
+	int rc;
+	struct battery_info info = {0};
+	struct qti_charger *chg = power_supply_get_drvdata(psy);
+
+	pval->intval = -ENODATA;
+
+	rc = qti_charger_read(chg, OEM_PROP_BATT_INFO, &info,
+				sizeof(struct battery_info));
+	if (rc)
+		return rc;
+
+	switch (prop) {
+	case POWER_SUPPLY_PROP_STATUS:
+		pval->intval = info.batt_status;
+		break;
+	case POWER_SUPPLY_PROP_PRESENT:
+		pval->intval = info.batt_temp > BPD_TEMP_THRE? 1 : 0;
+		break;
+	case POWER_SUPPLY_PROP_VOLTAGE_NOW:
+		pval->intval = info.batt_uv;
+		break;
+	case POWER_SUPPLY_PROP_CURRENT_NOW:
+		pval->intval = info.batt_ua;
+		break;
+	case POWER_SUPPLY_PROP_CAPACITY:
+		pval->intval = info.batt_soc / 100;
+		break;
+	case POWER_SUPPLY_PROP_TEMP:
+		pval->intval = info.batt_temp / 10;
+		break;
+	case POWER_SUPPLY_PROP_CHARGE_FULL:
+		pval->intval = info.batt_full_uah;
+		break;
+	case POWER_SUPPLY_PROP_CHARGE_FULL_DESIGN:
+		pval->intval = info.batt_design_uah;
+		break;
+	default:
+		break;
+	}
+
+	return rc;
+}
+
+static int battery_psy_set_prop(struct power_supply *psy,
+		enum power_supply_property prop,
+		const union power_supply_propval *pval)
+{
+	struct qti_charger *chg = power_supply_get_drvdata(psy);
+
+	switch (prop) {
+	default:
+		mmi_err(chg, "Not supported property: %d\n", prop);
+		return -EINVAL;
+	}
+
+	return 0;
+}
+
+static enum power_supply_property battery_props[] = {
+	POWER_SUPPLY_PROP_STATUS,
+	POWER_SUPPLY_PROP_PRESENT,
+	POWER_SUPPLY_PROP_CAPACITY,
+	POWER_SUPPLY_PROP_VOLTAGE_NOW,
+	POWER_SUPPLY_PROP_CURRENT_NOW,
+	POWER_SUPPLY_PROP_TEMP,
+	POWER_SUPPLY_PROP_CHARGE_FULL_DESIGN,
+	POWER_SUPPLY_PROP_CHARGE_FULL,
+};
+
+static const struct power_supply_desc batt_psy_desc = {
+	.name			= "main_battery",
+	.type			= POWER_SUPPLY_TYPE_MAINS,
+	.properties		= battery_props,
+	.num_properties		= ARRAY_SIZE(battery_props),
+	.get_property		= battery_psy_get_prop,
+	.set_property		= battery_psy_set_prop,
+};
+
 static int qti_charger_init(struct qti_charger *chg)
 {
 	int rc;
@@ -1781,6 +1866,22 @@ static int qti_charger_init(struct qti_charger *chg)
 		return rc;
 	}
 
+	if (of_property_read_bool(chg->dev->of_node,
+				"mmi,main-battery-enabled")) {
+		struct power_supply_config psy_cfg = {};
+		psy_cfg.drv_data = chg;
+		psy_cfg.of_node = chg->dev->of_node;
+		chg->batt_psy = devm_power_supply_register(chg->dev,
+						&batt_psy_desc,
+						&psy_cfg);
+		if (IS_ERR(chg->batt_psy)) {
+			rc = PTR_ERR(chg->batt_psy);
+			chg->batt_psy = NULL;
+			mmi_err(chg, "Failed to register main psy, rc=%d\n", rc);
+			return rc;
+		}
+	}
+
 	driver = devm_kzalloc(chg->dev,
 				sizeof(struct mmi_charger_driver),
 				GFP_KERNEL);