// SPDX-License-Identifier: GPL-2.0
/*
 *  Copyright (C) 2020 Texas Instruments Incorporated - http://www.ti.com
 *  Author: Peter Ujfalusi <peter.ujfalusi@ti.com>
 */

#include <linux/clk.h>
#include <linux/module.h>
#include <linux/of.h>
#include <linux/platform_device.h>

#include <sound/core.h>
#include <sound/pcm.h>
#include <sound/pcm_params.h>
#include <sound/soc.h>

#include "davinci-mcasp.h"

/*
 * Maximum number of configuration entries for prefixes:
 * CPB: 2 (mcasp10 + codec)
 * IVI: 3 (mcasp0 + 2x codec)
 */
#define J721E_CODEC_CONF_COUNT 5

enum j721e_audio_domain_id {
 J721E_AUDIO_DOMAIN_CPB = 0,
 J721E_AUDIO_DOMAIN_IVI,
 J721E_AUDIO_DOMAIN_LAST,
};

#define J721E_CLK_PARENT_48000 0
#define J721E_CLK_PARENT_44100 1

#define J721E_MAX_CLK_HSDIV 128
#define PCM1368A_MAX_SYSCLK 36864000

#define J721E_DAI_FMT  (SND_SOC_DAIFMT_RIGHT_J | \
     SND_SOC_DAIFMT_NB_NF |   \
     SND_SOC_DAIFMT_CBC_CFC)

enum j721e_board_type {
 J721E_BOARD_CPB = 1,
 J721E_BOARD_CPB_IVI,
};

struct j721e_audio_match_data {
 enum j721e_board_type board_type;
 int num_links;
 unsigned int pll_rates[2];
};

static unsigned int ratios_for_pcm3168a[] = {
 256,
 512,
 768,
};

struct j721e_audio_clocks {
 struct clk *target;
 struct clk *parent[2];
};

struct j721e_audio_domain {
 struct j721e_audio_clocks codec;
 struct j721e_audio_clocks mcasp;
 int parent_clk_id;

 int active;
 unsigned int active_link;
 unsigned int rate;
};

struct j721e_priv {
 struct device *dev;
 struct snd_soc_card card;
 struct snd_soc_dai_link *dai_links;
 struct snd_soc_codec_conf codec_conf[J721E_CODEC_CONF_COUNT];
 struct snd_interval rate_range;
 const struct j721e_audio_match_data *match_data;
 u32 pll_rates[2];
 unsigned int hsdiv_rates[2];

 struct j721e_audio_domain audio_domains[J721E_AUDIO_DOMAIN_LAST];

 struct mutex mutex;
};

static const struct snd_soc_dapm_widget j721e_cpb_dapm_widgets[] = {
 SND_SOC_DAPM_HP("CPB Stereo HP 1", NULL),
 SND_SOC_DAPM_HP("CPB Stereo HP 2", NULL),
 SND_SOC_DAPM_HP("CPB Stereo HP 3", NULL),
 SND_SOC_DAPM_LINE("CPB Line Out", NULL),
 SND_SOC_DAPM_MIC("CPB Stereo Mic 1", NULL),
 SND_SOC_DAPM_MIC("CPB Stereo Mic 2", NULL),
 SND_SOC_DAPM_LINE("CPB Line In", NULL),
};

static const struct snd_soc_dapm_route j721e_cpb_dapm_routes[] = {
 {"CPB Stereo HP 1", NULL, "codec-1 AOUT1L"},
 {"CPB Stereo HP 1", NULL, "codec-1 AOUT1R"},
 {"CPB Stereo HP 2", NULL, "codec-1 AOUT2L"},
 {"CPB Stereo HP 2", NULL, "codec-1 AOUT2R"},
 {"CPB Stereo HP 3", NULL, "codec-1 AOUT3L"},
 {"CPB Stereo HP 3", NULL, "codec-1 AOUT3R"},
 {"CPB Line Out", NULL, "codec-1 AOUT4L"},
 {"CPB Line Out", NULL, "codec-1 AOUT4R"},

 {"codec-1 AIN1L", NULL, "CPB Stereo Mic 1"},
 {"codec-1 AIN1R", NULL, "CPB Stereo Mic 1"},
 {"codec-1 AIN2L", NULL, "CPB Stereo Mic 2"},
 {"codec-1 AIN2R", NULL, "CPB Stereo Mic 2"},
 {"codec-1 AIN3L", NULL, "CPB Line In"},
 {"codec-1 AIN3R", NULL, "CPB Line In"},
};

static const struct snd_soc_dapm_widget j721e_ivi_codec_a_dapm_widgets[] = {
 SND_SOC_DAPM_LINE("IVI A Line Out 1", NULL),
 SND_SOC_DAPM_LINE("IVI A Line Out 2", NULL),
 SND_SOC_DAPM_LINE("IVI A Line Out 3", NULL),
 SND_SOC_DAPM_LINE("IVI A Line Out 4", NULL),
 SND_SOC_DAPM_MIC("IVI A Stereo Mic 1", NULL),
 SND_SOC_DAPM_MIC("IVI A Stereo Mic 2", NULL),
 SND_SOC_DAPM_LINE("IVI A Line In", NULL),
};

