// SPDX-License-Identifier: (GPL-2.0 OR BSD-3-Clause)
//
// This file is provided under a dual BSD/GPLv2 license.  When using or
// redistributing this file, you may do so under either license.
//
// Copyright(c) 2018 Intel Corporation. All rights reserved.
//
// Authors: Keyon Jie <yang.jie@linux.intel.com>
//

#include <sound/pcm_params.h>
#include <sound/hdaudio_ext.h>
#include "../sof-priv.h"
#include "hda.h"

#if IS_ENABLED(CONFIG_SND_SOC_SOF_HDA)

struct hda_pipe_params {
	u8 host_dma_id;
	u8 link_dma_id;
	u32 ch;
	u32 s_freq;
	u32 s_fmt;
	u8 linktype;
	snd_pcm_format_t format;
	int link_index;
	int stream;
	unsigned int host_bps;
	unsigned int link_bps;
};

/*
 * Unlike GP dma, there is a set of stream registers in hda controller
 * to control the link dma channels. Each register controls one link
 * dma channel and the relation is fixed. To make sure FW uses correct
 * link dma channels, host allocates stream registers and sends the
 * corresponding link dma channels to FW to allocate link dma channel
 *
 * FIXME: this API is abused in the sense that tx_num and rx_num are
 * passed as arguments, not returned. We need to find a better way to
 * retrieve the stream tag allocated for the link DMA
 */
static int hda_link_dma_get_channels(struct snd_soc_dai *dai,
				     unsigned int *tx_num,
				     unsigned int *tx_slot,
				     unsigned int *rx_num,
				     unsigned int *rx_slot)
{
	struct hdac_bus *bus;
	struct hdac_ext_stream *stream;
	struct snd_pcm_substream substream;
	struct snd_sof_dev *sdev =
		snd_soc_component_get_drvdata(dai->component);

	bus = sof_to_bus(sdev);

	memset(&substream, 0, sizeof(substream));
	if (*tx_num == 1) {
		substream.stream = SNDRV_PCM_STREAM_PLAYBACK;
		stream = snd_hdac_ext_stream_assign(bus, &substream,
						    HDAC_EXT_STREAM_TYPE_LINK);
		if (!stream) {
			dev_err(bus->dev, "error: failed to find a free hda ext stream for playback");
			return -EBUSY;
		}

		snd_soc_dai_set_dma_data(dai, &substream, stream);
		*tx_slot = hdac_stream(stream)->stream_tag - 1;

		dev_dbg(bus->dev, "link dma channel %d for playback", *tx_slot);
	}

	if (*rx_num == 1) {
		substream.stream = SNDRV_PCM_STREAM_CAPTURE;
		stream = snd_hdac_ext_stream_assign(bus, &substream,
						    HDAC_EXT_STREAM_TYPE_LINK);
		if (!stream) {
			dev_err(bus->dev, "error: failed to find a free hda ext stream for capture");
			return -EBUSY;
		}

		snd_soc_dai_set_dma_data(dai, &substream, stream);
		*rx_slot = hdac_stream(stream)->stream_tag - 1;

		dev_dbg(bus->dev, "link dma channel %d for capture", *rx_slot);
	}

	return 0;
}

static int hda_link_dma_params(struct hdac_ext_stream *stream,
			       struct hda_pipe_params *params)
{
	struct hdac_stream *hstream = &stream->hstream;
	unsigned char stream_tag = hstream->stream_tag;
	struct hdac_bus *bus = hstream->bus;
	struct hdac_ext_link *link;
	unsigned int format_val;

	snd_hdac_ext_stream_decouple(bus, stream, true);
	snd_hdac_ext_link_stream_reset(stream);

	format_val = snd_hdac_calc_stream_format(params->s_freq, params->ch,
						 params->format,
						 params->link_bps, 0);

	dev_dbg(bus->dev, "format_val=%d, rate=%d, ch=%d, format=%d\n",
		format_val, params->s_freq, params->ch, params->format);

	snd_hdac_ext_link_stream_setup(stream, format_val);

	if (stream->hstream.direction == SNDRV_PCM_STREAM_PLAYBACK) {
		list_for_each_entry(link, &bus->hlink_list, list) {
			if (link->index == params->link_index)
				snd_hdac_ext_link_set_stream_id(link,
								stream_tag);
		}
	}

