Merge 'upstream' branch of rsync://rsync.kernel.org/pub/scm/linux/kernel/git/jgarzik...
[linux-2.6] / drivers / input / joystick / tmdc.c
1 /*
2  * $Id: tmdc.c,v 1.31 2002/01/22 20:29:52 vojtech Exp $
3  *
4  *  Copyright (c) 1998-2001 Vojtech Pavlik
5  *
6  *   Based on the work of:
7  *      Trystan Larey-Williams
8  */
9
10 /*
11  * ThrustMaster DirectConnect (BSP) joystick family driver for Linux
12  */
13
14 /*
15  * This program is free software; you can redistribute it and/or modify
16  * it under the terms of the GNU General Public License as published by
17  * the Free Software Foundation; either version 2 of the License, or
18  * (at your option) any later version.
19  *
20  * This program is distributed in the hope that it will be useful,
21  * but WITHOUT ANY WARRANTY; without even the implied warranty of
22  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
23  * GNU General Public License for more details.
24  *
25  * You should have received a copy of the GNU General Public License
26  * along with this program; if not, write to the Free Software
27  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
28  *
29  * Should you need to contact me, the author, you can do so either by
30  * e-mail - mail your message to <vojtech@ucw.cz>, or by paper mail:
31  * Vojtech Pavlik, Simunkova 1594, Prague 8, 182 00 Czech Republic
32  */
33
34 #include <linux/delay.h>
35 #include <linux/kernel.h>
36 #include <linux/slab.h>
37 #include <linux/module.h>
38 #include <linux/init.h>
39 #include <linux/gameport.h>
40 #include <linux/input.h>
41
42 #define DRIVER_DESC     "ThrustMaster DirectConnect joystick driver"
43
44 MODULE_AUTHOR("Vojtech Pavlik <vojtech@ucw.cz>");
45 MODULE_DESCRIPTION(DRIVER_DESC);
46 MODULE_LICENSE("GPL");
47
48 #define TMDC_MAX_START          600     /* 600 us */
49 #define TMDC_MAX_STROBE         60      /* 60 us */
50 #define TMDC_MAX_LENGTH         13
51
52 #define TMDC_MODE_M3DI          1
53 #define TMDC_MODE_3DRP          3
54 #define TMDC_MODE_AT            4
55 #define TMDC_MODE_FM            8
56 #define TMDC_MODE_FGP           163
57
58 #define TMDC_BYTE_ID            10
59 #define TMDC_BYTE_REV           11
60 #define TMDC_BYTE_DEF           12
61
62 #define TMDC_ABS                7
63 #define TMDC_ABS_HAT            4
64 #define TMDC_BTN                16
65
66 static unsigned char tmdc_byte_a[16] = { 0, 1, 3, 4, 6, 7 };
67 static unsigned char tmdc_byte_d[16] = { 2, 5, 8, 9 };
68
69 static signed char tmdc_abs[TMDC_ABS] =
70         { ABS_X, ABS_Y, ABS_RUDDER, ABS_THROTTLE, ABS_RX, ABS_RY, ABS_RZ };
71 static signed char tmdc_abs_hat[TMDC_ABS_HAT] =
72         { ABS_HAT0X, ABS_HAT0Y, ABS_HAT1X, ABS_HAT1Y };
73 static signed char tmdc_abs_at[TMDC_ABS] =
74         { ABS_X, ABS_Y, ABS_RUDDER, -1, ABS_THROTTLE };
75 static signed char tmdc_abs_fm[TMDC_ABS] =
76         { ABS_RX, ABS_RY, ABS_X, ABS_Y };
77
78 static short tmdc_btn_pad[TMDC_BTN] =
79         { BTN_A, BTN_B, BTN_C, BTN_X, BTN_Y, BTN_Z, BTN_START, BTN_SELECT, BTN_TL, BTN_TR };
80 static short tmdc_btn_joy[TMDC_BTN] =
81         { BTN_TRIGGER, BTN_THUMB, BTN_TOP, BTN_TOP2, BTN_BASE, BTN_BASE2, BTN_THUMB2, BTN_PINKIE,
82           BTN_BASE3, BTN_BASE4, BTN_A, BTN_B, BTN_C, BTN_X, BTN_Y, BTN_Z };
83 static short tmdc_btn_fm[TMDC_BTN] =
84         { BTN_TRIGGER, BTN_C, BTN_B, BTN_A, BTN_THUMB, BTN_X, BTN_Y, BTN_Z, BTN_TOP, BTN_TOP2 };
85 static short tmdc_btn_at[TMDC_BTN] =
86         { BTN_TRIGGER, BTN_THUMB2, BTN_PINKIE, BTN_THUMB, BTN_BASE6, BTN_BASE5, BTN_BASE4,
87           BTN_BASE3, BTN_BASE2, BTN_BASE };
88
89 static struct {
90         int x;
91         int y;
92 } tmdc_hat_to_axis[] = {{ 0, 0}, { 1, 0}, { 0,-1}, {-1, 0}, { 0, 1}};
93
94 struct tmdc {
95         struct gameport *gameport;
96         struct input_dev dev[2];
97         char name[2][64];
98         char phys[2][32];
99         int mode[2];
100         signed char *abs[2];
101         short *btn[2];
102         unsigned char absc[2];
103         unsigned char btnc[2][4];
104         unsigned char btno[2][4];
105         int reads;
106         int bads;
107         unsigned char exists;
108 };
109
110 /*
111  * tmdc_read_packet() reads a ThrustMaster packet.
112  */
113
114 static int tmdc_read_packet(struct gameport *gameport, unsigned char data[2][TMDC_MAX_LENGTH])
115 {
116         unsigned char u, v, w, x;
117         unsigned long flags;
118         int i[2], j[2], t[2], p, k;
119
120         p = gameport_time(gameport, TMDC_MAX_STROBE);
121
122         for (k = 0; k < 2; k++) {
123                 t[k] = gameport_time(gameport, TMDC_MAX_START);
124                 i[k] = j[k] = 0;
125         }
126
127         local_irq_save(flags);
128         gameport_trigger(gameport);
129
130         w = gameport_read(gameport) >> 4;
131
132         do {
133                 x = w;
134                 w = gameport_read(gameport) >> 4;
135
136                 for (k = 0, v = w, u = x; k < 2; k++, v >>= 2, u >>= 2) {
137                         if (~v & u & 2) {
138                                 if (t[k] <= 0 || i[k] >= TMDC_MAX_LENGTH) continue;
139                                 t[k] = p;
140                                 if (j[k] == 0) {                                 /* Start bit */
141                                         if (~v & 1) t[k] = 0;
142                                         data[k][i[k]] = 0; j[k]++; continue;
143                                 }
144                                 if (j[k] == 9) {                                /* Stop bit */
145                                         if (v & 1) t[k] = 0;
146                                         j[k] = 0; i[k]++; continue;
147                                 }
148                                 data[k][i[k]] |= (~v & 1) << (j[k]++ - 1);      /* Data bit */
149                         }
150                         t[k]--;
151                 }
152         } while (t[0] > 0 || t[1] > 0);
153
154         local_irq_restore(flags);
155
156         return (i[0] == TMDC_MAX_LENGTH) | ((i[1] == TMDC_MAX_LENGTH) << 1);
157 }
158
159 /*
160  * tmdc_poll() reads and analyzes ThrustMaster joystick data.
161  */
162
163 static void tmdc_poll(struct gameport *gameport)
164 {
165         unsigned char data[2][TMDC_MAX_LENGTH];
166         struct tmdc *tmdc = gameport_get_drvdata(gameport);
167         struct input_dev *dev;
168         unsigned char r, bad = 0;
169         int i, j, k, l;
170
171         tmdc->reads++;
172
173         if ((r = tmdc_read_packet(tmdc->gameport, data)) != tmdc->exists)
174                 bad = 1;
175         else
176
177         for (j = 0; j < 2; j++)
178                 if (r & (1 << j) & tmdc->exists) {
179
180                         if (data[j][TMDC_BYTE_ID] != tmdc->mode[j]) {
181                                 bad = 1;
182                                 continue;
183                         }
184
185                         dev = tmdc->dev + j;
186
187                         for (i = 0; i < tmdc->absc[j]; i++) {
188                                 if (tmdc->abs[j][i] < 0) continue;
189                                 input_report_abs(dev, tmdc->abs[j][i], data[j][tmdc_byte_a[i]]);
190                         }
191
192                         switch (tmdc->mode[j]) {
193
194                                 case TMDC_MODE_M3DI:
195
196                                         i = tmdc_byte_d[0];
197                                         input_report_abs(dev, ABS_HAT0X, ((data[j][i] >> 3) & 1) - ((data[j][i] >> 1) & 1));
198                                         input_report_abs(dev, ABS_HAT0Y, ((data[j][i] >> 2) & 1) - ( data[j][i]       & 1));
199                                         break;
200
201                                 case TMDC_MODE_AT:
202
203                                         i = tmdc_byte_a[3];
204                                         input_report_abs(dev, ABS_HAT0X, tmdc_hat_to_axis[(data[j][i] - 141) / 25].x);
205                                         input_report_abs(dev, ABS_HAT0Y, tmdc_hat_to_axis[(data[j][i] - 141) / 25].y);
206                                         break;
207
208                         }
209
210                         for (k = l = 0; k < 4; k++) {
211                                 for (i = 0; i < tmdc->btnc[j][k]; i++)
212                                         input_report_key(dev, tmdc->btn[j][i + l],
213                                                 ((data[j][tmdc_byte_d[k]] >> (i + tmdc->btno[j][k])) & 1));
214                                 l += tmdc->btnc[j][k];
215                         }
216
217                         input_sync(dev);
218         }
219
220         tmdc->bads += bad;
221 }
222
223 static int tmdc_open(struct input_dev *dev)
224 {
225         struct tmdc *tmdc = dev->private;
226
227         gameport_start_polling(tmdc->gameport);
228         return 0;
229 }
230
231 static void tmdc_close(struct input_dev *dev)
232 {
233         struct tmdc *tmdc = dev->private;
234
235         gameport_stop_polling(tmdc->gameport);
236 }
237
238 /*
239  * tmdc_probe() probes for ThrustMaster type joysticks.
240  */
241
242 static int tmdc_connect(struct gameport *gameport, struct gameport_driver *drv)
243 {
244         static struct models {
245                 unsigned char id;
246                 char *name;
247                 char abs;
248                 char hats;
249                 char btnc[4];
250                 char btno[4];
251                 signed char *axes;
252                 short *buttons;
253         } models[] = {  {   1, "ThrustMaster Millenium 3D Inceptor",      6, 2, { 4, 2 }, { 4, 6 }, tmdc_abs, tmdc_btn_joy },
254                         {   3, "ThrustMaster Rage 3D Gamepad",            2, 0, { 8, 2 }, { 0, 0 }, tmdc_abs, tmdc_btn_pad },
255                         {   4, "ThrustMaster Attack Throttle",            5, 2, { 4, 6 }, { 4, 2 }, tmdc_abs_at, tmdc_btn_at },
256                         {   8, "ThrustMaster FragMaster",                 4, 0, { 8, 2 }, { 0, 0 }, tmdc_abs_fm, tmdc_btn_fm },
257                         { 163, "Thrustmaster Fusion GamePad",             2, 0, { 8, 2 }, { 0, 0 }, tmdc_abs, tmdc_btn_pad },
258                         {   0, "Unknown %d-axis, %d-button TM device %d", 0, 0, { 0, 0 }, { 0, 0 }, tmdc_abs, tmdc_btn_joy }};
259
260         unsigned char data[2][TMDC_MAX_LENGTH];
261         struct tmdc *tmdc;
262         int i, j, k, l, m;
263         int err;
264
265         if (!(tmdc = kcalloc(1, sizeof(struct tmdc), GFP_KERNEL)))
266                 return -ENOMEM;
267
268         tmdc->gameport = gameport;
269
270         gameport_set_drvdata(gameport, tmdc);
271
272         err = gameport_open(gameport, drv, GAMEPORT_MODE_RAW);
273         if (err)
274                 goto fail1;
275
276         if (!(tmdc->exists = tmdc_read_packet(gameport, data))) {
277                 err = -ENODEV;
278                 goto fail2;
279         }
280
281         gameport_set_poll_handler(gameport, tmdc_poll);
282         gameport_set_poll_interval(gameport, 20);
283
284         for (j = 0; j < 2; j++)
285                 if (tmdc->exists & (1 << j)) {
286
287                         tmdc->mode[j] = data[j][TMDC_BYTE_ID];
288
289                         for (m = 0; models[m].id && models[m].id != tmdc->mode[j]; m++);
290
291                         tmdc->abs[j] = models[m].axes;
292                         tmdc->btn[j] = models[m].buttons;
293
294                         if (!models[m].id) {
295                                 models[m].abs = data[j][TMDC_BYTE_DEF] >> 4;
296                                 for (k = 0; k < 4; k++)
297                                         models[m].btnc[k] = k < (data[j][TMDC_BYTE_DEF] & 0xf) ? 8 : 0;
298                         }
299
300                         tmdc->absc[j] = models[m].abs;
301                         for (k = 0; k < 4; k++) {
302                                 tmdc->btnc[j][k] = models[m].btnc[k];
303                                 tmdc->btno[j][k] = models[m].btno[k];
304                         }
305
306                         sprintf(tmdc->name[j], models[m].name, models[m].abs,
307                                 (data[j][TMDC_BYTE_DEF] & 0xf) << 3, tmdc->mode[j]);
308
309                         sprintf(tmdc->phys[j], "%s/input%d", gameport->phys, j);
310
311                         tmdc->dev[j].private = tmdc;
312                         tmdc->dev[j].open = tmdc_open;
313                         tmdc->dev[j].close = tmdc_close;
314
315                         tmdc->dev[j].name = tmdc->name[j];
316                         tmdc->dev[j].phys = tmdc->phys[j];
317                         tmdc->dev[j].id.bustype = BUS_GAMEPORT;
318                         tmdc->dev[j].id.vendor = GAMEPORT_ID_VENDOR_THRUSTMASTER;
319                         tmdc->dev[j].id.product = models[m].id;
320                         tmdc->dev[j].id.version = 0x0100;
321
322                         tmdc->dev[j].evbit[0] = BIT(EV_KEY) | BIT(EV_ABS);
323
324                         for (i = 0; i < models[m].abs && i < TMDC_ABS; i++)
325                                 if (tmdc->abs[j][i] >= 0)
326                                         input_set_abs_params(&tmdc->dev[j], tmdc->abs[j][i], 8, 248, 2, 4);
327
328                         for (i = 0; i < models[m].hats && i < TMDC_ABS_HAT; i++)
329                                 input_set_abs_params(&tmdc->dev[j], tmdc_abs_hat[i], -1, 1, 0, 0);
330
331
332                         for (k = l = 0; k < 4; k++) {
333                                 for (i = 0; i < models[m].btnc[k] && i < TMDC_BTN; i++)
334                                         set_bit(tmdc->btn[j][i + l], tmdc->dev[j].keybit);
335                                 l += models[m].btnc[k];
336                         }
337
338                         input_register_device(tmdc->dev + j);
339                         printk(KERN_INFO "input: %s on %s\n", tmdc->name[j], gameport->phys);
340                 }
341
342         return 0;
343
344 fail2:  gameport_close(gameport);
345 fail1:  gameport_set_drvdata(gameport, NULL);
346         kfree(tmdc);
347         return err;
348 }
349
350 static void tmdc_disconnect(struct gameport *gameport)
351 {
352         struct tmdc *tmdc = gameport_get_drvdata(gameport);
353         int i;
354
355         for (i = 0; i < 2; i++)
356                 if (tmdc->exists & (1 << i))
357                         input_unregister_device(tmdc->dev + i);
358         gameport_close(gameport);
359         gameport_set_drvdata(gameport, NULL);
360         kfree(tmdc);
361 }
362
363 static struct gameport_driver tmdc_drv = {
364         .driver         = {
365                 .name   = "tmdc",
366         },
367         .description    = DRIVER_DESC,
368         .connect        = tmdc_connect,
369         .disconnect     = tmdc_disconnect,
370 };
371
372 static int __init tmdc_init(void)
373 {
374         gameport_register_driver(&tmdc_drv);
375         return 0;
376 }
377
378 static void __exit tmdc_exit(void)
379 {
380         gameport_unregister_driver(&tmdc_drv);
381 }
382
383 module_init(tmdc_init);
384 module_exit(tmdc_exit);