1/*
2 *   Copyright (c) 2006,2007 Daniel Mack, Tim Ruetz
3 *
4 *   This program is free software; you can redistribute it and/or modify
5 *   it under the terms of the GNU General Public License as published by
6 *   the Free Software Foundation; either version 2 of the License, or
7 *   (at your option) any later version.
8 *
9 *   This program is distributed in the hope that it will be useful,
10 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
11 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 *   GNU General Public License for more details.
13 *
14 *   You should have received a copy of the GNU General Public License
15 *   along with this program; if not, write to the Free Software
16 *   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
17*/
18
19#include <linux/device.h>
20#include <linux/gfp.h>
21#include <linux/init.h>
22#include <linux/usb.h>
23#include <linux/usb/input.h>
24#include <sound/core.h>
25#include <sound/pcm.h>
26
27#include "device.h"
28#include "input.h"
29
30static unsigned short keycode_ak1[] =  { KEY_C, KEY_B, KEY_A };
31static unsigned short keycode_rk2[] =  { KEY_1, KEY_2, KEY_3, KEY_4,
32					 KEY_5, KEY_6, KEY_7 };
33static unsigned short keycode_rk3[] =  { KEY_1, KEY_2, KEY_3, KEY_4,
34					 KEY_5, KEY_6, KEY_7, KEY_8, KEY_9 };
35
36static unsigned short keycode_kore[] = {
37	KEY_FN_F1,      /* "menu"               */
38	KEY_FN_F7,      /* "lcd backlight       */
39	KEY_FN_F2,      /* "control"            */
40	KEY_FN_F3,      /* "enter"              */
41	KEY_FN_F4,      /* "view"               */
42	KEY_FN_F5,      /* "esc"                */
43	KEY_FN_F6,      /* "sound"              */
44	KEY_FN_F8,      /* array spacer, never triggered. */
45	KEY_RIGHT,
46	KEY_DOWN,
47	KEY_UP,
48	KEY_LEFT,
49	KEY_SOUND,      /* "listen"             */
50	KEY_RECORD,
51	KEY_PLAYPAUSE,
52	KEY_STOP,
53	BTN_4,          /* 8 softkeys */
54	BTN_3,
55	BTN_2,
56	BTN_1,
57	BTN_8,
58	BTN_7,
59	BTN_6,
60	BTN_5,
61	KEY_BRL_DOT4,   /* touch sensitive knobs */
62	KEY_BRL_DOT3,
63	KEY_BRL_DOT2,
64	KEY_BRL_DOT1,
65	KEY_BRL_DOT8,
66	KEY_BRL_DOT7,
67	KEY_BRL_DOT6,
68	KEY_BRL_DOT5
69};
70
71#define MASCHINE_BUTTONS   (42)
72#define MASCHINE_BUTTON(X) ((X) + BTN_MISC)
73#define MASCHINE_PADS      (16)
74#define MASCHINE_PAD(X)    ((X) + ABS_PRESSURE)
75
76static unsigned short keycode_maschine[] = {
77	MASCHINE_BUTTON(40), /* mute       */
78	MASCHINE_BUTTON(39), /* solo       */
79	MASCHINE_BUTTON(38), /* select     */
80	MASCHINE_BUTTON(37), /* duplicate  */
81	MASCHINE_BUTTON(36), /* navigate   */
82	MASCHINE_BUTTON(35), /* pad mode   */
83	MASCHINE_BUTTON(34), /* pattern    */
84	MASCHINE_BUTTON(33), /* scene      */
85	KEY_RESERVED, /* spacer */
86
87	MASCHINE_BUTTON(30), /* rec        */
88	MASCHINE_BUTTON(31), /* erase      */
89	MASCHINE_BUTTON(32), /* shift      */
90	MASCHINE_BUTTON(28), /* grid       */
91	MASCHINE_BUTTON(27), /* >          */
92	MASCHINE_BUTTON(26), /* <          */
93	MASCHINE_BUTTON(25), /* restart    */
94
95	MASCHINE_BUTTON(21), /* E          */
96	MASCHINE_BUTTON(22), /* F          */
97	MASCHINE_BUTTON(23), /* G          */
98	MASCHINE_BUTTON(24), /* H          */
99	MASCHINE_BUTTON(20), /* D          */
100	MASCHINE_BUTTON(19), /* C          */
101	MASCHINE_BUTTON(18), /* B          */
102	MASCHINE_BUTTON(17), /* A          */
103
104	MASCHINE_BUTTON(0),  /* control    */
105	MASCHINE_BUTTON(2),  /* browse     */
106	MASCHINE_BUTTON(4),  /* <          */
107	MASCHINE_BUTTON(6),  /* snap       */
108	MASCHINE_BUTTON(7),  /* autowrite  */
109	MASCHINE_BUTTON(5),  /* >          */
110	MASCHINE_BUTTON(3),  /* sampling   */
111	MASCHINE_BUTTON(1),  /* step       */
112
113	MASCHINE_BUTTON(15), /* 8 softkeys */
114	MASCHINE_BUTTON(14),
115	MASCHINE_BUTTON(13),
116	MASCHINE_BUTTON(12),
117	MASCHINE_BUTTON(11),
118	MASCHINE_BUTTON(10),
119	MASCHINE_BUTTON(9),
120	MASCHINE_BUTTON(8),
121
122	MASCHINE_BUTTON(16), /* note repeat */
123	MASCHINE_BUTTON(29)  /* play        */
124};
125
126#define KONTROLX1_INPUTS	(40)
127#define KONTROLS4_BUTTONS	(12 * 8)
128#define KONTROLS4_AXIS		(46)
129
130#define KONTROLS4_BUTTON(X)	((X) + BTN_MISC)
131#define KONTROLS4_ABS(X)	((X) + ABS_HAT0X)
132
133#define DEG90		(range / 2)
134#define DEG180		(range)
135#define DEG270		(DEG90 + DEG180)
136#define DEG360		(DEG180 * 2)
137#define HIGH_PEAK	(268)
138#define LOW_PEAK	(-7)
139
140/* some of these devices have endless rotation potentiometers
141 * built in which use two tapers, 90 degrees phase shifted.
142 * this algorithm decodes them to one single value, ranging
143 * from 0 to 999 */
144static unsigned int decode_erp(unsigned char a, unsigned char b)
145{
146	int weight_a, weight_b;
147	int pos_a, pos_b;
148	int ret;
149	int range = HIGH_PEAK - LOW_PEAK;
150	int mid_value = (HIGH_PEAK + LOW_PEAK) / 2;
151
152	weight_b = abs(mid_value - a) - (range / 2 - 100) / 2;
153
154	if (weight_b < 0)
155		weight_b = 0;
156
157	if (weight_b > 100)
158		weight_b = 100;
159
160	weight_a = 100 - weight_b;
161
162	if (a < mid_value) {
163		/* 0..90 and 270..360 degrees */
164		pos_b = b - LOW_PEAK + DEG270;
165		if (pos_b >= DEG360)
166			pos_b -= DEG360;
167	} else
168		/* 90..270 degrees */
169		pos_b = HIGH_PEAK - b + DEG90;
170
171
172	if (b > mid_value)
173		/* 0..180 degrees */
174		pos_a = a - LOW_PEAK;
175	else
176		/* 180..360 degrees */
177		pos_a = HIGH_PEAK - a + DEG180;
178
179	/* interpolate both slider values, depending on weight factors */
180	/* 0..99 x DEG360 */
181	ret = pos_a * weight_a + pos_b * weight_b;
182
183	/* normalize to 0..999 */
184	ret *= 10;
185	ret /= DEG360;
186
187	if (ret < 0)
188		ret += 1000;
189
190	if (ret >= 1000)
191		ret -= 1000;
192
193	return ret;
194}
195
196#undef DEG90
197#undef DEG180
198#undef DEG270
199#undef DEG360
200#undef HIGH_PEAK
201#undef LOW_PEAK
202
203static inline void snd_caiaq_input_report_abs(struct snd_usb_caiaqdev *cdev,
204					      int axis, const unsigned char *buf,
205					      int offset)
206{
207	input_report_abs(cdev->input_dev, axis,
208			 (buf[offset * 2] << 8) | buf[offset * 2 + 1]);
209}
210
211static void snd_caiaq_input_read_analog(struct snd_usb_caiaqdev *cdev,
212					const unsigned char *buf,
213					unsigned int len)
214{
215	struct input_dev *input_dev = cdev->input_dev;
216
217	switch (cdev->chip.usb_id) {
218	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_RIGKONTROL2):
219		snd_caiaq_input_report_abs(cdev, ABS_X, buf, 2);
220		snd_caiaq_input_report_abs(cdev, ABS_Y, buf, 0);
221		snd_caiaq_input_report_abs(cdev, ABS_Z, buf, 1);
222		break;
223	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_RIGKONTROL3):
224	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_KORECONTROLLER):
225	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_KORECONTROLLER2):
226		snd_caiaq_input_report_abs(cdev, ABS_X, buf, 0);
227		snd_caiaq_input_report_abs(cdev, ABS_Y, buf, 1);
228		snd_caiaq_input_report_abs(cdev, ABS_Z, buf, 2);
229		break;
230	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_TRAKTORKONTROLX1):
231		snd_caiaq_input_report_abs(cdev, ABS_HAT0X, buf, 4);
232		snd_caiaq_input_report_abs(cdev, ABS_HAT0Y, buf, 2);
233		snd_caiaq_input_report_abs(cdev, ABS_HAT1X, buf, 6);
234		snd_caiaq_input_report_abs(cdev, ABS_HAT1Y, buf, 1);
235		snd_caiaq_input_report_abs(cdev, ABS_HAT2X, buf, 7);
236		snd_caiaq_input_report_abs(cdev, ABS_HAT2Y, buf, 0);
237		snd_caiaq_input_report_abs(cdev, ABS_HAT3X, buf, 5);
238		snd_caiaq_input_report_abs(cdev, ABS_HAT3Y, buf, 3);
239		break;
240	}
241
242	input_sync(input_dev);
243}
244
245static void snd_caiaq_input_read_erp(struct snd_usb_caiaqdev *cdev,
246				     const char *buf, unsigned int len)
247{
248	struct input_dev *input_dev = cdev->input_dev;
249	int i;
250
251	switch (cdev->chip.usb_id) {
252	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_AK1):
253		i = decode_erp(buf[0], buf[1]);
254		input_report_abs(input_dev, ABS_X, i);
255		input_sync(input_dev);
256		break;
257	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_KORECONTROLLER):
258	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_KORECONTROLLER2):
259		i = decode_erp(buf[7], buf[5]);
260		input_report_abs(input_dev, ABS_HAT0X, i);
261		i = decode_erp(buf[12], buf[14]);
262		input_report_abs(input_dev, ABS_HAT0Y, i);
263		i = decode_erp(buf[15], buf[13]);
264		input_report_abs(input_dev, ABS_HAT1X, i);
265		i = decode_erp(buf[0], buf[2]);
266		input_report_abs(input_dev, ABS_HAT1Y, i);
267		i = decode_erp(buf[3], buf[1]);
268		input_report_abs(input_dev, ABS_HAT2X, i);
269		i = decode_erp(buf[8], buf[10]);
270		input_report_abs(input_dev, ABS_HAT2Y, i);
271		i = decode_erp(buf[11], buf[9]);
272		input_report_abs(input_dev, ABS_HAT3X, i);
273		i = decode_erp(buf[4], buf[6]);
274		input_report_abs(input_dev, ABS_HAT3Y, i);
275		input_sync(input_dev);
276		break;
277
278	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_MASCHINECONTROLLER):
279		/* 4 under the left screen */
280		input_report_abs(input_dev, ABS_HAT0X, decode_erp(buf[21], buf[20]));
281		input_report_abs(input_dev, ABS_HAT0Y, decode_erp(buf[15], buf[14]));
282		input_report_abs(input_dev, ABS_HAT1X, decode_erp(buf[9],  buf[8]));
283		input_report_abs(input_dev, ABS_HAT1Y, decode_erp(buf[3],  buf[2]));
284
285		/* 4 under the right screen */
286		input_report_abs(input_dev, ABS_HAT2X, decode_erp(buf[19], buf[18]));
287		input_report_abs(input_dev, ABS_HAT2Y, decode_erp(buf[13], buf[12]));
288		input_report_abs(input_dev, ABS_HAT3X, decode_erp(buf[7],  buf[6]));
289		input_report_abs(input_dev, ABS_HAT3Y, decode_erp(buf[1],  buf[0]));
290
291		/* volume */
292		input_report_abs(input_dev, ABS_RX, decode_erp(buf[17], buf[16]));
293		/* tempo */
294		input_report_abs(input_dev, ABS_RY, decode_erp(buf[11], buf[10]));
295		/* swing */
296		input_report_abs(input_dev, ABS_RZ, decode_erp(buf[5],  buf[4]));
297
298		input_sync(input_dev);
299		break;
300	}
301}
302
303static void snd_caiaq_input_read_io(struct snd_usb_caiaqdev *cdev,
304				    unsigned char *buf, unsigned int len)
305{
306	struct input_dev *input_dev = cdev->input_dev;
307	unsigned short *keycode = input_dev->keycode;
308	int i;
309
310	if (!keycode)
311		return;
312
313	if (input_dev->id.product == USB_PID_RIGKONTROL2)
314		for (i = 0; i < len; i++)
315			buf[i] = ~buf[i];
316
317	for (i = 0; i < input_dev->keycodemax && i < len * 8; i++)
318		input_report_key(input_dev, keycode[i],
319				 buf[i / 8] & (1 << (i % 8)));
320
321	switch (cdev->chip.usb_id) {
322	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_KORECONTROLLER):
323	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_KORECONTROLLER2):
324		input_report_abs(cdev->input_dev, ABS_MISC, 255 - buf[4]);
325		break;
326	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_TRAKTORKONTROLX1):
327		/* rotary encoders */
328		input_report_abs(cdev->input_dev, ABS_X, buf[5] & 0xf);
329		input_report_abs(cdev->input_dev, ABS_Y, buf[5] >> 4);
330		input_report_abs(cdev->input_dev, ABS_Z, buf[6] & 0xf);
331		input_report_abs(cdev->input_dev, ABS_MISC, buf[6] >> 4);
332		break;
333	}
334
335	input_sync(input_dev);
336}
337
338#define TKS4_MSGBLOCK_SIZE	16
339
340static void snd_usb_caiaq_tks4_dispatch(struct snd_usb_caiaqdev *cdev,
341					const unsigned char *buf,
342					unsigned int len)
343{
344	struct device *dev = caiaqdev_to_dev(cdev);
345
346	while (len) {
347		unsigned int i, block_id = (buf[0] << 8) | buf[1];
348
349		switch (block_id) {
350		case 0:
351			/* buttons */
352			for (i = 0; i < KONTROLS4_BUTTONS; i++)
353				input_report_key(cdev->input_dev, KONTROLS4_BUTTON(i),
354						 (buf[4 + (i / 8)] >> (i % 8)) & 1);
355			break;
356
357		case 1:
358			/* left wheel */
359			input_report_abs(cdev->input_dev, KONTROLS4_ABS(36), buf[9] | ((buf[8] & 0x3) << 8));
360			/* right wheel */
361			input_report_abs(cdev->input_dev, KONTROLS4_ABS(37), buf[13] | ((buf[12] & 0x3) << 8));
362
363			/* rotary encoders */
364			input_report_abs(cdev->input_dev, KONTROLS4_ABS(38), buf[3] & 0xf);
365			input_report_abs(cdev->input_dev, KONTROLS4_ABS(39), buf[4] >> 4);
366			input_report_abs(cdev->input_dev, KONTROLS4_ABS(40), buf[4] & 0xf);
367			input_report_abs(cdev->input_dev, KONTROLS4_ABS(41), buf[5] >> 4);
368			input_report_abs(cdev->input_dev, KONTROLS4_ABS(42), buf[5] & 0xf);
369			input_report_abs(cdev->input_dev, KONTROLS4_ABS(43), buf[6] >> 4);
370			input_report_abs(cdev->input_dev, KONTROLS4_ABS(44), buf[6] & 0xf);
371			input_report_abs(cdev->input_dev, KONTROLS4_ABS(45), buf[7] >> 4);
372			input_report_abs(cdev->input_dev, KONTROLS4_ABS(46), buf[7] & 0xf);
373
374			break;
375		case 2:
376			/* Volume Fader Channel D */
377			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(0), buf, 1);
378			/* Volume Fader Channel B */
379			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(1), buf, 2);
380			/* Volume Fader Channel A */
381			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(2), buf, 3);
382			/* Volume Fader Channel C */
383			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(3), buf, 4);
384			/* Loop Volume */
385			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(4), buf, 6);
386			/* Crossfader */
387			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(7), buf, 7);
388
389			break;
390
391		case 3:
392			/* Tempo Fader R */
393			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(6), buf, 3);
394			/* Tempo Fader L */
395			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(5), buf, 4);
396			/* Mic Volume */
397			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(8), buf, 6);
398			/* Cue Mix */
399			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(9), buf, 7);
400
401			break;
402
403		case 4:
404			/* Wheel distance sensor L */
405			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(10), buf, 1);
406			/* Wheel distance sensor R */
407			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(11), buf, 2);
408			/* Channel D EQ - Filter */
409			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(12), buf, 3);
410			/* Channel D EQ - Low */
411			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(13), buf, 4);
412			/* Channel D EQ - Mid */
413			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(14), buf, 5);
414			/* Channel D EQ - Hi */
415			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(15), buf, 6);
416			/* FX2 - dry/wet */
417			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(16), buf, 7);
418
419			break;
420
421		case 5:
422			/* FX2 - 1 */
423			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(17), buf, 1);
424			/* FX2 - 2 */
425			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(18), buf, 2);
426			/* FX2 - 3 */
427			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(19), buf, 3);
428			/* Channel B EQ - Filter */
429			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(20), buf, 4);
430			/* Channel B EQ - Low */
431			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(21), buf, 5);
432			/* Channel B EQ - Mid */
433			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(22), buf, 6);
434			/* Channel B EQ - Hi */
435			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(23), buf, 7);
436
437			break;
438
439		case 6:
440			/* Channel A EQ - Filter */
441			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(24), buf, 1);
442			/* Channel A EQ - Low */
443			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(25), buf, 2);
444			/* Channel A EQ - Mid */
445			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(26), buf, 3);
446			/* Channel A EQ - Hi */
447			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(27), buf, 4);
448			/* Channel C EQ - Filter */
449			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(28), buf, 5);
450			/* Channel C EQ - Low */
451			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(29), buf, 6);
452			/* Channel C EQ - Mid */
453			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(30), buf, 7);
454
455			break;
456
457		case 7:
458			/* Channel C EQ - Hi */
459			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(31), buf, 1);
460			/* FX1 - wet/dry */
461			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(32), buf, 2);
462			/* FX1 - 1 */
463			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(33), buf, 3);
464			/* FX1 - 2 */
465			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(34), buf, 4);
466			/* FX1 - 3 */
467			snd_caiaq_input_report_abs(cdev, KONTROLS4_ABS(35), buf, 5);
468
469			break;
470
471		default:
472			dev_dbg(dev, "%s(): bogus block (id %d)\n",
473				__func__, block_id);
474			return;
475		}
476
477		len -= TKS4_MSGBLOCK_SIZE;
478		buf += TKS4_MSGBLOCK_SIZE;
479	}
480
481	input_sync(cdev->input_dev);
482}
483
484#define MASCHINE_MSGBLOCK_SIZE 2
485
486static void snd_usb_caiaq_maschine_dispatch(struct snd_usb_caiaqdev *cdev,
487					const unsigned char *buf,
488					unsigned int len)
489{
490	unsigned int i, pad_id;
491	__le16 *pressure = (__le16 *) buf;
492
493	for (i = 0; i < MASCHINE_PADS; i++) {
494		pad_id = le16_to_cpu(*pressure) >> 12;
495		input_report_abs(cdev->input_dev, MASCHINE_PAD(pad_id),
496				 le16_to_cpu(*pressure) & 0xfff);
497		pressure++;
498	}
499
500	input_sync(cdev->input_dev);
501}
502
503static void snd_usb_caiaq_ep4_reply_dispatch(struct urb *urb)
504{
505	struct snd_usb_caiaqdev *cdev = urb->context;
506	unsigned char *buf = urb->transfer_buffer;
507	struct device *dev = &urb->dev->dev;
508	int ret;
509
510	if (urb->status || !cdev || urb != cdev->ep4_in_urb)
511		return;
512
513	switch (cdev->chip.usb_id) {
514	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_TRAKTORKONTROLX1):
515		if (urb->actual_length < 24)
516			goto requeue;
517
518		if (buf[0] & 0x3)
519			snd_caiaq_input_read_io(cdev, buf + 1, 7);
520
521		if (buf[0] & 0x4)
522			snd_caiaq_input_read_analog(cdev, buf + 8, 16);
523
524		break;
525
526	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_TRAKTORKONTROLS4):
527		snd_usb_caiaq_tks4_dispatch(cdev, buf, urb->actual_length);
528		break;
529
530	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_MASCHINECONTROLLER):
531		if (urb->actual_length < (MASCHINE_PADS * MASCHINE_MSGBLOCK_SIZE))
532			goto requeue;
533
534		snd_usb_caiaq_maschine_dispatch(cdev, buf, urb->actual_length);
535		break;
536	}
537
538requeue:
539	cdev->ep4_in_urb->actual_length = 0;
540	ret = usb_submit_urb(cdev->ep4_in_urb, GFP_ATOMIC);
541	if (ret < 0)
542		dev_err(dev, "unable to submit urb. OOM!?\n");
543}
544
545static int snd_usb_caiaq_input_open(struct input_dev *idev)
546{
547	struct snd_usb_caiaqdev *cdev = input_get_drvdata(idev);
548
549	if (!cdev)
550		return -EINVAL;
551
552	switch (cdev->chip.usb_id) {
553	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_TRAKTORKONTROLX1):
554	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_TRAKTORKONTROLS4):
555	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_MASCHINECONTROLLER):
556		if (usb_submit_urb(cdev->ep4_in_urb, GFP_KERNEL) != 0)
557			return -EIO;
558		break;
559	}
560
561	return 0;
562}
563
564static void snd_usb_caiaq_input_close(struct input_dev *idev)
565{
566	struct snd_usb_caiaqdev *cdev = input_get_drvdata(idev);
567
568	if (!cdev)
569		return;
570
571	switch (cdev->chip.usb_id) {
572	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_TRAKTORKONTROLX1):
573	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_TRAKTORKONTROLS4):
574	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_MASCHINECONTROLLER):
575		usb_kill_urb(cdev->ep4_in_urb);
576		break;
577	}
578}
579
580void snd_usb_caiaq_input_dispatch(struct snd_usb_caiaqdev *cdev,
581				  char *buf,
582				  unsigned int len)
583{
584	if (!cdev->input_dev || len < 1)
585		return;
586
587	switch (buf[0]) {
588	case EP1_CMD_READ_ANALOG:
589		snd_caiaq_input_read_analog(cdev, buf + 1, len - 1);
590		break;
591	case EP1_CMD_READ_ERP:
592		snd_caiaq_input_read_erp(cdev, buf + 1, len - 1);
593		break;
594	case EP1_CMD_READ_IO:
595		snd_caiaq_input_read_io(cdev, buf + 1, len - 1);
596		break;
597	}
598}
599
600int snd_usb_caiaq_input_init(struct snd_usb_caiaqdev *cdev)
601{
602	struct usb_device *usb_dev = cdev->chip.dev;
603	struct input_dev *input;
604	int i, ret = 0;
605
606	input = input_allocate_device();
607	if (!input)
608		return -ENOMEM;
609
610	usb_make_path(usb_dev, cdev->phys, sizeof(cdev->phys));
611	strlcat(cdev->phys, "/input0", sizeof(cdev->phys));
612
613	input->name = cdev->product_name;
614	input->phys = cdev->phys;
615	usb_to_input_id(usb_dev, &input->id);
616	input->dev.parent = &usb_dev->dev;
617
618	input_set_drvdata(input, cdev);
619
620	switch (cdev->chip.usb_id) {
621	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_RIGKONTROL2):
622		input->evbit[0] = BIT_MASK(EV_KEY) | BIT_MASK(EV_ABS);
623		input->absbit[0] = BIT_MASK(ABS_X) | BIT_MASK(ABS_Y) |
624			BIT_MASK(ABS_Z);
625		BUILD_BUG_ON(sizeof(cdev->keycode) < sizeof(keycode_rk2));
626		memcpy(cdev->keycode, keycode_rk2, sizeof(keycode_rk2));
627		input->keycodemax = ARRAY_SIZE(keycode_rk2);
628		input_set_abs_params(input, ABS_X, 0, 4096, 0, 10);
629		input_set_abs_params(input, ABS_Y, 0, 4096, 0, 10);
630		input_set_abs_params(input, ABS_Z, 0, 4096, 0, 10);
631		snd_usb_caiaq_set_auto_msg(cdev, 1, 10, 0);
632		break;
633	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_RIGKONTROL3):
634		input->evbit[0] = BIT_MASK(EV_KEY) | BIT_MASK(EV_ABS);
635		input->absbit[0] = BIT_MASK(ABS_X) | BIT_MASK(ABS_Y) |
636			BIT_MASK(ABS_Z);
637		BUILD_BUG_ON(sizeof(cdev->keycode) < sizeof(keycode_rk3));
638		memcpy(cdev->keycode, keycode_rk3, sizeof(keycode_rk3));
639		input->keycodemax = ARRAY_SIZE(keycode_rk3);
640		input_set_abs_params(input, ABS_X, 0, 1024, 0, 10);
641		input_set_abs_params(input, ABS_Y, 0, 1024, 0, 10);
642		input_set_abs_params(input, ABS_Z, 0, 1024, 0, 10);
643		snd_usb_caiaq_set_auto_msg(cdev, 1, 10, 0);
644		break;
645	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_AK1):
646		input->evbit[0] = BIT_MASK(EV_KEY) | BIT_MASK(EV_ABS);
647		input->absbit[0] = BIT_MASK(ABS_X);
648		BUILD_BUG_ON(sizeof(cdev->keycode) < sizeof(keycode_ak1));
649		memcpy(cdev->keycode, keycode_ak1, sizeof(keycode_ak1));
650		input->keycodemax = ARRAY_SIZE(keycode_ak1);
651		input_set_abs_params(input, ABS_X, 0, 999, 0, 10);
652		snd_usb_caiaq_set_auto_msg(cdev, 1, 0, 5);
653		break;
654	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_KORECONTROLLER):
655	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_KORECONTROLLER2):
656		input->evbit[0] = BIT_MASK(EV_KEY) | BIT_MASK(EV_ABS);
657		input->absbit[0] = BIT_MASK(ABS_HAT0X) | BIT_MASK(ABS_HAT0Y) |
658				   BIT_MASK(ABS_HAT1X) | BIT_MASK(ABS_HAT1Y) |
659				   BIT_MASK(ABS_HAT2X) | BIT_MASK(ABS_HAT2Y) |
660				   BIT_MASK(ABS_HAT3X) | BIT_MASK(ABS_HAT3Y) |
661				   BIT_MASK(ABS_X) | BIT_MASK(ABS_Y) |
662				   BIT_MASK(ABS_Z);
663		input->absbit[BIT_WORD(ABS_MISC)] |= BIT_MASK(ABS_MISC);
664		BUILD_BUG_ON(sizeof(cdev->keycode) < sizeof(keycode_kore));
665		memcpy(cdev->keycode, keycode_kore, sizeof(keycode_kore));
666		input->keycodemax = ARRAY_SIZE(keycode_kore);
667		input_set_abs_params(input, ABS_HAT0X, 0, 999, 0, 10);
668		input_set_abs_params(input, ABS_HAT0Y, 0, 999, 0, 10);
669		input_set_abs_params(input, ABS_HAT1X, 0, 999, 0, 10);
670		input_set_abs_params(input, ABS_HAT1Y, 0, 999, 0, 10);
671		input_set_abs_params(input, ABS_HAT2X, 0, 999, 0, 10);
672		input_set_abs_params(input, ABS_HAT2Y, 0, 999, 0, 10);
673		input_set_abs_params(input, ABS_HAT3X, 0, 999, 0, 10);
674		input_set_abs_params(input, ABS_HAT3Y, 0, 999, 0, 10);
675		input_set_abs_params(input, ABS_X, 0, 4096, 0, 10);
676		input_set_abs_params(input, ABS_Y, 0, 4096, 0, 10);
677		input_set_abs_params(input, ABS_Z, 0, 4096, 0, 10);
678		input_set_abs_params(input, ABS_MISC, 0, 255, 0, 1);
679		snd_usb_caiaq_set_auto_msg(cdev, 1, 10, 5);
680		break;
681	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_TRAKTORKONTROLX1):
682		input->evbit[0] = BIT_MASK(EV_KEY) | BIT_MASK(EV_ABS);
683		input->absbit[0] = BIT_MASK(ABS_HAT0X) | BIT_MASK(ABS_HAT0Y) |
684				   BIT_MASK(ABS_HAT1X) | BIT_MASK(ABS_HAT1Y) |
685				   BIT_MASK(ABS_HAT2X) | BIT_MASK(ABS_HAT2Y) |
686				   BIT_MASK(ABS_HAT3X) | BIT_MASK(ABS_HAT3Y) |
687				   BIT_MASK(ABS_X) | BIT_MASK(ABS_Y) |
688				   BIT_MASK(ABS_Z);
689		input->absbit[BIT_WORD(ABS_MISC)] |= BIT_MASK(ABS_MISC);
690		BUILD_BUG_ON(sizeof(cdev->keycode) < KONTROLX1_INPUTS);
691		for (i = 0; i < KONTROLX1_INPUTS; i++)
692			cdev->keycode[i] = BTN_MISC + i;
693		input->keycodemax = KONTROLX1_INPUTS;
694
695		/* analog potentiometers */
696		input_set_abs_params(input, ABS_HAT0X, 0, 4096, 0, 10);
697		input_set_abs_params(input, ABS_HAT0Y, 0, 4096, 0, 10);
698		input_set_abs_params(input, ABS_HAT1X, 0, 4096, 0, 10);
699		input_set_abs_params(input, ABS_HAT1Y, 0, 4096, 0, 10);
700		input_set_abs_params(input, ABS_HAT2X, 0, 4096, 0, 10);
701		input_set_abs_params(input, ABS_HAT2Y, 0, 4096, 0, 10);
702		input_set_abs_params(input, ABS_HAT3X, 0, 4096, 0, 10);
703		input_set_abs_params(input, ABS_HAT3Y, 0, 4096, 0, 10);
704
705		/* rotary encoders */
706		input_set_abs_params(input, ABS_X, 0, 0xf, 0, 1);
707		input_set_abs_params(input, ABS_Y, 0, 0xf, 0, 1);
708		input_set_abs_params(input, ABS_Z, 0, 0xf, 0, 1);
709		input_set_abs_params(input, ABS_MISC, 0, 0xf, 0, 1);
710
711		cdev->ep4_in_urb = usb_alloc_urb(0, GFP_KERNEL);
712		if (!cdev->ep4_in_urb) {
713			ret = -ENOMEM;
714			goto exit_free_idev;
715		}
716
717		usb_fill_bulk_urb(cdev->ep4_in_urb, usb_dev,
718				  usb_rcvbulkpipe(usb_dev, 0x4),
719				  cdev->ep4_in_buf, EP4_BUFSIZE,
720				  snd_usb_caiaq_ep4_reply_dispatch, cdev);
721
722		snd_usb_caiaq_set_auto_msg(cdev, 1, 10, 5);
723
724		break;
725
726	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_TRAKTORKONTROLS4):
727		input->evbit[0] = BIT_MASK(EV_KEY) | BIT_MASK(EV_ABS);
728		BUILD_BUG_ON(sizeof(cdev->keycode) < KONTROLS4_BUTTONS);
729		for (i = 0; i < KONTROLS4_BUTTONS; i++)
730			cdev->keycode[i] = KONTROLS4_BUTTON(i);
731		input->keycodemax = KONTROLS4_BUTTONS;
732
733		for (i = 0; i < KONTROLS4_AXIS; i++) {
734			int axis = KONTROLS4_ABS(i);
735			input->absbit[BIT_WORD(axis)] |= BIT_MASK(axis);
736		}
737
738		/* 36 analog potentiometers and faders */
739		for (i = 0; i < 36; i++)
740			input_set_abs_params(input, KONTROLS4_ABS(i), 0, 0xfff, 0, 10);
741
742		/* 2 encoder wheels */
743		input_set_abs_params(input, KONTROLS4_ABS(36), 0, 0x3ff, 0, 1);
744		input_set_abs_params(input, KONTROLS4_ABS(37), 0, 0x3ff, 0, 1);
745
746		/* 9 rotary encoders */
747		for (i = 0; i < 9; i++)
748			input_set_abs_params(input, KONTROLS4_ABS(38+i), 0, 0xf, 0, 1);
749
750		cdev->ep4_in_urb = usb_alloc_urb(0, GFP_KERNEL);
751		if (!cdev->ep4_in_urb) {
752			ret = -ENOMEM;
753			goto exit_free_idev;
754		}
755
756		usb_fill_bulk_urb(cdev->ep4_in_urb, usb_dev,
757				  usb_rcvbulkpipe(usb_dev, 0x4),
758				  cdev->ep4_in_buf, EP4_BUFSIZE,
759				  snd_usb_caiaq_ep4_reply_dispatch, cdev);
760
761		snd_usb_caiaq_set_auto_msg(cdev, 1, 10, 5);
762
763		break;
764
765	case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_MASCHINECONTROLLER):
766		input->evbit[0] = BIT_MASK(EV_KEY) | BIT_MASK(EV_ABS);
767		input->absbit[0] = BIT_MASK(ABS_HAT0X) | BIT_MASK(ABS_HAT0Y) |
768			BIT_MASK(ABS_HAT1X) | BIT_MASK(ABS_HAT1Y) |
769			BIT_MASK(ABS_HAT2X) | BIT_MASK(ABS_HAT2Y) |
770			BIT_MASK(ABS_HAT3X) | BIT_MASK(ABS_HAT3Y) |
771			BIT_MASK(ABS_RX) | BIT_MASK(ABS_RY) |
772			BIT_MASK(ABS_RZ);
773
774		BUILD_BUG_ON(sizeof(cdev->keycode) < sizeof(keycode_maschine));
775		memcpy(cdev->keycode, keycode_maschine, sizeof(keycode_maschine));
776		input->keycodemax = ARRAY_SIZE(keycode_maschine);
777
778		for (i = 0; i < MASCHINE_PADS; i++) {
779			input->absbit[0] |= MASCHINE_PAD(i);
780			input_set_abs_params(input, MASCHINE_PAD(i), 0, 0xfff, 5, 10);
781		}
782
783		input_set_abs_params(input, ABS_HAT0X, 0, 999, 0, 10);
784		input_set_abs_params(input, ABS_HAT0Y, 0, 999, 0, 10);
785		input_set_abs_params(input, ABS_HAT1X, 0, 999, 0, 10);
786		input_set_abs_params(input, ABS_HAT1Y, 0, 999, 0, 10);
787		input_set_abs_params(input, ABS_HAT2X, 0, 999, 0, 10);
788		input_set_abs_params(input, ABS_HAT2Y, 0, 999, 0, 10);
789		input_set_abs_params(input, ABS_HAT3X, 0, 999, 0, 10);
790		input_set_abs_params(input, ABS_HAT3Y, 0, 999, 0, 10);
791		input_set_abs_params(input, ABS_RX, 0, 999, 0, 10);
792		input_set_abs_params(input, ABS_RY, 0, 999, 0, 10);
793		input_set_abs_params(input, ABS_RZ, 0, 999, 0, 10);
794
795		cdev->ep4_in_urb = usb_alloc_urb(0, GFP_KERNEL);
796		if (!cdev->ep4_in_urb) {
797			ret = -ENOMEM;
798			goto exit_free_idev;
799		}
800
801		usb_fill_bulk_urb(cdev->ep4_in_urb, usb_dev,
802				  usb_rcvbulkpipe(usb_dev, 0x4),
803				  cdev->ep4_in_buf, EP4_BUFSIZE,
804				  snd_usb_caiaq_ep4_reply_dispatch, cdev);
805
806		snd_usb_caiaq_set_auto_msg(cdev, 1, 10, 5);
807		break;
808
809	default:
810		/* no input methods supported on this device */
811		goto exit_free_idev;
812	}
813
814	input->open = snd_usb_caiaq_input_open;
815	input->close = snd_usb_caiaq_input_close;
816	input->keycode = cdev->keycode;
817	input->keycodesize = sizeof(unsigned short);
818	for (i = 0; i < input->keycodemax; i++)
819		__set_bit(cdev->keycode[i], input->keybit);
820
821	cdev->input_dev = input;
822
823	ret = input_register_device(input);
824	if (ret < 0)
825		goto exit_free_idev;
826
827	return 0;
828
829exit_free_idev:
830	input_free_device(input);
831	cdev->input_dev = NULL;
832	return ret;
833}
834
835void snd_usb_caiaq_input_free(struct snd_usb_caiaqdev *cdev)
836{
837	if (!cdev || !cdev->input_dev)
838		return;
839
840	usb_kill_urb(cdev->ep4_in_urb);
841	usb_free_urb(cdev->ep4_in_urb);
842	cdev->ep4_in_urb = NULL;
843
844	input_unregister_device(cdev->input_dev);
845	cdev->input_dev = NULL;
846}
847