[linux-2.6-block.git] / drivers / input / misc / kxtj9.c

// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (C) 2011 Kionix, Inc.
 * Written by Chris Hudson <chudson@kionix.com>
 */

#include <linux/delay.h>
#include <linux/i2c.h>
#include <linux/input.h>
#include <linux/interrupt.h>
#include <linux/module.h>
#include <linux/slab.h>
#include <linux/input/kxtj9.h>

#define NAME			"kxtj9"
#define G_MAX			8000
/* OUTPUT REGISTERS */
#define XOUT_L			0x06
#define WHO_AM_I		0x0F
/* CONTROL REGISTERS */
#define INT_REL			0x1A
#define CTRL_REG1		0x1B
#define INT_CTRL1		0x1E
#define DATA_CTRL		0x21
/* CONTROL REGISTER 1 BITS */
#define PC1_OFF			0x7F
#define PC1_ON			(1 << 7)
/* Data ready function enable bit: set during probe if using irq mode */
#define DRDYE			(1 << 5)
/* DATA CONTROL REGISTER BITS */
#define ODR12_5F		0
#define ODR25F			1
#define ODR50F			2
#define ODR100F		3
#define ODR200F		4
#define ODR400F		5
#define ODR800F		6
/* INTERRUPT CONTROL REGISTER 1 BITS */
/* Set these during probe if using irq mode */
#define KXTJ9_IEL		(1 << 3)
#define KXTJ9_IEA		(1 << 4)
#define KXTJ9_IEN		(1 << 5)
/* INPUT_ABS CONSTANTS */
#define FUZZ			3
#define FLAT			3
/* RESUME STATE INDICES */
#define RES_DATA_CTRL		0
#define RES_CTRL_REG1		1
#define RES_INT_CTRL1		2
#define RESUME_ENTRIES		3

/*
 * The following table lists the maximum appropriate poll interval for each
 * available output data rate.
 */
static const struct {
	unsigned int cutoff;
	u8 mask;
} kxtj9_odr_table[] = {
	{ 3,	ODR800F },
	{ 5,	ODR400F },
	{ 10,	ODR200F },
	{ 20,	ODR100F },
	{ 40,	ODR50F  },
	{ 80,	ODR25F  },
	{ 0,	ODR12_5F},
};

struct kxtj9_data {
	struct i2c_client *client;
	struct kxtj9_platform_data pdata;
	struct input_dev *input_dev;
	unsigned int last_poll_interval;
	u8 shift;
	u8 ctrl_reg1;
	u8 data_ctrl;
	u8 int_ctrl;
};

static int kxtj9_i2c_read(struct kxtj9_data *tj9, u8 addr, u8 *data, int len)
{
	struct i2c_msg msgs[] = {
		{
			.addr = tj9->client->addr,
			.flags = tj9->client->flags,
			.len = 1,
			.buf = &addr,
		},
		{
			.addr = tj9->client->addr,
			.flags = tj9->client->flags | I2C_M_RD,
			.len = len,
			.buf = data,
		},
	};

	return i2c_transfer(tj9->client->adapter, msgs, 2);
}

static void kxtj9_report_acceleration_data(struct kxtj9_data *tj9)
{
	s16 acc_data[3]; /* Data bytes from hardware xL, xH, yL, yH, zL, zH */
	s16 x, y, z;
	int err;

	err = kxtj9_i2c_read(tj9, XOUT_L, (u8 *)acc_data, 6);
	if (err < 0)
		dev_err(&tj9->client->dev, "accelerometer data read failed\n");

	x = le16_to_cpu(acc_data[tj9->pdata.axis_map_x]);
	y = le16_to_cpu(acc_data[tj9->pdata.axis_map_y]);
	z = le16_to_cpu(acc_data[tj9->pdata.axis_map_z]);

	x >>= tj9->shift;
	y >>= tj9->shift;
	z >>= tj9->shift;

	input_report_abs(tj9->input_dev, ABS_X, tj9->pdata.negate_x ? -x : x);
	input_report_abs(tj9->input_dev, ABS_Y, tj9->pdata.negate_y ? -y : y);
	input_report_abs(tj9->input_dev, ABS_Z, tj9->pdata.negate_z ? -z : z);
	input_sync(tj9->input_dev);
}

static irqreturn_t kxtj9_isr(int irq, void *dev)
{
	struct kxtj9_data *tj9 = dev;
	int err;

	/* data ready is the only possible interrupt type */
	kxtj9_report_acceleration_data(tj9);

	err = i2c_smbus_read_byte_data(tj9->client, INT_REL);
	if (err < 0)
		dev_err(&tj9->client->dev,
			"error clearing interrupt status: %d\n", err);

	return IRQ_HANDLED;
}

