[deliverable/linux.git] / drivers / input / touchscreen / wacom_w8001.c

/*
 * Wacom W8001 penabled serial touchscreen driver
 *
 * Copyright (c) 2008 Jaya Kumar
 * Copyright (c) 2010 Red Hat, Inc.
 *
 * This file is subject to the terms and conditions of the GNU General Public
 * License. See the file COPYING in the main directory of this archive for
 * more details.
 *
 * Layout based on Elo serial touchscreen driver by Vojtech Pavlik
 */

#include <linux/errno.h>
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/slab.h>
#include <linux/input/mt.h>
#include <linux/serio.h>
#include <linux/init.h>
#include <linux/ctype.h>

#define DRIVER_DESC	"Wacom W8001 serial touchscreen driver"

MODULE_AUTHOR("Jaya Kumar <jayakumar.lkml@gmail.com>");
MODULE_DESCRIPTION(DRIVER_DESC);
MODULE_LICENSE("GPL");

#define W8001_MAX_LENGTH	11
#define W8001_LEAD_MASK		0x80
#define W8001_LEAD_BYTE		0x80
#define W8001_TAB_MASK		0x40
#define W8001_TAB_BYTE		0x40
/* set in first byte of touch data packets */
#define W8001_TOUCH_MASK	(0x10 | W8001_LEAD_MASK)
#define W8001_TOUCH_BYTE	(0x10 | W8001_LEAD_BYTE)

#define W8001_QUERY_PACKET	0x20

#define W8001_CMD_START		'1'
#define W8001_CMD_QUERY		'*'
#define W8001_CMD_TOUCHQUERY	'%'

/* length of data packets in bytes, depends on device. */
#define W8001_PKTLEN_TOUCH93	5
#define W8001_PKTLEN_TOUCH9A	7
#define W8001_PKTLEN_TPCPEN	9
#define W8001_PKTLEN_TPCCTL	11	/* control packet */
#define W8001_PKTLEN_TOUCH2FG	13

#define MAX_TRACKING_ID		0xFF	/* arbitrarily chosen */

struct w8001_coord {
	u8 rdy;
	u8 tsw;
	u8 f1;
	u8 f2;
	u16 x;
	u16 y;
	u16 pen_pressure;
	u8 tilt_x;
	u8 tilt_y;
};

/* touch query reply packet */
struct w8001_touch_query {
	u8 panel_res;
	u8 capacity_res;
	u8 sensor_id;
	u16 x;
	u16 y;
};

/*
 * Per-touchscreen data.
 */

struct w8001 {
	struct input_dev *dev;
	struct serio *serio;
	struct completion cmd_done;
	int id;
	int idx;
	unsigned char response_type;
	unsigned char response[W8001_MAX_LENGTH];
	unsigned char data[W8001_MAX_LENGTH];
	char phys[32];
	int type;
	unsigned int pktlen;
	int trkid[2];
};

static void parse_data(u8 *data, struct w8001_coord *coord)
{
	memset(coord, 0, sizeof(*coord));

	coord->rdy = data[0] & 0x20;
	coord->tsw = data[0] & 0x01;
	coord->f1 = data[0] & 0x02;
	coord->f2 = data[0] & 0x04;

	coord->x = (data[1] & 0x7F) << 9;
	coord->x |= (data[2] & 0x7F) << 2;
	coord->x |= (data[6] & 0x60) >> 5;

	coord->y = (data[3] & 0x7F) << 9;
	coord->y |= (data[4] & 0x7F) << 2;
	coord->y |= (data[6] & 0x18) >> 3;

	coord->pen_pressure = data[5] & 0x7F;
	coord->pen_pressure |= (data[6] & 0x07) << 7 ;

	coord->tilt_x = data[7] & 0x7F;
	coord->tilt_y = data[8] & 0x7F;
}

static void parse_touch(struct w8001 *w8001)
{
	static int trkid;
	struct input_dev *dev = w8001->dev;
	unsigned char *data = w8001->data;
	int i;

	for (i = 0; i < 2; i++) {
		input_mt_slot(dev, i);

		if (data[0] & (1 << i)) {
			int x = (data[6 * i + 1] << 7) | (data[6 * i + 2]);
			int y = (data[6 * i + 3] << 7) | (data[6 * i + 4]);
			/* data[5,6] and [11,12] is finger capacity */

			input_report_abs(dev, ABS_MT_POSITION_X, x);
			input_report_abs(dev, ABS_MT_POSITION_Y, y);
			input_report_abs(dev, ABS_MT_TOOL_TYPE, MT_TOOL_FINGER);
			if (w8001->trkid[i] < 0)
				w8001->trkid[i] = trkid++ & MAX_TRACKING_ID;
		} else {
			w8001->trkid[i] = -1;
		}
		input_report_abs(dev, ABS_MT_TRACKING_ID, w8001->trkid[i]);
	}

	input_sync(dev);
}

