root/sound/soc/intel/avs/boards/nau8825.c
// SPDX-License-Identifier: GPL-2.0-only
//
// Copyright(c) 2021-2022 Intel Corporation
//
// Authors: Cezary Rojewski <cezary.rojewski@intel.com>
//          Amadeusz Slawinski <amadeuszx.slawinski@linux.intel.com>
//

#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/pcm_params.h>
#include <sound/soc.h>
#include <sound/soc-acpi.h>
#include "../../../codecs/nau8825.h"
#include "../utils.h"

#define SKL_NUVOTON_CODEC_DAI   "nau8825-hifi"

static int
avs_nau8825_clock_control(struct snd_soc_dapm_widget *w, struct snd_kcontrol *control, int event)
{
        struct snd_soc_card *card = snd_soc_dapm_to_card(w->dapm);
        struct snd_soc_dai *codec_dai;
        int ret;

        codec_dai = snd_soc_card_get_codec_dai(card, SKL_NUVOTON_CODEC_DAI);
        if (!codec_dai) {
                dev_err(card->dev, "Codec dai not found\n");
                return -EINVAL;
        }

        if (SND_SOC_DAPM_EVENT_ON(event))
                ret = snd_soc_dai_set_sysclk(codec_dai, NAU8825_CLK_MCLK, 24000000,
                                             SND_SOC_CLOCK_IN);
        else
                ret = snd_soc_dai_set_sysclk(codec_dai, NAU8825_CLK_INTERNAL, 0, SND_SOC_CLOCK_IN);
        if (ret < 0)
                dev_err(card->dev, "Set sysclk failed: %d\n", ret);

        return ret;
}

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

static const struct snd_soc_dapm_widget card_widgets[] = {
        SND_SOC_DAPM_HP("Headphone Jack", NULL),
        SND_SOC_DAPM_MIC("Headset Mic", NULL),
        SND_SOC_DAPM_SUPPLY("Platform Clock", SND_SOC_NOPM, 0, 0, avs_nau8825_clock_control,
                            SND_SOC_DAPM_PRE_PMU | SND_SOC_DAPM_POST_PMD),
};

static const struct snd_soc_dapm_route card_base_routes[] = {
        { "Headphone Jack", NULL, "HPOL" },
        { "Headphone Jack", NULL, "HPOR" },

        { "MIC", NULL, "Headset Mic" },

        { "Headphone Jack", NULL, "Platform Clock" },
        { "Headset Mic", NULL, "Platform Clock" },
};

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

static int avs_nau8825_codec_init(struct snd_soc_pcm_runtime *runtime)
{
        struct snd_soc_card *card = runtime->card;
        struct snd_soc_jack_pin *pins;
        struct snd_soc_jack *jack;
        int num_pins, ret;

        jack = snd_soc_card_get_drvdata(card);
        num_pins = ARRAY_SIZE(card_headset_pins);

        pins = devm_kmemdup_array(card->dev, card_headset_pins, num_pins,
                                  sizeof(card_headset_pins[0]), GFP_KERNEL);
        if (!pins)
                return -ENOMEM;

        /*
         * 4 buttons here map to the google Reference headset.
         * The use of these buttons can be decided by the user space.
         */
        ret = snd_soc_card_jack_new_pins(card, "Headset Jack", SND_JACK_HEADSET | SND_JACK_BTN_0 |
                                         SND_JACK_BTN_1 | SND_JACK_BTN_2 | SND_JACK_BTN_3,
                                         jack, pins, num_pins);
        if (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);

        return snd_soc_component_set_jack(snd_soc_rtd_to_codec(runtime, 0)->component, jack, NULL);
}

static void avs_nau8825_codec_exit(struct snd_soc_pcm_runtime *rtd)
{
        snd_soc_component_set_jack(snd_soc_rtd_to_codec(rtd, 0)->component, NULL, NULL);
}