static int kxtj9_update_g_range(struct kxtj9_data *tj9, u8 new_g_range)
{
	switch (new_g_range) {
	case KXTJ9_G_2G:
		tj9->shift = 4;
		break;
	case KXTJ9_G_4G:
		tj9->shift = 3;
		break;
	case KXTJ9_G_8G:
		tj9->shift = 2;
		break;
	default:
		return -EINVAL;
	}

	tj9->ctrl_reg1 &= 0xe7;
	tj9->ctrl_reg1 |= new_g_range;

	return 0;
}

static int kxtj9_update_odr(struct kxtj9_data *tj9, unsigned int poll_interval)
{
	int err;
	int i;

	/* Use the lowest ODR that can support the requested poll interval */
	for (i = 0; i < ARRAY_SIZE(kxtj9_odr_table); i++) {
		tj9->data_ctrl = kxtj9_odr_table[i].mask;
		if (poll_interval < kxtj9_odr_table[i].cutoff)
			break;
	}

	err = i2c_smbus_write_byte_data(tj9->client, CTRL_REG1, 0);
	if (err < 0)
		return err;

	err = i2c_smbus_write_byte_data(tj9->client, DATA_CTRL, tj9->data_ctrl);
	if (err < 0)
		return err;

	err = i2c_smbus_write_byte_data(tj9->client, CTRL_REG1, tj9->ctrl_reg1);
	if (err < 0)
		return err;

	return 0;
}

static int kxtj9_device_power_on(struct kxtj9_data *tj9)
{
	if (tj9->pdata.power_on)
		return tj9->pdata.power_on();

	return 0;
}

static void kxtj9_device_power_off(struct kxtj9_data *tj9)
{
	int err;

	tj9->ctrl_reg1 &= PC1_OFF;
	err = i2c_smbus_write_byte_data(tj9->client, CTRL_REG1, tj9->ctrl_reg1);
	if (err < 0)
		dev_err(&tj9->client->dev, "soft power off failed\n");

	if (tj9->pdata.power_off)
		tj9->pdata.power_off();
}

static int kxtj9_enable(struct kxtj9_data *tj9)
{
	int err;

	err = kxtj9_device_power_on(tj9);
	if (err < 0)
		return err;

	/* ensure that PC1 is cleared before updating control registers */
	err = i2c_smbus_write_byte_data(tj9->client, CTRL_REG1, 0);
	if (err < 0)
		return err;

	/* only write INT_CTRL_REG1 if in irq mode */
	if (tj9->client->irq) {
		err = i2c_smbus_write_byte_data(tj9->client,
						INT_CTRL1, tj9->int_ctrl);
		if (err < 0)
			return err;
	}

	err = kxtj9_update_g_range(tj9, tj9->pdata.g_range);
	if (err < 0)
		return err;

	/* turn on outputs */
	tj9->ctrl_reg1 |= PC1_ON;
	err = i2c_smbus_write_byte_data(tj9->client, CTRL_REG1, tj9->ctrl_reg1);
	if (err < 0)
		return err;

	err = kxtj9_update_odr(tj9, tj9->last_poll_interval);
	if (err < 0)
		return err;

	/* clear initial interrupt if in irq mode */
	if (tj9->client->irq) {
		err = i2c_smbus_read_byte_data(tj9->client, INT_REL);
		if (err < 0) {
			dev_err(&tj9->client->dev,
				"error clearing interrupt: %d\n", err);
			goto fail;
		}
	}

	return 0;

fail:
	kxtj9_device_power_off(tj9);
	return err;
}

static void kxtj9_disable(struct kxtj9_data *tj9)
{
	kxtj9_device_power_off(tj9);
}

static int kxtj9_input_open(struct input_dev *input)
{
	struct kxtj9_data *tj9 = input_get_drvdata(input);

	return kxtj9_enable(tj9);
}

static void kxtj9_input_close(struct input_dev *dev)
{
	struct kxtj9_data *tj9 = input_get_drvdata(dev);

	kxtj9_disable(tj9);
}

/*
 * When IRQ mode is selected, we need to provide an interface to allow the user
 * to change the output data rate of the part.  For consistency, we are using
 * the set_poll method, which accepts a poll interval in milliseconds, and then
 * calls update_odr() while passing this value as an argument.  In IRQ mode, the
 * data outputs will not be read AT the requested poll interval, rather, the
 * lowest ODR that can support the requested interval.  The client application
 * will be responsible for retrieving data from the input node at the desired
 * interval.
 */

/* Returns currently selected poll interval (in ms) */
static ssize_t kxtj9_get_poll(struct device *dev,
				struct device_attribute *attr, char *buf)
{
	struct i2c_client *client = to_i2c_client(dev);
	struct kxtj9_data *tj9 = i2c_get_clientdata(client);

	return sprintf(buf, "%d\n", tj9->last_poll_interval);
}

