// SPDX-License-Identifier: GPL-2.0+
//
// Speyside audio support
//
// Copyright 2011 Wolfson Microelectronics

#include <sound/soc.h>
#include <sound/soc-dapm.h>
#include <sound/jack.h>
#include <linux/gpio/machine.h>
#include <linux/gpio/consumer.h>
#include <linux/module.h>

#include "../codecs/wm8996.h"
#include "../codecs/wm9081.h"

#define MCLK_AUDIO_RATE (512 * 48000)

static int speyside_set_bias_level(struct snd_soc_card *card,
                                   struct snd_soc_dapm_context *dapm,
                                   enum snd_soc_bias_level level)
{
        struct snd_soc_pcm_runtime *rtd;
        struct snd_soc_dai *codec_dai;
        int ret;

        rtd = snd_soc_get_pcm_runtime(card, &card->dai_link[1]);
        codec_dai = snd_soc_rtd_to_codec(rtd, 0);

        if (snd_soc_dapm_to_dev(dapm) != codec_dai->dev)
                return 0;

        switch (level) {
        case SND_SOC_BIAS_STANDBY:
                ret = snd_soc_dai_set_sysclk(codec_dai, WM8996_SYSCLK_MCLK2,
                                             32768, SND_SOC_CLOCK_IN);
                if (ret < 0)
                        return ret;

                ret = snd_soc_dai_set_pll(codec_dai, WM8996_FLL_MCLK2,
                                          0, 0, 0);
                if (ret < 0) {
                        pr_err("Failed to stop FLL\n");
                        return ret;
                }
                break;

        default:
                break;
        }

        return 0;
}

static int speyside_set_bias_level_post(struct snd_soc_card *card,
                                        struct snd_soc_dapm_context *dapm,
                                        enum snd_soc_bias_level level)
{
        struct snd_soc_pcm_runtime *rtd;
        struct snd_soc_dai *codec_dai;
        int ret;

        rtd = snd_soc_get_pcm_runtime(card, &card->dai_link[1]);
        codec_dai = snd_soc_rtd_to_codec(rtd, 0);

        if (snd_soc_dapm_to_dev(dapm) != codec_dai->dev)
                return 0;

        switch (level) {
        case SND_SOC_BIAS_PREPARE:
                if (snd_soc_dapm_get_bias_level(dapm) == SND_SOC_BIAS_STANDBY) {
                        ret = snd_soc_dai_set_pll(codec_dai, 0,
                                                  WM8996_FLL_MCLK2,
                                                  32768, MCLK_AUDIO_RATE);
                        if (ret < 0) {
                                pr_err("Failed to start FLL\n");
                                return ret;
                        }

                        ret = snd_soc_dai_set_sysclk(codec_dai,
                                                     WM8996_SYSCLK_FLL,
                                                     MCLK_AUDIO_RATE,
                                                     SND_SOC_CLOCK_IN);
                        if (ret < 0)
                                return ret;
                }
                break;

        default:
                break;
        }

        return 0;
}

static struct snd_soc_jack speyside_headset;

/* Headset jack detection DAPM pins */
static struct snd_soc_jack_pin speyside_headset_pins[] = {
        {
                .pin = "Headset Mic",
                .mask = SND_JACK_MICROPHONE,
        },
};

static struct gpio_desc *speyside_hpsel_gpio;
/* Default the headphone selection to active high */
static int speyside_jack_polarity;

static int speyside_get_micbias(struct snd_soc_dapm_widget *source,
                                struct snd_soc_dapm_widget *sink)
{
        if (speyside_jack_polarity && (snd_soc_dapm_widget_name_cmp(source, "MICB1") == 0))
                return 1;
        if (!speyside_jack_polarity && (snd_soc_dapm_widget_name_cmp(source, "MICB2") == 0))
                return 1;

        return 0;
}

static void speyside_set_polarity(struct snd_soc_component *component,
                                  int polarity)
{
        speyside_jack_polarity = !polarity;
        gpiod_direction_output(speyside_hpsel_gpio, speyside_jack_polarity);

        /* Re-run DAPM to make sure we're using the correct mic bias */
        snd_soc_dapm_sync(snd_soc_component_to_dapm(component));
}

static int speyside_wm0010_init(struct snd_soc_pcm_runtime *rtd)
{
        struct snd_soc_dai *dai = snd_soc_rtd_to_codec(rtd, 0);
        int ret;

        ret = snd_soc_dai_set_sysclk(dai, 0, MCLK_AUDIO_RATE, 0);
        if (ret < 0)
                return ret;

        return 0;
}