	stream->link_prepared = 1;

	return 0;
}

static int hda_link_hw_params(struct snd_pcm_substream *substream,
			      struct snd_pcm_hw_params *params,
			      struct snd_soc_dai *dai)
{
	struct hdac_stream *hstream = substream->runtime->private_data;
	struct hdac_bus *bus = hstream->bus;
	struct hdac_ext_stream *link_dev;
	struct snd_soc_pcm_runtime *rtd = snd_pcm_substream_chip(substream);
	struct snd_soc_dai *codec_dai = rtd->codec_dai;
	struct sof_intel_hda_stream *hda_stream;
	struct hda_pipe_params p_params = {0};
	struct hdac_ext_link *link;
	int stream_tag;

	link_dev = snd_soc_dai_get_dma_data(dai, substream);

	hda_stream = container_of(link_dev, struct sof_intel_hda_stream,
				  hda_stream);
	hda_stream->hw_params_upon_resume = 0;

	link = snd_hdac_ext_bus_get_link(bus, codec_dai->component->name);
	if (!link)
		return -EINVAL;

	stream_tag = hdac_stream(link_dev)->stream_tag;

	/* set the stream tag in the codec dai dma params  */
	if (substream->stream == SNDRV_PCM_STREAM_PLAYBACK)
		snd_soc_dai_set_tdm_slot(codec_dai, stream_tag, 0, 0, 0);
	else
		snd_soc_dai_set_tdm_slot(codec_dai, 0, stream_tag, 0, 0);

	p_params.s_fmt = snd_pcm_format_width(params_format(params));
	p_params.ch = params_channels(params);
	p_params.s_freq = params_rate(params);
	p_params.stream = substream->stream;
	p_params.link_dma_id = stream_tag - 1;
	p_params.link_index = link->index;
	p_params.format = params_format(params);

	if (substream->stream == SNDRV_PCM_STREAM_PLAYBACK)
		p_params.link_bps = codec_dai->driver->playback.sig_bits;
	else
		p_params.link_bps = codec_dai->driver->capture.sig_bits;

	return hda_link_dma_params(link_dev, &p_params);
}

static int hda_link_pcm_prepare(struct snd_pcm_substream *substream,
				struct snd_soc_dai *dai)
{
	struct hdac_ext_stream *link_dev =
				snd_soc_dai_get_dma_data(dai, substream);
	struct sof_intel_hda_stream *hda_stream;
	struct snd_sof_dev *sdev =
				snd_soc_component_get_drvdata(dai->component);
	struct snd_soc_pcm_runtime *rtd = snd_pcm_substream_chip(substream);
	int stream = substream->stream;

	hda_stream = container_of(link_dev, struct sof_intel_hda_stream,
				  hda_stream);

	/* setup hw_params again only if resuming from system suspend */
	if (!hda_stream->hw_params_upon_resume)
		return 0;

	dev_dbg(sdev->dev, "hda: prepare stream dir %d\n", substream->stream);

	return hda_link_hw_params(substream, &rtd->dpcm[stream].hw_params,
				  dai);
}

static int hda_link_pcm_trigger(struct snd_pcm_substream *substream,
				int cmd, struct snd_soc_dai *dai)
{
	struct hdac_ext_stream *link_dev =
				snd_soc_dai_get_dma_data(dai, substream);
	int ret;

	dev_dbg(dai->dev, "In %s cmd=%d\n", __func__, cmd);
	switch (cmd) {
	case SNDRV_PCM_TRIGGER_RESUME:
		/* set up hw_params */
		ret = hda_link_pcm_prepare(substream, dai);
		if (ret < 0) {
			dev_err(dai->dev,
				"error: setting up hw_params during resume\n");
			return ret;
		}

		/* fallthrough */
	case SNDRV_PCM_TRIGGER_START:
	case SNDRV_PCM_TRIGGER_PAUSE_RELEASE:
		snd_hdac_ext_link_stream_start(link_dev);
		break;
	case SNDRV_PCM_TRIGGER_PAUSE_PUSH:
	case SNDRV_PCM_TRIGGER_SUSPEND:
	case SNDRV_PCM_TRIGGER_STOP:
		snd_hdac_ext_link_stream_clear(link_dev);
		break;
	default:
		return -EINVAL;
	}
	return 0;
}