/* Allow users to select a new poll interval (in ms) */
static ssize_t kxtj9_set_poll(struct device *dev, struct device_attribute *attr,
						const char *buf, size_t count)
{
	struct i2c_client *client = to_i2c_client(dev);
	struct kxtj9_data *tj9 = i2c_get_clientdata(client);
	struct input_dev *input_dev = tj9->input_dev;
	unsigned int interval;
	int error;

	error = kstrtouint(buf, 10, &interval);
	if (error < 0)
		return error;

	/* Lock the device to prevent races with open/close (and itself) */
	guard(mutex)(&input_dev->mutex);
	guard(disable_irq)(&client->irq);

	/*
	 * Set current interval to the greater of the minimum interval or
	 * the requested interval
	 */
	tj9->last_poll_interval = max(interval, tj9->pdata.min_interval);

	kxtj9_update_odr(tj9, tj9->last_poll_interval);

	return count;
}

static DEVICE_ATTR(poll, S_IRUGO|S_IWUSR, kxtj9_get_poll, kxtj9_set_poll);

static struct attribute *kxtj9_attrs[] = {
	&dev_attr_poll.attr,
	NULL
};

static umode_t kxtj9_attr_is_visible(struct kobject *kobj,
				     struct attribute *attr, int n)
{
	struct device *dev = kobj_to_dev(kobj);
	struct i2c_client *client = to_i2c_client(dev);

	return client->irq ? attr->mode : 0;
}

static struct attribute_group kxtj9_group = {
	.attrs = kxtj9_attrs,
	.is_visible = kxtj9_attr_is_visible,
};
__ATTRIBUTE_GROUPS(kxtj9);

static void kxtj9_poll(struct input_dev *input)
{
	struct kxtj9_data *tj9 = input_get_drvdata(input);
	unsigned int poll_interval = input_get_poll_interval(input);

	kxtj9_report_acceleration_data(tj9);

	if (poll_interval != tj9->last_poll_interval) {
		kxtj9_update_odr(tj9, poll_interval);
		tj9->last_poll_interval = poll_interval;
	}
}

static void kxtj9_platform_exit(void *data)
{
	struct kxtj9_data *tj9 = data;

	if (tj9->pdata.exit)
		tj9->pdata.exit();
}

static int kxtj9_verify(struct kxtj9_data *tj9)
{
	int retval;

	retval = kxtj9_device_power_on(tj9);
	if (retval < 0)
		return retval;

	retval = i2c_smbus_read_byte_data(tj9->client, WHO_AM_I);
	if (retval < 0) {
		dev_err(&tj9->client->dev, "read err int source\n");
		goto out;
	}

	retval = (retval != 0x07 && retval != 0x08) ? -EIO : 0;

out:
	kxtj9_device_power_off(tj9);
	return retval;
}

static int kxtj9_probe(struct i2c_client *client)
{
	const struct kxtj9_platform_data *pdata =
			dev_get_platdata(&client->dev);
	struct kxtj9_data *tj9;
	struct input_dev *input_dev;
	int err;

	if (!i2c_check_functionality(client->adapter,
				I2C_FUNC_I2C | I2C_FUNC_SMBUS_BYTE_DATA)) {
		dev_err(&client->dev, "client is not i2c capable\n");
		return -ENXIO;
	}

	if (!pdata) {
		dev_err(&client->dev, "platform data is NULL; exiting\n");
		return -EINVAL;
	}

	tj9 = devm_kzalloc(&client->dev, sizeof(*tj9), GFP_KERNEL);
	if (!tj9) {
		dev_err(&client->dev,
			"failed to allocate memory for module data\n");
		return -ENOMEM;
	}

	tj9->client = client;
	tj9->pdata = *pdata;

	if (pdata->init) {
		err = pdata->init();
		if (err < 0)
			return err;
	}

	err = devm_add_action_or_reset(&client->dev, kxtj9_platform_exit, tj9);
	if (err)
		return err;

	err = kxtj9_verify(tj9);
	if (err < 0) {
		dev_err(&client->dev, "device not recognized\n");
		return err;
	}

	i2c_set_clientdata(client, tj9);

	tj9->ctrl_reg1 = tj9->pdata.res_12bit | tj9->pdata.g_range;
	tj9->last_poll_interval = tj9->pdata.init_interval;

	input_dev = devm_input_allocate_device(&client->dev);
	if (!input_dev) {
		dev_err(&client->dev, "input device allocate failed\n");
		return -ENOMEM;
	}

	input_set_drvdata(input_dev, tj9);
	tj9->input_dev = input_dev;

	input_dev->name = "kxtj9_accel";
	input_dev->id.bustype = BUS_I2C;

	input_dev->open = kxtj9_input_open;
	input_dev->close = kxtj9_input_close;

	input_set_abs_params(input_dev, ABS_X, -G_MAX, G_MAX, FUZZ, FLAT);
	input_set_abs_params(input_dev, ABS_Y, -G_MAX, G_MAX, FUZZ, FLAT);
	input_set_abs_params(input_dev, ABS_Z, -G_MAX, G_MAX, FUZZ, FLAT);

	if (client->irq <= 0) {
		err = input_setup_polling(input_dev, kxtj9_poll);
		if (err)
			return err;
	}

	err = input_register_device(input_dev);
	if (err) {
		dev_err(&client->dev,
			"unable to register input polled device %s: %d\n",
			input_dev->name, err);
		return err;
	}

	if (client->irq) {
		/* If in irq mode, populate INT_CTRL_REG1 and enable DRDY. */
		tj9->int_ctrl |= KXTJ9_IEN | KXTJ9_IEA | KXTJ9_IEL;
		tj9->ctrl_reg1 |= DRDYE;

		err = devm_request_threaded_irq(&client->dev, client->irq,
						NULL, kxtj9_isr,
						IRQF_TRIGGER_RISING |
							IRQF_ONESHOT,
						"kxtj9-irq", tj9);
		if (err) {
			dev_err(&client->dev, "request irq failed: %d\n", err);
			return err;
		}
	}

	return 0;
}

