// SPDX-License-Identifier: GPL-2.0-only
// Copyright(c) 2021 Intel Corporation.
// Copyright(c) 2021 Nuvoton Corporation.

/*
 * Intel SOF Machine Driver with Nuvoton headphone codec NAU8825
 * and speaker codec RT1019P MAX98360a or MAX98373
 */
#include <linux/i2c.h>
#include <linux/input.h>
#include <linux/module.h>
#include <linux/platform_device.h>
#include <linux/dmi.h>
#include <sound/core.h>
#include <sound/jack.h>
#include <sound/pcm.h>
#include <sound/pcm_params.h>
#include <sound/soc.h>
#include <sound/sof.h>
#include <sound/soc-acpi.h>
#include "../../codecs/nau8825.h"
#include "../common/soc-intel-quirks.h"
#include "sof_board_helpers.h"
#include "sof_realtek_common.h"
#include "sof_maxim_common.h"
#include "sof_nuvoton_common.h"

static unsigned long sof_nau8825_quirk = SOF_SSP_PORT_CODEC(0);

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

static int sof_nau8825_codec_init(struct snd_soc_pcm_runtime *rtd)
{
 struct sof_card_private *ctx = snd_soc_card_get_drvdata(rtd->card);
 struct snd_soc_component *component = snd_soc_rtd_to_codec(rtd, 0)->component;
 struct snd_soc_jack *jack = &ctx->headset_jack;
 int ret;

 /*
 * Headset buttons map to the google Reference headset.
 * These can be configured by userspace.
 */
 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,
      jack_pins,
      ARRAY_SIZE(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);

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

 return ret;
};

static void sof_nau8825_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 sof_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);
 int clk_freq, ret;

 clk_freq = sof_dai_get_bclk(rtd); /* BCLK freq */

 if (clk_freq <= 0) {
  dev_err(rtd->dev, "get bclk freq failed: %d\n", clk_freq);
  return -EINVAL;
 }

 /* 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 ret;
}

static const struct snd_soc_ops sof_nau8825_ops = {
 .hw_params = sof_nau8825_hw_params,
};

static int sof_card_late_probe(struct snd_soc_card *card)
{
 struct sof_card_private *ctx = snd_soc_card_get_drvdata(card);
 struct snd_soc_dapm_context *dapm = &card->dapm;
 int err;

 if (ctx->amp_type == CODEC_MAX98373) {
  /* Disable Left and Right Spk pin after boot */
  snd_soc_dapm_disable_pin(dapm, "Left Spk");
  snd_soc_dapm_disable_pin(dapm, "Right Spk");
  err = snd_soc_dapm_sync(dapm);
  if (err < 0)
   return err;
 }

 return sof_intel_board_card_late_probe(card);
}

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

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

static const struct snd_soc_dapm_route sof_map[] = {
 /* HP jack connectors - unknown if we have jack detection */
 { "Headphone Jack", NULL, "HPOL" },
 { "Headphone Jack", NULL, "HPOR" },

 /* other jacks */
 { "MIC", NULL, "Headset Mic" },
};

/* sof audio machine driver for nau8825 codec */
static struct snd_soc_card sof_audio_card_nau8825 = {
 .name = "nau8825", /* the sof- prefix is added by the core */
 .owner = THIS_MODULE,
 .controls = sof_controls,
 .num_controls = ARRAY_SIZE(sof_controls),
 .dapm_widgets = sof_widgets,
 .num_dapm_widgets = ARRAY_SIZE(sof_widgets),
 .dapm_routes = sof_map,
 .num_dapm_routes = ARRAY_SIZE(sof_map),
 .fully_routed = true,
 .late_probe = sof_card_late_probe,
};

static struct snd_soc_dai_link_component nau8825_component[] = {
 {
  .name = "i2c-10508825:00",
  .dai_name = "nau8825-hifi",
 }
};

static int
sof_card_dai_links_create(struct device *dev, struct snd_soc_card *card,
     struct sof_card_private *ctx)
{
 int ret;

 ret = sof_intel_board_set_dai_link(dev, card, ctx);
 if (ret)
  return ret;

 if (!ctx->codec_link) {
  dev_err(dev, "codec link not available");
  return -EINVAL;
 }

 /* codec-specific fields for headphone codec */
 ctx->codec_link->codecs = nau8825_component;
 ctx->codec_link->num_codecs = ARRAY_SIZE(nau8825_component);
 ctx->codec_link->init = sof_nau8825_codec_init;
 ctx->codec_link->exit = sof_nau8825_codec_exit;
 ctx->codec_link->ops = &sof_nau8825_ops;

 if (ctx->amp_type == CODEC_NONE)
  return 0;

 if (!ctx->amp_link) {
  dev_err(dev, "amp link not available");
  return -EINVAL;
 }

