// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Copyright 2011 Freescale Semiconductor, Inc.
 */

#include <linux/module.h>
#include <linux/device.h>
#include <linux/of.h>
#include <sound/core.h>
#include <sound/pcm.h>
#include <sound/soc.h>
#include <sound/jack.h>
#include <sound/soc-dapm.h>

#include "../codecs/sgtl5000.h"
#include "mxs-saif.h"

static int mxs_sgtl5000_hw_params(struct snd_pcm_substream *substream,
        struct snd_pcm_hw_params *params)
{
        struct snd_soc_pcm_runtime *rtd = snd_soc_substream_to_rtd(substream);
        struct snd_soc_dai *codec_dai = snd_soc_rtd_to_codec(rtd, 0);
        struct snd_soc_dai *cpu_dai = snd_soc_rtd_to_cpu(rtd, 0);
        unsigned int rate = params_rate(params);
        u32 mclk;
        int ret;

        /* sgtl5000 does not support 512*rate when in 96000 fs */
        switch (rate) {
        case 96000:
                mclk = 256 * rate;
                break;
        default:
                mclk = 512 * rate;
                break;
        }

        /* Set SGTL5000's SYSCLK (provided by SAIF MCLK) */
        ret = snd_soc_dai_set_sysclk(codec_dai, SGTL5000_SYSCLK, mclk, 0);
        if (ret) {
                dev_err(codec_dai->dev, "Failed to set sysclk to %u.%03uMHz\n",
                        mclk / 1000000, mclk / 1000 % 1000);
                return ret;
        }

        /* The SAIF MCLK should be the same as SGTL5000_SYSCLK */
        ret = snd_soc_dai_set_sysclk(cpu_dai, MXS_SAIF_MCLK, mclk, 0);
        if (ret) {
                dev_err(cpu_dai->dev, "Failed to set sysclk to %u.%03uMHz\n",
                        mclk / 1000000, mclk / 1000 % 1000);
                return ret;
        }

        return 0;
}

static const struct snd_soc_ops mxs_sgtl5000_hifi_ops = {
        .hw_params = mxs_sgtl5000_hw_params,
};

#define MXS_SGTL5000_DAI_FMT (SND_SOC_DAIFMT_I2S | SND_SOC_DAIFMT_NB_NF | \
        SND_SOC_DAIFMT_CBC_CFC)


SND_SOC_DAILINK_DEFS(hifi_tx,
        DAILINK_COMP_ARRAY(COMP_EMPTY()),
        DAILINK_COMP_ARRAY(COMP_CODEC(NULL, "sgtl5000")),
        DAILINK_COMP_ARRAY(COMP_EMPTY()));

SND_SOC_DAILINK_DEFS(hifi_rx,
        DAILINK_COMP_ARRAY(COMP_EMPTY()),
        DAILINK_COMP_ARRAY(COMP_CODEC(NULL, "sgtl5000")),
        DAILINK_COMP_ARRAY(COMP_EMPTY()));

static struct snd_soc_dai_link mxs_sgtl5000_dai[] = {
        {
                .name           = "HiFi Tx",
                .stream_name    = "HiFi Playback",
                .dai_fmt        = MXS_SGTL5000_DAI_FMT,
                .ops            = &mxs_sgtl5000_hifi_ops,
                .playback_only  = true,
                SND_SOC_DAILINK_REG(hifi_tx),
        }, {
                .name           = "HiFi Rx",
                .stream_name    = "HiFi Capture",
                .dai_fmt        = MXS_SGTL5000_DAI_FMT,
                .ops            = &mxs_sgtl5000_hifi_ops,
                .capture_only   = true,
                SND_SOC_DAILINK_REG(hifi_rx),
        },
};

static const struct snd_soc_dapm_widget mxs_sgtl5000_dapm_widgets[] = {
        SND_SOC_DAPM_MIC("Mic Jack", NULL),
        SND_SOC_DAPM_LINE("Line In Jack", NULL),
        SND_SOC_DAPM_HP("Headphone Jack", NULL),
        SND_SOC_DAPM_SPK("Line Out Jack", NULL),
        SND_SOC_DAPM_SPK("Ext Spk", NULL),
};