static int kxtj9_suspend(struct device *dev)
{
	struct i2c_client *client = to_i2c_client(dev);
	struct kxtj9_data *tj9 = i2c_get_clientdata(client);
	struct input_dev *input_dev = tj9->input_dev;

	guard(mutex)(&input_dev->mutex);

	if (input_device_enabled(input_dev))
		kxtj9_disable(tj9);

	return 0;
}

static int kxtj9_resume(struct device *dev)
{
	struct i2c_client *client = to_i2c_client(dev);
	struct kxtj9_data *tj9 = i2c_get_clientdata(client);
	struct input_dev *input_dev = tj9->input_dev;

	guard(mutex)(&input_dev->mutex);

	if (input_device_enabled(input_dev))
		kxtj9_enable(tj9);

	return 0;
}

static DEFINE_SIMPLE_DEV_PM_OPS(kxtj9_pm_ops, kxtj9_suspend, kxtj9_resume);

static const struct i2c_device_id kxtj9_id[] = {
	{ NAME },
	{ }
};

MODULE_DEVICE_TABLE(i2c, kxtj9_id);

static struct i2c_driver kxtj9_driver = {
	.driver = {
		.name		= NAME,
		.dev_groups	= kxtj9_groups,
		.pm		= pm_sleep_ptr(&kxtj9_pm_ops),
	},
	.probe		= kxtj9_probe,
	.id_table	= kxtj9_id,
};

module_i2c_driver(kxtj9_driver);

MODULE_DESCRIPTION("KXTJ9 accelerometer driver");
MODULE_AUTHOR("Chris Hudson <chudson@kionix.com>");
MODULE_LICENSE("GPL");
Commit	Line	Data
45051539	1	// SPDX-License-Identifier: GPL-2.0-only
e8e70d83 CH	2	/*
	3	* Copyright (C) 2011 Kionix, Inc.
	4	* Written by Chris Hudson <chudson@kionix.com>
e8e70d83 CH	5	*/
	6
	7	#include <linux/delay.h>
	8	#include <linux/i2c.h>
	9	#include <linux/input.h>
	10	#include <linux/interrupt.h>
2501ec97	11	#include <linux/module.h>
e8e70d83 CH	12	#include <linux/slab.h>
e8e70d83 CH	13	#include <linux/input/kxtj9.h>
e8e70d83 CH	14
	15	#define NAME "kxtj9"
	16	#define G_MAX 8000
	17	/* OUTPUT REGISTERS */
	18	#define XOUT_L 0x06
	19	#define WHO_AM_I 0x0F
	20	/* CONTROL REGISTERS */
	21	#define INT_REL 0x1A
	22	#define CTRL_REG1 0x1B
	23	#define INT_CTRL1 0x1E
	24	#define DATA_CTRL 0x21
	25	/* CONTROL REGISTER 1 BITS */
	26	#define PC1_OFF 0x7F
	27	#define PC1_ON (1 << 7)
515ef92b	28	/* Data ready function enable bit: set during probe if using irq mode */
e8e70d83	29	#define DRDYE (1 << 5)
04391660 CH	30	/* DATA CONTROL REGISTER BITS */
	31	#define ODR12_5F 0
	32	#define ODR25F 1
	33	#define ODR50F 2
	34	#define ODR100F 3
	35	#define ODR200F 4
	36	#define ODR400F 5
	37	#define ODR800F 6