static void parse_touchquery(u8 *data, struct w8001_touch_query *query)
{
	memset(query, 0, sizeof(*query));

	query->panel_res = data[1];
	query->sensor_id = data[2] & 0x7;
	query->capacity_res = data[7];

	query->x = data[3] << 9;
	query->x |= data[4] << 2;
	query->x |= (data[2] >> 5) & 0x3;

	query->y = data[5] << 9;
	query->y |= data[6] << 2;
	query->y |= (data[2] >> 3) & 0x3;
}

static void report_pen_events(struct w8001 *w8001, struct w8001_coord *coord)
{
	struct input_dev *dev = w8001->dev;

	/*
	 * We have 1 bit for proximity (rdy) and 3 bits for tip, side,
	 * side2/eraser. If rdy && f2 are set, this can be either pen + side2,
	 * or eraser. assume
	 * - if dev is already in proximity and f2 is toggled → pen + side2
	 * - if dev comes into proximity with f2 set → eraser
	 * If f2 disappears after assuming eraser, fake proximity out for
	 * eraser and in for pen.
	 */

	if (!w8001->type) {
		w8001->type = coord->f2 ? BTN_TOOL_RUBBER : BTN_TOOL_PEN;
	} else if (w8001->type == BTN_TOOL_RUBBER) {
		if (!coord->f2) {
			input_report_abs(dev, ABS_PRESSURE, 0);
			input_report_key(dev, BTN_TOUCH, 0);
			input_report_key(dev, BTN_STYLUS, 0);
			input_report_key(dev, BTN_STYLUS2, 0);
			input_report_key(dev, BTN_TOOL_RUBBER, 0);
			input_sync(dev);
			w8001->type = BTN_TOOL_PEN;
		}
	} else {
		input_report_key(dev, BTN_STYLUS2, coord->f2);
	}

	input_report_abs(dev, ABS_X, coord->x);
	input_report_abs(dev, ABS_Y, coord->y);
	input_report_abs(dev, ABS_PRESSURE, coord->pen_pressure);
	input_report_key(dev, BTN_TOUCH, coord->tsw);
	input_report_key(dev, BTN_STYLUS, coord->f1);
	input_report_key(dev, w8001->type, coord->rdy);
	input_sync(dev);

	if (!coord->rdy)
		w8001->type = 0;
}

static irqreturn_t w8001_interrupt(struct serio *serio,
				   unsigned char data, unsigned int flags)
{
	struct w8001 *w8001 = serio_get_drvdata(serio);
	struct w8001_coord coord;
	unsigned char tmp;

	w8001->data[w8001->idx] = data;
	switch (w8001->idx++) {
	case 0:
		if ((data & W8001_LEAD_MASK) != W8001_LEAD_BYTE) {
			pr_debug("w8001: unsynchronized data: 0x%02x\n", data);
			w8001->idx = 0;
		}
		break;

	case W8001_PKTLEN_TOUCH93 - 1:
	case W8001_PKTLEN_TOUCH9A - 1:
		/* ignore one-finger touch packet. */
		if (w8001->pktlen == w8001->idx)
			w8001->idx = 0;
		break;

	/* Pen coordinates packet */
	case W8001_PKTLEN_TPCPEN - 1:
		tmp = w8001->data[0] & W8001_TAB_MASK;
		if (unlikely(tmp == W8001_TAB_BYTE))
			break;

		tmp = (w8001->data[0] & W8001_TOUCH_BYTE);
		if (tmp == W8001_TOUCH_BYTE)
			break;

		w8001->idx = 0;
		parse_data(w8001->data, &coord);
		report_pen_events(w8001, &coord);
		break;

	/* control packet */
	case W8001_PKTLEN_TPCCTL - 1:
		tmp = (w8001->data[0] & W8001_TOUCH_MASK);
		if (tmp == W8001_TOUCH_BYTE)
			break;

		w8001->idx = 0;
		memcpy(w8001->response, w8001->data, W8001_MAX_LENGTH);
		w8001->response_type = W8001_QUERY_PACKET;
		complete(&w8001->cmd_done);
		break;

	/* 2 finger touch packet */
	case W8001_PKTLEN_TOUCH2FG - 1:
		w8001->idx = 0;
		parse_touch(w8001);
		break;
	}

	return IRQ_HANDLED;
}

