/*
 * sound\soc\sunxi\sun8iw19-snddaudio.c
 * (C) Copyright 2014-2019
 * Allwinner Technology Co., Ltd. <www.allwinnertech.com>
 * yumingfeng <yumingfeng@allwinnertech.com>
 *
 * some simple description for this code
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of
 * the License, or (at your option) any later version.
 *
 */
#include <linux/module.h>
#include <linux/clk.h>
#include <linux/mutex.h>
#include <sound/pcm.h>
#include <sound/soc.h>
#include <sound/pcm_params.h>
#include <sound/soc-dapm.h>
#include <linux/io.h>
#include <linux/of.h>
#include <linux/debugfs.h>
#include "sun8iw19-snddaudio.h"

static atomic_t daudio_count = ATOMIC_INIT(-1);

static int sunxi_snddaudio_hw_params(struct snd_pcm_substream *substream,
		struct snd_pcm_hw_params *params)
{
	struct snd_soc_pcm_runtime *rtd = substream->private_data;
	struct snd_soc_dai *codec_dai = rtd->codec_dai;
	struct snd_soc_dai *cpu_dai = rtd->cpu_dai;
	struct snd_soc_card *card = rtd->card;
	unsigned int freq, clk_div;
	int ret;
#ifdef USE_ASOC_FMT_SETTING
	struct sunxi_snddaudio_priv *snddaudio_priv =
				snd_soc_card_get_drvdata(card);
	struct platform_data *pdata = &(snddaudio_priv->pdata);
#endif

	switch (params_rate(params)) {
	case 8000:
	case 16000:
	case 32000:
	case 64000:
	case 12000:
	case 24000:
	case 48000:
	case 96000:
	case 192000:
	freq = 24576000;
		break;
	case 11025:
	case 22050:
	case 44100:
	case 88200:
	case 176400:
	freq = 22579200;
		break;
	default:
		dev_err(card->dev, "unsupport params rate:%d\n", freq);
		return -EINVAL;
	}

	/* set platform clk source freq and set the mode as daudio or pcm */
	ret = snd_soc_dai_set_sysclk(cpu_dai, 0, freq, 0);
	if (ret < 0) {
		dev_warn(card->dev, "cpu_dai set sysclk failed\n");
		return ret;
	}

	/* set codec clk source freq and set the mode as daudio or pcm */
	ret = snd_soc_dai_set_sysclk(codec_dai, 0, freq, 0);
	if (ret < 0)
		dev_warn(card->dev, "codec_dai set sysclk failed\n");

	ret = snd_soc_dai_set_pll(codec_dai, 0, 0, freq, freq);
	if (ret < 0)
		dev_warn(card->dev, "codec_dai set set_pll failed.\n");

	/*
	 * SND_SOC_DAIFMT_I2S
	 * SND_SOC_DAIFMT_NB_NF
	 * SND_SOC_DAIFMT_CBS_CFS
	 */
#ifdef USE_ASOC_FMT_SETTING
	/* set codec dai fmt */
	ret = snd_soc_dai_set_fmt(codec_dai, pdata->audio_format
			| (pdata->signal_inversion << SND_SOC_DAIFMT_SIG_SHIFT)
			| (pdata->daudio_master << SND_SOC_DAIFMT_MASTER_SHIFT));
	if (ret < 0)
		dev_warn(card->dev, "codec dai set fmt failed\n");

	/* set cpu dai fmt */
	ret = snd_soc_dai_set_fmt(cpu_dai, pdata->audio_format
			| (pdata->signal_inversion << SND_SOC_DAIFMT_SIG_SHIFT)
			| (pdata->daudio_master << SND_SOC_DAIFMT_MASTER_SHIFT));
	if (ret < 0)
		dev_warn(card->dev, "cpu dai set fmt failed\n");
#else
	/* set codec dai fmt */
	ret = snd_soc_dai_set_fmt(codec_dai, SND_SOC_DAIFMT_I2S |
			SND_SOC_DAIFMT_NB_NF | SND_SOC_DAIFMT_CBS_CFS);
	if (ret < 0)
		dev_warn(card->dev, "codec dai set fmt failed\n");
#endif

