[mirror_ubuntu-bionic-kernel.git] / drivers / input / joystick / adi.c

/*
 *  Copyright (c) 1998-2005 Vojtech Pavlik
 */

/*
 * Logitech ADI joystick family driver for Linux
 */

/*
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 *
 * Should you need to contact me, the author, you can do so either by
 * e-mail - mail your message to <vojtech@ucw.cz>, or by paper mail:
 * Vojtech Pavlik, Simunkova 1594, Prague 8, 182 00 Czech Republic
 */

#include <linux/delay.h>
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/string.h>
#include <linux/slab.h>
#include <linux/input.h>
#include <linux/gameport.h>
#include <linux/jiffies.h>

#define DRIVER_DESC	"Logitech ADI joystick family driver"

MODULE_AUTHOR("Vojtech Pavlik <vojtech@ucw.cz>");
MODULE_DESCRIPTION(DRIVER_DESC);
MODULE_LICENSE("GPL");

/*
 * Times, array sizes, flags, ids.
 */

#define ADI_MAX_START		200	/* Trigger to packet timeout [200us] */
#define ADI_MAX_STROBE		40	/* Single bit timeout [40us] */
#define ADI_INIT_DELAY		10	/* Delay after init packet [10ms] */
#define ADI_DATA_DELAY		4	/* Delay after data packet [4ms] */

#define ADI_MAX_LENGTH		256
#define ADI_MIN_LENGTH		8
#define ADI_MIN_LEN_LENGTH	10
#define ADI_MIN_ID_LENGTH	66
#define ADI_MAX_NAME_LENGTH	64
#define ADI_MAX_CNAME_LENGTH	16
#define ADI_MAX_PHYS_LENGTH	64

#define ADI_FLAG_HAT		0x04
#define ADI_FLAG_10BIT		0x08

#define ADI_ID_TPD		0x01
#define ADI_ID_WGP		0x06
#define ADI_ID_WGPE		0x08
#define ADI_ID_MAX		0x0a

/*
 * Names, buttons, axes ...
 */

static char *adi_names[] = {	"WingMan Extreme Digital", "ThunderPad Digital", "SideCar", "CyberMan 2",
				"WingMan Interceptor", "WingMan Formula", "WingMan GamePad",
				"WingMan Extreme Digital 3D", "WingMan GamePad Extreme",
				"WingMan GamePad USB", "Unknown Device %#x" };

static char adi_wmgpe_abs[] =	{ ABS_X, ABS_Y, ABS_HAT0X, ABS_HAT0Y };
static char adi_wmi_abs[] =	{ ABS_X, ABS_Y, ABS_THROTTLE, ABS_HAT0X, ABS_HAT0Y, ABS_HAT1X, ABS_HAT1Y, ABS_HAT2X, ABS_HAT2Y };
static char adi_wmed3d_abs[] =	{ ABS_X, ABS_Y, ABS_THROTTLE, ABS_RZ, ABS_HAT0X, ABS_HAT0Y };
static char adi_cm2_abs[] =	{ ABS_X, ABS_Y, ABS_Z, ABS_RX, ABS_RY, ABS_RZ };
static char adi_wmf_abs[] =	{ ABS_WHEEL, ABS_GAS, ABS_BRAKE, ABS_HAT0X, ABS_HAT0Y, ABS_HAT1X, ABS_HAT1Y, ABS_HAT2X, ABS_HAT2Y };

static short adi_wmgpe_key[] =	{ BTN_A, BTN_B, BTN_C, BTN_X, BTN_Y, BTN_Z,  BTN_TL, BTN_TR, BTN_START, BTN_MODE, BTN_SELECT };
static short adi_wmi_key[] =	{ BTN_TRIGGER,  BTN_TOP, BTN_THUMB, BTN_TOP2, BTN_BASE, BTN_BASE2, BTN_BASE3, BTN_BASE4, BTN_EXTRA };
static short adi_wmed3d_key[] =	{ BTN_TRIGGER, BTN_THUMB, BTN_THUMB2, BTN_TOP, BTN_TOP2, BTN_BASE, BTN_BASE2 };
static short adi_cm2_key[] =	{ BTN_1, BTN_2, BTN_3, BTN_4, BTN_5, BTN_6, BTN_7, BTN_8 };

static char* adi_abs[] = { adi_wmi_abs, adi_wmgpe_abs, adi_wmf_abs, adi_cm2_abs, adi_wmi_abs, adi_wmf_abs,
			   adi_wmgpe_abs, adi_wmed3d_abs, adi_wmgpe_abs, adi_wmgpe_abs, adi_wmi_abs };

static short* adi_key[] = { adi_wmi_key, adi_wmgpe_key, adi_cm2_key, adi_cm2_key, adi_wmi_key, adi_cm2_key,
			    adi_wmgpe_key, adi_wmed3d_key, adi_wmgpe_key, adi_wmgpe_key, adi_wmi_key };

/*
 * Hat to axis conversion arrays.
 */

static struct {
	int x;
	int y;
} adi_hat_to_axis[] = {{ 0, 0}, { 0,-1}, { 1,-1}, { 1, 0}, { 1, 1}, { 0, 1}, {-1, 1}, {-1, 0}, {-1,-1}};

/*
 * Per-port information.
 */

