[GitHub/LineageOS/android_kernel_motorola_exynos9610.git] / drivers / input / keyboard / cap11xx.c

/*
 * Input driver for Microchip CAP11xx based capacitive touch sensors
 *
 * (c) 2014 Daniel Mack <linux@zonque.org>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */

#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/interrupt.h>
#include <linux/input.h>
#include <linux/leds.h>
#include <linux/of_irq.h>
#include <linux/regmap.h>
#include <linux/i2c.h>
#include <linux/gpio/consumer.h>

#define CAP11XX_REG_MAIN_CONTROL	0x00
#define CAP11XX_REG_MAIN_CONTROL_GAIN_SHIFT	(6)
#define CAP11XX_REG_MAIN_CONTROL_GAIN_MASK	(0xc0)
#define CAP11XX_REG_MAIN_CONTROL_DLSEEP		BIT(4)
#define CAP11XX_REG_GENERAL_STATUS	0x02
#define CAP11XX_REG_SENSOR_INPUT	0x03
#define CAP11XX_REG_NOISE_FLAG_STATUS	0x0a
#define CAP11XX_REG_SENOR_DELTA(X)	(0x10 + (X))
#define CAP11XX_REG_SENSITIVITY_CONTROL	0x1f
#define CAP11XX_REG_CONFIG		0x20
#define CAP11XX_REG_SENSOR_ENABLE	0x21
#define CAP11XX_REG_SENSOR_CONFIG	0x22
#define CAP11XX_REG_SENSOR_CONFIG2	0x23
#define CAP11XX_REG_SAMPLING_CONFIG	0x24
#define CAP11XX_REG_CALIBRATION		0x26
#define CAP11XX_REG_INT_ENABLE		0x27
#define CAP11XX_REG_REPEAT_RATE		0x28
#define CAP11XX_REG_MT_CONFIG		0x2a
#define CAP11XX_REG_MT_PATTERN_CONFIG	0x2b
#define CAP11XX_REG_MT_PATTERN		0x2d
#define CAP11XX_REG_RECALIB_CONFIG	0x2f
#define CAP11XX_REG_SENSOR_THRESH(X)	(0x30 + (X))
#define CAP11XX_REG_SENSOR_NOISE_THRESH	0x38
#define CAP11XX_REG_STANDBY_CHANNEL	0x40
#define CAP11XX_REG_STANDBY_CONFIG	0x41
#define CAP11XX_REG_STANDBY_SENSITIVITY	0x42
#define CAP11XX_REG_STANDBY_THRESH	0x43
#define CAP11XX_REG_CONFIG2		0x44
#define CAP11XX_REG_CONFIG2_ALT_POL	BIT(6)
#define CAP11XX_REG_SENSOR_BASE_CNT(X)	(0x50 + (X))
#define CAP11XX_REG_LED_POLARITY	0x73
#define CAP11XX_REG_LED_OUTPUT_CONTROL	0x74

#define CAP11XX_REG_LED_DUTY_CYCLE_1	0x90
#define CAP11XX_REG_LED_DUTY_CYCLE_2	0x91
#define CAP11XX_REG_LED_DUTY_CYCLE_3	0x92
#define CAP11XX_REG_LED_DUTY_CYCLE_4	0x93

#define CAP11XX_REG_LED_DUTY_MIN_MASK	(0x0f)
#define CAP11XX_REG_LED_DUTY_MIN_MASK_SHIFT	(0)
#define CAP11XX_REG_LED_DUTY_MAX_MASK	(0xf0)
#define CAP11XX_REG_LED_DUTY_MAX_MASK_SHIFT	(4)
#define CAP11XX_REG_LED_DUTY_MAX_VALUE	(15)

#define CAP11XX_REG_SENSOR_CALIB	(0xb1 + (X))
#define CAP11XX_REG_SENSOR_CALIB_LSB1	0xb9
#define CAP11XX_REG_SENSOR_CALIB_LSB2	0xba
#define CAP11XX_REG_PRODUCT_ID		0xfd
#define CAP11XX_REG_MANUFACTURER_ID	0xfe
#define CAP11XX_REG_REVISION		0xff

#define CAP11XX_MANUFACTURER_ID	0x5d

#ifdef CONFIG_LEDS_CLASS
struct cap11xx_led {
	struct cap11xx_priv *priv;
	struct led_classdev cdev;
	u32 reg;
};
#endif

struct cap11xx_priv {
	struct regmap *regmap;
	struct input_dev *idev;

	struct cap11xx_led *leds;
	int num_leds;

	/* config */
	u32 keycodes[];
};

struct cap11xx_hw_model {
	u8 product_id;
	unsigned int num_channels;
	unsigned int num_leds;
};

enum {
	CAP1106,
	CAP1126,
	CAP1188,
};

static const struct cap11xx_hw_model cap11xx_devices[] = {
	[CAP1106] = { .product_id = 0x55, .num_channels = 6, .num_leds = 0 },
	[CAP1126] = { .product_id = 0x53, .num_channels = 6, .num_leds = 2 },
	[CAP1188] = { .product_id = 0x50, .num_channels = 8, .num_leds = 8 },
};