static int
avs_nau8825_be_fixup(struct snd_soc_pcm_runtime *runtime, struct snd_pcm_hw_params *params)
{
        struct snd_interval *rate, *channels;
        struct snd_mask *fmt;

        rate = hw_param_interval(params, SNDRV_PCM_HW_PARAM_RATE);
        channels = hw_param_interval(params, SNDRV_PCM_HW_PARAM_CHANNELS);
        fmt = hw_param_mask(params, SNDRV_PCM_HW_PARAM_FORMAT);

        /* The ADSP will convert the FE rate to 48k, stereo */
        rate->min = rate->max = 48000;
        channels->min = channels->max = 2;

        /* set SSP to 24 bit */
        snd_mask_none(fmt);
        snd_mask_set_format(fmt, SNDRV_PCM_FORMAT_S24_LE);

        return 0;
}

static int avs_nau8825_trigger(struct snd_pcm_substream *substream, int cmd)
{
        struct snd_pcm_runtime *runtime = substream->runtime;
        struct snd_soc_pcm_runtime *rtm = snd_soc_substream_to_rtd(substream);
        struct snd_soc_dai *codec_dai = snd_soc_rtd_to_codec(rtm, 0);
        int ret = 0;

        switch (cmd) {
        case SNDRV_PCM_TRIGGER_START:
                ret = snd_soc_dai_set_sysclk(codec_dai, NAU8825_CLK_FLL_FS, 0, SND_SOC_CLOCK_IN);
                if (ret < 0) {
                        dev_err(codec_dai->dev, "can't set FS clock %d\n", ret);
                        break;
                }

                ret = snd_soc_dai_set_pll(codec_dai, 0, 0, runtime->rate, runtime->rate * 256);
                if (ret < 0)
                        dev_err(codec_dai->dev, "can't set FLL: %d\n", ret);
                break;

        case SNDRV_PCM_TRIGGER_RESUME:
                ret = snd_soc_dai_set_pll(codec_dai, 0, 0, runtime->rate, runtime->rate * 256);
                if (ret < 0)
                        dev_err(codec_dai->dev, "can't set FLL: %d\n", ret);
                break;
        }

        return ret;
}


static const struct snd_soc_ops avs_nau8825_ops = {
        .trigger = avs_nau8825_trigger,
};

static int avs_create_dai_link(struct device *dev, int ssp_port, int tdm_slot,
                               struct snd_soc_dai_link **dai_link)
{
        struct snd_soc_dai_link_component *platform;
        struct snd_soc_dai_link *dl;

        dl = devm_kzalloc(dev, sizeof(*dl), GFP_KERNEL);
        platform = devm_kzalloc(dev, sizeof(*platform), GFP_KERNEL);
        if (!dl || !platform)
                return -ENOMEM;

        dl->name = devm_kasprintf(dev, GFP_KERNEL,
                                  AVS_STRING_FMT("SSP", "-Codec", ssp_port, tdm_slot));
        dl->cpus = devm_kzalloc(dev, sizeof(*dl->cpus), GFP_KERNEL);
        dl->codecs = devm_kzalloc(dev, sizeof(*dl->codecs), GFP_KERNEL);
        if (!dl->name || !dl->cpus || !dl->codecs)
                return -ENOMEM;

        dl->cpus->dai_name = devm_kasprintf(dev, GFP_KERNEL,
                                            AVS_STRING_FMT("SSP", " Pin", ssp_port, tdm_slot));
        dl->codecs->name = devm_kasprintf(dev, GFP_KERNEL, "i2c-10508825:00");
        dl->codecs->dai_name = devm_kasprintf(dev, GFP_KERNEL, SKL_NUVOTON_CODEC_DAI);
        if (!dl->cpus->dai_name || !dl->codecs->name || !dl->codecs->dai_name)
                return -ENOMEM;

        platform->name = dev_name(dev);
        dl->num_cpus = 1;
        dl->num_codecs = 1;
        dl->platforms = platform;
        dl->num_platforms = 1;
        dl->id = 0;
        dl->dai_fmt = SND_SOC_DAIFMT_I2S | SND_SOC_DAIFMT_NB_NF | SND_SOC_DAIFMT_CBC_CFC;
        dl->init = avs_nau8825_codec_init;
        dl->exit = avs_nau8825_codec_exit;
        dl->be_hw_params_fixup = avs_nau8825_be_fixup;
        dl->ops = &avs_nau8825_ops;
        dl->nonatomic = 1;
        dl->no_pcm = 1;

