Add linux-next specific files for 20110831
[linux-2.6/next.git] / drivers / media / video / hdpvr / hdpvr-video.c
blob087f7c08cb851ee50485dd2a50e80f30d9a82674
1 /*
2 * Hauppauge HD PVR USB driver - video 4 linux 2 interface
4 * Copyright (C) 2008 Janne Grunau (j@jannau.net)
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License as
8 * published by the Free Software Foundation, version 2.
12 #include <linux/kernel.h>
13 #include <linux/errno.h>
14 #include <linux/init.h>
15 #include <linux/slab.h>
16 #include <linux/module.h>
17 #include <linux/uaccess.h>
18 #include <linux/usb.h>
19 #include <linux/mutex.h>
20 #include <linux/workqueue.h>
22 #include <linux/videodev2.h>
23 #include <media/v4l2-dev.h>
24 #include <media/v4l2-common.h>
25 #include <media/v4l2-ioctl.h>
26 #include "hdpvr.h"
28 #define BULK_URB_TIMEOUT 90 /* 0.09 seconds */
30 #define print_buffer_status() { \
31 v4l2_dbg(MSG_BUFFER, hdpvr_debug, &dev->v4l2_dev, \
32 "%s:%d buffer stat: %d free, %d proc\n", \
33 __func__, __LINE__, \
34 list_size(&dev->free_buff_list), \
35 list_size(&dev->rec_buff_list)); }
37 struct hdpvr_fh {
38 struct hdpvr_device *dev;
41 static uint list_size(struct list_head *list)
43 struct list_head *tmp;
44 uint count = 0;
46 list_for_each(tmp, list) {
47 count++;
50 return count;
53 /*=========================================================================*/
54 /* urb callback */
55 static void hdpvr_read_bulk_callback(struct urb *urb)
57 struct hdpvr_buffer *buf = (struct hdpvr_buffer *)urb->context;
58 struct hdpvr_device *dev = buf->dev;
60 /* marking buffer as received and wake waiting */
61 buf->status = BUFSTAT_READY;
62 wake_up_interruptible(&dev->wait_data);
65 /*=========================================================================*/
66 /* bufffer bits */
68 /* function expects dev->io_mutex to be hold by caller */
69 int hdpvr_cancel_queue(struct hdpvr_device *dev)
71 struct hdpvr_buffer *buf;
73 list_for_each_entry(buf, &dev->rec_buff_list, buff_list) {
74 usb_kill_urb(buf->urb);
75 buf->status = BUFSTAT_AVAILABLE;
78 list_splice_init(&dev->rec_buff_list, dev->free_buff_list.prev);
80 return 0;
83 static int hdpvr_free_queue(struct list_head *q)
85 struct list_head *tmp;
86 struct list_head *p;
87 struct hdpvr_buffer *buf;
88 struct urb *urb;
90 for (p = q->next; p != q;) {
91 buf = list_entry(p, struct hdpvr_buffer, buff_list);
93 urb = buf->urb;
94 usb_free_coherent(urb->dev, urb->transfer_buffer_length,
95 urb->transfer_buffer, urb->transfer_dma);
96 usb_free_urb(urb);
97 tmp = p->next;
98 list_del(p);
99 kfree(buf);
100 p = tmp;
103 return 0;
106 /* function expects dev->io_mutex to be hold by caller */
107 int hdpvr_free_buffers(struct hdpvr_device *dev)
109 hdpvr_cancel_queue(dev);
111 hdpvr_free_queue(&dev->free_buff_list);
112 hdpvr_free_queue(&dev->rec_buff_list);
114 return 0;
117 /* function expects dev->io_mutex to be hold by caller */
118 int hdpvr_alloc_buffers(struct hdpvr_device *dev, uint count)
120 uint i;
121 int retval = -ENOMEM;
122 u8 *mem;
123 struct hdpvr_buffer *buf;
124 struct urb *urb;
126 v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
127 "allocating %u buffers\n", count);
129 for (i = 0; i < count; i++) {
131 buf = kzalloc(sizeof(struct hdpvr_buffer), GFP_KERNEL);
132 if (!buf) {
133 v4l2_err(&dev->v4l2_dev, "cannot allocate buffer\n");
134 goto exit;
136 buf->dev = dev;
138 urb = usb_alloc_urb(0, GFP_KERNEL);
139 if (!urb) {
140 v4l2_err(&dev->v4l2_dev, "cannot allocate urb\n");
141 goto exit_urb;
143 buf->urb = urb;
145 mem = usb_alloc_coherent(dev->udev, dev->bulk_in_size, GFP_KERNEL,
146 &urb->transfer_dma);
147 if (!mem) {
148 v4l2_err(&dev->v4l2_dev,
149 "cannot allocate usb transfer buffer\n");
150 goto exit_urb_buffer;
153 usb_fill_bulk_urb(buf->urb, dev->udev,
154 usb_rcvbulkpipe(dev->udev,
155 dev->bulk_in_endpointAddr),
156 mem, dev->bulk_in_size,
157 hdpvr_read_bulk_callback, buf);
159 buf->urb->transfer_flags |= URB_NO_TRANSFER_DMA_MAP;
160 buf->status = BUFSTAT_AVAILABLE;
161 list_add_tail(&buf->buff_list, &dev->free_buff_list);
163 return 0;
164 exit_urb_buffer:
165 usb_free_urb(urb);
166 exit_urb:
167 kfree(buf);
168 exit:
169 hdpvr_free_buffers(dev);
170 return retval;
173 static int hdpvr_submit_buffers(struct hdpvr_device *dev)
175 struct hdpvr_buffer *buf;
176 struct urb *urb;
177 int ret = 0, err_count = 0;
179 mutex_lock(&dev->io_mutex);
181 while (dev->status == STATUS_STREAMING &&
182 !list_empty(&dev->free_buff_list)) {
184 buf = list_entry(dev->free_buff_list.next, struct hdpvr_buffer,
185 buff_list);
186 if (buf->status != BUFSTAT_AVAILABLE) {
187 v4l2_err(&dev->v4l2_dev,
188 "buffer not marked as available\n");
189 ret = -EFAULT;
190 goto err;
193 urb = buf->urb;
194 urb->status = 0;
195 urb->actual_length = 0;
196 ret = usb_submit_urb(urb, GFP_KERNEL);
197 if (ret) {
198 v4l2_err(&dev->v4l2_dev,
199 "usb_submit_urb in %s returned %d\n",
200 __func__, ret);
201 if (++err_count > 2)
202 break;
203 continue;
205 buf->status = BUFSTAT_INPROGRESS;
206 list_move_tail(&buf->buff_list, &dev->rec_buff_list);
208 err:
209 print_buffer_status();
210 mutex_unlock(&dev->io_mutex);
211 return ret;
214 static struct hdpvr_buffer *hdpvr_get_next_buffer(struct hdpvr_device *dev)
216 struct hdpvr_buffer *buf;
218 mutex_lock(&dev->io_mutex);
220 if (list_empty(&dev->rec_buff_list)) {
221 mutex_unlock(&dev->io_mutex);
222 return NULL;
225 buf = list_entry(dev->rec_buff_list.next, struct hdpvr_buffer,
226 buff_list);
227 mutex_unlock(&dev->io_mutex);
229 return buf;
232 static void hdpvr_transmit_buffers(struct work_struct *work)
234 struct hdpvr_device *dev = container_of(work, struct hdpvr_device,
235 worker);
237 while (dev->status == STATUS_STREAMING) {
239 if (hdpvr_submit_buffers(dev)) {
240 v4l2_err(&dev->v4l2_dev, "couldn't submit buffers\n");
241 goto error;
243 if (wait_event_interruptible(dev->wait_buffer,
244 !list_empty(&dev->free_buff_list) ||
245 dev->status != STATUS_STREAMING))
246 goto error;
249 v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
250 "transmit worker exited\n");
251 return;
252 error:
253 v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
254 "transmit buffers errored\n");
255 dev->status = STATUS_ERROR;
258 /* function expects dev->io_mutex to be hold by caller */
259 static int hdpvr_start_streaming(struct hdpvr_device *dev)
261 int ret;
262 struct hdpvr_video_info *vidinf;
264 if (dev->status == STATUS_STREAMING)
265 return 0;
266 else if (dev->status != STATUS_IDLE)
267 return -EAGAIN;
269 vidinf = get_video_info(dev);
271 if (vidinf) {
272 v4l2_dbg(MSG_BUFFER, hdpvr_debug, &dev->v4l2_dev,
273 "video signal: %dx%d@%dhz\n", vidinf->width,
274 vidinf->height, vidinf->fps);
275 kfree(vidinf);
277 /* start streaming 2 request */
278 ret = usb_control_msg(dev->udev,
279 usb_sndctrlpipe(dev->udev, 0),
280 0xb8, 0x38, 0x1, 0, NULL, 0, 8000);
281 v4l2_dbg(MSG_BUFFER, hdpvr_debug, &dev->v4l2_dev,
282 "encoder start control request returned %d\n", ret);
284 hdpvr_config_call(dev, CTRL_START_STREAMING_VALUE, 0x00);
286 INIT_WORK(&dev->worker, hdpvr_transmit_buffers);
287 queue_work(dev->workqueue, &dev->worker);
289 v4l2_dbg(MSG_BUFFER, hdpvr_debug, &dev->v4l2_dev,
290 "streaming started\n");
291 dev->status = STATUS_STREAMING;
293 return 0;
295 msleep(250);
296 v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
297 "no video signal at input %d\n", dev->options.video_input);
298 return -EAGAIN;
302 /* function expects dev->io_mutex to be hold by caller */
303 static int hdpvr_stop_streaming(struct hdpvr_device *dev)
305 int actual_length;
306 uint c = 0;
307 u8 *buf;
309 if (dev->status == STATUS_IDLE)
310 return 0;
311 else if (dev->status != STATUS_STREAMING)
312 return -EAGAIN;
314 buf = kmalloc(dev->bulk_in_size, GFP_KERNEL);
315 if (!buf)
316 v4l2_err(&dev->v4l2_dev, "failed to allocate temporary buffer "
317 "for emptying the internal device buffer. "
318 "Next capture start will be slow\n");
320 dev->status = STATUS_SHUTTING_DOWN;
321 hdpvr_config_call(dev, CTRL_STOP_STREAMING_VALUE, 0x00);
322 mutex_unlock(&dev->io_mutex);
324 wake_up_interruptible(&dev->wait_buffer);
325 msleep(50);
327 flush_workqueue(dev->workqueue);
329 mutex_lock(&dev->io_mutex);
330 /* kill the still outstanding urbs */
331 hdpvr_cancel_queue(dev);
333 /* emptying the device buffer beforeshutting it down */
334 while (buf && ++c < 500 &&
335 !usb_bulk_msg(dev->udev,
336 usb_rcvbulkpipe(dev->udev,
337 dev->bulk_in_endpointAddr),
338 buf, dev->bulk_in_size, &actual_length,
339 BULK_URB_TIMEOUT)) {
340 v4l2_dbg(MSG_BUFFER, hdpvr_debug, &dev->v4l2_dev,
341 "%2d: got %d bytes\n", c, actual_length);
343 kfree(buf);
344 v4l2_dbg(MSG_BUFFER, hdpvr_debug, &dev->v4l2_dev,
345 "used %d urbs to empty device buffers\n", c-1);
346 msleep(10);
348 dev->status = STATUS_IDLE;
350 return 0;
354 /*=======================================================================*/
356 * video 4 linux 2 file operations
359 static int hdpvr_open(struct file *file)
361 struct hdpvr_device *dev;
362 struct hdpvr_fh *fh;
363 int retval = -ENOMEM;
365 dev = (struct hdpvr_device *)video_get_drvdata(video_devdata(file));
366 if (!dev) {
367 pr_err("open failing with with ENODEV\n");
368 retval = -ENODEV;
369 goto err;
372 fh = kzalloc(sizeof(struct hdpvr_fh), GFP_KERNEL);
373 if (!fh) {
374 v4l2_err(&dev->v4l2_dev, "Out of memory\n");
375 goto err;
377 /* lock the device to allow correctly handling errors
378 * in resumption */
379 mutex_lock(&dev->io_mutex);
380 dev->open_count++;
381 mutex_unlock(&dev->io_mutex);
383 fh->dev = dev;
385 /* save our object in the file's private structure */
386 file->private_data = fh;
388 retval = 0;
389 err:
390 return retval;
393 static int hdpvr_release(struct file *file)
395 struct hdpvr_fh *fh = file->private_data;
396 struct hdpvr_device *dev = fh->dev;
398 if (!dev)
399 return -ENODEV;
401 mutex_lock(&dev->io_mutex);
402 if (!(--dev->open_count) && dev->status == STATUS_STREAMING)
403 hdpvr_stop_streaming(dev);
405 mutex_unlock(&dev->io_mutex);
407 return 0;
411 * hdpvr_v4l2_read()
412 * will allocate buffers when called for the first time
414 static ssize_t hdpvr_read(struct file *file, char __user *buffer, size_t count,
415 loff_t *pos)
417 struct hdpvr_fh *fh = file->private_data;
418 struct hdpvr_device *dev = fh->dev;
419 struct hdpvr_buffer *buf = NULL;
420 struct urb *urb;
421 unsigned int ret = 0;
422 int rem, cnt;
424 if (*pos)
425 return -ESPIPE;
427 if (!dev)
428 return -ENODEV;
430 mutex_lock(&dev->io_mutex);
431 if (dev->status == STATUS_IDLE) {
432 if (hdpvr_start_streaming(dev)) {
433 v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
434 "start_streaming failed\n");
435 ret = -EIO;
436 msleep(200);
437 dev->status = STATUS_IDLE;
438 mutex_unlock(&dev->io_mutex);
439 goto err;
441 print_buffer_status();
443 mutex_unlock(&dev->io_mutex);
445 /* wait for the first buffer */
446 if (!(file->f_flags & O_NONBLOCK)) {
447 if (wait_event_interruptible(dev->wait_data,
448 hdpvr_get_next_buffer(dev)))
449 return -ERESTARTSYS;
452 buf = hdpvr_get_next_buffer(dev);
454 while (count > 0 && buf) {
456 if (buf->status != BUFSTAT_READY &&
457 dev->status != STATUS_DISCONNECTED) {
458 /* return nonblocking */
459 if (file->f_flags & O_NONBLOCK) {
460 if (!ret)
461 ret = -EAGAIN;
462 goto err;
465 if (wait_event_interruptible(dev->wait_data,
466 buf->status == BUFSTAT_READY)) {
467 ret = -ERESTARTSYS;
468 goto err;
472 if (buf->status != BUFSTAT_READY)
473 break;
475 /* set remaining bytes to copy */
476 urb = buf->urb;
477 rem = urb->actual_length - buf->pos;
478 cnt = rem > count ? count : rem;
480 if (copy_to_user(buffer, urb->transfer_buffer + buf->pos,
481 cnt)) {
482 v4l2_err(&dev->v4l2_dev, "read: copy_to_user failed\n");
483 if (!ret)
484 ret = -EFAULT;
485 goto err;
488 buf->pos += cnt;
489 count -= cnt;
490 buffer += cnt;
491 ret += cnt;
493 /* finished, take next buffer */
494 if (buf->pos == urb->actual_length) {
495 mutex_lock(&dev->io_mutex);
496 buf->pos = 0;
497 buf->status = BUFSTAT_AVAILABLE;
499 list_move_tail(&buf->buff_list, &dev->free_buff_list);
501 print_buffer_status();
503 mutex_unlock(&dev->io_mutex);
505 wake_up_interruptible(&dev->wait_buffer);
507 buf = hdpvr_get_next_buffer(dev);
510 err:
511 if (!ret && !buf)
512 ret = -EAGAIN;
513 return ret;
516 static unsigned int hdpvr_poll(struct file *filp, poll_table *wait)
518 struct hdpvr_buffer *buf = NULL;
519 struct hdpvr_fh *fh = filp->private_data;
520 struct hdpvr_device *dev = fh->dev;
521 unsigned int mask = 0;
523 mutex_lock(&dev->io_mutex);
525 if (!video_is_registered(dev->video_dev)) {
526 mutex_unlock(&dev->io_mutex);
527 return -EIO;
530 if (dev->status == STATUS_IDLE) {
531 if (hdpvr_start_streaming(dev)) {
532 v4l2_dbg(MSG_BUFFER, hdpvr_debug, &dev->v4l2_dev,
533 "start_streaming failed\n");
534 dev->status = STATUS_IDLE;
537 print_buffer_status();
539 mutex_unlock(&dev->io_mutex);
541 buf = hdpvr_get_next_buffer(dev);
542 /* only wait if no data is available */
543 if (!buf || buf->status != BUFSTAT_READY) {
544 poll_wait(filp, &dev->wait_data, wait);
545 buf = hdpvr_get_next_buffer(dev);
547 if (buf && buf->status == BUFSTAT_READY)
548 mask |= POLLIN | POLLRDNORM;
550 return mask;
554 static const struct v4l2_file_operations hdpvr_fops = {
555 .owner = THIS_MODULE,
556 .open = hdpvr_open,
557 .release = hdpvr_release,
558 .read = hdpvr_read,
559 .poll = hdpvr_poll,
560 .unlocked_ioctl = video_ioctl2,
563 /*=======================================================================*/
565 * V4L2 ioctl handling
568 static int vidioc_querycap(struct file *file, void *priv,
569 struct v4l2_capability *cap)
571 struct hdpvr_device *dev = video_drvdata(file);
573 strcpy(cap->driver, "hdpvr");
574 strcpy(cap->card, "Hauppauge HD PVR");
575 usb_make_path(dev->udev, cap->bus_info, sizeof(cap->bus_info));
576 cap->capabilities = V4L2_CAP_VIDEO_CAPTURE |
577 V4L2_CAP_AUDIO |
578 V4L2_CAP_READWRITE;
579 return 0;
582 static int vidioc_s_std(struct file *file, void *private_data,
583 v4l2_std_id *std)
585 struct hdpvr_fh *fh = file->private_data;
586 struct hdpvr_device *dev = fh->dev;
587 u8 std_type = 1;
589 if (*std & (V4L2_STD_NTSC | V4L2_STD_PAL_60))
590 std_type = 0;
592 return hdpvr_config_call(dev, CTRL_VIDEO_STD_TYPE, std_type);
595 static const char *iname[] = {
596 [HDPVR_COMPONENT] = "Component",
597 [HDPVR_SVIDEO] = "S-Video",
598 [HDPVR_COMPOSITE] = "Composite",
601 static int vidioc_enum_input(struct file *file, void *priv,
602 struct v4l2_input *i)
604 struct hdpvr_fh *fh = file->private_data;
605 struct hdpvr_device *dev = fh->dev;
606 unsigned int n;
608 n = i->index;
609 if (n >= HDPVR_VIDEO_INPUTS)
610 return -EINVAL;
612 i->type = V4L2_INPUT_TYPE_CAMERA;
614 strncpy(i->name, iname[n], sizeof(i->name) - 1);
615 i->name[sizeof(i->name) - 1] = '\0';
617 i->audioset = 1<<HDPVR_RCA_FRONT | 1<<HDPVR_RCA_BACK | 1<<HDPVR_SPDIF;
619 i->std = dev->video_dev->tvnorms;
621 return 0;
624 static int vidioc_s_input(struct file *file, void *private_data,
625 unsigned int index)
627 struct hdpvr_fh *fh = file->private_data;
628 struct hdpvr_device *dev = fh->dev;
629 int retval;
631 if (index >= HDPVR_VIDEO_INPUTS)
632 return -EINVAL;
634 if (dev->status != STATUS_IDLE)
635 return -EAGAIN;
637 retval = hdpvr_config_call(dev, CTRL_VIDEO_INPUT_VALUE, index+1);
638 if (!retval)
639 dev->options.video_input = index;
641 return retval;
644 static int vidioc_g_input(struct file *file, void *private_data,
645 unsigned int *index)
647 struct hdpvr_fh *fh = file->private_data;
648 struct hdpvr_device *dev = fh->dev;
650 *index = dev->options.video_input;
651 return 0;
655 static const char *audio_iname[] = {
656 [HDPVR_RCA_FRONT] = "RCA front",
657 [HDPVR_RCA_BACK] = "RCA back",
658 [HDPVR_SPDIF] = "SPDIF",
661 static int vidioc_enumaudio(struct file *file, void *priv,
662 struct v4l2_audio *audio)
664 unsigned int n;
666 n = audio->index;
667 if (n >= HDPVR_AUDIO_INPUTS)
668 return -EINVAL;
670 audio->capability = V4L2_AUDCAP_STEREO;
672 strncpy(audio->name, audio_iname[n], sizeof(audio->name) - 1);
673 audio->name[sizeof(audio->name) - 1] = '\0';
675 return 0;
678 static int vidioc_s_audio(struct file *file, void *private_data,
679 struct v4l2_audio *audio)
681 struct hdpvr_fh *fh = file->private_data;
682 struct hdpvr_device *dev = fh->dev;
683 int retval;
685 if (audio->index >= HDPVR_AUDIO_INPUTS)
686 return -EINVAL;
688 if (dev->status != STATUS_IDLE)
689 return -EAGAIN;
691 retval = hdpvr_set_audio(dev, audio->index+1, dev->options.audio_codec);
692 if (!retval)
693 dev->options.audio_input = audio->index;
695 return retval;
698 static int vidioc_g_audio(struct file *file, void *private_data,
699 struct v4l2_audio *audio)
701 struct hdpvr_fh *fh = file->private_data;
702 struct hdpvr_device *dev = fh->dev;
704 audio->index = dev->options.audio_input;
705 audio->capability = V4L2_AUDCAP_STEREO;
706 strncpy(audio->name, audio_iname[audio->index], sizeof(audio->name));
707 audio->name[sizeof(audio->name) - 1] = '\0';
708 return 0;
711 static const s32 supported_v4l2_ctrls[] = {
712 V4L2_CID_BRIGHTNESS,
713 V4L2_CID_CONTRAST,
714 V4L2_CID_SATURATION,
715 V4L2_CID_HUE,
716 V4L2_CID_SHARPNESS,
717 V4L2_CID_MPEG_AUDIO_ENCODING,
718 V4L2_CID_MPEG_VIDEO_ENCODING,
719 V4L2_CID_MPEG_VIDEO_BITRATE_MODE,
720 V4L2_CID_MPEG_VIDEO_BITRATE,
721 V4L2_CID_MPEG_VIDEO_BITRATE_PEAK,
724 static int fill_queryctrl(struct hdpvr_options *opt, struct v4l2_queryctrl *qc,
725 int ac3)
727 int err;
729 switch (qc->id) {
730 case V4L2_CID_BRIGHTNESS:
731 return v4l2_ctrl_query_fill(qc, 0x0, 0xff, 1, 0x86);
732 case V4L2_CID_CONTRAST:
733 return v4l2_ctrl_query_fill(qc, 0x0, 0xff, 1, 0x80);
734 case V4L2_CID_SATURATION:
735 return v4l2_ctrl_query_fill(qc, 0x0, 0xff, 1, 0x80);
736 case V4L2_CID_HUE:
737 return v4l2_ctrl_query_fill(qc, 0x0, 0xff, 1, 0x80);
738 case V4L2_CID_SHARPNESS:
739 return v4l2_ctrl_query_fill(qc, 0x0, 0xff, 1, 0x80);
740 case V4L2_CID_MPEG_AUDIO_ENCODING:
741 return v4l2_ctrl_query_fill(
742 qc, V4L2_MPEG_AUDIO_ENCODING_AAC,
743 ac3 ? V4L2_MPEG_AUDIO_ENCODING_AC3
744 : V4L2_MPEG_AUDIO_ENCODING_AAC,
745 1, V4L2_MPEG_AUDIO_ENCODING_AAC);
746 case V4L2_CID_MPEG_VIDEO_ENCODING:
747 return v4l2_ctrl_query_fill(
748 qc, V4L2_MPEG_VIDEO_ENCODING_MPEG_4_AVC,
749 V4L2_MPEG_VIDEO_ENCODING_MPEG_4_AVC, 1,
750 V4L2_MPEG_VIDEO_ENCODING_MPEG_4_AVC);
752 /* case V4L2_CID_MPEG_VIDEO_? maybe keyframe interval: */
753 /* return v4l2_ctrl_query_fill(qc, 0, 128, 128, 0); */
754 case V4L2_CID_MPEG_VIDEO_BITRATE_MODE:
755 return v4l2_ctrl_query_fill(
756 qc, V4L2_MPEG_VIDEO_BITRATE_MODE_VBR,
757 V4L2_MPEG_VIDEO_BITRATE_MODE_CBR, 1,
758 V4L2_MPEG_VIDEO_BITRATE_MODE_CBR);
760 case V4L2_CID_MPEG_VIDEO_BITRATE:
761 return v4l2_ctrl_query_fill(qc, 1000000, 13500000, 100000,
762 6500000);
763 case V4L2_CID_MPEG_VIDEO_BITRATE_PEAK:
764 err = v4l2_ctrl_query_fill(qc, 1100000, 20200000, 100000,
765 9000000);
766 if (!err && opt->bitrate_mode == HDPVR_CONSTANT)
767 qc->flags |= V4L2_CTRL_FLAG_INACTIVE;
768 return err;
769 default:
770 return -EINVAL;
774 static int vidioc_queryctrl(struct file *file, void *private_data,
775 struct v4l2_queryctrl *qc)
777 struct hdpvr_fh *fh = file->private_data;
778 struct hdpvr_device *dev = fh->dev;
779 int i, next;
780 u32 id = qc->id;
782 memset(qc, 0, sizeof(*qc));
784 next = !!(id & V4L2_CTRL_FLAG_NEXT_CTRL);
785 qc->id = id & ~V4L2_CTRL_FLAG_NEXT_CTRL;
787 for (i = 0; i < ARRAY_SIZE(supported_v4l2_ctrls); i++) {
788 if (next) {
789 if (qc->id < supported_v4l2_ctrls[i])
790 qc->id = supported_v4l2_ctrls[i];
791 else
792 continue;
795 if (qc->id == supported_v4l2_ctrls[i])
796 return fill_queryctrl(&dev->options, qc,
797 dev->flags & HDPVR_FLAG_AC3_CAP);
799 if (qc->id < supported_v4l2_ctrls[i])
800 break;
803 return -EINVAL;
806 static int vidioc_g_ctrl(struct file *file, void *private_data,
807 struct v4l2_control *ctrl)
809 struct hdpvr_fh *fh = file->private_data;
810 struct hdpvr_device *dev = fh->dev;
812 switch (ctrl->id) {
813 case V4L2_CID_BRIGHTNESS:
814 ctrl->value = dev->options.brightness;
815 break;
816 case V4L2_CID_CONTRAST:
817 ctrl->value = dev->options.contrast;
818 break;
819 case V4L2_CID_SATURATION:
820 ctrl->value = dev->options.saturation;
821 break;
822 case V4L2_CID_HUE:
823 ctrl->value = dev->options.hue;
824 break;
825 case V4L2_CID_SHARPNESS:
826 ctrl->value = dev->options.sharpness;
827 break;
828 default:
829 return -EINVAL;
831 return 0;
834 static int vidioc_s_ctrl(struct file *file, void *private_data,
835 struct v4l2_control *ctrl)
837 struct hdpvr_fh *fh = file->private_data;
838 struct hdpvr_device *dev = fh->dev;
839 int retval;
841 switch (ctrl->id) {
842 case V4L2_CID_BRIGHTNESS:
843 retval = hdpvr_config_call(dev, CTRL_BRIGHTNESS, ctrl->value);
844 if (!retval)
845 dev->options.brightness = ctrl->value;
846 break;
847 case V4L2_CID_CONTRAST:
848 retval = hdpvr_config_call(dev, CTRL_CONTRAST, ctrl->value);
849 if (!retval)
850 dev->options.contrast = ctrl->value;
851 break;
852 case V4L2_CID_SATURATION:
853 retval = hdpvr_config_call(dev, CTRL_SATURATION, ctrl->value);
854 if (!retval)
855 dev->options.saturation = ctrl->value;
856 break;
857 case V4L2_CID_HUE:
858 retval = hdpvr_config_call(dev, CTRL_HUE, ctrl->value);
859 if (!retval)
860 dev->options.hue = ctrl->value;
861 break;
862 case V4L2_CID_SHARPNESS:
863 retval = hdpvr_config_call(dev, CTRL_SHARPNESS, ctrl->value);
864 if (!retval)
865 dev->options.sharpness = ctrl->value;
866 break;
867 default:
868 return -EINVAL;
871 return retval;
875 static int hdpvr_get_ctrl(struct hdpvr_options *opt,
876 struct v4l2_ext_control *ctrl)
878 switch (ctrl->id) {
879 case V4L2_CID_MPEG_AUDIO_ENCODING:
880 ctrl->value = opt->audio_codec;
881 break;
882 case V4L2_CID_MPEG_VIDEO_ENCODING:
883 ctrl->value = V4L2_MPEG_VIDEO_ENCODING_MPEG_4_AVC;
884 break;
885 /* case V4L2_CID_MPEG_VIDEO_B_FRAMES: */
886 /* ctrl->value = (opt->gop_mode & 0x2) ? 0 : 128; */
887 /* break; */
888 case V4L2_CID_MPEG_VIDEO_BITRATE_MODE:
889 ctrl->value = opt->bitrate_mode == HDPVR_CONSTANT
890 ? V4L2_MPEG_VIDEO_BITRATE_MODE_CBR
891 : V4L2_MPEG_VIDEO_BITRATE_MODE_VBR;
892 break;
893 case V4L2_CID_MPEG_VIDEO_BITRATE:
894 ctrl->value = opt->bitrate * 100000;
895 break;
896 case V4L2_CID_MPEG_VIDEO_BITRATE_PEAK:
897 ctrl->value = opt->peak_bitrate * 100000;
898 break;
899 case V4L2_CID_MPEG_STREAM_TYPE:
900 ctrl->value = V4L2_MPEG_STREAM_TYPE_MPEG2_TS;
901 break;
902 default:
903 return -EINVAL;
905 return 0;
908 static int vidioc_g_ext_ctrls(struct file *file, void *priv,
909 struct v4l2_ext_controls *ctrls)
911 struct hdpvr_fh *fh = file->private_data;
912 struct hdpvr_device *dev = fh->dev;
913 int i, err = 0;
915 if (ctrls->ctrl_class == V4L2_CTRL_CLASS_MPEG) {
916 for (i = 0; i < ctrls->count; i++) {
917 struct v4l2_ext_control *ctrl = ctrls->controls + i;
919 err = hdpvr_get_ctrl(&dev->options, ctrl);
920 if (err) {
921 ctrls->error_idx = i;
922 break;
925 return err;
929 return -EINVAL;
933 static int hdpvr_try_ctrl(struct v4l2_ext_control *ctrl, int ac3)
935 int ret = -EINVAL;
937 switch (ctrl->id) {
938 case V4L2_CID_MPEG_AUDIO_ENCODING:
939 if (ctrl->value == V4L2_MPEG_AUDIO_ENCODING_AAC ||
940 (ac3 && ctrl->value == V4L2_MPEG_AUDIO_ENCODING_AC3))
941 ret = 0;
942 break;
943 case V4L2_CID_MPEG_VIDEO_ENCODING:
944 if (ctrl->value == V4L2_MPEG_VIDEO_ENCODING_MPEG_4_AVC)
945 ret = 0;
946 break;
947 /* case V4L2_CID_MPEG_VIDEO_B_FRAMES: */
948 /* if (ctrl->value == 0 || ctrl->value == 128) */
949 /* ret = 0; */
950 /* break; */
951 case V4L2_CID_MPEG_VIDEO_BITRATE_MODE:
952 if (ctrl->value == V4L2_MPEG_VIDEO_BITRATE_MODE_CBR ||
953 ctrl->value == V4L2_MPEG_VIDEO_BITRATE_MODE_VBR)
954 ret = 0;
955 break;
956 case V4L2_CID_MPEG_VIDEO_BITRATE:
958 uint bitrate = ctrl->value / 100000;
959 if (bitrate >= 10 && bitrate <= 135)
960 ret = 0;
961 break;
963 case V4L2_CID_MPEG_VIDEO_BITRATE_PEAK:
965 uint peak_bitrate = ctrl->value / 100000;
966 if (peak_bitrate >= 10 && peak_bitrate <= 202)
967 ret = 0;
968 break;
970 case V4L2_CID_MPEG_STREAM_TYPE:
971 if (ctrl->value == V4L2_MPEG_STREAM_TYPE_MPEG2_TS)
972 ret = 0;
973 break;
974 default:
975 return -EINVAL;
977 return 0;
980 static int vidioc_try_ext_ctrls(struct file *file, void *priv,
981 struct v4l2_ext_controls *ctrls)
983 struct hdpvr_fh *fh = file->private_data;
984 struct hdpvr_device *dev = fh->dev;
985 int i, err = 0;
987 if (ctrls->ctrl_class == V4L2_CTRL_CLASS_MPEG) {
988 for (i = 0; i < ctrls->count; i++) {
989 struct v4l2_ext_control *ctrl = ctrls->controls + i;
991 err = hdpvr_try_ctrl(ctrl,
992 dev->flags & HDPVR_FLAG_AC3_CAP);
993 if (err) {
994 ctrls->error_idx = i;
995 break;
998 return err;
1001 return -EINVAL;
1005 static int hdpvr_set_ctrl(struct hdpvr_device *dev,
1006 struct v4l2_ext_control *ctrl)
1008 struct hdpvr_options *opt = &dev->options;
1009 int ret = 0;
1011 switch (ctrl->id) {
1012 case V4L2_CID_MPEG_AUDIO_ENCODING:
1013 if (dev->flags & HDPVR_FLAG_AC3_CAP) {
1014 opt->audio_codec = ctrl->value;
1015 ret = hdpvr_set_audio(dev, opt->audio_input,
1016 opt->audio_codec);
1018 break;
1019 case V4L2_CID_MPEG_VIDEO_ENCODING:
1020 break;
1021 /* case V4L2_CID_MPEG_VIDEO_B_FRAMES: */
1022 /* if (ctrl->value == 0 && !(opt->gop_mode & 0x2)) { */
1023 /* opt->gop_mode |= 0x2; */
1024 /* hdpvr_config_call(dev, CTRL_GOP_MODE_VALUE, */
1025 /* opt->gop_mode); */
1026 /* } */
1027 /* if (ctrl->value == 128 && opt->gop_mode & 0x2) { */
1028 /* opt->gop_mode &= ~0x2; */
1029 /* hdpvr_config_call(dev, CTRL_GOP_MODE_VALUE, */
1030 /* opt->gop_mode); */
1031 /* } */
1032 /* break; */
1033 case V4L2_CID_MPEG_VIDEO_BITRATE_MODE:
1034 if (ctrl->value == V4L2_MPEG_VIDEO_BITRATE_MODE_CBR &&
1035 opt->bitrate_mode != HDPVR_CONSTANT) {
1036 opt->bitrate_mode = HDPVR_CONSTANT;
1037 hdpvr_config_call(dev, CTRL_BITRATE_MODE_VALUE,
1038 opt->bitrate_mode);
1040 if (ctrl->value == V4L2_MPEG_VIDEO_BITRATE_MODE_VBR &&
1041 opt->bitrate_mode == HDPVR_CONSTANT) {
1042 opt->bitrate_mode = HDPVR_VARIABLE_AVERAGE;
1043 hdpvr_config_call(dev, CTRL_BITRATE_MODE_VALUE,
1044 opt->bitrate_mode);
1046 break;
1047 case V4L2_CID_MPEG_VIDEO_BITRATE: {
1048 uint bitrate = ctrl->value / 100000;
1050 opt->bitrate = bitrate;
1051 if (bitrate >= opt->peak_bitrate)
1052 opt->peak_bitrate = bitrate+1;
1054 hdpvr_set_bitrate(dev);
1055 break;
1057 case V4L2_CID_MPEG_VIDEO_BITRATE_PEAK: {
1058 uint peak_bitrate = ctrl->value / 100000;
1060 if (opt->bitrate_mode == HDPVR_CONSTANT)
1061 break;
1063 if (opt->bitrate < peak_bitrate) {
1064 opt->peak_bitrate = peak_bitrate;
1065 hdpvr_set_bitrate(dev);
1066 } else
1067 ret = -EINVAL;
1068 break;
1070 case V4L2_CID_MPEG_STREAM_TYPE:
1071 break;
1072 default:
1073 return -EINVAL;
1075 return ret;
1078 static int vidioc_s_ext_ctrls(struct file *file, void *priv,
1079 struct v4l2_ext_controls *ctrls)
1081 struct hdpvr_fh *fh = file->private_data;
1082 struct hdpvr_device *dev = fh->dev;
1083 int i, err = 0;
1085 if (ctrls->ctrl_class == V4L2_CTRL_CLASS_MPEG) {
1086 for (i = 0; i < ctrls->count; i++) {
1087 struct v4l2_ext_control *ctrl = ctrls->controls + i;
1089 err = hdpvr_try_ctrl(ctrl,
1090 dev->flags & HDPVR_FLAG_AC3_CAP);
1091 if (err) {
1092 ctrls->error_idx = i;
1093 break;
1095 err = hdpvr_set_ctrl(dev, ctrl);
1096 if (err) {
1097 ctrls->error_idx = i;
1098 break;
1101 return err;
1105 return -EINVAL;
1108 static int vidioc_enum_fmt_vid_cap(struct file *file, void *private_data,
1109 struct v4l2_fmtdesc *f)
1112 if (f->index != 0 || f->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
1113 return -EINVAL;
1115 f->flags = V4L2_FMT_FLAG_COMPRESSED;
1116 strncpy(f->description, "MPEG2-TS with AVC/AAC streams", 32);
1117 f->pixelformat = V4L2_PIX_FMT_MPEG;
1119 return 0;
1122 static int vidioc_g_fmt_vid_cap(struct file *file, void *private_data,
1123 struct v4l2_format *f)
1125 struct hdpvr_fh *fh = file->private_data;
1126 struct hdpvr_device *dev = fh->dev;
1127 struct hdpvr_video_info *vid_info;
1129 if (!dev)
1130 return -ENODEV;
1132 vid_info = get_video_info(dev);
1133 if (!vid_info)
1134 return -EFAULT;
1136 f->type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
1137 f->fmt.pix.pixelformat = V4L2_PIX_FMT_MPEG;
1138 f->fmt.pix.width = vid_info->width;
1139 f->fmt.pix.height = vid_info->height;
1140 f->fmt.pix.sizeimage = dev->bulk_in_size;
1141 f->fmt.pix.colorspace = 0;
1142 f->fmt.pix.bytesperline = 0;
1143 f->fmt.pix.field = V4L2_FIELD_ANY;
1145 kfree(vid_info);
1146 return 0;
1149 static int vidioc_encoder_cmd(struct file *filp, void *priv,
1150 struct v4l2_encoder_cmd *a)
1152 struct hdpvr_fh *fh = filp->private_data;
1153 struct hdpvr_device *dev = fh->dev;
1154 int res;
1156 mutex_lock(&dev->io_mutex);
1158 memset(&a->raw, 0, sizeof(a->raw));
1159 switch (a->cmd) {
1160 case V4L2_ENC_CMD_START:
1161 a->flags = 0;
1162 res = hdpvr_start_streaming(dev);
1163 break;
1164 case V4L2_ENC_CMD_STOP:
1165 res = hdpvr_stop_streaming(dev);
1166 break;
1167 default:
1168 v4l2_dbg(MSG_INFO, hdpvr_debug, &dev->v4l2_dev,
1169 "Unsupported encoder cmd %d\n", a->cmd);
1170 res = -EINVAL;
1172 mutex_unlock(&dev->io_mutex);
1173 return res;
1176 static int vidioc_try_encoder_cmd(struct file *filp, void *priv,
1177 struct v4l2_encoder_cmd *a)
1179 switch (a->cmd) {
1180 case V4L2_ENC_CMD_START:
1181 case V4L2_ENC_CMD_STOP:
1182 return 0;
1183 default:
1184 return -EINVAL;
1188 static const struct v4l2_ioctl_ops hdpvr_ioctl_ops = {
1189 .vidioc_querycap = vidioc_querycap,
1190 .vidioc_s_std = vidioc_s_std,
1191 .vidioc_enum_input = vidioc_enum_input,
1192 .vidioc_g_input = vidioc_g_input,
1193 .vidioc_s_input = vidioc_s_input,
1194 .vidioc_enumaudio = vidioc_enumaudio,
1195 .vidioc_g_audio = vidioc_g_audio,
1196 .vidioc_s_audio = vidioc_s_audio,
1197 .vidioc_queryctrl = vidioc_queryctrl,
1198 .vidioc_g_ctrl = vidioc_g_ctrl,
1199 .vidioc_s_ctrl = vidioc_s_ctrl,
1200 .vidioc_g_ext_ctrls = vidioc_g_ext_ctrls,
1201 .vidioc_s_ext_ctrls = vidioc_s_ext_ctrls,
1202 .vidioc_try_ext_ctrls = vidioc_try_ext_ctrls,
1203 .vidioc_enum_fmt_vid_cap = vidioc_enum_fmt_vid_cap,
1204 .vidioc_g_fmt_vid_cap = vidioc_g_fmt_vid_cap,
1205 .vidioc_encoder_cmd = vidioc_encoder_cmd,
1206 .vidioc_try_encoder_cmd = vidioc_try_encoder_cmd,
1209 static void hdpvr_device_release(struct video_device *vdev)
1211 struct hdpvr_device *dev = video_get_drvdata(vdev);
1213 hdpvr_delete(dev);
1214 mutex_lock(&dev->io_mutex);
1215 destroy_workqueue(dev->workqueue);
1216 mutex_unlock(&dev->io_mutex);
1218 v4l2_device_unregister(&dev->v4l2_dev);
1220 /* deregister I2C adapter */
1221 #if defined(CONFIG_I2C) || (CONFIG_I2C_MODULE)
1222 mutex_lock(&dev->i2c_mutex);
1223 i2c_del_adapter(&dev->i2c_adapter);
1224 mutex_unlock(&dev->i2c_mutex);
1225 #endif /* CONFIG_I2C */
1227 kfree(dev->usbc_buf);
1228 kfree(dev);
1231 static const struct video_device hdpvr_video_template = {
1232 /* .type = VFL_TYPE_GRABBER, */
1233 /* .type2 = VID_TYPE_CAPTURE | VID_TYPE_MPEG_ENCODER, */
1234 .fops = &hdpvr_fops,
1235 .release = hdpvr_device_release,
1236 .ioctl_ops = &hdpvr_ioctl_ops,
1237 .tvnorms =
1238 V4L2_STD_NTSC | V4L2_STD_SECAM | V4L2_STD_PAL_B |
1239 V4L2_STD_PAL_G | V4L2_STD_PAL_H | V4L2_STD_PAL_I |
1240 V4L2_STD_PAL_D | V4L2_STD_PAL_M | V4L2_STD_PAL_N |
1241 V4L2_STD_PAL_60,
1242 .current_norm = V4L2_STD_NTSC | V4L2_STD_PAL_M |
1243 V4L2_STD_PAL_60,
1246 int hdpvr_register_videodev(struct hdpvr_device *dev, struct device *parent,
1247 int devnum)
1249 /* setup and register video device */
1250 dev->video_dev = video_device_alloc();
1251 if (!dev->video_dev) {
1252 v4l2_err(&dev->v4l2_dev, "video_device_alloc() failed\n");
1253 goto error;
1256 *(dev->video_dev) = hdpvr_video_template;
1257 strcpy(dev->video_dev->name, "Hauppauge HD PVR");
1258 dev->video_dev->parent = parent;
1259 video_set_drvdata(dev->video_dev, dev);
1261 if (video_register_device(dev->video_dev, VFL_TYPE_GRABBER, devnum)) {
1262 v4l2_err(&dev->v4l2_dev, "video_device registration failed\n");
1263 goto error;
1266 return 0;
1267 error:
1268 return -ENOMEM;