 /* codec-specific fields for speaker amplifier */
 switch (ctx->amp_type) {
 case CODEC_MAX98360A:
  max_98360a_dai_link(ctx->amp_link);
  break;
 case CODEC_MAX98373:
  max_98373_dai_link(dev, ctx->amp_link);
  break;
 case CODEC_NAU8318:
  nau8318_set_dai_link(ctx->amp_link);
  break;
 case CODEC_RT1015P:
  sof_rt1015p_dai_link(ctx->amp_link);
  break;
 case CODEC_RT1019P:
  sof_rt1019p_dai_link(ctx->amp_link);
  break;
 default:
  dev_err(dev, "invalid amp type %d\n", ctx->amp_type);
  return -EINVAL;
 }

 return 0;
}

static int sof_audio_probe(struct platform_device *pdev)
{
 struct snd_soc_acpi_mach *mach = pdev->dev.platform_data;
 struct sof_card_private *ctx;
 int ret;

 if (pdev->id_entry && pdev->id_entry->driver_data)
  sof_nau8825_quirk = (unsigned long)pdev->id_entry->driver_data;

 dev_dbg(&pdev->dev, "sof_nau8825_quirk = %lx\n", sof_nau8825_quirk);

 /* initialize ctx with board quirk */
 ctx = sof_intel_board_get_ctx(&pdev->dev, sof_nau8825_quirk);
 if (!ctx)
  return -ENOMEM;

 if (mach->mach_params.codec_mask & IDISP_CODEC_MASK)
  ctx->hdmi.idisp_codec = true;

 /* update dai_link */
 ret = sof_card_dai_links_create(&pdev->dev, &sof_audio_card_nau8825, ctx);
 if (ret)
  return ret;

 /* update codec_conf */
 switch (ctx->amp_type) {
 case CODEC_MAX98373:
  max_98373_set_codec_conf(&sof_audio_card_nau8825);
  break;
 case CODEC_RT1015P:
  sof_rt1015p_codec_conf(&sof_audio_card_nau8825);
  break;
 case CODEC_MAX98360A:
 case CODEC_NAU8318:
 case CODEC_RT1019P:
 case CODEC_NONE:
  /* no codec conf required */
  break;
 default:
  dev_err(&pdev->dev, "invalid amp type %d\n", ctx->amp_type);
  return -EINVAL;
 }

 sof_audio_card_nau8825.dev = &pdev->dev;

 /* set platform name for each dailink */
 ret = snd_soc_fixup_dai_links_platform_name(&sof_audio_card_nau8825,
          mach->mach_params.platform);
 if (ret)
  return ret;

 snd_soc_card_set_drvdata(&sof_audio_card_nau8825, ctx);

 return devm_snd_soc_register_card(&pdev->dev,
       &sof_audio_card_nau8825);
}

static const struct platform_device_id board_ids[] = {
 {
  .name = "adl_rt1019p_8825",
  .driver_data = (kernel_ulong_t)(SOF_SSP_PORT_CODEC(0) |
     SOF_SSP_PORT_AMP(2) |
     SOF_NUM_IDISP_HDMI(4)),
 },
 {
  .name = "adl_nau8825_def",
  .driver_data = (kernel_ulong_t)(SOF_SSP_PORT_CODEC(0) |
     SOF_SSP_PORT_AMP(1) |
     SOF_NUM_IDISP_HDMI(4) |
     SOF_SSP_PORT_BT_OFFLOAD(2) |
     SOF_BT_OFFLOAD_PRESENT),
 },
 {
  .name = "rpl_nau8825_def",
  .driver_data = (kernel_ulong_t)(SOF_SSP_PORT_CODEC(0) |
     SOF_SSP_PORT_AMP(1) |
     SOF_NUM_IDISP_HDMI(4) |
     SOF_SSP_PORT_BT_OFFLOAD(2) |
     SOF_BT_OFFLOAD_PRESENT),
 },
 {
  .name = "mtl_nau8825_def",
  .driver_data = (kernel_ulong_t)(SOF_SSP_PORT_CODEC(2) |
     SOF_SSP_PORT_AMP(0) |
     SOF_SSP_PORT_BT_OFFLOAD(1) |
     SOF_BT_OFFLOAD_PRESENT),
 },
 { }
};
MODULE_DEVICE_TABLE(platform, board_ids);

static struct platform_driver sof_audio = {
 .probe = sof_audio_probe,
 .driver = {
  .name = "sof_nau8825",
  .pm = &snd_soc_pm_ops,
 },
 .id_table = board_ids,
};
module_platform_driver(sof_audio)

/* Module information */
MODULE_DESCRIPTION("SOF Audio Machine driver for NAU8825");
MODULE_AUTHOR("David Lin <ctlin0@nuvoton.com>");
MODULE_AUTHOR("Mac Chiang <mac.chiang@intel.com>");
MODULE_AUTHOR("Brent Lu <brent.lu@intel.com>");
MODULE_LICENSE("GPL");
MODULE_IMPORT_NS("SND_SOC_INTEL_SOF_BOARD_HELPERS");
MODULE_IMPORT_NS("SND_SOC_INTEL_SOF_MAXIM_COMMON");
MODULE_IMPORT_NS("SND_SOC_INTEL_SOF_NUVOTON_COMMON");
MODULE_IMPORT_NS("SND_SOC_INTEL_SOF_REALTEK_COMMON");