/*
 * FIXME: This API is also abused since it's used for two purposes.
 * when the substream argument is NULL this function is used for cleanups
 * that aren't necessarily required, and called explicitly by handling
 * ASoC core structures, which is not recommended.
 * This part will be reworked in follow-up patches.
 */
static int hda_link_hw_free(struct snd_pcm_substream *substream,
			    struct snd_soc_dai *dai)
{
	const char *name;
	unsigned int stream_tag;
	struct hdac_bus *bus;
	struct hdac_ext_link *link;
	struct hdac_stream *hstream;
	struct hdac_ext_stream *stream;
	struct snd_soc_pcm_runtime *rtd;
	struct hdac_ext_stream *link_dev;
	struct snd_pcm_substream pcm_substream;

	memset(&pcm_substream, 0, sizeof(pcm_substream));
	if (substream) {
		hstream = substream->runtime->private_data;
		bus = hstream->bus;
		rtd = snd_pcm_substream_chip(substream);
		link_dev = snd_soc_dai_get_dma_data(dai, substream);
		snd_hdac_ext_stream_decouple(bus, link_dev, false);
		name = rtd->codec_dai->component->name;
		link = snd_hdac_ext_bus_get_link(bus, name);
		if (!link)
			return -EINVAL;

		if (substream->stream == SNDRV_PCM_STREAM_PLAYBACK) {
			stream_tag = hdac_stream(link_dev)->stream_tag;
			snd_hdac_ext_link_clear_stream_id(link, stream_tag);
		}

		link_dev->link_prepared = 0;
	} else {
		/* release all hda streams when dai link is unloaded */
		pcm_substream.stream = SNDRV_PCM_STREAM_PLAYBACK;
		stream = snd_soc_dai_get_dma_data(dai, &pcm_substream);
		if (stream) {
			snd_soc_dai_set_dma_data(dai, &pcm_substream, NULL);
			snd_hdac_ext_stream_release(stream,
						    HDAC_EXT_STREAM_TYPE_LINK);
		}

		pcm_substream.stream = SNDRV_PCM_STREAM_CAPTURE;
		stream = snd_soc_dai_get_dma_data(dai, &pcm_substream);
		if (stream) {
			snd_soc_dai_set_dma_data(dai, &pcm_substream, NULL);
			snd_hdac_ext_stream_release(stream,
						    HDAC_EXT_STREAM_TYPE_LINK);
		}
	}

	return 0;
}

static const struct snd_soc_dai_ops hda_link_dai_ops = {
	.hw_params = hda_link_hw_params,
	.hw_free = hda_link_hw_free,
	.trigger = hda_link_pcm_trigger,
	.prepare = hda_link_pcm_prepare,
	.get_channel_map = hda_link_dma_get_channels,
};
#endif

/*
 * common dai driver for skl+ platforms.
 * some products who use this DAI array only physically have a subset of
 * the DAIs, but no harm is done here by adding the whole set.
 */
struct snd_soc_dai_driver skl_dai[] = {
{
	.name = "SSP0 Pin",
},
{
	.name = "SSP1 Pin",
},
{
	.name = "SSP2 Pin",
},
{
	.name = "SSP3 Pin",
},
{
	.name = "SSP4 Pin",
},
{
	.name = "SSP5 Pin",
},
{
	.name = "DMIC01 Pin",
},
{
	.name = "DMIC16k Pin",
},
#if IS_ENABLED(CONFIG_SND_SOC_SOF_HDA)
{
	.name = "iDisp1 Pin",
	.ops = &hda_link_dai_ops,
},
{
	.name = "iDisp2 Pin",
	.ops = &hda_link_dai_ops,
},
{
	.name = "iDisp3 Pin",
	.ops = &hda_link_dai_ops,
},
{
	.name = "Analog CPU DAI",
	.ops = &hda_link_dai_ops,
},
{
	.name = "Digital CPU DAI",
	.ops = &hda_link_dai_ops,
},
{
	.name = "Alt Analog CPU DAI",
	.ops = &hda_link_dai_ops,
},
#endif
};