// SPDX-License-Identifier: GPL-2.0
//
// Copyright (C) 2018 MediaTek Inc.

#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/err.h>
#include <linux/i2c.h>
/* alsa sound header */
#include <sound/soc.h>
#include <sound/pcm_params.h>

#include "mtk-sp-spk-amp.h"
#if IS_ENABLED(CONFIG_SND_SOC_RT5509)
#include "../../codecs/rt5509.h"
#endif

#if IS_ENABLED(CONFIG_SND_SOC_RT5512)
#include "../../codecs/richtek/rt5512.h"
#endif

#if IS_ENABLED(CONFIG_SND_SOC_TFA9874)
#include "../../codecs/tfa98xx/inc/tfa98xx_ext.h"
#endif

#if IS_ENABLED(CONFIG_SND_SOC_AW87339)
#include "aw87339.h"
#endif

/* adsp relate */
#if IS_ENABLED(CONFIG_SND_SOC_MTK_AUDIO_DSP)
#include "../audio_dsp/mtk-dsp-common.h"
#include "audio_messenger_ipi.h"
#endif

/* rv relate */
#if IS_ENABLED(CONFIG_MTK_SCP_AUDIO)
#include "../audio_scp/mtk-scp-audio-pcm.h"
#endif

//drv add by pengzhipeng 20230105 start
#if IS_ENABLED(CONFIG_SND_SOC_FS1599)
#include "../../codecs/fs1599/fsm_public.h"
#endif
#if IS_ENABLED(CONFIG_SND_SOC_AW8839X)
#include "../../codecs/aw883xx/aw883xx.h"
#endif
//drv add by pengzhipeng 20230105 start

#define MTK_SPK_NAME "Speaker Codec"
#define MTK_SPK_REF_NAME "Speaker Codec Ref"

static unsigned int mtk_spk_type;
static int mtk_spk_i2s_out = MTK_SPK_I2S_3, mtk_spk_i2s_in = MTK_SPK_I2S_0;
static struct mtk_spk_i2c_ctrl mtk_spk_list[MTK_SPK_TYPE_NUM] = {
	[MTK_SPK_NOT_SMARTPA] = {
//drv add by pengzhipeng 20230105 start
#if IS_ENABLED(CONFIG_SND_SOC_FS1599)
		.i2c_probe = exfsm_i2c_probe,
		.i2c_remove = exfsm_i2c_remove,
#endif
//drv add by pengzhipeng 20230105 end
		.codec_dai_name = "snd-soc-dummy-dai",
		.codec_name = "snd-soc-dummy",
	},
#if IS_ENABLED(CONFIG_SND_SOC_RT5509)
	[MTK_SPK_RICHTEK_RT5509] = {
		.i2c_probe = rt5509_i2c_probe,
		.i2c_remove = rt5509_i2c_remove,
		.i2c_shutdown = rt5509_i2c_shutdown,
		.codec_dai_name = "rt5509-aif1",
		.codec_name = "RT5509_MT_0",
	},
#endif

#if IS_ENABLED(CONFIG_SND_SOC_RT5512)
	[MTK_SPK_MEDIATEK_RT5512] = {
		.codec_dai_name = "rt5512-aif",
		.codec_name = "RT5512_MT_0",
	},
#endif /* CONFIG_SND_SOC_RT5512 */

#if IS_ENABLED(CONFIG_SND_SOC_TFA9874)
	[MTK_SPK_GOODIX_TFA98XX] = {
		.codec_dai_name = "tfa98xx-aif",
		.codec_name = "tfa98xx",
	},
#endif /* CONFIG_SND_SOC_TFA9874 */
//prize add by lipengpeng 20220607 start
#if IS_ENABLED(CONFIG_SND_SOC_AW8839X)
        [MTK_SPK_AWINIC_AW883XX] = {
                //.i2c_probe = aw883xx_i2c_probe,
                //.i2c_remove = aw883xx_i2c_remove,
                .codec_dai_name = "aw883xx-aif-6-34",
                .codec_name = "aw883xx_smartpa.6-0034",
        },
#endif 
//prize add by lipengpeng 20220607 end
};