e8e70d83 CH	38	/* INTERRUPT CONTROL REGISTER 1 BITS */
	39	/* Set these during probe if using irq mode */
	40	#define KXTJ9_IEL (1 << 3)
	41	#define KXTJ9_IEA (1 << 4)
	42	#define KXTJ9_IEN (1 << 5)
	43	/* INPUT_ABS CONSTANTS */
	44	#define FUZZ 3
	45	#define FLAT 3
	46	/* RESUME STATE INDICES */
	47	#define RES_DATA_CTRL 0
	48	#define RES_CTRL_REG1 1
	49	#define RES_INT_CTRL1 2
	50	#define RESUME_ENTRIES 3
	51
	52	/*
	53	* The following table lists the maximum appropriate poll interval for each
	54	* available output data rate.
	55	*/
	56	static const struct {
	57	unsigned int cutoff;
	58	u8 mask;
	59	} kxtj9_odr_table[] = {
	60	{ 3, ODR800F },
	61	{ 5, ODR400F },
	62	{ 10, ODR200F },
	63	{ 20, ODR100F },
	64	{ 40, ODR50F },
	65	{ 80, ODR25F },
	66	{ 0, ODR12_5F},
	67	};
	68
	69	struct kxtj9_data {
	70	struct i2c_client *client;
	71	struct kxtj9_platform_data pdata;
	72	struct input_dev *input_dev;
e8e70d83 CH	73	unsigned int last_poll_interval;
	74	u8 shift;
	75	u8 ctrl_reg1;
	76	u8 data_ctrl;
	77	u8 int_ctrl;
	78	};
	79
	80	static int kxtj9_i2c_read(struct kxtj9_data tj9, u8 addr, u8 data, int len)
	81	{
	82	struct i2c_msg msgs[] = {
	83	{
	84	.addr = tj9->client->addr,
	85	.flags = tj9->client->flags,
	86	.len = 1,
	87	.buf = &addr,
	88	},
	89	{
	90	.addr = tj9->client->addr,
	91	.flags = tj9->client->flags \| I2C_M_RD,
	92	.len = len,
	93	.buf = data,
	94	},
	95	};
	96
	97	return i2c_transfer(tj9->client->adapter, msgs, 2);
	98	}
	99
	100	static void kxtj9_report_acceleration_data(struct kxtj9_data *tj9)
	101	{
	102	s16 acc_data[3]; /* Data bytes from hardware xL, xH, yL, yH, zL, zH */
	103	s16 x, y, z;
	104	int err;
	105
	106	err = kxtj9_i2c_read(tj9, XOUT_L, (u8 *)acc_data, 6);
	107	if (err < 0)
	108	dev_err(&tj9->client->dev, "accelerometer data read failed\n");
	109
04391660 CH	110	x = le16_to_cpu(acc_data[tj9->pdata.axis_map_x]);
	111	y = le16_to_cpu(acc_data[tj9->pdata.axis_map_y]);
	112	z = le16_to_cpu(acc_data[tj9->pdata.axis_map_z]);
	113
	114	x >>= tj9->shift;
	115	y >>= tj9->shift;
	116	z >>= tj9->shift;
e8e70d83 CH	117
	118	input_report_abs(tj9->input_dev, ABS_X, tj9->pdata.negate_x ? -x : x);
	119	input_report_abs(tj9->input_dev, ABS_Y, tj9->pdata.negate_y ? -y : y);
	120	input_report_abs(tj9->input_dev, ABS_Z, tj9->pdata.negate_z ? -z : z);
	121	input_sync(tj9->input_dev);
	122	}
	123
	124	static irqreturn_t kxtj9_isr(int irq, void *dev)
	125	{
	126	struct kxtj9_data *tj9 = dev;
	127	int err;
	128
	129	/* data ready is the only possible interrupt type */
	130	kxtj9_report_acceleration_data(tj9);
	131
	132	err = i2c_smbus_read_byte_data(tj9->client, INT_REL);
	133	if (err < 0)
	134	dev_err(&tj9->client->dev,
	135	"error clearing interrupt status: %d\n", err);
	136
	137	return IRQ_HANDLED;
	138	}
	139
	140	static int kxtj9_update_g_range(struct kxtj9_data *tj9, u8 new_g_range)
	141	{
	142	switch (new_g_range) {
	143	case KXTJ9_G_2G:
	144	tj9->shift = 4;
	145	break;
	146	case KXTJ9_G_4G:
	147	tj9->shift = 3;
	148	break;
	149	case KXTJ9_G_8G:
	150	tj9->shift = 2;
	151	break;
	152	default:
	153	return -EINVAL;
	154	}
	155
	156	tj9->ctrl_reg1 &= 0xe7;
	157	tj9->ctrl_reg1 \|= new_g_range;
	158
	159	return 0;
	160	}
	161
	162	static int kxtj9_update_odr(struct kxtj9_data *tj9, unsigned int poll_interval)
	163	{
	164	int err;
	165	int i;
	166
	167	/* Use the lowest ODR that can support the requested poll interval */
	168	for (i = 0; i < ARRAY_SIZE(kxtj9_odr_table); i++) {
	169	tj9->data_ctrl = kxtj9_odr_table[i].mask;
	170	if (poll_interval < kxtj9_odr_table[i].cutoff)
	171	break;
	172	}
	173
	174	err = i2c_smbus_write_byte_data(tj9->client, CTRL_REG1, 0);
	175	if (err < 0)
	176	return err;
	177
	178	err = i2c_smbus_write_byte_data(tj9->client, DATA_CTRL, tj9->data_ctrl);
	179	if (err < 0)
	180	return err;
181
182	err = i2c_smbus_write_byte_data(tj9->client, CTRL_REG1, tj9->ctrl_reg1);
183	if (err < 0)
184	return err;
185
186	return 0;
187	}
188
189	static int kxtj9_device_power_on(struct kxtj9_data *tj9)
190	{
191	if (tj9->pdata.power_on)
192	return tj9->pdata.power_on();
193
194	return 0;
195	}
196
197	static void kxtj9_device_power_off(struct kxtj9_data *tj9)
198	{
199	int err;
200
201	tj9->ctrl_reg1 &= PC1_OFF;
202	err = i2c_smbus_write_byte_data(tj9->client, CTRL_REG1, tj9->ctrl_reg1);
203	if (err < 0)
204	dev_err(&tj9->client->dev, "soft power off failed\n");
205
206	if (tj9->pdata.power_off)
207	tj9->pdata.power_off();
208	}
209
210	static int kxtj9_enable(struct kxtj9_data *tj9)
211	{
212	int err;
213
214	err = kxtj9_device_power_on(tj9);
215	if (err < 0)
216	return err;
217
218	/* ensure that PC1 is cleared before updating control registers */
219	err = i2c_smbus_write_byte_data(tj9->client, CTRL_REG1, 0);
220	if (err < 0)
221	return err;
222
223	/* only write INT_CTRL_REG1 if in irq mode */
224	if (tj9->client->irq) {
225	err = i2c_smbus_write_byte_data(tj9->client,
226	INT_CTRL1, tj9->int_ctrl);
227	if (err < 0)
228	return err;
229	}
230
231	err = kxtj9_update_g_range(tj9, tj9->pdata.g_range);
232	if (err < 0)
233	return err;
234
235	/* turn on outputs */
236	tj9->ctrl_reg1 \|= PC1_ON;
237	err = i2c_smbus_write_byte_data(tj9->client, CTRL_REG1, tj9->ctrl_reg1);
238	if (err < 0)
239	return err;
240
241	err = kxtj9_update_odr(tj9, tj9->last_poll_interval);
242	if (err < 0)
243	return err;
244
245	/* clear initial interrupt if in irq mode */
246	if (tj9->client->irq) {
247	err = i2c_smbus_read_byte_data(tj9->client, INT_REL);
248	if (err < 0) {
249	dev_err(&tj9->client->dev,
250	"error clearing interrupt: %d\n", err);
251	goto fail;
252	}
253	}
254
255	return 0;
256
257	fail:
258	kxtj9_device_power_off(tj9);
259	return err;
260	}
261
262	static void kxtj9_disable(struct kxtj9_data *tj9)
263	{
264	kxtj9_device_power_off(tj9);
265	}
266
267	static int kxtj9_input_open(struct input_dev *input)
268	{
269	struct kxtj9_data *tj9 = input_get_drvdata(input);
270
271	return kxtj9_enable(tj9);
272	}
273
274	static void kxtj9_input_close(struct input_dev *dev)
275	{
276	struct kxtj9_data *tj9 = input_get_drvdata(dev);
277
278	kxtj9_disable(tj9);
279	}
280
e8e70d83 CH	281	/*
	282	* When IRQ mode is selected, we need to provide an interface to allow the user
	283	* to change the output data rate of the part. For consistency, we are using
	284	* the set_poll method, which accepts a poll interval in milliseconds, and then
	285	* calls update_odr() while passing this value as an argument. In IRQ mode, the
	286	* data outputs will not be read AT the requested poll interval, rather, the
	287	* lowest ODR that can support the requested interval. The client application
	288	* will be responsible for retrieving data from the input node at the desired
	289	* interval.
	290	*/
	291
	292	/* Returns currently selected poll interval (in ms) */
	293	static ssize_t kxtj9_get_poll(struct device *dev,
	294	struct device_attribute attr, char buf)
	295	{
	296	struct i2c_client *client = to_i2c_client(dev);
	297	struct kxtj9_data *tj9 = i2c_get_clientdata(client);
	298
	299	return sprintf(buf, "%d\n", tj9->last_poll_interval);
	300	}
	301
	302	/* Allow users to select a new poll interval (in ms) */
	303	static ssize_t kxtj9_set_poll(struct device dev, struct device_attribute attr,
	304	const char *buf, size_t count)
	305	{
	306	struct i2c_client *client = to_i2c_client(dev);
	307	struct kxtj9_data *tj9 = i2c_get_clientdata(client);
	308	struct input_dev *input_dev = tj9->input_dev;
	309	unsigned int interval;
	310	int error;
	311
	312	error = kstrtouint(buf, 10, &interval);
	313	if (error < 0)
	314	return error;
	315
	316	/* Lock the device to prevent races with open/close (and itself) */
6bbf7efc DT	317	guard(mutex)(&input_dev->mutex);
6bbf7efc DT	318	guard(disable_irq)(&client->irq);
e8e70d83 CH	319
	320	/*
	321	* Set current interval to the greater of the minimum interval or
	322	* the requested interval
	323	*/
	324	tj9->last_poll_interval = max(interval, tj9->pdata.min_interval);
	325
	326	kxtj9_update_odr(tj9, tj9->last_poll_interval);
	327
e8e70d83 CH	328	return count;
	329	}
	330
	331	static DEVICE_ATTR(poll, S_IRUGO\|S_IWUSR, kxtj9_get_poll, kxtj9_set_poll);
	332
