linux/drivers/staging/media/imx/imx-media-capture.c
<<
>>
Prefs
   1// SPDX-License-Identifier: GPL-2.0+
   2/*
   3 * Video Capture Subdev for Freescale i.MX5/6 SOC
   4 *
   5 * Copyright (c) 2012-2016 Mentor Graphics Inc.
   6 */
   7#include <linux/delay.h>
   8#include <linux/fs.h>
   9#include <linux/module.h>
  10#include <linux/of_platform.h>
  11#include <linux/pinctrl/consumer.h>
  12#include <linux/platform_device.h>
  13#include <linux/sched.h>
  14#include <linux/slab.h>
  15#include <linux/spinlock.h>
  16#include <linux/timer.h>
  17#include <media/v4l2-ctrls.h>
  18#include <media/v4l2-device.h>
  19#include <media/v4l2-event.h>
  20#include <media/v4l2-fwnode.h>
  21#include <media/v4l2-ioctl.h>
  22#include <media/v4l2-mc.h>
  23#include <media/v4l2-subdev.h>
  24#include <media/videobuf2-dma-contig.h>
  25#include <video/imx-ipu-v3.h>
  26#include <media/imx.h>
  27#include "imx-media.h"
  28
  29#define IMX_CAPTURE_NAME "imx-capture"
  30
  31struct capture_priv {
  32        struct imx_media_video_dev vdev;
  33
  34        struct v4l2_subdev    *src_sd;
  35        int                   src_sd_pad;
  36        struct device         *dev;
  37
  38        struct imx_media_dev  *md;
  39
  40        struct media_pad      vdev_pad;
  41
  42        struct mutex          mutex;       /* capture device mutex */
  43
  44        /* the videobuf2 queue */
  45        struct vb2_queue       q;
  46        /* list of ready imx_media_buffer's from q */
  47        struct list_head       ready_q;
  48        /* protect ready_q */
  49        spinlock_t             q_lock;
  50
  51        /* controls inherited from subdevs */
  52        struct v4l2_ctrl_handler ctrl_hdlr;
  53
  54        /* misc status */
  55        bool                  stop;          /* streaming is stopping */
  56};
  57
  58#define to_capture_priv(v) container_of(v, struct capture_priv, vdev)
  59
  60/* In bytes, per queue */
  61#define VID_MEM_LIMIT   SZ_64M
  62
  63static const struct vb2_ops capture_qops;
  64
  65/*
  66 * Video ioctls follow
  67 */
  68
  69static int vidioc_querycap(struct file *file, void *fh,
  70                           struct v4l2_capability *cap)
  71{
  72        struct capture_priv *priv = video_drvdata(file);
  73
  74        strscpy(cap->driver, IMX_CAPTURE_NAME, sizeof(cap->driver));
  75        strscpy(cap->card, IMX_CAPTURE_NAME, sizeof(cap->card));
  76        snprintf(cap->bus_info, sizeof(cap->bus_info),
  77                 "platform:%s", priv->src_sd->name);
  78
  79        return 0;
  80}
  81
  82static int capture_enum_framesizes(struct file *file, void *fh,
  83                                   struct v4l2_frmsizeenum *fsize)
  84{
  85        struct capture_priv *priv = video_drvdata(file);
  86        const struct imx_media_pixfmt *cc;
  87        struct v4l2_subdev_frame_size_enum fse = {
  88                .index = fsize->index,
  89                .pad = priv->src_sd_pad,
  90                .which = V4L2_SUBDEV_FORMAT_ACTIVE,
  91        };
  92        int ret;
  93
  94        cc = imx_media_find_pixel_format(fsize->pixel_format, PIXFMT_SEL_ANY);
  95        if (!cc)
  96                return -EINVAL;
  97
  98        fse.code = cc->codes ? cc->codes[0] : 0;
  99
 100        ret = v4l2_subdev_call(priv->src_sd, pad, enum_frame_size, NULL, &fse);
 101        if (ret)
 102                return ret;
 103
 104        if (fse.min_width == fse.max_width &&
 105            fse.min_height == fse.max_height) {
 106                fsize->type = V4L2_FRMSIZE_TYPE_DISCRETE;
 107                fsize->discrete.width = fse.min_width;
 108                fsize->discrete.height = fse.min_height;
 109        } else {
 110                fsize->type = V4L2_FRMSIZE_TYPE_CONTINUOUS;
 111                fsize->stepwise.min_width = fse.min_width;
 112                fsize->stepwise.max_width = fse.max_width;
 113                fsize->stepwise.min_height = fse.min_height;
 114                fsize->stepwise.max_height = fse.max_height;
 115                fsize->stepwise.step_width = 1;
 116                fsize->stepwise.step_height = 1;
 117        }
 118
 119        return 0;
 120}
 121
 122static int capture_enum_frameintervals(struct file *file, void *fh,
 123                                       struct v4l2_frmivalenum *fival)
 124{
 125        struct capture_priv *priv = video_drvdata(file);
 126        const struct imx_media_pixfmt *cc;
 127        struct v4l2_subdev_frame_interval_enum fie = {
 128                .index = fival->index,
 129                .pad = priv->src_sd_pad,
 130                .width = fival->width,
 131                .height = fival->height,
 132                .which = V4L2_SUBDEV_FORMAT_ACTIVE,
 133        };
 134        int ret;
 135
 136        cc = imx_media_find_pixel_format(fival->pixel_format, PIXFMT_SEL_ANY);
 137        if (!cc)
 138                return -EINVAL;
 139
 140        fie.code = cc->codes ? cc->codes[0] : 0;
 141
 142        ret = v4l2_subdev_call(priv->src_sd, pad, enum_frame_interval,
 143                               NULL, &fie);
 144        if (ret)
 145                return ret;
 146
 147        fival->type = V4L2_FRMIVAL_TYPE_DISCRETE;
 148        fival->discrete = fie.interval;
 149
 150        return 0;
 151}
 152
 153static int capture_enum_fmt_vid_cap(struct file *file, void *fh,
 154                                    struct v4l2_fmtdesc *f)
 155{
 156        struct capture_priv *priv = video_drvdata(file);
 157        const struct imx_media_pixfmt *cc_src;
 158        struct v4l2_subdev_format fmt_src;
 159        u32 fourcc;
 160        int ret;
 161
 162        fmt_src.pad = priv->src_sd_pad;
 163        fmt_src.which = V4L2_SUBDEV_FORMAT_ACTIVE;
 164        ret = v4l2_subdev_call(priv->src_sd, pad, get_fmt, NULL, &fmt_src);
 165        if (ret) {
 166                v4l2_err(priv->src_sd, "failed to get src_sd format\n");
 167                return ret;
 168        }
 169
 170        cc_src = imx_media_find_ipu_format(fmt_src.format.code,
 171                                           PIXFMT_SEL_YUV_RGB);
 172        if (cc_src) {
 173                enum imx_pixfmt_sel fmt_sel =
 174                        (cc_src->cs == IPUV3_COLORSPACE_YUV) ?
 175                        PIXFMT_SEL_YUV : PIXFMT_SEL_RGB;
 176
 177                ret = imx_media_enum_pixel_formats(&fourcc, f->index, fmt_sel);
 178                if (ret)
 179                        return ret;
 180        } else {
 181                cc_src = imx_media_find_mbus_format(fmt_src.format.code,
 182                                                    PIXFMT_SEL_ANY);
 183                if (WARN_ON(!cc_src))
 184                        return -EINVAL;
 185
 186                if (f->index != 0)
 187                        return -EINVAL;
 188                fourcc = cc_src->fourcc;
 189        }
 190
 191        f->pixelformat = fourcc;
 192
 193        return 0;
 194}
 195
 196static int capture_g_fmt_vid_cap(struct file *file, void *fh,
 197                                 struct v4l2_format *f)
 198{
 199        struct capture_priv *priv = video_drvdata(file);
 200
 201        *f = priv->vdev.fmt;
 202
 203        return 0;
 204}
 205
 206static int __capture_try_fmt_vid_cap(struct capture_priv *priv,
 207                                     struct v4l2_subdev_format *fmt_src,
 208                                     struct v4l2_format *f,
 209                                     const struct imx_media_pixfmt **retcc,
 210                                     struct v4l2_rect *compose)
 211{
 212        const struct imx_media_pixfmt *cc, *cc_src;
 213
 214        cc_src = imx_media_find_ipu_format(fmt_src->format.code,
 215                                           PIXFMT_SEL_YUV_RGB);
 216        if (cc_src) {
 217                enum imx_pixfmt_sel fmt_sel;
 218                u32 fourcc;
 219
 220                fmt_sel = (cc_src->cs == IPUV3_COLORSPACE_YUV) ?
 221                        PIXFMT_SEL_YUV : PIXFMT_SEL_RGB;
 222                fourcc = f->fmt.pix.pixelformat;
 223
 224                cc = imx_media_find_pixel_format(fourcc, fmt_sel);
 225                if (!cc) {
 226                        imx_media_enum_pixel_formats(&fourcc, 0, fmt_sel);
 227                        cc = imx_media_find_pixel_format(fourcc, fmt_sel);
 228                }
 229        } else {
 230                cc_src = imx_media_find_mbus_format(fmt_src->format.code,
 231                                                    PIXFMT_SEL_ANY);
 232                if (WARN_ON(!cc_src))
 233                        return -EINVAL;
 234
 235                cc = cc_src;
 236        }
 237
 238        /* allow IDMAC interweave but enforce field order from source */
 239        if (V4L2_FIELD_IS_INTERLACED(f->fmt.pix.field)) {
 240                switch (fmt_src->format.field) {
 241                case V4L2_FIELD_SEQ_TB:
 242                        fmt_src->format.field = V4L2_FIELD_INTERLACED_TB;
 243                        break;
 244                case V4L2_FIELD_SEQ_BT:
 245                        fmt_src->format.field = V4L2_FIELD_INTERLACED_BT;
 246                        break;
 247                default:
 248                        break;
 249                }
 250        }
 251
 252        imx_media_mbus_fmt_to_pix_fmt(&f->fmt.pix, &fmt_src->format, cc);
 253
 254        if (retcc)
 255                *retcc = cc;
 256
 257        if (compose) {
 258                compose->left = 0;
 259                compose->top = 0;
 260                compose->width = fmt_src->format.width;
 261                compose->height = fmt_src->format.height;
 262        }
 263
 264        return 0;
 265}
 266
 267static int capture_try_fmt_vid_cap(struct file *file, void *fh,
 268                                   struct v4l2_format *f)
 269{
 270        struct capture_priv *priv = video_drvdata(file);
 271        struct v4l2_subdev_format fmt_src;
 272        int ret;
 273
 274        fmt_src.pad = priv->src_sd_pad;
 275        fmt_src.which = V4L2_SUBDEV_FORMAT_ACTIVE;
 276        ret = v4l2_subdev_call(priv->src_sd, pad, get_fmt, NULL, &fmt_src);
 277        if (ret)
 278                return ret;
 279
 280        return __capture_try_fmt_vid_cap(priv, &fmt_src, f, NULL, NULL);
 281}
 282
 283static int capture_s_fmt_vid_cap(struct file *file, void *fh,
 284                                 struct v4l2_format *f)
 285{
 286        struct capture_priv *priv = video_drvdata(file);
 287        struct v4l2_subdev_format fmt_src;
 288        int ret;
 289
 290        if (vb2_is_busy(&priv->q)) {
 291                v4l2_err(priv->src_sd, "%s queue busy\n", __func__);
 292                return -EBUSY;
 293        }
 294
 295        fmt_src.pad = priv->src_sd_pad;
 296        fmt_src.which = V4L2_SUBDEV_FORMAT_ACTIVE;
 297        ret = v4l2_subdev_call(priv->src_sd, pad, get_fmt, NULL, &fmt_src);
 298        if (ret)
 299                return ret;
 300
 301        ret = __capture_try_fmt_vid_cap(priv, &fmt_src, f, &priv->vdev.cc,
 302                                        &priv->vdev.compose);
 303        if (ret)
 304                return ret;
 305
 306        priv->vdev.fmt.fmt.pix = f->fmt.pix;
 307
 308        return 0;
 309}
 310
 311static int capture_querystd(struct file *file, void *fh, v4l2_std_id *std)
 312{
 313        struct capture_priv *priv = video_drvdata(file);
 314
 315        return v4l2_subdev_call(priv->src_sd, video, querystd, std);
 316}
 317
 318static int capture_g_std(struct file *file, void *fh, v4l2_std_id *std)
 319{
 320        struct capture_priv *priv = video_drvdata(file);
 321
 322        return v4l2_subdev_call(priv->src_sd, video, g_std, std);
 323}
 324
 325static int capture_s_std(struct file *file, void *fh, v4l2_std_id std)
 326{
 327        struct capture_priv *priv = video_drvdata(file);
 328
 329        if (vb2_is_busy(&priv->q))
 330                return -EBUSY;
 331
 332        return v4l2_subdev_call(priv->src_sd, video, s_std, std);
 333}
 334
 335static int capture_g_selection(struct file *file, void *fh,
 336                               struct v4l2_selection *s)
 337{
 338        struct capture_priv *priv = video_drvdata(file);
 339
 340        switch (s->target) {
 341        case V4L2_SEL_TGT_COMPOSE:
 342        case V4L2_SEL_TGT_COMPOSE_DEFAULT:
 343        case V4L2_SEL_TGT_COMPOSE_BOUNDS:
 344                /* The compose rectangle is fixed to the source format. */
 345                s->r = priv->vdev.compose;
 346                break;
 347        case V4L2_SEL_TGT_COMPOSE_PADDED:
 348                /*
 349                 * The hardware writes with a configurable but fixed DMA burst
 350                 * size. If the source format width is not burst size aligned,
 351                 * the written frame contains padding to the right.
 352                 */
 353                s->r.left = 0;
 354                s->r.top = 0;
 355                s->r.width = priv->vdev.fmt.fmt.pix.width;
 356                s->r.height = priv->vdev.fmt.fmt.pix.height;
 357                break;
 358        default:
 359                return -EINVAL;
 360        }
 361
 362        return 0;
 363}
 364
 365static int capture_g_parm(struct file *file, void *fh,
 366                          struct v4l2_streamparm *a)
 367{
 368        struct capture_priv *priv = video_drvdata(file);
 369        struct v4l2_subdev_frame_interval fi;
 370        int ret;
 371
 372        if (a->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
 373                return -EINVAL;
 374
 375        memset(&fi, 0, sizeof(fi));
 376        fi.pad = priv->src_sd_pad;
 377        ret = v4l2_subdev_call(priv->src_sd, video, g_frame_interval, &fi);
 378        if (ret < 0)
 379                return ret;
 380
 381        a->parm.capture.capability = V4L2_CAP_TIMEPERFRAME;
 382        a->parm.capture.timeperframe = fi.interval;
 383
 384        return 0;
 385}
 386
 387static int capture_s_parm(struct file *file, void *fh,
 388                          struct v4l2_streamparm *a)
 389{
 390        struct capture_priv *priv = video_drvdata(file);
 391        struct v4l2_subdev_frame_interval fi;
 392        int ret;
 393
 394        if (a->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
 395                return -EINVAL;
 396
 397        memset(&fi, 0, sizeof(fi));
 398        fi.pad = priv->src_sd_pad;
 399        fi.interval = a->parm.capture.timeperframe;
 400        ret = v4l2_subdev_call(priv->src_sd, video, s_frame_interval, &fi);
 401        if (ret < 0)
 402                return ret;
 403
 404        a->parm.capture.capability = V4L2_CAP_TIMEPERFRAME;
 405        a->parm.capture.timeperframe = fi.interval;
 406
 407        return 0;
 408}
 409
 410static int capture_subscribe_event(struct v4l2_fh *fh,
 411                                   const struct v4l2_event_subscription *sub)
 412{
 413        switch (sub->type) {
 414        case V4L2_EVENT_IMX_FRAME_INTERVAL_ERROR:
 415                return v4l2_event_subscribe(fh, sub, 0, NULL);
 416        case V4L2_EVENT_SOURCE_CHANGE:
 417                return v4l2_src_change_event_subscribe(fh, sub);
 418        case V4L2_EVENT_CTRL:
 419                return v4l2_ctrl_subscribe_event(fh, sub);
 420        default:
 421                return -EINVAL;
 422        }
 423}
 424
 425static const struct v4l2_ioctl_ops capture_ioctl_ops = {
 426        .vidioc_querycap        = vidioc_querycap,
 427
 428        .vidioc_enum_framesizes = capture_enum_framesizes,
 429        .vidioc_enum_frameintervals = capture_enum_frameintervals,
 430
 431        .vidioc_enum_fmt_vid_cap        = capture_enum_fmt_vid_cap,
 432        .vidioc_g_fmt_vid_cap           = capture_g_fmt_vid_cap,
 433        .vidioc_try_fmt_vid_cap         = capture_try_fmt_vid_cap,
 434        .vidioc_s_fmt_vid_cap           = capture_s_fmt_vid_cap,
 435
 436        .vidioc_querystd        = capture_querystd,
 437        .vidioc_g_std           = capture_g_std,
 438        .vidioc_s_std           = capture_s_std,
 439
 440        .vidioc_g_selection     = capture_g_selection,
 441
 442        .vidioc_g_parm          = capture_g_parm,
 443        .vidioc_s_parm          = capture_s_parm,
 444
 445        .vidioc_reqbufs         = vb2_ioctl_reqbufs,
 446        .vidioc_create_bufs     = vb2_ioctl_create_bufs,
 447        .vidioc_prepare_buf     = vb2_ioctl_prepare_buf,
 448        .vidioc_querybuf        = vb2_ioctl_querybuf,
 449        .vidioc_qbuf            = vb2_ioctl_qbuf,
 450        .vidioc_dqbuf           = vb2_ioctl_dqbuf,
 451        .vidioc_expbuf          = vb2_ioctl_expbuf,
 452        .vidioc_streamon        = vb2_ioctl_streamon,
 453        .vidioc_streamoff       = vb2_ioctl_streamoff,
 454
 455        .vidioc_subscribe_event = capture_subscribe_event,
 456        .vidioc_unsubscribe_event = v4l2_event_unsubscribe,
 457};
 458
 459/*
 460 * Queue operations
 461 */
 462
 463static int capture_queue_setup(struct vb2_queue *vq,
 464                               unsigned int *nbuffers,
 465                               unsigned int *nplanes,
 466                               unsigned int sizes[],
 467                               struct device *alloc_devs[])
 468{
 469        struct capture_priv *priv = vb2_get_drv_priv(vq);
 470        struct v4l2_pix_format *pix = &priv->vdev.fmt.fmt.pix;
 471        unsigned int count = *nbuffers;
 472
 473        if (vq->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
 474                return -EINVAL;
 475
 476        if (*nplanes) {
 477                if (*nplanes != 1 || sizes[0] < pix->sizeimage)
 478                        return -EINVAL;
 479                count += vq->num_buffers;
 480        }
 481
 482        count = min_t(__u32, VID_MEM_LIMIT / pix->sizeimage, count);
 483
 484        if (*nplanes)
 485                *nbuffers = (count < vq->num_buffers) ? 0 :
 486                        count - vq->num_buffers;
 487        else
 488                *nbuffers = count;
 489
 490        *nplanes = 1;
 491        sizes[0] = pix->sizeimage;
 492
 493        return 0;
 494}
 495
 496static int capture_buf_init(struct vb2_buffer *vb)
 497{
 498        struct imx_media_buffer *buf = to_imx_media_vb(vb);
 499
 500        INIT_LIST_HEAD(&buf->list);
 501
 502        return 0;
 503}
 504
 505static int capture_buf_prepare(struct vb2_buffer *vb)
 506{
 507        struct vb2_queue *vq = vb->vb2_queue;
 508        struct capture_priv *priv = vb2_get_drv_priv(vq);
 509        struct v4l2_pix_format *pix = &priv->vdev.fmt.fmt.pix;
 510
 511        if (vb2_plane_size(vb, 0) < pix->sizeimage) {
 512                v4l2_err(priv->src_sd,
 513                         "data will not fit into plane (%lu < %lu)\n",
 514                         vb2_plane_size(vb, 0), (long)pix->sizeimage);
 515                return -EINVAL;
 516        }
 517
 518        vb2_set_plane_payload(vb, 0, pix->sizeimage);
 519
 520        return 0;
 521}
 522
 523static void capture_buf_queue(struct vb2_buffer *vb)
 524{
 525        struct capture_priv *priv = vb2_get_drv_priv(vb->vb2_queue);
 526        struct imx_media_buffer *buf = to_imx_media_vb(vb);
 527        unsigned long flags;
 528
 529        spin_lock_irqsave(&priv->q_lock, flags);
 530
 531        list_add_tail(&buf->list, &priv->ready_q);
 532
 533        spin_unlock_irqrestore(&priv->q_lock, flags);
 534}
 535
 536static int capture_validate_fmt(struct capture_priv *priv)
 537{
 538        struct v4l2_subdev_format fmt_src;
 539        const struct imx_media_pixfmt *cc;
 540        struct v4l2_rect compose;
 541        struct v4l2_format f;
 542        int ret;
 543
 544        fmt_src.pad = priv->src_sd_pad;
 545        fmt_src.which = V4L2_SUBDEV_FORMAT_ACTIVE;
 546        ret = v4l2_subdev_call(priv->src_sd, pad, get_fmt, NULL, &fmt_src);
 547        if (ret)
 548                return ret;
 549
 550        v4l2_fill_pix_format(&f.fmt.pix, &fmt_src.format);
 551
 552        ret = __capture_try_fmt_vid_cap(priv, &fmt_src, &f, &cc, &compose);
 553        if (ret)
 554                return ret;
 555
 556        return (priv->vdev.fmt.fmt.pix.width != f.fmt.pix.width ||
 557                priv->vdev.fmt.fmt.pix.height != f.fmt.pix.height ||
 558                priv->vdev.cc->cs != cc->cs ||
 559                priv->vdev.compose.width != compose.width ||
 560                priv->vdev.compose.height != compose.height) ? -EINVAL : 0;
 561}
 562
 563static int capture_start_streaming(struct vb2_queue *vq, unsigned int count)
 564{
 565        struct capture_priv *priv = vb2_get_drv_priv(vq);
 566        struct imx_media_buffer *buf, *tmp;
 567        unsigned long flags;
 568        int ret;
 569
 570        ret = capture_validate_fmt(priv);
 571        if (ret) {
 572                v4l2_err(priv->src_sd, "capture format not valid\n");
 573                goto return_bufs;
 574        }
 575
 576        ret = imx_media_pipeline_set_stream(priv->md, &priv->src_sd->entity,
 577                                            true);
 578        if (ret) {
 579                v4l2_err(priv->src_sd, "pipeline start failed with %d\n", ret);
 580                goto return_bufs;
 581        }
 582
 583        priv->stop = false;
 584
 585        return 0;
 586
 587return_bufs:
 588        spin_lock_irqsave(&priv->q_lock, flags);
 589        list_for_each_entry_safe(buf, tmp, &priv->ready_q, list) {
 590                list_del(&buf->list);
 591                vb2_buffer_done(&buf->vbuf.vb2_buf, VB2_BUF_STATE_QUEUED);
 592        }
 593        spin_unlock_irqrestore(&priv->q_lock, flags);
 594        return ret;
 595}
 596
 597static void capture_stop_streaming(struct vb2_queue *vq)
 598{
 599        struct capture_priv *priv = vb2_get_drv_priv(vq);
 600        struct imx_media_buffer *frame;
 601        struct imx_media_buffer *tmp;
 602        unsigned long flags;
 603        int ret;
 604
 605        spin_lock_irqsave(&priv->q_lock, flags);
 606        priv->stop = true;
 607        spin_unlock_irqrestore(&priv->q_lock, flags);
 608
 609        ret = imx_media_pipeline_set_stream(priv->md, &priv->src_sd->entity,
 610                                            false);
 611        if (ret)
 612                v4l2_warn(priv->src_sd, "pipeline stop failed with %d\n", ret);
 613
 614        /* release all active buffers */
 615        spin_lock_irqsave(&priv->q_lock, flags);
 616        list_for_each_entry_safe(frame, tmp, &priv->ready_q, list) {
 617                list_del(&frame->list);
 618                vb2_buffer_done(&frame->vbuf.vb2_buf, VB2_BUF_STATE_ERROR);
 619        }
 620        spin_unlock_irqrestore(&priv->q_lock, flags);
 621}
 622
 623static const struct vb2_ops capture_qops = {
 624        .queue_setup     = capture_queue_setup,
 625        .buf_init        = capture_buf_init,
 626        .buf_prepare     = capture_buf_prepare,
 627        .buf_queue       = capture_buf_queue,
 628        .wait_prepare    = vb2_ops_wait_prepare,
 629        .wait_finish     = vb2_ops_wait_finish,
 630        .start_streaming = capture_start_streaming,
 631        .stop_streaming  = capture_stop_streaming,
 632};
 633
 634/*
 635 * File operations
 636 */
 637static int capture_open(struct file *file)
 638{
 639        struct capture_priv *priv = video_drvdata(file);
 640        struct video_device *vfd = priv->vdev.vfd;
 641        int ret;
 642
 643        if (mutex_lock_interruptible(&priv->mutex))
 644                return -ERESTARTSYS;
 645
 646        ret = v4l2_fh_open(file);
 647        if (ret)
 648                v4l2_err(priv->src_sd, "v4l2_fh_open failed\n");
 649
 650        ret = v4l2_pipeline_pm_get(&vfd->entity);
 651        if (ret)
 652                v4l2_fh_release(file);
 653
 654        mutex_unlock(&priv->mutex);
 655        return ret;
 656}
 657
 658static int capture_release(struct file *file)
 659{
 660        struct capture_priv *priv = video_drvdata(file);
 661        struct video_device *vfd = priv->vdev.vfd;
 662        struct vb2_queue *vq = &priv->q;
 663
 664        mutex_lock(&priv->mutex);
 665
 666        if (file->private_data == vq->owner) {
 667                vb2_queue_release(vq);
 668                vq->owner = NULL;
 669        }
 670
 671        v4l2_pipeline_pm_put(&vfd->entity);
 672
 673        v4l2_fh_release(file);
 674        mutex_unlock(&priv->mutex);
 675        return 0;
 676}
 677
 678static const struct v4l2_file_operations capture_fops = {
 679        .owner          = THIS_MODULE,
 680        .open           = capture_open,
 681        .release        = capture_release,
 682        .poll           = vb2_fop_poll,
 683        .unlocked_ioctl = video_ioctl2,
 684        .mmap           = vb2_fop_mmap,
 685};
 686
 687static struct video_device capture_videodev = {
 688        .fops           = &capture_fops,
 689        .ioctl_ops      = &capture_ioctl_ops,
 690        .minor          = -1,
 691        .release        = video_device_release,
 692        .vfl_dir        = VFL_DIR_RX,
 693        .tvnorms        = V4L2_STD_NTSC | V4L2_STD_PAL | V4L2_STD_SECAM,
 694        .device_caps    = V4L2_CAP_VIDEO_CAPTURE | V4L2_CAP_STREAMING,
 695};
 696
 697struct imx_media_buffer *
 698imx_media_capture_device_next_buf(struct imx_media_video_dev *vdev)
 699{
 700        struct capture_priv *priv = to_capture_priv(vdev);
 701        struct imx_media_buffer *buf = NULL;
 702        unsigned long flags;
 703
 704        spin_lock_irqsave(&priv->q_lock, flags);
 705
 706        /* get next queued buffer */
 707        if (!list_empty(&priv->ready_q)) {
 708                buf = list_entry(priv->ready_q.next, struct imx_media_buffer,
 709                                 list);
 710                list_del(&buf->list);
 711        }
 712
 713        spin_unlock_irqrestore(&priv->q_lock, flags);
 714
 715        return buf;
 716}
 717EXPORT_SYMBOL_GPL(imx_media_capture_device_next_buf);
 718
 719void imx_media_capture_device_error(struct imx_media_video_dev *vdev)
 720{
 721        struct capture_priv *priv = to_capture_priv(vdev);
 722        struct vb2_queue *vq = &priv->q;
 723        unsigned long flags;
 724
 725        if (!vb2_is_streaming(vq))
 726                return;
 727
 728        spin_lock_irqsave(&priv->q_lock, flags);
 729        vb2_queue_error(vq);
 730        spin_unlock_irqrestore(&priv->q_lock, flags);
 731}
 732EXPORT_SYMBOL_GPL(imx_media_capture_device_error);
 733
 734int imx_media_capture_device_register(struct imx_media_video_dev *vdev)
 735{
 736        struct capture_priv *priv = to_capture_priv(vdev);
 737        struct v4l2_subdev *sd = priv->src_sd;
 738        struct v4l2_device *v4l2_dev = sd->v4l2_dev;
 739        struct video_device *vfd = vdev->vfd;
 740        struct vb2_queue *vq = &priv->q;
 741        struct v4l2_subdev_format fmt_src;
 742        int ret;
 743
 744        /* get media device */
 745        priv->md = container_of(v4l2_dev->mdev, struct imx_media_dev, md);
 746
 747        vfd->v4l2_dev = v4l2_dev;
 748
 749        ret = video_register_device(vfd, VFL_TYPE_VIDEO, -1);
 750        if (ret) {
 751                v4l2_err(sd, "Failed to register video device\n");
 752                return ret;
 753        }
 754
 755        vq->type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
 756        vq->io_modes = VB2_MMAP | VB2_DMABUF;
 757        vq->drv_priv = priv;
 758        vq->buf_struct_size = sizeof(struct imx_media_buffer);
 759        vq->ops = &capture_qops;
 760        vq->mem_ops = &vb2_dma_contig_memops;
 761        vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_MONOTONIC;
 762        vq->lock = &priv->mutex;
 763        vq->min_buffers_needed = 2;
 764        vq->dev = priv->dev;
 765
 766        ret = vb2_queue_init(vq);
 767        if (ret) {
 768                v4l2_err(sd, "vb2_queue_init failed\n");
 769                goto unreg;
 770        }
 771
 772        INIT_LIST_HEAD(&priv->ready_q);
 773
 774        /* create the link from the src_sd devnode pad to device node */
 775        ret = media_create_pad_link(&sd->entity, priv->src_sd_pad,
 776                                    &vfd->entity, 0, 0);
 777        if (ret) {
 778                v4l2_err(sd, "failed to create link to device node\n");
 779                goto unreg;
 780        }
 781
 782        /* setup default format */
 783        fmt_src.pad = priv->src_sd_pad;
 784        fmt_src.which = V4L2_SUBDEV_FORMAT_ACTIVE;
 785        ret = v4l2_subdev_call(sd, pad, get_fmt, NULL, &fmt_src);
 786        if (ret) {
 787                v4l2_err(sd, "failed to get src_sd format\n");
 788                goto unreg;
 789        }
 790
 791        vdev->fmt.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
 792        imx_media_mbus_fmt_to_pix_fmt(&vdev->fmt.fmt.pix,
 793                                      &fmt_src.format, NULL);
 794        vdev->compose.width = fmt_src.format.width;
 795        vdev->compose.height = fmt_src.format.height;
 796        vdev->cc = imx_media_find_pixel_format(vdev->fmt.fmt.pix.pixelformat,
 797                                               PIXFMT_SEL_ANY);
 798
 799        v4l2_info(sd, "Registered %s as /dev/%s\n", vfd->name,
 800                  video_device_node_name(vfd));
 801
 802        vfd->ctrl_handler = &priv->ctrl_hdlr;
 803
 804        /* add vdev to the video device list */
 805        imx_media_add_video_device(priv->md, vdev);
 806
 807        return 0;
 808unreg:
 809        video_unregister_device(vfd);
 810        return ret;
 811}
 812EXPORT_SYMBOL_GPL(imx_media_capture_device_register);
 813
 814void imx_media_capture_device_unregister(struct imx_media_video_dev *vdev)
 815{
 816        struct capture_priv *priv = to_capture_priv(vdev);
 817        struct video_device *vfd = priv->vdev.vfd;
 818
 819        mutex_lock(&priv->mutex);
 820
 821        if (video_is_registered(vfd)) {
 822                video_unregister_device(vfd);
 823                media_entity_cleanup(&vfd->entity);
 824        }
 825
 826        mutex_unlock(&priv->mutex);
 827}
 828EXPORT_SYMBOL_GPL(imx_media_capture_device_unregister);
 829
 830struct imx_media_video_dev *
 831imx_media_capture_device_init(struct device *dev, struct v4l2_subdev *src_sd,
 832                              int pad)
 833{
 834        struct capture_priv *priv;
 835        struct video_device *vfd;
 836        int ret;
 837
 838        priv = devm_kzalloc(dev, sizeof(*priv), GFP_KERNEL);
 839        if (!priv)
 840                return ERR_PTR(-ENOMEM);
 841
 842        priv->src_sd = src_sd;
 843        priv->src_sd_pad = pad;
 844        priv->dev = dev;
 845
 846        mutex_init(&priv->mutex);
 847        spin_lock_init(&priv->q_lock);
 848
 849        snprintf(capture_videodev.name, sizeof(capture_videodev.name),
 850                 "%s capture", src_sd->name);
 851
 852        vfd = video_device_alloc();
 853        if (!vfd)
 854                return ERR_PTR(-ENOMEM);
 855
 856        *vfd = capture_videodev;
 857        vfd->lock = &priv->mutex;
 858        vfd->queue = &priv->q;
 859        priv->vdev.vfd = vfd;
 860
 861        priv->vdev_pad.flags = MEDIA_PAD_FL_SINK;
 862        ret = media_entity_pads_init(&vfd->entity, 1, &priv->vdev_pad);
 863        if (ret) {
 864                video_device_release(vfd);
 865                return ERR_PTR(ret);
 866        }
 867
 868        INIT_LIST_HEAD(&priv->vdev.list);
 869
 870        video_set_drvdata(vfd, priv);
 871
 872        v4l2_ctrl_handler_init(&priv->ctrl_hdlr, 0);
 873
 874        return &priv->vdev;
 875}
 876EXPORT_SYMBOL_GPL(imx_media_capture_device_init);
 877
 878void imx_media_capture_device_remove(struct imx_media_video_dev *vdev)
 879{
 880        struct capture_priv *priv = to_capture_priv(vdev);
 881
 882        v4l2_ctrl_handler_free(&priv->ctrl_hdlr);
 883}
 884EXPORT_SYMBOL_GPL(imx_media_capture_device_remove);
 885
 886MODULE_DESCRIPTION("i.MX5/6 v4l2 video capture interface driver");
 887MODULE_AUTHOR("Steve Longerbeam <steve_longerbeam@mentor.com>");
 888MODULE_LICENSE("GPL");
 889