root/sound/soc/mediatek/mt8196/mt8196-nau8825.c
// SPDX-License-Identifier: GPL-2.0
/*
 *  mt8196-nau8825.c  --  mt8196 nau8825 ALSA SoC machine driver
 *
 *  Copyright (c) 2025 MediaTek Inc.
 *  Author: Darren Ye <darren.ye@mediatek.com>
 */

#include <linux/input.h>
#include <linux/module.h>
#include <linux/of_device.h>
#include <linux/pm_runtime.h>

#include <sound/soc.h>
#include <sound/jack.h>
#include <sound/pcm_params.h>

#include "mt8196-afe-common.h"

#include "../../codecs/nau8825.h"
#include "../../codecs/rt5682s.h"

#include "../common/mtk-soc-card.h"
#include "../common/mtk-dsp-sof-common.h"
#include "../common/mtk-soundcard-driver.h"
#include "../common/mtk-afe-platform-driver.h"

#define NAU8825_HS_PRESENT      BIT(0)
#define RT5682S_HS_PRESENT      BIT(1)
#define RT5650_HS_PRESENT       BIT(2)

/*
 * Nau88l25
 */
#define NAU8825_CODEC_DAI  "nau8825-hifi"

/*
 * Rt5682s
 */
#define RT5682S_CODEC_DAI     "rt5682s-aif1"

/*
 * Rt5650
 */
#define RT5650_CODEC_DAI     "rt5645-aif1"

#define SOF_DMA_DL1 "SOF_DMA_DL1"
#define SOF_DMA_DL_24CH "SOF_DMA_DL_24CH"
#define SOF_DMA_UL0 "SOF_DMA_UL0"
#define SOF_DMA_UL1 "SOF_DMA_UL1"
#define SOF_DMA_UL2 "SOF_DMA_UL2"

enum mt8196_jacks {
        MT8196_JACK_HEADSET,
        MT8196_JACK_DP,
        MT8196_JACK_HDMI,
        MT8196_JACK_MAX,
};

static struct snd_soc_jack_pin mt8196_dp_jack_pins[] = {
        {
                .pin = "DP",
                .mask = SND_JACK_AVOUT,
        },
};

static struct snd_soc_jack_pin mt8196_hdmi_jack_pins[] = {
        {
                .pin = "HDMI",
                .mask = SND_JACK_AVOUT,
        },
};

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

static const struct snd_kcontrol_new mt8196_dumb_spk_controls[] = {
        SOC_DAPM_PIN_SWITCH("Ext Spk"),
};

static const struct snd_soc_dapm_widget mt8196_dumb_spk_widgets[] = {
        SND_SOC_DAPM_SPK("Ext Spk", NULL),
};

static const struct snd_soc_dapm_widget mt8196_nau8825_widgets[] = {
        SND_SOC_DAPM_HP("Headphone Jack", NULL),
        SND_SOC_DAPM_MIC("Headset Mic", NULL),
        SND_SOC_DAPM_SPK("Ext Spk", NULL),
        SND_SOC_DAPM_SINK("DP"),
};

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

#define EXT_SPK_AMP_W_NAME "Ext_Speaker_Amp"

static struct snd_soc_card mt8196_nau8825_soc_card;

static const struct snd_soc_dapm_widget mt8196_nau8825_card_widgets[] = {
        /* SOF Uplink */
        SND_SOC_DAPM_MIXER("SOF_DMA_UL0", SND_SOC_NOPM, 0, 0, NULL, 0),
        SND_SOC_DAPM_MIXER("SOF_DMA_UL1", SND_SOC_NOPM, 0, 0, NULL, 0),
        SND_SOC_DAPM_MIXER("SOF_DMA_UL2", SND_SOC_NOPM, 0, 0, NULL, 0),

        /*
         * SOF Downlink
         * the widgets on the machine driver cannot use the parameter with kcontrol
         * because the widget domain is its platform driver. so sof downlink route
         * is written in the i2s dai driver.
         */
};