b4f2ad7b	333	static struct attribute *kxtj9_attrs[] = {
e8e70d83 CH	334	&dev_attr_poll.attr,
	335	NULL
	336	};
	337
b4f2ad7b DT	338	static umode_t kxtj9_attr_is_visible(struct kobject *kobj,
	339	struct attribute *attr, int n)
	340	{
	341	struct device *dev = kobj_to_dev(kobj);
	342	struct i2c_client *client = to_i2c_client(dev);
	343
	344	return client->irq ? attr->mode : 0;
	345	}
	346
	347	static struct attribute_group kxtj9_group = {
	348	.attrs = kxtj9_attrs,
	349	.is_visible = kxtj9_attr_is_visible,
e8e70d83	350	};
b4f2ad7b	351	__ATTRIBUTE_GROUPS(kxtj9);
e8e70d83	352
78713dfa	353	static void kxtj9_poll(struct input_dev *input)
e8e70d83	354	{
78713dfa DT	355	struct kxtj9_data *tj9 = input_get_drvdata(input);
78713dfa DT	356	unsigned int poll_interval = input_get_poll_interval(input);
e8e70d83 CH	357
	358	kxtj9_report_acceleration_data(tj9);
	359
	360	if (poll_interval != tj9->last_poll_interval) {
	361	kxtj9_update_odr(tj9, poll_interval);
	362	tj9->last_poll_interval = poll_interval;
	363	}
	364	}
	365