struct adi {
	struct input_dev *dev;
	int length;
	int ret;
	int idx;
	unsigned char id;
	char buttons;
	char axes10;
	char axes8;
	signed char pad;
	char hats;
	char *abs;
	short *key;
	char name[ADI_MAX_NAME_LENGTH];
	char cname[ADI_MAX_CNAME_LENGTH];
	char phys[ADI_MAX_PHYS_LENGTH];
	unsigned char data[ADI_MAX_LENGTH];
};

struct adi_port {
	struct gameport *gameport;
	struct adi adi[2];
	int bad;
	int reads;
};

/*
 * adi_read_packet() reads a Logitech ADI packet.
 */

static void adi_read_packet(struct adi_port *port)
{
	struct adi *adi = port->adi;
	struct gameport *gameport = port->gameport;
	unsigned char u, v, w, x, z;
	int t[2], s[2], i;
	unsigned long flags;

	for (i = 0; i < 2; i++) {
		adi[i].ret = -1;
		t[i] = gameport_time(gameport, ADI_MAX_START);
		s[i] = 0;
	}

	local_irq_save(flags);

	gameport_trigger(gameport);
	v = z = gameport_read(gameport);

	do {
		u = v;
		w = u ^ (v = x = gameport_read(gameport));
		for (i = 0; i < 2; i++, w >>= 2, x >>= 2) {
			t[i]--;
			if ((w & 0x30) && s[i]) {
				if ((w & 0x30) < 0x30 && adi[i].ret < ADI_MAX_LENGTH && t[i] > 0) {
					adi[i].data[++adi[i].ret] = w;
					t[i] = gameport_time(gameport, ADI_MAX_STROBE);
				} else t[i] = 0;
			} else if (!(x & 0x30)) s[i] = 1;
		}
	} while (t[0] > 0 || t[1] > 0);

	local_irq_restore(flags);

	return;
}

/*
 * adi_move_bits() detects a possible 2-stream mode, and moves
 * the bits accordingly.
 */

static void adi_move_bits(struct adi_port *port, int length)
{
	int i;
	struct adi *adi = port->adi;

	adi[0].idx = adi[1].idx = 0;

	if (adi[0].ret <= 0 || adi[1].ret <= 0) return;
	if (adi[0].data[0] & 0x20 || ~adi[1].data[0] & 0x20) return;

	for (i = 1; i <= adi[1].ret; i++)
		adi[0].data[((length - 1) >> 1) + i + 1] = adi[1].data[i];

	adi[0].ret += adi[1].ret;
	adi[1].ret = -1;
}

/*
 * adi_get_bits() gathers bits from the data packet.
 */

static inline int adi_get_bits(struct adi *adi, int count)
{
	int bits = 0;
	int i;
	if ((adi->idx += count) > adi->ret) return 0;
	for (i = 0; i < count; i++)
		bits |= ((adi->data[adi->idx - i] >> 5) & 1) << i;
	return bits;
}

/*
 * adi_decode() decodes Logitech joystick data into input events.
 */

static int adi_decode(struct adi *adi)
{
	struct input_dev *dev = adi->dev;
	char *abs = adi->abs;
	short *key = adi->key;
	int i, t;

	if (adi->ret < adi->length || adi->id != (adi_get_bits(adi, 4) | (adi_get_bits(adi, 4) << 4)))
		return -1;

	for (i = 0; i < adi->axes10; i++)
		input_report_abs(dev, *abs++, adi_get_bits(adi, 10));

	for (i = 0; i < adi->axes8; i++)
		input_report_abs(dev, *abs++, adi_get_bits(adi, 8));

	for (i = 0; i < adi->buttons && i < 63; i++) {
		if (i == adi->pad) {
			t = adi_get_bits(adi, 4);
			input_report_abs(dev, *abs++, ((t >> 2) & 1) - ( t       & 1));
			input_report_abs(dev, *abs++, ((t >> 1) & 1) - ((t >> 3) & 1));
		}
		input_report_key(dev, *key++, adi_get_bits(adi, 1));
	}

	for (i = 0; i < adi->hats; i++) {
		if ((t = adi_get_bits(adi, 4)) > 8) t = 0;
		input_report_abs(dev, *abs++, adi_hat_to_axis[t].x);
		input_report_abs(dev, *abs++, adi_hat_to_axis[t].y);
	}

	for (i = 63; i < adi->buttons; i++)
		input_report_key(dev, *key++, adi_get_bits(adi, 1));

	input_sync(dev);

	return 0;
}

/*
 * adi_read() reads the data packet and decodes it.
 */

static int adi_read(struct adi_port *port)
{
	int i;
	int result = 0;

	adi_read_packet(port);
	adi_move_bits(port, port->adi[0].length);

	for (i = 0; i < 2; i++)
		if (port->adi[i].length)
			 result |= adi_decode(port->adi + i);

	return result;
}

/*
 * adi_poll() repeatedly polls the Logitech joysticks.
 */

static void adi_poll(struct gameport *gameport)
{
	struct adi_port *port = gameport_get_drvdata(gameport);

	port->bad -= adi_read(port);
	port->reads++;
}

/*
 * adi_open() is a callback from the input open routine.
 */

static int adi_open(struct input_dev *dev)
{
	struct adi_port *port = input_get_drvdata(dev);

	gameport_start_polling(port->gameport);
	return 0;
}

/*
 * adi_close() is a callback from the input close routine.
 */

static void adi_close(struct input_dev *dev)
{
	struct adi_port *port = input_get_drvdata(dev);

	gameport_stop_polling(port->gameport);
}

