162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0+
262306a36Sopenharmony_ci//
362306a36Sopenharmony_ci// wm831x-isink.c  --  Current sink driver for the WM831x series
462306a36Sopenharmony_ci//
562306a36Sopenharmony_ci// Copyright 2009 Wolfson Microelectronics PLC.
662306a36Sopenharmony_ci//
762306a36Sopenharmony_ci// Author: Mark Brown <broonie@opensource.wolfsonmicro.com>
862306a36Sopenharmony_ci
962306a36Sopenharmony_ci#include <linux/module.h>
1062306a36Sopenharmony_ci#include <linux/moduleparam.h>
1162306a36Sopenharmony_ci#include <linux/init.h>
1262306a36Sopenharmony_ci#include <linux/bitops.h>
1362306a36Sopenharmony_ci#include <linux/err.h>
1462306a36Sopenharmony_ci#include <linux/i2c.h>
1562306a36Sopenharmony_ci#include <linux/platform_device.h>
1662306a36Sopenharmony_ci#include <linux/regulator/driver.h>
1762306a36Sopenharmony_ci#include <linux/slab.h>
1862306a36Sopenharmony_ci
1962306a36Sopenharmony_ci#include <linux/mfd/wm831x/core.h>
2062306a36Sopenharmony_ci#include <linux/mfd/wm831x/regulator.h>
2162306a36Sopenharmony_ci#include <linux/mfd/wm831x/pdata.h>
2262306a36Sopenharmony_ci
2362306a36Sopenharmony_ci#define WM831X_ISINK_MAX_NAME 7
2462306a36Sopenharmony_ci
2562306a36Sopenharmony_cistruct wm831x_isink {
2662306a36Sopenharmony_ci	char name[WM831X_ISINK_MAX_NAME];
2762306a36Sopenharmony_ci	struct regulator_desc desc;
2862306a36Sopenharmony_ci	int reg;
2962306a36Sopenharmony_ci	struct wm831x *wm831x;
3062306a36Sopenharmony_ci	struct regulator_dev *regulator;
3162306a36Sopenharmony_ci};
3262306a36Sopenharmony_ci
3362306a36Sopenharmony_cistatic int wm831x_isink_enable(struct regulator_dev *rdev)
3462306a36Sopenharmony_ci{
3562306a36Sopenharmony_ci	struct wm831x_isink *isink = rdev_get_drvdata(rdev);
3662306a36Sopenharmony_ci	struct wm831x *wm831x = isink->wm831x;
3762306a36Sopenharmony_ci	int ret;
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_ci	/* We have a two stage enable: first start the ISINK... */
4062306a36Sopenharmony_ci	ret = wm831x_set_bits(wm831x, isink->reg, WM831X_CS1_ENA,
4162306a36Sopenharmony_ci			      WM831X_CS1_ENA);
4262306a36Sopenharmony_ci	if (ret != 0)
4362306a36Sopenharmony_ci		return ret;
4462306a36Sopenharmony_ci
4562306a36Sopenharmony_ci	/* ...then enable drive */
4662306a36Sopenharmony_ci	ret = wm831x_set_bits(wm831x, isink->reg, WM831X_CS1_DRIVE,
4762306a36Sopenharmony_ci			      WM831X_CS1_DRIVE);
4862306a36Sopenharmony_ci	if (ret != 0)
4962306a36Sopenharmony_ci		wm831x_set_bits(wm831x, isink->reg, WM831X_CS1_ENA, 0);
5062306a36Sopenharmony_ci
5162306a36Sopenharmony_ci	return ret;
5262306a36Sopenharmony_ci
5362306a36Sopenharmony_ci}
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_cistatic int wm831x_isink_disable(struct regulator_dev *rdev)
5662306a36Sopenharmony_ci{
5762306a36Sopenharmony_ci	struct wm831x_isink *isink = rdev_get_drvdata(rdev);
5862306a36Sopenharmony_ci	struct wm831x *wm831x = isink->wm831x;
5962306a36Sopenharmony_ci	int ret;
6062306a36Sopenharmony_ci
6162306a36Sopenharmony_ci	ret = wm831x_set_bits(wm831x, isink->reg, WM831X_CS1_DRIVE, 0);
6262306a36Sopenharmony_ci	if (ret < 0)
6362306a36Sopenharmony_ci		return ret;
6462306a36Sopenharmony_ci
6562306a36Sopenharmony_ci	ret = wm831x_set_bits(wm831x, isink->reg, WM831X_CS1_ENA, 0);
6662306a36Sopenharmony_ci	if (ret < 0)
6762306a36Sopenharmony_ci		return ret;
6862306a36Sopenharmony_ci
6962306a36Sopenharmony_ci	return ret;
7062306a36Sopenharmony_ci
7162306a36Sopenharmony_ci}
7262306a36Sopenharmony_ci
7362306a36Sopenharmony_cistatic int wm831x_isink_is_enabled(struct regulator_dev *rdev)
7462306a36Sopenharmony_ci{
7562306a36Sopenharmony_ci	struct wm831x_isink *isink = rdev_get_drvdata(rdev);
7662306a36Sopenharmony_ci	struct wm831x *wm831x = isink->wm831x;
7762306a36Sopenharmony_ci	int ret;
7862306a36Sopenharmony_ci
7962306a36Sopenharmony_ci	ret = wm831x_reg_read(wm831x, isink->reg);
8062306a36Sopenharmony_ci	if (ret < 0)
8162306a36Sopenharmony_ci		return ret;
8262306a36Sopenharmony_ci
8362306a36Sopenharmony_ci	if ((ret & (WM831X_CS1_ENA | WM831X_CS1_DRIVE)) ==
8462306a36Sopenharmony_ci	    (WM831X_CS1_ENA | WM831X_CS1_DRIVE))
8562306a36Sopenharmony_ci		return 1;
8662306a36Sopenharmony_ci	else
8762306a36Sopenharmony_ci		return 0;
8862306a36Sopenharmony_ci}
8962306a36Sopenharmony_ci
9062306a36Sopenharmony_cistatic const struct regulator_ops wm831x_isink_ops = {
9162306a36Sopenharmony_ci	.is_enabled = wm831x_isink_is_enabled,
9262306a36Sopenharmony_ci	.enable = wm831x_isink_enable,
9362306a36Sopenharmony_ci	.disable = wm831x_isink_disable,
9462306a36Sopenharmony_ci	.set_current_limit = regulator_set_current_limit_regmap,
9562306a36Sopenharmony_ci	.get_current_limit = regulator_get_current_limit_regmap,
9662306a36Sopenharmony_ci};
9762306a36Sopenharmony_ci
9862306a36Sopenharmony_cistatic irqreturn_t wm831x_isink_irq(int irq, void *data)
9962306a36Sopenharmony_ci{
10062306a36Sopenharmony_ci	struct wm831x_isink *isink = data;
10162306a36Sopenharmony_ci
10262306a36Sopenharmony_ci	regulator_notifier_call_chain(isink->regulator,
10362306a36Sopenharmony_ci				      REGULATOR_EVENT_OVER_CURRENT,
10462306a36Sopenharmony_ci				      NULL);
10562306a36Sopenharmony_ci
10662306a36Sopenharmony_ci	return IRQ_HANDLED;
10762306a36Sopenharmony_ci}
10862306a36Sopenharmony_ci
10962306a36Sopenharmony_ci
11062306a36Sopenharmony_cistatic int wm831x_isink_probe(struct platform_device *pdev)
11162306a36Sopenharmony_ci{
11262306a36Sopenharmony_ci	struct wm831x *wm831x = dev_get_drvdata(pdev->dev.parent);
11362306a36Sopenharmony_ci	struct wm831x_pdata *pdata = dev_get_platdata(wm831x->dev);
11462306a36Sopenharmony_ci	struct wm831x_isink *isink;
11562306a36Sopenharmony_ci	int id = pdev->id % ARRAY_SIZE(pdata->isink);
11662306a36Sopenharmony_ci	struct regulator_config config = { };
11762306a36Sopenharmony_ci	struct resource *res;
11862306a36Sopenharmony_ci	int ret, irq;
11962306a36Sopenharmony_ci
12062306a36Sopenharmony_ci	dev_dbg(&pdev->dev, "Probing ISINK%d\n", id + 1);
12162306a36Sopenharmony_ci
12262306a36Sopenharmony_ci	if (pdata == NULL || pdata->isink[id] == NULL)
12362306a36Sopenharmony_ci		return -ENODEV;
12462306a36Sopenharmony_ci
12562306a36Sopenharmony_ci	isink = devm_kzalloc(&pdev->dev, sizeof(struct wm831x_isink),
12662306a36Sopenharmony_ci			     GFP_KERNEL);
12762306a36Sopenharmony_ci	if (!isink)
12862306a36Sopenharmony_ci		return -ENOMEM;
12962306a36Sopenharmony_ci
13062306a36Sopenharmony_ci	isink->wm831x = wm831x;
13162306a36Sopenharmony_ci
13262306a36Sopenharmony_ci	res = platform_get_resource(pdev, IORESOURCE_REG, 0);
13362306a36Sopenharmony_ci	if (res == NULL) {
13462306a36Sopenharmony_ci		dev_err(&pdev->dev, "No REG resource\n");
13562306a36Sopenharmony_ci		ret = -EINVAL;
13662306a36Sopenharmony_ci		goto err;
13762306a36Sopenharmony_ci	}
13862306a36Sopenharmony_ci	isink->reg = res->start;
13962306a36Sopenharmony_ci
14062306a36Sopenharmony_ci	/* For current parts this is correct; probably need to revisit
14162306a36Sopenharmony_ci	 * in future.
14262306a36Sopenharmony_ci	 */
14362306a36Sopenharmony_ci	snprintf(isink->name, sizeof(isink->name), "ISINK%d", id + 1);
14462306a36Sopenharmony_ci	isink->desc.name = isink->name;
14562306a36Sopenharmony_ci	isink->desc.id = id;
14662306a36Sopenharmony_ci	isink->desc.ops = &wm831x_isink_ops;
14762306a36Sopenharmony_ci	isink->desc.type = REGULATOR_CURRENT;
14862306a36Sopenharmony_ci	isink->desc.owner = THIS_MODULE;
14962306a36Sopenharmony_ci	isink->desc.curr_table = wm831x_isinkv_values,
15062306a36Sopenharmony_ci	isink->desc.n_current_limits = ARRAY_SIZE(wm831x_isinkv_values),
15162306a36Sopenharmony_ci	isink->desc.csel_reg = isink->reg,
15262306a36Sopenharmony_ci	isink->desc.csel_mask = WM831X_CS1_ISEL_MASK,
15362306a36Sopenharmony_ci
15462306a36Sopenharmony_ci	config.dev = pdev->dev.parent;
15562306a36Sopenharmony_ci	config.init_data = pdata->isink[id];
15662306a36Sopenharmony_ci	config.driver_data = isink;
15762306a36Sopenharmony_ci	config.regmap = wm831x->regmap;
15862306a36Sopenharmony_ci
15962306a36Sopenharmony_ci	isink->regulator = devm_regulator_register(&pdev->dev, &isink->desc,
16062306a36Sopenharmony_ci						   &config);
16162306a36Sopenharmony_ci	if (IS_ERR(isink->regulator)) {
16262306a36Sopenharmony_ci		ret = PTR_ERR(isink->regulator);
16362306a36Sopenharmony_ci		dev_err(wm831x->dev, "Failed to register ISINK%d: %d\n",
16462306a36Sopenharmony_ci			id + 1, ret);
16562306a36Sopenharmony_ci		goto err;
16662306a36Sopenharmony_ci	}
16762306a36Sopenharmony_ci
16862306a36Sopenharmony_ci	irq = wm831x_irq(wm831x, platform_get_irq(pdev, 0));
16962306a36Sopenharmony_ci	ret = devm_request_threaded_irq(&pdev->dev, irq, NULL,
17062306a36Sopenharmony_ci					wm831x_isink_irq,
17162306a36Sopenharmony_ci					IRQF_TRIGGER_RISING | IRQF_ONESHOT,
17262306a36Sopenharmony_ci					isink->name,
17362306a36Sopenharmony_ci					isink);
17462306a36Sopenharmony_ci	if (ret != 0) {
17562306a36Sopenharmony_ci		dev_err(&pdev->dev, "Failed to request ISINK IRQ %d: %d\n",
17662306a36Sopenharmony_ci			irq, ret);
17762306a36Sopenharmony_ci		goto err;
17862306a36Sopenharmony_ci	}
17962306a36Sopenharmony_ci
18062306a36Sopenharmony_ci	platform_set_drvdata(pdev, isink);
18162306a36Sopenharmony_ci
18262306a36Sopenharmony_ci	return 0;
18362306a36Sopenharmony_ci
18462306a36Sopenharmony_cierr:
18562306a36Sopenharmony_ci	return ret;
18662306a36Sopenharmony_ci}
18762306a36Sopenharmony_ci
18862306a36Sopenharmony_cistatic struct platform_driver wm831x_isink_driver = {
18962306a36Sopenharmony_ci	.probe = wm831x_isink_probe,
19062306a36Sopenharmony_ci	.driver		= {
19162306a36Sopenharmony_ci		.name	= "wm831x-isink",
19262306a36Sopenharmony_ci		.probe_type = PROBE_PREFER_ASYNCHRONOUS,
19362306a36Sopenharmony_ci	},
19462306a36Sopenharmony_ci};
19562306a36Sopenharmony_ci
19662306a36Sopenharmony_cistatic int __init wm831x_isink_init(void)
19762306a36Sopenharmony_ci{
19862306a36Sopenharmony_ci	int ret;
19962306a36Sopenharmony_ci	ret = platform_driver_register(&wm831x_isink_driver);
20062306a36Sopenharmony_ci	if (ret != 0)
20162306a36Sopenharmony_ci		pr_err("Failed to register WM831x ISINK driver: %d\n", ret);
20262306a36Sopenharmony_ci
20362306a36Sopenharmony_ci	return ret;
20462306a36Sopenharmony_ci}
20562306a36Sopenharmony_cisubsys_initcall(wm831x_isink_init);
20662306a36Sopenharmony_ci
20762306a36Sopenharmony_cistatic void __exit wm831x_isink_exit(void)
20862306a36Sopenharmony_ci{
20962306a36Sopenharmony_ci	platform_driver_unregister(&wm831x_isink_driver);
21062306a36Sopenharmony_ci}
21162306a36Sopenharmony_cimodule_exit(wm831x_isink_exit);
21262306a36Sopenharmony_ci
21362306a36Sopenharmony_ci/* Module information */
21462306a36Sopenharmony_ciMODULE_AUTHOR("Mark Brown");
21562306a36Sopenharmony_ciMODULE_DESCRIPTION("WM831x current sink driver");
21662306a36Sopenharmony_ciMODULE_LICENSE("GPL");
21762306a36Sopenharmony_ciMODULE_ALIAS("platform:wm831x-isink");
218