static const struct reg_default cap11xx_reg_defaults[] = {
	{ CAP11XX_REG_MAIN_CONTROL,		0x00 },
	{ CAP11XX_REG_GENERAL_STATUS,		0x00 },
	{ CAP11XX_REG_SENSOR_INPUT,		0x00 },
	{ CAP11XX_REG_NOISE_FLAG_STATUS,	0x00 },
	{ CAP11XX_REG_SENSITIVITY_CONTROL,	0x2f },
	{ CAP11XX_REG_CONFIG,			0x20 },
	{ CAP11XX_REG_SENSOR_ENABLE,		0x3f },
	{ CAP11XX_REG_SENSOR_CONFIG,		0xa4 },
	{ CAP11XX_REG_SENSOR_CONFIG2,		0x07 },
	{ CAP11XX_REG_SAMPLING_CONFIG,		0x39 },
	{ CAP11XX_REG_CALIBRATION,		0x00 },
	{ CAP11XX_REG_INT_ENABLE,		0x3f },
	{ CAP11XX_REG_REPEAT_RATE,		0x3f },
	{ CAP11XX_REG_MT_CONFIG,		0x80 },
	{ CAP11XX_REG_MT_PATTERN_CONFIG,	0x00 },
	{ CAP11XX_REG_MT_PATTERN,		0x3f },
	{ CAP11XX_REG_RECALIB_CONFIG,		0x8a },
	{ CAP11XX_REG_SENSOR_THRESH(0),		0x40 },
	{ CAP11XX_REG_SENSOR_THRESH(1),		0x40 },
	{ CAP11XX_REG_SENSOR_THRESH(2),		0x40 },
	{ CAP11XX_REG_SENSOR_THRESH(3),		0x40 },
	{ CAP11XX_REG_SENSOR_THRESH(4),		0x40 },
	{ CAP11XX_REG_SENSOR_THRESH(5),		0x40 },
	{ CAP11XX_REG_SENSOR_NOISE_THRESH,	0x01 },
	{ CAP11XX_REG_STANDBY_CHANNEL,		0x00 },
	{ CAP11XX_REG_STANDBY_CONFIG,		0x39 },
	{ CAP11XX_REG_STANDBY_SENSITIVITY,	0x02 },
	{ CAP11XX_REG_STANDBY_THRESH,		0x40 },
	{ CAP11XX_REG_CONFIG2,			0x40 },
	{ CAP11XX_REG_LED_POLARITY,		0x00 },
	{ CAP11XX_REG_SENSOR_CALIB_LSB1,	0x00 },
	{ CAP11XX_REG_SENSOR_CALIB_LSB2,	0x00 },
};

static bool cap11xx_volatile_reg(struct device *dev, unsigned int reg)
{
	switch (reg) {
	case CAP11XX_REG_MAIN_CONTROL:
	case CAP11XX_REG_SENSOR_INPUT:
	case CAP11XX_REG_SENOR_DELTA(0):
	case CAP11XX_REG_SENOR_DELTA(1):
	case CAP11XX_REG_SENOR_DELTA(2):
	case CAP11XX_REG_SENOR_DELTA(3):
	case CAP11XX_REG_SENOR_DELTA(4):
	case CAP11XX_REG_SENOR_DELTA(5):
	case CAP11XX_REG_PRODUCT_ID:
	case CAP11XX_REG_MANUFACTURER_ID:
	case CAP11XX_REG_REVISION:
		return true;
	}

	return false;
}

static const struct regmap_config cap11xx_regmap_config = {
	.reg_bits = 8,
	.val_bits = 8,

	.max_register = CAP11XX_REG_REVISION,
	.reg_defaults = cap11xx_reg_defaults,

	.num_reg_defaults = ARRAY_SIZE(cap11xx_reg_defaults),
	.cache_type = REGCACHE_RBTREE,
	.volatile_reg = cap11xx_volatile_reg,
};

static irqreturn_t cap11xx_thread_func(int irq_num, void *data)
{
	struct cap11xx_priv *priv = data;
	unsigned int status;
	int ret, i;

	/*
	 * Deassert interrupt. This needs to be done before reading the status
	 * registers, which will not carry valid values otherwise.
	 */
	ret = regmap_update_bits(priv->regmap, CAP11XX_REG_MAIN_CONTROL, 1, 0);
	if (ret < 0)
		goto out;

	ret = regmap_read(priv->regmap, CAP11XX_REG_SENSOR_INPUT, &status);
	if (ret < 0)
		goto out;

	for (i = 0; i < priv->idev->keycodemax; i++)
		input_report_key(priv->idev, priv->keycodes[i],
				 status & (1 << i));

	input_sync(priv->idev);

out:
	return IRQ_HANDLED;
}

static int cap11xx_set_sleep(struct cap11xx_priv *priv, bool sleep)
{
	/*
	 * DLSEEP mode will turn off all LEDS, prevent this
	 */
	if (IS_ENABLED(CONFIG_LEDS_CLASS) && priv->num_leds)
		return 0;

	return regmap_update_bits(priv->regmap, CAP11XX_REG_MAIN_CONTROL,
				  CAP11XX_REG_MAIN_CONTROL_DLSEEP,
				  sleep ? CAP11XX_REG_MAIN_CONTROL_DLSEEP : 0);
}

static int cap11xx_input_open(struct input_dev *idev)
{
	struct cap11xx_priv *priv = input_get_drvdata(idev);

	return cap11xx_set_sleep(priv, false);
}

static void cap11xx_input_close(struct input_dev *idev)
{
	struct cap11xx_priv *priv = input_get_drvdata(idev);

	cap11xx_set_sleep(priv, true);
}

#ifdef CONFIG_LEDS_CLASS
static int cap11xx_led_set(struct led_classdev *cdev,
			    enum led_brightness value)
{
	struct cap11xx_led *led = container_of(cdev, struct cap11xx_led, cdev);
	struct cap11xx_priv *priv = led->priv;

