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

#include <linux/clk.h>
#include <linux/input.h>
#include <linux/module.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/rt5663.h"
#include "../utils.h"

#define RT5663_CODEC_DAI "rt5663-aif"

struct rt5663_private {
 struct snd_soc_jack jack;
};

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),
};

static const struct snd_soc_dapm_route card_routes[] = {
 /* HP jack connectors */
 { "Headphone Jack", NULL, "HPOL" },
 { "Headphone Jack", NULL, "HPOR" },

 /* Mic jacks */
 { "IN1P", NULL, "Headset Mic" },
 { "IN1N", NULL, "Headset Mic" },
};

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_rt5663_codec_init(struct snd_soc_pcm_runtime *runtime)
{
 struct snd_soc_card *card = runtime->card;
 struct rt5663_private *priv = snd_soc_card_get_drvdata(card);
 struct snd_soc_jack_pin *pins;
 struct snd_soc_jack *jack;
 int num_pins, ret;

 jack = &priv->jack;
 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;

 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);

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

 return 0;
}

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

static int
avs_rt5663_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 SSPN to 24 bit */
 snd_mask_none(fmt);
 snd_mask_set_format(fmt, SNDRV_PCM_FORMAT_S24_LE);

 return 0;
}

static int avs_rt5663_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);
 int ret;

 /* use ASRC for internal clocks, as PLL rate isn't multiple of BCLK */
 rt5663_sel_asrc_clk_src(codec_dai->component,
    RT5663_DA_STEREO_FILTER | RT5663_AD_STEREO_FILTER,
    RT5663_CLK_SEL_I2S1_ASRC);

 ret = snd_soc_dai_set_sysclk(codec_dai, RT5663_SCLK_S_MCLK, 24576000, SND_SOC_CLOCK_IN);

 return ret;
}

static const struct snd_soc_ops avs_rt5663_ops = {
 .hw_params = avs_rt5663_hw_params,
};


static int avs_create_dai_link(struct device *dev, const char *platform_name, 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;

 platform->name = platform_name;

 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-10EC5663:00");
 dl->codecs->dai_name = devm_kasprintf(dev, GFP_KERNEL, RT5663_CODEC_DAI);
 if (!dl->cpus->dai_name || !dl->codecs->name || !dl->codecs->dai_name)
  return -ENOMEM;

 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_rt5663_codec_init;
 dl->exit = avs_rt5663_codec_exit;
 dl->be_hw_params_fixup = avs_rt5663_be_fixup;
 dl->nonatomic = 1;
 dl->no_pcm = 1;
 dl->ops = &avs_rt5663_ops;

 *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, RT5663_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, RT5663_CODEC_DAI);
 struct snd_soc_jack *jack = snd_soc_card_get_drvdata(card);

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

static int avs_rt5663_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 rt5663_private *priv;
 struct device *dev = &pdev->dev;
 const char *pname;
 int ssp_port, tdm_slot, ret;

 mach = dev_get_platdata(dev);
 pname = mach->mach_params.platform;
 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, pname, ssp_port, tdm_slot, &dai_link);
 if (ret) {
  dev_err(dev, "Failed to create dai link: %d", ret);
  return ret;
 }

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

 if (pdata->obsolete_card_names) {
  card->name = "avs_rt5663";
 } else {
  card->driver_name = "avs_rt5663";
  card->long_name = card->name = "AVS I2S ALC5663";
 }
 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_routes;
 card->num_dapm_routes = ARRAY_SIZE(card_routes);
 card->fully_routed = true;
 snd_soc_card_set_drvdata(card, priv);

 ret = snd_soc_fixup_dai_links_platform_name(card, pname);
 if (ret)
  return ret;

 return devm_snd_soc_register_deferrable_card(dev, card);
}

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

static struct platform_driver avs_rt5663_driver = {
 .probe = avs_rt5663_probe,
 .driver = {
  .name = "avs_rt5663",
  .pm = &snd_soc_pm_ops,
 },
 .id_table = avs_rt5663_driver_ids,
};

module_platform_driver(avs_rt5663_driver);

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