static const struct snd_soc_dapm_route j721e_codec_a_dapm_routes[] = {
 {"IVI A Line Out 1", NULL, "codec-a AOUT1L"},
 {"IVI A Line Out 1", NULL, "codec-a AOUT1R"},
 {"IVI A Line Out 2", NULL, "codec-a AOUT2L"},
 {"IVI A Line Out 2", NULL, "codec-a AOUT2R"},
 {"IVI A Line Out 3", NULL, "codec-a AOUT3L"},
 {"IVI A Line Out 3", NULL, "codec-a AOUT3R"},
 {"IVI A Line Out 4", NULL, "codec-a AOUT4L"},
 {"IVI A Line Out 4", NULL, "codec-a AOUT4R"},

 {"codec-a AIN1L", NULL, "IVI A Stereo Mic 1"},
 {"codec-a AIN1R", NULL, "IVI A Stereo Mic 1"},
 {"codec-a AIN2L", NULL, "IVI A Stereo Mic 2"},
 {"codec-a AIN2R", NULL, "IVI A Stereo Mic 2"},
 {"codec-a AIN3L", NULL, "IVI A Line In"},
 {"codec-a AIN3R", NULL, "IVI A Line In"},
};

static const struct snd_soc_dapm_widget j721e_ivi_codec_b_dapm_widgets[] = {
 SND_SOC_DAPM_LINE("IVI B Line Out 1", NULL),
 SND_SOC_DAPM_LINE("IVI B Line Out 2", NULL),
 SND_SOC_DAPM_LINE("IVI B Line Out 3", NULL),
 SND_SOC_DAPM_LINE("IVI B Line Out 4", NULL),
 SND_SOC_DAPM_MIC("IVI B Stereo Mic 1", NULL),
 SND_SOC_DAPM_MIC("IVI B Stereo Mic 2", NULL),
 SND_SOC_DAPM_LINE("IVI B Line In", NULL),
};

static const struct snd_soc_dapm_route j721e_codec_b_dapm_routes[] = {
 {"IVI B Line Out 1", NULL, "codec-b AOUT1L"},
 {"IVI B Line Out 1", NULL, "codec-b AOUT1R"},
 {"IVI B Line Out 2", NULL, "codec-b AOUT2L"},
 {"IVI B Line Out 2", NULL, "codec-b AOUT2R"},
 {"IVI B Line Out 3", NULL, "codec-b AOUT3L"},
 {"IVI B Line Out 3", NULL, "codec-b AOUT3R"},
 {"IVI B Line Out 4", NULL, "codec-b AOUT4L"},
 {"IVI B Line Out 4", NULL, "codec-b AOUT4R"},

 {"codec-b AIN1L", NULL, "IVI B Stereo Mic 1"},
 {"codec-b AIN1R", NULL, "IVI B Stereo Mic 1"},
 {"codec-b AIN2L", NULL, "IVI B Stereo Mic 2"},
 {"codec-b AIN2R", NULL, "IVI B Stereo Mic 2"},
 {"codec-b AIN3L", NULL, "IVI B Line In"},
 {"codec-b AIN3R", NULL, "IVI B Line In"},
};

static int j721e_configure_refclk(struct j721e_priv *priv,
      unsigned int audio_domain, unsigned int rate)
{
 struct j721e_audio_domain *domain = &priv->audio_domains[audio_domain];
 unsigned int scki;
 int ret = -EINVAL;
 int i, clk_id;

 if (!(rate % 8000) && priv->pll_rates[J721E_CLK_PARENT_48000])
  clk_id = J721E_CLK_PARENT_48000;
 else if (!(rate % 11025) && priv->pll_rates[J721E_CLK_PARENT_44100])
  clk_id = J721E_CLK_PARENT_44100;
 else if (!(rate % 11025) && priv->pll_rates[J721E_CLK_PARENT_48000])
  clk_id = J721E_CLK_PARENT_48000;
 else
  return ret;

 for (i = 0; i < ARRAY_SIZE(ratios_for_pcm3168a); i++) {
  scki = ratios_for_pcm3168a[i] * rate;

  if (priv->pll_rates[clk_id] / scki <= J721E_MAX_CLK_HSDIV) {
   ret = 0;
   break;
  }
 }

 if (ret) {
  dev_err(priv->dev, "No valid clock configuration for %u Hz\n",
   rate);
  return ret;
 }

 if (domain->parent_clk_id == -1 || priv->hsdiv_rates[domain->parent_clk_id] != scki) {
  dev_dbg(priv->dev,
   "domain%u configuration for %u Hz: %s, %dxFS (SCKI: %u Hz)\n",
   audio_domain, rate,
   clk_id == J721E_CLK_PARENT_48000 ? "PLL4" : "PLL15",
   ratios_for_pcm3168a[i], scki);

  if (domain->parent_clk_id != clk_id) {
   ret = clk_set_parent(domain->codec.target,
          domain->codec.parent[clk_id]);
   if (ret)
    return ret;

   ret = clk_set_parent(domain->mcasp.target,
          domain->mcasp.parent[clk_id]);
   if (ret)
    return ret;

   domain->parent_clk_id = clk_id;
  }

  ret = clk_set_rate(domain->codec.target, scki);
  if (ret) {
   dev_err(priv->dev, "codec set rate failed for %u Hz\n",
    scki);
   return ret;
  }

  ret = clk_set_rate(domain->mcasp.target, scki);
  if (!ret) {
   priv->hsdiv_rates[domain->parent_clk_id] = scki;
  } else {
   dev_err(priv->dev, "mcasp set rate failed for %u Hz\n",
    scki);
   return ret;
  }
 }

 return ret;
}