	/*
	 * All LEDs share the same duty cycle as this is a HW
	 * limitation. Brightness levels per LED are either
	 * 0 (OFF) and 1 (ON).
	 */
	return regmap_update_bits(priv->regmap,
				  CAP11XX_REG_LED_OUTPUT_CONTROL,
				  BIT(led->reg),
				  value ? BIT(led->reg) : 0);
}

static int cap11xx_init_leds(struct device *dev,
			     struct cap11xx_priv *priv, int num_leds)
{
	struct device_node *node = dev->of_node, *child;
	struct cap11xx_led *led;
	int cnt = of_get_child_count(node);
	int error;

	if (!num_leds || !cnt)
		return 0;

	if (cnt > num_leds)
		return -EINVAL;

	led = devm_kcalloc(dev, cnt, sizeof(struct cap11xx_led), GFP_KERNEL);
	if (!led)
		return -ENOMEM;

	priv->leds = led;

	error = regmap_update_bits(priv->regmap,
				CAP11XX_REG_LED_OUTPUT_CONTROL, 0xff, 0);
	if (error)
		return error;

	error = regmap_update_bits(priv->regmap, CAP11XX_REG_LED_DUTY_CYCLE_4,
				CAP11XX_REG_LED_DUTY_MAX_MASK,
				CAP11XX_REG_LED_DUTY_MAX_VALUE <<
				CAP11XX_REG_LED_DUTY_MAX_MASK_SHIFT);
	if (error)
		return error;

	for_each_child_of_node(node, child) {
		u32 reg;

		led->cdev.name =
			of_get_property(child, "label", NULL) ? : child->name;
		led->cdev.default_trigger =
			of_get_property(child, "linux,default-trigger", NULL);
		led->cdev.flags = 0;
		led->cdev.brightness_set_blocking = cap11xx_led_set;
		led->cdev.max_brightness = 1;
		led->cdev.brightness = LED_OFF;

		error = of_property_read_u32(child, "reg", &reg);
		if (error != 0 || reg >= num_leds) {
			of_node_put(child);
			return -EINVAL;
		}

		led->reg = reg;
		led->priv = priv;

		error = devm_led_classdev_register(dev, &led->cdev);
		if (error) {
			of_node_put(child);
			return error;
		}

		priv->num_leds++;
		led++;
	}

	return 0;
}
#else
static int cap11xx_init_leds(struct device *dev,
			     struct cap11xx_priv *priv, int num_leds)
{
	return 0;
}
#endif

static int cap11xx_i2c_probe(struct i2c_client *i2c_client,
			     const struct i2c_device_id *id)
{
	struct device *dev = &i2c_client->dev;
	struct cap11xx_priv *priv;
	struct device_node *node;
	const struct cap11xx_hw_model *cap;
	int i, error, irq, gain = 0;
	unsigned int val, rev;
	u32 gain32;

	if (id->driver_data >= ARRAY_SIZE(cap11xx_devices)) {
		dev_err(dev, "Invalid device ID %lu\n", id->driver_data);
		return -EINVAL;
	}

	cap = &cap11xx_devices[id->driver_data];
	if (!cap || !cap->num_channels) {
		dev_err(dev, "Invalid device configuration\n");
		return -EINVAL;
	}

	priv = devm_kzalloc(dev,
			    sizeof(*priv) +
				cap->num_channels * sizeof(priv->keycodes[0]),
			    GFP_KERNEL);
	if (!priv)
		return -ENOMEM;

	priv->regmap = devm_regmap_init_i2c(i2c_client, &cap11xx_regmap_config);
	if (IS_ERR(priv->regmap))
		return PTR_ERR(priv->regmap);

	error = regmap_read(priv->regmap, CAP11XX_REG_PRODUCT_ID, &val);
	if (error)
		return error;

	if (val != cap->product_id) {
		dev_err(dev, "Product ID: Got 0x%02x, expected 0x%02x\n",
			val, cap->product_id);
		return -ENXIO;
	}

	error = regmap_read(priv->regmap, CAP11XX_REG_MANUFACTURER_ID, &val);
	if (error)
		return error;

	if (val != CAP11XX_MANUFACTURER_ID) {
		dev_err(dev, "Manufacturer ID: Got 0x%02x, expected 0x%02x\n",
			val, CAP11XX_MANUFACTURER_ID);
		return -ENXIO;
	}

	error = regmap_read(priv->regmap, CAP11XX_REG_REVISION, &rev);
	if (error < 0)
		return error;

	dev_info(dev, "CAP11XX detected, revision 0x%02x\n", rev);
	node = dev->of_node;

	if (!of_property_read_u32(node, "microchip,sensor-gain", &gain32)) {
		if (is_power_of_2(gain32) && gain32 <= 8)
			gain = ilog2(gain32);
		else
			dev_err(dev, "Invalid sensor-gain value %d\n", gain32);
	}

	if (of_property_read_bool(node, "microchip,irq-active-high")) {
		error = regmap_update_bits(priv->regmap, CAP11XX_REG_CONFIG2,
					   CAP11XX_REG_CONFIG2_ALT_POL, 0);
		if (error)
			return error;
	}

	/* Provide some useful defaults */
	for (i = 0; i < cap->num_channels; i++)
		priv->keycodes[i] = KEY_A + i;

	of_property_read_u32_array(node, "linux,keycodes",
				   priv->keycodes, cap->num_channels);

	error = regmap_update_bits(priv->regmap, CAP11XX_REG_MAIN_CONTROL,
				   CAP11XX_REG_MAIN_CONTROL_GAIN_MASK,
				   gain << CAP11XX_REG_MAIN_CONTROL_GAIN_SHIFT);
	if (error)
		return error;

	/* Disable autorepeat. The Linux input system has its own handling. */
	error = regmap_write(priv->regmap, CAP11XX_REG_REPEAT_RATE, 0);
	if (error)
		return error;

	priv->idev = devm_input_allocate_device(dev);
	if (!priv->idev)
		return -ENOMEM;

	priv->idev->name = "CAP11XX capacitive touch sensor";
	priv->idev->id.bustype = BUS_I2C;
	priv->idev->evbit[0] = BIT_MASK(EV_KEY);

	if (of_property_read_bool(node, "autorepeat"))
		__set_bit(EV_REP, priv->idev->evbit);

	for (i = 0; i < cap->num_channels; i++)
		__set_bit(priv->keycodes[i], priv->idev->keybit);

	__clear_bit(KEY_RESERVED, priv->idev->keybit);

	priv->idev->keycode = priv->keycodes;
	priv->idev->keycodesize = sizeof(priv->keycodes[0]);
	priv->idev->keycodemax = cap->num_channels;

	priv->idev->id.vendor = CAP11XX_MANUFACTURER_ID;
	priv->idev->id.product = cap->product_id;
	priv->idev->id.version = rev;

	priv->idev->open = cap11xx_input_open;
	priv->idev->close = cap11xx_input_close;

	error = cap11xx_init_leds(dev, priv, cap->num_leds);
	if (error)
		return error;

	input_set_drvdata(priv->idev, priv);

	/*
	 * Put the device in deep sleep mode for now.
	 * ->open() will bring it back once the it is actually needed.
	 */
	cap11xx_set_sleep(priv, true);

	error = input_register_device(priv->idev);
	if (error)
		return error;

	irq = irq_of_parse_and_map(node, 0);
	if (!irq) {
		dev_err(dev, "Unable to parse or map IRQ\n");
		return -ENXIO;
	}

	error = devm_request_threaded_irq(dev, irq, NULL, cap11xx_thread_func,
					  IRQF_ONESHOT, dev_name(dev), priv);
	if (error)
		return error;

	return 0;
}

