// SPDX-License-Identifier: GPL-2.0
/*
 *  mt6877-afe-gpio.c  --  Mediatek 6833 afe gpio ctrl
 *
 *  Copyright (c) 2020 MediaTek Inc.
 *  Author: Eason Yen <eason.yen@mediatek.com>
 */

#include <linux/gpio.h>
#include <linux/pinctrl/consumer.h>

#include "mt6877-afe-common.h"
#include "mt6877-afe-gpio.h"

struct pinctrl *aud_pinctrl;

struct audio_gpio_attr {
	const char *name;
	bool gpio_prepare;
	struct pinctrl_state *gpioctrl;
};

static struct audio_gpio_attr aud_gpios[MT6877_AFE_GPIO_GPIO_NUM] = {
	[MT6877_AFE_GPIO_DAT_MISO0_OFF] = {"aud_dat_miso0_off", false, NULL},
	[MT6877_AFE_GPIO_DAT_MISO0_ON] = {"aud_dat_miso0_on", false, NULL},
	[MT6877_AFE_GPIO_DAT_MISO1_OFF] = {"aud_dat_miso1_off", false, NULL},
	[MT6877_AFE_GPIO_DAT_MISO1_ON] = {"aud_dat_miso1_on", false, NULL},
	[MT6877_AFE_GPIO_DAT_MISO2_OFF] = {"aud_dat_miso2_off", false, NULL},
	[MT6877_AFE_GPIO_DAT_MISO2_ON] = {"aud_dat_miso2_on", false, NULL},
	[MT6877_AFE_GPIO_DAT_MOSI_OFF] = {"aud_dat_mosi_off", false, NULL},
	[MT6877_AFE_GPIO_DAT_MOSI_ON] = {"aud_dat_mosi_on", false, NULL},
	[MT6877_AFE_GPIO_DAT_MOSI_CH34_OFF] = {"aud_dat_mosi_ch34_off", false, NULL},
	[MT6877_AFE_GPIO_DAT_MOSI_CH34_ON] = {"aud_dat_mosi_ch34_on", false, NULL},
	[MT6877_AFE_GPIO_I2S0_OFF] = {"aud_gpio_i2s0_off", false, NULL},
	[MT6877_AFE_GPIO_I2S0_ON] = {"aud_gpio_i2s0_on", false, NULL},
	[MT6877_AFE_GPIO_I2S1_OFF] = {"aud_gpio_i2s1_off", false, NULL},
	[MT6877_AFE_GPIO_I2S1_ON] = {"aud_gpio_i2s1_on", false, NULL},
	[MT6877_AFE_GPIO_I2S2_OFF] = {"aud_gpio_i2s2_off", false, NULL},
	[MT6877_AFE_GPIO_I2S2_ON] = {"aud_gpio_i2s2_on", false, NULL},
	[MT6877_AFE_GPIO_I2S3_OFF] = {"aud_gpio_i2s3_off", false, NULL},
	[MT6877_AFE_GPIO_I2S3_ON] = {"aud_gpio_i2s3_on", false, NULL},
	[MT6877_AFE_GPIO_I2S5_OFF] = {"aud_gpio_i2s5_off", false, NULL},
	[MT6877_AFE_GPIO_I2S5_ON] = {"aud_gpio_i2s5_on", false, NULL},
	[MT6877_AFE_GPIO_I2S6_OFF] = {"aud_gpio_i2s6_off", false, NULL},
	[MT6877_AFE_GPIO_I2S6_ON] = {"aud_gpio_i2s6_on", false, NULL},
	[MT6877_AFE_GPIO_I2S7_OFF] = {"aud_gpio_i2s7_off", false, NULL},
	[MT6877_AFE_GPIO_I2S7_ON] = {"aud_gpio_i2s7_on", false, NULL},
	[MT6877_AFE_GPIO_I2S8_OFF] = {"aud_gpio_i2s8_off", false, NULL},
	[MT6877_AFE_GPIO_I2S8_ON] = {"aud_gpio_i2s8_on", false, NULL},
	[MT6877_AFE_GPIO_I2S9_OFF] = {"aud_gpio_i2s9_off", false, NULL},
	[MT6877_AFE_GPIO_I2S9_ON] = {"aud_gpio_i2s9_on", false, NULL},
	[MT6877_AFE_GPIO_VOW_OFF] = {"vow_gpio_off", false, NULL},
	[MT6877_AFE_GPIO_VOW_ON] = {"vow_gpio_on", false, NULL},
};

static DEFINE_MUTEX(gpio_request_mutex);

int mt6877_afe_gpio_init(struct mtk_base_afe *afe)
{
	int ret;
	int i = 0;

	aud_pinctrl = devm_pinctrl_get(afe->dev);
	if (IS_ERR(aud_pinctrl)) {
		ret = PTR_ERR(aud_pinctrl);
		dev_err(afe->dev, "%s(), ret %d, cannot get aud_pinctrl!\n",
			__func__, ret);
		return -ENODEV;
	}

	for (i = 0; i < ARRAY_SIZE(aud_gpios); i++) {
		aud_gpios[i].gpioctrl = pinctrl_lookup_state(aud_pinctrl,
							     aud_gpios[i].name);
		if (IS_ERR(aud_gpios[i].gpioctrl)) {
			ret = PTR_ERR(aud_gpios[i].gpioctrl);
			dev_err(afe->dev, "%s(), pinctrl_lookup_state %s fail, ret %d\n",
				__func__, aud_gpios[i].name, ret);
		} else {
			aud_gpios[i].gpio_prepare = true;
		}
	}

	/* gpio status init */
	mt6877_afe_gpio_request(afe, false, MT6877_DAI_ADDA, 0);
	mt6877_afe_gpio_request(afe, false, MT6877_DAI_ADDA, 1);
	mt6877_afe_gpio_request(afe, false, MT6877_DAI_I2S_0, 0);
	mt6877_afe_gpio_request(afe, false, MT6877_DAI_I2S_3, 0);

	return 0;
}