        *dai_link = dl;

        return 0;
}

static int avs_card_suspend_pre(struct snd_soc_card *card)
{
        struct snd_soc_dai *codec_dai = snd_soc_card_get_codec_dai(card, SKL_NUVOTON_CODEC_DAI);

        return snd_soc_component_set_jack(codec_dai->component, NULL, NULL);
}

static int avs_card_resume_post(struct snd_soc_card *card)
{
        struct snd_soc_dai *codec_dai = snd_soc_card_get_codec_dai(card, SKL_NUVOTON_CODEC_DAI);
        struct snd_soc_jack *jack = snd_soc_card_get_drvdata(card);
        int stream = SNDRV_PCM_STREAM_PLAYBACK;

        if (!codec_dai) {
                dev_err(card->dev, "Codec dai not found\n");
                return -EINVAL;
        }

        if (snd_soc_dai_stream_active(codec_dai, stream) &&
            snd_soc_dai_get_widget(codec_dai, stream)->active)
                snd_soc_dai_set_sysclk(codec_dai, NAU8825_CLK_FLL_FS, 0, SND_SOC_CLOCK_IN);

        return snd_soc_component_set_jack(codec_dai->component, jack, NULL);
}

static int avs_nau8825_probe(struct platform_device *pdev)
{
        struct snd_soc_dai_link *dai_link;
        struct snd_soc_acpi_mach *mach;
        struct avs_mach_pdata *pdata;
        struct snd_soc_card *card;
        struct snd_soc_jack *jack;
        struct device *dev = &pdev->dev;
        int ssp_port, tdm_slot, ret;

        mach = dev_get_platdata(dev);
        pdata = mach->pdata;

        ret = avs_mach_get_ssp_tdm(dev, mach, &ssp_port, &tdm_slot);
        if (ret)
                return ret;

        ret = avs_create_dai_link(dev, ssp_port, tdm_slot, &dai_link);
        if (ret) {
                dev_err(dev, "Failed to create dai link: %d", ret);
                return ret;
        }

        jack = devm_kzalloc(dev, sizeof(*jack), GFP_KERNEL);
        card = devm_kzalloc(dev, sizeof(*card), GFP_KERNEL);
        if (!jack || !card)
                return -ENOMEM;

        if (pdata->obsolete_card_names) {
                card->name = "avs_nau8825";
        } else {
                card->driver_name = "avs_nau8825";
                card->long_name = card->name = "AVS I2S NAU8825";
        }
        card->dev = dev;
        card->owner = THIS_MODULE;
        card->suspend_pre = avs_card_suspend_pre;
        card->resume_post = avs_card_resume_post;
        card->dai_link = dai_link;
        card->num_links = 1;
        card->controls = card_controls;
        card->num_controls = ARRAY_SIZE(card_controls);
        card->dapm_widgets = card_widgets;
        card->num_dapm_widgets = ARRAY_SIZE(card_widgets);
        card->dapm_routes = card_base_routes;
        card->num_dapm_routes = ARRAY_SIZE(card_base_routes);
        card->fully_routed = true;
        snd_soc_card_set_drvdata(card, jack);

        return devm_snd_soc_register_deferrable_card(dev, card);
}

static const struct platform_device_id avs_nau8825_driver_ids[] = {
        {
                .name = "avs_nau8825",
        },
        {},
};
MODULE_DEVICE_TABLE(platform, avs_nau8825_driver_ids);

static struct platform_driver avs_nau8825_driver = {
        .probe = avs_nau8825_probe,
        .driver = {
                .name = "avs_nau8825",
                .pm = &snd_soc_pm_ops,
        },
        .id_table = avs_nau8825_driver_ids,
};

module_platform_driver(avs_nau8825_driver)

MODULE_DESCRIPTION("Intel nau8825 machine driver");
MODULE_LICENSE("GPL");