static int j721e_rule_rate(struct snd_pcm_hw_params *params,
      struct snd_pcm_hw_rule *rule)
{
 struct snd_interval *t = rule->private;

 return snd_interval_refine(hw_param_interval(params, rule->var), t);
}

static int j721e_audio_startup(struct snd_pcm_substream *substream)
{
 struct snd_soc_pcm_runtime *rtd = snd_soc_substream_to_rtd(substream);
 struct j721e_priv *priv = snd_soc_card_get_drvdata(rtd->card);
 unsigned int domain_id = rtd->dai_link->id;
 struct j721e_audio_domain *domain = &priv->audio_domains[domain_id];
 struct snd_soc_dai *cpu_dai = snd_soc_rtd_to_cpu(rtd, 0);
 struct snd_soc_dai *codec_dai;
 unsigned int active_rate;
 int ret = 0;
 int i;

 mutex_lock(&priv->mutex);

 domain->active++;

 for (i = 0; i < J721E_AUDIO_DOMAIN_LAST; i++) {
  active_rate = priv->audio_domains[i].rate;
  if (active_rate)
   break;
 }

 if (active_rate)
  ret = snd_pcm_hw_constraint_single(substream->runtime,
         SNDRV_PCM_HW_PARAM_RATE,
         active_rate);
 else
  ret = snd_pcm_hw_rule_add(substream->runtime, 0,
       SNDRV_PCM_HW_PARAM_RATE,
       j721e_rule_rate, &priv->rate_range,
       SNDRV_PCM_HW_PARAM_RATE, -1);


 if (ret)
  goto out;

 /* Reset TDM slots to 32 */
 ret = snd_soc_dai_set_tdm_slot(cpu_dai, 0x3, 0x3, 2, 32);
 if (ret && ret != -ENOTSUPP)
  goto out;

 for_each_rtd_codec_dais(rtd, i, codec_dai) {
  ret = snd_soc_dai_set_tdm_slot(codec_dai, 0x3, 0x3, 2, 32);
  if (ret && ret != -ENOTSUPP)
   goto out;
 }

 if (ret == -ENOTSUPP)
  ret = 0;
out:
 if (ret)
  domain->active--;
 mutex_unlock(&priv->mutex);

 return ret;
}

static int j721e_audio_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_card *card = rtd->card;
 struct j721e_priv *priv = snd_soc_card_get_drvdata(card);
 unsigned int domain_id = rtd->dai_link->id;
 struct j721e_audio_domain *domain = &priv->audio_domains[domain_id];
 struct snd_soc_dai *cpu_dai = snd_soc_rtd_to_cpu(rtd, 0);
 struct snd_soc_dai *codec_dai;
 unsigned int sysclk_rate;
 int slot_width = 32;
 int ret;
 int i;

 mutex_lock(&priv->mutex);

 if (domain->rate && domain->rate != params_rate(params)) {
  ret = -EINVAL;
  goto out;
 }

 if (params_width(params) == 16)
  slot_width = 16;

 ret = snd_soc_dai_set_tdm_slot(cpu_dai, 0x3, 0x3, 2, slot_width);
 if (ret && ret != -ENOTSUPP)
  goto out;

 for_each_rtd_codec_dais(rtd, i, codec_dai) {
  ret = snd_soc_dai_set_tdm_slot(codec_dai, 0x3, 0x3, 2,
            slot_width);
  if (ret && ret != -ENOTSUPP)
   goto out;
 }

 ret = j721e_configure_refclk(priv, domain_id, params_rate(params));
 if (ret)
  goto out;

 sysclk_rate = priv->hsdiv_rates[domain->parent_clk_id];
 for_each_rtd_codec_dais(rtd, i, codec_dai) {
  ret = snd_soc_dai_set_sysclk(codec_dai, 0, sysclk_rate,
          SND_SOC_CLOCK_IN);
  if (ret && ret != -ENOTSUPP) {
   dev_err(priv->dev,
    "codec set_sysclk failed for %u Hz\n",
    sysclk_rate);
   goto out;
  }
 }

 ret = snd_soc_dai_set_sysclk(cpu_dai, MCASP_CLK_HCLK_AUXCLK,
         sysclk_rate, SND_SOC_CLOCK_IN);

 if (ret && ret != -ENOTSUPP) {
  dev_err(priv->dev, "mcasp set_sysclk failed for %u Hz\n",
   sysclk_rate);
 } else {
  domain->rate = params_rate(params);
  ret = 0;
 }

