// SPDX-License-Identifier: GPL-2.0
//
// Loongson I2S controller master mode dirver(platform device)
//
// Copyright (C) 2023-2024 Loongson Technology Corporation Limited
//
// Author: Yingkun Meng <mengyingkun@loongson.cn>
//         Binbin Zhou <zhoubinbin@loongson.cn>

#include <linux/clk.h>
#include <linux/dma-mapping.h>
#include <linux/module.h>
#include <linux/of_dma.h>
#include <linux/platform_device.h>
#include <linux/pm_runtime.h>
#include <sound/dmaengine_pcm.h>
#include <sound/pcm.h>
#include <sound/pcm_params.h>
#include <sound/soc.h>

#include "loongson_i2s.h"

#define LOONGSON_I2S_RX_DMA_OFFSET	21
#define LOONGSON_I2S_TX_DMA_OFFSET	18

#define LOONGSON_DMA0_CONF	0x0
#define LOONGSON_DMA1_CONF	0x1
#define LOONGSON_DMA2_CONF	0x2
#define LOONGSON_DMA3_CONF	0x3
#define LOONGSON_DMA4_CONF	0x4

/* periods_max = PAGE_SIZE / sizeof(struct ls_dma_chan_reg) */
static const struct snd_pcm_hardware loongson_pcm_hardware = {
	.info = SNDRV_PCM_INFO_MMAP |
		SNDRV_PCM_INFO_INTERLEAVED |
		SNDRV_PCM_INFO_MMAP_VALID |
		SNDRV_PCM_INFO_RESUME |
		SNDRV_PCM_INFO_PAUSE,
	.formats = SNDRV_PCM_FMTBIT_S16_LE |
		   SNDRV_PCM_FMTBIT_S20_3LE |
		   SNDRV_PCM_FMTBIT_S24_LE,
	.period_bytes_min = 128,
	.period_bytes_max = 128 * 1024,
	.periods_min = 1,
	.periods_max = 64,
	.buffer_bytes_max = 1024 * 1024,
};

static const struct snd_dmaengine_pcm_config loongson_dmaengine_pcm_config = {
	.pcm_hardware = &loongson_pcm_hardware,
	.prepare_slave_config = snd_dmaengine_pcm_prepare_slave_config,
	.prealloc_buffer_size = 128 * 1024,
};

static int loongson_pcm_open(struct snd_soc_component *component,
			     struct snd_pcm_substream *substream)
{
	struct snd_pcm_runtime *runtime = substream->runtime;

	if (substream->pcm->device & 1) {
		runtime->hw.info &= ~SNDRV_PCM_INFO_INTERLEAVED;
		runtime->hw.info |= SNDRV_PCM_INFO_NONINTERLEAVED;
	}

	if (substream->pcm->device & 2)
		runtime->hw.info &= ~(SNDRV_PCM_INFO_MMAP |
				      SNDRV_PCM_INFO_MMAP_VALID);
	/*
	 * For mysterious reasons (and despite what the manual says)
	 * playback samples are lost if the DMA count is not a multiple
	 * of the DMA burst size.  Let's add a rule to enforce that.
	 */
	snd_pcm_hw_constraint_step(runtime, 0,
				   SNDRV_PCM_HW_PARAM_PERIOD_BYTES, 128);
	snd_pcm_hw_constraint_step(runtime, 0,
				   SNDRV_PCM_HW_PARAM_BUFFER_BYTES, 128);
	snd_pcm_hw_constraint_integer(substream->runtime,
				      SNDRV_PCM_HW_PARAM_PERIODS);

	return 0;
}

static const struct snd_soc_component_driver loongson_i2s_component_driver = {
	.name   = LS_I2S_DRVNAME,
	.open	= loongson_pcm_open,
};

static const struct regmap_config loongson_i2s_regmap_config = {
	.reg_bits = 32,
	.reg_stride = 4,
	.val_bits = 32,
	.max_register = 0x14,
	.cache_type = REGCACHE_FLAT,
};