static const struct of_device_id cap11xx_dt_ids[] = {
	{ .compatible = "microchip,cap1106", },
	{ .compatible = "microchip,cap1126", },
	{ .compatible = "microchip,cap1188", },
	{}
};
MODULE_DEVICE_TABLE(of, cap11xx_dt_ids);

static const struct i2c_device_id cap11xx_i2c_ids[] = {
	{ "cap1106", CAP1106 },
	{ "cap1126", CAP1126 },
	{ "cap1188", CAP1188 },
	{}
};
MODULE_DEVICE_TABLE(i2c, cap11xx_i2c_ids);

static struct i2c_driver cap11xx_i2c_driver = {
	.driver = {
		.name	= "cap11xx",
		.of_match_table = cap11xx_dt_ids,
	},
	.id_table	= cap11xx_i2c_ids,
	.probe		= cap11xx_i2c_probe,
};

module_i2c_driver(cap11xx_i2c_driver);

MODULE_DESCRIPTION("Microchip CAP11XX driver");
MODULE_AUTHOR("Daniel Mack <linux@zonque.org>");
MODULE_LICENSE("GPL v2");
Commit	Line	Data
c77fd0a4 MR	1	/*
	2	* Input driver for Microchip CAP11xx based capacitive touch sensors
	3	*
c77fd0a4 MR	4	* (c) 2014 Daniel Mack <linux@zonque.org>
	5	*
	6	* This program is free software; you can redistribute it and/or modify
	7	* it under the terms of the GNU General Public License version 2 as
	8	* published by the Free Software Foundation.
	9	*/
	10
	11	#include <linux/kernel.h>
	12	#include <linux/module.h>
	13	#include <linux/interrupt.h>
	14	#include <linux/input.h>
efe3b616	15	#include <linux/leds.h>
c77fd0a4 MR	16	#include <linux/of_irq.h>
	17	#include <linux/regmap.h>
	18	#include <linux/i2c.h>
	19	#include <linux/gpio/consumer.h>
	20
	21	#define CAP11XX_REG_MAIN_CONTROL 0x00
	22	#define CAP11XX_REG_MAIN_CONTROL_GAIN_SHIFT (6)
	23	#define CAP11XX_REG_MAIN_CONTROL_GAIN_MASK (0xc0)
	24	#define CAP11XX_REG_MAIN_CONTROL_DLSEEP BIT(4)
	25	#define CAP11XX_REG_GENERAL_STATUS 0x02
	26	#define CAP11XX_REG_SENSOR_INPUT 0x03
	27	#define CAP11XX_REG_NOISE_FLAG_STATUS 0x0a
	28	#define CAP11XX_REG_SENOR_DELTA(X) (0x10 + (X))
	29	#define CAP11XX_REG_SENSITIVITY_CONTROL 0x1f
	30	#define CAP11XX_REG_CONFIG 0x20
	31	#define CAP11XX_REG_SENSOR_ENABLE 0x21
	32	#define CAP11XX_REG_SENSOR_CONFIG 0x22
	33	#define CAP11XX_REG_SENSOR_CONFIG2 0x23
	34	#define CAP11XX_REG_SAMPLING_CONFIG 0x24
	35	#define CAP11XX_REG_CALIBRATION 0x26
	36	#define CAP11XX_REG_INT_ENABLE 0x27
	37	#define CAP11XX_REG_REPEAT_RATE 0x28
	38	#define CAP11XX_REG_MT_CONFIG 0x2a
	39	#define CAP11XX_REG_MT_PATTERN_CONFIG 0x2b
	40	#define CAP11XX_REG_MT_PATTERN 0x2d
	41	#define CAP11XX_REG_RECALIB_CONFIG 0x2f
	42	#define CAP11XX_REG_SENSOR_THRESH(X) (0x30 + (X))
	43	#define CAP11XX_REG_SENSOR_NOISE_THRESH 0x38
	44	#define CAP11XX_REG_STANDBY_CHANNEL 0x40
	45	#define CAP11XX_REG_STANDBY_CONFIG 0x41
	46	#define CAP11XX_REG_STANDBY_SENSITIVITY 0x42
	47	#define CAP11XX_REG_STANDBY_THRESH 0x43
	48	#define CAP11XX_REG_CONFIG2 0x44