out:
 mutex_unlock(&priv->mutex);
 return ret;
}

static void j721e_audio_shutdown(struct snd_pcm_substream *substream)
{
 struct snd_soc_pcm_runtime *rtd = snd_soc_substream_to_rtd(substream);
 struct j721e_priv *priv = snd_soc_card_get_drvdata(rtd->card);
 unsigned int domain_id = rtd->dai_link->id;
 struct j721e_audio_domain *domain = &priv->audio_domains[domain_id];

 mutex_lock(&priv->mutex);

 domain->active--;
 if (!domain->active) {
  domain->rate = 0;
  domain->active_link = 0;
 }

 mutex_unlock(&priv->mutex);
}

static const struct snd_soc_ops j721e_audio_ops = {
 .startup = j721e_audio_startup,
 .hw_params = j721e_audio_hw_params,
 .shutdown = j721e_audio_shutdown,
};

static int j721e_audio_init(struct snd_soc_pcm_runtime *rtd)
{
 struct j721e_priv *priv = snd_soc_card_get_drvdata(rtd->card);
 unsigned int domain_id = rtd->dai_link->id;
 struct j721e_audio_domain *domain = &priv->audio_domains[domain_id];
 struct snd_soc_dai *cpu_dai = snd_soc_rtd_to_cpu(rtd, 0);
 struct snd_soc_dai *codec_dai;
 unsigned int sysclk_rate;
 int i, ret;

 /* Set up initial clock configuration */
 ret = j721e_configure_refclk(priv, domain_id, 48000);
 if (ret)
  return ret;

 sysclk_rate = priv->hsdiv_rates[domain->parent_clk_id];
 for_each_rtd_codec_dais(rtd, i, codec_dai) {
  ret = snd_soc_dai_set_sysclk(codec_dai, 0, sysclk_rate,
          SND_SOC_CLOCK_IN);
  if (ret && ret != -ENOTSUPP)
   return ret;
 }

 ret = snd_soc_dai_set_sysclk(cpu_dai, MCASP_CLK_HCLK_AUXCLK,
         sysclk_rate, SND_SOC_CLOCK_IN);
 if (ret && ret != -ENOTSUPP)
  return ret;

 /* Set initial tdm slots */
 ret = snd_soc_dai_set_tdm_slot(cpu_dai, 0x3, 0x3, 2, 32);
 if (ret && ret != -ENOTSUPP)
  return ret;

 for_each_rtd_codec_dais(rtd, i, codec_dai) {
  ret = snd_soc_dai_set_tdm_slot(codec_dai, 0x3, 0x3, 2, 32);
  if (ret && ret != -ENOTSUPP)
   return ret;
 }

 return 0;
}

static int j721e_audio_init_ivi(struct snd_soc_pcm_runtime *rtd)
{
 struct snd_soc_dapm_context *dapm = &rtd->card->dapm;

 snd_soc_dapm_new_controls(dapm, j721e_ivi_codec_a_dapm_widgets,
      ARRAY_SIZE(j721e_ivi_codec_a_dapm_widgets));
 snd_soc_dapm_add_routes(dapm, j721e_codec_a_dapm_routes,
    ARRAY_SIZE(j721e_codec_a_dapm_routes));
 snd_soc_dapm_new_controls(dapm, j721e_ivi_codec_b_dapm_widgets,
      ARRAY_SIZE(j721e_ivi_codec_b_dapm_widgets));
 snd_soc_dapm_add_routes(dapm, j721e_codec_b_dapm_routes,
    ARRAY_SIZE(j721e_codec_b_dapm_routes));

 return j721e_audio_init(rtd);
}

static int j721e_get_clocks(struct device *dev,
       struct j721e_audio_clocks *clocks, char *prefix)
{
 struct clk *parent;
 char *clk_name;
 int ret;

 clocks->target = devm_clk_get(dev, prefix);
 if (IS_ERR(clocks->target))
  return dev_err_probe(dev, PTR_ERR(clocks->target),
         "failed to acquire %s\n", prefix);