static int speyside_wm8996_init(struct snd_soc_pcm_runtime *rtd)
{
        struct snd_soc_dai *dai = snd_soc_rtd_to_codec(rtd, 0);
        struct snd_soc_component *component = dai->component;
        enum gpiod_flags flags;
        int ret;

        ret = snd_soc_dai_set_sysclk(dai, WM8996_SYSCLK_MCLK2, 32768, 0);
        if (ret < 0)
                return ret;

        if (speyside_jack_polarity)
                flags = GPIOD_OUT_HIGH;
        else
                flags = GPIOD_OUT_LOW;
        speyside_hpsel_gpio = devm_gpiod_get(rtd->card->dev,
                                             "hp-sel",
                                             flags);
        if (IS_ERR(speyside_hpsel_gpio))
                return PTR_ERR(speyside_hpsel_gpio);

        ret = snd_soc_card_jack_new_pins(rtd->card, "Headset",
                                         SND_JACK_LINEOUT | SND_JACK_HEADSET |
                                         SND_JACK_BTN_0,
                                         &speyside_headset,
                                         speyside_headset_pins,
                                         ARRAY_SIZE(speyside_headset_pins));
        if (ret)
                return ret;

        wm8996_detect(component, &speyside_headset, speyside_set_polarity);

        return 0;
}

static int speyside_late_probe(struct snd_soc_card *card)
{
        struct snd_soc_dapm_context *dapm = snd_soc_card_to_dapm(card);

        snd_soc_dapm_ignore_suspend(dapm, "Headphone");
        snd_soc_dapm_ignore_suspend(dapm, "Headset Mic");
        snd_soc_dapm_ignore_suspend(dapm, "Main AMIC");
        snd_soc_dapm_ignore_suspend(dapm, "Main DMIC");
        snd_soc_dapm_ignore_suspend(dapm, "Main Speaker");
        snd_soc_dapm_ignore_suspend(dapm, "WM1250 Output");
        snd_soc_dapm_ignore_suspend(dapm, "WM1250 Input");

        return 0;
}

static const struct snd_soc_pcm_stream dsp_codec_params = {
        .formats = SNDRV_PCM_FMTBIT_S32_LE,
        .rate_min = 48000,
        .rate_max = 48000,
        .channels_min = 2,
        .channels_max = 2,
};

SND_SOC_DAILINK_DEFS(cpu_dsp,
        DAILINK_COMP_ARRAY(COMP_CPU("samsung-i2s.0")),
        DAILINK_COMP_ARRAY(COMP_CODEC("spi0.0", "wm0010-sdi1")),
        DAILINK_COMP_ARRAY(COMP_PLATFORM("samsung-i2s.0")));

SND_SOC_DAILINK_DEFS(dsp_codec,
        DAILINK_COMP_ARRAY(COMP_CPU("wm0010-sdi2")),
        DAILINK_COMP_ARRAY(COMP_CODEC("wm8996.1-001a", "wm8996-aif1")));

SND_SOC_DAILINK_DEFS(baseband,
        DAILINK_COMP_ARRAY(COMP_CPU("wm8996-aif2")),
        DAILINK_COMP_ARRAY(COMP_CODEC("wm1250-ev1.1-0027", "wm1250-ev1")));

static struct snd_soc_dai_link speyside_dai[] = {
        {
                .name = "CPU-DSP",
                .stream_name = "CPU-DSP",
                .init = speyside_wm0010_init,
                .dai_fmt = SND_SOC_DAIFMT_I2S | SND_SOC_DAIFMT_NB_NF
                                | SND_SOC_DAIFMT_CBP_CFP,
                SND_SOC_DAILINK_REG(cpu_dsp),
        },
        {
                .name = "DSP-CODEC",
                .stream_name = "DSP-CODEC",
                .init = speyside_wm8996_init,
                .dai_fmt = SND_SOC_DAIFMT_I2S | SND_SOC_DAIFMT_NB_NF
                                | SND_SOC_DAIFMT_CBP_CFP,
                .c2c_params = &dsp_codec_params,
                .num_c2c_params = 1,
                .ignore_suspend = 1,
                SND_SOC_DAILINK_REG(dsp_codec),
        },
        {
                .name = "Baseband",
                .stream_name = "Baseband",
                .dai_fmt = SND_SOC_DAIFMT_I2S | SND_SOC_DAIFMT_NB_NF
                                | SND_SOC_DAIFMT_CBP_CFP,
                .ignore_suspend = 1,
                SND_SOC_DAILINK_REG(baseband),
        },
};

static int speyside_wm9081_init(struct snd_soc_component *component)
{
        /* At any time the WM9081 is active it will have this clock */
        return snd_soc_component_set_sysclk(component, WM9081_SYSCLK_MCLK, 0,
                                        MCLK_AUDIO_RATE, 0);
}

static struct snd_soc_aux_dev speyside_aux_dev[] = {
        {
                .dlc = COMP_AUX("wm9081.1-006c"),
                .init = speyside_wm9081_init,
        },
};

