root/sound/soc/qcom/sc7280.c
// SPDX-License-Identifier: GPL-2.0-only
//
// Copyright (c) 2020-2021, The Linux Foundation. All rights reserved.
//
// ALSA SoC Machine driver for sc7280

#include <dt-bindings/sound/qcom,lpass.h>
#include <dt-bindings/sound/qcom,q6afe.h>
#include <linux/input.h>
#include <linux/module.h>
#include <linux/platform_device.h>
#include <sound/core.h>
#include <sound/jack.h>
#include <sound/pcm.h>
#include <sound/soc.h>
#include <sound/rt5682s.h>
#include <linux/soundwire/sdw.h>
#include <sound/pcm_params.h>

#include "../codecs/rt5682.h"
#include "../codecs/rt5682s.h"
#include "common.h"
#include "lpass.h"
#include "qdsp6/q6afe.h"
#include "sdw.h"

#define DEFAULT_MCLK_RATE              19200000
#define RT5682_PLL_FREQ (48000 * 512)
#define MI2S_BCLK_RATE          1536000

struct sc7280_snd_data {
        struct snd_soc_card card;
        u32 pri_mi2s_clk_count;
        struct snd_soc_jack hs_jack;
        struct snd_soc_jack hdmi_jack;
        bool jack_setup;
        bool stream_prepared[LPASS_MAX_PORTS];
};

static void sc7280_jack_free(struct snd_jack *jack)
{
        struct snd_soc_component *component = jack->private_data;

        snd_soc_component_set_jack(component, NULL, NULL);
}

static struct snd_soc_jack_pin sc7280_jack_pins[] = {
        {
                .pin = "Headphone Jack",
                .mask = SND_JACK_HEADPHONE,
        },
        {
                .pin = "Headset Mic",
                .mask = SND_JACK_MICROPHONE,
        },
};

static int sc7280_headset_init(struct snd_soc_pcm_runtime *rtd)
{
        struct snd_soc_card *card = rtd->card;
        struct sc7280_snd_data *pdata = snd_soc_card_get_drvdata(card);
        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);
        struct snd_soc_component *component = codec_dai->component;
        struct snd_jack *jack;
        int rval, i;

        if (!pdata->jack_setup) {
                rval = snd_soc_card_jack_new_pins(card, "Headset Jack",
                                                  SND_JACK_HEADSET | SND_JACK_LINEOUT |
                                                  SND_JACK_MECHANICAL |
                                                  SND_JACK_BTN_0 | SND_JACK_BTN_1 |
                                                  SND_JACK_BTN_2 | SND_JACK_BTN_3 |
                                                  SND_JACK_BTN_4 | SND_JACK_BTN_5,
                                                  &pdata->hs_jack,
                                                  sc7280_jack_pins,
                                                  ARRAY_SIZE(sc7280_jack_pins));

                if (rval < 0) {
                        dev_err(card->dev, "Unable to add Headset Jack\n");
                        return rval;
                }

                jack = pdata->hs_jack.jack;

                snd_jack_set_key(jack, SND_JACK_BTN_0, KEY_PLAYPAUSE);
                snd_jack_set_key(jack, SND_JACK_BTN_1, KEY_VOICECOMMAND);
                snd_jack_set_key(jack, SND_JACK_BTN_2, KEY_VOLUMEUP);
                snd_jack_set_key(jack, SND_JACK_BTN_3, KEY_VOLUMEDOWN);

                jack->private_data = component;
                jack->private_free = sc7280_jack_free;
                pdata->jack_setup = true;
        }
        switch (cpu_dai->id) {
        case MI2S_PRIMARY:
        case LPASS_CDC_DMA_RX0:
        case LPASS_CDC_DMA_TX3:
        case TX_CODEC_DMA_TX_3:
                for_each_rtd_codec_dais(rtd, i, codec_dai) {
                        rval = snd_soc_component_set_jack(component, &pdata->hs_jack, NULL);
                        if (rval != 0 && rval != -ENOTSUPP) {
                                dev_err(card->dev, "Failed to set jack: %d\n", rval);
                                return rval;
                        }
                }
                break;
        default:
                break;
        }

        return 0;
}