 clk_name = kasprintf(GFP_KERNEL, "%s-48000", prefix);
 if (clk_name) {
  parent = devm_clk_get(dev, clk_name);
  kfree(clk_name);
  if (IS_ERR(parent)) {
   ret = PTR_ERR(parent);
   if (ret == -EPROBE_DEFER)
    return ret;

   dev_dbg(dev, "no 48KHz parent for %s: %d\n", prefix, ret);
   parent = NULL;
  }
  clocks->parent[J721E_CLK_PARENT_48000] = parent;
 } else {
  return -ENOMEM;
 }

 clk_name = kasprintf(GFP_KERNEL, "%s-44100", prefix);
 if (clk_name) {
  parent = devm_clk_get(dev, clk_name);
  kfree(clk_name);
  if (IS_ERR(parent)) {
   ret = PTR_ERR(parent);
   if (ret == -EPROBE_DEFER)
    return ret;

   dev_dbg(dev, "no 44.1KHz parent for %s: %d\n", prefix, ret);
   parent = NULL;
  }
  clocks->parent[J721E_CLK_PARENT_44100] = parent;
 } else {
  return -ENOMEM;
 }

 if (!clocks->parent[J721E_CLK_PARENT_44100] &&
     !clocks->parent[J721E_CLK_PARENT_48000]) {
  dev_err(dev, "At least one parent clock is needed for %s\n",
   prefix);
  return -EINVAL;
 }

 return 0;
}

static const struct j721e_audio_match_data j721e_cpb_data = {
 .board_type = J721E_BOARD_CPB,
 .num_links = 2, /* CPB pcm3168a */
 .pll_rates = {
  [J721E_CLK_PARENT_44100] = 1083801600, /* PLL15 */
  [J721E_CLK_PARENT_48000] = 1179648000, /* PLL4 */
 },
};

static const struct j721e_audio_match_data j721e_cpb_ivi_data = {
 .board_type = J721E_BOARD_CPB_IVI,
 .num_links = 4, /* CPB pcm3168a + 2x pcm3168a on IVI */
 .pll_rates = {
  [J721E_CLK_PARENT_44100] = 1083801600, /* PLL15 */
  [J721E_CLK_PARENT_48000] = 1179648000, /* PLL4 */
 },
};

static const struct j721e_audio_match_data j7200_cpb_data = {
 .board_type = J721E_BOARD_CPB,
 .num_links = 2, /* CPB pcm3168a */
 .pll_rates = {
  [J721E_CLK_PARENT_48000] = 2359296000u, /* PLL4 */
 },
};

static const struct of_device_id j721e_audio_of_match[] = {
 {
  .compatible = "ti,j721e-cpb-audio",
  .data = &j721e_cpb_data,
 }, {
  .compatible = "ti,j721e-cpb-ivi-audio",
  .data = &j721e_cpb_ivi_data,
 }, {
  .compatible = "ti,j7200-cpb-audio",
  .data = &j7200_cpb_data,
 },
 { },
};
MODULE_DEVICE_TABLE(of, j721e_audio_of_match);

static int j721e_calculate_rate_range(struct j721e_priv *priv)
{
 const struct j721e_audio_match_data *match_data = priv->match_data;
 struct j721e_audio_clocks *domain_clocks;
 unsigned int min_rate, max_rate, pll_rate;
 struct clk *pll;

 domain_clocks = &priv->audio_domains[J721E_AUDIO_DOMAIN_CPB].mcasp;

 pll = clk_get_parent(domain_clocks->parent[J721E_CLK_PARENT_44100]);
 if (IS_ERR_OR_NULL(pll)) {
  priv->pll_rates[J721E_CLK_PARENT_44100] =
    match_data->pll_rates[J721E_CLK_PARENT_44100];
 } else {
  priv->pll_rates[J721E_CLK_PARENT_44100] = clk_get_rate(pll);
  clk_put(pll);
 }

 pll = clk_get_parent(domain_clocks->parent[J721E_CLK_PARENT_48000]);
 if (IS_ERR_OR_NULL(pll)) {
  priv->pll_rates[J721E_CLK_PARENT_48000] =
    match_data->pll_rates[J721E_CLK_PARENT_48000];
 } else {
  priv->pll_rates[J721E_CLK_PARENT_48000] = clk_get_rate(pll);
  clk_put(pll);
 }

 if (!priv->pll_rates[J721E_CLK_PARENT_44100] &&
     !priv->pll_rates[J721E_CLK_PARENT_48000]) {
  dev_err(priv->dev, "At least one PLL is needed\n");
  return -EINVAL;
 }

 if (priv->pll_rates[J721E_CLK_PARENT_44100])
  pll_rate = priv->pll_rates[J721E_CLK_PARENT_44100];
 else
  pll_rate = priv->pll_rates[J721E_CLK_PARENT_48000];

 min_rate = pll_rate / J721E_MAX_CLK_HSDIV;
 min_rate /= ratios_for_pcm3168a[ARRAY_SIZE(ratios_for_pcm3168a) - 1];

 if (priv->pll_rates[J721E_CLK_PARENT_48000])
  pll_rate = priv->pll_rates[J721E_CLK_PARENT_48000];
 else
  pll_rate = priv->pll_rates[J721E_CLK_PARENT_44100];

 if (pll_rate > PCM1368A_MAX_SYSCLK)
  pll_rate = PCM1368A_MAX_SYSCLK;

 max_rate = pll_rate / ratios_for_pcm3168a[0];

 snd_interval_any(&priv->rate_range);
 priv->rate_range.min = min_rate;
 priv->rate_range.max = max_rate;

 return 0;
}