6bdd2fd1	49	#define CAP11XX_REG_CONFIG2_ALT_POL BIT(6)
c77fd0a4	50	#define CAP11XX_REG_SENSOR_BASE_CNT(X) (0x50 + (X))
efe3b616 MR	51	#define CAP11XX_REG_LED_POLARITY 0x73
	52	#define CAP11XX_REG_LED_OUTPUT_CONTROL 0x74
	53
	54	#define CAP11XX_REG_LED_DUTY_CYCLE_1 0x90
	55	#define CAP11XX_REG_LED_DUTY_CYCLE_2 0x91
	56	#define CAP11XX_REG_LED_DUTY_CYCLE_3 0x92
	57	#define CAP11XX_REG_LED_DUTY_CYCLE_4 0x93
	58
	59	#define CAP11XX_REG_LED_DUTY_MIN_MASK (0x0f)
	60	#define CAP11XX_REG_LED_DUTY_MIN_MASK_SHIFT (0)
	61	#define CAP11XX_REG_LED_DUTY_MAX_MASK (0xf0)
	62	#define CAP11XX_REG_LED_DUTY_MAX_MASK_SHIFT (4)
	63	#define CAP11XX_REG_LED_DUTY_MAX_VALUE (15)
	64
c77fd0a4 MR	65	#define CAP11XX_REG_SENSOR_CALIB (0xb1 + (X))
	66	#define CAP11XX_REG_SENSOR_CALIB_LSB1 0xb9
	67	#define CAP11XX_REG_SENSOR_CALIB_LSB2 0xba
	68	#define CAP11XX_REG_PRODUCT_ID 0xfd
	69	#define CAP11XX_REG_MANUFACTURER_ID 0xfe
	70	#define CAP11XX_REG_REVISION 0xff
	71
c77fd0a4 MR	72	#define CAP11XX_MANUFACTURER_ID 0x5d
c77fd0a4 MR	73
efe3b616 MR	74	#ifdef CONFIG_LEDS_CLASS
	75	struct cap11xx_led {
	76	struct cap11xx_priv *priv;
	77	struct led_classdev cdev;
efe3b616	78	u32 reg;
efe3b616 MR	79	};
	80	#endif
	81
