/* SPDX-License-Identifier: GPL-2.0-only */
/*
 * Copyright (c) 2024, Qualcomm Innovation Center, Inc. All rights reserved.
 */

#ifndef __FG_GEN4_IIO_H
#define __FG_GEN4_IIO_H

#include <linux/iio/iio.h>
#include <dt-bindings/iio/qti_power_supply_iio.h>

struct fg_gen4_iio_prop_channels {
	const char *datasheet_name;
	int channel_num;
	enum iio_chan_type type;
	long info_mask;
};

#define FG_GEN4_IIO_CHAN(_dname, _chan, _type, _mask)		\
	{								\
		.datasheet_name = _dname,				\
		.channel_num = _chan,				\
		.type = _type,						\
		.info_mask = _mask,					\
	},								\

#define FG_GEN4_CHAN_VOLT(_name, _num)			\
	FG_GEN4_IIO_CHAN(_name, _num, IIO_VOLTAGE,		\
		BIT(IIO_CHAN_INFO_PROCESSED))

#define FG_GEN4_CHAN_CUR(_name, _num)			\
	FG_GEN4_IIO_CHAN(_name, _num, IIO_CURRENT,		\
		BIT(IIO_CHAN_INFO_PROCESSED))

#define FG_GEN4_CHAN_RES(_name, _num)			\
	FG_GEN4_IIO_CHAN(_name, _num, IIO_RESISTANCE,	\
		BIT(IIO_CHAN_INFO_PROCESSED))

#define FG_GEN4_CHAN_TEMP(_name, _num)			\
	FG_GEN4_IIO_CHAN(_name, _num, IIO_TEMP,		\
		BIT(IIO_CHAN_INFO_PROCESSED))

#define FG_GEN4_CHAN_POW(_name, _num)			\
	FG_GEN4_IIO_CHAN(_name, _num, IIO_POWER,		\
		BIT(IIO_CHAN_INFO_PROCESSED))

#define FG_GEN4_CHAN_ENERGY(_name, _num)			\
	FG_GEN4_IIO_CHAN(_name, _num, IIO_ENERGY,		\
		BIT(IIO_CHAN_INFO_PROCESSED))

#define FG_GEN4_CHAN_INDEX(_name, _num)			\
	FG_GEN4_IIO_CHAN(_name, _num, IIO_INDEX,		\
		BIT(IIO_CHAN_INFO_PROCESSED))

#define FG_GEN4_CHAN_ACT(_name, _num)			\
	FG_GEN4_IIO_CHAN(_name, _num, IIO_ACTIVITY,		\
		BIT(IIO_CHAN_INFO_PROCESSED))

#define FG_GEN4_CHAN_TSTAMP(_name, _num)			\
	FG_GEN4_IIO_CHAN(_name, _num, IIO_TIMESTAMP,		\
		BIT(IIO_CHAN_INFO_PROCESSED))

#define FG_GEN4_CHAN_COUNT(_name, _num)			\
	FG_GEN4_IIO_CHAN(_name, _num, IIO_COUNT,		\
		BIT(IIO_CHAN_INFO_PROCESSED))