static int j721e_soc_probe_cpb(struct j721e_priv *priv, int *link_idx,
          int *conf_idx)
{
 struct device_node *node = priv->dev->of_node;
 struct snd_soc_dai_link_component *compnent;
 struct device_node *dai_node, *codec_node;
 struct j721e_audio_domain *domain;
 int comp_count, comp_idx;
 int ret;

 dai_node = of_parse_phandle(node, "ti,cpb-mcasp", 0);
 if (!dai_node) {
  dev_err(priv->dev, "CPB McASP node is not provided\n");
  return -EINVAL;
 }

 codec_node = of_parse_phandle(node, "ti,cpb-codec", 0);
 if (!codec_node) {
  dev_err(priv->dev, "CPB codec node is not provided\n");
  ret = -EINVAL;
  goto put_dai_node;
 }

 domain = &priv->audio_domains[J721E_AUDIO_DOMAIN_CPB];
 ret = j721e_get_clocks(priv->dev, &domain->codec, "cpb-codec-scki");
 if (ret)
  goto put_codec_node;

 ret = j721e_get_clocks(priv->dev, &domain->mcasp, "cpb-mcasp-auxclk");
 if (ret)
  goto put_codec_node;

 /*
  * Common Processor Board, two links
  * Link 1: McASP10 -> pcm3168a_1 DAC
  * Link 2: McASP10 <- pcm3168a_1 ADC
 */
 comp_count = 6;
 compnent = devm_kcalloc(priv->dev, comp_count, sizeof(*compnent),
    GFP_KERNEL);
 if (!compnent) {
  ret = -ENOMEM;
  goto put_codec_node;
 }

 comp_idx = 0;
 priv->dai_links[*link_idx].cpus = &compnent[comp_idx++];
 priv->dai_links[*link_idx].num_cpus = 1;
 priv->dai_links[*link_idx].codecs = &compnent[comp_idx++];
 priv->dai_links[*link_idx].num_codecs = 1;
 priv->dai_links[*link_idx].platforms = &compnent[comp_idx++];
 priv->dai_links[*link_idx].num_platforms = 1;

 priv->dai_links[*link_idx].name = "CPB PCM3168A Playback";
 priv->dai_links[*link_idx].stream_name = "CPB PCM3168A Analog";
 priv->dai_links[*link_idx].cpus->of_node = dai_node;
 priv->dai_links[*link_idx].platforms->of_node = dai_node;
 priv->dai_links[*link_idx].codecs->of_node = codec_node;
 priv->dai_links[*link_idx].codecs->dai_name = "pcm3168a-dac";
 priv->dai_links[*link_idx].playback_only = 1;
 priv->dai_links[*link_idx].id = J721E_AUDIO_DOMAIN_CPB;
 priv->dai_links[*link_idx].dai_fmt = J721E_DAI_FMT;
 priv->dai_links[*link_idx].init = j721e_audio_init;
 priv->dai_links[*link_idx].ops = &j721e_audio_ops;
 (*link_idx)++;

 priv->dai_links[*link_idx].cpus = &compnent[comp_idx++];
 priv->dai_links[*link_idx].num_cpus = 1;
 priv->dai_links[*link_idx].codecs = &compnent[comp_idx++];
 priv->dai_links[*link_idx].num_codecs = 1;
 priv->dai_links[*link_idx].platforms = &compnent[comp_idx++];
 priv->dai_links[*link_idx].num_platforms = 1;

 priv->dai_links[*link_idx].name = "CPB PCM3168A Capture";
 priv->dai_links[*link_idx].stream_name = "CPB PCM3168A Analog";
 priv->dai_links[*link_idx].cpus->of_node = dai_node;
 priv->dai_links[*link_idx].platforms->of_node = dai_node;
 priv->dai_links[*link_idx].codecs->of_node = codec_node;
 priv->dai_links[*link_idx].codecs->dai_name = "pcm3168a-adc";
 priv->dai_links[*link_idx].capture_only = 1;
 priv->dai_links[*link_idx].id = J721E_AUDIO_DOMAIN_CPB;
 priv->dai_links[*link_idx].dai_fmt = J721E_DAI_FMT;
 priv->dai_links[*link_idx].init = j721e_audio_init;
 priv->dai_links[*link_idx].ops = &j721e_audio_ops;
 (*link_idx)++;

 priv->codec_conf[*conf_idx].dlc.of_node = codec_node;
 priv->codec_conf[*conf_idx].name_prefix = "codec-1";
 (*conf_idx)++;
 priv->codec_conf[*conf_idx].dlc.of_node = dai_node;
 priv->codec_conf[*conf_idx].name_prefix = "McASP10";
 (*conf_idx)++;

 return 0;

put_codec_node:
 of_node_put(codec_node);
put_dai_node:
 of_node_put(dai_node);
 return ret;
}