static int mtk_spk_i2c_probe(struct i2c_client *client,
			     const struct i2c_device_id *id)
{
	int i, ret = 0;

	dev_info(&client->dev, "%s()\n", __func__);

	mtk_spk_type = MTK_SPK_NOT_SMARTPA;
	for (i = 0; i < MTK_SPK_TYPE_NUM; i++) {
		if (!mtk_spk_list[i].i2c_probe)
			continue;

		ret = mtk_spk_list[i].i2c_probe(client, id);
		if (ret)
			continue;

		mtk_spk_type = i;
		break;
	}

	return ret;
}

static int mtk_spk_i2c_remove(struct i2c_client *client)
{
	dev_info(&client->dev, "%s()\n", __func__);

	if (mtk_spk_list[mtk_spk_type].i2c_remove)
		mtk_spk_list[mtk_spk_type].i2c_remove(client);

	return 0;
}

static void mtk_spk_i2c_shutdown(struct i2c_client *client)
{
	dev_info(&client->dev, "%s()\n", __func__);

	if (mtk_spk_list[mtk_spk_type].i2c_shutdown)
		mtk_spk_list[mtk_spk_type].i2c_shutdown(client);
}

int mtk_spk_get_type(void)
{
	return mtk_spk_type;
}
EXPORT_SYMBOL(mtk_spk_get_type);

void mtk_spk_set_type(int spk_type)
{
//prize add by pengzhipeng 20230105 start
#if IS_ENABLED(CONFIG_SND_SOC_FS1599)
	mtk_spk_type = spk_type;//prize add by pengzhipeng 20230105
#else
	mtk_spk_type = 5;//prize add by lipengpeng 20220607
#endif
//prize add by pengzhipeng 20230105 end
}
EXPORT_SYMBOL(mtk_spk_set_type);

int mtk_spk_get_i2s_out_type(void)
{
	return mtk_spk_i2s_out;
}
EXPORT_SYMBOL(mtk_spk_get_i2s_out_type);

int mtk_spk_get_i2s_in_type(void)
{
	return mtk_spk_i2s_in;
}
EXPORT_SYMBOL(mtk_spk_get_i2s_in_type);

int mtk_ext_spk_get_status(void)
{
#ifdef CONFIG_SND_SOC_AW87339
	return aw87339_spk_status_get();
#else
	return 0;
#endif
}
EXPORT_SYMBOL(mtk_ext_spk_get_status);

void mtk_ext_spk_enable(int enable)
{
#ifdef CONFIG_SND_SOC_AW87339
	aw87339_spk_enable_set(enable);
#endif
}
EXPORT_SYMBOL(mtk_ext_spk_enable);

int mtk_spk_update_info(struct snd_soc_card *card,
			struct platform_device *pdev)
{
	struct snd_soc_dai_link *dai_link;
	int ret, i;
	int i2s_out_dai_link_idx = -1;
	int i2s_in_dai_link_idx = -1;
	const int i2s_num = 2;
	unsigned int i2s_set[2];
//prize add by pengzhipeng 20230105 start
#if IS_ENABLED(CONFIG_SND_SOC_FS1599) || IS_ENABLED(CONFIG_SND_SOC_AW87XXX)
	mtk_spk_type = MTK_SPK_NOT_SMARTPA;
#else
    mtk_spk_type=5;//prize add by lipengpeng 20220607
#endif
//prize add by pengzhipeng 20230105 end

	if (mtk_spk_type == MTK_SPK_NOT_SMARTPA)
		goto BYPASS_UPDATE;