c77fd0a4 MR	82	struct cap11xx_priv {
	83	struct regmap *regmap;
	84	struct input_dev *idev;
	85
efe3b616 MR	86	struct cap11xx_led *leds;
	87	int num_leds;
	88
c77fd0a4	89	/* config */
7609a5e9 MR	90	u32 keycodes[];
	91	};
	92
	93	struct cap11xx_hw_model {
	94	u8 product_id;
	95	unsigned int num_channels;
efe3b616	96	unsigned int num_leds;
7609a5e9 MR	97	};
	98
	99	enum {
	100	CAP1106,
	101	CAP1126,
	102	CAP1188,
	103	};
	104
	105	static const struct cap11xx_hw_model cap11xx_devices[] = {
efe3b616 MR	106	[CAP1106] = { .product_id = 0x55, .num_channels = 6, .num_leds = 0 },
	107	[CAP1126] = { .product_id = 0x53, .num_channels = 6, .num_leds = 2 },
	108	[CAP1188] = { .product_id = 0x50, .num_channels = 8, .num_leds = 8 },
c77fd0a4 MR	109	};
	110
	111	static const struct reg_default cap11xx_reg_defaults[] = {
	112	{ CAP11XX_REG_MAIN_CONTROL, 0x00 },
	113	{ CAP11XX_REG_GENERAL_STATUS, 0x00 },
	114	{ CAP11XX_REG_SENSOR_INPUT, 0x00 },
	115	{ CAP11XX_REG_NOISE_FLAG_STATUS, 0x00 },
	116	{ CAP11XX_REG_SENSITIVITY_CONTROL, 0x2f },
	117	{ CAP11XX_REG_CONFIG, 0x20 },
	118	{ CAP11XX_REG_SENSOR_ENABLE, 0x3f },
	119	{ CAP11XX_REG_SENSOR_CONFIG, 0xa4 },
	120	{ CAP11XX_REG_SENSOR_CONFIG2, 0x07 },
	121	{ CAP11XX_REG_SAMPLING_CONFIG, 0x39 },
	122	{ CAP11XX_REG_CALIBRATION, 0x00 },
	123	{ CAP11XX_REG_INT_ENABLE, 0x3f },
	124	{ CAP11XX_REG_REPEAT_RATE, 0x3f },
	125	{ CAP11XX_REG_MT_CONFIG, 0x80 },
	126	{ CAP11XX_REG_MT_PATTERN_CONFIG, 0x00 },
	127	{ CAP11XX_REG_MT_PATTERN, 0x3f },
	128	{ CAP11XX_REG_RECALIB_CONFIG, 0x8a },
	129	{ CAP11XX_REG_SENSOR_THRESH(0), 0x40 },
	130	{ CAP11XX_REG_SENSOR_THRESH(1), 0x40 },
	131	{ CAP11XX_REG_SENSOR_THRESH(2), 0x40 },
	132	{ CAP11XX_REG_SENSOR_THRESH(3), 0x40 },
	133	{ CAP11XX_REG_SENSOR_THRESH(4), 0x40 },
	134	{ CAP11XX_REG_SENSOR_THRESH(5), 0x40 },
	135	{ CAP11XX_REG_SENSOR_NOISE_THRESH, 0x01 },
	136	{ CAP11XX_REG_STANDBY_CHANNEL, 0x00 },
	137	{ CAP11XX_REG_STANDBY_CONFIG, 0x39 },
	138	{ CAP11XX_REG_STANDBY_SENSITIVITY, 0x02 },
	139	{ CAP11XX_REG_STANDBY_THRESH, 0x40 },
	140	{ CAP11XX_REG_CONFIG2, 0x40 },
efe3b616	141	{ CAP11XX_REG_LED_POLARITY, 0x00 },
c77fd0a4 MR	142	{ CAP11XX_REG_SENSOR_CALIB_LSB1, 0x00 },
	143	{ CAP11XX_REG_SENSOR_CALIB_LSB2, 0x00 },
	144	};
	145
	146	static bool cap11xx_volatile_reg(struct device *dev, unsigned int reg)
	147	{
	148	switch (reg) {
	149	case CAP11XX_REG_MAIN_CONTROL:
	150	case CAP11XX_REG_SENSOR_INPUT:
	151	case CAP11XX_REG_SENOR_DELTA(0):
	152	case CAP11XX_REG_SENOR_DELTA(1):
	153	case CAP11XX_REG_SENOR_DELTA(2):
	154	case CAP11XX_REG_SENOR_DELTA(3):
	155	case CAP11XX_REG_SENOR_DELTA(4):
	156	case CAP11XX_REG_SENOR_DELTA(5):
	157	case CAP11XX_REG_PRODUCT_ID:
	158	case CAP11XX_REG_MANUFACTURER_ID:
	159	case CAP11XX_REG_REVISION:
	160	return true;
	161	}
	162
	163	return false;
	164	}
	165
	166	static const struct regmap_config cap11xx_regmap_config = {
	167	.reg_bits = 8,
	168	.val_bits = 8,
	169
	170	.max_register = CAP11XX_REG_REVISION,
	171	.reg_defaults = cap11xx_reg_defaults,
	172
	173	.num_reg_defaults = ARRAY_SIZE(cap11xx_reg_defaults),
	174	.cache_type = REGCACHE_RBTREE,
	175	.volatile_reg = cap11xx_volatile_reg,
	176	};
	177
	178	static irqreturn_t cap11xx_thread_func(int irq_num, void *data)
	179	{
	180	struct cap11xx_priv *priv = data;
	181	unsigned int status;
	182	int ret, i;
	183
	184	/*
	185	* Deassert interrupt. This needs to be done before reading the status
	186	* registers, which will not carry valid values otherwise.
	187	*/
	188	ret = regmap_update_bits(priv->regmap, CAP11XX_REG_MAIN_CONTROL, 1, 0);
	189	if (ret < 0)
	190	goto out;
	191
	192	ret = regmap_read(priv->regmap, CAP11XX_REG_SENSOR_INPUT, &status);
	193	if (ret < 0)
	194	goto out;
	195
7609a5e9	196	for (i = 0; i < priv->idev->keycodemax; i++)
c77fd0a4 MR	197	input_report_key(priv->idev, priv->keycodes[i],
	198	status & (1 << i));
	199
	200	input_sync(priv->idev);
	201
	202	out:
	203	return IRQ_HANDLED;
	204	}
	205
	206	static int cap11xx_set_sleep(struct cap11xx_priv *priv, bool sleep)
	207	{
efe3b616 MR	208	/*
	209	* DLSEEP mode will turn off all LEDS, prevent this
	210	*/
	211	if (IS_ENABLED(CONFIG_LEDS_CLASS) && priv->num_leds)
	212	return 0;
	213
c77fd0a4 MR	214	return regmap_update_bits(priv->regmap, CAP11XX_REG_MAIN_CONTROL,
	215	CAP11XX_REG_MAIN_CONTROL_DLSEEP,
	216	sleep ? CAP11XX_REG_MAIN_CONTROL_DLSEEP : 0);
	217	}
	218
	219	static int cap11xx_input_open(struct input_dev *idev)
	220	{
	221	struct cap11xx_priv *priv = input_get_drvdata(idev);
	222
	223	return cap11xx_set_sleep(priv, false);
	224	}
	225
	226	static void cap11xx_input_close(struct input_dev *idev)
	227	{
	228	struct cap11xx_priv *priv = input_get_drvdata(idev);
	229
	230	cap11xx_set_sleep(priv, true);
	231	}
	232