e7339118	366	static void kxtj9_platform_exit(void *data)
e8e70d83	367	{
e7339118	368	struct kxtj9_data *tj9 = data;
e8e70d83	369
e7339118 DT	370	if (tj9->pdata.exit)
	371	tj9->pdata.exit();
	372	}
e8e70d83	373
5298cc4c	374	static int kxtj9_verify(struct kxtj9_data *tj9)
e8e70d83 CH	375	{
	376	int retval;
	377
	378	retval = kxtj9_device_power_on(tj9);
	379	if (retval < 0)
	380	return retval;
	381
	382	retval = i2c_smbus_read_byte_data(tj9->client, WHO_AM_I);
	383	if (retval < 0) {
	384	dev_err(&tj9->client->dev, "read err int source\n");
	385	goto out;
	386	}
	387
04391660	388	retval = (retval != 0x07 && retval != 0x08) ? -EIO : 0;
e8e70d83 CH	389
	390	out:
	391	kxtj9_device_power_off(tj9);
	392	return retval;
	393	}
	394
b05989d7	395	static int kxtj9_probe(struct i2c_client *client)
e8e70d83	396	{
c838cb3d JH	397	const struct kxtj9_platform_data *pdata =
c838cb3d JH	398	dev_get_platdata(&client->dev);
e8e70d83	399	struct kxtj9_data *tj9;
78713dfa	400	struct input_dev *input_dev;
e8e70d83 CH	401	int err;
	402
	403	if (!i2c_check_functionality(client->adapter,
	404	I2C_FUNC_I2C \| I2C_FUNC_SMBUS_BYTE_DATA)) {
	405	dev_err(&client->dev, "client is not i2c capable\n");
	406	return -ENXIO;
	407	}
	408
	409	if (!pdata) {
	410	dev_err(&client->dev, "platform data is NULL; exiting\n");
	411	return -EINVAL;
	412	}
	413
e7339118	414	tj9 = devm_kzalloc(&client->dev, sizeof(*tj9), GFP_KERNEL);
e8e70d83 CH	415	if (!tj9) {
	416	dev_err(&client->dev,
	417	"failed to allocate memory for module data\n");
	418	return -ENOMEM;
	419	}
	420
	421	tj9->client = client;
	422	tj9->pdata = *pdata;
	423
	424	if (pdata->init) {
	425	err = pdata->init();
	426	if (err < 0)
e7339118	427	return err;
e8e70d83 CH	428	}
e8e70d83 CH	429
e7339118 DT	430	err = devm_add_action_or_reset(&client->dev, kxtj9_platform_exit, tj9);
	431	if (err)
	432	return err;
	433
e8e70d83 CH	434	err = kxtj9_verify(tj9);
	435	if (err < 0) {
	436	dev_err(&client->dev, "device not recognized\n");
e7339118	437	return err;
e8e70d83 CH	438	}
	439
	440	i2c_set_clientdata(client, tj9);
	441
	442	tj9->ctrl_reg1 = tj9->pdata.res_12bit \| tj9->pdata.g_range;
04391660	443	tj9->last_poll_interval = tj9->pdata.init_interval;
e8e70d83	444
78713dfa DT	445	input_dev = devm_input_allocate_device(&client->dev);
	446	if (!input_dev) {
	447	dev_err(&client->dev, "input device allocate failed\n");
	448	return -ENOMEM;
	449	}
	450
	451	input_set_drvdata(input_dev, tj9);
	452	tj9->input_dev = input_dev;
	453
	454	input_dev->name = "kxtj9_accel";
	455	input_dev->id.bustype = BUS_I2C;
	456
	457	input_dev->open = kxtj9_input_open;
	458	input_dev->close = kxtj9_input_close;
	459
	460	input_set_abs_params(input_dev, ABS_X, -G_MAX, G_MAX, FUZZ, FLAT);
	461	input_set_abs_params(input_dev, ABS_Y, -G_MAX, G_MAX, FUZZ, FLAT);
	462	input_set_abs_params(input_dev, ABS_Z, -G_MAX, G_MAX, FUZZ, FLAT);
	463
	464	if (client->irq <= 0) {
	465	err = input_setup_polling(input_dev, kxtj9_poll);
	466	if (err)
	467	return err;
	468	}
	469
	470	err = input_register_device(input_dev);
	471	if (err) {
	472	dev_err(&client->dev,
	473	"unable to register input polled device %s: %d\n",
	474	input_dev->name, err);
	475	return err;
	476	}
	477
e8e70d83 CH	478	if (client->irq) {
	479	/* If in irq mode, populate INT_CTRL_REG1 and enable DRDY. */
	480	tj9->int_ctrl \|= KXTJ9_IEN \| KXTJ9_IEA \| KXTJ9_IEL;
	481	tj9->ctrl_reg1 \|= DRDYE;
	482
e7339118 DT	483	err = devm_request_threaded_irq(&client->dev, client->irq,
	484	NULL, kxtj9_isr,
	485	IRQF_TRIGGER_RISING \|
	486	IRQF_ONESHOT,
	487	"kxtj9-irq", tj9);
e8e70d83 CH	488	if (err) {
e8e70d83 CH	489	dev_err(&client->dev, "request irq failed: %d\n", err);
e7339118	490	return err;
e8e70d83	491	}
e8e70d83 CH	492	}
e8e70d83 CH	493
e8e70d83 CH	494	return 0;
	495	}
	496
