162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0+
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright (c) 2021 Richtek Technology Corp.
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Author: ChiYuan Huang <cy_huang@richtek.com>
662306a36Sopenharmony_ci */
762306a36Sopenharmony_ci
862306a36Sopenharmony_ci#include <linux/gpio/consumer.h>
962306a36Sopenharmony_ci#include <linux/i2c.h>
1062306a36Sopenharmony_ci#include <linux/kernel.h>
1162306a36Sopenharmony_ci#include <linux/mfd/core.h>
1262306a36Sopenharmony_ci#include <linux/module.h>
1362306a36Sopenharmony_ci#include <linux/regmap.h>
1462306a36Sopenharmony_ci
1562306a36Sopenharmony_ci#define RT4831_REG_REVISION	0x01
1662306a36Sopenharmony_ci#define RT4831_REG_ENABLE	0x08
1762306a36Sopenharmony_ci#define RT4831_REG_I2CPROT	0x15
1862306a36Sopenharmony_ci
1962306a36Sopenharmony_ci#define RICHTEK_VENDOR_ID	0x03
2062306a36Sopenharmony_ci#define RT4831_VID_MASK		GENMASK(1, 0)
2162306a36Sopenharmony_ci#define RT4831_RESET_MASK	BIT(7)
2262306a36Sopenharmony_ci#define RT4831_I2CSAFETMR_MASK	BIT(0)
2362306a36Sopenharmony_ci
2462306a36Sopenharmony_cistatic const struct mfd_cell rt4831_subdevs[] = {
2562306a36Sopenharmony_ci	MFD_CELL_OF("rt4831-backlight", NULL, NULL, 0, 0, "richtek,rt4831-backlight"),
2662306a36Sopenharmony_ci	MFD_CELL_NAME("rt4831-regulator")
2762306a36Sopenharmony_ci};
2862306a36Sopenharmony_ci
2962306a36Sopenharmony_cistatic bool rt4831_is_accessible_reg(struct device *dev, unsigned int reg)
3062306a36Sopenharmony_ci{
3162306a36Sopenharmony_ci	if (reg >= RT4831_REG_REVISION && reg <= RT4831_REG_I2CPROT)
3262306a36Sopenharmony_ci		return true;
3362306a36Sopenharmony_ci	return false;
3462306a36Sopenharmony_ci}
3562306a36Sopenharmony_ci
3662306a36Sopenharmony_cistatic const struct regmap_config rt4831_regmap_config = {
3762306a36Sopenharmony_ci	.reg_bits = 8,
3862306a36Sopenharmony_ci	.val_bits = 8,
3962306a36Sopenharmony_ci	.max_register = RT4831_REG_I2CPROT,
4062306a36Sopenharmony_ci
4162306a36Sopenharmony_ci	.readable_reg = rt4831_is_accessible_reg,
4262306a36Sopenharmony_ci	.writeable_reg = rt4831_is_accessible_reg,
4362306a36Sopenharmony_ci};
4462306a36Sopenharmony_ci
4562306a36Sopenharmony_cistatic int rt4831_probe(struct i2c_client *client)
4662306a36Sopenharmony_ci{
4762306a36Sopenharmony_ci	struct gpio_desc *enable_gpio;
4862306a36Sopenharmony_ci	struct regmap *regmap;
4962306a36Sopenharmony_ci	unsigned int chip_id;
5062306a36Sopenharmony_ci	int ret;
5162306a36Sopenharmony_ci
5262306a36Sopenharmony_ci	enable_gpio = devm_gpiod_get_optional(&client->dev, "enable", GPIOD_OUT_HIGH);
5362306a36Sopenharmony_ci	if (IS_ERR(enable_gpio)) {
5462306a36Sopenharmony_ci		dev_err(&client->dev, "Failed to get 'enable' GPIO\n");
5562306a36Sopenharmony_ci		return PTR_ERR(enable_gpio);
5662306a36Sopenharmony_ci	}
5762306a36Sopenharmony_ci
5862306a36Sopenharmony_ci	regmap = devm_regmap_init_i2c(client, &rt4831_regmap_config);
5962306a36Sopenharmony_ci	if (IS_ERR(regmap)) {
6062306a36Sopenharmony_ci		dev_err(&client->dev, "Failed to initialize regmap\n");
6162306a36Sopenharmony_ci		return PTR_ERR(regmap);
6262306a36Sopenharmony_ci	}
6362306a36Sopenharmony_ci
6462306a36Sopenharmony_ci	ret = regmap_read(regmap, RT4831_REG_REVISION, &chip_id);
6562306a36Sopenharmony_ci	if (ret) {
6662306a36Sopenharmony_ci		dev_err(&client->dev, "Failed to get H/W revision\n");
6762306a36Sopenharmony_ci		return ret;
6862306a36Sopenharmony_ci	}
6962306a36Sopenharmony_ci
7062306a36Sopenharmony_ci	if ((chip_id & RT4831_VID_MASK) != RICHTEK_VENDOR_ID) {
7162306a36Sopenharmony_ci		dev_err(&client->dev, "Chip vendor ID 0x%02x not matched\n", chip_id);
7262306a36Sopenharmony_ci		return -ENODEV;
7362306a36Sopenharmony_ci	}
7462306a36Sopenharmony_ci
7562306a36Sopenharmony_ci	/*
7662306a36Sopenharmony_ci	 * Used to prevent the abnormal shutdown.
7762306a36Sopenharmony_ci	 * If SCL/SDA both keep low for one second to reset HW.
7862306a36Sopenharmony_ci	 */
7962306a36Sopenharmony_ci	ret = regmap_update_bits(regmap, RT4831_REG_I2CPROT, RT4831_I2CSAFETMR_MASK,
8062306a36Sopenharmony_ci				 RT4831_I2CSAFETMR_MASK);
8162306a36Sopenharmony_ci	if (ret) {
8262306a36Sopenharmony_ci		dev_err(&client->dev, "Failed to enable I2C safety timer\n");
8362306a36Sopenharmony_ci		return ret;
8462306a36Sopenharmony_ci	}
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_ci	return devm_mfd_add_devices(&client->dev, PLATFORM_DEVID_AUTO, rt4831_subdevs,
8762306a36Sopenharmony_ci				    ARRAY_SIZE(rt4831_subdevs), NULL, 0, NULL);
8862306a36Sopenharmony_ci}
8962306a36Sopenharmony_ci
9062306a36Sopenharmony_cistatic void rt4831_remove(struct i2c_client *client)
9162306a36Sopenharmony_ci{
9262306a36Sopenharmony_ci	struct regmap *regmap = dev_get_regmap(&client->dev, NULL);
9362306a36Sopenharmony_ci	int ret;
9462306a36Sopenharmony_ci
9562306a36Sopenharmony_ci	/* Disable WLED and DSV outputs */
9662306a36Sopenharmony_ci	ret = regmap_update_bits(regmap, RT4831_REG_ENABLE, RT4831_RESET_MASK, RT4831_RESET_MASK);
9762306a36Sopenharmony_ci	if (ret)
9862306a36Sopenharmony_ci		dev_warn(&client->dev, "Failed to disable outputs (%pe)\n", ERR_PTR(ret));
9962306a36Sopenharmony_ci}
10062306a36Sopenharmony_ci
10162306a36Sopenharmony_cistatic const struct of_device_id __maybe_unused rt4831_of_match[] = {
10262306a36Sopenharmony_ci	{ .compatible = "richtek,rt4831", },
10362306a36Sopenharmony_ci	{}
10462306a36Sopenharmony_ci};
10562306a36Sopenharmony_ciMODULE_DEVICE_TABLE(of, rt4831_of_match);
10662306a36Sopenharmony_ci
10762306a36Sopenharmony_cistatic struct i2c_driver rt4831_driver = {
10862306a36Sopenharmony_ci	.driver = {
10962306a36Sopenharmony_ci		.name = "rt4831",
11062306a36Sopenharmony_ci		.of_match_table = rt4831_of_match,
11162306a36Sopenharmony_ci	},
11262306a36Sopenharmony_ci	.probe = rt4831_probe,
11362306a36Sopenharmony_ci	.remove = rt4831_remove,
11462306a36Sopenharmony_ci};
11562306a36Sopenharmony_cimodule_i2c_driver(rt4831_driver);
11662306a36Sopenharmony_ci
11762306a36Sopenharmony_ciMODULE_AUTHOR("ChiYuan Huang <cy_huang@richtek.com>");
11862306a36Sopenharmony_ciMODULE_LICENSE("GPL v2");
119