/*
 * adi_init_digital() sends a trigger & delay sequence
 * to reset and initialize a Logitech joystick into digital mode.
 */

static void adi_init_digital(struct gameport *gameport)
{
	int seq[] = { 4, -2, -3, 10, -6, -11, -7, -9, 11, 0 };
	int i;

	for (i = 0; seq[i]; i++) {
		gameport_trigger(gameport);
		if (seq[i] > 0)
			msleep(seq[i]);
		if (seq[i] < 0) {
			mdelay(-seq[i]);
			udelay(-seq[i]*14);	/* It looks like mdelay() is off by approx 1.4% */
		}
	}
}

static void adi_id_decode(struct adi *adi, struct adi_port *port)
{
	int i, t;

	if (adi->ret < ADI_MIN_ID_LENGTH) /* Minimum ID packet length */
		return;

	if (adi->ret < (t = adi_get_bits(adi, 10))) {
		printk(KERN_WARNING "adi: Short ID packet: reported: %d != read: %d\n", t, adi->ret);
		return;
	}

	adi->id = adi_get_bits(adi, 4) | (adi_get_bits(adi, 4) << 4);

	if ((t = adi_get_bits(adi, 4)) & ADI_FLAG_HAT) adi->hats++;

	adi->length = adi_get_bits(adi, 10);

	if (adi->length >= ADI_MAX_LENGTH || adi->length < ADI_MIN_LENGTH) {
		printk(KERN_WARNING "adi: Bad data packet length (%d).\n", adi->length);
		adi->length = 0;
		return;
	}

	adi->axes8 = adi_get_bits(adi, 4);
	adi->buttons = adi_get_bits(adi, 6);

	if (adi_get_bits(adi, 6) != 8 && adi->hats) {
		printk(KERN_WARNING "adi: Other than 8-dir POVs not supported yet.\n");
		adi->length = 0;
		return;
	}

	adi->buttons += adi_get_bits(adi, 6);
	adi->hats += adi_get_bits(adi, 4);

	i = adi_get_bits(adi, 4);

	if (t & ADI_FLAG_10BIT) {
		adi->axes10 = adi->axes8 - i;
		adi->axes8 = i;
	}

	t = adi_get_bits(adi, 4);

	for (i = 0; i < t; i++)
		adi->cname[i] = adi_get_bits(adi, 8);
	adi->cname[i] = 0;

	t = 8 + adi->buttons + adi->axes10 * 10 + adi->axes8 * 8 + adi->hats * 4;
	if (adi->length != t && adi->length != t + (t & 1)) {
		printk(KERN_WARNING "adi: Expected length %d != data length %d\n", t, adi->length);
		adi->length = 0;
		return;
	}

	switch (adi->id) {
		case ADI_ID_TPD:
			adi->pad = 4;
			adi->buttons -= 4;
			break;
		case ADI_ID_WGP:
			adi->pad = 0;
			adi->buttons -= 4;
			break;
		default:
			adi->pad = -1;
			break;
	}
}

static int adi_init_input(struct adi *adi, struct adi_port *port, int half)
{
	struct input_dev *input_dev;
	char buf[ADI_MAX_NAME_LENGTH];
	int i, t;

	adi->dev = input_dev = input_allocate_device();
	if (!input_dev)
		return -ENOMEM;

	t = adi->id < ADI_ID_MAX ? adi->id : ADI_ID_MAX;

	snprintf(buf, ADI_MAX_PHYS_LENGTH, adi_names[t], adi->id);
	snprintf(adi->name, ADI_MAX_NAME_LENGTH, "Logitech %s [%s]", buf, adi->cname);
	snprintf(adi->phys, ADI_MAX_PHYS_LENGTH, "%s/input%d", port->gameport->phys, half);

	adi->abs = adi_abs[t];
	adi->key = adi_key[t];

	input_dev->name = adi->name;
	input_dev->phys = adi->phys;
	input_dev->id.bustype = BUS_GAMEPORT;
	input_dev->id.vendor = GAMEPORT_ID_VENDOR_LOGITECH;
	input_dev->id.product = adi->id;
	input_dev->id.version = 0x0100;
	input_dev->dev.parent = &port->gameport->dev;

	input_set_drvdata(input_dev, port);

	input_dev->open = adi_open;
	input_dev->close = adi_close;

	input_dev->evbit[0] = BIT_MASK(EV_KEY) | BIT_MASK(EV_ABS);

	for (i = 0; i < adi->axes10 + adi->axes8 + (adi->hats + (adi->pad != -1)) * 2; i++)
		set_bit(adi->abs[i], input_dev->absbit);

	for (i = 0; i < adi->buttons; i++)
		set_bit(adi->key[i], input_dev->keybit);

	return 0;
}

static void adi_init_center(struct adi *adi)
{
	int i, t, x;

	if (!adi->length)
		return;

	for (i = 0; i < adi->axes10 + adi->axes8 + (adi->hats + (adi->pad != -1)) * 2; i++) {

		t = adi->abs[i];
		x = input_abs_get_val(adi->dev, t);

		if (t == ABS_THROTTLE || t == ABS_RUDDER || adi->id == ADI_ID_WGPE)
			x = i < adi->axes10 ? 512 : 128;

		if (i < adi->axes10)
			input_set_abs_params(adi->dev, t, 64, x * 2 - 64, 2, 16);
		else if (i < adi->axes10 + adi->axes8)
			input_set_abs_params(adi->dev, t, 48, x * 2 - 48, 1, 16);
		else
			input_set_abs_params(adi->dev, t, -1, 1, 0, 0);
	}
}