static int sc7280_hdmi_init(struct snd_soc_pcm_runtime *rtd)
{
        struct snd_soc_card *card = rtd->card;
        struct sc7280_snd_data *pdata = snd_soc_card_get_drvdata(card);
        struct snd_soc_dai *codec_dai = snd_soc_rtd_to_codec(rtd, 0);
        struct snd_soc_component *component = codec_dai->component;
        struct snd_jack *jack;
        int rval;

        rval = snd_soc_card_jack_new(card, "HDMI Jack", SND_JACK_LINEOUT,
                                     &pdata->hdmi_jack);

        if (rval < 0) {
                dev_err(card->dev, "Unable to add HDMI Jack\n");
                return rval;
        }

        jack = pdata->hdmi_jack.jack;
        jack->private_data = component;
        jack->private_free = sc7280_jack_free;

        return snd_soc_component_set_jack(component, &pdata->hdmi_jack, NULL);
}

static int sc7280_rt5682_init(struct snd_soc_pcm_runtime *rtd)
{
        struct snd_soc_dai *cpu_dai = snd_soc_rtd_to_cpu(rtd, 0);
        struct snd_soc_dai *codec_dai = snd_soc_rtd_to_codec(rtd, 0);
        struct snd_soc_card *card = rtd->card;
        struct sc7280_snd_data *data = snd_soc_card_get_drvdata(card);
        int ret;

        if (++data->pri_mi2s_clk_count == 1) {
                snd_soc_dai_set_sysclk(cpu_dai,
                        LPASS_MCLK0,
                        DEFAULT_MCLK_RATE,
                        SNDRV_PCM_STREAM_PLAYBACK);
        }
        snd_soc_dai_set_fmt(codec_dai,
                                SND_SOC_DAIFMT_CBC_CFC |
                                SND_SOC_DAIFMT_NB_NF |
                                SND_SOC_DAIFMT_I2S);

        ret = snd_soc_dai_set_pll(codec_dai, RT5682S_PLL2, RT5682S_PLL_S_MCLK,
                                        DEFAULT_MCLK_RATE, RT5682_PLL_FREQ);
        if (ret) {
                dev_err(rtd->dev, "can't set codec pll: %d\n", ret);
                return ret;
        }

        ret = snd_soc_dai_set_sysclk(codec_dai, RT5682S_SCLK_S_PLL2,
                                        RT5682_PLL_FREQ,
                                        SND_SOC_CLOCK_IN);

        if (ret) {
                dev_err(rtd->dev, "snd_soc_dai_set_sysclk err = %d\n",
                        ret);
                return ret;
        }

        return 0;
}

static int sc7280_init(struct snd_soc_pcm_runtime *rtd)
{
        struct snd_soc_dai *cpu_dai = snd_soc_rtd_to_cpu(rtd, 0);

        switch (cpu_dai->id) {
        case MI2S_PRIMARY:
        case LPASS_CDC_DMA_TX3:
        case TX_CODEC_DMA_TX_3:
                return sc7280_headset_init(rtd);
        case LPASS_CDC_DMA_RX0:
        case LPASS_CDC_DMA_VA_TX0:
        case MI2S_SECONDARY:
        case RX_CODEC_DMA_RX_0:
        case SECONDARY_MI2S_RX:
        case VA_CODEC_DMA_TX_0:
                return 0;
        case LPASS_DP_RX:
                return sc7280_hdmi_init(rtd);
        default:
                dev_err(rtd->dev, "%s: invalid dai id 0x%x\n", __func__, cpu_dai->id);
        }

        return -EINVAL;
}

static int sc7280_snd_hw_params(struct snd_pcm_substream *substream,
                                struct snd_pcm_hw_params *params)
{
        struct snd_pcm_runtime *runtime = substream->runtime;
        struct snd_soc_pcm_runtime *rtd = snd_soc_substream_to_rtd(substream);