	/* get spk i2s set */
	ret = of_property_read_u32_array(pdev->dev.of_node, "mediatek,spk-i2s",
					 i2s_set, i2s_num);
	if (ret) {
		dev_info(&pdev->dev,
			 "%s(), fail read mediatek,spk-i2s, use defalut i2s3/i2s0\n",
			 __func__);
		mtk_spk_i2s_out = MTK_SPK_I2S_3;
		mtk_spk_i2s_in = MTK_SPK_I2S_0;
	} else {
		mtk_spk_i2s_out = i2s_set[0];
		mtk_spk_i2s_in = i2s_set[1];
	}

	if (mtk_spk_i2s_out > MTK_SPK_I2S_TYPE_NUM ||
	    mtk_spk_i2s_in > MTK_SPK_I2S_TYPE_NUM) {
		dev_err(&pdev->dev, "%s(), get mtk spk i2s fail\n",
			__func__);
		return -ENODEV;
	}

	/* find dai link of i2s in and i2s out */
	for_each_card_prelinks(card, i, dai_link) {
		if (i2s_out_dai_link_idx < 0 &&
		    strcmp(dai_link->cpus->dai_name, "I2S1") == 0 &&
		    mtk_spk_i2s_out == MTK_SPK_I2S_1) {
			i2s_out_dai_link_idx = i;
			dai_link->name = MTK_SPK_NAME;
			dai_link->codecs->name = NULL;
			dai_link->codecs->dai_name = NULL;
		} else if (i2s_out_dai_link_idx < 0 &&
			   strcmp(dai_link->cpus->dai_name, "I2S3") == 0 &&
			   mtk_spk_i2s_out == MTK_SPK_I2S_3) {
			i2s_out_dai_link_idx = i;
			dai_link->name = MTK_SPK_NAME;
		} else if (i2s_out_dai_link_idx < 0 &&
			   strcmp(dai_link->cpus->dai_name, "I2S5") == 0 &&
			   mtk_spk_i2s_out == MTK_SPK_I2S_5) {
			i2s_out_dai_link_idx = i;
			dai_link->name = MTK_SPK_NAME;
			dai_link->codecs->name = NULL;
			dai_link->codecs->dai_name = NULL;
		} else if (i2s_out_dai_link_idx < 0 &&
				strcmp(dai_link->cpus->dai_name, "ETDMOUT") == 0 &&
				mtk_spk_i2s_out == MTK_SPK_ETDM_OUT) {
			i2s_out_dai_link_idx = i;
			dai_link->name = MTK_SPK_NAME;
		}

		if (i2s_in_dai_link_idx < 0 &&
		    strcmp(dai_link->cpus->dai_name, "I2S0") == 0 &&
		    (mtk_spk_i2s_in == MTK_SPK_I2S_0 ||
		     mtk_spk_i2s_in == MTK_SPK_TINYCONN_I2S_0)) {
			i2s_in_dai_link_idx = i;
			dai_link->name = MTK_SPK_REF_NAME;
		} else if (i2s_in_dai_link_idx < 0 &&
			   strcmp(dai_link->cpus->dai_name, "I2S2") == 0 &&
			   (mtk_spk_i2s_in == MTK_SPK_I2S_2 ||
			    mtk_spk_i2s_in == MTK_SPK_TINYCONN_I2S_2)) {
			i2s_in_dai_link_idx = i;
			dai_link->name = MTK_SPK_REF_NAME;
			dai_link->codecs->name = NULL;
			dai_link->codecs->dai_name = NULL;
		} else if (i2s_in_dai_link_idx < 0 &&
				strcmp(dai_link->cpus->dai_name, "ETDMIN") == 0 &&
				mtk_spk_i2s_in == MTK_SPK_ETDM_IN) {
			i2s_in_dai_link_idx = i;
			dai_link->name = MTK_SPK_REF_NAME;
		}

		if (i2s_out_dai_link_idx >= 0 && i2s_in_dai_link_idx >= 0)
			break;
	}

