sound/soc/jz4740/jz4740-i2s.c - linux - Git at Google

 // SPDX-License-Identifier: GPL-2.0-or-later
 /*
  *  Copyright (C) 2010, Lars-Peter Clausen <lars@metafoo.de>
  */

 #include <linux/init.h>
 #include <linux/io.h>
 #include <linux/kernel.h>
 #include <linux/module.h>
 #include <linux/mod_devicetable.h>
 #include <linux/platform_device.h>
 #include <linux/slab.h>

 #include <linux/clk.h>
 #include <linux/delay.h>

 #include <linux/dma-mapping.h>

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

 #include "jz4740-i2s.h"

 #define JZ_REG_AIC_CONF		0x00
 #define JZ_REG_AIC_CTRL		0x04
 #define JZ_REG_AIC_I2S_FMT	0x10
 #define JZ_REG_AIC_FIFO_STATUS	0x14
 #define JZ_REG_AIC_I2S_STATUS	0x1c
 #define JZ_REG_AIC_CLK_DIV	0x30
 #define JZ_REG_AIC_FIFO		0x34

 #define JZ_AIC_CONF_FIFO_RX_THRESHOLD_MASK (0xf << 12)
 #define JZ_AIC_CONF_FIFO_TX_THRESHOLD_MASK (0xf <<  8)
 #define JZ_AIC_CONF_OVERFLOW_PLAY_LAST BIT(6)
 #define JZ_AIC_CONF_INTERNAL_CODEC BIT(5)
 #define JZ_AIC_CONF_I2S BIT(4)
 #define JZ_AIC_CONF_RESET BIT(3)
 #define JZ_AIC_CONF_BIT_CLK_MASTER BIT(2)
 #define JZ_AIC_CONF_SYNC_CLK_MASTER BIT(1)
 #define JZ_AIC_CONF_ENABLE BIT(0)

 #define JZ_AIC_CONF_FIFO_RX_THRESHOLD_OFFSET 12
 #define JZ_AIC_CONF_FIFO_TX_THRESHOLD_OFFSET 8
 #define JZ4760_AIC_CONF_FIFO_RX_THRESHOLD_OFFSET 24
 #define JZ4760_AIC_CONF_FIFO_TX_THRESHOLD_OFFSET 16

 #define JZ_AIC_CTRL_OUTPUT_SAMPLE_SIZE_MASK (0x7 << 19)
 #define JZ_AIC_CTRL_INPUT_SAMPLE_SIZE_MASK (0x7 << 16)
 #define JZ_AIC_CTRL_ENABLE_RX_DMA BIT(15)
 #define JZ_AIC_CTRL_ENABLE_TX_DMA BIT(14)
 #define JZ_AIC_CTRL_MONO_TO_STEREO BIT(11)
 #define JZ_AIC_CTRL_SWITCH_ENDIANNESS BIT(10)
 #define JZ_AIC_CTRL_SIGNED_TO_UNSIGNED BIT(9)
 #define JZ_AIC_CTRL_TFLUSH		BIT(8)
 #define JZ_AIC_CTRL_RFLUSH		BIT(7)
 #define JZ_AIC_CTRL_ENABLE_ROR_INT BIT(6)
 #define JZ_AIC_CTRL_ENABLE_TUR_INT BIT(5)
 #define JZ_AIC_CTRL_ENABLE_RFS_INT BIT(4)
 #define JZ_AIC_CTRL_ENABLE_TFS_INT BIT(3)
 #define JZ_AIC_CTRL_ENABLE_LOOPBACK BIT(2)
 #define JZ_AIC_CTRL_ENABLE_PLAYBACK BIT(1)
 #define JZ_AIC_CTRL_ENABLE_CAPTURE BIT(0)

 #define JZ_AIC_CTRL_OUTPUT_SAMPLE_SIZE_OFFSET 19
 #define JZ_AIC_CTRL_INPUT_SAMPLE_SIZE_OFFSET  16

 #define JZ_AIC_I2S_FMT_DISABLE_BIT_CLK BIT(12)
 #define JZ_AIC_I2S_FMT_DISABLE_BIT_ICLK BIT(13)
 #define JZ_AIC_I2S_FMT_ENABLE_SYS_CLK BIT(4)
 #define JZ_AIC_I2S_FMT_MSB BIT(0)

 #define JZ_AIC_I2S_STATUS_BUSY BIT(2)

 #define JZ_AIC_CLK_DIV_MASK 0xf
 #define I2SDIV_DV_SHIFT 0
 #define I2SDIV_DV_MASK (0xf << I2SDIV_DV_SHIFT)
 #define I2SDIV_IDV_SHIFT 8
 #define I2SDIV_IDV_MASK (0xf << I2SDIV_IDV_SHIFT)

 enum jz47xx_i2s_version {
 	JZ_I2S_JZ4740,
 	JZ_I2S_JZ4760,
 	JZ_I2S_JZ4770,
 	JZ_I2S_JZ4780,
 };

 struct i2s_soc_info {
 	enum jz47xx_i2s_version version;
 	struct snd_soc_dai_driver *dai;

 	bool shared_fifo_flush;
 };

 struct jz4740_i2s {
 	void __iomem *base;

 	struct clk *clk_aic;
 	struct clk *clk_i2s;

 	struct snd_dmaengine_dai_dma_data playback_dma_data;
 	struct snd_dmaengine_dai_dma_data capture_dma_data;

 	const struct i2s_soc_info *soc_info;
 };

 static inline uint32_t jz4740_i2s_read(const struct jz4740_i2s *i2s,
 	unsigned int reg)
 {
 	return readl(i2s->base + reg);
 }

 static inline void jz4740_i2s_write(const struct jz4740_i2s *i2s,
 	unsigned int reg, uint32_t value)
 {
 	writel(value, i2s->base + reg);
 }

 static inline void jz4740_i2s_set_bits(const struct jz4740_i2s *i2s,
 	unsigned int reg, uint32_t bits)
 {
 	uint32_t value = jz4740_i2s_read(i2s, reg);
 	value |= bits;
 	jz4740_i2s_write(i2s, reg, value);
 }

 static int jz4740_i2s_startup(struct snd_pcm_substream *substream,
 	struct snd_soc_dai *dai)
 {
 	struct jz4740_i2s *i2s = snd_soc_dai_get_drvdata(dai);
 	uint32_t conf;
 	int ret;

 	/*
 	 * When we can flush FIFOs independently, only flush the FIFO
 	 * that is starting up. We can do this when the DAI is active
 	 * because it does not disturb other active substreams.
 	 */
 	if (!i2s->soc_info->shared_fifo_flush) {
 		if (substream->stream == SNDRV_PCM_STREAM_PLAYBACK)
 			jz4740_i2s_set_bits(i2s, JZ_REG_AIC_CTRL, JZ_AIC_CTRL_TFLUSH);
 		else
 			jz4740_i2s_set_bits(i2s, JZ_REG_AIC_CTRL, JZ_AIC_CTRL_RFLUSH);
 	}

 	if (snd_soc_dai_active(dai))
 		return 0;

 	/*
 	 * When there is a shared flush bit for both FIFOs, the TFLUSH
 	 * bit flushes both FIFOs. Flushing while the DAI is active would
 	 * cause FIFO underruns in other active substreams so we have to
 	 * guard this behind the snd_soc_dai_active() check.
 	 */
 	if (i2s->soc_info->shared_fifo_flush)
 		jz4740_i2s_set_bits(i2s, JZ_REG_AIC_CTRL, JZ_AIC_CTRL_TFLUSH);

 	ret = clk_prepare_enable(i2s->clk_i2s);
 	if (ret)
 		return ret;

 	conf = jz4740_i2s_read(i2s, JZ_REG_AIC_CONF);
 	conf |= JZ_AIC_CONF_ENABLE;
 	jz4740_i2s_write(i2s, JZ_REG_AIC_CONF, conf);

 	return 0;
 }

 static void jz4740_i2s_shutdown(struct snd_pcm_substream *substream,
 	struct snd_soc_dai *dai)
 {
 	struct jz4740_i2s *i2s = snd_soc_dai_get_drvdata(dai);
 	uint32_t conf;

 	if (snd_soc_dai_active(dai))
 		return;

 	conf = jz4740_i2s_read(i2s, JZ_REG_AIC_CONF);
 	conf &= ~JZ_AIC_CONF_ENABLE;
 	jz4740_i2s_write(i2s, JZ_REG_AIC_CONF, conf);

 	clk_disable_unprepare(i2s->clk_i2s);
 }

 static int jz4740_i2s_trigger(struct snd_pcm_substream *substream, int cmd,
 	struct snd_soc_dai *dai)
 {
 	struct jz4740_i2s *i2s = snd_soc_dai_get_drvdata(dai);

 	uint32_t ctrl;
 	uint32_t mask;

 	if (substream->stream == SNDRV_PCM_STREAM_PLAYBACK)
 		mask = JZ_AIC_CTRL_ENABLE_PLAYBACK | JZ_AIC_CTRL_ENABLE_TX_DMA;
 	else
 		mask = JZ_AIC_CTRL_ENABLE_CAPTURE | JZ_AIC_CTRL_ENABLE_RX_DMA;

 	ctrl = jz4740_i2s_read(i2s, JZ_REG_AIC_CTRL);

 	switch (cmd) {
 	case SNDRV_PCM_TRIGGER_START:
 	case SNDRV_PCM_TRIGGER_RESUME:
 	case SNDRV_PCM_TRIGGER_PAUSE_RELEASE:
 		ctrl |= mask;
 		break;
 	case SNDRV_PCM_TRIGGER_STOP:
 	case SNDRV_PCM_TRIGGER_SUSPEND:
 	case SNDRV_PCM_TRIGGER_PAUSE_PUSH:
 		ctrl &= ~mask;
 		break;
 	default:
 		return -EINVAL;
 	}

 	jz4740_i2s_write(i2s, JZ_REG_AIC_CTRL, ctrl);

 	return 0;
 }

 static int jz4740_i2s_set_fmt(struct snd_soc_dai *dai, unsigned int fmt)
 {
 	struct jz4740_i2s *i2s = snd_soc_dai_get_drvdata(dai);

 	uint32_t format = 0;
 	uint32_t conf;

 	conf = jz4740_i2s_read(i2s, JZ_REG_AIC_CONF);

 	conf &= ~(JZ_AIC_CONF_BIT_CLK_MASTER | JZ_AIC_CONF_SYNC_CLK_MASTER);

 	switch (fmt & SND_SOC_DAIFMT_CLOCK_PROVIDER_MASK) {
 	case SND_SOC_DAIFMT_BP_FP:
 		conf |= JZ_AIC_CONF_BIT_CLK_MASTER | JZ_AIC_CONF_SYNC_CLK_MASTER;
 		format |= JZ_AIC_I2S_FMT_ENABLE_SYS_CLK;
 		break;
 	case SND_SOC_DAIFMT_BC_FP:
 		conf |= JZ_AIC_CONF_SYNC_CLK_MASTER;
 		break;
 	case SND_SOC_DAIFMT_BP_FC:
 		conf |= JZ_AIC_CONF_BIT_CLK_MASTER;
 		break;
 	case SND_SOC_DAIFMT_BC_FC:
 		break;
 	default:
 		return -EINVAL;
 	}

 	switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
 	case SND_SOC_DAIFMT_MSB:
 		format |= JZ_AIC_I2S_FMT_MSB;
 		break;
 	case SND_SOC_DAIFMT_I2S:
 		break;
 	default:
 		return -EINVAL;
 	}

 	switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
 	case SND_SOC_DAIFMT_NB_NF:
 		break;
 	default:
 		return -EINVAL;
 	}

 	jz4740_i2s_write(i2s, JZ_REG_AIC_CONF, conf);
 	jz4740_i2s_write(i2s, JZ_REG_AIC_I2S_FMT, format);

 	return 0;
 }

 static int jz4740_i2s_hw_params(struct snd_pcm_substream *substream,
 	struct snd_pcm_hw_params *params, struct snd_soc_dai *dai)
 {
 	struct jz4740_i2s *i2s = snd_soc_dai_get_drvdata(dai);
 	unsigned int sample_size;
 	uint32_t ctrl, div_reg;
 	int div;

 	ctrl = jz4740_i2s_read(i2s, JZ_REG_AIC_CTRL);

 	div_reg = jz4740_i2s_read(i2s, JZ_REG_AIC_CLK_DIV);
 	div = clk_get_rate(i2s->clk_i2s) / (64 * params_rate(params));

 	switch (params_format(params)) {
 	case SNDRV_PCM_FORMAT_S8:
 		sample_size = 0;
 		break;
 	case SNDRV_PCM_FORMAT_S16:
 		sample_size = 1;
 		break;
 	default:
 		return -EINVAL;
 	}

 	if (substream->stream == SNDRV_PCM_STREAM_PLAYBACK) {
 		ctrl &= ~JZ_AIC_CTRL_OUTPUT_SAMPLE_SIZE_MASK;
 		ctrl |= sample_size << JZ_AIC_CTRL_OUTPUT_SAMPLE_SIZE_OFFSET;
 		if (params_channels(params) == 1)
 			ctrl |= JZ_AIC_CTRL_MONO_TO_STEREO;
 		else
 			ctrl &= ~JZ_AIC_CTRL_MONO_TO_STEREO;

 		div_reg &= ~I2SDIV_DV_MASK;
 		div_reg |= (div - 1) << I2SDIV_DV_SHIFT;
 	} else {
 		ctrl &= ~JZ_AIC_CTRL_INPUT_SAMPLE_SIZE_MASK;
 		ctrl |= sample_size << JZ_AIC_CTRL_INPUT_SAMPLE_SIZE_OFFSET;

 		if (i2s->soc_info->version >= JZ_I2S_JZ4770) {
 			div_reg &= ~I2SDIV_IDV_MASK;
 			div_reg |= (div - 1) << I2SDIV_IDV_SHIFT;
 		} else {
 			div_reg &= ~I2SDIV_DV_MASK;
 			div_reg |= (div - 1) << I2SDIV_DV_SHIFT;
 		}
 	}

 	jz4740_i2s_write(i2s, JZ_REG_AIC_CTRL, ctrl);
 	jz4740_i2s_write(i2s, JZ_REG_AIC_CLK_DIV, div_reg);

 	return 0;
 }

 static int jz4740_i2s_set_sysclk(struct snd_soc_dai *dai, int clk_id,
 	unsigned int freq, int dir)
 {
 	struct jz4740_i2s *i2s = snd_soc_dai_get_drvdata(dai);
 	struct clk *parent;
 	int ret = 0;

 	switch (clk_id) {
 	case JZ4740_I2S_CLKSRC_EXT:
 		parent = clk_get(NULL, "ext");
 		if (IS_ERR(parent))
 			return PTR_ERR(parent);
 		clk_set_parent(i2s->clk_i2s, parent);
 		break;
 	case JZ4740_I2S_CLKSRC_PLL:
 		parent = clk_get(NULL, "pll half");
 		if (IS_ERR(parent))
 			return PTR_ERR(parent);
 		clk_set_parent(i2s->clk_i2s, parent);
 		ret = clk_set_rate(i2s->clk_i2s, freq);
 		break;
 	default:
 		return -EINVAL;
 	}
 	clk_put(parent);

 	return ret;
 }

 static int jz4740_i2s_suspend(struct snd_soc_component *component)
 {
 	struct jz4740_i2s *i2s = snd_soc_component_get_drvdata(component);
 	uint32_t conf;

 	if (snd_soc_component_active(component)) {
 		conf = jz4740_i2s_read(i2s, JZ_REG_AIC_CONF);
 		conf &= ~JZ_AIC_CONF_ENABLE;
 		jz4740_i2s_write(i2s, JZ_REG_AIC_CONF, conf);

 		clk_disable_unprepare(i2s->clk_i2s);
 	}

 	clk_disable_unprepare(i2s->clk_aic);

 	return 0;
 }

 static int jz4740_i2s_resume(struct snd_soc_component *component)
 {
 	struct jz4740_i2s *i2s = snd_soc_component_get_drvdata(component);
 	uint32_t conf;
 	int ret;

 	ret = clk_prepare_enable(i2s->clk_aic);
 	if (ret)
 		return ret;

 	if (snd_soc_component_active(component)) {
 		ret = clk_prepare_enable(i2s->clk_i2s);
 		if (ret) {
 			clk_disable_unprepare(i2s->clk_aic);
 			return ret;
 		}

 		conf = jz4740_i2s_read(i2s, JZ_REG_AIC_CONF);
 		conf |= JZ_AIC_CONF_ENABLE;
 		jz4740_i2s_write(i2s, JZ_REG_AIC_CONF, conf);
 	}

 	return 0;
 }

 static int jz4740_i2s_dai_probe(struct snd_soc_dai *dai)
 {
 	struct jz4740_i2s *i2s = snd_soc_dai_get_drvdata(dai);
 	uint32_t conf;
 	int ret;

 	ret = clk_prepare_enable(i2s->clk_aic);
 	if (ret)
 		return ret;

 	snd_soc_dai_init_dma_data(dai, &i2s->playback_dma_data,
 		&i2s->capture_dma_data);

 	if (i2s->soc_info->version >= JZ_I2S_JZ4760) {
 		conf = (7 << JZ4760_AIC_CONF_FIFO_RX_THRESHOLD_OFFSET) |
 			(8 << JZ4760_AIC_CONF_FIFO_TX_THRESHOLD_OFFSET) |
 			JZ_AIC_CONF_OVERFLOW_PLAY_LAST |
 			JZ_AIC_CONF_I2S |
 			JZ_AIC_CONF_INTERNAL_CODEC;
 	} else {
 		conf = (7 << JZ_AIC_CONF_FIFO_RX_THRESHOLD_OFFSET) |
 			(8 << JZ_AIC_CONF_FIFO_TX_THRESHOLD_OFFSET) |
 			JZ_AIC_CONF_OVERFLOW_PLAY_LAST |
 			JZ_AIC_CONF_I2S |
 			JZ_AIC_CONF_INTERNAL_CODEC;
 	}

 	jz4740_i2s_write(i2s, JZ_REG_AIC_CONF, JZ_AIC_CONF_RESET);
 	jz4740_i2s_write(i2s, JZ_REG_AIC_CONF, conf);

 	return 0;
 }

 static int jz4740_i2s_dai_remove(struct snd_soc_dai *dai)
 {
 	struct jz4740_i2s *i2s = snd_soc_dai_get_drvdata(dai);

 	clk_disable_unprepare(i2s->clk_aic);
 	return 0;
 }

 static const struct snd_soc_dai_ops jz4740_i2s_dai_ops = {
 	.startup = jz4740_i2s_startup,
 	.shutdown = jz4740_i2s_shutdown,
 	.trigger = jz4740_i2s_trigger,
 	.hw_params = jz4740_i2s_hw_params,
 	.set_fmt = jz4740_i2s_set_fmt,
 	.set_sysclk = jz4740_i2s_set_sysclk,
 };

 #define JZ4740_I2S_FMTS (SNDRV_PCM_FMTBIT_S8 | \
 		SNDRV_PCM_FMTBIT_S16_LE)

 static struct snd_soc_dai_driver jz4740_i2s_dai = {
 	.probe = jz4740_i2s_dai_probe,
 	.remove = jz4740_i2s_dai_remove,
 	.playback = {
 		.channels_min = 1,
 		.channels_max = 2,
 		.rates = SNDRV_PCM_RATE_8000_48000,
 		.formats = JZ4740_I2S_FMTS,
 	},
 	.capture = {
 		.channels_min = 2,
 		.channels_max = 2,
 		.rates = SNDRV_PCM_RATE_8000_48000,
 		.formats = JZ4740_I2S_FMTS,
 	},
 	.symmetric_rate = 1,
 	.ops = &jz4740_i2s_dai_ops,
 };

 static const struct i2s_soc_info jz4740_i2s_soc_info = {
 	.version = JZ_I2S_JZ4740,
 	.dai = &jz4740_i2s_dai,
 	.shared_fifo_flush = true,
 };

 static const struct i2s_soc_info jz4760_i2s_soc_info = {
 	.version = JZ_I2S_JZ4760,
 	.dai = &jz4740_i2s_dai,
 };

 static struct snd_soc_dai_driver jz4770_i2s_dai = {
 	.probe = jz4740_i2s_dai_probe,
 	.remove = jz4740_i2s_dai_remove,
 	.playback = {
 		.channels_min = 1,
 		.channels_max = 2,
 		.rates = SNDRV_PCM_RATE_8000_48000,
 		.formats = JZ4740_I2S_FMTS,
 	},
 	.capture = {
 		.channels_min = 2,
 		.channels_max = 2,
 		.rates = SNDRV_PCM_RATE_8000_48000,
 		.formats = JZ4740_I2S_FMTS,
 	},
 	.ops = &jz4740_i2s_dai_ops,
 };

 static const struct i2s_soc_info jz4770_i2s_soc_info = {
 	.version = JZ_I2S_JZ4770,
 	.dai = &jz4770_i2s_dai,
 };

 static const struct i2s_soc_info jz4780_i2s_soc_info = {
 	.version = JZ_I2S_JZ4780,
 	.dai = &jz4770_i2s_dai,
 };

 static const struct snd_soc_component_driver jz4740_i2s_component = {
 	.name			= "jz4740-i2s",
 	.suspend		= jz4740_i2s_suspend,
 	.resume			= jz4740_i2s_resume,
 	.legacy_dai_naming	= 1,
 };

 static const struct of_device_id jz4740_of_matches[] = {
 	{ .compatible = "ingenic,jz4740-i2s", .data = &jz4740_i2s_soc_info },
 	{ .compatible = "ingenic,jz4760-i2s", .data = &jz4760_i2s_soc_info },
 	{ .compatible = "ingenic,jz4770-i2s", .data = &jz4770_i2s_soc_info },
 	{ .compatible = "ingenic,jz4780-i2s", .data = &jz4780_i2s_soc_info },
 	{ /* sentinel */ }
 };
 MODULE_DEVICE_TABLE(of, jz4740_of_matches);

 static int jz4740_i2s_dev_probe(struct platform_device *pdev)
 {
 	struct device *dev = &pdev->dev;
 	struct jz4740_i2s *i2s;
 	struct resource *mem;
 	int ret;

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

 	i2s->soc_info = device_get_match_data(dev);

 	i2s->base = devm_platform_get_and_ioremap_resource(pdev, 0, &mem);
 	if (IS_ERR(i2s->base))
 		return PTR_ERR(i2s->base);

 	i2s->playback_dma_data.maxburst = 16;
 	i2s->playback_dma_data.addr = mem->start + JZ_REG_AIC_FIFO;

 	i2s->capture_dma_data.maxburst = 16;
 	i2s->capture_dma_data.addr = mem->start + JZ_REG_AIC_FIFO;

 	i2s->clk_aic = devm_clk_get(dev, "aic");
 	if (IS_ERR(i2s->clk_aic))
 		return PTR_ERR(i2s->clk_aic);

 	i2s->clk_i2s = devm_clk_get(dev, "i2s");
 	if (IS_ERR(i2s->clk_i2s))
 		return PTR_ERR(i2s->clk_i2s);

 	platform_set_drvdata(pdev, i2s);

 	ret = devm_snd_soc_register_component(dev, &jz4740_i2s_component,
 					      i2s->soc_info->dai, 1);
 	if (ret)
 		return ret;

 	return devm_snd_dmaengine_pcm_register(dev, NULL,
 		SND_DMAENGINE_PCM_FLAG_COMPAT);
 }

 static struct platform_driver jz4740_i2s_driver = {
 	.probe = jz4740_i2s_dev_probe,
 	.driver = {
 		.name = "jz4740-i2s",
 		.of_match_table = jz4740_of_matches,
 	},
 };

 module_platform_driver(jz4740_i2s_driver);

 MODULE_AUTHOR("Lars-Peter Clausen, <lars@metafoo.de>");
 MODULE_DESCRIPTION("Ingenic JZ4740 SoC I2S driver");
 MODULE_LICENSE("GPL");
 MODULE_ALIAS("platform:jz4740-i2s");
	// SPDX-License-Identifier: GPL-2.0-or-later
	/*
	* Copyright (C) 2010, Lars-Peter Clausen <lars@metafoo.de>
	*/

	#include <linux/init.h>
	#include <linux/io.h>
	#include <linux/kernel.h>
	#include <linux/module.h>
	#include <linux/mod_devicetable.h>
	#include <linux/platform_device.h>
	#include <linux/slab.h>

	#include <linux/clk.h>
	#include <linux/delay.h>

	#include <linux/dma-mapping.h>

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

	#include "jz4740-i2s.h"

	#define JZ_REG_AIC_CONF 0x00
	#define JZ_REG_AIC_CTRL 0x04
	#define JZ_REG_AIC_I2S_FMT 0x10
	#define JZ_REG_AIC_FIFO_STATUS 0x14
	#define JZ_REG_AIC_I2S_STATUS 0x1c
	#define JZ_REG_AIC_CLK_DIV 0x30
	#define JZ_REG_AIC_FIFO 0x34

	#define JZ_AIC_CONF_FIFO_RX_THRESHOLD_MASK (0xf << 12)
	#define JZ_AIC_CONF_FIFO_TX_THRESHOLD_MASK (0xf << 8)
	#define JZ_AIC_CONF_OVERFLOW_PLAY_LAST BIT(6)
	#define JZ_AIC_CONF_INTERNAL_CODEC BIT(5)
	#define JZ_AIC_CONF_I2S BIT(4)
	#define JZ_AIC_CONF_RESET BIT(3)
	#define JZ_AIC_CONF_BIT_CLK_MASTER BIT(2)
	#define JZ_AIC_CONF_SYNC_CLK_MASTER BIT(1)
	#define JZ_AIC_CONF_ENABLE BIT(0)

	#define JZ_AIC_CONF_FIFO_RX_THRESHOLD_OFFSET 12
	#define JZ_AIC_CONF_FIFO_TX_THRESHOLD_OFFSET 8
	#define JZ4760_AIC_CONF_FIFO_RX_THRESHOLD_OFFSET 24
	#define JZ4760_AIC_CONF_FIFO_TX_THRESHOLD_OFFSET 16

	#define JZ_AIC_CTRL_OUTPUT_SAMPLE_SIZE_MASK (0x7 << 19)
	#define JZ_AIC_CTRL_INPUT_SAMPLE_SIZE_MASK (0x7 << 16)
	#define JZ_AIC_CTRL_ENABLE_RX_DMA BIT(15)
	#define JZ_AIC_CTRL_ENABLE_TX_DMA BIT(14)
	#define JZ_AIC_CTRL_MONO_TO_STEREO BIT(11)
	#define JZ_AIC_CTRL_SWITCH_ENDIANNESS BIT(10)
	#define JZ_AIC_CTRL_SIGNED_TO_UNSIGNED BIT(9)
	#define JZ_AIC_CTRL_TFLUSH BIT(8)
	#define JZ_AIC_CTRL_RFLUSH BIT(7)
	#define JZ_AIC_CTRL_ENABLE_ROR_INT BIT(6)
	#define JZ_AIC_CTRL_ENABLE_TUR_INT BIT(5)
	#define JZ_AIC_CTRL_ENABLE_RFS_INT BIT(4)
	#define JZ_AIC_CTRL_ENABLE_TFS_INT BIT(3)
	#define JZ_AIC_CTRL_ENABLE_LOOPBACK BIT(2)
	#define JZ_AIC_CTRL_ENABLE_PLAYBACK BIT(1)
	#define JZ_AIC_CTRL_ENABLE_CAPTURE BIT(0)

	#define JZ_AIC_CTRL_OUTPUT_SAMPLE_SIZE_OFFSET 19
	#define JZ_AIC_CTRL_INPUT_SAMPLE_SIZE_OFFSET 16

	#define JZ_AIC_I2S_FMT_DISABLE_BIT_CLK BIT(12)
	#define JZ_AIC_I2S_FMT_DISABLE_BIT_ICLK BIT(13)
	#define JZ_AIC_I2S_FMT_ENABLE_SYS_CLK BIT(4)
	#define JZ_AIC_I2S_FMT_MSB BIT(0)

	#define JZ_AIC_I2S_STATUS_BUSY BIT(2)

	#define JZ_AIC_CLK_DIV_MASK 0xf
	#define I2SDIV_DV_SHIFT 0
	#define I2SDIV_DV_MASK (0xf << I2SDIV_DV_SHIFT)
	#define I2SDIV_IDV_SHIFT 8
	#define I2SDIV_IDV_MASK (0xf << I2SDIV_IDV_SHIFT)

	enum jz47xx_i2s_version {
	JZ_I2S_JZ4740,
	JZ_I2S_JZ4760,
	JZ_I2S_JZ4770,
	JZ_I2S_JZ4780,
	};

	struct i2s_soc_info {
	enum jz47xx_i2s_version version;
	struct snd_soc_dai_driver *dai;

	bool shared_fifo_flush;
	};

	struct jz4740_i2s {
	void __iomem *base;

	struct clk *clk_aic;
	struct clk *clk_i2s;

	struct snd_dmaengine_dai_dma_data playback_dma_data;
	struct snd_dmaengine_dai_dma_data capture_dma_data;

	const struct i2s_soc_info *soc_info;
	};

	static inline uint32_t jz4740_i2s_read(const struct jz4740_i2s *i2s,
	unsigned int reg)
	{
	return readl(i2s->base + reg);
	}

	static inline void jz4740_i2s_write(const struct jz4740_i2s *i2s,
	unsigned int reg, uint32_t value)
	{
	writel(value, i2s->base + reg);
	}

	static inline void jz4740_i2s_set_bits(const struct jz4740_i2s *i2s,
	unsigned int reg, uint32_t bits)
	{
	uint32_t value = jz4740_i2s_read(i2s, reg);
	value \|= bits;
	jz4740_i2s_write(i2s, reg, value);
	}

	static int jz4740_i2s_startup(struct snd_pcm_substream *substream,
	struct snd_soc_dai *dai)
	{
	struct jz4740_i2s *i2s = snd_soc_dai_get_drvdata(dai);
	uint32_t conf;
	int ret;

	/*
	* When we can flush FIFOs independently, only flush the FIFO
	* that is starting up. We can do this when the DAI is active
	* because it does not disturb other active substreams.
	*/
	if (!i2s->soc_info->shared_fifo_flush) {
	if (substream->stream == SNDRV_PCM_STREAM_PLAYBACK)
	jz4740_i2s_set_bits(i2s, JZ_REG_AIC_CTRL, JZ_AIC_CTRL_TFLUSH);
	else
	jz4740_i2s_set_bits(i2s, JZ_REG_AIC_CTRL, JZ_AIC_CTRL_RFLUSH);
	}

	if (snd_soc_dai_active(dai))
	return 0;

	/*
	* When there is a shared flush bit for both FIFOs, the TFLUSH
	* bit flushes both FIFOs. Flushing while the DAI is active would
	* cause FIFO underruns in other active substreams so we have to
	* guard this behind the snd_soc_dai_active() check.
	*/
	if (i2s->soc_info->shared_fifo_flush)
	jz4740_i2s_set_bits(i2s, JZ_REG_AIC_CTRL, JZ_AIC_CTRL_TFLUSH);

	ret = clk_prepare_enable(i2s->clk_i2s);
	if (ret)
	return ret;

	conf = jz4740_i2s_read(i2s, JZ_REG_AIC_CONF);
	conf \|= JZ_AIC_CONF_ENABLE;
	jz4740_i2s_write(i2s, JZ_REG_AIC_CONF, conf);

	return 0;
	}

	static void jz4740_i2s_shutdown(struct snd_pcm_substream *substream,
	struct snd_soc_dai *dai)
	{
	struct jz4740_i2s *i2s = snd_soc_dai_get_drvdata(dai);
	uint32_t conf;

	if (snd_soc_dai_active(dai))
	return;

	conf = jz4740_i2s_read(i2s, JZ_REG_AIC_CONF);
	conf &= ~JZ_AIC_CONF_ENABLE;
	jz4740_i2s_write(i2s, JZ_REG_AIC_CONF, conf);

	clk_disable_unprepare(i2s->clk_i2s);
	}

	static int jz4740_i2s_trigger(struct snd_pcm_substream *substream, int cmd,
	struct snd_soc_dai *dai)
	{
	struct jz4740_i2s *i2s = snd_soc_dai_get_drvdata(dai);

	uint32_t ctrl;
	uint32_t mask;

	if (substream->stream == SNDRV_PCM_STREAM_PLAYBACK)
	mask = JZ_AIC_CTRL_ENABLE_PLAYBACK \| JZ_AIC_CTRL_ENABLE_TX_DMA;
	else
	mask = JZ_AIC_CTRL_ENABLE_CAPTURE \| JZ_AIC_CTRL_ENABLE_RX_DMA;

	ctrl = jz4740_i2s_read(i2s, JZ_REG_AIC_CTRL);

	switch (cmd) {
	case SNDRV_PCM_TRIGGER_START:
	case SNDRV_PCM_TRIGGER_RESUME:
	case SNDRV_PCM_TRIGGER_PAUSE_RELEASE:
	ctrl \|= mask;
	break;
	case SNDRV_PCM_TRIGGER_STOP:
	case SNDRV_PCM_TRIGGER_SUSPEND:
	case SNDRV_PCM_TRIGGER_PAUSE_PUSH:
	ctrl &= ~mask;
	break;
	default:
	return -EINVAL;
	}

	jz4740_i2s_write(i2s, JZ_REG_AIC_CTRL, ctrl);

	return 0;
	}

	static int jz4740_i2s_set_fmt(struct snd_soc_dai *dai, unsigned int fmt)
	{
	struct jz4740_i2s *i2s = snd_soc_dai_get_drvdata(dai);

	uint32_t format = 0;
	uint32_t conf;

	conf = jz4740_i2s_read(i2s, JZ_REG_AIC_CONF);

	conf &= ~(JZ_AIC_CONF_BIT_CLK_MASTER \| JZ_AIC_CONF_SYNC_CLK_MASTER);

	switch (fmt & SND_SOC_DAIFMT_CLOCK_PROVIDER_MASK) {
	case SND_SOC_DAIFMT_BP_FP:
	conf \|= JZ_AIC_CONF_BIT_CLK_MASTER \| JZ_AIC_CONF_SYNC_CLK_MASTER;
	format \|= JZ_AIC_I2S_FMT_ENABLE_SYS_CLK;
	break;
	case SND_SOC_DAIFMT_BC_FP:
	conf \|= JZ_AIC_CONF_SYNC_CLK_MASTER;
	break;
	case SND_SOC_DAIFMT_BP_FC:
	conf \|= JZ_AIC_CONF_BIT_CLK_MASTER;
	break;
	case SND_SOC_DAIFMT_BC_FC:
	break;
	default:
	return -EINVAL;
	}

	switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
	case SND_SOC_DAIFMT_MSB:
	format \|= JZ_AIC_I2S_FMT_MSB;
	break;
	case SND_SOC_DAIFMT_I2S:
	break;
	default:
	return -EINVAL;
	}

	switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
	case SND_SOC_DAIFMT_NB_NF:
	break;
	default:
	return -EINVAL;
	}

	jz4740_i2s_write(i2s, JZ_REG_AIC_CONF, conf);
	jz4740_i2s_write(i2s, JZ_REG_AIC_I2S_FMT, format);

	return 0;
	}

	static int jz4740_i2s_hw_params(struct snd_pcm_substream *substream,
	struct snd_pcm_hw_params params, struct snd_soc_dai dai)
	{
	struct jz4740_i2s *i2s = snd_soc_dai_get_drvdata(dai);
	unsigned int sample_size;
	uint32_t ctrl, div_reg;
	int div;

	ctrl = jz4740_i2s_read(i2s, JZ_REG_AIC_CTRL);

	div_reg = jz4740_i2s_read(i2s, JZ_REG_AIC_CLK_DIV);
	div = clk_get_rate(i2s->clk_i2s) / (64 * params_rate(params));

	switch (params_format(params)) {
	case SNDRV_PCM_FORMAT_S8:
	sample_size = 0;
	break;
	case SNDRV_PCM_FORMAT_S16:
	sample_size = 1;
	break;
	default:
	return -EINVAL;
	}

	if (substream->stream == SNDRV_PCM_STREAM_PLAYBACK) {
	ctrl &= ~JZ_AIC_CTRL_OUTPUT_SAMPLE_SIZE_MASK;
	ctrl \|= sample_size << JZ_AIC_CTRL_OUTPUT_SAMPLE_SIZE_OFFSET;
	if (params_channels(params) == 1)
	ctrl \|= JZ_AIC_CTRL_MONO_TO_STEREO;
	else
	ctrl &= ~JZ_AIC_CTRL_MONO_TO_STEREO;

	div_reg &= ~I2SDIV_DV_MASK;
	div_reg \|= (div - 1) << I2SDIV_DV_SHIFT;
	} else {
	ctrl &= ~JZ_AIC_CTRL_INPUT_SAMPLE_SIZE_MASK;
	ctrl \|= sample_size << JZ_AIC_CTRL_INPUT_SAMPLE_SIZE_OFFSET;

	if (i2s->soc_info->version >= JZ_I2S_JZ4770) {
	div_reg &= ~I2SDIV_IDV_MASK;
	div_reg \|= (div - 1) << I2SDIV_IDV_SHIFT;
	} else {
	div_reg &= ~I2SDIV_DV_MASK;
	div_reg \|= (div - 1) << I2SDIV_DV_SHIFT;
	}
	}

	jz4740_i2s_write(i2s, JZ_REG_AIC_CTRL, ctrl);
	jz4740_i2s_write(i2s, JZ_REG_AIC_CLK_DIV, div_reg);

	return 0;
	}

	static int jz4740_i2s_set_sysclk(struct snd_soc_dai *dai, int clk_id,
	unsigned int freq, int dir)
	{
	struct jz4740_i2s *i2s = snd_soc_dai_get_drvdata(dai);
	struct clk *parent;
	int ret = 0;

	switch (clk_id) {
	case JZ4740_I2S_CLKSRC_EXT:
	parent = clk_get(NULL, "ext");
	if (IS_ERR(parent))
	return PTR_ERR(parent);
	clk_set_parent(i2s->clk_i2s, parent);
	break;
	case JZ4740_I2S_CLKSRC_PLL:
	parent = clk_get(NULL, "pll half");
	if (IS_ERR(parent))
	return PTR_ERR(parent);
	clk_set_parent(i2s->clk_i2s, parent);
	ret = clk_set_rate(i2s->clk_i2s, freq);
	break;
	default:
	return -EINVAL;
	}
	clk_put(parent);

	return ret;
	}

	static int jz4740_i2s_suspend(struct snd_soc_component *component)
	{
	struct jz4740_i2s *i2s = snd_soc_component_get_drvdata(component);
	uint32_t conf;

	if (snd_soc_component_active(component)) {
	conf = jz4740_i2s_read(i2s, JZ_REG_AIC_CONF);
	conf &= ~JZ_AIC_CONF_ENABLE;
	jz4740_i2s_write(i2s, JZ_REG_AIC_CONF, conf);

	clk_disable_unprepare(i2s->clk_i2s);
	}

	clk_disable_unprepare(i2s->clk_aic);

	return 0;
	}

	static int jz4740_i2s_resume(struct snd_soc_component *component)
	{
	struct jz4740_i2s *i2s = snd_soc_component_get_drvdata(component);
	uint32_t conf;
	int ret;

	ret = clk_prepare_enable(i2s->clk_aic);
	if (ret)
	return ret;

	if (snd_soc_component_active(component)) {
	ret = clk_prepare_enable(i2s->clk_i2s);
	if (ret) {
	clk_disable_unprepare(i2s->clk_aic);
	return ret;
	}

	conf = jz4740_i2s_read(i2s, JZ_REG_AIC_CONF);
	conf \|= JZ_AIC_CONF_ENABLE;
	jz4740_i2s_write(i2s, JZ_REG_AIC_CONF, conf);
	}

	return 0;
	}

	static int jz4740_i2s_dai_probe(struct snd_soc_dai *dai)
	{
	struct jz4740_i2s *i2s = snd_soc_dai_get_drvdata(dai);
	uint32_t conf;
	int ret;

	ret = clk_prepare_enable(i2s->clk_aic);
	if (ret)
	return ret;

	snd_soc_dai_init_dma_data(dai, &i2s->playback_dma_data,
	&i2s->capture_dma_data);

	if (i2s->soc_info->version >= JZ_I2S_JZ4760) {
	conf = (7 << JZ4760_AIC_CONF_FIFO_RX_THRESHOLD_OFFSET) \|
	(8 << JZ4760_AIC_CONF_FIFO_TX_THRESHOLD_OFFSET) \|
	JZ_AIC_CONF_OVERFLOW_PLAY_LAST \|
	JZ_AIC_CONF_I2S \|
	JZ_AIC_CONF_INTERNAL_CODEC;
	} else {
	conf = (7 << JZ_AIC_CONF_FIFO_RX_THRESHOLD_OFFSET) \|
	(8 << JZ_AIC_CONF_FIFO_TX_THRESHOLD_OFFSET) \|
	JZ_AIC_CONF_OVERFLOW_PLAY_LAST \|
	JZ_AIC_CONF_I2S \|
	JZ_AIC_CONF_INTERNAL_CODEC;
	}

	jz4740_i2s_write(i2s, JZ_REG_AIC_CONF, JZ_AIC_CONF_RESET);
	jz4740_i2s_write(i2s, JZ_REG_AIC_CONF, conf);

	return 0;
	}

	static int jz4740_i2s_dai_remove(struct snd_soc_dai *dai)
	{
	struct jz4740_i2s *i2s = snd_soc_dai_get_drvdata(dai);

	clk_disable_unprepare(i2s->clk_aic);
	return 0;
	}

	static const struct snd_soc_dai_ops jz4740_i2s_dai_ops = {
	.startup = jz4740_i2s_startup,
	.shutdown = jz4740_i2s_shutdown,
	.trigger = jz4740_i2s_trigger,
	.hw_params = jz4740_i2s_hw_params,
	.set_fmt = jz4740_i2s_set_fmt,
	.set_sysclk = jz4740_i2s_set_sysclk,
	};

	#define JZ4740_I2S_FMTS (SNDRV_PCM_FMTBIT_S8 \| \
	SNDRV_PCM_FMTBIT_S16_LE)

	static struct snd_soc_dai_driver jz4740_i2s_dai = {
	.probe = jz4740_i2s_dai_probe,
	.remove = jz4740_i2s_dai_remove,
	.playback = {
	.channels_min = 1,
	.channels_max = 2,
	.rates = SNDRV_PCM_RATE_8000_48000,
	.formats = JZ4740_I2S_FMTS,
	},
	.capture = {
	.channels_min = 2,
	.channels_max = 2,
	.rates = SNDRV_PCM_RATE_8000_48000,
	.formats = JZ4740_I2S_FMTS,
	},
	.symmetric_rate = 1,
	.ops = &jz4740_i2s_dai_ops,
	};

	static const struct i2s_soc_info jz4740_i2s_soc_info = {
	.version = JZ_I2S_JZ4740,
	.dai = &jz4740_i2s_dai,
	.shared_fifo_flush = true,
	};

	static const struct i2s_soc_info jz4760_i2s_soc_info = {
	.version = JZ_I2S_JZ4760,
	.dai = &jz4740_i2s_dai,
	};

	static struct snd_soc_dai_driver jz4770_i2s_dai = {
	.probe = jz4740_i2s_dai_probe,
	.remove = jz4740_i2s_dai_remove,
	.playback = {
	.channels_min = 1,
	.channels_max = 2,
	.rates = SNDRV_PCM_RATE_8000_48000,
	.formats = JZ4740_I2S_FMTS,
	},
	.capture = {
	.channels_min = 2,
	.channels_max = 2,
	.rates = SNDRV_PCM_RATE_8000_48000,
	.formats = JZ4740_I2S_FMTS,
	},
	.ops = &jz4740_i2s_dai_ops,
	};

	static const struct i2s_soc_info jz4770_i2s_soc_info = {
	.version = JZ_I2S_JZ4770,
	.dai = &jz4770_i2s_dai,
	};

	static const struct i2s_soc_info jz4780_i2s_soc_info = {
	.version = JZ_I2S_JZ4780,
	.dai = &jz4770_i2s_dai,
	};

	static const struct snd_soc_component_driver jz4740_i2s_component = {
	.name = "jz4740-i2s",
	.suspend = jz4740_i2s_suspend,
	.resume = jz4740_i2s_resume,
	.legacy_dai_naming = 1,
	};

	static const struct of_device_id jz4740_of_matches[] = {
	{ .compatible = "ingenic,jz4740-i2s", .data = &jz4740_i2s_soc_info },
	{ .compatible = "ingenic,jz4760-i2s", .data = &jz4760_i2s_soc_info },
	{ .compatible = "ingenic,jz4770-i2s", .data = &jz4770_i2s_soc_info },
	{ .compatible = "ingenic,jz4780-i2s", .data = &jz4780_i2s_soc_info },
	{ /* sentinel */ }
	};
	MODULE_DEVICE_TABLE(of, jz4740_of_matches);

	static int jz4740_i2s_dev_probe(struct platform_device *pdev)
	{
	struct device *dev = &pdev->dev;
	struct jz4740_i2s *i2s;
	struct resource *mem;
	int ret;

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

	i2s->soc_info = device_get_match_data(dev);

	i2s->base = devm_platform_get_and_ioremap_resource(pdev, 0, &mem);
	if (IS_ERR(i2s->base))
	return PTR_ERR(i2s->base);

	i2s->playback_dma_data.maxburst = 16;
	i2s->playback_dma_data.addr = mem->start + JZ_REG_AIC_FIFO;

	i2s->capture_dma_data.maxburst = 16;
	i2s->capture_dma_data.addr = mem->start + JZ_REG_AIC_FIFO;

	i2s->clk_aic = devm_clk_get(dev, "aic");
	if (IS_ERR(i2s->clk_aic))
	return PTR_ERR(i2s->clk_aic);

	i2s->clk_i2s = devm_clk_get(dev, "i2s");
	if (IS_ERR(i2s->clk_i2s))
	return PTR_ERR(i2s->clk_i2s);

	platform_set_drvdata(pdev, i2s);

	ret = devm_snd_soc_register_component(dev, &jz4740_i2s_component,
	i2s->soc_info->dai, 1);
	if (ret)
	return ret;

	return devm_snd_dmaengine_pcm_register(dev, NULL,
	SND_DMAENGINE_PCM_FLAG_COMPAT);
	}

	static struct platform_driver jz4740_i2s_driver = {
	.probe = jz4740_i2s_dev_probe,
	.driver = {
	.name = "jz4740-i2s",
	.of_match_table = jz4740_of_matches,
	},
	};

	module_platform_driver(jz4740_i2s_driver);

	MODULE_AUTHOR("Lars-Peter Clausen, <lars@metafoo.de>");
	MODULE_DESCRIPTION("Ingenic JZ4740 SoC I2S driver");
	MODULE_LICENSE("GPL");
	MODULE_ALIAS("platform:jz4740-i2s");