drivers/media/usb/stk1160/stk1160-v4l.c

   1 /*
   2  * STK1160 driver
   3  *
   4  * Copyright (C) 2012 Ezequiel Garcia
   5  * <elezegarcia--a.t--gmail.com>
   6  *
   7  * Based on Easycap driver by R.M. Thomas
   8  *      Copyright (C) 2010 R.M. Thomas
   9  *      <rmthomas--a.t--sciolus.org>
  10  *
  11  * This program is free software; you can redistribute it and/or modify
  12  * it under the terms of the GNU General Public License as published by
  13  * the Free Software Foundation; either version 2 of the License, or
  14  * (at your option) any later version.
  15  *
  16  * This program is distributed in the hope that it will be useful,
  17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19  * GNU General Public License for more details.
  20  *
  21  */
  22
  23 #include <linux/module.h>
  24 #include <linux/usb.h>
  25 #include <linux/mm.h>
  26 #include <linux/slab.h>
  27
  28 #include <linux/videodev2.h>
  29 #include <media/v4l2-device.h>
  30 #include <media/v4l2-common.h>
  31 #include <media/v4l2-ioctl.h>
  32 #include <media/v4l2-fh.h>
  33 #include <media/v4l2-event.h>
  34 #include <media/videobuf2-vmalloc.h>
  35
  36 #include <media/i2c/saa7115.h>
  37
  38 #include "stk1160.h"
  39 #include "stk1160-reg.h"
  40
  41 static bool keep_buffers;
  42 module_param(keep_buffers, bool, 0644);
  43 MODULE_PARM_DESC(keep_buffers, "don't release buffers upon stop streaming");
  44
  45 enum stk1160_decimate_mode {
  46         STK1160_DECIMATE_MORE_THAN_HALF,
  47         STK1160_DECIMATE_LESS_THAN_HALF,
  48 };
  49
  50 struct stk1160_decimate_ctrl {
  51         bool col_en, row_en;
  52         enum stk1160_decimate_mode col_mode, row_mode;
  53         unsigned int col_n, row_n;
  54 };
  55
  56 /* supported video standards */
  57 static struct stk1160_fmt format[] = {
  58         {
  59                 .name     = "16 bpp YUY2, 4:2:2, packed",
  60                 .fourcc   = V4L2_PIX_FMT_UYVY,
  61                 .depth    = 16,
  62         }
  63 };
  64
  65 /*
  66  * Helper to find the next divisor that results in modulo being zero.
  67  * This is required to guarantee valid decimation unit counts.
  68  */
  69 static unsigned int
  70 div_round_integer(unsigned int x, unsigned int y)
  71 {
  72         for (;; y++) {
  73                 if (x % y == 0)
  74                         return x / y;
  75         }
  76 }
  77
  78 static void stk1160_set_std(struct stk1160 *dev)
  79 {
  80         int i;
  81
  82         static struct regval std525[] = {
  83
  84                 /* 720x480 */
  85
  86                 /* Frame start */
  87                 {STK116_CFSPO_STX_L, 0x0000},
  88                 {STK116_CFSPO_STX_H, 0x0000},
  89                 {STK116_CFSPO_STY_L, 0x0003},
  90                 {STK116_CFSPO_STY_H, 0x0000},
  91
  92                 /* Frame end */
  93                 {STK116_CFEPO_ENX_L, 0x05a0},
  94                 {STK116_CFEPO_ENX_H, 0x0005},
  95                 {STK116_CFEPO_ENY_L, 0x00f3},
  96                 {STK116_CFEPO_ENY_H, 0x0000},
  97
  98                 {0xffff, 0xffff}
  99         };
 100
 101         static struct regval std625[] = {
 102
 103                 /* 720x576 */
 104
 105                 /* TODO: Each line of frame has some junk at the end */
 106                 /* Frame start */
 107                 {STK116_CFSPO,   0x0000},
 108                 {STK116_CFSPO+1, 0x0000},
 109                 {STK116_CFSPO+2, 0x0001},
 110                 {STK116_CFSPO+3, 0x0000},
 111
 112                 /* Frame end */
 113                 {STK116_CFEPO,   0x05a0},
 114                 {STK116_CFEPO+1, 0x0005},
 115                 {STK116_CFEPO+2, 0x0121},
 116                 {STK116_CFEPO+3, 0x0001},
 117
 118                 {0xffff, 0xffff}
 119         };
 120
 121         if (dev->norm & V4L2_STD_525_60) {
 122                 stk1160_dbg("registers to NTSC like standard\n");
 123                 for (i = 0; std525[i].reg != 0xffff; i++)
 124                         stk1160_write_reg(dev, std525[i].reg, std525[i].val);
 125         } else {
 126                 stk1160_dbg("registers to PAL like standard\n");
 127                 for (i = 0; std625[i].reg != 0xffff; i++)
 128                         stk1160_write_reg(dev, std625[i].reg, std625[i].val);
 129         }
 130
 131 }
 132
 133 static void stk1160_set_fmt(struct stk1160 *dev,
 134                             struct stk1160_decimate_ctrl *ctrl)
 135 {
 136         u32 val = 0;
 137
 138         if (ctrl) {
 139                 /*
 140                  * Since the format is UYVY, the device must skip or send
 141                  * a number of rows/columns multiple of four. This way, the
 142                  * colour format is preserved. The STK1160_DEC_UNIT_SIZE bit
 143                  * does exactly this.
 144                  */
 145                 val |= STK1160_DEC_UNIT_SIZE;
 146                 val |= ctrl->col_en ? STK1160_H_DEC_EN : 0;
 147                 val |= ctrl->row_en ? STK1160_V_DEC_EN : 0;
 148                 val |= ctrl->col_mode ==
 149                         STK1160_DECIMATE_MORE_THAN_HALF ?
 150                         STK1160_H_DEC_MODE : 0;
 151                 val |= ctrl->row_mode ==
 152                         STK1160_DECIMATE_MORE_THAN_HALF ?
 153                         STK1160_V_DEC_MODE : 0;
 154
 155                 /* Horizontal count units */
 156                 stk1160_write_reg(dev, STK1160_DMCTRL_H_UNITS, ctrl->col_n);
 157                 /* Vertical count units */
 158                 stk1160_write_reg(dev, STK1160_DMCTRL_V_UNITS, ctrl->row_n);
 159
 160                 stk1160_dbg("decimate 0x%x, column units %d, row units %d\n",
 161                             val, ctrl->col_n, ctrl->row_n);
 162         }
 163
 164         /* Decimation control */
 165         stk1160_write_reg(dev, STK1160_DMCTRL, val);
 166 }
 167
 168 /*
 169  * Set a new alternate setting.
 170  * Returns true is dev->max_pkt_size has changed, false otherwise.
 171  */
 172 static bool stk1160_set_alternate(struct stk1160 *dev)
 173 {
 174         int i, prev_alt = dev->alt;
 175         unsigned int min_pkt_size;
 176         bool new_pkt_size;
 177
 178         /*
 179          * If we don't set right alternate,
 180          * then we will get a green screen with junk.
 181          */
 182         min_pkt_size = STK1160_MIN_PKT_SIZE;
 183
 184         for (i = 0; i < dev->num_alt; i++) {
 185                 /* stop when the selected alt setting offers enough bandwidth */
 186                 if (dev->alt_max_pkt_size[i] >= min_pkt_size) {
 187                         dev->alt = i;
 188                         break;
 189                 /*
 190                  * otherwise make sure that we end up with the maximum bandwidth
 191                  * because the min_pkt_size equation might be wrong...
 192                  */
 193                 } else if (dev->alt_max_pkt_size[i] >
 194                            dev->alt_max_pkt_size[dev->alt])
 195                         dev->alt = i;
 196         }
 197
 198         stk1160_dbg("setting alternate %d\n", dev->alt);
 199
 200         if (dev->alt != prev_alt) {
 201                 stk1160_dbg("minimum isoc packet size: %u (alt=%d)\n",
 202                                 min_pkt_size, dev->alt);
 203                 stk1160_dbg("setting alt %d with wMaxPacketSize=%u\n",
 204                                dev->alt, dev->alt_max_pkt_size[dev->alt]);
 205                 usb_set_interface(dev->udev, 0, dev->alt);
 206         }
 207
 208         new_pkt_size = dev->max_pkt_size != dev->alt_max_pkt_size[dev->alt];
 209         dev->max_pkt_size = dev->alt_max_pkt_size[dev->alt];
 210
 211         return new_pkt_size;
 212 }
 213
 214 static int stk1160_start_streaming(struct stk1160 *dev)
 215 {
 216         bool new_pkt_size;
 217         int rc = 0;
 218         int i;
 219
 220         /* Check device presence */
 221         if (!dev->udev)
 222                 return -ENODEV;
 223
 224         if (mutex_lock_interruptible(&dev->v4l_lock))
 225                 return -ERESTARTSYS;
 226         /*
 227          * For some reason it is mandatory to set alternate *first*
 228          * and only *then* initialize isoc urbs.
 229          * Someone please explain me why ;)
 230          */
 231         new_pkt_size = stk1160_set_alternate(dev);
 232
 233         /*
 234          * We (re)allocate isoc urbs if:
 235          * there is no allocated isoc urbs, OR
 236          * a new dev->max_pkt_size is detected
 237          */
 238         if (!dev->isoc_ctl.num_bufs || new_pkt_size) {
 239                 rc = stk1160_alloc_isoc(dev);
 240                 if (rc < 0)
 241                         goto out_stop_hw;
 242         }
 243
 244         /* submit urbs and enables IRQ */
 245         for (i = 0; i < dev->isoc_ctl.num_bufs; i++) {
 246                 rc = usb_submit_urb(dev->isoc_ctl.urb[i], GFP_KERNEL);
 247                 if (rc) {
 248                         stk1160_err("cannot submit urb[%d] (%d)\n", i, rc);
 249                         goto out_uninit;
 250                 }
 251         }
 252
 253         /* Start saa711x */
 254         v4l2_device_call_all(&dev->v4l2_dev, 0, video, s_stream, 1);
 255
 256         dev->sequence = 0;
 257
 258         /* Start stk1160 */
 259         stk1160_write_reg(dev, STK1160_DCTRL, 0xb3);
 260         stk1160_write_reg(dev, STK1160_DCTRL+3, 0x00);
 261
 262         stk1160_dbg("streaming started\n");
 263
 264         mutex_unlock(&dev->v4l_lock);
 265
 266         return 0;
 267
 268 out_uninit:
 269         stk1160_uninit_isoc(dev);
 270 out_stop_hw:
 271         usb_set_interface(dev->udev, 0, 0);
 272         stk1160_clear_queue(dev);
 273
 274         mutex_unlock(&dev->v4l_lock);
 275
 276         return rc;
 277 }
 278
 279 /* Must be called with v4l_lock hold */
 280 static void stk1160_stop_hw(struct stk1160 *dev)
 281 {
 282         /* If the device is not physically present, there is nothing to do */
 283         if (!dev->udev)
 284                 return;
 285
 286         /* set alternate 0 */
 287         dev->alt = 0;
 288         stk1160_dbg("setting alternate %d\n", dev->alt);
 289         usb_set_interface(dev->udev, 0, 0);
 290
 291         /* Stop stk1160 */
 292         stk1160_write_reg(dev, STK1160_DCTRL, 0x00);
 293         stk1160_write_reg(dev, STK1160_DCTRL+3, 0x00);
 294
 295         /* Stop saa711x */
 296         v4l2_device_call_all(&dev->v4l2_dev, 0, video, s_stream, 0);
 297 }
 298
 299 static int stk1160_stop_streaming(struct stk1160 *dev)
 300 {
 301         if (mutex_lock_interruptible(&dev->v4l_lock))
 302                 return -ERESTARTSYS;
 303
 304         /*
 305          * Once URBs are cancelled, the URB complete handler
 306          * won't be running. This is required to safely release the
 307          * current buffer (dev->isoc_ctl.buf).
 308          */
 309         stk1160_cancel_isoc(dev);
 310
 311         /*
 312          * It is possible to keep buffers around using a module parameter.
 313          * This is intended to avoid memory fragmentation.
 314          */
 315         if (!keep_buffers)
 316                 stk1160_free_isoc(dev);
 317
 318         stk1160_stop_hw(dev);
 319
 320         stk1160_clear_queue(dev);
 321
 322         stk1160_dbg("streaming stopped\n");
 323
 324         mutex_unlock(&dev->v4l_lock);
 325
 326         return 0;
 327 }
 328
 329 static const struct v4l2_file_operations stk1160_fops = {
 330         .owner = THIS_MODULE,
 331         .open = v4l2_fh_open,
 332         .release = vb2_fop_release,
 333         .read = vb2_fop_read,
 334         .poll = vb2_fop_poll,
 335         .mmap = vb2_fop_mmap,
 336         .unlocked_ioctl = video_ioctl2,
 337 };
 338
 339 /*
 340  * vidioc ioctls
 341  */
 342 static int vidioc_querycap(struct file *file,
 343                 void *priv, struct v4l2_capability *cap)
 344 {
 345         struct stk1160 *dev = video_drvdata(file);
 346
 347         strcpy(cap->driver, "stk1160");
 348         strcpy(cap->card, "stk1160");
 349         usb_make_path(dev->udev, cap->bus_info, sizeof(cap->bus_info));
 350         cap->device_caps =
 351                 V4L2_CAP_VIDEO_CAPTURE |
 352                 V4L2_CAP_STREAMING |
 353                 V4L2_CAP_READWRITE;
 354         cap->capabilities = cap->device_caps | V4L2_CAP_DEVICE_CAPS;
 355         return 0;
 356 }
 357
 358 static int vidioc_enum_fmt_vid_cap(struct file *file, void  *priv,
 359                 struct v4l2_fmtdesc *f)
 360 {
 361         if (f->index != 0)
 362                 return -EINVAL;
 363
 364         strlcpy(f->description, format[f->index].name, sizeof(f->description));
 365         f->pixelformat = format[f->index].fourcc;
 366         return 0;
 367 }
 368
 369 static int vidioc_g_fmt_vid_cap(struct file *file, void *priv,
 370                                         struct v4l2_format *f)
 371 {
 372         struct stk1160 *dev = video_drvdata(file);
 373
 374         f->fmt.pix.width = dev->width;
 375         f->fmt.pix.height = dev->height;
 376         f->fmt.pix.field = V4L2_FIELD_INTERLACED;
 377         f->fmt.pix.pixelformat = dev->fmt->fourcc;
 378         f->fmt.pix.bytesperline = dev->width * 2;
 379         f->fmt.pix.sizeimage = dev->height * f->fmt.pix.bytesperline;
 380         f->fmt.pix.colorspace = V4L2_COLORSPACE_SMPTE170M;
 381
 382         return 0;
 383 }
 384
 385 static int stk1160_try_fmt(struct stk1160 *dev, struct v4l2_format *f,
 386                             struct stk1160_decimate_ctrl *ctrl)
 387 {
 388         unsigned int width, height;
 389         unsigned int base_width, base_height;
 390         unsigned int col_n, row_n;
 391         enum stk1160_decimate_mode col_mode, row_mode;
 392         bool col_en, row_en;
 393
 394         base_width = 720;
 395         base_height = (dev->norm & V4L2_STD_525_60) ? 480 : 576;
 396
 397         /* Minimum width and height is 5% the frame size */
 398         width = clamp_t(unsigned int, f->fmt.pix.width,
 399                         base_width / 20, base_width);
 400         height = clamp_t(unsigned int, f->fmt.pix.height,
 401                         base_height / 20, base_height);
 402
 403         /* Let's set default no decimation values */
 404         col_n = 0;
 405         row_n = 0;
 406         col_en = false;
 407         row_en = false;
 408         f->fmt.pix.width = base_width;
 409         f->fmt.pix.height = base_height;
 410         row_mode = STK1160_DECIMATE_LESS_THAN_HALF;
 411         col_mode = STK1160_DECIMATE_LESS_THAN_HALF;
 412
 413         if (width < base_width && width > base_width / 2) {
 414                 /*
 415                  * The device will send count units for each
 416                  * unit skipped. This means count unit is:
 417                  *
 418                  * n = width / (frame width - width)
 419                  *
 420                  * And the width is:
 421                  *
 422                  * width = (n / n + 1) * frame width
 423                  */
 424                 col_n = div_round_integer(width, base_width - width);
 425                 if (col_n > 0 && col_n <= 255) {
 426                         col_en = true;
 427                         col_mode = STK1160_DECIMATE_LESS_THAN_HALF;
 428                         f->fmt.pix.width = (base_width * col_n) / (col_n + 1);
 429                 }
 430
 431         } else if (width <= base_width / 2) {
 432
 433                 /*
 434                  * The device will skip count units for each
 435                  * unit sent. This means count is:
 436                  *
 437                  * n = (frame width / width) - 1
 438                  *
 439                  * And the width is:
 440                  *
 441                  * width = frame width / (n + 1)
 442                  */
 443                 col_n = div_round_integer(base_width, width) - 1;
 444                 if (col_n > 0 && col_n <= 255) {
 445                         col_en = true;
 446                         col_mode = STK1160_DECIMATE_MORE_THAN_HALF;
 447                         f->fmt.pix.width = base_width / (col_n + 1);
 448                 }
 449         }
 450
 451         if (height < base_height && height > base_height / 2) {
 452                 row_n = div_round_integer(height, base_height - height);
 453                 if (row_n > 0 && row_n <= 255) {
 454                         row_en = true;
 455                         row_mode = STK1160_DECIMATE_LESS_THAN_HALF;
 456                         f->fmt.pix.height = (base_height * row_n) / (row_n + 1);
 457                 }
 458
 459         } else if (height <= base_height / 2) {
 460                 row_n = div_round_integer(base_height, height) - 1;
 461                 if (row_n > 0 && row_n <= 255) {
 462                         row_en = true;
 463                         row_mode = STK1160_DECIMATE_MORE_THAN_HALF;
 464                         f->fmt.pix.height = base_height / (row_n + 1);
 465                 }
 466         }
 467
 468         f->fmt.pix.pixelformat = dev->fmt->fourcc;
 469         f->fmt.pix.field = V4L2_FIELD_INTERLACED;
 470         f->fmt.pix.bytesperline = f->fmt.pix.width * 2;
 471         f->fmt.pix.sizeimage = f->fmt.pix.height * f->fmt.pix.bytesperline;
 472         f->fmt.pix.colorspace = V4L2_COLORSPACE_SMPTE170M;
 473
 474         if (ctrl) {
 475                 ctrl->col_en = col_en;
 476                 ctrl->col_n = col_n;
 477                 ctrl->col_mode = col_mode;
 478                 ctrl->row_en = row_en;
 479                 ctrl->row_n = row_n;
 480                 ctrl->row_mode = row_mode;
 481         }
 482
 483         stk1160_dbg("width %d, height %d\n",
 484                     f->fmt.pix.width, f->fmt.pix.height);
 485         return 0;
 486 }
 487
 488 static int vidioc_try_fmt_vid_cap(struct file *file, void *priv,
 489                                   struct v4l2_format *f)
 490 {
 491         struct stk1160 *dev = video_drvdata(file);
 492
 493         return stk1160_try_fmt(dev, f, NULL);
 494 }
 495
 496 static int vidioc_s_fmt_vid_cap(struct file *file, void *priv,
 497                                         struct v4l2_format *f)
 498 {
 499         struct stk1160 *dev = video_drvdata(file);
 500         struct vb2_queue *q = &dev->vb_vidq;
 501         struct stk1160_decimate_ctrl ctrl;
 502         int rc;
 503
 504         if (vb2_is_busy(q))
 505                 return -EBUSY;
 506
 507         rc = stk1160_try_fmt(dev, f, &ctrl);
 508         if (rc < 0)
 509                 return rc;
 510         dev->width = f->fmt.pix.width;
 511         dev->height = f->fmt.pix.height;
 512         stk1160_set_fmt(dev, &ctrl);
 513
 514         return 0;
 515 }
 516
 517 static int vidioc_querystd(struct file *file, void *priv, v4l2_std_id *norm)
 518 {
 519         struct stk1160 *dev = video_drvdata(file);
 520         v4l2_device_call_all(&dev->v4l2_dev, 0, video, querystd, norm);
 521         return 0;
 522 }
 523
 524 static int vidioc_g_std(struct file *file, void *priv, v4l2_std_id *norm)
 525 {
 526         struct stk1160 *dev = video_drvdata(file);
 527
 528         *norm = dev->norm;
 529         return 0;
 530 }
 531
 532 static int vidioc_s_std(struct file *file, void *priv, v4l2_std_id norm)
 533 {
 534         struct stk1160 *dev = video_drvdata(file);
 535         struct vb2_queue *q = &dev->vb_vidq;
 536
 537         if (dev->norm == norm)
 538                 return 0;
 539
 540         if (vb2_is_busy(q))
 541                 return -EBUSY;
 542
 543         /* Check device presence */
 544         if (!dev->udev)
 545                 return -ENODEV;
 546
 547         /* We need to set this now, before we call stk1160_set_std */
 548         dev->width = 720;
 549         dev->height = (norm & V4L2_STD_525_60) ? 480 : 576;
 550         dev->norm = norm;
 551
 552         stk1160_set_std(dev);
 553
 554         /* Calling with NULL disables frame decimation */
 555         stk1160_set_fmt(dev, NULL);
 556
 557         v4l2_device_call_all(&dev->v4l2_dev, 0, video, s_std,
 558                         dev->norm);
 559
 560         return 0;
 561 }
 562
 563
 564 static int vidioc_enum_input(struct file *file, void *priv,
 565                                 struct v4l2_input *i)
 566 {
 567         struct stk1160 *dev = video_drvdata(file);
 568
 569         if (i->index > STK1160_MAX_INPUT)
 570                 return -EINVAL;
 571
 572         /* S-Video special handling */
 573         if (i->index == STK1160_SVIDEO_INPUT)
 574                 sprintf(i->name, "S-Video");
 575         else
 576                 sprintf(i->name, "Composite%d", i->index);
 577
 578         i->type = V4L2_INPUT_TYPE_CAMERA;
 579         i->std = dev->vdev.tvnorms;
 580         return 0;
 581 }
 582
 583 static int vidioc_g_input(struct file *file, void *priv, unsigned int *i)
 584 {
 585         struct stk1160 *dev = video_drvdata(file);
 586         *i = dev->ctl_input;
 587         return 0;
 588 }
 589
 590 static int vidioc_s_input(struct file *file, void *priv, unsigned int i)
 591 {
 592         struct stk1160 *dev = video_drvdata(file);
 593
 594         if (i > STK1160_MAX_INPUT)
 595                 return -EINVAL;
 596
 597         dev->ctl_input = i;
 598
 599         stk1160_select_input(dev);
 600
 601         return 0;
 602 }
 603
 604 #ifdef CONFIG_VIDEO_ADV_DEBUG
 605 static int vidioc_g_register(struct file *file, void *priv,
 606                              struct v4l2_dbg_register *reg)
 607 {
 608         struct stk1160 *dev = video_drvdata(file);
 609         int rc;
 610         u8 val;
 611
 612         /* Match host */
 613         rc = stk1160_read_reg(dev, reg->reg, &val);
 614         reg->val = val;
 615         reg->size = 1;
 616
 617         return rc;
 618 }
 619
 620 static int vidioc_s_register(struct file *file, void *priv,
 621                              const struct v4l2_dbg_register *reg)
 622 {
 623         struct stk1160 *dev = video_drvdata(file);
 624
 625         /* Match host */
 626         return stk1160_write_reg(dev, reg->reg, reg->val);
 627 }
 628 #endif
 629
 630 static const struct v4l2_ioctl_ops stk1160_ioctl_ops = {
 631         .vidioc_querycap      = vidioc_querycap,
 632         .vidioc_enum_fmt_vid_cap  = vidioc_enum_fmt_vid_cap,
 633         .vidioc_g_fmt_vid_cap     = vidioc_g_fmt_vid_cap,
 634         .vidioc_try_fmt_vid_cap   = vidioc_try_fmt_vid_cap,
 635         .vidioc_s_fmt_vid_cap     = vidioc_s_fmt_vid_cap,
 636         .vidioc_querystd      = vidioc_querystd,
 637         .vidioc_g_std         = vidioc_g_std,
 638         .vidioc_s_std         = vidioc_s_std,
 639         .vidioc_enum_input    = vidioc_enum_input,
 640         .vidioc_g_input       = vidioc_g_input,
 641         .vidioc_s_input       = vidioc_s_input,
 642
 643         /* vb2 takes care of these */
 644         .vidioc_reqbufs       = vb2_ioctl_reqbufs,
 645         .vidioc_querybuf      = vb2_ioctl_querybuf,
 646         .vidioc_qbuf          = vb2_ioctl_qbuf,
 647         .vidioc_dqbuf         = vb2_ioctl_dqbuf,
 648         .vidioc_streamon      = vb2_ioctl_streamon,
 649         .vidioc_streamoff     = vb2_ioctl_streamoff,
 650         .vidioc_expbuf        = vb2_ioctl_expbuf,
 651
 652         .vidioc_log_status  = v4l2_ctrl_log_status,
 653         .vidioc_subscribe_event = v4l2_ctrl_subscribe_event,
 654         .vidioc_unsubscribe_event = v4l2_event_unsubscribe,
 655
 656 #ifdef CONFIG_VIDEO_ADV_DEBUG
 657         .vidioc_g_register = vidioc_g_register,
 658         .vidioc_s_register = vidioc_s_register,
 659 #endif
 660 };
 661
 662 /********************************************************************/
 663
 664 /*
 665  * Videobuf2 operations
 666  */
 667 static int queue_setup(struct vb2_queue *vq,
 668                                 unsigned int *nbuffers, unsigned int *nplanes,
 669                                 unsigned int sizes[], struct device *alloc_devs[])
 670 {
 671         struct stk1160 *dev = vb2_get_drv_priv(vq);
 672         unsigned long size;
 673
 674         size = dev->width * dev->height * 2;
 675
 676         /*
 677          * Here we can change the number of buffers being requested.
 678          * So, we set a minimum and a maximum like this:
 679          */
 680         *nbuffers = clamp_t(unsigned int, *nbuffers,
 681                         STK1160_MIN_VIDEO_BUFFERS, STK1160_MAX_VIDEO_BUFFERS);
 682
 683         if (*nplanes)
 684                 return sizes[0] < size ? -EINVAL : 0;
 685
 686         /* This means a packed colorformat */
 687         *nplanes = 1;
 688
 689         sizes[0] = size;
 690
 691         stk1160_dbg("%s: buffer count %d, each %ld bytes\n",
 692                     __func__, *nbuffers, size);
 693
 694         return 0;
 695 }
 696
 697 static void buffer_queue(struct vb2_buffer *vb)
 698 {
 699         unsigned long flags;
 700         struct stk1160 *dev = vb2_get_drv_priv(vb->vb2_queue);
 701         struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
 702         struct stk1160_buffer *buf =
 703                 container_of(vbuf, struct stk1160_buffer, vb);
 704
 705         spin_lock_irqsave(&dev->buf_lock, flags);
 706         if (!dev->udev) {
 707                 /*
 708                  * If the device is disconnected return the buffer to userspace
 709                  * directly. The next QBUF call will fail with -ENODEV.
 710                  */
 711                 vb2_buffer_done(&buf->vb.vb2_buf, VB2_BUF_STATE_ERROR);
 712         } else {
 713
 714                 buf->mem = vb2_plane_vaddr(vb, 0);
 715                 buf->length = vb2_plane_size(vb, 0);
 716                 buf->bytesused = 0;
 717                 buf->pos = 0;
 718
 719                 /*
 720                  * If buffer length is less from expected then we return
 721                  * the buffer to userspace directly.
 722                  */
 723                 if (buf->length < dev->width * dev->height * 2)
 724                         vb2_buffer_done(&buf->vb.vb2_buf, VB2_BUF_STATE_ERROR);
 725                 else
 726                         list_add_tail(&buf->list, &dev->avail_bufs);
 727
 728         }
 729         spin_unlock_irqrestore(&dev->buf_lock, flags);
 730 }
 731
 732 static int start_streaming(struct vb2_queue *vq, unsigned int count)
 733 {
 734         struct stk1160 *dev = vb2_get_drv_priv(vq);
 735         return stk1160_start_streaming(dev);
 736 }
 737
 738 /* abort streaming and wait for last buffer */
 739 static void stop_streaming(struct vb2_queue *vq)
 740 {
 741         struct stk1160 *dev = vb2_get_drv_priv(vq);
 742         stk1160_stop_streaming(dev);
 743 }
 744
 745 static const struct vb2_ops stk1160_video_qops = {
 746         .queue_setup            = queue_setup,
 747         .buf_queue              = buffer_queue,
 748         .start_streaming        = start_streaming,
 749         .stop_streaming         = stop_streaming,
 750         .wait_prepare           = vb2_ops_wait_prepare,
 751         .wait_finish            = vb2_ops_wait_finish,
 752 };
 753
 754 static const struct video_device v4l_template = {
 755         .name = "stk1160",
 756         .tvnorms = V4L2_STD_525_60 | V4L2_STD_625_50,
 757         .fops = &stk1160_fops,
 758         .ioctl_ops = &stk1160_ioctl_ops,
 759         .release = video_device_release_empty,
 760 };
 761
 762 /********************************************************************/
 763
 764 /* Must be called with both v4l_lock and vb_queue_lock hold */
 765 void stk1160_clear_queue(struct stk1160 *dev)
 766 {
 767         struct stk1160_buffer *buf;
 768         unsigned long flags;
 769
 770         /* Release all active buffers */
 771         spin_lock_irqsave(&dev->buf_lock, flags);
 772         while (!list_empty(&dev->avail_bufs)) {
 773                 buf = list_first_entry(&dev->avail_bufs,
 774                         struct stk1160_buffer, list);
 775                 list_del(&buf->list);
 776                 vb2_buffer_done(&buf->vb.vb2_buf, VB2_BUF_STATE_ERROR);
 777                 stk1160_dbg("buffer [%p/%d] aborted\n",
 778                             buf, buf->vb.vb2_buf.index);
 779         }
 780
 781         /* It's important to release the current buffer */
 782         if (dev->isoc_ctl.buf) {
 783                 buf = dev->isoc_ctl.buf;
 784                 dev->isoc_ctl.buf = NULL;
 785
 786                 vb2_buffer_done(&buf->vb.vb2_buf, VB2_BUF_STATE_ERROR);
 787                 stk1160_dbg("buffer [%p/%d] aborted\n",
 788                             buf, buf->vb.vb2_buf.index);
 789         }
 790         spin_unlock_irqrestore(&dev->buf_lock, flags);
 791 }
 792
 793 int stk1160_vb2_setup(struct stk1160 *dev)
 794 {
 795         int rc;
 796         struct vb2_queue *q;
 797
 798         q = &dev->vb_vidq;
 799         q->type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
 800         q->io_modes = VB2_READ | VB2_MMAP | VB2_USERPTR | VB2_DMABUF;
 801         q->drv_priv = dev;
 802         q->buf_struct_size = sizeof(struct stk1160_buffer);
 803         q->ops = &stk1160_video_qops;
 804         q->mem_ops = &vb2_vmalloc_memops;
 805         q->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_MONOTONIC;
 806
 807         rc = vb2_queue_init(q);
 808         if (rc < 0)
 809                 return rc;
 810
 811         /* initialize video dma queue */
 812         INIT_LIST_HEAD(&dev->avail_bufs);
 813
 814         return 0;
 815 }
 816
 817 int stk1160_video_register(struct stk1160 *dev)
 818 {
 819         int rc;
 820
 821         /* Initialize video_device with a template structure */
 822         dev->vdev = v4l_template;
 823         dev->vdev.queue = &dev->vb_vidq;
 824
 825         /*
 826          * Provide mutexes for v4l2 core and for videobuf2 queue.
 827          * It will be used to protect *only* v4l2 ioctls.
 828          */
 829         dev->vdev.lock = &dev->v4l_lock;
 830         dev->vdev.queue->lock = &dev->vb_queue_lock;
 831
 832         /* This will be used to set video_device parent */
 833         dev->vdev.v4l2_dev = &dev->v4l2_dev;
 834
 835         /* NTSC is default */
 836         dev->norm = V4L2_STD_NTSC_M;
 837         dev->width = 720;
 838         dev->height = 480;
 839
 840         /* set default format */
 841         dev->fmt = &format[0];
 842         stk1160_set_std(dev);
 843
 844         v4l2_device_call_all(&dev->v4l2_dev, 0, video, s_std,
 845                         dev->norm);
 846
 847         video_set_drvdata(&dev->vdev, dev);
 848         rc = video_register_device(&dev->vdev, VFL_TYPE_GRABBER, -1);
 849         if (rc < 0) {
 850                 stk1160_err("video_register_device failed (%d)\n", rc);
 851                 return rc;
 852         }
 853
 854         v4l2_info(&dev->v4l2_dev, "V4L2 device registered as %s\n",
 855                   video_device_node_name(&dev->vdev));
 856
 857         return 0;
 858 }