static int w8001_command(struct w8001 *w8001, unsigned char command,
			 bool wait_response)
{
	int rc;

	w8001->response_type = 0;
	init_completion(&w8001->cmd_done);

	rc = serio_write(w8001->serio, command);
	if (rc == 0 && wait_response) {

		wait_for_completion_timeout(&w8001->cmd_done, HZ);
		if (w8001->response_type != W8001_QUERY_PACKET)
			rc = -EIO;
	}

	return rc;
}

static int w8001_setup(struct w8001 *w8001)
{
	struct input_dev *dev = w8001->dev;
	struct w8001_coord coord;
	int error;

	error = w8001_command(w8001, W8001_CMD_QUERY, true);
	if (error)
		return error;

	parse_data(w8001->response, &coord);

	input_set_abs_params(dev, ABS_X, 0, coord.x, 0, 0);
	input_set_abs_params(dev, ABS_Y, 0, coord.y, 0, 0);
	input_set_abs_params(dev, ABS_PRESSURE, 0, coord.pen_pressure, 0, 0);
	input_set_abs_params(dev, ABS_TILT_X, 0, coord.tilt_x, 0, 0);
	input_set_abs_params(dev, ABS_TILT_Y, 0, coord.tilt_y, 0, 0);

	error = w8001_command(w8001, W8001_CMD_TOUCHQUERY, true);
	if (!error) {
		struct w8001_touch_query touch;

		parse_touchquery(w8001->response, &touch);

		switch (touch.sensor_id) {
		case 0:
		case 2:
			w8001->pktlen = W8001_PKTLEN_TOUCH93;
			break;
		case 1:
		case 3:
		case 4:
			w8001->pktlen = W8001_PKTLEN_TOUCH9A;
			break;
		case 5:
			w8001->pktlen = W8001_PKTLEN_TOUCH2FG;

			input_mt_init_slots(dev, 2);
			input_set_abs_params(dev, ABS_MT_TRACKING_ID,
						0, MAX_TRACKING_ID, 0, 0);
			input_set_abs_params(dev, ABS_MT_POSITION_X,
						0, touch.x, 0, 0);
			input_set_abs_params(dev, ABS_MT_POSITION_Y,
						0, touch.y, 0, 0);
			input_set_abs_params(dev, ABS_MT_TOOL_TYPE,
						0, 0, 0, 0);
			break;
		}
	}

	return w8001_command(w8001, W8001_CMD_START, false);
}

/*
 * w8001_disconnect() is the opposite of w8001_connect()
 */

static void w8001_disconnect(struct serio *serio)
{
	struct w8001 *w8001 = serio_get_drvdata(serio);

	input_get_device(w8001->dev);
	input_unregister_device(w8001->dev);
	serio_close(serio);
	serio_set_drvdata(serio, NULL);
	input_put_device(w8001->dev);
	kfree(w8001);
}

/*
 * w8001_connect() is the routine that is called when someone adds a
 * new serio device that supports the w8001 protocol and registers it as
 * an input device.
 */

