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