/*
 * adi_connect() probes for Logitech ADI joysticks.
 */

static int adi_connect(struct gameport *gameport, struct gameport_driver *drv)
{
	struct adi_port *port;
	int i;
	int err;

	port = kzalloc(sizeof(struct adi_port), GFP_KERNEL);
	if (!port)
		return -ENOMEM;

	port->gameport = gameport;

	gameport_set_drvdata(gameport, port);

	err = gameport_open(gameport, drv, GAMEPORT_MODE_RAW);
	if (err)
		goto fail1;

	adi_init_digital(gameport);
	adi_read_packet(port);

	if (port->adi[0].ret >= ADI_MIN_LEN_LENGTH)
		adi_move_bits(port, adi_get_bits(port->adi, 10));

	for (i = 0; i < 2; i++) {
		adi_id_decode(port->adi + i, port);

		if (!port->adi[i].length)
			continue;

		err = adi_init_input(port->adi + i, port, i);
		if (err)
			goto fail2;
	}

	if (!port->adi[0].length && !port->adi[1].length) {
		err = -ENODEV;
		goto fail2;
	}

	gameport_set_poll_handler(gameport, adi_poll);
	gameport_set_poll_interval(gameport, 20);

	msleep(ADI_INIT_DELAY);
	if (adi_read(port)) {
		msleep(ADI_DATA_DELAY);
		adi_read(port);
	}

	for (i = 0; i < 2; i++)
		if (port->adi[i].length > 0) {
			adi_init_center(port->adi + i);
			err = input_register_device(port->adi[i].dev);
			if (err)
				goto fail3;
		}

	return 0;

 fail3: while (--i >= 0) {
		if (port->adi[i].length > 0) {
			input_unregister_device(port->adi[i].dev);
			port->adi[i].dev = NULL;
		}
	}
 fail2:	for (i = 0; i < 2; i++)
		input_free_device(port->adi[i].dev);
	gameport_close(gameport);
 fail1:	gameport_set_drvdata(gameport, NULL);
	kfree(port);
	return err;
}

static void adi_disconnect(struct gameport *gameport)
{
	int i;
	struct adi_port *port = gameport_get_drvdata(gameport);

	for (i = 0; i < 2; i++)
		if (port->adi[i].length > 0)
			input_unregister_device(port->adi[i].dev);
	gameport_close(gameport);
	gameport_set_drvdata(gameport, NULL);
	kfree(port);
}

static struct gameport_driver adi_drv = {
	.driver		= {
		.name	= "adi",
	},
	.description	= DRIVER_DESC,
	.connect	= adi_connect,
	.disconnect	= adi_disconnect,
};

module_gameport_driver(adi_drv);
Commit	Line	Data
1da177e4 LT	1	/*
	2	* Copyright (c) 1998-2005 Vojtech Pavlik
	3	*/
	4
	5	/*
	6	* Logitech ADI joystick family driver for Linux
	7	*/
	8
	9	/*
	10	* This program is free software; you can redistribute it and/or modify
	11	* it under the terms of the GNU General Public License as published by
	12	* the Free Software Foundation; either version 2 of the License, or
	13	* (at your option) any later version.
	14	*
	15	* This program is distributed in the hope that it will be useful,
	16	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	17	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	18	* GNU General Public License for more details.
	19	*
	20	* You should have received a copy of the GNU General Public License
	21	* along with this program; if not, write to the Free Software
	22	* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
	23	*
	24	* Should you need to contact me, the author, you can do so either by
	25	* e-mail - mail your message to <vojtech@ucw.cz>, or by paper mail:
	26	* Vojtech Pavlik, Simunkova 1594, Prague 8, 182 00 Czech Republic
	27	*/
	28
	29	#include <linux/delay.h>
	30	#include <linux/kernel.h>
	31	#include <linux/module.h>
	32	#include <linux/string.h>
	33	#include <linux/slab.h>
	34	#include <linux/input.h>
	35	#include <linux/gameport.h>
4e57b681	36	#include <linux/jiffies.h>
1da177e4 LT	37
	38	#define DRIVER_DESC "Logitech ADI joystick family driver"
	39
	40	MODULE_AUTHOR("Vojtech Pavlik <vojtech@ucw.cz>");
	41	MODULE_DESCRIPTION(DRIVER_DESC);
	42	MODULE_LICENSE("GPL");
	43
	44	/*
	45	* Times, array sizes, flags, ids.
	46	*/
	47
	48	#define ADI_MAX_START 200 /* Trigger to packet timeout [200us] */
	49	#define ADI_MAX_STROBE 40 /* Single bit timeout [40us] */
	50	#define ADI_INIT_DELAY 10 /* Delay after init packet [10ms] */
	51	#define ADI_DATA_DELAY 4 /* Delay after data packet [4ms] */
	52
	53	#define ADI_MAX_LENGTH 256
	54	#define ADI_MIN_LENGTH 8
	55	#define ADI_MIN_LEN_LENGTH 10
	56	#define ADI_MIN_ID_LENGTH 66