static int j721e_soc_probe_ivi(struct j721e_priv *priv, int *link_idx,
          int *conf_idx)
{
 struct device_node *node = priv->dev->of_node;
 struct snd_soc_dai_link_component *compnent;
 struct device_node *dai_node, *codeca_node, *codecb_node;
 struct j721e_audio_domain *domain;
 int comp_count, comp_idx;
 int ret;

 if (priv->match_data->board_type != J721E_BOARD_CPB_IVI)
  return 0;

 dai_node = of_parse_phandle(node, "ti,ivi-mcasp", 0);
 if (!dai_node) {
  dev_err(priv->dev, "IVI McASP node is not provided\n");
  return -EINVAL;
 }

 codeca_node = of_parse_phandle(node, "ti,ivi-codec-a", 0);
 if (!codeca_node) {
  dev_err(priv->dev, "IVI codec-a node is not provided\n");
  ret = -EINVAL;
  goto put_dai_node;
 }

 codecb_node = of_parse_phandle(node, "ti,ivi-codec-b", 0);
 if (!codecb_node) {
  dev_warn(priv->dev, "IVI codec-b node is not provided\n");
  ret = 0;
  goto put_codeca_node;
 }

 domain = &priv->audio_domains[J721E_AUDIO_DOMAIN_IVI];
 ret = j721e_get_clocks(priv->dev, &domain->codec, "ivi-codec-scki");
 if (ret)
  goto put_codecb_node;

 ret = j721e_get_clocks(priv->dev, &domain->mcasp, "ivi-mcasp-auxclk");
 if (ret)
  goto put_codecb_node;

 /*
  * IVI extension, two links
  * Link 1: McASP0 -> pcm3168a_a DAC
  *    \> pcm3168a_b DAC
  * Link 2: McASP0 <- pcm3168a_a ADC
  *     \ pcm3168a_b ADC
 */
 comp_count = 8;
 compnent = devm_kcalloc(priv->dev, comp_count, sizeof(*compnent),
    GFP_KERNEL);
 if (!compnent) {
  ret = -ENOMEM;
  goto put_codecb_node;
 }

 comp_idx = 0;
 priv->dai_links[*link_idx].cpus = &compnent[comp_idx++];
 priv->dai_links[*link_idx].num_cpus = 1;
 priv->dai_links[*link_idx].platforms = &compnent[comp_idx++];
 priv->dai_links[*link_idx].num_platforms = 1;
 priv->dai_links[*link_idx].codecs = &compnent[comp_idx];
 priv->dai_links[*link_idx].num_codecs = 2;
 comp_idx += 2;

 priv->dai_links[*link_idx].name = "IVI 2xPCM3168A Playback";
 priv->dai_links[*link_idx].stream_name = "IVI 2xPCM3168A Analog";
 priv->dai_links[*link_idx].cpus->of_node = dai_node;
 priv->dai_links[*link_idx].platforms->of_node = dai_node;
 priv->dai_links[*link_idx].codecs[0].of_node = codeca_node;
 priv->dai_links[*link_idx].codecs[0].dai_name = "pcm3168a-dac";
 priv->dai_links[*link_idx].codecs[1].of_node = codecb_node;
 priv->dai_links[*link_idx].codecs[1].dai_name = "pcm3168a-dac";
 priv->dai_links[*link_idx].playback_only = 1;
 priv->dai_links[*link_idx].id = J721E_AUDIO_DOMAIN_IVI;
 priv->dai_links[*link_idx].dai_fmt = J721E_DAI_FMT;
 priv->dai_links[*link_idx].init = j721e_audio_init_ivi;
 priv->dai_links[*link_idx].ops = &j721e_audio_ops;
 (*link_idx)++;

 priv->dai_links[*link_idx].cpus = &compnent[comp_idx++];
 priv->dai_links[*link_idx].num_cpus = 1;
 priv->dai_links[*link_idx].platforms = &compnent[comp_idx++];
 priv->dai_links[*link_idx].num_platforms = 1;
 priv->dai_links[*link_idx].codecs = &compnent[comp_idx];
 priv->dai_links[*link_idx].num_codecs = 2;

 priv->dai_links[*link_idx].name = "IVI 2xPCM3168A Capture";
 priv->dai_links[*link_idx].stream_name = "IVI 2xPCM3168A Analog";
 priv->dai_links[*link_idx].cpus->of_node = dai_node;
 priv->dai_links[*link_idx].platforms->of_node = dai_node;
 priv->dai_links[*link_idx].codecs[0].of_node = codeca_node;
 priv->dai_links[*link_idx].codecs[0].dai_name = "pcm3168a-adc";
 priv->dai_links[*link_idx].codecs[1].of_node = codecb_node;
 priv->dai_links[*link_idx].codecs[1].dai_name = "pcm3168a-adc";
 priv->dai_links[*link_idx].capture_only = 1;
 priv->dai_links[*link_idx].id = J721E_AUDIO_DOMAIN_IVI;
 priv->dai_links[*link_idx].dai_fmt = J721E_DAI_FMT;
 priv->dai_links[*link_idx].init = j721e_audio_init;
 priv->dai_links[*link_idx].ops = &j721e_audio_ops;
 (*link_idx)++;

 priv->codec_conf[*conf_idx].dlc.of_node = codeca_node;
 priv->codec_conf[*conf_idx].name_prefix = "codec-a";
 (*conf_idx)++;

 priv->codec_conf[*conf_idx].dlc.of_node = codecb_node;
 priv->codec_conf[*conf_idx].name_prefix = "codec-b";
 (*conf_idx)++;

 priv->codec_conf[*conf_idx].dlc.of_node = dai_node;
 priv->codec_conf[*conf_idx].name_prefix = "McASP0";
 (*conf_idx)++;

 return 0;


put_codecb_node:
 of_node_put(codecb_node);
put_codeca_node:
 of_node_put(codeca_node);
put_dai_node:
 of_node_put(dai_node);
 return ret;
}