static int w8001_connect(struct serio *serio, struct serio_driver *drv)
{
	struct w8001 *w8001;
	struct input_dev *input_dev;
	int err;

	w8001 = kzalloc(sizeof(struct w8001), GFP_KERNEL);
	input_dev = input_allocate_device();
	if (!w8001 || !input_dev) {
		err = -ENOMEM;
		goto fail1;
	}

	w8001->serio = serio;
	w8001->id = serio->id.id;
	w8001->dev = input_dev;
	w8001->trkid[0] = w8001->trkid[1] = -1;
	init_completion(&w8001->cmd_done);
	snprintf(w8001->phys, sizeof(w8001->phys), "%s/input0", serio->phys);

	input_dev->name = "Wacom W8001 Penabled Serial TouchScreen";
	input_dev->phys = w8001->phys;
	input_dev->id.bustype = BUS_RS232;
	input_dev->id.vendor = SERIO_W8001;
	input_dev->id.product = w8001->id;
	input_dev->id.version = 0x0100;
	input_dev->dev.parent = &serio->dev;

	input_dev->evbit[0] = BIT_MASK(EV_KEY) | BIT_MASK(EV_ABS);
	input_dev->keybit[BIT_WORD(BTN_TOUCH)] = BIT_MASK(BTN_TOUCH);
	input_dev->keybit[BIT_WORD(BTN_TOOL_PEN)] |= BIT_MASK(BTN_TOOL_PEN);
	input_dev->keybit[BIT_WORD(BTN_TOOL_RUBBER)] |= BIT_MASK(BTN_TOOL_RUBBER);
	input_dev->keybit[BIT_WORD(BTN_STYLUS)] |= BIT_MASK(BTN_STYLUS);
	input_dev->keybit[BIT_WORD(BTN_STYLUS2)] |= BIT_MASK(BTN_STYLUS2);

	serio_set_drvdata(serio, w8001);
	err = serio_open(serio, drv);
	if (err)
		goto fail2;

	err = w8001_setup(w8001);
	if (err)
		goto fail3;

	err = input_register_device(w8001->dev);
	if (err)
		goto fail3;

	return 0;

fail3:
	serio_close(serio);
fail2:
	serio_set_drvdata(serio, NULL);
fail1:
	input_free_device(input_dev);
	kfree(w8001);
	return err;
}

static struct serio_device_id w8001_serio_ids[] = {
	{
		.type	= SERIO_RS232,
		.proto	= SERIO_W8001,
		.id	= SERIO_ANY,
		.extra	= SERIO_ANY,
	},
	{ 0 }
};

MODULE_DEVICE_TABLE(serio, w8001_serio_ids);

static struct serio_driver w8001_drv = {
	.driver		= {
		.name	= "w8001",
	},
	.description	= DRIVER_DESC,
	.id_table	= w8001_serio_ids,
	.interrupt	= w8001_interrupt,
	.connect	= w8001_connect,
	.disconnect	= w8001_disconnect,
};

static int __init w8001_init(void)
{
	return serio_register_driver(&w8001_drv);
}

static void __exit w8001_exit(void)
{
	serio_unregister_driver(&w8001_drv);
}

module_init(w8001_init);
module_exit(w8001_exit);
Commit	Line	Data
3eb1aa43 JK	1	/*
	2	* Wacom W8001 penabled serial touchscreen driver
	3	*
	4	* Copyright (c) 2008 Jaya Kumar
aaba933e	5	* Copyright (c) 2010 Red Hat, Inc.
3eb1aa43 JK	6	*
	7	* This file is subject to the terms and conditions of the GNU General Public
	8	* License. See the file COPYING in the main directory of this archive for
	9	* more details.
	10	*
	11	* Layout based on Elo serial touchscreen driver by Vojtech Pavlik
	12	*/
	13
	14	#include <linux/errno.h>
	15	#include <linux/kernel.h>
	16	#include <linux/module.h>
	17	#include <linux/slab.h>
47c78e89	18	#include <linux/input/mt.h>
3eb1aa43 JK	19	#include <linux/serio.h>
	20	#include <linux/init.h>
	21	#include <linux/ctype.h>
	22
	23	#define DRIVER_DESC "Wacom W8001 serial touchscreen driver"
	24
	25	MODULE_AUTHOR("Jaya Kumar <jayakumar.lkml@gmail.com>");
	26	MODULE_DESCRIPTION(DRIVER_DESC);
	27	MODULE_LICENSE("GPL");
	28
3eb1aa43	29	#define W8001_MAX_LENGTH 11
41c372dc DT	30	#define W8001_LEAD_MASK 0x80
	31	#define W8001_LEAD_BYTE 0x80
	32	#define W8001_TAB_MASK 0x40
	33	#define W8001_TAB_BYTE 0x40
aaba933e PH	34	/* set in first byte of touch data packets */
	35	#define W8001_TOUCH_MASK (0x10 \| W8001_LEAD_MASK)
	36	#define W8001_TOUCH_BYTE (0x10 \| W8001_LEAD_BYTE)
3eb1aa43	37
41c372dc DT	38	#define W8001_QUERY_PACKET 0x20
	39
	40	#define W8001_CMD_START '1'
	41	#define W8001_CMD_QUERY '*'
aaba933e PH	42	#define W8001_CMD_TOUCHQUERY '%'
	43
	44	/* length of data packets in bytes, depends on device. */
	45	#define W8001_PKTLEN_TOUCH93 5
	46	#define W8001_PKTLEN_TOUCH9A 7
	47	#define W8001_PKTLEN_TPCPEN 9
	48	#define W8001_PKTLEN_TPCCTL 11 /* control packet */
	49	#define W8001_PKTLEN_TOUCH2FG 13
