162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci#include <linux/err.h>
362306a36Sopenharmony_ci#include <linux/i2c.h>
462306a36Sopenharmony_ci#include <linux/i2c-mux.h>
562306a36Sopenharmony_ci#include <linux/iio/iio.h>
662306a36Sopenharmony_ci#include <linux/module.h>
762306a36Sopenharmony_ci#include <linux/regmap.h>
862306a36Sopenharmony_ci#include <linux/pm_runtime.h>
962306a36Sopenharmony_ci
1062306a36Sopenharmony_ci#include "mpu3050.h"
1162306a36Sopenharmony_ci
1262306a36Sopenharmony_cistatic const struct regmap_config mpu3050_i2c_regmap_config = {
1362306a36Sopenharmony_ci	.reg_bits = 8,
1462306a36Sopenharmony_ci	.val_bits = 8,
1562306a36Sopenharmony_ci};
1662306a36Sopenharmony_ci
1762306a36Sopenharmony_cistatic int mpu3050_i2c_bypass_select(struct i2c_mux_core *mux, u32 chan_id)
1862306a36Sopenharmony_ci{
1962306a36Sopenharmony_ci	struct mpu3050 *mpu3050 = i2c_mux_priv(mux);
2062306a36Sopenharmony_ci
2162306a36Sopenharmony_ci	/* Just power up the device, that is all that is needed */
2262306a36Sopenharmony_ci	pm_runtime_get_sync(mpu3050->dev);
2362306a36Sopenharmony_ci	return 0;
2462306a36Sopenharmony_ci}
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_cistatic int mpu3050_i2c_bypass_deselect(struct i2c_mux_core *mux, u32 chan_id)
2762306a36Sopenharmony_ci{
2862306a36Sopenharmony_ci	struct mpu3050 *mpu3050 = i2c_mux_priv(mux);
2962306a36Sopenharmony_ci
3062306a36Sopenharmony_ci	pm_runtime_mark_last_busy(mpu3050->dev);
3162306a36Sopenharmony_ci	pm_runtime_put_autosuspend(mpu3050->dev);
3262306a36Sopenharmony_ci	return 0;
3362306a36Sopenharmony_ci}
3462306a36Sopenharmony_ci
3562306a36Sopenharmony_cistatic int mpu3050_i2c_probe(struct i2c_client *client)
3662306a36Sopenharmony_ci{
3762306a36Sopenharmony_ci	const struct i2c_device_id *id = i2c_client_get_device_id(client);
3862306a36Sopenharmony_ci	struct regmap *regmap;
3962306a36Sopenharmony_ci	const char *name;
4062306a36Sopenharmony_ci	struct mpu3050 *mpu3050;
4162306a36Sopenharmony_ci	int ret;
4262306a36Sopenharmony_ci
4362306a36Sopenharmony_ci	if (!i2c_check_functionality(client->adapter,
4462306a36Sopenharmony_ci				     I2C_FUNC_SMBUS_I2C_BLOCK))
4562306a36Sopenharmony_ci		return -EOPNOTSUPP;
4662306a36Sopenharmony_ci
4762306a36Sopenharmony_ci	if (id)
4862306a36Sopenharmony_ci		name = id->name;
4962306a36Sopenharmony_ci	else
5062306a36Sopenharmony_ci		return -ENODEV;
5162306a36Sopenharmony_ci
5262306a36Sopenharmony_ci	regmap = devm_regmap_init_i2c(client, &mpu3050_i2c_regmap_config);
5362306a36Sopenharmony_ci	if (IS_ERR(regmap)) {
5462306a36Sopenharmony_ci		dev_err(&client->dev, "Failed to register i2c regmap: %pe\n",
5562306a36Sopenharmony_ci			regmap);
5662306a36Sopenharmony_ci		return PTR_ERR(regmap);
5762306a36Sopenharmony_ci	}
5862306a36Sopenharmony_ci
5962306a36Sopenharmony_ci	ret = mpu3050_common_probe(&client->dev, regmap, client->irq, name);
6062306a36Sopenharmony_ci	if (ret)
6162306a36Sopenharmony_ci		return ret;
6262306a36Sopenharmony_ci
6362306a36Sopenharmony_ci	/* The main driver is up, now register the I2C mux */
6462306a36Sopenharmony_ci	mpu3050 = iio_priv(dev_get_drvdata(&client->dev));
6562306a36Sopenharmony_ci	mpu3050->i2cmux = i2c_mux_alloc(client->adapter, &client->dev,
6662306a36Sopenharmony_ci					1, 0, I2C_MUX_LOCKED | I2C_MUX_GATE,
6762306a36Sopenharmony_ci					mpu3050_i2c_bypass_select,
6862306a36Sopenharmony_ci					mpu3050_i2c_bypass_deselect);
6962306a36Sopenharmony_ci	/* Just fail the mux, there is no point in killing the driver */
7062306a36Sopenharmony_ci	if (!mpu3050->i2cmux)
7162306a36Sopenharmony_ci		dev_err(&client->dev, "failed to allocate I2C mux\n");
7262306a36Sopenharmony_ci	else {
7362306a36Sopenharmony_ci		mpu3050->i2cmux->priv = mpu3050;
7462306a36Sopenharmony_ci		/* Ignore failure, not critical */
7562306a36Sopenharmony_ci		i2c_mux_add_adapter(mpu3050->i2cmux, 0, 0, 0);
7662306a36Sopenharmony_ci	}
7762306a36Sopenharmony_ci
7862306a36Sopenharmony_ci	return 0;
7962306a36Sopenharmony_ci}
8062306a36Sopenharmony_ci
8162306a36Sopenharmony_cistatic void mpu3050_i2c_remove(struct i2c_client *client)
8262306a36Sopenharmony_ci{
8362306a36Sopenharmony_ci	struct iio_dev *indio_dev = dev_get_drvdata(&client->dev);
8462306a36Sopenharmony_ci	struct mpu3050 *mpu3050 = iio_priv(indio_dev);
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_ci	if (mpu3050->i2cmux)
8762306a36Sopenharmony_ci		i2c_mux_del_adapters(mpu3050->i2cmux);
8862306a36Sopenharmony_ci
8962306a36Sopenharmony_ci	mpu3050_common_remove(&client->dev);
9062306a36Sopenharmony_ci}
9162306a36Sopenharmony_ci
9262306a36Sopenharmony_ci/*
9362306a36Sopenharmony_ci * device id table is used to identify what device can be
9462306a36Sopenharmony_ci * supported by this driver
9562306a36Sopenharmony_ci */
9662306a36Sopenharmony_cistatic const struct i2c_device_id mpu3050_i2c_id[] = {
9762306a36Sopenharmony_ci	{ "mpu3050" },
9862306a36Sopenharmony_ci	{}
9962306a36Sopenharmony_ci};
10062306a36Sopenharmony_ciMODULE_DEVICE_TABLE(i2c, mpu3050_i2c_id);
10162306a36Sopenharmony_ci
10262306a36Sopenharmony_cistatic const struct of_device_id mpu3050_i2c_of_match[] = {
10362306a36Sopenharmony_ci	{ .compatible = "invensense,mpu3050", .data = "mpu3050" },
10462306a36Sopenharmony_ci	/* Deprecated vendor ID from the Input driver */
10562306a36Sopenharmony_ci	{ .compatible = "invn,mpu3050", .data = "mpu3050" },
10662306a36Sopenharmony_ci	{ },
10762306a36Sopenharmony_ci};
10862306a36Sopenharmony_ciMODULE_DEVICE_TABLE(of, mpu3050_i2c_of_match);
10962306a36Sopenharmony_ci
11062306a36Sopenharmony_cistatic struct i2c_driver mpu3050_i2c_driver = {
11162306a36Sopenharmony_ci	.probe = mpu3050_i2c_probe,
11262306a36Sopenharmony_ci	.remove = mpu3050_i2c_remove,
11362306a36Sopenharmony_ci	.id_table = mpu3050_i2c_id,
11462306a36Sopenharmony_ci	.driver = {
11562306a36Sopenharmony_ci		.of_match_table = mpu3050_i2c_of_match,
11662306a36Sopenharmony_ci		.name = "mpu3050-i2c",
11762306a36Sopenharmony_ci		.pm = pm_ptr(&mpu3050_dev_pm_ops),
11862306a36Sopenharmony_ci	},
11962306a36Sopenharmony_ci};
12062306a36Sopenharmony_cimodule_i2c_driver(mpu3050_i2c_driver);
12162306a36Sopenharmony_ci
12262306a36Sopenharmony_ciMODULE_AUTHOR("Linus Walleij");
12362306a36Sopenharmony_ciMODULE_DESCRIPTION("Invensense MPU3050 gyroscope driver");
12462306a36Sopenharmony_ciMODULE_LICENSE("GPL");
125