	/* set system clk div */
	clk_div = freq / params_rate(params);
	ret = snd_soc_dai_set_clkdiv(cpu_dai, 0, clk_div);
	if (ret < 0) {
		dev_warn(card->dev, "cpu_dai set clkdiv failed\n");
		return ret;
	}

	ret = snd_soc_dai_set_clkdiv(codec_dai, 0, clk_div);
	if (ret < 0)
		dev_warn(card->dev, "codec_dai set clkdiv failed\n");

	return 0;
}

static struct snd_soc_ops sunxi_snddaudio_ops = {
	.hw_params      = sunxi_snddaudio_hw_params,
};

static struct snd_soc_dai_link sunxi_snddaudio_dai_link = {
	.name           = "sysvoice",
	.stream_name    = "SUNXI-AUDIO",
	.cpu_dai_name   = "sunxi-daudio",
	.platform_name  = "sunxi-daudio",
	.codec_dai_name = "snd-soc-dummy-dai",
	.codec_name     = "snd-soc-dummy",
	.ops            = &sunxi_snddaudio_ops,
};

static struct snd_soc_card snd_soc_sunxi_snddaudio = {
	.name           = "snddaudio",
	.owner          = THIS_MODULE,
};

static int sunxi_snddaudio_dev_probe(struct platform_device *pdev)
{
	struct device_node *np = pdev->dev.of_node;
	struct snd_soc_dai_link	*dai_link;
	struct snd_soc_card *card;
	struct sunxi_snddaudio_priv *snddaudio_priv;
	char name[30] = "snddaudio";
	unsigned int temp_val;
	int ret = 0;

	snddaudio_priv = devm_kzalloc(&pdev->dev,
			sizeof(struct sunxi_snddaudio_priv), GFP_KERNEL);
	if (!snddaudio_priv) {
		dev_err(&pdev->dev, "Can't malloc the sunxi_snddaudio_priv!\n");
		return -ENOMEM;
	}

	snddaudio_priv->card = devm_kzalloc(&pdev->dev,
					sizeof(struct snd_soc_card),
					GFP_KERNEL);
	if (!snddaudio_priv->card) {
		dev_err(&pdev->dev, "Can't malloc the snd_soc_card!\n");
		goto err_kfree_sndaudio_priv;
	}

	card = snddaudio_priv->card;
	memcpy(card, &snd_soc_sunxi_snddaudio, sizeof(struct snd_soc_card));

	card->dev = &pdev->dev;

	dai_link = devm_kzalloc(&pdev->dev,
			sizeof(struct snd_soc_dai_link), GFP_KERNEL);
	if (!dai_link) {
		ret = -ENOMEM;
		goto err_kfree_card;
	}
	memcpy(dai_link, &sunxi_snddaudio_dai_link,
			sizeof(struct snd_soc_dai_link));
	card->dai_link = dai_link;
	card->num_links = 1;

	ret = of_property_read_u32(np, "audio_format", &temp_val);
	if (ret < 0) {
		dev_warn(&pdev->dev, "audio_format config missing or invalid\n");
		snddaudio_priv->pdata.audio_format = 1;
	} else {
		snddaudio_priv->pdata.audio_format = temp_val;
	}

	ret = of_property_read_u32(np, "signal_inversion", &temp_val);
	if (ret < 0) {
		dev_warn(&pdev->dev, "signal_inversion config missing or invalid\n");
		snddaudio_priv->pdata.signal_inversion = 1;
	} else {
		snddaudio_priv->pdata.signal_inversion = temp_val;
	}

	ret = of_property_read_u32(np, "daudio_master", &temp_val);
	if (ret < 0) {
		dev_warn(&pdev->dev, "daudio_master config missing or invalid\n");
		/*
		 * default setting SND_SOC_DAIFMT_CBS_CFS mode
		 * codec clk & FRM slave
		 */
		snddaudio_priv->pdata.daudio_master = 4;
	} else {
		snddaudio_priv->pdata.daudio_master = temp_val;
	}

	dai_link->cpu_dai_name = NULL;
	dai_link->cpu_of_node = of_parse_phandle(np,
			"sunxi,daudio0-controller", 0);
	if (dai_link->cpu_of_node)
		goto cpu_node_find;

	dai_link->cpu_of_node = of_parse_phandle(np,
			"sunxi,daudio1-controller", 0);
	if (dai_link->cpu_of_node)
		goto cpu_node_find;

	dai_link->cpu_of_node = of_parse_phandle(np,
			"sunxi,daudio-controller", 0);
	if (dai_link->cpu_of_node)
		goto cpu_node_find;

	dev_err(card->dev, "Property 'sunxi,daudio-controller' missing\n");

	goto err_kfree_link;

cpu_node_find:
	dai_link->cpu_dai_name = NULL;
	dai_link->platform_name = NULL;
	dai_link->platform_of_node =
				dai_link->cpu_of_node;

	ret = of_property_read_string(np, "sunxi,snddaudio-codec",
			&dai_link->codec_name);
	/*
	 * As we setting codec & codec_dai in dtb, we just setting the
	 * codec & codec_dai in the dai_link. But if we just not setting,
	 * we then using the snd-soc-dummy as the codec & codec_dai to
	 * construct the snd-card for audio playback & capture.
	 */
	if (!ret) {
		ret = of_property_read_string(np, "sunxi,snddaudio-codec-dai",
				&dai_link->codec_dai_name);
		if (ret < 0) {
			dev_err(card->dev, "codec_dai name invaild in dtb\n");
			ret = -EINVAL;
			goto err_kfree_link;
		}
		sprintf(name+3, "%s", dai_link->codec_name);
	} else {
		if (dai_link->cpu_of_node && of_property_read_u32(
			dai_link->cpu_of_node, "tdm_num", &temp_val) >= 0)
			sprintf(name+9, "%u", temp_val);
		else
			sprintf(name+9, "%d", atomic_inc_return(&daudio_count));
	}

	card->name = name;
	dev_info(card->dev, "codec: %s, codec_dai: %s.\n",
			dai_link->codec_name,
			dai_link->codec_dai_name);

	snd_soc_card_set_drvdata(card, snddaudio_priv);

	ret = snd_soc_register_card(card);
	if (ret) {
		dev_err(card->dev, "snd_soc_register_card failed\n");
		goto err_kfree_link;
	}

	return ret;

err_kfree_link:
	devm_kfree(&pdev->dev, card->dai_link);
err_kfree_card:
	devm_kfree(&pdev->dev, card);
err_kfree_sndaudio_priv:
	devm_kfree(&pdev->dev, snddaudio_priv);
	return ret;
}