3eb1aa43	50
5e8b9140 PH	51	#define MAX_TRACKING_ID 0xFF /* arbitrarily chosen */
5e8b9140 PH	52
3eb1aa43 JK	53	struct w8001_coord {
	54	u8 rdy;
	55	u8 tsw;
	56	u8 f1;
	57	u8 f2;
	58	u16 x;
	59	u16 y;
	60	u16 pen_pressure;
	61	u8 tilt_x;
	62	u8 tilt_y;
	63	};
	64
aaba933e PH	65	/* touch query reply packet */
	66	struct w8001_touch_query {
	67	u8 panel_res;
	68	u8 capacity_res;
	69	u8 sensor_id;
	70	u16 x;
	71	u16 y;
	72	};
	73
3eb1aa43 JK	74	/*
	75	* Per-touchscreen data.
	76	*/
	77
	78	struct w8001 {
	79	struct input_dev *dev;
	80	struct serio *serio;
3eb1aa43 JK	81	struct completion cmd_done;
	82	int id;
	83	int idx;
41c372dc DT	84	unsigned char response_type;
41c372dc DT	85	unsigned char response[W8001_MAX_LENGTH];
3eb1aa43	86	unsigned char data[W8001_MAX_LENGTH];
3eb1aa43	87	char phys[32];
2072f8db	88	int type;
aaba933e	89	unsigned int pktlen;
5e8b9140	90	int trkid[2];
3eb1aa43 JK	91	};
3eb1aa43 JK	92
41c372dc	93	static void parse_data(u8 data, struct w8001_coord coord)
3eb1aa43	94	{
41c372dc DT	95	memset(coord, 0, sizeof(*coord));
41c372dc DT	96
3eb1aa43 JK	97	coord->rdy = data[0] & 0x20;
	98	coord->tsw = data[0] & 0x01;
	99	coord->f1 = data[0] & 0x02;
	100	coord->f2 = data[0] & 0x04;
	101
	102	coord->x = (data[1] & 0x7F) << 9;
	103	coord->x \|= (data[2] & 0x7F) << 2;
	104	coord->x \|= (data[6] & 0x60) >> 5;
	105
	106	coord->y = (data[3] & 0x7F) << 9;
	107	coord->y \|= (data[4] & 0x7F) << 2;
	108	coord->y \|= (data[6] & 0x18) >> 3;
	109
	110	coord->pen_pressure = data[5] & 0x7F;
	111	coord->pen_pressure \|= (data[6] & 0x07) << 7 ;
	112
	113	coord->tilt_x = data[7] & 0x7F;
	114	coord->tilt_y = data[8] & 0x7F;
3eb1aa43 JK	115	}
3eb1aa43 JK	116
5e8b9140 PH	117	static void parse_touch(struct w8001 *w8001)
	118	{
	119	static int trkid;
	120	struct input_dev *dev = w8001->dev;
	121	unsigned char *data = w8001->data;
	122	int i;
	123
	124	for (i = 0; i < 2; i++) {
	125	input_mt_slot(dev, i);
	126
	127	if (data[0] & (1 << i)) {
	128	int x = (data[6 * i + 1] << 7) \| (data[6 * i + 2]);
	129	int y = (data[6 * i + 3] << 7) \| (data[6 * i + 4]);
	130	/* data[5,6] and [11,12] is finger capacity */
	131
	132	input_report_abs(dev, ABS_MT_POSITION_X, x);
	133	input_report_abs(dev, ABS_MT_POSITION_Y, y);
	134	input_report_abs(dev, ABS_MT_TOOL_TYPE, MT_TOOL_FINGER);
	135	if (w8001->trkid[i] < 0)
	136	w8001->trkid[i] = trkid++ & MAX_TRACKING_ID;
	137	} else {
	138	w8001->trkid[i] = -1;
	139	}
	140	input_report_abs(dev, ABS_MT_TRACKING_ID, w8001->trkid[i]);
	141	}
	142
	143	input_sync(dev);
	144	}
	145