efe3b616	233	#ifdef CONFIG_LEDS_CLASS
0d523d67 DT	234	static int cap11xx_led_set(struct led_classdev *cdev,
0d523d67 DT	235	enum led_brightness value)
efe3b616	236	{
0d523d67	237	struct cap11xx_led *led = container_of(cdev, struct cap11xx_led, cdev);
efe3b616	238	struct cap11xx_priv *priv = led->priv;
efe3b616 MR	239
efe3b616 MR	240	/*
0d523d67 DT	241	* All LEDs share the same duty cycle as this is a HW
	242	* limitation. Brightness levels per LED are either
	243	* 0 (OFF) and 1 (ON).
efe3b616	244	*/
0d523d67 DT	245	return regmap_update_bits(priv->regmap,
	246	CAP11XX_REG_LED_OUTPUT_CONTROL,
	247	BIT(led->reg),
	248	value ? BIT(led->reg) : 0);
efe3b616 MR	249	}
	250
	251	static int cap11xx_init_leds(struct device *dev,
	252	struct cap11xx_priv *priv, int num_leds)
	253	{
	254	struct device_node node = dev->of_node, child;
	255	struct cap11xx_led *led;
	256	int cnt = of_get_child_count(node);
	257	int error;
	258
	259	if (!num_leds \|\| !cnt)
	260	return 0;
	261
	262	if (cnt > num_leds)
	263	return -EINVAL;
	264
	265	led = devm_kcalloc(dev, cnt, sizeof(struct cap11xx_led), GFP_KERNEL);
	266	if (!led)
	267	return -ENOMEM;
	268
	269	priv->leds = led;
	270
	271	error = regmap_update_bits(priv->regmap,
	272	CAP11XX_REG_LED_OUTPUT_CONTROL, 0xff, 0);
	273	if (error)
	274	return error;
	275
	276	error = regmap_update_bits(priv->regmap, CAP11XX_REG_LED_DUTY_CYCLE_4,
	277	CAP11XX_REG_LED_DUTY_MAX_MASK,
	278	CAP11XX_REG_LED_DUTY_MAX_VALUE <<
	279	CAP11XX_REG_LED_DUTY_MAX_MASK_SHIFT);
	280	if (error)
	281	return error;
	282
	283	for_each_child_of_node(node, child) {
	284	u32 reg;
	285
	286	led->cdev.name =
	287	of_get_property(child, "label", NULL) ? : child->name;
	288	led->cdev.default_trigger =
	289	of_get_property(child, "linux,default-trigger", NULL);
	290	led->cdev.flags = 0;
0d523d67	291	led->cdev.brightness_set_blocking = cap11xx_led_set;
efe3b616 MR	292	led->cdev.max_brightness = 1;
	293	led->cdev.brightness = LED_OFF;
	294
	295	error = of_property_read_u32(child, "reg", &reg);
497e1b3f AKC	296	if (error != 0 \|\| reg >= num_leds) {
497e1b3f AKC	297	of_node_put(child);
efe3b616	298	return -EINVAL;
497e1b3f	299	}
efe3b616 MR	300
	301	led->reg = reg;
	302	led->priv = priv;
	303
efe3b616	304	error = devm_led_classdev_register(dev, &led->cdev);
497e1b3f AKC	305	if (error) {
497e1b3f AKC	306	of_node_put(child);
efe3b616	307	return error;
497e1b3f	308	}
efe3b616 MR	309
	310	priv->num_leds++;
	311	led++;
	312	}
	313
	314	return 0;
	315	}
	316	#else
	317	static int cap11xx_init_leds(struct device *dev,
	318	struct cap11xx_priv *priv, int num_leds)
	319	{
	320	return 0;
	321	}
	322	#endif
	323