static int j721e_soc_probe(struct platform_device *pdev)
{
 struct device_node *node = pdev->dev.of_node;
 struct snd_soc_card *card;
 const struct of_device_id *match;
 struct j721e_priv *priv;
 int link_cnt, conf_cnt, ret, i;

 if (!node) {
  dev_err(&pdev->dev, "of node is missing.\n");
  return -ENODEV;
 }

 match = of_match_node(j721e_audio_of_match, node);
 if (!match) {
  dev_err(&pdev->dev, "No compatible match found\n");
  return -ENODEV;
 }

 priv = devm_kzalloc(&pdev->dev, sizeof(*priv), GFP_KERNEL);
 if (!priv)
  return -ENOMEM;

 priv->match_data = match->data;

 priv->dai_links = devm_kcalloc(&pdev->dev, priv->match_data->num_links,
           sizeof(*priv->dai_links), GFP_KERNEL);
 if (!priv->dai_links)
  return -ENOMEM;

 for (i = 0; i < J721E_AUDIO_DOMAIN_LAST; i++)
  priv->audio_domains[i].parent_clk_id = -1;

 priv->dev = &pdev->dev;
 card = &priv->card;
 card->dev = &pdev->dev;
 card->owner = THIS_MODULE;
 card->dapm_widgets = j721e_cpb_dapm_widgets;
 card->num_dapm_widgets = ARRAY_SIZE(j721e_cpb_dapm_widgets);
 card->dapm_routes = j721e_cpb_dapm_routes;
 card->num_dapm_routes = ARRAY_SIZE(j721e_cpb_dapm_routes);
 card->fully_routed = 1;

 if (snd_soc_of_parse_card_name(card, "model")) {
  dev_err(&pdev->dev, "Card name is not provided\n");
  return -ENODEV;
 }

 link_cnt = 0;
 conf_cnt = 0;
 ret = j721e_soc_probe_cpb(priv, &link_cnt, &conf_cnt);
 if (ret)
  return ret;

 ret = j721e_soc_probe_ivi(priv, &link_cnt, &conf_cnt);
 if (ret)
  return ret;

 card->dai_link = priv->dai_links;
 card->num_links = link_cnt;

 card->codec_conf = priv->codec_conf;
 card->num_configs = conf_cnt;

 ret = j721e_calculate_rate_range(priv);
 if (ret)
  return ret;

 snd_soc_card_set_drvdata(card, priv);

 mutex_init(&priv->mutex);
 ret = devm_snd_soc_register_card(&pdev->dev, card);
 if (ret)
  dev_err_probe(&pdev->dev, ret,
         "devm_snd_soc_register_card() failed: %d\n",
         ret);

 return ret;
}

static struct platform_driver j721e_soc_driver = {
 .driver = {
  .name = "j721e-audio",
  .pm = &snd_soc_pm_ops,
  .of_match_table = j721e_audio_of_match,
 },
 .probe = j721e_soc_probe,
};

module_platform_driver(j721e_soc_driver);

MODULE_AUTHOR("Peter Ujfalusi <peter.ujfalusi@ti.com>");
MODULE_DESCRIPTION("ASoC machine driver for j721e Common Processor Board");
MODULE_LICENSE("GPL v2");