static struct snd_soc_card mxs_sgtl5000 = {
        .name           = "mxs_sgtl5000",
        .owner          = THIS_MODULE,
        .dai_link       = mxs_sgtl5000_dai,
        .num_links      = ARRAY_SIZE(mxs_sgtl5000_dai),
};

static int mxs_sgtl5000_probe(struct platform_device *pdev)
{
        struct snd_soc_card *card = &mxs_sgtl5000;
        int ret, i;
        struct device_node *np = pdev->dev.of_node;
        struct device_node *saif_np[2], *codec_np;

        saif_np[0] = of_parse_phandle(np, "saif-controllers", 0);
        saif_np[1] = of_parse_phandle(np, "saif-controllers", 1);
        codec_np = of_parse_phandle(np, "audio-codec", 0);
        if (!saif_np[0] || !saif_np[1] || !codec_np) {
                dev_err(&pdev->dev, "phandle missing or invalid\n");
                of_node_put(codec_np);
                of_node_put(saif_np[0]);
                of_node_put(saif_np[1]);
                return -EINVAL;
        }

        for (i = 0; i < 2; i++) {
                mxs_sgtl5000_dai[i].codecs->name = NULL;
                mxs_sgtl5000_dai[i].codecs->of_node = codec_np;
                mxs_sgtl5000_dai[i].cpus->dai_name = NULL;
                mxs_sgtl5000_dai[i].cpus->of_node = saif_np[i];
                mxs_sgtl5000_dai[i].platforms->name = NULL;
                mxs_sgtl5000_dai[i].platforms->of_node = saif_np[i];
        }

        of_node_put(codec_np);
        of_node_put(saif_np[0]);
        of_node_put(saif_np[1]);

        /*
         * Set an init clock(11.28Mhz) for sgtl5000 initialization(i2c r/w).
         * The Sgtl5000 sysclk is derived from saif0 mclk and it's range
         * should be >= 8MHz and <= 27M.
         */
        ret = mxs_saif_get_mclk(0, 44100 * 256, 44100);
        if (ret) {
                dev_err(&pdev->dev, "failed to get mclk\n");
                return ret;
        }

        card->dev = &pdev->dev;

        if (of_property_present(np, "audio-routing")) {
                card->dapm_widgets = mxs_sgtl5000_dapm_widgets;
                card->num_dapm_widgets = ARRAY_SIZE(mxs_sgtl5000_dapm_widgets);

                ret = snd_soc_of_parse_audio_routing(card, "audio-routing");
                if (ret) {
                        dev_err(&pdev->dev, "failed to parse audio-routing (%d)\n",
                                ret);
                        return ret;
                }
        }

        ret = devm_snd_soc_register_card(&pdev->dev, card);
        if (ret)
                return dev_err_probe(&pdev->dev, ret, "snd_soc_register_card failed\n");

        return 0;
}

static void mxs_sgtl5000_remove(struct platform_device *pdev)
{
        mxs_saif_put_mclk(0);
}

static const struct of_device_id mxs_sgtl5000_dt_ids[] = {
        { .compatible = "fsl,mxs-audio-sgtl5000", },
        { /* sentinel */ }
};
MODULE_DEVICE_TABLE(of, mxs_sgtl5000_dt_ids);

static struct platform_driver mxs_sgtl5000_audio_driver = {
        .driver = {
                .name = "mxs-sgtl5000",
                .of_match_table = mxs_sgtl5000_dt_ids,
        },
        .probe = mxs_sgtl5000_probe,
        .remove = mxs_sgtl5000_remove,
};

module_platform_driver(mxs_sgtl5000_audio_driver);

MODULE_AUTHOR("Freescale Semiconductor, Inc.");
MODULE_DESCRIPTION("MXS ALSA SoC Machine driver");
MODULE_LICENSE("GPL");
MODULE_ALIAS("platform:mxs-sgtl5000");