aaba933e PH	146	static void parse_touchquery(u8 data, struct w8001_touch_query query)
	147	{
	148	memset(query, 0, sizeof(*query));
	149
	150	query->panel_res = data[1];
	151	query->sensor_id = data[2] & 0x7;
	152	query->capacity_res = data[7];
	153
	154	query->x = data[3] << 9;
	155	query->x \|= data[4] << 2;
	156	query->x \|= (data[2] >> 5) & 0x3;
	157
	158	query->y = data[5] << 9;
	159	query->y \|= data[6] << 2;
	160	query->y \|= (data[2] >> 3) & 0x3;
	161	}
	162
2072f8db PH	163	static void report_pen_events(struct w8001 w8001, struct w8001_coord coord)
	164	{
	165	struct input_dev *dev = w8001->dev;
	166
	167	/*
	168	* We have 1 bit for proximity (rdy) and 3 bits for tip, side,
	169	* side2/eraser. If rdy && f2 are set, this can be either pen + side2,
	170	* or eraser. assume
	171	* - if dev is already in proximity and f2 is toggled → pen + side2
	172	* - if dev comes into proximity with f2 set → eraser
	173	* If f2 disappears after assuming eraser, fake proximity out for
	174	* eraser and in for pen.
	175	*/
	176
	177	if (!w8001->type) {
	178	w8001->type = coord->f2 ? BTN_TOOL_RUBBER : BTN_TOOL_PEN;
	179	} else if (w8001->type == BTN_TOOL_RUBBER) {
	180	if (!coord->f2) {
	181	input_report_abs(dev, ABS_PRESSURE, 0);
	182	input_report_key(dev, BTN_TOUCH, 0);
	183	input_report_key(dev, BTN_STYLUS, 0);
	184	input_report_key(dev, BTN_STYLUS2, 0);
	185	input_report_key(dev, BTN_TOOL_RUBBER, 0);
	186	input_sync(dev);
	187	w8001->type = BTN_TOOL_PEN;
	188	}
	189	} else {
	190	input_report_key(dev, BTN_STYLUS2, coord->f2);
	191	}
	192
	193	input_report_abs(dev, ABS_X, coord->x);
	194	input_report_abs(dev, ABS_Y, coord->y);
	195	input_report_abs(dev, ABS_PRESSURE, coord->pen_pressure);
	196	input_report_key(dev, BTN_TOUCH, coord->tsw);
	197	input_report_key(dev, BTN_STYLUS, coord->f1);
	198	input_report_key(dev, w8001->type, coord->rdy);
	199	input_sync(dev);
	200
	201	if (!coord->rdy)
	202	w8001->type = 0;
	203	}
	204
41c372dc DT	205	static irqreturn_t w8001_interrupt(struct serio *serio,
41c372dc DT	206	unsigned char data, unsigned int flags)
3eb1aa43	207	{
41c372dc	208	struct w8001 *w8001 = serio_get_drvdata(serio);
3eb1aa43	209	struct w8001_coord coord;
41c372dc	210	unsigned char tmp;
3eb1aa43 JK	211
	212	w8001->data[w8001->idx] = data;
	213	switch (w8001->idx++) {
	214	case 0:
	215	if ((data & W8001_LEAD_MASK) != W8001_LEAD_BYTE) {
	216	pr_debug("w8001: unsynchronized data: 0x%02x\n", data);
	217	w8001->idx = 0;
	218	}
	219	break;
41c372dc	220
aaba933e PH	221	case W8001_PKTLEN_TOUCH93 - 1:
	222	case W8001_PKTLEN_TOUCH9A - 1:
	223	/* ignore one-finger touch packet. */
	224	if (w8001->pktlen == w8001->idx)
	225	w8001->idx = 0;
	226	break;
	227
	228	/* Pen coordinates packet */
	229	case W8001_PKTLEN_TPCPEN - 1:
3eb1aa43 JK	230	tmp = w8001->data[0] & W8001_TAB_MASK;
	231	if (unlikely(tmp == W8001_TAB_BYTE))
	232	break;
41c372dc	233
aaba933e PH	234	tmp = (w8001->data[0] & W8001_TOUCH_BYTE);
	235	if (tmp == W8001_TOUCH_BYTE)
	236	break;
	237
3eb1aa43	238	w8001->idx = 0;
3eb1aa43	239	parse_data(w8001->data, &coord);
2072f8db	240	report_pen_events(w8001, &coord);
3eb1aa43	241	break;
41c372dc	242
aaba933e PH	243	/* control packet */
	244	case W8001_PKTLEN_TPCCTL - 1:
	245	tmp = (w8001->data[0] & W8001_TOUCH_MASK);
	246	if (tmp == W8001_TOUCH_BYTE)
	247	break;
	248
3eb1aa43	249	w8001->idx = 0;
41c372dc DT	250	memcpy(w8001->response, w8001->data, W8001_MAX_LENGTH);
41c372dc DT	251	w8001->response_type = W8001_QUERY_PACKET;
3eb1aa43 JK	252	complete(&w8001->cmd_done);
3eb1aa43 JK	253	break;
aaba933e	254
5e8b9140	255	/* 2 finger touch packet */
aaba933e	256	case W8001_PKTLEN_TOUCH2FG - 1:
5e8b9140 PH	257	w8001->idx = 0;
5e8b9140 PH	258	parse_touch(w8001);
aaba933e	259	break;
3eb1aa43	260	}
3eb1aa43 JK	261
	262	return IRQ_HANDLED;
	263	}
	264