	if (i2s_out_dai_link_idx < 0 || i2s_in_dai_link_idx < 0) {
		dev_err(&pdev->dev,
			"%s(), i2s cpu dai name error, i2s_out_dai_link_idx = %d, i2s_in_dai_link_idx = %d",
			__func__, i2s_out_dai_link_idx, i2s_in_dai_link_idx);
		return -ENODEV;
	}

BYPASS_UPDATE:
	dev_info(&pdev->dev,
		 "%s(), mtk_spk_type %d, spk_in_dai_link_idx %d, spk_out_dai_link_idx %d\n",
		 __func__,
		 mtk_spk_type, i2s_in_dai_link_idx,
		 i2s_out_dai_link_idx);

	return 0;
}
EXPORT_SYMBOL(mtk_spk_update_info);

int mtk_spk_send_ipi_buf_to_dsp(void *data_buffer, uint32_t data_size)
{
	int result = -1, task_scene = -1;
	struct ipi_msg_t ipi_msg;

#if IS_ENABLED(CONFIG_SND_SOC_MTK_AUDIO_DSP)
	task_scene = mtk_get_ipi_buf_scene_adsp();
#endif

#if IS_ENABLED(CONFIG_MTK_SCP_AUDIO)
	if (task_scene == -1)
		task_scene = mtk_get_ipi_buf_scene_rv();
#endif

	if (task_scene >= 0) {
		memset((void *)&ipi_msg, 0, sizeof(struct ipi_msg_t));

		result = audio_send_ipi_buf_to_dsp(&ipi_msg, task_scene,
						   AUDIO_DSP_TASK_AURISYS_SET_BUF,
						   mtk_spk_type,
						   data_buffer, data_size);
	}
	return result;
}
EXPORT_SYMBOL(mtk_spk_send_ipi_buf_to_dsp);

int mtk_spk_recv_ipi_buf_from_dsp(int8_t *buffer,
				  int16_t size,
				  uint32_t *buf_len)
{
	int result = -1, task_scene = -1;
	struct ipi_msg_t ipi_msg;


#if IS_ENABLED(CONFIG_SND_SOC_MTK_AUDIO_DSP)
	task_scene = mtk_get_ipi_buf_scene_adsp();
#endif
#if IS_ENABLED(CONFIG_MTK_SCP_AUDIO)
	if (task_scene == -1)
		task_scene = mtk_get_ipi_buf_scene_rv();
#endif

	if (task_scene >= 0) {
		memset((void *)&ipi_msg, 0, sizeof(struct ipi_msg_t));

		result = audio_recv_ipi_buf_from_dsp(&ipi_msg,
						     task_scene,
						     AUDIO_DSP_TASK_AURISYS_GET_BUF,
						     mtk_spk_type,
						     buffer, size, buf_len);
	}
	return result;
}
EXPORT_SYMBOL(mtk_spk_recv_ipi_buf_from_dsp);

static const struct i2c_device_id mtk_spk_i2c_id[] = {
	{ "tfa9874", 0},
	{ "speaker_amp", 0},
	{}
};
MODULE_DEVICE_TABLE(i2c, mtk_spk_i2c_id);

#ifdef CONFIG_OF
static const struct of_device_id mtk_spk_match_table[] = {
	{.compatible = "mediatek,speaker_amp",},
	{},
};
MODULE_DEVICE_TABLE(of, mtk_spk_match_table);
#endif /* #ifdef CONFIG_OF */

static struct i2c_driver mtk_spk_i2c_driver = {
	.driver = {
		.name = "speaker_amp",
		.owner = THIS_MODULE,
		.of_match_table = of_match_ptr(mtk_spk_match_table),
	},
	.probe = mtk_spk_i2c_probe,
	.remove = mtk_spk_i2c_remove,
	.shutdown = mtk_spk_i2c_shutdown,
	.id_table = mtk_spk_i2c_id,
};

module_i2c_driver(mtk_spk_i2c_driver);

MODULE_DESCRIPTION("Mediatek speaker amp register driver");
MODULE_AUTHOR("Shane Chien <shane.chien@mediatek.com>");
MODULE_LICENSE("GPL v2");