0ac787e0	497	static int kxtj9_suspend(struct device *dev)
e8e70d83 CH	498	{
	499	struct i2c_client *client = to_i2c_client(dev);
	500	struct kxtj9_data *tj9 = i2c_get_clientdata(client);
	501	struct input_dev *input_dev = tj9->input_dev;
	502
6bbf7efc	503	guard(mutex)(&input_dev->mutex);
e8e70d83	504
d69f0a43	505	if (input_device_enabled(input_dev))
e8e70d83 CH	506	kxtj9_disable(tj9);
e8e70d83 CH	507
e8e70d83 CH	508	return 0;
	509	}
	510
0ac787e0	511	static int kxtj9_resume(struct device *dev)
e8e70d83 CH	512	{
	513	struct i2c_client *client = to_i2c_client(dev);
	514	struct kxtj9_data *tj9 = i2c_get_clientdata(client);
	515	struct input_dev *input_dev = tj9->input_dev;
e8e70d83	516
6bbf7efc	517	guard(mutex)(&input_dev->mutex);
e8e70d83	518
d69f0a43	519	if (input_device_enabled(input_dev))
e8e70d83 CH	520	kxtj9_enable(tj9);
e8e70d83 CH	521
5ccd9abd	522	return 0;
e8e70d83	523	}
e8e70d83	524
0ac787e0	525	static DEFINE_SIMPLE_DEV_PM_OPS(kxtj9_pm_ops, kxtj9_suspend, kxtj9_resume);
e8e70d83 CH	526
e8e70d83 CH	527	static const struct i2c_device_id kxtj9_id[] = {
5852f2af UKK	528	{ NAME },
5852f2af UKK	529	{ }
e8e70d83 CH	530	};
	531
	532	MODULE_DEVICE_TABLE(i2c, kxtj9_id);
	533
	534	static struct i2c_driver kxtj9_driver = {
	535	.driver = {
b4f2ad7b DT	536	.name = NAME,
	537	.dev_groups = kxtj9_groups,
	538	.pm = pm_sleep_ptr(&kxtj9_pm_ops),
e8e70d83	539	},
d8bde56d	540	.probe = kxtj9_probe,
e8e70d83 CH	541	.id_table = kxtj9_id,
	542	};
	543
1b92c1cf	544	module_i2c_driver(kxtj9_driver);
e8e70d83 CH	545
	546	MODULE_DESCRIPTION("KXTJ9 accelerometer driver");
	547	MODULE_AUTHOR("Chris Hudson <chudson@kionix.com>");
	548	MODULE_LICENSE("GPL");