17dd3f0f	57	#define ADI_MAX_NAME_LENGTH 64
1da177e4 LT	58	#define ADI_MAX_CNAME_LENGTH 16
	59	#define ADI_MAX_PHYS_LENGTH 64
	60
	61	#define ADI_FLAG_HAT 0x04
	62	#define ADI_FLAG_10BIT 0x08
	63
	64	#define ADI_ID_TPD 0x01
	65	#define ADI_ID_WGP 0x06
	66	#define ADI_ID_WGPE 0x08
	67	#define ADI_ID_MAX 0x0a
	68
	69	/*
	70	* Names, buttons, axes ...
	71	*/
	72
	73	static char *adi_names[] = { "WingMan Extreme Digital", "ThunderPad Digital", "SideCar", "CyberMan 2",
	74	"WingMan Interceptor", "WingMan Formula", "WingMan GamePad",
	75	"WingMan Extreme Digital 3D", "WingMan GamePad Extreme",
	76	"WingMan GamePad USB", "Unknown Device %#x" };
	77
	78	static char adi_wmgpe_abs[] = { ABS_X, ABS_Y, ABS_HAT0X, ABS_HAT0Y };
	79	static char adi_wmi_abs[] = { ABS_X, ABS_Y, ABS_THROTTLE, ABS_HAT0X, ABS_HAT0Y, ABS_HAT1X, ABS_HAT1Y, ABS_HAT2X, ABS_HAT2Y };
	80	static char adi_wmed3d_abs[] = { ABS_X, ABS_Y, ABS_THROTTLE, ABS_RZ, ABS_HAT0X, ABS_HAT0Y };
	81	static char adi_cm2_abs[] = { ABS_X, ABS_Y, ABS_Z, ABS_RX, ABS_RY, ABS_RZ };
	82	static char adi_wmf_abs[] = { ABS_WHEEL, ABS_GAS, ABS_BRAKE, ABS_HAT0X, ABS_HAT0Y, ABS_HAT1X, ABS_HAT1Y, ABS_HAT2X, ABS_HAT2Y };
	83
	84	static short adi_wmgpe_key[] = { BTN_A, BTN_B, BTN_C, BTN_X, BTN_Y, BTN_Z, BTN_TL, BTN_TR, BTN_START, BTN_MODE, BTN_SELECT };
