Linux Audio

Check our new training course

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