Merge branch 'fix/misc' into topic/misc
[pandora-kernel.git] / sound / usb / caiaq / audio.c
1 /*
2  *   Copyright (c) 2006-2008 Daniel Mack, Karsten Wiese
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/spinlock.h>
20 #include <linux/slab.h>
21 #include <linux/init.h>
22 #include <linux/usb.h>
23 #include <sound/core.h>
24 #include <sound/pcm.h>
25
26 #include "device.h"
27 #include "audio.h"
28
29 #define N_URBS                  32
30 #define CLOCK_DRIFT_TOLERANCE   5
31 #define FRAMES_PER_URB          8
32 #define BYTES_PER_FRAME         512
33 #define CHANNELS_PER_STREAM     2
34 #define BYTES_PER_SAMPLE        3
35 #define BYTES_PER_SAMPLE_USB    4
36 #define MAX_BUFFER_SIZE         (128*1024)
37 #define MAX_ENDPOINT_SIZE       512
38
39 #define ENDPOINT_CAPTURE        2
40 #define ENDPOINT_PLAYBACK       6
41
42 #define MAKE_CHECKBYTE(dev,stream,i) \
43         (stream << 1) | (~(i / (dev->n_streams * BYTES_PER_SAMPLE_USB)) & 1)
44
45 static struct snd_pcm_hardware snd_usb_caiaq_pcm_hardware = {
46         .info           = (SNDRV_PCM_INFO_MMAP | SNDRV_PCM_INFO_INTERLEAVED |
47                            SNDRV_PCM_INFO_BLOCK_TRANSFER),
48         .formats        = SNDRV_PCM_FMTBIT_S24_3BE,
49         .rates          = (SNDRV_PCM_RATE_44100 | SNDRV_PCM_RATE_48000 |
50                            SNDRV_PCM_RATE_96000),
51         .rate_min       = 44100,
52         .rate_max       = 0, /* will overwrite later */
53         .channels_min   = CHANNELS_PER_STREAM,
54         .channels_max   = CHANNELS_PER_STREAM,
55         .buffer_bytes_max = MAX_BUFFER_SIZE,
56         .period_bytes_min = 128,
57         .period_bytes_max = MAX_BUFFER_SIZE,
58         .periods_min    = 1,
59         .periods_max    = 1024,
60 };
61
62 static void
63 activate_substream(struct snd_usb_caiaqdev *dev,
64                    struct snd_pcm_substream *sub)
65 {
66         spin_lock(&dev->spinlock);
67
68         if (sub->stream == SNDRV_PCM_STREAM_PLAYBACK)
69                 dev->sub_playback[sub->number] = sub;
70         else
71                 dev->sub_capture[sub->number] = sub;
72
73         spin_unlock(&dev->spinlock);
74 }
75
76 static void
77 deactivate_substream(struct snd_usb_caiaqdev *dev,
78                      struct snd_pcm_substream *sub)
79 {
80         unsigned long flags;
81         spin_lock_irqsave(&dev->spinlock, flags);
82
83         if (sub->stream == SNDRV_PCM_STREAM_PLAYBACK)
84                 dev->sub_playback[sub->number] = NULL;
85         else
86                 dev->sub_capture[sub->number] = NULL;
87
88         spin_unlock_irqrestore(&dev->spinlock, flags);
89 }
90
91 static int
92 all_substreams_zero(struct snd_pcm_substream **subs)
93 {
94         int i;
95         for (i = 0; i < MAX_STREAMS; i++)
96                 if (subs[i] != NULL)
97                         return 0;
98         return 1;
99 }
100
101 static int stream_start(struct snd_usb_caiaqdev *dev)
102 {
103         int i, ret;
104
105         debug("%s(%p)\n", __func__, dev);
106
107         if (dev->streaming)
108                 return -EINVAL;
109
110         memset(dev->sub_playback, 0, sizeof(dev->sub_playback));
111         memset(dev->sub_capture, 0, sizeof(dev->sub_capture));
112         dev->input_panic = 0;
113         dev->output_panic = 0;
114         dev->first_packet = 4;
115         dev->streaming = 1;
116         dev->warned = 0;
117
118         for (i = 0; i < N_URBS; i++) {
119                 ret = usb_submit_urb(dev->data_urbs_in[i], GFP_ATOMIC);
120                 if (ret) {
121                         log("unable to trigger read #%d! (ret %d)\n", i, ret);
122                         dev->streaming = 0;
123                         return -EPIPE;
124                 }
125         }
126
127         return 0;
128 }
129
130 static void stream_stop(struct snd_usb_caiaqdev *dev)
131 {
132         int i;
133
134         debug("%s(%p)\n", __func__, dev);
135         if (!dev->streaming)
136                 return;
137
138         dev->streaming = 0;
139
140         for (i = 0; i < N_URBS; i++) {
141                 usb_kill_urb(dev->data_urbs_in[i]);
142                 usb_kill_urb(dev->data_urbs_out[i]);
143         }
144 }
145
146 static int snd_usb_caiaq_substream_open(struct snd_pcm_substream *substream)
147 {
148         struct snd_usb_caiaqdev *dev = snd_pcm_substream_chip(substream);
149         debug("%s(%p)\n", __func__, substream);
150         substream->runtime->hw = dev->pcm_info;
151         snd_pcm_limit_hw_rates(substream->runtime);
152         return 0;
153 }
154
155 static int snd_usb_caiaq_substream_close(struct snd_pcm_substream *substream)
156 {
157         struct snd_usb_caiaqdev *dev = snd_pcm_substream_chip(substream);
158
159         debug("%s(%p)\n", __func__, substream);
160         if (all_substreams_zero(dev->sub_playback) &&
161             all_substreams_zero(dev->sub_capture)) {
162                 /* when the last client has stopped streaming,
163                  * all sample rates are allowed again */
164                 stream_stop(dev);
165                 dev->pcm_info.rates = dev->samplerates;
166         }
167
168         return 0;
169 }
170
171 static int snd_usb_caiaq_pcm_hw_params(struct snd_pcm_substream *sub,
172                                        struct snd_pcm_hw_params *hw_params)
173 {
174         debug("%s(%p)\n", __func__, sub);
175         return snd_pcm_lib_malloc_pages(sub, params_buffer_bytes(hw_params));
176 }
177
178 static int snd_usb_caiaq_pcm_hw_free(struct snd_pcm_substream *sub)
179 {
180         struct snd_usb_caiaqdev *dev = snd_pcm_substream_chip(sub);
181         debug("%s(%p)\n", __func__, sub);
182         deactivate_substream(dev, sub);
183         return snd_pcm_lib_free_pages(sub);
184 }
185
186 /* this should probably go upstream */
187 #if SNDRV_PCM_RATE_5512 != 1 << 0 || SNDRV_PCM_RATE_192000 != 1 << 12
188 #error "Change this table"
189 #endif
190
191 static unsigned int rates[] = { 5512, 8000, 11025, 16000, 22050, 32000, 44100,
192                                 48000, 64000, 88200, 96000, 176400, 192000 };
193
194 static int snd_usb_caiaq_pcm_prepare(struct snd_pcm_substream *substream)
195 {
196         int bytes_per_sample, bpp, ret, i;
197         int index = substream->number;
198         struct snd_usb_caiaqdev *dev = snd_pcm_substream_chip(substream);
199         struct snd_pcm_runtime *runtime = substream->runtime;
200
201         debug("%s(%p)\n", __func__, substream);
202
203         if (substream->stream == SNDRV_PCM_STREAM_PLAYBACK) {
204                 int out_pos;
205
206                 switch (dev->spec.data_alignment) {
207                 case 0:
208                 case 2:
209                         out_pos = BYTES_PER_SAMPLE + 1;
210                         break;
211                 case 3:
212                 default:
213                         out_pos = 0;
214                         break;
215                 }
216
217                 dev->period_out_count[index] = out_pos;
218                 dev->audio_out_buf_pos[index] = out_pos;
219         } else {
220                 int in_pos;
221
222                 switch (dev->spec.data_alignment) {
223                 case 0:
224                         in_pos = BYTES_PER_SAMPLE + 2;
225                         break;
226                 case 2:
227                         in_pos = BYTES_PER_SAMPLE;
228                         break;
229                 case 3:
230                 default:
231                         in_pos = 0;
232                         break;
233                 }
234
235                 dev->period_in_count[index] = in_pos;
236                 dev->audio_in_buf_pos[index] = in_pos;
237         }
238
239         if (dev->streaming)
240                 return 0;
241
242         /* the first client that opens a stream defines the sample rate
243          * setting for all subsequent calls, until the last client closed. */
244         for (i=0; i < ARRAY_SIZE(rates); i++)
245                 if (runtime->rate == rates[i])
246                         dev->pcm_info.rates = 1 << i;
247
248         snd_pcm_limit_hw_rates(runtime);
249
250         bytes_per_sample = BYTES_PER_SAMPLE;
251         if (dev->spec.data_alignment >= 2)
252                 bytes_per_sample++;
253
254         bpp = ((runtime->rate / 8000) + CLOCK_DRIFT_TOLERANCE)
255                 * bytes_per_sample * CHANNELS_PER_STREAM * dev->n_streams;
256
257         if (bpp > MAX_ENDPOINT_SIZE)
258                 bpp = MAX_ENDPOINT_SIZE;
259
260         ret = snd_usb_caiaq_set_audio_params(dev, runtime->rate,
261                                              runtime->sample_bits, bpp);
262         if (ret)
263                 return ret;
264
265         ret = stream_start(dev);
266         if (ret)
267                 return ret;
268
269         dev->output_running = 0;
270         wait_event_timeout(dev->prepare_wait_queue, dev->output_running, HZ);
271         if (!dev->output_running) {
272                 stream_stop(dev);
273                 return -EPIPE;
274         }
275
276         return 0;
277 }
278
279 static int snd_usb_caiaq_pcm_trigger(struct snd_pcm_substream *sub, int cmd)
280 {
281         struct snd_usb_caiaqdev *dev = snd_pcm_substream_chip(sub);
282
283         debug("%s(%p) cmd %d\n", __func__, sub, cmd);
284
285         switch (cmd) {
286         case SNDRV_PCM_TRIGGER_START:
287         case SNDRV_PCM_TRIGGER_PAUSE_RELEASE:
288                 activate_substream(dev, sub);
289                 break;
290         case SNDRV_PCM_TRIGGER_STOP:
291         case SNDRV_PCM_TRIGGER_PAUSE_PUSH:
292                 deactivate_substream(dev, sub);
293                 break;
294         default:
295                 return -EINVAL;
296         }
297
298         return 0;
299 }
300
301 static snd_pcm_uframes_t
302 snd_usb_caiaq_pcm_pointer(struct snd_pcm_substream *sub)
303 {
304         int index = sub->number;
305         struct snd_usb_caiaqdev *dev = snd_pcm_substream_chip(sub);
306         snd_pcm_uframes_t ptr;
307
308         spin_lock(&dev->spinlock);
309
310         if (dev->input_panic || dev->output_panic)
311                 ptr = SNDRV_PCM_POS_XRUN;
312
313         if (sub->stream == SNDRV_PCM_STREAM_PLAYBACK)
314                 ptr = bytes_to_frames(sub->runtime,
315                                         dev->audio_out_buf_pos[index]);
316         else
317                 ptr = bytes_to_frames(sub->runtime,
318                                         dev->audio_in_buf_pos[index]);
319
320         spin_unlock(&dev->spinlock);
321         return ptr;
322 }
323
324 /* operators for both playback and capture */
325 static struct snd_pcm_ops snd_usb_caiaq_ops = {
326         .open =         snd_usb_caiaq_substream_open,
327         .close =        snd_usb_caiaq_substream_close,
328         .ioctl =        snd_pcm_lib_ioctl,
329         .hw_params =    snd_usb_caiaq_pcm_hw_params,
330         .hw_free =      snd_usb_caiaq_pcm_hw_free,
331         .prepare =      snd_usb_caiaq_pcm_prepare,
332         .trigger =      snd_usb_caiaq_pcm_trigger,
333         .pointer =      snd_usb_caiaq_pcm_pointer
334 };
335
336 static void check_for_elapsed_periods(struct snd_usb_caiaqdev *dev,
337                                       struct snd_pcm_substream **subs)
338 {
339         int stream, pb, *cnt;
340         struct snd_pcm_substream *sub;
341
342         for (stream = 0; stream < dev->n_streams; stream++) {
343                 sub = subs[stream];
344                 if (!sub)
345                         continue;
346
347                 pb = snd_pcm_lib_period_bytes(sub);
348                 cnt = (sub->stream == SNDRV_PCM_STREAM_PLAYBACK) ?
349                                         &dev->period_out_count[stream] :
350                                         &dev->period_in_count[stream];
351
352                 if (*cnt >= pb) {
353                         snd_pcm_period_elapsed(sub);
354                         *cnt %= pb;
355                 }
356         }
357 }
358
359 static void read_in_urb_mode0(struct snd_usb_caiaqdev *dev,
360                               const struct urb *urb,
361                               const struct usb_iso_packet_descriptor *iso)
362 {
363         unsigned char *usb_buf = urb->transfer_buffer + iso->offset;
364         struct snd_pcm_substream *sub;
365         int stream, i;
366
367         if (all_substreams_zero(dev->sub_capture))
368                 return;
369
370         for (i = 0; i < iso->actual_length;) {
371                 for (stream = 0; stream < dev->n_streams; stream++, i++) {
372                         sub = dev->sub_capture[stream];
373                         if (sub) {
374                                 struct snd_pcm_runtime *rt = sub->runtime;
375                                 char *audio_buf = rt->dma_area;
376                                 int sz = frames_to_bytes(rt, rt->buffer_size);
377                                 audio_buf[dev->audio_in_buf_pos[stream]++]
378                                         = usb_buf[i];
379                                 dev->period_in_count[stream]++;
380                                 if (dev->audio_in_buf_pos[stream] == sz)
381                                         dev->audio_in_buf_pos[stream] = 0;
382                         }
383                 }
384         }
385 }
386
387 static void read_in_urb_mode2(struct snd_usb_caiaqdev *dev,
388                               const struct urb *urb,
389                               const struct usb_iso_packet_descriptor *iso)
390 {
391         unsigned char *usb_buf = urb->transfer_buffer + iso->offset;
392         unsigned char check_byte;
393         struct snd_pcm_substream *sub;
394         int stream, i;
395
396         for (i = 0; i < iso->actual_length;) {
397                 if (i % (dev->n_streams * BYTES_PER_SAMPLE_USB) == 0) {
398                         for (stream = 0;
399                              stream < dev->n_streams;
400                              stream++, i++) {
401                                 if (dev->first_packet)
402                                         continue;
403
404                                 check_byte = MAKE_CHECKBYTE(dev, stream, i);
405
406                                 if ((usb_buf[i] & 0x3f) != check_byte)
407                                         dev->input_panic = 1;
408
409                                 if (usb_buf[i] & 0x80)
410                                         dev->output_panic = 1;
411                         }
412                 }
413                 dev->first_packet = 0;
414
415                 for (stream = 0; stream < dev->n_streams; stream++, i++) {
416                         sub = dev->sub_capture[stream];
417                         if (dev->input_panic)
418                                 usb_buf[i] = 0;
419
420                         if (sub) {
421                                 struct snd_pcm_runtime *rt = sub->runtime;
422                                 char *audio_buf = rt->dma_area;
423                                 int sz = frames_to_bytes(rt, rt->buffer_size);
424                                 audio_buf[dev->audio_in_buf_pos[stream]++] =
425                                         usb_buf[i];
426                                 dev->period_in_count[stream]++;
427                                 if (dev->audio_in_buf_pos[stream] == sz)
428                                         dev->audio_in_buf_pos[stream] = 0;
429                         }
430                 }
431         }
432 }
433
434 static void read_in_urb_mode3(struct snd_usb_caiaqdev *dev,
435                               const struct urb *urb,
436                               const struct usb_iso_packet_descriptor *iso)
437 {
438         unsigned char *usb_buf = urb->transfer_buffer + iso->offset;
439         int stream, i;
440
441         /* paranoia check */
442         if (iso->actual_length % (BYTES_PER_SAMPLE_USB * CHANNELS_PER_STREAM))
443                 return;
444
445         for (i = 0; i < iso->actual_length;) {
446                 for (stream = 0; stream < dev->n_streams; stream++) {
447                         struct snd_pcm_substream *sub = dev->sub_capture[stream];
448                         char *audio_buf = NULL;
449                         int c, n, sz = 0;
450
451                         if (sub && !dev->input_panic) {
452                                 struct snd_pcm_runtime *rt = sub->runtime;
453                                 audio_buf = rt->dma_area;
454                                 sz = frames_to_bytes(rt, rt->buffer_size);
455                         }
456
457                         for (c = 0; c < CHANNELS_PER_STREAM; c++) {
458                                 /* 3 audio data bytes, followed by 1 check byte */
459                                 if (audio_buf) {
460                                         for (n = 0; n < BYTES_PER_SAMPLE; n++) {
461                                                 audio_buf[dev->audio_in_buf_pos[stream]++] = usb_buf[i+n];
462
463                                                 if (dev->audio_in_buf_pos[stream] == sz)
464                                                         dev->audio_in_buf_pos[stream] = 0;
465                                         }
466
467                                         dev->period_in_count[stream] += BYTES_PER_SAMPLE;
468                                 }
469
470                                 i += BYTES_PER_SAMPLE;
471
472                                 if (usb_buf[i] != ((stream << 1) | c) &&
473                                     !dev->first_packet) {
474                                         if (!dev->input_panic)
475                                                 printk(" EXPECTED: %02x got %02x, c %d, stream %d, i %d\n",
476                                                         ((stream << 1) | c), usb_buf[i], c, stream, i);
477                                         dev->input_panic = 1;
478                                 }
479
480                                 i++;
481                         }
482                 }
483         }
484
485         if (dev->first_packet > 0)
486                 dev->first_packet--;
487 }
488
489 static void read_in_urb(struct snd_usb_caiaqdev *dev,
490                         const struct urb *urb,
491                         const struct usb_iso_packet_descriptor *iso)
492 {
493         if (!dev->streaming)
494                 return;
495
496         if (iso->actual_length < dev->bpp)
497                 return;
498
499         switch (dev->spec.data_alignment) {
500         case 0:
501                 read_in_urb_mode0(dev, urb, iso);
502                 break;
503         case 2:
504                 read_in_urb_mode2(dev, urb, iso);
505                 break;
506         case 3:
507                 read_in_urb_mode3(dev, urb, iso);
508                 break;
509         }
510
511         if ((dev->input_panic || dev->output_panic) && !dev->warned) {
512                 debug("streaming error detected %s %s\n",
513                                 dev->input_panic ? "(input)" : "",
514                                 dev->output_panic ? "(output)" : "");
515                 dev->warned = 1;
516         }
517 }
518
519 static void fill_out_urb_mode_0(struct snd_usb_caiaqdev *dev,
520                                 struct urb *urb,
521                                 const struct usb_iso_packet_descriptor *iso)
522 {
523         unsigned char *usb_buf = urb->transfer_buffer + iso->offset;
524         struct snd_pcm_substream *sub;
525         int stream, i;
526
527         for (i = 0; i < iso->length;) {
528                 for (stream = 0; stream < dev->n_streams; stream++, i++) {
529                         sub = dev->sub_playback[stream];
530                         if (sub) {
531                                 struct snd_pcm_runtime *rt = sub->runtime;
532                                 char *audio_buf = rt->dma_area;
533                                 int sz = frames_to_bytes(rt, rt->buffer_size);
534                                 usb_buf[i] =
535                                         audio_buf[dev->audio_out_buf_pos[stream]];
536                                 dev->period_out_count[stream]++;
537                                 dev->audio_out_buf_pos[stream]++;
538                                 if (dev->audio_out_buf_pos[stream] == sz)
539                                         dev->audio_out_buf_pos[stream] = 0;
540                         } else
541                                 usb_buf[i] = 0;
542                 }
543
544                 /* fill in the check bytes */
545                 if (dev->spec.data_alignment == 2 &&
546                     i % (dev->n_streams * BYTES_PER_SAMPLE_USB) ==
547                         (dev->n_streams * CHANNELS_PER_STREAM))
548                         for (stream = 0; stream < dev->n_streams; stream++, i++)
549                                 usb_buf[i] = MAKE_CHECKBYTE(dev, stream, i);
550         }
551 }
552
553 static void fill_out_urb_mode_3(struct snd_usb_caiaqdev *dev,
554                                 struct urb *urb,
555                                 const struct usb_iso_packet_descriptor *iso)
556 {
557         unsigned char *usb_buf = urb->transfer_buffer + iso->offset;
558         int stream, i;
559
560         for (i = 0; i < iso->length;) {
561                 for (stream = 0; stream < dev->n_streams; stream++) {
562                         struct snd_pcm_substream *sub = dev->sub_playback[stream];
563                         char *audio_buf = NULL;
564                         int c, n, sz = 0;
565
566                         if (sub) {
567                                 struct snd_pcm_runtime *rt = sub->runtime;
568                                 audio_buf = rt->dma_area;
569                                 sz = frames_to_bytes(rt, rt->buffer_size);
570                         }
571
572                         for (c = 0; c < CHANNELS_PER_STREAM; c++) {
573                                 for (n = 0; n < BYTES_PER_SAMPLE; n++) {
574                                         if (audio_buf) {
575                                                 usb_buf[i+n] = audio_buf[dev->audio_out_buf_pos[stream]++];
576
577                                                 if (dev->audio_out_buf_pos[stream] == sz)
578                                                         dev->audio_out_buf_pos[stream] = 0;
579                                         } else {
580                                                 usb_buf[i+n] = 0;
581                                         }
582                                 }
583
584                                 if (audio_buf)
585                                         dev->period_out_count[stream] += BYTES_PER_SAMPLE;
586
587                                 i += BYTES_PER_SAMPLE;
588
589                                 /* fill in the check byte pattern */
590                                 usb_buf[i++] = (stream << 1) | c;
591                         }
592                 }
593         }
594 }
595
596 static inline void fill_out_urb(struct snd_usb_caiaqdev *dev,
597                                 struct urb *urb,
598                                 const struct usb_iso_packet_descriptor *iso)
599 {
600         switch (dev->spec.data_alignment) {
601         case 0:
602         case 2:
603                 fill_out_urb_mode_0(dev, urb, iso);
604                 break;
605         case 3:
606                 fill_out_urb_mode_3(dev, urb, iso);
607                 break;
608         }
609 }
610
611 static void read_completed(struct urb *urb)
612 {
613         struct snd_usb_caiaq_cb_info *info = urb->context;
614         struct snd_usb_caiaqdev *dev;
615         struct urb *out;
616         int frame, len, send_it = 0, outframe = 0;
617
618         if (urb->status || !info)
619                 return;
620
621         dev = info->dev;
622
623         if (!dev->streaming)
624                 return;
625
626         out = dev->data_urbs_out[info->index];
627
628         /* read the recently received packet and send back one which has
629          * the same layout */
630         for (frame = 0; frame < FRAMES_PER_URB; frame++) {
631                 if (urb->iso_frame_desc[frame].status)
632                         continue;
633
634                 len = urb->iso_frame_desc[outframe].actual_length;
635                 out->iso_frame_desc[outframe].length = len;
636                 out->iso_frame_desc[outframe].actual_length = 0;
637                 out->iso_frame_desc[outframe].offset = BYTES_PER_FRAME * frame;
638
639                 if (len > 0) {
640                         spin_lock(&dev->spinlock);
641                         fill_out_urb(dev, out, &out->iso_frame_desc[outframe]);
642                         read_in_urb(dev, urb, &urb->iso_frame_desc[frame]);
643                         spin_unlock(&dev->spinlock);
644                         check_for_elapsed_periods(dev, dev->sub_playback);
645                         check_for_elapsed_periods(dev, dev->sub_capture);
646                         send_it = 1;
647                 }
648
649                 outframe++;
650         }
651
652         if (send_it) {
653                 out->number_of_packets = FRAMES_PER_URB;
654                 out->transfer_flags = URB_ISO_ASAP;
655                 usb_submit_urb(out, GFP_ATOMIC);
656         }
657
658         /* re-submit inbound urb */
659         for (frame = 0; frame < FRAMES_PER_URB; frame++) {
660                 urb->iso_frame_desc[frame].offset = BYTES_PER_FRAME * frame;
661                 urb->iso_frame_desc[frame].length = BYTES_PER_FRAME;
662                 urb->iso_frame_desc[frame].actual_length = 0;
663         }
664
665         urb->number_of_packets = FRAMES_PER_URB;
666         urb->transfer_flags = URB_ISO_ASAP;
667         usb_submit_urb(urb, GFP_ATOMIC);
668 }
669
670 static void write_completed(struct urb *urb)
671 {
672         struct snd_usb_caiaq_cb_info *info = urb->context;
673         struct snd_usb_caiaqdev *dev = info->dev;
674
675         if (!dev->output_running) {
676                 dev->output_running = 1;
677                 wake_up(&dev->prepare_wait_queue);
678         }
679 }
680
681 static struct urb **alloc_urbs(struct snd_usb_caiaqdev *dev, int dir, int *ret)
682 {
683         int i, frame;
684         struct urb **urbs;
685         struct usb_device *usb_dev = dev->chip.dev;
686         unsigned int pipe;
687
688         pipe = (dir == SNDRV_PCM_STREAM_PLAYBACK) ?
689                 usb_sndisocpipe(usb_dev, ENDPOINT_PLAYBACK) :
690                 usb_rcvisocpipe(usb_dev, ENDPOINT_CAPTURE);
691
692         urbs = kmalloc(N_URBS * sizeof(*urbs), GFP_KERNEL);
693         if (!urbs) {
694                 log("unable to kmalloc() urbs, OOM!?\n");
695                 *ret = -ENOMEM;
696                 return NULL;
697         }
698
699         for (i = 0; i < N_URBS; i++) {
700                 urbs[i] = usb_alloc_urb(FRAMES_PER_URB, GFP_KERNEL);
701                 if (!urbs[i]) {
702                         log("unable to usb_alloc_urb(), OOM!?\n");
703                         *ret = -ENOMEM;
704                         return urbs;
705                 }
706
707                 urbs[i]->transfer_buffer =
708                         kmalloc(FRAMES_PER_URB * BYTES_PER_FRAME, GFP_KERNEL);
709                 if (!urbs[i]->transfer_buffer) {
710                         log("unable to kmalloc() transfer buffer, OOM!?\n");
711                         *ret = -ENOMEM;
712                         return urbs;
713                 }
714
715                 for (frame = 0; frame < FRAMES_PER_URB; frame++) {
716                         struct usb_iso_packet_descriptor *iso =
717                                 &urbs[i]->iso_frame_desc[frame];
718
719                         iso->offset = BYTES_PER_FRAME * frame;
720                         iso->length = BYTES_PER_FRAME;
721                 }
722
723                 urbs[i]->dev = usb_dev;
724                 urbs[i]->pipe = pipe;
725                 urbs[i]->transfer_buffer_length = FRAMES_PER_URB
726                                                 * BYTES_PER_FRAME;
727                 urbs[i]->context = &dev->data_cb_info[i];
728                 urbs[i]->interval = 1;
729                 urbs[i]->transfer_flags = URB_ISO_ASAP;
730                 urbs[i]->number_of_packets = FRAMES_PER_URB;
731                 urbs[i]->complete = (dir == SNDRV_PCM_STREAM_CAPTURE) ?
732                                         read_completed : write_completed;
733         }
734
735         *ret = 0;
736         return urbs;
737 }
738
739 static void free_urbs(struct urb **urbs)
740 {
741         int i;
742
743         if (!urbs)
744                 return;
745
746         for (i = 0; i < N_URBS; i++) {
747                 if (!urbs[i])
748                         continue;
749
750                 usb_kill_urb(urbs[i]);
751                 kfree(urbs[i]->transfer_buffer);
752                 usb_free_urb(urbs[i]);
753         }
754
755         kfree(urbs);
756 }
757
758 int snd_usb_caiaq_audio_init(struct snd_usb_caiaqdev *dev)
759 {
760         int i, ret;
761
762         dev->n_audio_in  = max(dev->spec.num_analog_audio_in,
763                                dev->spec.num_digital_audio_in) /
764                                 CHANNELS_PER_STREAM;
765         dev->n_audio_out = max(dev->spec.num_analog_audio_out,
766                                dev->spec.num_digital_audio_out) /
767                                 CHANNELS_PER_STREAM;
768         dev->n_streams = max(dev->n_audio_in, dev->n_audio_out);
769
770         debug("dev->n_audio_in = %d\n", dev->n_audio_in);
771         debug("dev->n_audio_out = %d\n", dev->n_audio_out);
772         debug("dev->n_streams = %d\n", dev->n_streams);
773
774         if (dev->n_streams > MAX_STREAMS) {
775                 log("unable to initialize device, too many streams.\n");
776                 return -EINVAL;
777         }
778
779         ret = snd_pcm_new(dev->chip.card, dev->product_name, 0,
780                         dev->n_audio_out, dev->n_audio_in, &dev->pcm);
781
782         if (ret < 0) {
783                 log("snd_pcm_new() returned %d\n", ret);
784                 return ret;
785         }
786
787         dev->pcm->private_data = dev;
788         strlcpy(dev->pcm->name, dev->product_name, sizeof(dev->pcm->name));
789
790         memset(dev->sub_playback, 0, sizeof(dev->sub_playback));
791         memset(dev->sub_capture, 0, sizeof(dev->sub_capture));
792
793         memcpy(&dev->pcm_info, &snd_usb_caiaq_pcm_hardware,
794                         sizeof(snd_usb_caiaq_pcm_hardware));
795
796         /* setup samplerates */
797         dev->samplerates = dev->pcm_info.rates;
798         switch (dev->chip.usb_id) {
799         case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_AK1):
800         case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_RIGKONTROL3):
801         case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_SESSIONIO):
802         case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_GUITARRIGMOBILE):
803                 dev->samplerates |= SNDRV_PCM_RATE_192000;
804                 /* fall thru */
805         case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_AUDIO2DJ):
806         case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_AUDIO4DJ):
807         case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_AUDIO8DJ):
808         case USB_ID(USB_VID_NATIVEINSTRUMENTS, USB_PID_TRAKTORAUDIO2):
809                 dev->samplerates |= SNDRV_PCM_RATE_88200;
810                 break;
811         }
812
813         snd_pcm_set_ops(dev->pcm, SNDRV_PCM_STREAM_PLAYBACK,
814                                 &snd_usb_caiaq_ops);
815         snd_pcm_set_ops(dev->pcm, SNDRV_PCM_STREAM_CAPTURE,
816                                 &snd_usb_caiaq_ops);
817
818         snd_pcm_lib_preallocate_pages_for_all(dev->pcm,
819                                         SNDRV_DMA_TYPE_CONTINUOUS,
820                                         snd_dma_continuous_data(GFP_KERNEL),
821                                         MAX_BUFFER_SIZE, MAX_BUFFER_SIZE);
822
823         dev->data_cb_info =
824                 kmalloc(sizeof(struct snd_usb_caiaq_cb_info) * N_URBS,
825                                         GFP_KERNEL);
826
827         if (!dev->data_cb_info)
828                 return -ENOMEM;
829
830         for (i = 0; i < N_URBS; i++) {
831                 dev->data_cb_info[i].dev = dev;
832                 dev->data_cb_info[i].index = i;
833         }
834
835         dev->data_urbs_in = alloc_urbs(dev, SNDRV_PCM_STREAM_CAPTURE, &ret);
836         if (ret < 0) {
837                 kfree(dev->data_cb_info);
838                 free_urbs(dev->data_urbs_in);
839                 return ret;
840         }
841
842         dev->data_urbs_out = alloc_urbs(dev, SNDRV_PCM_STREAM_PLAYBACK, &ret);
843         if (ret < 0) {
844                 kfree(dev->data_cb_info);
845                 free_urbs(dev->data_urbs_in);
846                 free_urbs(dev->data_urbs_out);
847                 return ret;
848         }
849
850         return 0;
851 }
852
853 void snd_usb_caiaq_audio_free(struct snd_usb_caiaqdev *dev)
854 {
855         debug("%s(%p)\n", __func__, dev);
856         stream_stop(dev);
857         free_urbs(dev->data_urbs_in);
858         free_urbs(dev->data_urbs_out);
859         kfree(dev->data_cb_info);
860 }
861