// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * sound/soc/codecs/wm8782.c
 * simple, strap-pin configured 24bit 2ch ADC
 *
 * Copyright: 2011 Raumfeld GmbH
 * Author: Johannes Stezenbach <js@sig21.net>
 *
 * based on ad73311.c
 * Copyright:	Analog Devices Inc.
 * Author:	Cliff Cai <cliff.cai@analog.com>
 */

#include <linux/init.h>
#include <linux/slab.h>
#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/device.h>
#include <linux/regulator/consumer.h>
#include <sound/core.h>
#include <sound/pcm.h>
#include <sound/ac97_codec.h>
#include <sound/initval.h>
#include <sound/soc.h>

/* regulator power supply names */
static const char *supply_names[] = {
	"Vdda", /* analog supply, 2.7V - 3.6V */
	"Vdd",  /* digital supply, 2.7V - 5.5V */
};

struct wm8782_priv {
	struct regulator_bulk_data supplies[ARRAY_SIZE(supply_names)];
	int max_rate;
};

static int wm8782_dai_startup(struct snd_pcm_substream *sub, struct snd_soc_dai *dai)
{
	struct snd_pcm_runtime *runtime = sub->runtime;
	struct wm8782_priv *priv =
		snd_soc_component_get_drvdata(dai->component);

	return snd_pcm_hw_constraint_minmax(runtime, SNDRV_PCM_HW_PARAM_RATE,
					   8000, priv->max_rate);
}

static const struct snd_soc_dapm_widget wm8782_dapm_widgets[] = {
SND_SOC_DAPM_INPUT("AINL"),
SND_SOC_DAPM_INPUT("AINR"),
};

static const struct snd_soc_dapm_route wm8782_dapm_routes[] = {
	{ "Capture", NULL, "AINL" },
	{ "Capture", NULL, "AINR" },
};

static const struct snd_soc_dai_ops wm8782_dai_ops = {
	.startup = &wm8782_dai_startup,
};

static struct snd_soc_dai_driver wm8782_dai = {
	.name = "wm8782",
	.capture = {
		.stream_name = "Capture",
		.channels_min = 2,
		.channels_max = 2,
		.rates = SNDRV_PCM_RATE_8000_192000,
		.formats = SNDRV_PCM_FMTBIT_S16_LE |
			   SNDRV_PCM_FMTBIT_S20_3LE |
			   SNDRV_PCM_FMTBIT_S24_LE,
	},
	.ops = &wm8782_dai_ops,
};

static int wm8782_soc_probe(struct snd_soc_component *component)
{
	struct wm8782_priv *priv = snd_soc_component_get_drvdata(component);
	return regulator_bulk_enable(ARRAY_SIZE(priv->supplies), priv->supplies);
}

static void wm8782_soc_remove(struct snd_soc_component *component)
{
	struct wm8782_priv *priv = snd_soc_component_get_drvdata(component);
	regulator_bulk_disable(ARRAY_SIZE(priv->supplies), priv->supplies);
}

#ifdef CONFIG_PM
static int wm8782_soc_suspend(struct snd_soc_component *component)
{
	struct wm8782_priv *priv = snd_soc_component_get_drvdata(component);
	regulator_bulk_disable(ARRAY_SIZE(priv->supplies), priv->supplies);
	return 0;
}

static int wm8782_soc_resume(struct snd_soc_component *component)
{
	struct wm8782_priv *priv = snd_soc_component_get_drvdata(component);
	return regulator_bulk_enable(ARRAY_SIZE(priv->supplies), priv->supplies);
}
#else
#define wm8782_soc_suspend      NULL
#define wm8782_soc_resume       NULL
#endif /* CONFIG_PM */

static const struct snd_soc_component_driver soc_component_dev_wm8782 = {
	.probe			= wm8782_soc_probe,
	.remove			= wm8782_soc_remove,
	.suspend		= wm8782_soc_suspend,
	.resume			= wm8782_soc_resume,
	.dapm_widgets		= wm8782_dapm_widgets,
	.num_dapm_widgets	= ARRAY_SIZE(wm8782_dapm_widgets),
	.dapm_routes		= wm8782_dapm_routes,
	.num_dapm_routes	= ARRAY_SIZE(wm8782_dapm_routes),
	.idle_bias_on		= 1,
	.use_pmdown_time	= 1,
	.endianness		= 1,
};

static int wm8782_probe(struct platform_device *pdev)
{
	struct device *dev = &pdev->dev;
	struct device_node *np = dev->of_node;
	struct wm8782_priv *priv;
	int ret, i, fsampen;

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

	dev_set_drvdata(dev, priv);

	for (i = 0; i < ARRAY_SIZE(supply_names); i++)
		priv->supplies[i].supply = supply_names[i];

	ret = devm_regulator_bulk_get(dev, ARRAY_SIZE(priv->supplies),
				      priv->supplies);
	if (ret < 0)
		return ret;

	// Assume lowest value by default to avoid inadvertent overclocking
	fsampen = 0;

	if (np)
		of_property_read_u32(np, "wlf,fsampen", &fsampen);

	switch (fsampen) {
	case 0:
		priv->max_rate = 48000;
		break;
	case 1:
		priv->max_rate = 96000;
		break;
	case 2:
		priv->max_rate = 192000;
		break;
	default:
		dev_err(dev, "Invalid wlf,fsampen value");
		return -EINVAL;
	}

	return devm_snd_soc_register_component(&pdev->dev,
			&soc_component_dev_wm8782, &wm8782_dai, 1);
}

#ifdef CONFIG_OF
static const struct of_device_id wm8782_of_match[] = {
	{ .compatible = "wlf,wm8782", },
	{ }
};
MODULE_DEVICE_TABLE(of, wm8782_of_match);
#endif

static struct platform_driver wm8782_codec_driver = {
	.driver = {
		.name = "wm8782",
		.of_match_table = of_match_ptr(wm8782_of_match),
	},
	.probe = wm8782_probe,
};

module_platform_driver(wm8782_codec_driver);

MODULE_DESCRIPTION("ASoC WM8782 driver");
MODULE_AUTHOR("Johannes Stezenbach <js@sig21.net>");
MODULE_LICENSE("GPL");