static int sunxi_snddaudio_dev_remove(struct platform_device *pdev)
{
	struct snd_soc_card *card = platform_get_drvdata(pdev);
	struct sunxi_snddaudio_priv *snddaudio_priv;

	snddaudio_priv = snd_soc_card_get_drvdata(card);

	snd_soc_unregister_card(card);
	devm_kfree(&pdev->dev, card->dai_link);
	devm_kfree(&pdev->dev, card);
	devm_kfree(&pdev->dev, snddaudio_priv);

	return 0;
}

static const struct of_device_id sunxi_snddaudio_of_match[] = {
	{ .compatible = "allwinner,sunxi-daudio0-machine", },
	{ .compatible = "allwinner,sunxi-daudio1-machine", },
	{},
};

static struct platform_driver sunxi_snddaudio_driver = {
	.driver = {
		.name   = "snddaudio",
		.owner  = THIS_MODULE,
		.pm     = &snd_soc_pm_ops,
		.of_match_table = sunxi_snddaudio_of_match,
	},
	.probe  = sunxi_snddaudio_dev_probe,
	.remove = sunxi_snddaudio_dev_remove,
};

static int __init sunxi_snddaudio_driver_init(void)
{
	return platform_driver_register(&sunxi_snddaudio_driver);
}
late_initcall(sunxi_snddaudio_driver_init);

static void __exit sunxi_snddaudio_driver_exit(void)
{
	platform_driver_unregister(&sunxi_snddaudio_driver);
}
module_exit(sunxi_snddaudio_driver_exit);

MODULE_AUTHOR("yumingfeng <yumingfeng@allwinnertech.com>");
MODULE_DESCRIPTION("SUNXI snddaudio ALSA SoC Audio Card Driver");
MODULE_LICENSE("GPL");