41c372dc DT	265	static int w8001_command(struct w8001 *w8001, unsigned char command,
41c372dc DT	266	bool wait_response)
3eb1aa43	267	{
41c372dc	268	int rc;
3eb1aa43	269
41c372dc	270	w8001->response_type = 0;
3eb1aa43	271	init_completion(&w8001->cmd_done);
3eb1aa43	272
41c372dc DT	273	rc = serio_write(w8001->serio, command);
41c372dc DT	274	if (rc == 0 && wait_response) {
3eb1aa43	275
41c372dc DT	276	wait_for_completion_timeout(&w8001->cmd_done, HZ);
	277	if (w8001->response_type != W8001_QUERY_PACKET)
	278	rc = -EIO;
3eb1aa43 JK	279	}
3eb1aa43 JK	280
3eb1aa43 JK	281	return rc;
	282	}
	283
	284	static int w8001_setup(struct w8001 *w8001)
	285	{
3eb1aa43	286	struct input_dev *dev = w8001->dev;
41c372dc DT	287	struct w8001_coord coord;
41c372dc DT	288	int error;
3eb1aa43	289
41c372dc DT	290	error = w8001_command(w8001, W8001_CMD_QUERY, true);
	291	if (error)
	292	return error;
3eb1aa43	293
41c372dc	294	parse_data(w8001->response, &coord);
3eb1aa43 JK	295
	296	input_set_abs_params(dev, ABS_X, 0, coord.x, 0, 0);
	297	input_set_abs_params(dev, ABS_Y, 0, coord.y, 0, 0);
	298	input_set_abs_params(dev, ABS_PRESSURE, 0, coord.pen_pressure, 0, 0);
	299	input_set_abs_params(dev, ABS_TILT_X, 0, coord.tilt_x, 0, 0);
	300	input_set_abs_params(dev, ABS_TILT_Y, 0, coord.tilt_y, 0, 0);
	301
aaba933e PH	302	error = w8001_command(w8001, W8001_CMD_TOUCHQUERY, true);
	303	if (!error) {
	304	struct w8001_touch_query touch;
	305
	306	parse_touchquery(w8001->response, &touch);
	307
	308	switch (touch.sensor_id) {
	309	case 0:
	310	case 2:
	311	w8001->pktlen = W8001_PKTLEN_TOUCH93;
	312	break;
	313	case 1:
	314	case 3:
	315	case 4:
	316	w8001->pktlen = W8001_PKTLEN_TOUCH9A;
	317	break;
	318	case 5:
	319	w8001->pktlen = W8001_PKTLEN_TOUCH2FG;
5e8b9140	320
8cde8100	321	input_mt_init_slots(dev, 2);
5e8b9140 PH	322	input_set_abs_params(dev, ABS_MT_TRACKING_ID,
	323	0, MAX_TRACKING_ID, 0, 0);
	324	input_set_abs_params(dev, ABS_MT_POSITION_X,
	325	0, touch.x, 0, 0);
	326	input_set_abs_params(dev, ABS_MT_POSITION_Y,
	327	0, touch.y, 0, 0);
	328	input_set_abs_params(dev, ABS_MT_TOOL_TYPE,
	329	0, 0, 0, 0);
aaba933e PH	330	break;
	331	}
	332	}
	333
41c372dc	334	return w8001_command(w8001, W8001_CMD_START, false);
3eb1aa43 JK	335	}
	336
	337	/*
	338	* w8001_disconnect() is the opposite of w8001_connect()
	339	*/
	340
	341	static void w8001_disconnect(struct serio *serio)
	342	{
	343	struct w8001 *w8001 = serio_get_drvdata(serio);
	344
	345	input_get_device(w8001->dev);
	346	input_unregister_device(w8001->dev);
	347	serio_close(serio);
	348	serio_set_drvdata(serio, NULL);
	349	input_put_device(w8001->dev);
	350	kfree(w8001);
	351	}
	352
	353	/*
	354	* w8001_connect() is the routine that is called when someone adds a
	355	* new serio device that supports the w8001 protocol and registers it as
	356	* an input device.
	357	*/
	358
	359	static int w8001_connect(struct serio serio, struct serio_driver drv)
	360	{
	361	struct w8001 *w8001;
	362	struct input_dev *input_dev;
	363	int err;
	364
	365	w8001 = kzalloc(sizeof(struct w8001), GFP_KERNEL);
	366	input_dev = input_allocate_device();
	367	if (!w8001 \|\| !input_dev) {
	368	err = -ENOMEM;
	369	goto fail1;
	370	}
	371
	372	w8001->serio = serio;
	373	w8001->id = serio->id.id;
	374	w8001->dev = input_dev;
5e8b9140	375	w8001->trkid[0] = w8001->trkid[1] = -1;
3eb1aa43 JK	376	init_completion(&w8001->cmd_done);
	377	snprintf(w8001->phys, sizeof(w8001->phys), "%s/input0", serio->phys);
	378
	379	input_dev->name = "Wacom W8001 Penabled Serial TouchScreen";
	380	input_dev->phys = w8001->phys;
	381	input_dev->id.bustype = BUS_RS232;
	382	input_dev->id.vendor = SERIO_W8001;
	383	input_dev->id.product = w8001->id;
	384	input_dev->id.version = 0x0100;
	385	input_dev->dev.parent = &serio->dev;
	386
	387	input_dev->evbit[0] = BIT_MASK(EV_KEY) \| BIT_MASK(EV_ABS);
	388	input_dev->keybit[BIT_WORD(BTN_TOUCH)] = BIT_MASK(BTN_TOUCH);
2072f8db PH	389	input_dev->keybit[BIT_WORD(BTN_TOOL_PEN)] \|= BIT_MASK(BTN_TOOL_PEN);
	390	input_dev->keybit[BIT_WORD(BTN_TOOL_RUBBER)] \|= BIT_MASK(BTN_TOOL_RUBBER);
	391	input_dev->keybit[BIT_WORD(BTN_STYLUS)] \|= BIT_MASK(BTN_STYLUS);
	392	input_dev->keybit[BIT_WORD(BTN_STYLUS2)] \|= BIT_MASK(BTN_STYLUS2);
3eb1aa43 JK	393
	394	serio_set_drvdata(serio, w8001);
	395	err = serio_open(serio, drv);
	396	if (err)
	397	goto fail2;
	398
41c372dc DT	399	err = w8001_setup(w8001);
41c372dc DT	400	if (err)
3eb1aa43 JK	401	goto fail3;
	402
	403	err = input_register_device(w8001->dev);
	404	if (err)
	405	goto fail3;
	406
	407	return 0;
	408
	409	fail3:
	410	serio_close(serio);
	411	fail2:
	412	serio_set_drvdata(serio, NULL);
	413	fail1:
	414	input_free_device(input_dev);
	415	kfree(w8001);
	416	return err;
	417	}
	418
	419	static struct serio_device_id w8001_serio_ids[] = {
	420	{
	421	.type = SERIO_RS232,
	422	.proto = SERIO_W8001,
	423	.id = SERIO_ANY,
	424	.extra = SERIO_ANY,
	425	},
	426	{ 0 }
	427	};
	428
	429	MODULE_DEVICE_TABLE(serio, w8001_serio_ids);
	430
	431	static struct serio_driver w8001_drv = {
	432	.driver = {
	433	.name = "w8001",
	434	},
	435	.description = DRIVER_DESC,
	436	.id_table = w8001_serio_ids,
	437	.interrupt = w8001_interrupt,
	438	.connect = w8001_connect,
	439	.disconnect = w8001_disconnect,
	440	};
	441
	442	static int __init w8001_init(void)
	443	{
	444	return serio_register_driver(&w8001_drv);
	445	}
	446
	447	static void __exit w8001_exit(void)
	448	{
	449	serio_unregister_driver(&w8001_drv);
	450	}
	451
	452	module_init(w8001_init);
	453	module_exit(w8001_exit);