        if (!rtd->dai_link->no_pcm) {
                snd_pcm_hw_constraint_minmax(runtime, SNDRV_PCM_HW_PARAM_CHANNELS, 2, 2);
                snd_pcm_hw_constraint_minmax(runtime, SNDRV_PCM_HW_PARAM_RATE, 48000, 48000);
        }

        return 0;
}

static int sc7280_snd_swr_prepare(struct snd_pcm_substream *substream)
{
        struct snd_soc_pcm_runtime *rtd = snd_soc_substream_to_rtd(substream);
        const struct snd_soc_dai *cpu_dai = snd_soc_rtd_to_cpu(rtd, 0);
        struct sc7280_snd_data *data = snd_soc_card_get_drvdata(rtd->card);

        return qcom_snd_sdw_prepare(substream, &data->stream_prepared[cpu_dai->id]);
}

static int sc7280_snd_prepare(struct snd_pcm_substream *substream)
{
        struct snd_soc_pcm_runtime *rtd = snd_soc_substream_to_rtd(substream);
        const struct snd_soc_dai *cpu_dai = snd_soc_rtd_to_cpu(rtd, 0);

        switch (cpu_dai->id) {
        case LPASS_CDC_DMA_RX0:
        case LPASS_CDC_DMA_TX3:
        case RX_CODEC_DMA_RX_0:
        case TX_CODEC_DMA_TX_3:
        case VA_CODEC_DMA_TX_0:
                return sc7280_snd_swr_prepare(substream);
        default:
                break;
        }

        return 0;
}

static int sc7280_snd_hw_free(struct snd_pcm_substream *substream)
{
        struct snd_soc_pcm_runtime *rtd = snd_soc_substream_to_rtd(substream);
        struct sc7280_snd_data *data = snd_soc_card_get_drvdata(rtd->card);
        const struct snd_soc_dai *cpu_dai = snd_soc_rtd_to_cpu(rtd, 0);

        return qcom_snd_sdw_hw_free(substream, &data->stream_prepared[cpu_dai->id]);
}

static void sc7280_snd_shutdown(struct snd_pcm_substream *substream)
{
        struct snd_soc_pcm_runtime *rtd = snd_soc_substream_to_rtd(substream);
        struct snd_soc_card *card = rtd->card;
        struct sc7280_snd_data *data = snd_soc_card_get_drvdata(card);
        struct snd_soc_dai *cpu_dai = snd_soc_rtd_to_cpu(rtd, 0);

        switch (cpu_dai->id) {
        case MI2S_PRIMARY:
                if (--data->pri_mi2s_clk_count == 0) {
                        snd_soc_dai_set_sysclk(cpu_dai,
                                               LPASS_MCLK0,
                                               0,
                                               SNDRV_PCM_STREAM_PLAYBACK);
                }
                break;
        case SECONDARY_MI2S_RX:
                snd_soc_dai_set_sysclk(cpu_dai, Q6AFE_LPASS_CLK_ID_SEC_MI2S_IBIT,
                                               0, SNDRV_PCM_STREAM_PLAYBACK);
                break;
        default:
                break;
        }

        qcom_snd_sdw_shutdown(substream);
}

static int sc7280_snd_startup(struct snd_pcm_substream *substream)
{
        unsigned int fmt = SND_SOC_DAIFMT_CBC_CFC;
        unsigned int codec_dai_fmt = SND_SOC_DAIFMT_CBC_CFC;
        struct snd_soc_pcm_runtime *rtd = snd_soc_substream_to_rtd(substream);
        struct snd_soc_dai *cpu_dai = snd_soc_rtd_to_cpu(rtd, 0);
        struct snd_soc_dai *codec_dai = snd_soc_rtd_to_codec(rtd, 0);
        int ret = 0;

        switch (cpu_dai->id) {
        case MI2S_PRIMARY:
                ret = sc7280_rt5682_init(rtd);
                if (ret)
                        return ret;
                break;
        case SECONDARY_MI2S_RX:
                codec_dai_fmt |= SND_SOC_DAIFMT_NB_NF | SND_SOC_DAIFMT_I2S;

                snd_soc_dai_set_sysclk(cpu_dai, Q6AFE_LPASS_CLK_ID_SEC_MI2S_IBIT,
                        MI2S_BCLK_RATE, SNDRV_PCM_STREAM_PLAYBACK);

                snd_soc_dai_set_fmt(cpu_dai, fmt);
                snd_soc_dai_set_fmt(codec_dai, codec_dai_fmt);
                break;
        default:
                break;
        }

        return qcom_snd_sdw_startup(substream);
}