static int loongson_i2s_apbdma_config(struct platform_device *pdev)
{
	int val;
	void __iomem *regs;

	regs = devm_platform_ioremap_resource(pdev, 1);
	if (IS_ERR(regs))
		return PTR_ERR(regs);

	val = readl(regs);
	val |= LOONGSON_DMA2_CONF << LOONGSON_I2S_TX_DMA_OFFSET;
	val |= LOONGSON_DMA3_CONF << LOONGSON_I2S_RX_DMA_OFFSET;
	writel(val, regs);

	return 0;
}

static int loongson_i2s_plat_probe(struct platform_device *pdev)
{
	struct device *dev = &pdev->dev;
	struct loongson_i2s *i2s;
	struct resource *res;
	struct clk *i2s_clk;
	int ret;

	i2s = devm_kzalloc(dev, sizeof(*i2s), GFP_KERNEL);
	if (!i2s)
		return -ENOMEM;

	ret = loongson_i2s_apbdma_config(pdev);
	if (ret)
		return ret;

	res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
	i2s->reg_base = devm_ioremap_resource(&pdev->dev, res);
	if (IS_ERR(i2s->reg_base))
		return dev_err_probe(dev, PTR_ERR(i2s->reg_base),
				     "devm_ioremap_resource failed\n");

	i2s->regmap = devm_regmap_init_mmio(dev, i2s->reg_base,
					    &loongson_i2s_regmap_config);
	if (IS_ERR(i2s->regmap))
		return dev_err_probe(dev, PTR_ERR(i2s->regmap),
				     "devm_regmap_init_mmio failed\n");

	i2s->playback_dma_data.addr = res->start + LS_I2S_TX_DATA;
	i2s->playback_dma_data.addr_width = DMA_SLAVE_BUSWIDTH_4_BYTES;
	i2s->playback_dma_data.maxburst = 4;

	i2s->capture_dma_data.addr = res->start + LS_I2S_RX_DATA;
	i2s->capture_dma_data.addr_width = DMA_SLAVE_BUSWIDTH_4_BYTES;
	i2s->capture_dma_data.maxburst = 4;

	i2s_clk = devm_clk_get_enabled(dev, NULL);
	if (IS_ERR(i2s_clk))
		return dev_err_probe(dev, PTR_ERR(i2s_clk), "clock property invalid\n");
	i2s->clk_rate = clk_get_rate(i2s_clk);

	dma_set_mask_and_coherent(dev, DMA_BIT_MASK(64));
	dev_set_name(dev, LS_I2S_DRVNAME);
	dev_set_drvdata(dev, i2s);

	ret = devm_snd_soc_register_component(dev, &loongson_i2s_component_driver,
					      &loongson_i2s_dai, 1);
	if (ret)
		return dev_err_probe(dev, ret, "failed to register DAI\n");

	return devm_snd_dmaengine_pcm_register(dev, &loongson_dmaengine_pcm_config,
					       SND_DMAENGINE_PCM_FLAG_COMPAT);
}

static const struct of_device_id loongson_i2s_ids[] = {
	{ .compatible = "loongson,ls2k1000-i2s" },
	{ /* sentinel */ },
};
MODULE_DEVICE_TABLE(of, loongson_i2s_ids);

static struct platform_driver loongson_i2s_driver = {
	.probe = loongson_i2s_plat_probe,
	.driver = {
		.name = "loongson-i2s-plat",
		.pm = pm_sleep_ptr(&loongson_i2s_pm),
		.of_match_table = loongson_i2s_ids,
	},
};
module_platform_driver(loongson_i2s_driver);

MODULE_DESCRIPTION("Loongson I2S Master Mode ASoC Driver");
MODULE_AUTHOR("Loongson Technology Corporation Limited");
MODULE_LICENSE("GPL");