static const struct snd_soc_dapm_route mt8196_nau8825_card_routes[] = {
        /* SOF Uplink */
        {"SOF_DMA_UL0", NULL, "UL0_CH1"},
        {"SOF_DMA_UL0", NULL, "UL0_CH2"},
        /* SOF Uplink */
        {"SOF_DMA_UL1", NULL, "UL1_CH1"},
        {"SOF_DMA_UL1", NULL, "UL1_CH2"},
        /* SOF Uplink */
        {"SOF_DMA_UL2", NULL, "UL2_CH1"},
        {"SOF_DMA_UL2", NULL, "UL2_CH2"},
};

static const struct snd_kcontrol_new mt8196_nau8825_card_controls[] = {
        SOC_DAPM_PIN_SWITCH(EXT_SPK_AMP_W_NAME),
};

/*
 * define mtk_spk_i2s_mck node in dts when need mclk,
 * BE i2s need assign snd_soc_ops = mt8196_nau8825_i2s_ops
 */
static int mt8196_nau8825_i2s_hw_params(struct snd_pcm_substream *substream,
                                        struct snd_pcm_hw_params *params)
{
        struct snd_soc_pcm_runtime *rtd = substream->private_data;
        unsigned int rate = params_rate(params);
        unsigned int mclk_fs_ratio = 128;
        unsigned int mclk_fs = rate * mclk_fs_ratio;
        struct snd_soc_dai *cpu_dai = snd_soc_rtd_to_cpu(rtd, 0);

        return snd_soc_dai_set_sysclk(cpu_dai,
                                      0, mclk_fs, SND_SOC_CLOCK_OUT);
}

static const struct snd_soc_ops mt8196_nau8825_i2s_ops = {
        .hw_params = mt8196_nau8825_i2s_hw_params,
};

static int mt8196_dptx_hw_params(struct snd_pcm_substream *substream,
                                 struct snd_pcm_hw_params *params)
{
        struct snd_soc_pcm_runtime *rtd = substream->private_data;
        unsigned int rate = params_rate(params);
        unsigned int mclk_fs_ratio = 256;
        unsigned int mclk_fs = rate * mclk_fs_ratio;
        struct snd_soc_dai *dai = snd_soc_rtd_to_cpu(rtd, 0);

        return snd_soc_dai_set_sysclk(dai, 0, mclk_fs, SND_SOC_CLOCK_OUT);
}

static const struct snd_soc_ops mt8196_dptx_ops = {
        .hw_params = mt8196_dptx_hw_params,
};

static int mt8196_hw_params_fixup(struct snd_soc_pcm_runtime *rtd,
                                  struct snd_pcm_hw_params *params)
{
        dev_info(rtd->dev, "fix format to 32bit\n");

        /* fix BE i2s format to 32bit, clean param mask first */
        snd_mask_reset_range(hw_param_mask(params, SNDRV_PCM_HW_PARAM_FORMAT),
                             0, (__force unsigned int)SNDRV_PCM_FORMAT_LAST);

        params_set_format(params, SNDRV_PCM_FORMAT_S32_LE);
        return 0;
}

static int mt8196_sof_be_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_component *cmpnt_afe = NULL;
        struct snd_soc_pcm_runtime *runtime;

        /* find afe component */
        for_each_card_rtds(rtd->card, runtime) {
                cmpnt_afe = snd_soc_rtdcom_lookup(runtime, AFE_PCM_NAME);
                if (cmpnt_afe) {
                        dev_info(rtd->dev, "component->name: %s\n", cmpnt_afe->name);
                        break;
                }
        }

        if (cmpnt_afe && !pm_runtime_active(cmpnt_afe->dev)) {
                dev_err(rtd->dev, "afe pm runtime is not active!!\n");
                return -EINVAL;
        }

        return 0;
}

static const struct snd_soc_ops mt8196_sof_be_ops = {
        .hw_params = mt8196_sof_be_hw_params,
};