c77fd0a4 MR	324	static int cap11xx_i2c_probe(struct i2c_client *i2c_client,
	325	const struct i2c_device_id *id)
	326	{
	327	struct device *dev = &i2c_client->dev;
	328	struct cap11xx_priv *priv;
	329	struct device_node *node;
7609a5e9	330	const struct cap11xx_hw_model *cap;
c77fd0a4 MR	331	int i, error, irq, gain = 0;
c77fd0a4 MR	332	unsigned int val, rev;
7609a5e9	333	u32 gain32;
c77fd0a4	334
7609a5e9 MR	335	if (id->driver_data >= ARRAY_SIZE(cap11xx_devices)) {
	336	dev_err(dev, "Invalid device ID %lu\n", id->driver_data);
	337	return -EINVAL;
	338	}
	339
	340	cap = &cap11xx_devices[id->driver_data];
	341	if (!cap \|\| !cap->num_channels) {
	342	dev_err(dev, "Invalid device configuration\n");
	343	return -EINVAL;
	344	}
	345
	346	priv = devm_kzalloc(dev,
	347	sizeof(*priv) +
	348	cap->num_channels * sizeof(priv->keycodes[0]),
	349	GFP_KERNEL);
c77fd0a4 MR	350	if (!priv)
	351	return -ENOMEM;
	352
	353	priv->regmap = devm_regmap_init_i2c(i2c_client, &cap11xx_regmap_config);
	354	if (IS_ERR(priv->regmap))
	355	return PTR_ERR(priv->regmap);
	356
	357	error = regmap_read(priv->regmap, CAP11XX_REG_PRODUCT_ID, &val);
	358	if (error)
	359	return error;
	360
7609a5e9	361	if (val != cap->product_id) {
c77fd0a4	362	dev_err(dev, "Product ID: Got 0x%02x, expected 0x%02x\n",
7609a5e9 MR	363	val, cap->product_id);
7609a5e9 MR	364	return -ENXIO;
c77fd0a4 MR	365	}
	366
	367	error = regmap_read(priv->regmap, CAP11XX_REG_MANUFACTURER_ID, &val);
	368	if (error)
	369	return error;
	370
	371	if (val != CAP11XX_MANUFACTURER_ID) {
	372	dev_err(dev, "Manufacturer ID: Got 0x%02x, expected 0x%02x\n",
	373	val, CAP11XX_MANUFACTURER_ID);
7609a5e9	374	return -ENXIO;
c77fd0a4 MR	375	}
	376
	377	error = regmap_read(priv->regmap, CAP11XX_REG_REVISION, &rev);
	378	if (error < 0)
	379	return error;
	380
	381	dev_info(dev, "CAP11XX detected, revision 0x%02x\n", rev);
c77fd0a4 MR	382	node = dev->of_node;
	383
	384	if (!of_property_read_u32(node, "microchip,sensor-gain", &gain32)) {
	385	if (is_power_of_2(gain32) && gain32 <= 8)
	386	gain = ilog2(gain32);
	387	else
	388	dev_err(dev, "Invalid sensor-gain value %d\n", gain32);
	389	}
	390
6bdd2fd1 MR	391	if (of_property_read_bool(node, "microchip,irq-active-high")) {
	392	error = regmap_update_bits(priv->regmap, CAP11XX_REG_CONFIG2,
	393	CAP11XX_REG_CONFIG2_ALT_POL, 0);
	394	if (error)
	395	return error;
	396	}
	397
c77fd0a4	398	/* Provide some useful defaults */
7609a5e9 MR	399	for (i = 0; i < cap->num_channels; i++)
7609a5e9 MR	400	priv->keycodes[i] = KEY_A + i;
c77fd0a4 MR	401
c77fd0a4 MR	402	of_property_read_u32_array(node, "linux,keycodes",
7609a5e9	403	priv->keycodes, cap->num_channels);
c77fd0a4 MR	404
	405	error = regmap_update_bits(priv->regmap, CAP11XX_REG_MAIN_CONTROL,
	406	CAP11XX_REG_MAIN_CONTROL_GAIN_MASK,
	407	gain << CAP11XX_REG_MAIN_CONTROL_GAIN_SHIFT);
	408	if (error)
	409	return error;
	410
	411	/* Disable autorepeat. The Linux input system has its own handling. */
	412	error = regmap_write(priv->regmap, CAP11XX_REG_REPEAT_RATE, 0);
	413	if (error)
	414	return error;
	415
	416	priv->idev = devm_input_allocate_device(dev);
	417	if (!priv->idev)
	418	return -ENOMEM;
	419
	420	priv->idev->name = "CAP11XX capacitive touch sensor";
	421	priv->idev->id.bustype = BUS_I2C;
	422	priv->idev->evbit[0] = BIT_MASK(EV_KEY);
	423
	424	if (of_property_read_bool(node, "autorepeat"))
	425	__set_bit(EV_REP, priv->idev->evbit);
	426
7609a5e9	427	for (i = 0; i < cap->num_channels; i++)
c77fd0a4 MR	428	__set_bit(priv->keycodes[i], priv->idev->keybit);
	429
	430	__clear_bit(KEY_RESERVED, priv->idev->keybit);
	431
	432	priv->idev->keycode = priv->keycodes;
	433	priv->idev->keycodesize = sizeof(priv->keycodes[0]);
7609a5e9	434	priv->idev->keycodemax = cap->num_channels;
c77fd0a4 MR	435
c77fd0a4 MR	436	priv->idev->id.vendor = CAP11XX_MANUFACTURER_ID;
7609a5e9	437	priv->idev->id.product = cap->product_id;
c77fd0a4 MR	438	priv->idev->id.version = rev;
	439
	440	priv->idev->open = cap11xx_input_open;
	441	priv->idev->close = cap11xx_input_close;
	442
efe3b616 MR	443	error = cap11xx_init_leds(dev, priv, cap->num_leds);
	444	if (error)
	445	return error;
	446
c77fd0a4 MR	447	input_set_drvdata(priv->idev, priv);
	448
	449	/*
	450	* Put the device in deep sleep mode for now.
	451	* ->open() will bring it back once the it is actually needed.
	452	*/
	453	cap11xx_set_sleep(priv, true);
	454
	455	error = input_register_device(priv->idev);
	456	if (error)
	457	return error;
	458
	459	irq = irq_of_parse_and_map(node, 0);
	460	if (!irq) {
	461	dev_err(dev, "Unable to parse or map IRQ\n");
	462	return -ENXIO;
	463	}
	464
	465	error = devm_request_threaded_irq(dev, irq, NULL, cap11xx_thread_func,
	466	IRQF_ONESHOT, dev_name(dev), priv);
	467	if (error)
	468	return error;
	469
	470	return 0;
	471	}
	472
	473	static const struct of_device_id cap11xx_dt_ids[] = {
	474	{ .compatible = "microchip,cap1106", },
7609a5e9 MR	475	{ .compatible = "microchip,cap1126", },
7609a5e9 MR	476	{ .compatible = "microchip,cap1188", },
c77fd0a4 MR	477	{}
	478	};
	479	MODULE_DEVICE_TABLE(of, cap11xx_dt_ids);
	480
	481	static const struct i2c_device_id cap11xx_i2c_ids[] = {
7609a5e9 MR	482	{ "cap1106", CAP1106 },
	483	{ "cap1126", CAP1126 },
	484	{ "cap1188", CAP1188 },
c77fd0a4 MR	485	{}
	486	};
	487	MODULE_DEVICE_TABLE(i2c, cap11xx_i2c_ids);
	488
	489	static struct i2c_driver cap11xx_i2c_driver = {
	490	.driver = {
	491	.name = "cap11xx",
c77fd0a4 MR	492	.of_match_table = cap11xx_dt_ids,
	493	},
	494	.id_table = cap11xx_i2c_ids,
	495	.probe = cap11xx_i2c_probe,
	496	};
	497
	498	module_i2c_driver(cap11xx_i2c_driver);
	499
c77fd0a4 MR	500	MODULE_DESCRIPTION("Microchip CAP11XX driver");
	501	MODULE_AUTHOR("Daniel Mack <linux@zonque.org>");
	502	MODULE_LICENSE("GPL v2");