static int mt6877_afe_gpio_select(struct mtk_base_afe *afe,
				  enum mt6877_afe_gpio type)
{
	int ret = 0;

	if (type < 0 || type >= MT6877_AFE_GPIO_GPIO_NUM) {
		dev_err(afe->dev, "%s(), error, invaild gpio type %d\n",
			__func__, type);
		return -EINVAL;
	}

	if (!aud_gpios[type].gpio_prepare)
		return -EIO;

	ret = pinctrl_select_state(aud_pinctrl,
				   aud_gpios[type].gpioctrl);
	if (ret) {
		dev_err(afe->dev, "%s(), error, can not set gpio type %d\n",
			__func__, type);
		AUDIO_AEE("can not set gpio type");
	}
	return ret;
}

bool mt6877_afe_gpio_is_prepared(enum mt6877_afe_gpio type)
{
	return aud_gpios[type].gpio_prepare;
}

int mt6877_afe_gpio_request(struct mtk_base_afe *afe, bool enable,
			    int dai, int uplink)
{
	mutex_lock(&gpio_request_mutex);
	switch (dai) {
	case MT6877_DAI_ADDA:
		if (uplink) {
			mt6877_afe_gpio_select(afe, enable ?
			       MT6877_AFE_GPIO_DAT_MISO0_ON :
			       MT6877_AFE_GPIO_DAT_MISO0_OFF);
			mt6877_afe_gpio_select(afe, enable ?
			       MT6877_AFE_GPIO_DAT_MISO1_ON :
			       MT6877_AFE_GPIO_DAT_MISO1_OFF);
		} else
			mt6877_afe_gpio_select(afe, enable ?
			       MT6877_AFE_GPIO_DAT_MOSI_ON :
			       MT6877_AFE_GPIO_DAT_MOSI_OFF);
		break;
	case MT6877_DAI_ADDA_CH34:
		if (uplink)
			mt6877_afe_gpio_select(afe, enable ?
			       MT6877_AFE_GPIO_DAT_MISO2_ON :
			       MT6877_AFE_GPIO_DAT_MISO2_OFF);
		else
			mt6877_afe_gpio_select(afe, enable ?
			       MT6877_AFE_GPIO_DAT_MOSI_CH34_ON :
			       MT6877_AFE_GPIO_DAT_MOSI_CH34_OFF);
		break;
	case MT6877_DAI_I2S_0:
		mt6877_afe_gpio_select(afe, enable ?
			       MT6877_AFE_GPIO_I2S0_ON :
			       MT6877_AFE_GPIO_I2S0_OFF);
		break;
	case MT6877_DAI_I2S_1:
		mt6877_afe_gpio_select(afe, enable ?
			       MT6877_AFE_GPIO_I2S1_ON :
			       MT6877_AFE_GPIO_I2S1_OFF);
		break;
	case MT6877_DAI_I2S_2:
		mt6877_afe_gpio_select(afe, enable ?
			       MT6877_AFE_GPIO_I2S2_ON :
			       MT6877_AFE_GPIO_I2S2_OFF);
		break;
	case MT6877_DAI_I2S_3:
		mt6877_afe_gpio_select(afe, enable ?
			       MT6877_AFE_GPIO_I2S3_ON :
			       MT6877_AFE_GPIO_I2S3_OFF);
		break;
	case MT6877_DAI_I2S_5:
		mt6877_afe_gpio_select(afe, enable ?
			       MT6877_AFE_GPIO_I2S5_ON :
			       MT6877_AFE_GPIO_I2S5_OFF);
		break;
	case MT6877_DAI_I2S_6:
		mt6877_afe_gpio_select(afe, enable ?
			       MT6877_AFE_GPIO_I2S6_ON :
			       MT6877_AFE_GPIO_I2S6_OFF);
		break;
	case MT6877_DAI_I2S_7:
		mt6877_afe_gpio_select(afe, enable ?
			       MT6877_AFE_GPIO_I2S7_ON :
			       MT6877_AFE_GPIO_I2S7_OFF);
		break;
	case MT6877_DAI_I2S_8:
		mt6877_afe_gpio_select(afe, enable ?
			       MT6877_AFE_GPIO_I2S8_ON :
			       MT6877_AFE_GPIO_I2S8_OFF);
		break;
	case MT6877_DAI_I2S_9:
		mt6877_afe_gpio_select(afe, enable ?
			       MT6877_AFE_GPIO_I2S9_ON :
			       MT6877_AFE_GPIO_I2S9_OFF);
		break;
	case MT6877_DAI_VOW:
		mt6877_afe_gpio_select(afe, enable ?
			       MT6877_AFE_GPIO_VOW_ON :
			       MT6877_AFE_GPIO_VOW_OFF);
		break;
	default:
		mutex_unlock(&gpio_request_mutex);
		dev_warn(afe->dev, "%s(), invalid dai %d\n", __func__, dai);
		AUDIO_AEE("invalid dai");
		return -EINVAL;
	}
	mutex_unlock(&gpio_request_mutex);
	return 0;
}