static const struct snd_soc_ops sc7280_ops = {
        .startup = sc7280_snd_startup,
        .hw_params = sc7280_snd_hw_params,
        .hw_free = sc7280_snd_hw_free,
        .prepare = sc7280_snd_prepare,
        .shutdown = sc7280_snd_shutdown,
};

static const struct snd_soc_dapm_widget sc7280_snd_widgets[] = {
        SND_SOC_DAPM_HP("Headphone Jack", NULL),
        SND_SOC_DAPM_MIC("Headset Mic", NULL),
};

static const struct snd_kcontrol_new sc7280_snd_controls[] = {
        SOC_DAPM_PIN_SWITCH("Headphone Jack"),
        SOC_DAPM_PIN_SWITCH("Headset Mic"),
};

static int sc7280_snd_be_hw_params_fixup(struct snd_soc_pcm_runtime *rtd,
                                         struct snd_pcm_hw_params *params)
{
        struct snd_interval *rate = hw_param_interval(params, SNDRV_PCM_HW_PARAM_RATE);
        struct snd_interval *channels = hw_param_interval(params, SNDRV_PCM_HW_PARAM_CHANNELS);
        struct snd_mask *fmt = hw_param_mask(params, SNDRV_PCM_HW_PARAM_FORMAT);

        rate->min = rate->max = 48000;
        channels->min = channels->max = 2;
        snd_mask_set_format(fmt, SNDRV_PCM_FORMAT_S16_LE);

        return 0;
}

static int sc7280_snd_platform_probe(struct platform_device *pdev)
{
        struct snd_soc_card *card;
        struct sc7280_snd_data *data;
        struct device *dev = &pdev->dev;
        struct snd_soc_dai_link *link;
        int ret, i;

        data = devm_kzalloc(dev, sizeof(*data), GFP_KERNEL);
        if (!data)
                return -ENOMEM;

        card = &data->card;
        snd_soc_card_set_drvdata(card, data);

        card->owner = THIS_MODULE;
        card->driver_name = "SC7280";
        card->dev = dev;

        card->dapm_widgets = sc7280_snd_widgets;
        card->num_dapm_widgets = ARRAY_SIZE(sc7280_snd_widgets);
        card->controls = sc7280_snd_controls;
        card->num_controls = ARRAY_SIZE(sc7280_snd_controls);

        ret = qcom_snd_parse_of(card);
        if (ret)
                return ret;

        for_each_card_prelinks(card, i, link) {
                link->init = sc7280_init;
                link->ops = &sc7280_ops;
                if (link->no_pcm == 1)
                        link->be_hw_params_fixup = sc7280_snd_be_hw_params_fixup;
        }

        return devm_snd_soc_register_card(dev, card);
}

static const struct of_device_id sc7280_snd_device_id[]  = {
        { .compatible = "google,sc7280-herobrine" },
        {}
};
MODULE_DEVICE_TABLE(of, sc7280_snd_device_id);

static struct platform_driver sc7280_snd_driver = {
        .probe = sc7280_snd_platform_probe,
        .driver = {
                .name = "msm-snd-sc7280",
                .of_match_table = sc7280_snd_device_id,
                .pm = &snd_soc_pm_ops,
        },
};
module_platform_driver(sc7280_snd_driver);

MODULE_DESCRIPTION("sc7280 ASoC Machine Driver");
MODULE_LICENSE("GPL");