ab0c3443	85	static short adi_wmi_key[] = { BTN_TRIGGER, BTN_TOP, BTN_THUMB, BTN_TOP2, BTN_BASE, BTN_BASE2, BTN_BASE3, BTN_BASE4, BTN_EXTRA };
1da177e4 LT	86	static short adi_wmed3d_key[] = { BTN_TRIGGER, BTN_THUMB, BTN_THUMB2, BTN_TOP, BTN_TOP2, BTN_BASE, BTN_BASE2 };
	87	static short adi_cm2_key[] = { BTN_1, BTN_2, BTN_3, BTN_4, BTN_5, BTN_6, BTN_7, BTN_8 };
	88
	89	static char* adi_abs[] = { adi_wmi_abs, adi_wmgpe_abs, adi_wmf_abs, adi_cm2_abs, adi_wmi_abs, adi_wmf_abs,
	90	adi_wmgpe_abs, adi_wmed3d_abs, adi_wmgpe_abs, adi_wmgpe_abs, adi_wmi_abs };
	91
	92	static short* adi_key[] = { adi_wmi_key, adi_wmgpe_key, adi_cm2_key, adi_cm2_key, adi_wmi_key, adi_cm2_key,
	93	adi_wmgpe_key, adi_wmed3d_key, adi_wmgpe_key, adi_wmgpe_key, adi_wmi_key };
	94
	95	/*
	96	* Hat to axis conversion arrays.
	97	*/
	98
	99	static struct {
	100	int x;
	101	int y;
	102	} adi_hat_to_axis[] = {{ 0, 0}, { 0,-1}, { 1,-1}, { 1, 0}, { 1, 1}, { 0, 1}, {-1, 1}, {-1, 0}, {-1,-1}};
	103
	104	/*
	105	* Per-port information.
	106	*/
	107
	108	struct adi {
17dd3f0f	109	struct input_dev *dev;
1da177e4 LT	110	int length;
	111	int ret;
	112	int idx;
	113	unsigned char id;
	114	char buttons;
	115	char axes10;
	116	char axes8;
	117	signed char pad;
	118	char hats;
	119	char *abs;
	120	short *key;
	121	char name[ADI_MAX_NAME_LENGTH];
	122	char cname[ADI_MAX_CNAME_LENGTH];
	123	char phys[ADI_MAX_PHYS_LENGTH];
	124	unsigned char data[ADI_MAX_LENGTH];
	125	};
	126
	127	struct adi_port {
	128	struct gameport *gameport;
	129	struct adi adi[2];
	130	int bad;
	131	int reads;
	132	};
	133
	134	/*
	135	* adi_read_packet() reads a Logitech ADI packet.
	136	*/
	137
	138	static void adi_read_packet(struct adi_port *port)
	139	{
	140	struct adi *adi = port->adi;
	141	struct gameport *gameport = port->gameport;
	142	unsigned char u, v, w, x, z;
	143	int t[2], s[2], i;
	144	unsigned long flags;
	145
	146	for (i = 0; i < 2; i++) {
	147	adi[i].ret = -1;
	148	t[i] = gameport_time(gameport, ADI_MAX_START);
	149	s[i] = 0;
	150	}
	151
	152	local_irq_save(flags);
	153
	154	gameport_trigger(gameport);
	155	v = z = gameport_read(gameport);
	156
	157	do {
	158	u = v;
	159	w = u ^ (v = x = gameport_read(gameport));
	160	for (i = 0; i < 2; i++, w >>= 2, x >>= 2) {
	161	t[i]--;
	162	if ((w & 0x30) && s[i]) {
	163	if ((w & 0x30) < 0x30 && adi[i].ret < ADI_MAX_LENGTH && t[i] > 0) {
	164	adi[i].data[++adi[i].ret] = w;
	165	t[i] = gameport_time(gameport, ADI_MAX_STROBE);
	166	} else t[i] = 0;
	167	} else if (!(x & 0x30)) s[i] = 1;
	168	}
	169	} while (t[0] > 0 \|\| t[1] > 0);
	170
	171	local_irq_restore(flags);
	172
	173	return;
174	}
175
176	/*
177	* adi_move_bits() detects a possible 2-stream mode, and moves
178	* the bits accordingly.
179	*/
180
181	static void adi_move_bits(struct adi_port *port, int length)
182	{
183	int i;
184	struct adi *adi = port->adi;
185
ab0c3443	186	adi[0].idx = adi[1].idx = 0;
1da177e4 LT	187
	188	if (adi[0].ret <= 0 \|\| adi[1].ret <= 0) return;
	189	if (adi[0].data[0] & 0x20 \|\| ~adi[1].data[0] & 0x20) return;
	190
	191	for (i = 1; i <= adi[1].ret; i++)
	192	adi[0].data[((length - 1) >> 1) + i + 1] = adi[1].data[i];
	193
	194	adi[0].ret += adi[1].ret;
	195	adi[1].ret = -1;
	196	}
	197
	198	/*
	199	* adi_get_bits() gathers bits from the data packet.
	200	*/
	201
	202	static inline int adi_get_bits(struct adi *adi, int count)
	203	{
	204	int bits = 0;
	205	int i;
	206	if ((adi->idx += count) > adi->ret) return 0;
	207	for (i = 0; i < count; i++)
	208	bits \|= ((adi->data[adi->idx - i] >> 5) & 1) << i;
	209	return bits;
	210	}
	211
	212	/*
	213	* adi_decode() decodes Logitech joystick data into input events.
	214	*/
	215
	216	static int adi_decode(struct adi *adi)
	217	{
17dd3f0f	218	struct input_dev *dev = adi->dev;
1da177e4 LT	219	char *abs = adi->abs;
	220	short *key = adi->key;
	221	int i, t;
	222
	223	if (adi->ret < adi->length \|\| adi->id != (adi_get_bits(adi, 4) \| (adi_get_bits(adi, 4) << 4)))
	224	return -1;
	225
	226	for (i = 0; i < adi->axes10; i++)
	227	input_report_abs(dev, *abs++, adi_get_bits(adi, 10));
	228
	229	for (i = 0; i < adi->axes8; i++)
	230	input_report_abs(dev, *abs++, adi_get_bits(adi, 8));
	231
	232	for (i = 0; i < adi->buttons && i < 63; i++) {
	233	if (i == adi->pad) {
	234	t = adi_get_bits(adi, 4);
	235	input_report_abs(dev, *abs++, ((t >> 2) & 1) - ( t & 1));
	236	input_report_abs(dev, *abs++, ((t >> 1) & 1) - ((t >> 3) & 1));
	237	}
	238	input_report_key(dev, *key++, adi_get_bits(adi, 1));
	239	}
	240
	241	for (i = 0; i < adi->hats; i++) {
	242	if ((t = adi_get_bits(adi, 4)) > 8) t = 0;
	243	input_report_abs(dev, *abs++, adi_hat_to_axis[t].x);
	244	input_report_abs(dev, *abs++, adi_hat_to_axis[t].y);
	245	}
	246
	247	for (i = 63; i < adi->buttons; i++)
	248	input_report_key(dev, *key++, adi_get_bits(adi, 1));
	249
	250	input_sync(dev);
	251
	252	return 0;
	253	}
	254
	255	/*
	256	* adi_read() reads the data packet and decodes it.
	257	*/
	258
	259	static int adi_read(struct adi_port *port)
	260	{
	261	int i;
	262	int result = 0;
	263
	264	adi_read_packet(port);
	265	adi_move_bits(port, port->adi[0].length);
	266
	267	for (i = 0; i < 2; i++)
	268	if (port->adi[i].length)
	269	result \|= adi_decode(port->adi + i);
	270
	271	return result;
	272	}
	273
	274	/*
	275	* adi_poll() repeatedly polls the Logitech joysticks.
	276	*/
	277
	278	static void adi_poll(struct gameport *gameport)
	279	{
	280	struct adi_port *port = gameport_get_drvdata(gameport);
	281
	282	port->bad -= adi_read(port);
283	port->reads++;
284	}
285
286	/*
287	* adi_open() is a callback from the input open routine.
288	*/
289
290	static int adi_open(struct input_dev *dev)
291	{
8715c1cf	292	struct adi_port *port = input_get_drvdata(dev);
1da177e4 LT	293
	294	gameport_start_polling(port->gameport);
	295	return 0;
	296	}
	297
	298	/*
	299	* adi_close() is a callback from the input close routine.
	300	*/
	301
	302	static void adi_close(struct input_dev *dev)
	303	{
8715c1cf	304	struct adi_port *port = input_get_drvdata(dev);
1da177e4 LT	305
	306	gameport_stop_polling(port->gameport);
	307	}
	308
	309	/*
	310	* adi_init_digital() sends a trigger & delay sequence
	311	* to reset and initialize a Logitech joystick into digital mode.
	312	*/
	313
	314	static void adi_init_digital(struct gameport *gameport)
	315	{
	316	int seq[] = { 4, -2, -3, 10, -6, -11, -7, -9, 11, 0 };
	317	int i;
	318
	319	for (i = 0; seq[i]; i++) {
	320	gameport_trigger(gameport);
17dd3f0f DT	321	if (seq[i] > 0)
17dd3f0f DT	322	msleep(seq[i]);
1da177e4 LT	323	if (seq[i] < 0) {
	324	mdelay(-seq[i]);
	325	udelay(-seq[i]14); / It looks like mdelay() is off by approx 1.4% */
	326	}
	327	}
	328	}
	329
	330	static void adi_id_decode(struct adi adi, struct adi_port port)
	331	{
	332	int i, t;
	333
	334	if (adi->ret < ADI_MIN_ID_LENGTH) /* Minimum ID packet length */
	335	return;
	336
	337	if (adi->ret < (t = adi_get_bits(adi, 10))) {
	338	printk(KERN_WARNING "adi: Short ID packet: reported: %d != read: %d\n", t, adi->ret);
	339	return;
	340	}
	341
	342	adi->id = adi_get_bits(adi, 4) \| (adi_get_bits(adi, 4) << 4);
	343
	344	if ((t = adi_get_bits(adi, 4)) & ADI_FLAG_HAT) adi->hats++;
	345
	346	adi->length = adi_get_bits(adi, 10);
	347
	348	if (adi->length >= ADI_MAX_LENGTH \|\| adi->length < ADI_MIN_LENGTH) {
	349	printk(KERN_WARNING "adi: Bad data packet length (%d).\n", adi->length);
	350	adi->length = 0;
	351	return;
	352	}
	353
	354	adi->axes8 = adi_get_bits(adi, 4);
	355	adi->buttons = adi_get_bits(adi, 6);
	356
	357	if (adi_get_bits(adi, 6) != 8 && adi->hats) {
	358	printk(KERN_WARNING "adi: Other than 8-dir POVs not supported yet.\n");
	359	adi->length = 0;
	360	return;
	361	}
	362
	363	adi->buttons += adi_get_bits(adi, 6);
	364	adi->hats += adi_get_bits(adi, 4);
	365
	366	i = adi_get_bits(adi, 4);
	367
	368	if (t & ADI_FLAG_10BIT) {
	369	adi->axes10 = adi->axes8 - i;
	370	adi->axes8 = i;
	371	}
	372
	373	t = adi_get_bits(adi, 4);
	374
	375	for (i = 0; i < t; i++)
	376	adi->cname[i] = adi_get_bits(adi, 8);
	377	adi->cname[i] = 0;
	378
	379	t = 8 + adi->buttons + adi->axes10 * 10 + adi->axes8 * 8 + adi->hats * 4;
	380	if (adi->length != t && adi->length != t + (t & 1)) {
	381	printk(KERN_WARNING "adi: Expected length %d != data length %d\n", t, adi->length);
	382	adi->length = 0;
	383	return;
	384	}
	385
	386	switch (adi->id) {
387	case ADI_ID_TPD:
388	adi->pad = 4;
389	adi->buttons -= 4;
390	break;
391	case ADI_ID_WGP:
392	adi->pad = 0;
393	adi->buttons -= 4;
394	break;
395	default:
396	adi->pad = -1;
397	break;
398	}
399	}
400
17dd3f0f	401	static int adi_init_input(struct adi adi, struct adi_port port, int half)
1da177e4	402	{
17dd3f0f	403	struct input_dev *input_dev;
1da177e4	404	char buf[ADI_MAX_NAME_LENGTH];
17dd3f0f	405	int i, t;
1da177e4	406
17dd3f0f DT	407	adi->dev = input_dev = input_allocate_device();
	408	if (!input_dev)
	409	return -ENOMEM;
1da177e4 LT	410
	411	t = adi->id < ADI_ID_MAX ? adi->id : ADI_ID_MAX;
	412
	413	snprintf(buf, ADI_MAX_PHYS_LENGTH, adi_names[t], adi->id);
17dd3f0f	414	snprintf(adi->name, ADI_MAX_NAME_LENGTH, "Logitech %s [%s]", buf, adi->cname);
1da177e4 LT	415	snprintf(adi->phys, ADI_MAX_PHYS_LENGTH, "%s/input%d", port->gameport->phys, half);
	416
	417	adi->abs = adi_abs[t];
	418	adi->key = adi_key[t];
	419
17dd3f0f DT	420	input_dev->name = adi->name;
	421	input_dev->phys = adi->phys;
	422	input_dev->id.bustype = BUS_GAMEPORT;
	423	input_dev->id.vendor = GAMEPORT_ID_VENDOR_LOGITECH;
	424	input_dev->id.product = adi->id;
	425	input_dev->id.version = 0x0100;
935e658e	426	input_dev->dev.parent = &port->gameport->dev;
8715c1cf DT	427
8715c1cf DT	428	input_set_drvdata(input_dev, port);
1da177e4	429
17dd3f0f DT	430	input_dev->open = adi_open;
17dd3f0f DT	431	input_dev->close = adi_close;
1da177e4	432
7b19ada2	433	input_dev->evbit[0] = BIT_MASK(EV_KEY) \| BIT_MASK(EV_ABS);
1da177e4 LT	434
1da177e4 LT	435	for (i = 0; i < adi->axes10 + adi->axes8 + (adi->hats + (adi->pad != -1)) * 2; i++)
17dd3f0f	436	set_bit(adi->abs[i], input_dev->absbit);
1da177e4 LT	437
1da177e4 LT	438	for (i = 0; i < adi->buttons; i++)
17dd3f0f DT	439	set_bit(adi->key[i], input_dev->keybit);
	440
	441	return 0;
1da177e4 LT	442	}
	443
	444	static void adi_init_center(struct adi *adi)
	445	{
	446	int i, t, x;
	447
	448	if (!adi->length)
	449	return;
	450
	451	for (i = 0; i < adi->axes10 + adi->axes8 + (adi->hats + (adi->pad != -1)) * 2; i++) {
	452
	453	t = adi->abs[i];
987a6c02	454	x = input_abs_get_val(adi->dev, t);
1da177e4 LT	455
	456	if (t == ABS_THROTTLE \|\| t == ABS_RUDDER \|\| adi->id == ADI_ID_WGPE)
	457	x = i < adi->axes10 ? 512 : 128;
	458
	459	if (i < adi->axes10)
17dd3f0f	460	input_set_abs_params(adi->dev, t, 64, x * 2 - 64, 2, 16);
1da177e4	461	else if (i < adi->axes10 + adi->axes8)
17dd3f0f	462	input_set_abs_params(adi->dev, t, 48, x * 2 - 48, 1, 16);
1da177e4	463	else
17dd3f0f	464	input_set_abs_params(adi->dev, t, -1, 1, 0, 0);
1da177e4 LT	465	}
	466	}
	467
	468	/*
	469	* adi_connect() probes for Logitech ADI joysticks.
	470	*/
	471
	472	static int adi_connect(struct gameport gameport, struct gameport_driver drv)
	473	{
	474	struct adi_port *port;
	475	int i;
	476	int err;
	477
17dd3f0f DT	478	port = kzalloc(sizeof(struct adi_port), GFP_KERNEL);
17dd3f0f DT	479	if (!port)
1da177e4 LT	480	return -ENOMEM;
	481
	482	port->gameport = gameport;
	483
	484	gameport_set_drvdata(gameport, port);
	485
	486	err = gameport_open(gameport, drv, GAMEPORT_MODE_RAW);
17dd3f0f DT	487	if (err)
17dd3f0f DT	488	goto fail1;
1da177e4 LT	489
	490	adi_init_digital(gameport);
	491	adi_read_packet(port);
	492
	493	if (port->adi[0].ret >= ADI_MIN_LEN_LENGTH)
	494	adi_move_bits(port, adi_get_bits(port->adi, 10));
	495
	496	for (i = 0; i < 2; i++) {
	497	adi_id_decode(port->adi + i, port);
17dd3f0f DT	498
	499	if (!port->adi[i].length)
	500	continue;
	501
	502	err = adi_init_input(port->adi + i, port, i);
	503	if (err)
	504	goto fail2;
1da177e4 LT	505	}
	506
	507	if (!port->adi[0].length && !port->adi[1].length) {
17dd3f0f DT	508	err = -ENODEV;
17dd3f0f DT	509	goto fail2;
1da177e4 LT	510	}
	511
	512	gameport_set_poll_handler(gameport, adi_poll);
	513	gameport_set_poll_interval(gameport, 20);
	514
	515	msleep(ADI_INIT_DELAY);
	516	if (adi_read(port)) {
	517	msleep(ADI_DATA_DELAY);
	518	adi_read(port);
	519	}
	520
	521	for (i = 0; i < 2; i++)
	522	if (port->adi[i].length > 0) {
	523	adi_init_center(port->adi + i);
127278ce DT	524	err = input_register_device(port->adi[i].dev);
	525	if (err)
	526	goto fail3;
1da177e4 LT	527	}
	528
	529	return 0;
17dd3f0f	530
127278ce DT	531	fail3: while (--i >= 0) {
	532	if (port->adi[i].length > 0) {
	533	input_unregister_device(port->adi[i].dev);
	534	port->adi[i].dev = NULL;
	535	}
	536	}
17dd3f0f	537	fail2: for (i = 0; i < 2; i++)
de3748f6	538	input_free_device(port->adi[i].dev);
17dd3f0f DT	539	gameport_close(gameport);
	540	fail1: gameport_set_drvdata(gameport, NULL);
	541	kfree(port);
	542	return err;
1da177e4 LT	543	}
	544
	545	static void adi_disconnect(struct gameport *gameport)
	546	{
	547	int i;
	548	struct adi_port *port = gameport_get_drvdata(gameport);
	549
	550	for (i = 0; i < 2; i++)
	551	if (port->adi[i].length > 0)
17dd3f0f	552	input_unregister_device(port->adi[i].dev);
1da177e4 LT	553	gameport_close(gameport);
	554	gameport_set_drvdata(gameport, NULL);
	555	kfree(port);
	556	}
	557
1da177e4 LT	558	static struct gameport_driver adi_drv = {
	559	.driver = {
	560	.name = "adi",
	561	},
	562	.description = DRIVER_DESC,
	563	.connect = adi_connect,
	564	.disconnect = adi_disconnect,
	565	};
	566
98a84131	567	module_gameport_driver(adi_drv);