static const struct sof_conn_stream g_sof_conn_streams[] = {
        {
                .sof_link = "AFE_SOF_DL1",
                .sof_dma = SOF_DMA_DL1,
                .stream_dir = SNDRV_PCM_STREAM_PLAYBACK
        },
        {
                .sof_link = "AFE_SOF_DL_24CH",
                .sof_dma = SOF_DMA_DL_24CH,
                .stream_dir = SNDRV_PCM_STREAM_PLAYBACK
        },
        {
                .sof_link = "AFE_SOF_UL0",
                .sof_dma = SOF_DMA_UL0,
                .stream_dir = SNDRV_PCM_STREAM_CAPTURE
        },
        {
                .sof_link = "AFE_SOF_UL1",
                .sof_dma = SOF_DMA_UL1,
                .stream_dir = SNDRV_PCM_STREAM_CAPTURE
        },
        {
                .sof_link = "AFE_SOF_UL2",
                .sof_dma = SOF_DMA_UL2,
                .stream_dir = SNDRV_PCM_STREAM_CAPTURE
        },
};

/* FE */
SND_SOC_DAILINK_DEFS(playback1,
                     DAILINK_COMP_ARRAY(COMP_CPU("DL1")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(playback_24ch,
                     DAILINK_COMP_ARRAY(COMP_CPU("DL_24CH")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(capture0,
                     DAILINK_COMP_ARRAY(COMP_CPU("UL0")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(capture1,
                     DAILINK_COMP_ARRAY(COMP_CPU("UL1")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(capture2,
                     DAILINK_COMP_ARRAY(COMP_CPU("UL2")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(playback_hdmi,
                     DAILINK_COMP_ARRAY(COMP_CPU("HDMI")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(playback2,
                     DAILINK_COMP_ARRAY(COMP_CPU("DL2")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(capture_cm0,
                     DAILINK_COMP_ARRAY(COMP_CPU("UL_CM0")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
/* BE */
SND_SOC_DAILINK_DEFS(ap_dmic,
                     DAILINK_COMP_ARRAY(COMP_CPU("AP_DMIC")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(ap_dmic_ch34,
                     DAILINK_COMP_ARRAY(COMP_CPU("AP_DMIC_CH34")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(ap_dmic_multich,
                     DAILINK_COMP_ARRAY(COMP_CPU("AP_DMIC_MULTICH")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(i2sin6,
                     DAILINK_COMP_ARRAY(COMP_CPU("I2SIN6")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(i2sout3,
                     DAILINK_COMP_ARRAY(COMP_CPU("I2SOUT3")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(i2sout4,
                     DAILINK_COMP_ARRAY(COMP_CPU("I2SOUT4")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(i2sout6,
                     DAILINK_COMP_ARRAY(COMP_CPU("I2SOUT6")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(tdm_dptx,
                     DAILINK_COMP_ARRAY(COMP_CPU("TDM_DPTX")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(AFE_SOF_DL_24CH,
                     DAILINK_COMP_ARRAY(COMP_CPU("SOF_DL_24CH")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(AFE_SOF_DL1,
                     DAILINK_COMP_ARRAY(COMP_CPU("SOF_DL1")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(AFE_SOF_UL0,
                     DAILINK_COMP_ARRAY(COMP_CPU("SOF_UL0")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(AFE_SOF_UL1,
                     DAILINK_COMP_ARRAY(COMP_CPU("SOF_UL1")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));
SND_SOC_DAILINK_DEFS(AFE_SOF_UL2,
                     DAILINK_COMP_ARRAY(COMP_CPU("SOF_UL2")),
                     DAILINK_COMP_ARRAY(COMP_DUMMY()),
                     DAILINK_COMP_ARRAY(COMP_EMPTY()));

static struct snd_soc_dai_link mt8196_nau8825_dai_links[] = {
        /*
         * The SOF topology expects PCM streams 0~4 to be available
         * for the SOF PCM streams. Put the SOF BE definitions here
         * so that the PCM device numbers are skipped over.
         * (BE dailinks do not have PCM devices created.)
         */
        {
                .name = "AFE_SOF_DL_24CH",
                .no_pcm = 1,
                .playback_only = 1,
                .ops = &mt8196_sof_be_ops,
                SND_SOC_DAILINK_REG(AFE_SOF_DL_24CH),
        },
        {
                .name = "AFE_SOF_DL1",
                .no_pcm = 1,
                .playback_only = 1,
                .ops = &mt8196_sof_be_ops,
                SND_SOC_DAILINK_REG(AFE_SOF_DL1),
        },
        {
                .name = "AFE_SOF_UL0",
                .no_pcm = 1,
                .capture_only = 1,
                .ops = &mt8196_sof_be_ops,
                SND_SOC_DAILINK_REG(AFE_SOF_UL0),
        },
        {
                .name = "AFE_SOF_UL1",
                .no_pcm = 1,
                .capture_only = 1,
                .ops = &mt8196_sof_be_ops,
                SND_SOC_DAILINK_REG(AFE_SOF_UL1),
        },
        {
                .name = "AFE_SOF_UL2",
                .no_pcm = 1,
                .capture_only = 1,
                .ops = &mt8196_sof_be_ops,
                SND_SOC_DAILINK_REG(AFE_SOF_UL2),
        },
        /* Front End DAI links */
        {
                .name = "HDMI_FE",
                .stream_name = "HDMI Playback",
                .trigger = {SND_SOC_DPCM_TRIGGER_PRE,
                            SND_SOC_DPCM_TRIGGER_PRE},
                .dynamic = 1,
                .playback_only = 1,
                SND_SOC_DAILINK_REG(playback_hdmi),
        },
        {
                .name = "DL2_FE",
                .stream_name = "DL2 Playback",
                .trigger = {SND_SOC_DPCM_TRIGGER_PRE,
                            SND_SOC_DPCM_TRIGGER_PRE},
                .dynamic = 1,
                .playback_only = 1,
                SND_SOC_DAILINK_REG(playback2),
        },
        {
                .name = "UL_CM0_FE",
                .stream_name = "UL_CM0 Capture",
                .trigger = {SND_SOC_DPCM_TRIGGER_PRE,
                            SND_SOC_DPCM_TRIGGER_PRE},
                .dynamic = 1,
                .capture_only = 1,
                SND_SOC_DAILINK_REG(capture_cm0),
        },
        {
                .name = "DL_24CH_FE",
                .stream_name = "DL_24CH Playback",
                .trigger = {SND_SOC_DPCM_TRIGGER_PRE,
                                SND_SOC_DPCM_TRIGGER_PRE},
                .dynamic = 1,
                .playback_only = 1,
                SND_SOC_DAILINK_REG(playback_24ch),
        },
        {
                .name = "DL1_FE",
                .stream_name = "DL1 Playback",
                .trigger = {SND_SOC_DPCM_TRIGGER_PRE,
                                SND_SOC_DPCM_TRIGGER_PRE},
                .dynamic = 1,
                .playback_only = 1,
                SND_SOC_DAILINK_REG(playback1),
        },
        {
                .name = "UL0_FE",
                .stream_name = "UL0 Capture",
                .trigger = {SND_SOC_DPCM_TRIGGER_PRE,
                                SND_SOC_DPCM_TRIGGER_PRE},
                .dynamic = 1,
                .capture_only = 1,
                SND_SOC_DAILINK_REG(capture0),
        },
        {
                .name = "UL1_FE",
                .stream_name = "UL1 Capture",
                .trigger = {SND_SOC_DPCM_TRIGGER_PRE,
                                SND_SOC_DPCM_TRIGGER_PRE},
                .dynamic = 1,
                .capture_only = 1,
                SND_SOC_DAILINK_REG(capture1),
        },
        {
                .name = "UL2_FE",
                .stream_name = "UL2 Capture",
                .trigger = {SND_SOC_DPCM_TRIGGER_PRE,
                                SND_SOC_DPCM_TRIGGER_PRE},
                .dynamic = 1,
                .capture_only = 1,
                SND_SOC_DAILINK_REG(capture2),
        },
        /* Back End DAI links */
        {
                .name = "I2SIN6_BE",
                .dai_fmt = SND_SOC_DAIFMT_I2S | SND_SOC_DAIFMT_CBC_CFC
                        | SND_SOC_DAIFMT_GATED,
                .ops = &mt8196_nau8825_i2s_ops,
                .no_pcm = 1,
                .capture_only = 1,
                .ignore_suspend = 1,
                .be_hw_params_fixup = mt8196_hw_params_fixup,
                SND_SOC_DAILINK_REG(i2sin6),
        },
        {
                .name = "I2SOUT4_BE",
                .dai_fmt = SND_SOC_DAIFMT_I2S | SND_SOC_DAIFMT_CBC_CFC
                        | SND_SOC_DAIFMT_GATED,
                .ops = &mt8196_nau8825_i2s_ops,
                .no_pcm = 1,
                .playback_only = 1,
                .ignore_suspend = 1,
                .ignore_pmdown_time = 1,
                .be_hw_params_fixup = mt8196_hw_params_fixup,
                SND_SOC_DAILINK_REG(i2sout4),
        },
        {
                .name = "I2SOUT6_BE",
                .dai_fmt = SND_SOC_DAIFMT_I2S | SND_SOC_DAIFMT_CBC_CFC
                        | SND_SOC_DAIFMT_GATED,
                .ops = &mt8196_nau8825_i2s_ops,
                .no_pcm = 1,
                .playback_only = 1,
                .ignore_suspend = 1,
                .be_hw_params_fixup = mt8196_hw_params_fixup,
                SND_SOC_DAILINK_REG(i2sout6),
        },
        {
                .name = "AP_DMIC_BE",
                .no_pcm = 1,
                .capture_only = 1,
                .ignore_suspend = 1,
                SND_SOC_DAILINK_REG(ap_dmic),
        },
        {
                .name = "AP_DMIC_CH34_BE",
                .no_pcm = 1,
                .capture_only = 1,
                .ignore_suspend = 1,
                SND_SOC_DAILINK_REG(ap_dmic_ch34),
        },
        {
                .name = "AP_DMIC_MULTICH_BE",
                .no_pcm = 1,
                .capture_only = 1,
                .ignore_suspend = 1,
                SND_SOC_DAILINK_REG(ap_dmic_multich),
        },
        {
                .name = "TDM_DPTX_BE",
                .dai_fmt = SND_SOC_DAIFMT_I2S | SND_SOC_DAIFMT_CBC_CFC
                        | SND_SOC_DAIFMT_GATED,
                .ops = &mt8196_dptx_ops,
                .be_hw_params_fixup = mt8196_hw_params_fixup,
                .no_pcm = 1,
                .playback_only = 1,
                .ignore_suspend = 1,
                SND_SOC_DAILINK_REG(tdm_dptx),
        },
        {
                .name = "I2SOUT3_BE",
                .dai_fmt = SND_SOC_DAIFMT_I2S | SND_SOC_DAIFMT_CBC_CFC
                        | SND_SOC_DAIFMT_GATED,
                .ops = &mt8196_nau8825_i2s_ops,
                .no_pcm = 1,
                .playback_only = 1,
                .ignore_suspend = 1,
                SND_SOC_DAILINK_REG(i2sout3),
        },
};

static int mt8196_dumb_amp_init(struct snd_soc_pcm_runtime *rtd)
{
        struct snd_soc_card *card = rtd->card;
        struct snd_soc_dapm_context *dapm = snd_soc_card_to_dapm(card);
        int ret = 0;

        ret = snd_soc_dapm_new_controls(dapm, mt8196_dumb_spk_widgets,
                                        ARRAY_SIZE(mt8196_dumb_spk_widgets));
        if (ret) {
                dev_err(rtd->dev, "unable to add Dumb Speaker dapm, ret %d\n", ret);
                return ret;
        }

        ret = snd_soc_add_card_controls(card, mt8196_dumb_spk_controls,
                                        ARRAY_SIZE(mt8196_dumb_spk_controls));
        if (ret) {
                dev_err(rtd->dev, "unable to add Dumb card controls, ret %d\n", ret);
                return ret;
        }

        return 0;
}

static int mt8196_dptx_codec_init(struct snd_soc_pcm_runtime *rtd)
{
        struct mtk_soc_card_data *soc_card_data = snd_soc_card_get_drvdata(rtd->card);
        struct snd_soc_jack *jack = &soc_card_data->card_data->jacks[MT8196_JACK_DP];
        struct snd_soc_component *component = snd_soc_rtd_to_codec(rtd, 0)->component;
        int ret = 0;

        ret = snd_soc_card_jack_new_pins(rtd->card, "DP Jack", SND_JACK_AVOUT,
                                         jack, mt8196_dp_jack_pins,
                                         ARRAY_SIZE(mt8196_dp_jack_pins));
        if (ret) {
                dev_err(rtd->dev, "new jack failed: %d\n", ret);
                return ret;
        }

        ret = snd_soc_component_set_jack(component, jack, NULL);
        if (ret) {
                dev_err(rtd->dev, "set jack failed on %s (ret=%d)\n",
                        component->name, ret);
                return ret;
        }

        return 0;
}

static int mt8196_hdmi_codec_init(struct snd_soc_pcm_runtime *rtd)
{
        struct mtk_soc_card_data *soc_card_data = snd_soc_card_get_drvdata(rtd->card);
        struct snd_soc_jack *jack = &soc_card_data->card_data->jacks[MT8196_JACK_HDMI];
        struct snd_soc_component *component = snd_soc_rtd_to_codec(rtd, 0)->component;
        int ret = 0;

        ret = snd_soc_card_jack_new_pins(rtd->card, "HDMI Jack", SND_JACK_AVOUT,
                                         jack, mt8196_hdmi_jack_pins,
                                         ARRAY_SIZE(mt8196_hdmi_jack_pins));
        if (ret) {
                dev_err(rtd->dev, "new jack failed: %d\n", ret);
                return ret;
        }

        ret = snd_soc_component_set_jack(component, jack, NULL);
        if (ret) {
                dev_err(rtd->dev, "set jack failed on %s (ret=%d)\n",
                        component->name, ret);
                return ret;
        }

        return 0;
}

static int mt8196_headset_codec_init(struct snd_soc_pcm_runtime *rtd)
{
        struct snd_soc_card *card = rtd->card;
        struct snd_soc_dapm_context *dapm = snd_soc_card_to_dapm(card);
        struct mtk_soc_card_data *soc_card_data = snd_soc_card_get_drvdata(card);
        struct snd_soc_jack *jack = &soc_card_data->card_data->jacks[MT8196_JACK_HEADSET];
        struct snd_soc_component *component = snd_soc_rtd_to_codec(rtd, 0)->component;
        int ret;
        int type;

        ret = snd_soc_dapm_new_controls(dapm, mt8196_nau8825_widgets,
                                        ARRAY_SIZE(mt8196_nau8825_widgets));
        if (ret) {
                dev_err(rtd->dev, "unable to add nau8825 card widget, ret %d\n", ret);
                return ret;
        }

        ret = snd_soc_add_card_controls(card, mt8196_nau8825_controls,
                                        ARRAY_SIZE(mt8196_nau8825_controls));
        if (ret) {
                dev_err(rtd->dev, "unable to add nau8825 card controls, ret %d\n", ret);
                return ret;
        }

        ret = snd_soc_card_jack_new_pins(rtd->card, "Headset Jack",
                                         SND_JACK_HEADSET | SND_JACK_BTN_0 |
                                         SND_JACK_BTN_1 | SND_JACK_BTN_2 |
                                         SND_JACK_BTN_3,
                                         jack,
                                         nau8825_jack_pins,
                                         ARRAY_SIZE(nau8825_jack_pins));
        if (ret) {
                dev_err(rtd->dev, "Headset Jack creation failed: %d\n", ret);
                return ret;
        }

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

        type = SND_JACK_HEADSET | SND_JACK_BTN_0 | SND_JACK_BTN_1 | SND_JACK_BTN_2 | SND_JACK_BTN_3;
        ret = snd_soc_component_set_jack(component, jack, (void *)&type);

        if (ret) {
                dev_err(rtd->dev, "Headset Jack call-back failed: %d\n", ret);
                return ret;
        }

        return 0;
};

static void mt8196_headset_codec_exit(struct snd_soc_pcm_runtime *rtd)
{
        struct snd_soc_component *component = snd_soc_rtd_to_codec(rtd, 0)->component;

        snd_soc_component_set_jack(component, NULL, NULL);
}

static int mt8196_nau8825_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);
        unsigned int rate = params_rate(params);
        unsigned int bit_width = params_width(params);
        int clk_freq, ret;

        clk_freq = rate * 2 * bit_width;

        /* Configure clock for codec */
        ret = snd_soc_dai_set_sysclk(codec_dai, NAU8825_CLK_FLL_BLK, 0,
                                     SND_SOC_CLOCK_IN);
        if (ret < 0) {
                dev_err(codec_dai->dev, "can't set BCLK clock %d\n", ret);
                return ret;
        }

        /* Configure pll for codec */
        ret = snd_soc_dai_set_pll(codec_dai, 0, 0, clk_freq,
                                  params_rate(params) * 256);
        if (ret < 0) {
                dev_err(codec_dai->dev, "can't set BCLK: %d\n", ret);
                return ret;
        }

        return 0;
}

static const struct snd_soc_ops mt8196_nau8825_ops = {
        .hw_params = mt8196_nau8825_hw_params,
};

static int mt8196_rt5682s_i2s_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_card *card = rtd->card;
        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);
        unsigned int rate = params_rate(params);
        int bitwidth;
        int ret;

        bitwidth = snd_pcm_format_width(params_format(params));
        if (bitwidth < 0) {
                dev_err(card->dev, "invalid bit width: %d\n", bitwidth);
                return bitwidth;
        }

        ret = snd_soc_dai_set_tdm_slot(codec_dai, 0x00, 0x0, 0x2, bitwidth);
        if (ret) {
                dev_err(card->dev, "failed to set tdm slot\n");
                return ret;
        }

        ret = snd_soc_dai_set_pll(codec_dai, RT5682S_PLL1, RT5682S_PLL_S_BCLK1,
                                  rate * 32, rate * 512);
        if (ret) {
                dev_err(card->dev, "failed to set pll\n");
                return ret;
        }

        dev_info(card->dev, "%s set mclk rate: %d\n", __func__, rate * 512);

        ret = snd_soc_dai_set_sysclk(codec_dai, RT5682S_SCLK_S_MCLK,
                                     rate * 512, SND_SOC_CLOCK_IN);
        if (ret) {
                dev_err(card->dev, "failed to set sysclk\n");
                return ret;
        }

        return snd_soc_dai_set_sysclk(cpu_dai, 0, rate * 512,
                                      SND_SOC_CLOCK_OUT);
}

static const struct snd_soc_ops mt8196_rt5682s_i2s_ops = {
        .hw_params = mt8196_rt5682s_i2s_hw_params,
};

static int mt8196_nau8825_soc_card_probe(struct mtk_soc_card_data *soc_card_data, bool legacy)
{
        struct snd_soc_card *card = soc_card_data->card_data->card;
        struct snd_soc_dai_link *dai_link;
        bool init_nau8825 = false;
        bool init_rt5682s = false;
        bool init_rt5650 = false;
        bool init_dumb = false;
        int i;

        dev_info(card->dev, "legacy: %d\n", legacy);

        for_each_card_prelinks(card, i, dai_link) {
                if (strcmp(dai_link->name, "TDM_DPTX_BE") == 0) {
                        if (dai_link->num_codecs &&
                            strcmp(dai_link->codecs->dai_name, "snd-soc-dummy-dai"))
                                dai_link->init = mt8196_dptx_codec_init;
                } else if (strcmp(dai_link->name, "I2SOUT3_BE") == 0) {
                        if (dai_link->num_codecs &&
                            strcmp(dai_link->codecs->dai_name, "snd-soc-dummy-dai"))
                                dai_link->init = mt8196_hdmi_codec_init;
                } else if (strcmp(dai_link->name, "I2SOUT6_BE") == 0 ||
                           strcmp(dai_link->name, "I2SIN6_BE") == 0) {
                        if (!strcmp(dai_link->codecs->dai_name, NAU8825_CODEC_DAI)) {
                                dai_link->ops = &mt8196_nau8825_ops;
                                if (!init_nau8825) {
                                        dai_link->init = mt8196_headset_codec_init;
                                        dai_link->exit = mt8196_headset_codec_exit;
                                        init_nau8825 = true;
                                }
                        } else if (!strcmp(dai_link->codecs->dai_name, RT5682S_CODEC_DAI)) {
                                dai_link->ops = &mt8196_rt5682s_i2s_ops;
                                if (!init_rt5682s) {
                                        dai_link->init = mt8196_headset_codec_init;
                                        dai_link->exit = mt8196_headset_codec_exit;
                                        init_rt5682s = true;
                                }
                        } else if (!strcmp(dai_link->codecs->dai_name, RT5650_CODEC_DAI)) {
                                dai_link->ops = &mt8196_rt5682s_i2s_ops;
                                if (!init_rt5650) {
                                        dai_link->init = mt8196_headset_codec_init;
                                        dai_link->exit = mt8196_headset_codec_exit;
                                        init_rt5650 = true;
                                }
                        } else {
                                if (strcmp(dai_link->codecs->dai_name, "snd-soc-dummy-dai")) {
                                        if (!init_dumb) {
                                                dai_link->init = mt8196_dumb_amp_init;
                                                init_dumb = true;
                                        }
                                }
                        }
                }
        }

        return 0;
}

static const struct mtk_sof_priv mt8196_sof_priv = {
        .conn_streams = g_sof_conn_streams,
        .num_streams = ARRAY_SIZE(g_sof_conn_streams),
};

static struct snd_soc_card mt8196_nau8825_soc_card = {
        .owner = THIS_MODULE,
        .dai_link = mt8196_nau8825_dai_links,
        .num_links = ARRAY_SIZE(mt8196_nau8825_dai_links),
        .dapm_widgets = mt8196_nau8825_card_widgets,
        .num_dapm_widgets = ARRAY_SIZE(mt8196_nau8825_card_widgets),
        .dapm_routes = mt8196_nau8825_card_routes,
        .num_dapm_routes = ARRAY_SIZE(mt8196_nau8825_card_routes),
        .controls = mt8196_nau8825_card_controls,
        .num_controls = ARRAY_SIZE(mt8196_nau8825_card_controls),
};

static const struct mtk_soundcard_pdata mt8196_nau8825_card = {
        .card_name = "mt8196_nau8825",
        .card_data = &(struct mtk_platform_card_data) {
                .card = &mt8196_nau8825_soc_card,
                .num_jacks = MT8196_JACK_MAX,
                .flags = NAU8825_HS_PRESENT
        },
        .sof_priv = &mt8196_sof_priv,
        .soc_probe = mt8196_nau8825_soc_card_probe,
};

static const struct mtk_soundcard_pdata mt8196_rt5682s_card = {
        .card_name = "mt8196_rt5682s",
        .card_data = &(struct mtk_platform_card_data) {
                .card = &mt8196_nau8825_soc_card,
                .num_jacks = MT8196_JACK_MAX,
                .flags = RT5682S_HS_PRESENT
        },
        .sof_priv = &mt8196_sof_priv,
        .soc_probe = mt8196_nau8825_soc_card_probe,
};

static const struct mtk_soundcard_pdata mt8196_rt5650_card = {
        .card_name = "mt8196_rt5650",
        .card_data = &(struct mtk_platform_card_data) {
                .card = &mt8196_nau8825_soc_card,
                .num_jacks = MT8196_JACK_MAX,
                .flags = RT5650_HS_PRESENT
        },
        .sof_priv = &mt8196_sof_priv,
        .soc_probe = mt8196_nau8825_soc_card_probe,
};

static const struct of_device_id mt8196_nau8825_dt_match[] = {
        {.compatible = "mediatek,mt8196-nau8825-sound", .data = &mt8196_nau8825_card,},
        {.compatible = "mediatek,mt8196-rt5682s-sound", .data = &mt8196_rt5682s_card,},
        {.compatible = "mediatek,mt8196-rt5650-sound", .data = &mt8196_rt5650_card,},
        {}
};
MODULE_DEVICE_TABLE(of, mt8196_nau8825_dt_match);

static struct platform_driver mt8196_nau8825_driver = {
        .driver = {
                .name = "mt8196-nau8825",
                .of_match_table = mt8196_nau8825_dt_match,
                .pm = &snd_soc_pm_ops,
        },
        .probe = mtk_soundcard_common_probe,
};
module_platform_driver(mt8196_nau8825_driver);

/* Module information */
MODULE_DESCRIPTION("MT8196 nau8825 ALSA SoC machine driver");
MODULE_AUTHOR("Darren Ye <darren.ye@mediatek.com>");
MODULE_LICENSE("GPL");
MODULE_ALIAS("mt8196 nau8825 soc card");