VisionFive2 Linux kernel

diff --git a/sound/soc/starfive/Kconfig b/sound/soc/starfive/Kconfig
index c134aae82870..2c9465ab47bb 100644
--- a/sound/soc/starfive/Kconfig
+++ b/sound/soc/starfive/Kconfig
@@ -54,6 +54,14 @@ config SND_SOC_JH7110_SPDIF_PCM
 	  Say Y or N if you want to add a custom ALSA extension that registers
 	  a PCM and uses PIO to transfer data.
 
+config SND_SOC_JH7110_PDM
+	tristate "JH7110 PDM device driver"
+	depends on HAVE_CLK && SND_SOC_STARFIVE
+	select SND_SOC_JH7110_I2S
+	select REGMAP_MMIO
+	help
+	  Say Y or M if you want to add support for StarFive pdm driver.
+
 config SND_SOC_JH7110_TDM
 	tristate "JH7110 TDM device driver"
 	depends on HAVE_CLK && SND_SOC_STARFIVE
diff --git a/sound/soc/starfive/Makefile b/sound/soc/starfive/Makefile
index e39b557520f5..29604e0f1644 100644
--- a/sound/soc/starfive/Makefile
+++ b/sound/soc/starfive/Makefile
@@ -9,4 +9,5 @@ obj-$(CONFIG_SND_SOC_JH7110_SPDIF) += spdif.o
 spdif-y := jh7110_spdif.o
 spdif-$(CONFIG_SND_SOC_JH7110_SPDIF_PCM) += jh7110_spdif_pcm.o
 
+obj-$(CONFIG_SND_SOC_JH7110_PDM) += jh7110_pdm.o
 obj-$(CONFIG_SND_SOC_JH7110_TDM) += jh7110_tdm.o
