GNU Linux-libre 4.19.245-gnu1
[releases.git] / drivers / input / tablet / hanwang.c
1 /*
2  *  USB Hanwang tablet support
3  *
4  *  Copyright (c) 2010 Xing Wei <weixing@hanwang.com.cn>
5  *
6  */
7
8 /*
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22  *
23  */
24
25 #include <linux/types.h>
26 #include <linux/kernel.h>
27 #include <linux/slab.h>
28 #include <linux/module.h>
29 #include <linux/usb/input.h>
30
31 MODULE_AUTHOR("Xing Wei <weixing@hanwang.com.cn>");
32 MODULE_DESCRIPTION("USB Hanwang tablet driver");
33 MODULE_LICENSE("GPL");
34
35 #define USB_VENDOR_ID_HANWANG           0x0b57
36 #define HANWANG_TABLET_INT_CLASS        0x0003
37 #define HANWANG_TABLET_INT_SUB_CLASS    0x0001
38 #define HANWANG_TABLET_INT_PROTOCOL     0x0002
39
40 #define ART_MASTER_PKGLEN_MAX   10
41
42 /* device IDs */
43 #define STYLUS_DEVICE_ID        0x02
44 #define TOUCH_DEVICE_ID         0x03
45 #define CURSOR_DEVICE_ID        0x06
46 #define ERASER_DEVICE_ID        0x0A
47 #define PAD_DEVICE_ID           0x0F
48
49 /* match vendor and interface info  */
50 #define HANWANG_TABLET_DEVICE(vend, cl, sc, pr) \
51         .match_flags = USB_DEVICE_ID_MATCH_VENDOR \
52                 | USB_DEVICE_ID_MATCH_INT_INFO, \
53         .idVendor = (vend), \
54         .bInterfaceClass = (cl), \
55         .bInterfaceSubClass = (sc), \
56         .bInterfaceProtocol = (pr)
57
58 enum hanwang_tablet_type {
59         HANWANG_ART_MASTER_III,
60         HANWANG_ART_MASTER_HD,
61         HANWANG_ART_MASTER_II,
62 };
63
64 struct hanwang {
65         unsigned char *data;
66         dma_addr_t data_dma;
67         struct input_dev *dev;
68         struct usb_device *usbdev;
69         struct urb *irq;
70         const struct hanwang_features *features;
71         unsigned int current_tool;
72         unsigned int current_id;
73         char name[64];
74         char phys[32];
75 };
76
77 struct hanwang_features {
78         unsigned short pid;
79         char *name;
80         enum hanwang_tablet_type type;
81         int pkg_len;
82         int max_x;
83         int max_y;
84         int max_tilt_x;
85         int max_tilt_y;
86         int max_pressure;
87 };
88
89 static const struct hanwang_features features_array[] = {
90         { 0x8528, "Hanwang Art Master III 0906", HANWANG_ART_MASTER_III,
91           ART_MASTER_PKGLEN_MAX, 0x5757, 0x3692, 0x3f, 0x7f, 2048 },
92         { 0x8529, "Hanwang Art Master III 0604", HANWANG_ART_MASTER_III,
93           ART_MASTER_PKGLEN_MAX, 0x3d84, 0x2672, 0x3f, 0x7f, 2048 },
94         { 0x852a, "Hanwang Art Master III 1308", HANWANG_ART_MASTER_III,
95           ART_MASTER_PKGLEN_MAX, 0x7f00, 0x4f60, 0x3f, 0x7f, 2048 },
96         { 0x8401, "Hanwang Art Master HD 5012", HANWANG_ART_MASTER_HD,
97           ART_MASTER_PKGLEN_MAX, 0x678e, 0x4150, 0x3f, 0x7f, 1024 },
98         { 0x8503, "Hanwang Art Master II", HANWANG_ART_MASTER_II,
99           ART_MASTER_PKGLEN_MAX, 0x27de, 0x1cfe, 0x3f, 0x7f, 1024 },
100 };
101
102 static const int hw_eventtypes[] = {
103         EV_KEY, EV_ABS, EV_MSC,
104 };
105
106 static const int hw_absevents[] = {
107         ABS_X, ABS_Y, ABS_TILT_X, ABS_TILT_Y, ABS_WHEEL,
108         ABS_RX, ABS_RY, ABS_PRESSURE, ABS_MISC,
109 };
110
111 static const int hw_btnevents[] = {
112         BTN_STYLUS, BTN_STYLUS2, BTN_TOOL_PEN, BTN_TOOL_RUBBER,
113         BTN_TOOL_MOUSE, BTN_TOOL_FINGER,
114         BTN_0, BTN_1, BTN_2, BTN_3, BTN_4, BTN_5, BTN_6, BTN_7, BTN_8,
115 };
116
117 static const int hw_mscevents[] = {
118         MSC_SERIAL,
119 };
120
121 static void hanwang_parse_packet(struct hanwang *hanwang)
122 {
123         unsigned char *data = hanwang->data;
124         struct input_dev *input_dev = hanwang->dev;
125         struct usb_device *dev = hanwang->usbdev;
126         enum hanwang_tablet_type type = hanwang->features->type;
127         int i;
128         u16 p;
129
130         if (type == HANWANG_ART_MASTER_II) {
131                 hanwang->current_tool = BTN_TOOL_PEN;
132                 hanwang->current_id = STYLUS_DEVICE_ID;
133         }
134
135         switch (data[0]) {
136         case 0x02:      /* data packet */
137                 switch (data[1]) {
138                 case 0x80:      /* tool prox out */
139                         if (type != HANWANG_ART_MASTER_II) {
140                                 hanwang->current_id = 0;
141                                 input_report_key(input_dev,
142                                                  hanwang->current_tool, 0);
143                         }
144                         break;
145
146                 case 0x00:      /* artmaster ii pen leave */
147                         if (type == HANWANG_ART_MASTER_II) {
148                                 hanwang->current_id = 0;
149                                 input_report_key(input_dev,
150                                                  hanwang->current_tool, 0);
151                         }
152                         break;
153
154                 case 0xc2:      /* first time tool prox in */
155                         switch (data[3] & 0xf0) {
156                         case 0x20:      /* art_master III */
157                         case 0x30:      /* art_master_HD */
158                                 hanwang->current_id = STYLUS_DEVICE_ID;
159                                 hanwang->current_tool = BTN_TOOL_PEN;
160                                 input_report_key(input_dev, BTN_TOOL_PEN, 1);
161                                 break;
162                         case 0xa0:      /* art_master III */
163                         case 0xb0:      /* art_master_HD */
164                                 hanwang->current_id = ERASER_DEVICE_ID;
165                                 hanwang->current_tool = BTN_TOOL_RUBBER;
166                                 input_report_key(input_dev, BTN_TOOL_RUBBER, 1);
167                                 break;
168                         default:
169                                 hanwang->current_id = 0;
170                                 dev_dbg(&dev->dev,
171                                         "unknown tablet tool %02x\n", data[0]);
172                                 break;
173                         }
174                         break;
175
176                 default:        /* tool data packet */
177                         switch (type) {
178                         case HANWANG_ART_MASTER_III:
179                                 p = (data[6] << 3) |
180                                     ((data[7] & 0xc0) >> 5) |
181                                     (data[1] & 0x01);
182                                 break;
183
184                         case HANWANG_ART_MASTER_HD:
185                         case HANWANG_ART_MASTER_II:
186                                 p = (data[7] >> 6) | (data[6] << 2);
187                                 break;
188
189                         default:
190                                 p = 0;
191                                 break;
192                         }
193
194                         input_report_abs(input_dev, ABS_X,
195                                          be16_to_cpup((__be16 *)&data[2]));
196                         input_report_abs(input_dev, ABS_Y,
197                                          be16_to_cpup((__be16 *)&data[4]));
198                         input_report_abs(input_dev, ABS_PRESSURE, p);
199                         input_report_abs(input_dev, ABS_TILT_X, data[7] & 0x3f);
200                         input_report_abs(input_dev, ABS_TILT_Y, data[8] & 0x7f);
201                         input_report_key(input_dev, BTN_STYLUS, data[1] & 0x02);
202
203                         if (type != HANWANG_ART_MASTER_II)
204                                 input_report_key(input_dev, BTN_STYLUS2,
205                                                  data[1] & 0x04);
206                         else
207                                 input_report_key(input_dev, BTN_TOOL_PEN, 1);
208
209                         break;
210                 }
211
212                 input_report_abs(input_dev, ABS_MISC, hanwang->current_id);
213                 input_event(input_dev, EV_MSC, MSC_SERIAL,
214                                 hanwang->features->pid);
215                 break;
216
217         case 0x0c:
218                 /* roll wheel */
219                 hanwang->current_id = PAD_DEVICE_ID;
220
221                 switch (type) {
222                 case HANWANG_ART_MASTER_III:
223                         input_report_key(input_dev, BTN_TOOL_FINGER,
224                                          data[1] || data[2] || data[3]);
225                         input_report_abs(input_dev, ABS_WHEEL, data[1]);
226                         input_report_key(input_dev, BTN_0, data[2]);
227                         for (i = 0; i < 8; i++)
228                                 input_report_key(input_dev,
229                                          BTN_1 + i, data[3] & (1 << i));
230                         break;
231
232                 case HANWANG_ART_MASTER_HD:
233                         input_report_key(input_dev, BTN_TOOL_FINGER, data[1] ||
234                                         data[2] || data[3] || data[4] ||
235                                         data[5] || data[6]);
236                         input_report_abs(input_dev, ABS_RX,
237                                         ((data[1] & 0x1f) << 8) | data[2]);
238                         input_report_abs(input_dev, ABS_RY,
239                                         ((data[3] & 0x1f) << 8) | data[4]);
240                         input_report_key(input_dev, BTN_0, data[5] & 0x01);
241                         for (i = 0; i < 4; i++) {
242                                 input_report_key(input_dev,
243                                          BTN_1 + i, data[5] & (1 << i));
244                                 input_report_key(input_dev,
245                                          BTN_5 + i, data[6] & (1 << i));
246                         }
247                         break;
248
249                 case HANWANG_ART_MASTER_II:
250                         dev_dbg(&dev->dev, "error packet  %02x\n", data[0]);
251                         return;
252                 }
253
254                 input_report_abs(input_dev, ABS_MISC, hanwang->current_id);
255                 input_event(input_dev, EV_MSC, MSC_SERIAL, 0xffffffff);
256                 break;
257
258         default:
259                 dev_dbg(&dev->dev, "error packet  %02x\n", data[0]);
260                 break;
261         }
262
263         input_sync(input_dev);
264 }
265
266 static void hanwang_irq(struct urb *urb)
267 {
268         struct hanwang *hanwang = urb->context;
269         struct usb_device *dev = hanwang->usbdev;
270         int retval;
271
272         switch (urb->status) {
273         case 0:
274                 /* success */;
275                 hanwang_parse_packet(hanwang);
276                 break;
277         case -ECONNRESET:
278         case -ENOENT:
279         case -ESHUTDOWN:
280                 /* this urb is terminated, clean up */
281                 dev_err(&dev->dev, "%s - urb shutting down with status: %d",
282                         __func__, urb->status);
283                 return;
284         default:
285                 dev_err(&dev->dev, "%s - nonzero urb status received: %d",
286                         __func__, urb->status);
287                 break;
288         }
289
290         retval = usb_submit_urb(urb, GFP_ATOMIC);
291         if (retval)
292                 dev_err(&dev->dev, "%s - usb_submit_urb failed with result %d",
293                         __func__, retval);
294 }
295
296 static int hanwang_open(struct input_dev *dev)
297 {
298         struct hanwang *hanwang = input_get_drvdata(dev);
299
300         hanwang->irq->dev = hanwang->usbdev;
301         if (usb_submit_urb(hanwang->irq, GFP_KERNEL))
302                 return -EIO;
303
304         return 0;
305 }
306
307 static void hanwang_close(struct input_dev *dev)
308 {
309         struct hanwang *hanwang = input_get_drvdata(dev);
310
311         usb_kill_urb(hanwang->irq);
312 }
313
314 static bool get_features(struct usb_device *dev, struct hanwang *hanwang)
315 {
316         int i;
317
318         for (i = 0; i < ARRAY_SIZE(features_array); i++) {
319                 if (le16_to_cpu(dev->descriptor.idProduct) ==
320                                 features_array[i].pid) {
321                         hanwang->features = &features_array[i];
322                         return true;
323                 }
324         }
325
326         return false;
327 }
328
329
330 static int hanwang_probe(struct usb_interface *intf, const struct usb_device_id *id)
331 {
332         struct usb_device *dev = interface_to_usbdev(intf);
333         struct usb_endpoint_descriptor *endpoint;
334         struct hanwang *hanwang;
335         struct input_dev *input_dev;
336         int error;
337         int i;
338
339         if (intf->cur_altsetting->desc.bNumEndpoints < 1)
340                 return -ENODEV;
341
342         hanwang = kzalloc(sizeof(struct hanwang), GFP_KERNEL);
343         input_dev = input_allocate_device();
344         if (!hanwang || !input_dev) {
345                 error = -ENOMEM;
346                 goto fail1;
347         }
348
349         if (!get_features(dev, hanwang)) {
350                 error = -ENXIO;
351                 goto fail1;
352         }
353
354         hanwang->data = usb_alloc_coherent(dev, hanwang->features->pkg_len,
355                                         GFP_KERNEL, &hanwang->data_dma);
356         if (!hanwang->data) {
357                 error = -ENOMEM;
358                 goto fail1;
359         }
360
361         hanwang->irq = usb_alloc_urb(0, GFP_KERNEL);
362         if (!hanwang->irq) {
363                 error = -ENOMEM;
364                 goto fail2;
365         }
366
367         hanwang->usbdev = dev;
368         hanwang->dev = input_dev;
369
370         usb_make_path(dev, hanwang->phys, sizeof(hanwang->phys));
371         strlcat(hanwang->phys, "/input0", sizeof(hanwang->phys));
372
373         strlcpy(hanwang->name, hanwang->features->name, sizeof(hanwang->name));
374         input_dev->name = hanwang->name;
375         input_dev->phys = hanwang->phys;
376         usb_to_input_id(dev, &input_dev->id);
377         input_dev->dev.parent = &intf->dev;
378
379         input_set_drvdata(input_dev, hanwang);
380
381         input_dev->open = hanwang_open;
382         input_dev->close = hanwang_close;
383
384         for (i = 0; i < ARRAY_SIZE(hw_eventtypes); ++i)
385                 __set_bit(hw_eventtypes[i], input_dev->evbit);
386
387         for (i = 0; i < ARRAY_SIZE(hw_absevents); ++i)
388                 __set_bit(hw_absevents[i], input_dev->absbit);
389
390         for (i = 0; i < ARRAY_SIZE(hw_btnevents); ++i)
391                 __set_bit(hw_btnevents[i], input_dev->keybit);
392
393         for (i = 0; i < ARRAY_SIZE(hw_mscevents); ++i)
394                 __set_bit(hw_mscevents[i], input_dev->mscbit);
395
396         input_set_abs_params(input_dev, ABS_X,
397                              0, hanwang->features->max_x, 4, 0);
398         input_set_abs_params(input_dev, ABS_Y,
399                              0, hanwang->features->max_y, 4, 0);
400         input_set_abs_params(input_dev, ABS_TILT_X,
401                              0, hanwang->features->max_tilt_x, 0, 0);
402         input_set_abs_params(input_dev, ABS_TILT_Y,
403                              0, hanwang->features->max_tilt_y, 0, 0);
404         input_set_abs_params(input_dev, ABS_PRESSURE,
405                              0, hanwang->features->max_pressure, 0, 0);
406
407         endpoint = &intf->cur_altsetting->endpoint[0].desc;
408         usb_fill_int_urb(hanwang->irq, dev,
409                         usb_rcvintpipe(dev, endpoint->bEndpointAddress),
410                         hanwang->data, hanwang->features->pkg_len,
411                         hanwang_irq, hanwang, endpoint->bInterval);
412         hanwang->irq->transfer_dma = hanwang->data_dma;
413         hanwang->irq->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
414
415         error = input_register_device(hanwang->dev);
416         if (error)
417                 goto fail3;
418
419         usb_set_intfdata(intf, hanwang);
420
421         return 0;
422
423  fail3: usb_free_urb(hanwang->irq);
424  fail2: usb_free_coherent(dev, hanwang->features->pkg_len,
425                         hanwang->data, hanwang->data_dma);
426  fail1: input_free_device(input_dev);
427         kfree(hanwang);
428         return error;
429
430 }
431
432 static void hanwang_disconnect(struct usb_interface *intf)
433 {
434         struct hanwang *hanwang = usb_get_intfdata(intf);
435
436         input_unregister_device(hanwang->dev);
437         usb_free_urb(hanwang->irq);
438         usb_free_coherent(interface_to_usbdev(intf),
439                         hanwang->features->pkg_len, hanwang->data,
440                         hanwang->data_dma);
441         kfree(hanwang);
442         usb_set_intfdata(intf, NULL);
443 }
444
445 static const struct usb_device_id hanwang_ids[] = {
446         { HANWANG_TABLET_DEVICE(USB_VENDOR_ID_HANWANG, HANWANG_TABLET_INT_CLASS,
447                 HANWANG_TABLET_INT_SUB_CLASS, HANWANG_TABLET_INT_PROTOCOL) },
448         {}
449 };
450
451 MODULE_DEVICE_TABLE(usb, hanwang_ids);
452
453 static struct usb_driver hanwang_driver = {
454         .name           = "hanwang",
455         .probe          = hanwang_probe,
456         .disconnect     = hanwang_disconnect,
457         .id_table       = hanwang_ids,
458 };
459
460 module_usb_driver(hanwang_driver);