static struct snd_soc_codec_conf speyside_codec_conf[] = {
        {
                .dlc = COMP_CODEC_CONF("wm9081.1-006c"),
                .name_prefix = "Sub",
        },
};

static const struct snd_kcontrol_new controls[] = {
        SOC_DAPM_PIN_SWITCH("Main Speaker"),
        SOC_DAPM_PIN_SWITCH("Main DMIC"),
        SOC_DAPM_PIN_SWITCH("Main AMIC"),
        SOC_DAPM_PIN_SWITCH("WM1250 Input"),
        SOC_DAPM_PIN_SWITCH("WM1250 Output"),
        SOC_DAPM_PIN_SWITCH("Headphone"),
};

static const struct snd_soc_dapm_widget widgets[] = {
        SND_SOC_DAPM_HP("Headphone", NULL),
        SND_SOC_DAPM_MIC("Headset Mic", NULL),

        SND_SOC_DAPM_SPK("Main Speaker", NULL),

        SND_SOC_DAPM_MIC("Main AMIC", NULL),
        SND_SOC_DAPM_MIC("Main DMIC", NULL),
};

static const struct snd_soc_dapm_route audio_paths[] = {
        { "IN1RN", NULL, "MICB1" },
        { "IN1RP", NULL, "MICB1" },
        { "IN1RN", NULL, "MICB2" },
        { "IN1RP", NULL, "MICB2" },
        { "MICB1", NULL, "Headset Mic", speyside_get_micbias },
        { "MICB2", NULL, "Headset Mic", speyside_get_micbias },

        { "IN1LP", NULL, "MICB2" },
        { "IN1RN", NULL, "MICB1" },
        { "MICB2", NULL, "Main AMIC" },

        { "DMIC1DAT", NULL, "MICB1" },
        { "DMIC2DAT", NULL, "MICB1" },
        { "MICB1", NULL, "Main DMIC" },

        { "Headphone", NULL, "HPOUT1L" },
        { "Headphone", NULL, "HPOUT1R" },

        { "Sub IN1", NULL, "HPOUT2L" },
        { "Sub IN2", NULL, "HPOUT2R" },

        { "Main Speaker", NULL, "Sub SPKN" },
        { "Main Speaker", NULL, "Sub SPKP" },
        { "Main Speaker", NULL, "SPKDAT" },
};

static struct snd_soc_card speyside = {
        .name = "Speyside",
        .owner = THIS_MODULE,
        .dai_link = speyside_dai,
        .num_links = ARRAY_SIZE(speyside_dai),
        .aux_dev = speyside_aux_dev,
        .num_aux_devs = ARRAY_SIZE(speyside_aux_dev),
        .codec_conf = speyside_codec_conf,
        .num_configs = ARRAY_SIZE(speyside_codec_conf),

        .set_bias_level = speyside_set_bias_level,
        .set_bias_level_post = speyside_set_bias_level_post,

        .controls = controls,
        .num_controls = ARRAY_SIZE(controls),
        .dapm_widgets = widgets,
        .num_dapm_widgets = ARRAY_SIZE(widgets),
        .dapm_routes = audio_paths,
        .num_dapm_routes = ARRAY_SIZE(audio_paths),
        .fully_routed = true,

        .late_probe = speyside_late_probe,
};

static struct gpiod_lookup_table wm8996_gpiod_table = {
        /* Hardcoded device name in board file mach-crag6410.c */
        .dev_id = "speyside",
        .table = {
                /*
                 * This line was hardcoded to 214 in the global GPIO
                 * number space, S3C GPIO macros seems top set the
                 * wm8996 codec GPIO start offset to 212, so this will
                 * be GPIO 214 - 212 = 2 on the wm8996.
                 */
                GPIO_LOOKUP("wm8996", 2, "hp-sel", GPIO_ACTIVE_HIGH),
                { },
        },
};

static void speyside_gpiod_table_action(void *data)
{
        gpiod_remove_lookup_table(&wm8996_gpiod_table);
}

static int speyside_probe(struct platform_device *pdev)
{
        struct snd_soc_card *card = &speyside;
        int ret;

        card->dev = &pdev->dev;

        gpiod_add_lookup_table(&wm8996_gpiod_table);
        ret = devm_add_action_or_reset(&pdev->dev, speyside_gpiod_table_action,
                                       NULL);
        if (ret)
                return ret;

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

        return ret;
}

static struct platform_driver speyside_driver = {
        .driver = {
                .name = "speyside",
                .pm = &snd_soc_pm_ops,
        },
        .probe = speyside_probe,
};

module_platform_driver(speyside_driver);

MODULE_DESCRIPTION("Speyside audio support");
MODULE_AUTHOR("Mark Brown <broonie@opensource.wolfsonmicro.com>");
MODULE_LICENSE("GPL");
MODULE_ALIAS("platform:speyside");