diff --git a/sound/soc/starfive/jh7110_pdm.c b/sound/soc/starfive/jh7110_pdm.c
new file mode 100644
index 000000000000..74a8a2f2f675
--- /dev/null
+++ b/sound/soc/starfive/jh7110_pdm.c
@@ -0,0 +1,493 @@
+// SPDX-License-Identifier: GPL-2.0
+/*
+ * PDM driver for the StarFive JH7110 SoC
+ *
+ * Copyright (C) 2022 StarFive Technology Co., Ltd.
+ */
+#include <linux/clk.h>
+#include <linux/device.h>
+#include <linux/dmaengine.h>
+#include <linux/reset.h>
+#include <linux/module.h>
+#include <linux/of_irq.h>
+#include <linux/of_platform.h>
+#include <linux/regmap.h>
+#include <linux/pm_runtime.h>
+#include <linux/types.h>
+#include <sound/dmaengine_pcm.h>
+#include <sound/initval.h>
+#include <sound/pcm.h>
+#include <sound/pcm_params.h>
+#include <sound/soc.h>
+#include <sound/soc-dai.h>
+#include <sound/tlv.h>
+
+#define PDM_DMIC_CTRL0			0x00
+#define PDM_DC_SCALE0			0x04
+#define PDM_DMIC_CTRL1			0x10
+#define PDM_DC_SCALE1			0x14
+
+/* PDM CTRL OFFSET */
+#define PDM_DMIC_MSB_SHIFT		1
+#define PDM_DMIC_MSB_MASK		(0x7 << PDM_DMIC_MSB_SHIFT)
+#define PDM_DMIC_VOL_SHIFT		16
+#define PDM_DMIC_VOL_MASK		(0x3f << PDM_DMIC_VOL_SHIFT)
+#define PDM_VOL_DB_MUTE			(0x3f << PDM_DMIC_VOL_SHIFT)
+#define PDM_VOL_DB_MAX			0
+
+#define PDM_DMIC_RVOL_MASK		BIT(22)
+#define PDM_DMIC_LVOL_MASK		BIT(23)
+#define PDM_DMIC_I2S_SLAVE		BIT(24)
+#define PDM_DMIC_HPF_EN			BIT(28)
+#define PDM_DMIC_FASTMODE_MASK		BIT(29)
+#define PDM_DMIC_DC_BYPASS_MASK		BIT(30)
+#define PDM_SW_RST_MASK			BIT(31)
+#define PDM_SW_RST_RELEASE		BIT(31)
+
+/* PDM SCALE OFFSET */
+#define DMIC_DCOFF3_SHIFT		24
+#define DMIC_DCOFF2_SHIFT		16
+#define DMIC_DCOFF1_SHIFT		8
+
+#define DMIC_DCOFF3_MASK		(0xf << DMIC_DCOFF3_SHIFT)
+#define DMIC_DCOFF3_VAL			(0xc << DMIC_DCOFF3_SHIFT)
+#define DMIC_DCOFF1_MASK		(0xff << DMIC_DCOFF1_SHIFT)
+#define DMIC_DCOFF1_VAL			(0x5 << DMIC_DCOFF1_SHIFT)
+#define DMIC_SCALE_MASK			0x3f
+#define DMIC_SCALE_DEF_VAL		0x8
+
+enum PDM_MSB_SHIFT {
+	PDM_MSB_SHIFT_NONE = 0,
+	PDM_MSB_SHIFT_1,
+	PDM_MSB_SHIFT_2,
+	PDM_MSB_SHIFT_3,
+	PDM_MSB_SHIFT_4,
+	PDM_MSB_SHIFT_5,
+	PDM_MSB_SHIFT_6,
+	PDM_MSB_SHIFT_7,
+};
+
+struct sf_pdm {
+	struct regmap *pdm_map;
+	struct device *dev;
+	struct clk *clk_pdm_apb;
+	struct clk *clk_pdm_mclk;
+	struct clk *clk_mclk;
+	struct clk *clk_mclk_ext;
+	struct reset_control *rst_pdm_dmic;
+	struct reset_control *rst_pdm_apb;
+	unsigned char flag_first;
+	unsigned int saved_ctrl0;
+	unsigned int saved_scale0;
+};
+
+static const DECLARE_TLV_DB_SCALE(volume_tlv, -9450, 150, 0);
+
+static const struct snd_kcontrol_new sf_pdm_snd_controls[] = {
+	SOC_SINGLE("DC compensation Control", PDM_DMIC_CTRL0, 30, 1, 0),
+	SOC_SINGLE("High Pass Filter Control", PDM_DMIC_CTRL0, 28, 1, 0),
+	SOC_SINGLE("Left Channel Volume Control", PDM_DMIC_CTRL0, 23, 1, 0),
+	SOC_SINGLE("Right Channel Volume Control", PDM_DMIC_CTRL0, 22, 1, 0),
+	SOC_SINGLE_TLV("Volume", PDM_DMIC_CTRL0, 16, 0x3F, 1, volume_tlv),
+	SOC_SINGLE("Data MSB Shift", PDM_DMIC_CTRL0, 1, 7, 0),
+	SOC_SINGLE("SCALE", PDM_DC_SCALE0, 0, 0x3F, 0),
+	SOC_SINGLE("DC offset", PDM_DC_SCALE0, 8, 0xFFFFF, 0),
+};
+
+static void sf_pdm_enable(struct regmap *map)
+{
+	/* Left and Right Channel Volume Control Enable */
+	regmap_update_bits(map, PDM_DMIC_CTRL0, PDM_DMIC_RVOL_MASK, 0);
+	regmap_update_bits(map, PDM_DMIC_CTRL0, PDM_DMIC_LVOL_MASK, 0);
+}
+
+static void sf_pdm_disable(struct regmap *map)
+{
+	/* Left and Right Channel Volume Control Disable */
+	regmap_update_bits(map, PDM_DMIC_CTRL0,
+			   PDM_DMIC_RVOL_MASK, PDM_DMIC_RVOL_MASK);
+	regmap_update_bits(map, PDM_DMIC_CTRL0,
+			   PDM_DMIC_LVOL_MASK, PDM_DMIC_LVOL_MASK);
+}
+
+static int sf_pdm_trigger(struct snd_pcm_substream *substream, int cmd,
+			   struct snd_soc_dai *dai)
+{
+	struct sf_pdm *priv = snd_soc_dai_get_drvdata(dai);
+
+	switch (cmd) {
+	case SNDRV_PCM_TRIGGER_START:
+	case SNDRV_PCM_TRIGGER_RESUME:
+	case SNDRV_PCM_TRIGGER_PAUSE_RELEASE:
+		if (priv->flag_first) {
+			priv->flag_first = 0;
+			mdelay(200);
+		}
+
+		sf_pdm_enable(priv->pdm_map);
+		return 0;
+
+	case SNDRV_PCM_TRIGGER_STOP:
+	case SNDRV_PCM_TRIGGER_SUSPEND:
+	case SNDRV_PCM_TRIGGER_PAUSE_PUSH:
+		sf_pdm_disable(priv->pdm_map);
+		return 0;
+
+	default:
+		return -EINVAL;
+	}
+}
+
+static int sf_pdm_hw_params(struct snd_pcm_substream *substream,
+			     struct snd_pcm_hw_params *params,
+			     struct snd_soc_dai *dai)
+{
+	struct sf_pdm *priv = snd_soc_dai_get_drvdata(dai);
+	unsigned int sample_rate;
+	unsigned int data_width;
+	int ret;
+	const int pdm_mul = 128;
+
+	sample_rate = params_rate(params);
+	switch (sample_rate) {
+	case 8000:
+	case 11025:
+	case 16000:
+		break;
+	default:
+		dev_err(priv->dev, "can't support sample rate:%d\n", sample_rate);
+		return -EINVAL;
+	}
+
+	data_width = params_width(params);
+	switch (data_width) {
+	case 16:
+	case 32:
+		break;
+	default:
+		dev_err(priv->dev, "can't support bit width %d\n", data_width);
+		return -EINVAL;
+	}
+
+	/* set pdm_mclk,  PDM MCLK = 128 * LRCLK */
+	ret = clk_set_rate(priv->clk_pdm_mclk, pdm_mul * sample_rate);
+	if (ret) {
+		dev_err(priv->dev, "Can't set pdm_mclk: %d\n", ret);
+		return ret;
+	}
+
+	return 0;
+}
+
+static const struct snd_soc_dai_ops sf_pdm_dai_ops = {
+	.trigger	= sf_pdm_trigger,
+	.hw_params	= sf_pdm_hw_params,
+};
+
+static void sf_pdm_module_init(struct sf_pdm *priv)
+{
+	/* Reset */
+	regmap_update_bits(priv->pdm_map, PDM_DMIC_CTRL0,
+			   PDM_SW_RST_MASK, 0x00);
+	regmap_update_bits(priv->pdm_map, PDM_DMIC_CTRL0,
+			   PDM_SW_RST_MASK, PDM_SW_RST_RELEASE);
+
+	/* Make sure the device is initially disabled */
+	sf_pdm_disable(priv->pdm_map);
+
+	/* MUTE */
+	regmap_update_bits(priv->pdm_map, PDM_DMIC_CTRL0,
+			   PDM_DMIC_VOL_MASK, PDM_VOL_DB_MUTE);
+
+	/* UNMUTE */
+	regmap_update_bits(priv->pdm_map, PDM_DMIC_CTRL0,
+			   PDM_DMIC_VOL_MASK, PDM_VOL_DB_MAX);
+
+	/* enable high pass filter */
+	regmap_update_bits(priv->pdm_map, PDM_DMIC_CTRL0,
+			   PDM_DMIC_HPF_EN, PDM_DMIC_HPF_EN);
+
+	/* PDM work as slave mode */
+	regmap_update_bits(priv->pdm_map, PDM_DMIC_CTRL0,
+			   PDM_DMIC_I2S_SLAVE, PDM_DMIC_I2S_SLAVE);
+
+	/* disable fast mode */
+	regmap_update_bits(priv->pdm_map, PDM_DMIC_CTRL0,
+			   PDM_DMIC_FASTMODE_MASK, 0);
+
+	/* dmic msb shift 0 */
+	regmap_update_bits(priv->pdm_map, PDM_DMIC_CTRL0,
+			   PDM_DMIC_MSB_MASK, 0);
+
+	/* scale: 0x8 */
+	regmap_update_bits(priv->pdm_map, PDM_DC_SCALE0,
+			   DMIC_SCALE_MASK, DMIC_SCALE_DEF_VAL);
+
+	regmap_update_bits(priv->pdm_map, PDM_DC_SCALE0,
+			   DMIC_DCOFF1_MASK, DMIC_DCOFF1_VAL);
+
+	regmap_update_bits(priv->pdm_map, PDM_DC_SCALE0,
+			   DMIC_DCOFF3_MASK, DMIC_DCOFF3_VAL);
+
+	/* scale: 0x3f */
+	regmap_update_bits(priv->pdm_map, PDM_DC_SCALE0,
+			   DMIC_SCALE_MASK, DMIC_SCALE_MASK);
+
+	/* dmic msb shift 2 */
+	regmap_update_bits(priv->pdm_map, PDM_DMIC_CTRL0,
+			   PDM_DMIC_MSB_MASK, PDM_MSB_SHIFT_4);
+}
+
+#define SF_PDM_RATES	(SNDRV_PCM_RATE_8000 | \
+			SNDRV_PCM_RATE_11025 | \
+			SNDRV_PCM_RATE_16000)
+
+#define SF_PDM_FORMATS	(SNDRV_PCM_FMTBIT_S16_LE | \
+			SNDRV_PCM_FMTBIT_S32_LE)
+
+static struct snd_soc_dai_driver sf_pdm_dai_drv = {
+	.name = "PDM",
+	.id = 0,
+	.capture = {
+		.stream_name	= "Capture",
+		.channels_min	= 2,
+		.channels_max	= 2,
+		.rates		= SF_PDM_RATES,
+		.formats	= SF_PDM_FORMATS,
+	},
+	.ops = &sf_pdm_dai_ops,
+	.symmetric_rate = 1,
+};
+
+static int sf_pdm_component_probe(struct snd_soc_component *component)
+{
+	struct sf_pdm *priv = snd_soc_component_get_drvdata(component);
+
+	snd_soc_component_init_regmap(component, priv->pdm_map);
+	snd_soc_add_component_controls(component, sf_pdm_snd_controls,
+				       ARRAY_SIZE(sf_pdm_snd_controls));
+
+	return 0;
+}
+
+static int sf_pdm_clock_enable(struct sf_pdm *priv)
+{
+	int ret;
+
+	ret = clk_prepare_enable(priv->clk_pdm_mclk);
+	if (ret) {
+		dev_err(priv->dev, "failed to prepare enable clk_pdm_mclk\n");
+		return ret;
+	}
+
+	ret = clk_prepare_enable(priv->clk_pdm_apb);
+	if (ret) {
+		dev_err(priv->dev, "failed to prepare enable clk_pdm_apb\n");
+		goto disable_pdm_mclk;
+	}
+
+	ret = reset_control_deassert(priv->rst_pdm_dmic);
+	if (ret) {
+		dev_err(priv->dev, "failed to deassert pdm_dmic\n");
+		goto disable_pdm_apb;
+	}
+
+	ret = reset_control_deassert(priv->rst_pdm_apb);
+	if (ret) {
+		dev_err(priv->dev, "failed to deassert pdm_apb\n");
+		goto disable_pdm_apb;
+	}
+
+	ret = clk_set_parent(priv->clk_mclk, priv->clk_mclk_ext);
+	if (ret) {
+		dev_err(priv->dev, "failed to set parent clk_mclk ret=%d\n", ret);
+		goto disable_pdm_apb;
+	}
+
+	return 0;
+
+disable_pdm_apb:
+	clk_disable_unprepare(priv->clk_pdm_apb);
+disable_pdm_mclk:
+	clk_disable_unprepare(priv->clk_pdm_mclk);
+
+	return ret;
+}
+
+#ifdef CONFIG_PM
+static int sf_pdm_runtime_suspend(struct device *dev)
+{
+	struct sf_pdm *priv = dev_get_drvdata(dev);
+
+	clk_disable_unprepare(priv->clk_pdm_apb);
+	clk_disable_unprepare(priv->clk_pdm_mclk);
+
+	return 0;
+}
+
+static int sf_pdm_runtime_resume(struct device *dev)
+{
+	struct sf_pdm *priv = dev_get_drvdata(dev);
+	int ret;
+
+	ret = sf_pdm_clock_enable(priv);
+	if (!ret)
+		sf_pdm_module_init(priv);
+
+	return ret;
+}
+#endif
+
+#ifdef CONFIG_PM_SLEEP
+static int sf_pdm_suspend(struct snd_soc_component *component)
+{
+	return pm_runtime_force_suspend(component->dev);
+}
+
+static int sf_pdm_resume(struct snd_soc_component *component)
+{
+	return pm_runtime_force_resume(component->dev);
+}
+
+#else
+#define sf_pdm_suspend	NULL
+#define sf_pdm_resume	NULL
+#endif
+
+static const struct snd_soc_component_driver sf_pdm_component_drv = {
+	.name = "jh7110-pdm",
+	.probe = sf_pdm_component_probe,
+	.suspend = sf_pdm_suspend,
+	.resume = sf_pdm_resume,
+};
+
+static const struct regmap_config sf_pdm_regmap_cfg = {
+	.reg_bits	= 32,
+	.val_bits	= 32,
+	.reg_stride	= 4,
+	.max_register	= 0x20,
+};
+
+static int sf_pdm_clock_get(struct platform_device *pdev, struct sf_pdm *priv)
+{
+	int ret;
+
+	static struct clk_bulk_data clks[] = {
+		{ .id = "pdm_mclk" },
+		{ .id = "pdm_apb" },
+		{ .id = "clk_mclk" },
+		{ .id = "mclk_ext" },
+	};
+
+	ret = devm_clk_bulk_get(&pdev->dev, ARRAY_SIZE(clks), clks);
+	if (ret) {
+		dev_err(&pdev->dev, "failed to get pdm clocks\n");
+		goto exit;
+	}
+
+	priv->clk_pdm_mclk = clks[0].clk;
+	priv->clk_pdm_apb = clks[1].clk;
+	priv->clk_mclk = clks[2].clk;
+	priv->clk_mclk_ext = clks[3].clk;
+
+	priv->rst_pdm_dmic = devm_reset_control_get_exclusive(&pdev->dev, "pdm_dmic");
+	if (IS_ERR(priv->rst_pdm_dmic)) {
+		dev_err(&pdev->dev, "failed to get pdm_dmic reset control\n");
+		ret = PTR_ERR(priv->rst_pdm_dmic);
+		goto exit;
+	}
+
+	priv->rst_pdm_apb = devm_reset_control_get_exclusive(&pdev->dev, "pdm_apb");
+	if (IS_ERR(priv->rst_pdm_apb)) {
+		dev_err(&pdev->dev, "failed to get pdm_apb reset control\n");
+		ret = PTR_ERR(priv->rst_pdm_apb);
+		goto exit;
+	}
+
+	/*
+	 * pdm clock must always be enabled as hardware issue that
+	 * no data in the first 4 seconds of the first recording
+	 */
+	ret = sf_pdm_clock_enable(priv);
+
+exit:
+	return ret;
+}
+
+static int sf_pdm_probe(struct platform_device *pdev)
+{
+	struct sf_pdm *priv;
+	struct resource *res;
+	void __iomem *regs;
+	int ret;
+
+	priv = devm_kzalloc(&pdev->dev, sizeof(*priv), GFP_KERNEL);
+	if (!priv)
+		return -ENOMEM;
+	platform_set_drvdata(pdev, priv);
+
+	res = platform_get_resource_byname(pdev, IORESOURCE_MEM, "pdm");
+	regs = devm_ioremap_resource(&pdev->dev, res);
+	if (IS_ERR(regs))
+		return PTR_ERR(regs);
+
+	priv->pdm_map = devm_regmap_init_mmio(&pdev->dev, regs, &sf_pdm_regmap_cfg);
+	if (IS_ERR(priv->pdm_map)) {
+		dev_err(&pdev->dev, "failed to init regmap: %ld\n",
+			PTR_ERR(priv->pdm_map));
+		return PTR_ERR(priv->pdm_map);
+	}
+
+	priv->dev = &pdev->dev;
+	priv->flag_first = 1;
+
+	ret = sf_pdm_clock_get(pdev, priv);
+	if (ret) {
+		dev_err(&pdev->dev, "failed to enable audio-pdm clock\n");
+		return ret;
+	}
+
+	dev_set_drvdata(&pdev->dev, priv);
+
+	ret = devm_snd_soc_register_component(&pdev->dev, &sf_pdm_component_drv,
+					      &sf_pdm_dai_drv, 1);
+	if (ret) {
+		dev_err(&pdev->dev, "failed to register pdm dai\n");
+		return ret;
+	}
+	pm_runtime_enable(&pdev->dev);
+
+	return 0;
+}
+
+static int sf_pdm_dev_remove(struct platform_device *pdev)
+{
+	pm_runtime_disable(&pdev->dev);
+	return 0;
+}
+
+static const struct of_device_id sf_pdm_of_match[] = {
+	{.compatible = "starfive,jh7110-pdm",},
+	{}
+};
+MODULE_DEVICE_TABLE(of, sf_pdm_of_match);
+
+static const struct dev_pm_ops sf_pdm_pm_ops = {
+	SET_RUNTIME_PM_OPS(sf_pdm_runtime_suspend,
+			   sf_pdm_runtime_resume, NULL)
+};
+
+static struct platform_driver sf_pdm_driver = {
+	.driver = {
+		.name = "jh7110-pdm",
+		.of_match_table = sf_pdm_of_match,
+		.pm = &sf_pdm_pm_ops,
+	},
+	.probe = sf_pdm_probe,
+	.remove = sf_pdm_dev_remove,
+};
+module_platform_driver(sf_pdm_driver);
+
+MODULE_AUTHOR("Walker Chen <walker.chen@starfivetech.com>");
+MODULE_DESCRIPTION("Starfive PDM Controller Driver");
+MODULE_LICENSE("GPL v2");