static const struct fg_gen4_iio_prop_channels fg_gen4_iio_psy_channels[] = {
	FG_GEN4_CHAN_ENERGY("real_capacity", PSY_IIO_REAL_CAPACITY)
	FG_GEN4_CHAN_ENERGY("capacity_raw", PSY_IIO_CAPACITY_RAW)
	FG_GEN4_CHAN_ENERGY("cc_soc", PSY_IIO_CC_SOC)
	FG_GEN4_CHAN_RES("resistance_id", PSY_IIO_RESISTANCE_ID)
	FG_GEN4_CHAN_RES("resistance", PSY_IIO_RESISTANCE)
	FG_GEN4_CHAN_RES("esr_actual", PSY_IIO_ESR_ACTUAL)
	FG_GEN4_CHAN_RES("esr_nominal", PSY_IIO_ESR_NOMINAL)
	FG_GEN4_CHAN_ENERGY("charge_now_raw", PSY_IIO_CHARGE_NOW_RAW)
	FG_GEN4_CHAN_ENERGY("charge_counter_shadow", PSY_IIO_CHARGE_COUNTER_SHADOW)
	FG_GEN4_CHAN_ENERGY("soc_reporting_ready", PSY_IIO_SOC_REPORTING_READY)
	FG_GEN4_CHAN_INDEX("debug_battery", PSY_IIO_DEBUG_BATTERY)
	FG_GEN4_CHAN_INDEX("clear_soh", PSY_IIO_CLEAR_SOH)
	FG_GEN4_CHAN_INDEX("soh", PSY_IIO_SOH)
	FG_GEN4_CHAN_ENERGY("cc_step", PSY_IIO_CC_STEP)
	FG_GEN4_CHAN_ENERGY("cc_step_sel", PSY_IIO_CC_STEP_SEL)
	FG_GEN4_CHAN_INDEX("batt_age_level", PSY_IIO_BATT_AGE_LEVEL)
	FG_GEN4_CHAN_ACT("scale_mode_en", PSY_IIO_SCALE_MODE_EN)
	FG_GEN4_CHAN_ENERGY("capacity", PSY_IIO_CAPACITY)
	FG_GEN4_CHAN_CUR("current_now", PSY_IIO_CURRENT_NOW)
	FG_GEN4_CHAN_VOLT("voltage_now", PSY_IIO_VOLTAGE_NOW)
	FG_GEN4_CHAN_VOLT("voltage_max", PSY_IIO_VOLTAGE_MAX)
	FG_GEN4_CHAN_ENERGY("charge_full", PSY_IIO_CHARGE_FULL)
	FG_GEN4_CHAN_TEMP("temp", PSY_IIO_TEMP)
	FG_GEN4_CHAN_ENERGY("charge_counter", PSY_IIO_CHARGE_COUNTER)
	FG_GEN4_CHAN_COUNT("cycle_count", PSY_IIO_CYCLE_COUNT)
	FG_GEN4_CHAN_ENERGY("charge_full_design", PSY_IIO_CHARGE_FULL_DESIGN)
	FG_GEN4_CHAN_TSTAMP("time_to_full_now", PSY_IIO_TIME_TO_FULL_NOW)
	FG_GEN4_CHAN_VOLT("voltage_ocv", PSY_IIO_VOLTAGE_OCV)
	FG_GEN4_CHAN_VOLT("voltage_avg", PSY_IIO_VOLTAGE_AVG)
	FG_GEN4_CHAN_CUR("current_avg", PSY_IIO_CURRENT_AVG)
	FG_GEN4_CHAN_VOLT("voltage_max_design", PSY_IIO_VOLTAGE_MAX_DESIGN)
	FG_GEN4_CHAN_ENERGY("charge_now", PSY_IIO_CHARGE_NOW)
	FG_GEN4_CHAN_VOLT("constant_charge_voltage", PSY_IIO_CONSTANT_CHARGE_VOLTAGE)
	FG_GEN4_CHAN_TSTAMP("time_to_full_avg", PSY_IIO_TIME_TO_FULL_AVG)
	FG_GEN4_CHAN_TSTAMP("time_to_empty_avg", PSY_IIO_TIME_TO_EMPTY_AVG)
	FG_GEN4_CHAN_POW("power_avg", PSY_IIO_POWER_AVG)
	FG_GEN4_CHAN_POW("power_now", PSY_IIO_POWER_NOW)
	FG_GEN4_CHAN_CUR("calibrate", PSY_IIO_CALIBRATE)
};

enum fg_gen4_ext_iio_channels {
	FG_GEN4_QNOVO_ENABLE = 0,
	FG_GEN4_CHARGE_DONE,
	FG_GEN4_PARALLEL_CHARGING_ENABLED,
	FG_GEN4_RECHARGE_SOC,
};

static const char * const fg_gen4_ext_iio_chan_name[] = {
	[FG_GEN4_QNOVO_ENABLE]			= "qnovo_enable",
	[FG_GEN4_CHARGE_DONE]			= "charge_done",
	[FG_GEN4_PARALLEL_CHARGING_ENABLED]		= "cp_charging_enabled",
	[FG_GEN4_RECHARGE_SOC]			= "recharge_soc",
};

struct iio_channel **get_ext_channels(struct device *dev,
	const char *const *channel_map, int size);
#endif