drivers/media/platform/coda/coda-common.c

   1 /*
   2  * Coda multi-standard codec IP
   3  *
   4  * Copyright (C) 2012 Vista Silicon S.L.
   5  *    Javier Martin, <javier.martin@vista-silicon.com>
   6  *    Xavier Duret
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  */
  13
  14 #include <linux/clk.h>
  15 #include <linux/debugfs.h>
  16 #include <linux/delay.h>
  17 #include <linux/firmware.h>
  18 #include <linux/gcd.h>
  19 #include <linux/genalloc.h>
  20 #include <linux/interrupt.h>
  21 #include <linux/io.h>
  22 #include <linux/irq.h>
  23 #include <linux/kfifo.h>
  24 #include <linux/module.h>
  25 #include <linux/of_device.h>
  26 #include <linux/platform_device.h>
  27 #include <linux/pm_runtime.h>
  28 #include <linux/slab.h>
  29 #include <linux/videodev2.h>
  30 #include <linux/of.h>
  31 #include <linux/platform_data/media/coda.h>
  32 #include <linux/reset.h>
  33
  34 #include <media/v4l2-ctrls.h>
  35 #include <media/v4l2-device.h>
  36 #include <media/v4l2-event.h>
  37 #include <media/v4l2-ioctl.h>
  38 #include <media/v4l2-mem2mem.h>
  39 #include <media/videobuf2-v4l2.h>
  40 #include <media/videobuf2-dma-contig.h>
  41 #include <media/videobuf2-vmalloc.h>
  42
  43 #include "coda.h"
  44 #include "imx-vdoa.h"
  45
  46 #define CODA_NAME               "coda"
  47
  48 #define CODADX6_MAX_INSTANCES   4
  49 #define CODA_MAX_FORMATS        4
  50
  51 #define CODA_ISRAM_SIZE (2048 * 2)
  52
  53 #define MIN_W 176
  54 #define MIN_H 144
  55
  56 #define S_ALIGN         1 /* multiple of 2 */
  57 #define W_ALIGN         1 /* multiple of 2 */
  58 #define H_ALIGN         1 /* multiple of 2 */
  59
  60 #define fh_to_ctx(__fh) container_of(__fh, struct coda_ctx, fh)
  61
  62 int coda_debug;
  63 module_param(coda_debug, int, 0644);
  64 MODULE_PARM_DESC(coda_debug, "Debug level (0-2)");
  65
  66 static int disable_tiling;
  67 module_param(disable_tiling, int, 0644);
  68 MODULE_PARM_DESC(disable_tiling, "Disable tiled frame buffers");
  69
  70 static int disable_vdoa;
  71 module_param(disable_vdoa, int, 0644);
  72 MODULE_PARM_DESC(disable_vdoa, "Disable Video Data Order Adapter tiled to raster-scan conversion");
  73
  74 static int enable_bwb = 0;
  75 module_param(enable_bwb, int, 0644);
  76 MODULE_PARM_DESC(enable_bwb, "Enable BWB unit, may crash on certain streams");
  77
  78 void coda_write(struct coda_dev *dev, u32 data, u32 reg)
  79 {
  80         v4l2_dbg(2, coda_debug, &dev->v4l2_dev,
  81                  "%s: data=0x%x, reg=0x%x\n", __func__, data, reg);
  82         writel(data, dev->regs_base + reg);
  83 }
  84
  85 unsigned int coda_read(struct coda_dev *dev, u32 reg)
  86 {
  87         u32 data;
  88
  89         data = readl(dev->regs_base + reg);
  90         v4l2_dbg(2, coda_debug, &dev->v4l2_dev,
  91                  "%s: data=0x%x, reg=0x%x\n", __func__, data, reg);
  92         return data;
  93 }
  94
  95 void coda_write_base(struct coda_ctx *ctx, struct coda_q_data *q_data,
  96                      struct vb2_v4l2_buffer *buf, unsigned int reg_y)
  97 {
  98         u32 base_y = vb2_dma_contig_plane_dma_addr(&buf->vb2_buf, 0);
  99         u32 base_cb, base_cr;
 100
 101         switch (q_data->fourcc) {
 102         case V4L2_PIX_FMT_YUYV:
 103                 /* Fallthrough: IN -H264-> CODA -NV12 MB-> VDOA -YUYV-> OUT */
 104         case V4L2_PIX_FMT_NV12:
 105         case V4L2_PIX_FMT_YUV420:
 106         default:
 107                 base_cb = base_y + q_data->bytesperline * q_data->height;
 108                 base_cr = base_cb + q_data->bytesperline * q_data->height / 4;
 109                 break;
 110         case V4L2_PIX_FMT_YVU420:
 111                 /* Switch Cb and Cr for YVU420 format */
 112                 base_cr = base_y + q_data->bytesperline * q_data->height;
 113                 base_cb = base_cr + q_data->bytesperline * q_data->height / 4;
 114                 break;
 115         case V4L2_PIX_FMT_YUV422P:
 116                 base_cb = base_y + q_data->bytesperline * q_data->height;
 117                 base_cr = base_cb + q_data->bytesperline * q_data->height / 2;
 118         }
 119
 120         coda_write(ctx->dev, base_y, reg_y);
 121         coda_write(ctx->dev, base_cb, reg_y + 4);
 122         coda_write(ctx->dev, base_cr, reg_y + 8);
 123 }
 124
 125 #define CODA_CODEC(mode, src_fourcc, dst_fourcc, max_w, max_h) \
 126         { mode, src_fourcc, dst_fourcc, max_w, max_h }
 127
 128 /*
 129  * Arrays of codecs supported by each given version of Coda:
 130  *  i.MX27 -> codadx6
 131  *  i.MX5x -> coda7
 132  *  i.MX6  -> coda960
 133  * Use V4L2_PIX_FMT_YUV420 as placeholder for all supported YUV 4:2:0 variants
 134  */
 135 static const struct coda_codec codadx6_codecs[] = {
 136         CODA_CODEC(CODADX6_MODE_ENCODE_H264, V4L2_PIX_FMT_YUV420, V4L2_PIX_FMT_H264,  720, 576),
 137         CODA_CODEC(CODADX6_MODE_ENCODE_MP4,  V4L2_PIX_FMT_YUV420, V4L2_PIX_FMT_MPEG4, 720, 576),
 138 };
 139
 140 static const struct coda_codec coda7_codecs[] = {
 141         CODA_CODEC(CODA7_MODE_ENCODE_H264, V4L2_PIX_FMT_YUV420, V4L2_PIX_FMT_H264,   1280, 720),
 142         CODA_CODEC(CODA7_MODE_ENCODE_MP4,  V4L2_PIX_FMT_YUV420, V4L2_PIX_FMT_MPEG4,  1280, 720),
 143         CODA_CODEC(CODA7_MODE_ENCODE_MJPG, V4L2_PIX_FMT_YUV420, V4L2_PIX_FMT_JPEG,   8192, 8192),
 144         CODA_CODEC(CODA7_MODE_DECODE_H264, V4L2_PIX_FMT_H264,   V4L2_PIX_FMT_YUV420, 1920, 1088),
 145         CODA_CODEC(CODA7_MODE_DECODE_MP2,  V4L2_PIX_FMT_MPEG2,  V4L2_PIX_FMT_YUV420, 1920, 1088),
 146         CODA_CODEC(CODA7_MODE_DECODE_MP4,  V4L2_PIX_FMT_MPEG4,  V4L2_PIX_FMT_YUV420, 1920, 1088),
 147         CODA_CODEC(CODA7_MODE_DECODE_MJPG, V4L2_PIX_FMT_JPEG,   V4L2_PIX_FMT_YUV420, 8192, 8192),
 148 };
 149
 150 static const struct coda_codec coda9_codecs[] = {
 151         CODA_CODEC(CODA9_MODE_ENCODE_H264, V4L2_PIX_FMT_YUV420, V4L2_PIX_FMT_H264,   1920, 1088),
 152         CODA_CODEC(CODA9_MODE_ENCODE_MP4,  V4L2_PIX_FMT_YUV420, V4L2_PIX_FMT_MPEG4,  1920, 1088),
 153         CODA_CODEC(CODA9_MODE_DECODE_H264, V4L2_PIX_FMT_H264,   V4L2_PIX_FMT_YUV420, 1920, 1088),
 154         CODA_CODEC(CODA9_MODE_DECODE_MP2,  V4L2_PIX_FMT_MPEG2,  V4L2_PIX_FMT_YUV420, 1920, 1088),
 155         CODA_CODEC(CODA9_MODE_DECODE_MP4,  V4L2_PIX_FMT_MPEG4,  V4L2_PIX_FMT_YUV420, 1920, 1088),
 156 };
 157
 158 struct coda_video_device {
 159         const char *name;
 160         enum coda_inst_type type;
 161         const struct coda_context_ops *ops;
 162         bool direct;
 163         u32 src_formats[CODA_MAX_FORMATS];
 164         u32 dst_formats[CODA_MAX_FORMATS];
 165 };
 166
 167 static const struct coda_video_device coda_bit_encoder = {
 168         .name = "coda-encoder",
 169         .type = CODA_INST_ENCODER,
 170         .ops = &coda_bit_encode_ops,
 171         .src_formats = {
 172                 V4L2_PIX_FMT_NV12,
 173                 V4L2_PIX_FMT_YUV420,
 174                 V4L2_PIX_FMT_YVU420,
 175         },
 176         .dst_formats = {
 177                 V4L2_PIX_FMT_H264,
 178                 V4L2_PIX_FMT_MPEG4,
 179         },
 180 };
 181
 182 static const struct coda_video_device coda_bit_jpeg_encoder = {
 183         .name = "coda-jpeg-encoder",
 184         .type = CODA_INST_ENCODER,
 185         .ops = &coda_bit_encode_ops,
 186         .src_formats = {
 187                 V4L2_PIX_FMT_NV12,
 188                 V4L2_PIX_FMT_YUV420,
 189                 V4L2_PIX_FMT_YVU420,
 190                 V4L2_PIX_FMT_YUV422P,
 191         },
 192         .dst_formats = {
 193                 V4L2_PIX_FMT_JPEG,
 194         },
 195 };
 196
 197 static const struct coda_video_device coda_bit_decoder = {
 198         .name = "coda-decoder",
 199         .type = CODA_INST_DECODER,
 200         .ops = &coda_bit_decode_ops,
 201         .src_formats = {
 202                 V4L2_PIX_FMT_H264,
 203                 V4L2_PIX_FMT_MPEG2,
 204                 V4L2_PIX_FMT_MPEG4,
 205         },
 206         .dst_formats = {
 207                 V4L2_PIX_FMT_NV12,
 208                 V4L2_PIX_FMT_YUV420,
 209                 V4L2_PIX_FMT_YVU420,
 210                 /*
 211                  * If V4L2_PIX_FMT_YUYV should be default,
 212                  * set_default_params() must be adjusted.
 213                  */
 214                 V4L2_PIX_FMT_YUYV,
 215         },
 216 };
 217
 218 static const struct coda_video_device coda_bit_jpeg_decoder = {
 219         .name = "coda-jpeg-decoder",
 220         .type = CODA_INST_DECODER,
 221         .ops = &coda_bit_decode_ops,
 222         .src_formats = {
 223                 V4L2_PIX_FMT_JPEG,
 224         },
 225         .dst_formats = {
 226                 V4L2_PIX_FMT_NV12,
 227                 V4L2_PIX_FMT_YUV420,
 228                 V4L2_PIX_FMT_YVU420,
 229                 V4L2_PIX_FMT_YUV422P,
 230         },
 231 };
 232
 233 static const struct coda_video_device *codadx6_video_devices[] = {
 234         &coda_bit_encoder,
 235 };
 236
 237 static const struct coda_video_device *coda7_video_devices[] = {
 238         &coda_bit_jpeg_encoder,
 239         &coda_bit_jpeg_decoder,
 240         &coda_bit_encoder,
 241         &coda_bit_decoder,
 242 };
 243
 244 static const struct coda_video_device *coda9_video_devices[] = {
 245         &coda_bit_encoder,
 246         &coda_bit_decoder,
 247 };
 248
 249 /*
 250  * Normalize all supported YUV 4:2:0 formats to the value used in the codec
 251  * tables.
 252  */
 253 static u32 coda_format_normalize_yuv(u32 fourcc)
 254 {
 255         switch (fourcc) {
 256         case V4L2_PIX_FMT_NV12:
 257         case V4L2_PIX_FMT_YUV420:
 258         case V4L2_PIX_FMT_YVU420:
 259         case V4L2_PIX_FMT_YUV422P:
 260         case V4L2_PIX_FMT_YUYV:
 261                 return V4L2_PIX_FMT_YUV420;
 262         default:
 263                 return fourcc;
 264         }
 265 }
 266
 267 static const struct coda_codec *coda_find_codec(struct coda_dev *dev,
 268                                                 int src_fourcc, int dst_fourcc)
 269 {
 270         const struct coda_codec *codecs = dev->devtype->codecs;
 271         int num_codecs = dev->devtype->num_codecs;
 272         int k;
 273
 274         src_fourcc = coda_format_normalize_yuv(src_fourcc);
 275         dst_fourcc = coda_format_normalize_yuv(dst_fourcc);
 276         if (src_fourcc == dst_fourcc)
 277                 return NULL;
 278
 279         for (k = 0; k < num_codecs; k++) {
 280                 if (codecs[k].src_fourcc == src_fourcc &&
 281                     codecs[k].dst_fourcc == dst_fourcc)
 282                         break;
 283         }
 284
 285         if (k == num_codecs)
 286                 return NULL;
 287
 288         return &codecs[k];
 289 }
 290
 291 static void coda_get_max_dimensions(struct coda_dev *dev,
 292                                     const struct coda_codec *codec,
 293                                     int *max_w, int *max_h)
 294 {
 295         const struct coda_codec *codecs = dev->devtype->codecs;
 296         int num_codecs = dev->devtype->num_codecs;
 297         unsigned int w, h;
 298         int k;
 299
 300         if (codec) {
 301                 w = codec->max_w;
 302                 h = codec->max_h;
 303         } else {
 304                 for (k = 0, w = 0, h = 0; k < num_codecs; k++) {
 305                         w = max(w, codecs[k].max_w);
 306                         h = max(h, codecs[k].max_h);
 307                 }
 308         }
 309
 310         if (max_w)
 311                 *max_w = w;
 312         if (max_h)
 313                 *max_h = h;
 314 }
 315
 316 static const struct coda_video_device *to_coda_video_device(struct video_device
 317                                                             *vdev)
 318 {
 319         struct coda_dev *dev = video_get_drvdata(vdev);
 320         unsigned int i = vdev - dev->vfd;
 321
 322         if (i >= dev->devtype->num_vdevs)
 323                 return NULL;
 324
 325         return dev->devtype->vdevs[i];
 326 }
 327
 328 const char *coda_product_name(int product)
 329 {
 330         static char buf[9];
 331
 332         switch (product) {
 333         case CODA_DX6:
 334                 return "CodaDx6";
 335         case CODA_7541:
 336                 return "CODA7541";
 337         case CODA_960:
 338                 return "CODA960";
 339         default:
 340                 snprintf(buf, sizeof(buf), "(0x%04x)", product);
 341                 return buf;
 342         }
 343 }
 344
 345 static struct vdoa_data *coda_get_vdoa_data(void)
 346 {
 347         struct device_node *vdoa_node;
 348         struct platform_device *vdoa_pdev;
 349         struct vdoa_data *vdoa_data = NULL;
 350
 351         vdoa_node = of_find_compatible_node(NULL, NULL, "fsl,imx6q-vdoa");
 352         if (!vdoa_node)
 353                 return NULL;
 354
 355         vdoa_pdev = of_find_device_by_node(vdoa_node);
 356         if (!vdoa_pdev)
 357                 goto out;
 358
 359         vdoa_data = platform_get_drvdata(vdoa_pdev);
 360         if (!vdoa_data)
 361                 vdoa_data = ERR_PTR(-EPROBE_DEFER);
 362
 363 out:
 364         if (vdoa_node)
 365                 of_node_put(vdoa_node);
 366
 367         return vdoa_data;
 368 }
 369
 370 /*
 371  * V4L2 ioctl() operations.
 372  */
 373 static int coda_querycap(struct file *file, void *priv,
 374                          struct v4l2_capability *cap)
 375 {
 376         struct coda_ctx *ctx = fh_to_ctx(priv);
 377
 378         strlcpy(cap->driver, CODA_NAME, sizeof(cap->driver));
 379         strlcpy(cap->card, coda_product_name(ctx->dev->devtype->product),
 380                 sizeof(cap->card));
 381         strlcpy(cap->bus_info, "platform:" CODA_NAME, sizeof(cap->bus_info));
 382         cap->device_caps = V4L2_CAP_VIDEO_M2M | V4L2_CAP_STREAMING;
 383         cap->capabilities = cap->device_caps | V4L2_CAP_DEVICE_CAPS;
 384
 385         return 0;
 386 }
 387
 388 static int coda_enum_fmt(struct file *file, void *priv,
 389                          struct v4l2_fmtdesc *f)
 390 {
 391         struct video_device *vdev = video_devdata(file);
 392         const struct coda_video_device *cvd = to_coda_video_device(vdev);
 393         struct coda_ctx *ctx = fh_to_ctx(priv);
 394         const u32 *formats;
 395
 396         if (f->type == V4L2_BUF_TYPE_VIDEO_OUTPUT)
 397                 formats = cvd->src_formats;
 398         else if (f->type == V4L2_BUF_TYPE_VIDEO_CAPTURE)
 399                 formats = cvd->dst_formats;
 400         else
 401                 return -EINVAL;
 402
 403         if (f->index >= CODA_MAX_FORMATS || formats[f->index] == 0)
 404                 return -EINVAL;
 405
 406         /* Skip YUYV if the vdoa is not available */
 407         if (!ctx->vdoa && f->type == V4L2_BUF_TYPE_VIDEO_CAPTURE &&
 408             formats[f->index] == V4L2_PIX_FMT_YUYV)
 409                 return -EINVAL;
 410
 411         f->pixelformat = formats[f->index];
 412
 413         return 0;
 414 }
 415
 416 static int coda_g_fmt(struct file *file, void *priv,
 417                       struct v4l2_format *f)
 418 {
 419         struct coda_q_data *q_data;
 420         struct coda_ctx *ctx = fh_to_ctx(priv);
 421
 422         q_data = get_q_data(ctx, f->type);
 423         if (!q_data)
 424                 return -EINVAL;
 425
 426         f->fmt.pix.field        = V4L2_FIELD_NONE;
 427         f->fmt.pix.pixelformat  = q_data->fourcc;
 428         f->fmt.pix.width        = q_data->width;
 429         f->fmt.pix.height       = q_data->height;
 430         f->fmt.pix.bytesperline = q_data->bytesperline;
 431
 432         f->fmt.pix.sizeimage    = q_data->sizeimage;
 433         f->fmt.pix.colorspace   = ctx->colorspace;
 434         f->fmt.pix.xfer_func    = ctx->xfer_func;
 435         f->fmt.pix.ycbcr_enc    = ctx->ycbcr_enc;
 436         f->fmt.pix.quantization = ctx->quantization;
 437
 438         return 0;
 439 }
 440
 441 static int coda_try_pixelformat(struct coda_ctx *ctx, struct v4l2_format *f)
 442 {
 443         struct coda_q_data *q_data;
 444         const u32 *formats;
 445         int i;
 446
 447         if (f->type == V4L2_BUF_TYPE_VIDEO_OUTPUT)
 448                 formats = ctx->cvd->src_formats;
 449         else if (f->type == V4L2_BUF_TYPE_VIDEO_CAPTURE)
 450                 formats = ctx->cvd->dst_formats;
 451         else
 452                 return -EINVAL;
 453
 454         for (i = 0; i < CODA_MAX_FORMATS; i++) {
 455                 /* Skip YUYV if the vdoa is not available */
 456                 if (!ctx->vdoa && f->type == V4L2_BUF_TYPE_VIDEO_CAPTURE &&
 457                     formats[i] == V4L2_PIX_FMT_YUYV)
 458                         continue;
 459
 460                 if (formats[i] == f->fmt.pix.pixelformat) {
 461                         f->fmt.pix.pixelformat = formats[i];
 462                         return 0;
 463                 }
 464         }
 465
 466         /* Fall back to currently set pixelformat */
 467         q_data = get_q_data(ctx, f->type);
 468         f->fmt.pix.pixelformat = q_data->fourcc;
 469
 470         return 0;
 471 }
 472
 473 static int coda_try_fmt_vdoa(struct coda_ctx *ctx, struct v4l2_format *f,
 474                              bool *use_vdoa)
 475 {
 476         int err;
 477
 478         if (f->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
 479                 return -EINVAL;
 480
 481         if (!use_vdoa)
 482                 return -EINVAL;
 483
 484         if (!ctx->vdoa) {
 485                 *use_vdoa = false;
 486                 return 0;
 487         }
 488
 489         err = vdoa_context_configure(NULL, f->fmt.pix.width, f->fmt.pix.height,
 490                                      f->fmt.pix.pixelformat);
 491         if (err) {
 492                 *use_vdoa = false;
 493                 return 0;
 494         }
 495
 496         *use_vdoa = true;
 497         return 0;
 498 }
 499
 500 static unsigned int coda_estimate_sizeimage(struct coda_ctx *ctx, u32 sizeimage,
 501                                             u32 width, u32 height)
 502 {
 503         /*
 504          * This is a rough estimate for sensible compressed buffer
 505          * sizes (between 1 and 16 bits per pixel). This could be
 506          * improved by better format specific worst case estimates.
 507          */
 508         return round_up(clamp(sizeimage, width * height / 8,
 509                                          width * height * 2), PAGE_SIZE);
 510 }
 511
 512 static int coda_try_fmt(struct coda_ctx *ctx, const struct coda_codec *codec,
 513                         struct v4l2_format *f)
 514 {
 515         struct coda_dev *dev = ctx->dev;
 516         unsigned int max_w, max_h;
 517         enum v4l2_field field;
 518
 519         field = f->fmt.pix.field;
 520         if (field == V4L2_FIELD_ANY)
 521                 field = V4L2_FIELD_NONE;
 522         else if (V4L2_FIELD_NONE != field)
 523                 return -EINVAL;
 524
 525         /* V4L2 specification suggests the driver corrects the format struct
 526          * if any of the dimensions is unsupported */
 527         f->fmt.pix.field = field;
 528
 529         coda_get_max_dimensions(dev, codec, &max_w, &max_h);
 530         v4l_bound_align_image(&f->fmt.pix.width, MIN_W, max_w, W_ALIGN,
 531                               &f->fmt.pix.height, MIN_H, max_h, H_ALIGN,
 532                               S_ALIGN);
 533
 534         switch (f->fmt.pix.pixelformat) {
 535         case V4L2_PIX_FMT_NV12:
 536         case V4L2_PIX_FMT_YUV420:
 537         case V4L2_PIX_FMT_YVU420:
 538                 /*
 539                  * Frame stride must be at least multiple of 8,
 540                  * but multiple of 16 for h.264 or JPEG 4:2:x
 541                  */
 542                 f->fmt.pix.bytesperline = round_up(f->fmt.pix.width, 16);
 543                 f->fmt.pix.sizeimage = f->fmt.pix.bytesperline *
 544                                         f->fmt.pix.height * 3 / 2;
 545                 break;
 546         case V4L2_PIX_FMT_YUYV:
 547                 f->fmt.pix.bytesperline = round_up(f->fmt.pix.width, 16) * 2;
 548                 f->fmt.pix.sizeimage = f->fmt.pix.bytesperline *
 549                                         f->fmt.pix.height;
 550                 break;
 551         case V4L2_PIX_FMT_YUV422P:
 552                 f->fmt.pix.bytesperline = round_up(f->fmt.pix.width, 16);
 553                 f->fmt.pix.sizeimage = f->fmt.pix.bytesperline *
 554                                         f->fmt.pix.height * 2;
 555                 break;
 556         case V4L2_PIX_FMT_JPEG:
 557                 f->fmt.pix.colorspace = V4L2_COLORSPACE_JPEG;
 558                 /* fallthrough */
 559         case V4L2_PIX_FMT_H264:
 560         case V4L2_PIX_FMT_MPEG4:
 561         case V4L2_PIX_FMT_MPEG2:
 562                 f->fmt.pix.bytesperline = 0;
 563                 f->fmt.pix.sizeimage = coda_estimate_sizeimage(ctx,
 564                                                         f->fmt.pix.sizeimage,
 565                                                         f->fmt.pix.width,
 566                                                         f->fmt.pix.height);
 567                 break;
 568         default:
 569                 BUG();
 570         }
 571
 572         return 0;
 573 }
 574
 575 static int coda_try_fmt_vid_cap(struct file *file, void *priv,
 576                                 struct v4l2_format *f)
 577 {
 578         struct coda_ctx *ctx = fh_to_ctx(priv);
 579         const struct coda_q_data *q_data_src;
 580         const struct coda_codec *codec;
 581         struct vb2_queue *src_vq;
 582         int ret;
 583         bool use_vdoa;
 584
 585         ret = coda_try_pixelformat(ctx, f);
 586         if (ret < 0)
 587                 return ret;
 588
 589         q_data_src = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT);
 590
 591         /*
 592          * If the source format is already fixed, only allow the same output
 593          * resolution
 594          */
 595         src_vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT);
 596         if (vb2_is_streaming(src_vq)) {
 597                 f->fmt.pix.width = q_data_src->width;
 598                 f->fmt.pix.height = q_data_src->height;
 599         }
 600
 601         f->fmt.pix.colorspace = ctx->colorspace;
 602         f->fmt.pix.xfer_func = ctx->xfer_func;
 603         f->fmt.pix.ycbcr_enc = ctx->ycbcr_enc;
 604         f->fmt.pix.quantization = ctx->quantization;
 605
 606         q_data_src = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT);
 607         codec = coda_find_codec(ctx->dev, q_data_src->fourcc,
 608                                 f->fmt.pix.pixelformat);
 609         if (!codec)
 610                 return -EINVAL;
 611
 612         ret = coda_try_fmt(ctx, codec, f);
 613         if (ret < 0)
 614                 return ret;
 615
 616         /* The h.264 decoder only returns complete 16x16 macroblocks */
 617         if (codec && codec->src_fourcc == V4L2_PIX_FMT_H264) {
 618                 f->fmt.pix.height = round_up(f->fmt.pix.height, 16);
 619                 f->fmt.pix.bytesperline = round_up(f->fmt.pix.width, 16);
 620                 f->fmt.pix.sizeimage = f->fmt.pix.bytesperline *
 621                                        f->fmt.pix.height * 3 / 2;
 622
 623                 ret = coda_try_fmt_vdoa(ctx, f, &use_vdoa);
 624                 if (ret < 0)
 625                         return ret;
 626
 627                 if (f->fmt.pix.pixelformat == V4L2_PIX_FMT_YUYV) {
 628                         if (!use_vdoa)
 629                                 return -EINVAL;
 630
 631                         f->fmt.pix.bytesperline = round_up(f->fmt.pix.width, 16) * 2;
 632                         f->fmt.pix.sizeimage = f->fmt.pix.bytesperline *
 633                                 f->fmt.pix.height;
 634                 }
 635         }
 636
 637         return 0;
 638 }
 639
 640 static void coda_set_default_colorspace(struct v4l2_pix_format *fmt)
 641 {
 642         enum v4l2_colorspace colorspace;
 643
 644         if (fmt->pixelformat == V4L2_PIX_FMT_JPEG)
 645                 colorspace = V4L2_COLORSPACE_JPEG;
 646         else if (fmt->width <= 720 && fmt->height <= 576)
 647                 colorspace = V4L2_COLORSPACE_SMPTE170M;
 648         else
 649                 colorspace = V4L2_COLORSPACE_REC709;
 650
 651         fmt->colorspace = colorspace;
 652         fmt->xfer_func = V4L2_XFER_FUNC_DEFAULT;
 653         fmt->ycbcr_enc = V4L2_YCBCR_ENC_DEFAULT;
 654         fmt->quantization = V4L2_QUANTIZATION_DEFAULT;
 655 }
 656
 657 static int coda_try_fmt_vid_out(struct file *file, void *priv,
 658                                 struct v4l2_format *f)
 659 {
 660         struct coda_ctx *ctx = fh_to_ctx(priv);
 661         struct coda_dev *dev = ctx->dev;
 662         const struct coda_q_data *q_data_dst;
 663         const struct coda_codec *codec;
 664         int ret;
 665
 666         ret = coda_try_pixelformat(ctx, f);
 667         if (ret < 0)
 668                 return ret;
 669
 670         if (f->fmt.pix.colorspace == V4L2_COLORSPACE_DEFAULT)
 671                 coda_set_default_colorspace(&f->fmt.pix);
 672
 673         q_data_dst = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE);
 674         codec = coda_find_codec(dev, f->fmt.pix.pixelformat, q_data_dst->fourcc);
 675
 676         return coda_try_fmt(ctx, codec, f);
 677 }
 678
 679 static int coda_s_fmt(struct coda_ctx *ctx, struct v4l2_format *f,
 680                       struct v4l2_rect *r)
 681 {
 682         struct coda_q_data *q_data;
 683         struct vb2_queue *vq;
 684
 685         vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, f->type);
 686         if (!vq)
 687                 return -EINVAL;
 688
 689         q_data = get_q_data(ctx, f->type);
 690         if (!q_data)
 691                 return -EINVAL;
 692
 693         if (vb2_is_busy(vq)) {
 694                 v4l2_err(&ctx->dev->v4l2_dev, "%s queue busy\n", __func__);
 695                 return -EBUSY;
 696         }
 697
 698         q_data->fourcc = f->fmt.pix.pixelformat;
 699         q_data->width = f->fmt.pix.width;
 700         q_data->height = f->fmt.pix.height;
 701         q_data->bytesperline = f->fmt.pix.bytesperline;
 702         q_data->sizeimage = f->fmt.pix.sizeimage;
 703         if (r) {
 704                 q_data->rect = *r;
 705         } else {
 706                 q_data->rect.left = 0;
 707                 q_data->rect.top = 0;
 708                 q_data->rect.width = f->fmt.pix.width;
 709                 q_data->rect.height = f->fmt.pix.height;
 710         }
 711
 712         switch (f->fmt.pix.pixelformat) {
 713         case V4L2_PIX_FMT_YUYV:
 714                 ctx->tiled_map_type = GDI_TILED_FRAME_MB_RASTER_MAP;
 715                 break;
 716         case V4L2_PIX_FMT_NV12:
 717                 ctx->tiled_map_type = GDI_TILED_FRAME_MB_RASTER_MAP;
 718                 if (!disable_tiling)
 719                         break;
 720                 /* else fall through */
 721         case V4L2_PIX_FMT_YUV420:
 722         case V4L2_PIX_FMT_YVU420:
 723                 ctx->tiled_map_type = GDI_LINEAR_FRAME_MAP;
 724                 break;
 725         default:
 726                 break;
 727         }
 728
 729         if (ctx->tiled_map_type == GDI_TILED_FRAME_MB_RASTER_MAP &&
 730             !coda_try_fmt_vdoa(ctx, f, &ctx->use_vdoa) &&
 731             ctx->use_vdoa)
 732                 vdoa_context_configure(ctx->vdoa, f->fmt.pix.width,
 733                                        f->fmt.pix.height,
 734                                        f->fmt.pix.pixelformat);
 735         else
 736                 ctx->use_vdoa = false;
 737
 738         v4l2_dbg(1, coda_debug, &ctx->dev->v4l2_dev,
 739                 "Setting format for type %d, wxh: %dx%d, fmt: %4.4s %c\n",
 740                 f->type, q_data->width, q_data->height,
 741                 (char *)&q_data->fourcc,
 742                 (ctx->tiled_map_type == GDI_LINEAR_FRAME_MAP) ? 'L' : 'T');
 743
 744         return 0;
 745 }
 746
 747 static int coda_s_fmt_vid_cap(struct file *file, void *priv,
 748                               struct v4l2_format *f)
 749 {
 750         struct coda_ctx *ctx = fh_to_ctx(priv);
 751         struct coda_q_data *q_data_src;
 752         struct v4l2_rect r;
 753         int ret;
 754
 755         ret = coda_try_fmt_vid_cap(file, priv, f);
 756         if (ret)
 757                 return ret;
 758
 759         q_data_src = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT);
 760         r.left = 0;
 761         r.top = 0;
 762         r.width = q_data_src->width;
 763         r.height = q_data_src->height;
 764
 765         return coda_s_fmt(ctx, f, &r);
 766 }
 767
 768 static int coda_s_fmt_vid_out(struct file *file, void *priv,
 769                               struct v4l2_format *f)
 770 {
 771         struct coda_ctx *ctx = fh_to_ctx(priv);
 772         struct coda_q_data *q_data_src;
 773         struct v4l2_format f_cap;
 774         struct v4l2_rect r;
 775         int ret;
 776
 777         ret = coda_try_fmt_vid_out(file, priv, f);
 778         if (ret)
 779                 return ret;
 780
 781         ret = coda_s_fmt(ctx, f, NULL);
 782         if (ret)
 783                 return ret;
 784
 785         ctx->colorspace = f->fmt.pix.colorspace;
 786         ctx->xfer_func = f->fmt.pix.xfer_func;
 787         ctx->ycbcr_enc = f->fmt.pix.ycbcr_enc;
 788         ctx->quantization = f->fmt.pix.quantization;
 789
 790         memset(&f_cap, 0, sizeof(f_cap));
 791         f_cap.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
 792         coda_g_fmt(file, priv, &f_cap);
 793         f_cap.fmt.pix.width = f->fmt.pix.width;
 794         f_cap.fmt.pix.height = f->fmt.pix.height;
 795
 796         ret = coda_try_fmt_vid_cap(file, priv, &f_cap);
 797         if (ret)
 798                 return ret;
 799
 800         q_data_src = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT);
 801         r.left = 0;
 802         r.top = 0;
 803         r.width = q_data_src->width;
 804         r.height = q_data_src->height;
 805
 806         return coda_s_fmt(ctx, &f_cap, &r);
 807 }
 808
 809 static int coda_reqbufs(struct file *file, void *priv,
 810                         struct v4l2_requestbuffers *rb)
 811 {
 812         struct coda_ctx *ctx = fh_to_ctx(priv);
 813         int ret;
 814
 815         ret = v4l2_m2m_reqbufs(file, ctx->fh.m2m_ctx, rb);
 816         if (ret)
 817                 return ret;
 818
 819         /*
 820          * Allow to allocate instance specific per-context buffers, such as
 821          * bitstream ringbuffer, slice buffer, work buffer, etc. if needed.
 822          */
 823         if (rb->type == V4L2_BUF_TYPE_VIDEO_OUTPUT && ctx->ops->reqbufs)
 824                 return ctx->ops->reqbufs(ctx, rb);
 825
 826         return 0;
 827 }
 828
 829 static int coda_qbuf(struct file *file, void *priv,
 830                      struct v4l2_buffer *buf)
 831 {
 832         struct coda_ctx *ctx = fh_to_ctx(priv);
 833
 834         return v4l2_m2m_qbuf(file, ctx->fh.m2m_ctx, buf);
 835 }
 836
 837 static bool coda_buf_is_end_of_stream(struct coda_ctx *ctx,
 838                                       struct vb2_v4l2_buffer *buf)
 839 {
 840         return ((ctx->bit_stream_param & CODA_BIT_STREAM_END_FLAG) &&
 841                 (buf->sequence == (ctx->qsequence - 1)));
 842 }
 843
 844 void coda_m2m_buf_done(struct coda_ctx *ctx, struct vb2_v4l2_buffer *buf,
 845                        enum vb2_buffer_state state)
 846 {
 847         const struct v4l2_event eos_event = {
 848                 .type = V4L2_EVENT_EOS
 849         };
 850
 851         if (coda_buf_is_end_of_stream(ctx, buf)) {
 852                 buf->flags |= V4L2_BUF_FLAG_LAST;
 853
 854                 v4l2_event_queue_fh(&ctx->fh, &eos_event);
 855         }
 856
 857         v4l2_m2m_buf_done(buf, state);
 858 }
 859
 860 static int coda_g_selection(struct file *file, void *fh,
 861                             struct v4l2_selection *s)
 862 {
 863         struct coda_ctx *ctx = fh_to_ctx(fh);
 864         struct coda_q_data *q_data;
 865         struct v4l2_rect r, *rsel;
 866
 867         q_data = get_q_data(ctx, s->type);
 868         if (!q_data)
 869                 return -EINVAL;
 870
 871         r.left = 0;
 872         r.top = 0;
 873         r.width = q_data->width;
 874         r.height = q_data->height;
 875         rsel = &q_data->rect;
 876
 877         switch (s->target) {
 878         case V4L2_SEL_TGT_CROP_DEFAULT:
 879         case V4L2_SEL_TGT_CROP_BOUNDS:
 880                 rsel = &r;
 881                 /* fallthrough */
 882         case V4L2_SEL_TGT_CROP:
 883                 if (s->type != V4L2_BUF_TYPE_VIDEO_OUTPUT)
 884                         return -EINVAL;
 885                 break;
 886         case V4L2_SEL_TGT_COMPOSE_BOUNDS:
 887         case V4L2_SEL_TGT_COMPOSE_PADDED:
 888                 rsel = &r;
 889                 /* fallthrough */
 890         case V4L2_SEL_TGT_COMPOSE:
 891         case V4L2_SEL_TGT_COMPOSE_DEFAULT:
 892                 if (s->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
 893                         return -EINVAL;
 894                 break;
 895         default:
 896                 return -EINVAL;
 897         }
 898
 899         s->r = *rsel;
 900
 901         return 0;
 902 }
 903
 904 static int coda_try_encoder_cmd(struct file *file, void *fh,
 905                                 struct v4l2_encoder_cmd *ec)
 906 {
 907         if (ec->cmd != V4L2_ENC_CMD_STOP)
 908                 return -EINVAL;
 909
 910         if (ec->flags & V4L2_ENC_CMD_STOP_AT_GOP_END)
 911                 return -EINVAL;
 912
 913         return 0;
 914 }
 915
 916 static int coda_encoder_cmd(struct file *file, void *fh,
 917                             struct v4l2_encoder_cmd *ec)
 918 {
 919         struct coda_ctx *ctx = fh_to_ctx(fh);
 920         struct vb2_queue *dst_vq;
 921         int ret;
 922
 923         ret = coda_try_encoder_cmd(file, fh, ec);
 924         if (ret < 0)
 925                 return ret;
 926
 927         /* Ignore encoder stop command silently in decoder context */
 928         if (ctx->inst_type != CODA_INST_ENCODER)
 929                 return 0;
 930
 931         /* Set the stream-end flag on this context */
 932         ctx->bit_stream_param |= CODA_BIT_STREAM_END_FLAG;
 933
 934         /* If there is no buffer in flight, wake up */
 935         if (ctx->qsequence == ctx->osequence) {
 936                 dst_vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx,
 937                                          V4L2_BUF_TYPE_VIDEO_CAPTURE);
 938                 dst_vq->last_buffer_dequeued = true;
 939                 wake_up(&dst_vq->done_wq);
 940         }
 941
 942         return 0;
 943 }
 944
 945 static int coda_try_decoder_cmd(struct file *file, void *fh,
 946                                 struct v4l2_decoder_cmd *dc)
 947 {
 948         if (dc->cmd != V4L2_DEC_CMD_STOP)
 949                 return -EINVAL;
 950
 951         if (dc->flags & V4L2_DEC_CMD_STOP_TO_BLACK)
 952                 return -EINVAL;
 953
 954         if (!(dc->flags & V4L2_DEC_CMD_STOP_IMMEDIATELY) && (dc->stop.pts != 0))
 955                 return -EINVAL;
 956
 957         return 0;
 958 }
 959
 960 static int coda_decoder_cmd(struct file *file, void *fh,
 961                             struct v4l2_decoder_cmd *dc)
 962 {
 963         struct coda_ctx *ctx = fh_to_ctx(fh);
 964         int ret;
 965
 966         ret = coda_try_decoder_cmd(file, fh, dc);
 967         if (ret < 0)
 968                 return ret;
 969
 970         /* Ignore decoder stop command silently in encoder context */
 971         if (ctx->inst_type != CODA_INST_DECODER)
 972                 return 0;
 973
 974         /* Set the stream-end flag on this context */
 975         coda_bit_stream_end_flag(ctx);
 976         ctx->hold = false;
 977         v4l2_m2m_try_schedule(ctx->fh.m2m_ctx);
 978
 979         return 0;
 980 }
 981
 982 static int coda_g_parm(struct file *file, void *fh, struct v4l2_streamparm *a)
 983 {
 984         struct coda_ctx *ctx = fh_to_ctx(fh);
 985         struct v4l2_fract *tpf;
 986
 987         if (a->type != V4L2_BUF_TYPE_VIDEO_OUTPUT)
 988                 return -EINVAL;
 989
 990         a->parm.output.capability = V4L2_CAP_TIMEPERFRAME;
 991         tpf = &a->parm.output.timeperframe;
 992         tpf->denominator = ctx->params.framerate & CODA_FRATE_RES_MASK;
 993         tpf->numerator = 1 + (ctx->params.framerate >>
 994                               CODA_FRATE_DIV_OFFSET);
 995
 996         return 0;
 997 }
 998
 999 /*
1000  * Approximate timeperframe v4l2_fract with values that can be written
1001  * into the 16-bit CODA_FRATE_DIV and CODA_FRATE_RES fields.
1002  */
1003 static void coda_approximate_timeperframe(struct v4l2_fract *timeperframe)
1004 {
1005         struct v4l2_fract s = *timeperframe;
1006         struct v4l2_fract f0;
1007         struct v4l2_fract f1 = { 1, 0 };
1008         struct v4l2_fract f2 = { 0, 1 };
1009         unsigned int i, div, s_denominator;
1010
1011         /* Lower bound is 1/65535 */
1012         if (s.numerator == 0 || s.denominator / s.numerator > 65535) {
1013                 timeperframe->numerator = 1;
1014                 timeperframe->denominator = 65535;
1015                 return;
1016         }
1017
1018         /* Upper bound is 65536/1, map everything above to infinity */
1019         if (s.denominator == 0 || s.numerator / s.denominator > 65536) {
1020                 timeperframe->numerator = 1;
1021                 timeperframe->denominator = 0;
1022                 return;
1023         }
1024
1025         /* Reduce fraction to lowest terms */
1026         div = gcd(s.numerator, s.denominator);
1027         if (div > 1) {
1028                 s.numerator /= div;
1029                 s.denominator /= div;
1030         }
1031
1032         if (s.numerator <= 65536 && s.denominator < 65536) {
1033                 *timeperframe = s;
1034                 return;
1035         }
1036
1037         /* Find successive convergents from continued fraction expansion */
1038         while (f2.numerator <= 65536 && f2.denominator < 65536) {
1039                 f0 = f1;
1040                 f1 = f2;
1041
1042                 /* Stop when f2 exactly equals timeperframe */
1043                 if (s.numerator == 0)
1044                         break;
1045
1046                 i = s.denominator / s.numerator;
1047
1048                 f2.numerator = f0.numerator + i * f1.numerator;
1049                 f2.denominator = f0.denominator + i * f2.denominator;
1050
1051                 s_denominator = s.numerator;
1052                 s.numerator = s.denominator % s.numerator;
1053                 s.denominator = s_denominator;
1054         }
1055
1056         *timeperframe = f1;
1057 }
1058
1059 static uint32_t coda_timeperframe_to_frate(struct v4l2_fract *timeperframe)
1060 {
1061         return ((timeperframe->numerator - 1) << CODA_FRATE_DIV_OFFSET) |
1062                 timeperframe->denominator;
1063 }
1064
1065 static int coda_s_parm(struct file *file, void *fh, struct v4l2_streamparm *a)
1066 {
1067         struct coda_ctx *ctx = fh_to_ctx(fh);
1068         struct v4l2_fract *tpf;
1069
1070         if (a->type != V4L2_BUF_TYPE_VIDEO_OUTPUT)
1071                 return -EINVAL;
1072
1073         tpf = &a->parm.output.timeperframe;
1074         coda_approximate_timeperframe(tpf);
1075         ctx->params.framerate = coda_timeperframe_to_frate(tpf);
1076
1077         return 0;
1078 }
1079
1080 static int coda_subscribe_event(struct v4l2_fh *fh,
1081                                 const struct v4l2_event_subscription *sub)
1082 {
1083         switch (sub->type) {
1084         case V4L2_EVENT_EOS:
1085                 return v4l2_event_subscribe(fh, sub, 0, NULL);
1086         default:
1087                 return v4l2_ctrl_subscribe_event(fh, sub);
1088         }
1089 }
1090
1091 static const struct v4l2_ioctl_ops coda_ioctl_ops = {
1092         .vidioc_querycap        = coda_querycap,
1093
1094         .vidioc_enum_fmt_vid_cap = coda_enum_fmt,
1095         .vidioc_g_fmt_vid_cap   = coda_g_fmt,
1096         .vidioc_try_fmt_vid_cap = coda_try_fmt_vid_cap,
1097         .vidioc_s_fmt_vid_cap   = coda_s_fmt_vid_cap,
1098
1099         .vidioc_enum_fmt_vid_out = coda_enum_fmt,
1100         .vidioc_g_fmt_vid_out   = coda_g_fmt,
1101         .vidioc_try_fmt_vid_out = coda_try_fmt_vid_out,
1102         .vidioc_s_fmt_vid_out   = coda_s_fmt_vid_out,
1103
1104         .vidioc_reqbufs         = coda_reqbufs,
1105         .vidioc_querybuf        = v4l2_m2m_ioctl_querybuf,
1106
1107         .vidioc_qbuf            = coda_qbuf,
1108         .vidioc_expbuf          = v4l2_m2m_ioctl_expbuf,
1109         .vidioc_dqbuf           = v4l2_m2m_ioctl_dqbuf,
1110         .vidioc_create_bufs     = v4l2_m2m_ioctl_create_bufs,
1111         .vidioc_prepare_buf     = v4l2_m2m_ioctl_prepare_buf,
1112
1113         .vidioc_streamon        = v4l2_m2m_ioctl_streamon,
1114         .vidioc_streamoff       = v4l2_m2m_ioctl_streamoff,
1115
1116         .vidioc_g_selection     = coda_g_selection,
1117
1118         .vidioc_try_encoder_cmd = coda_try_encoder_cmd,
1119         .vidioc_encoder_cmd     = coda_encoder_cmd,
1120         .vidioc_try_decoder_cmd = coda_try_decoder_cmd,
1121         .vidioc_decoder_cmd     = coda_decoder_cmd,
1122
1123         .vidioc_g_parm          = coda_g_parm,
1124         .vidioc_s_parm          = coda_s_parm,
1125
1126         .vidioc_subscribe_event = coda_subscribe_event,
1127         .vidioc_unsubscribe_event = v4l2_event_unsubscribe,
1128 };
1129
1130 /*
1131  * Mem-to-mem operations.
1132  */
1133
1134 static void coda_device_run(void *m2m_priv)
1135 {
1136         struct coda_ctx *ctx = m2m_priv;
1137         struct coda_dev *dev = ctx->dev;
1138
1139         queue_work(dev->workqueue, &ctx->pic_run_work);
1140 }
1141
1142 static void coda_pic_run_work(struct work_struct *work)
1143 {
1144         struct coda_ctx *ctx = container_of(work, struct coda_ctx, pic_run_work);
1145         struct coda_dev *dev = ctx->dev;
1146         int ret;
1147
1148         mutex_lock(&ctx->buffer_mutex);
1149         mutex_lock(&dev->coda_mutex);
1150
1151         ret = ctx->ops->prepare_run(ctx);
1152         if (ret < 0 && ctx->inst_type == CODA_INST_DECODER) {
1153                 mutex_unlock(&dev->coda_mutex);
1154                 mutex_unlock(&ctx->buffer_mutex);
1155                 /* job_finish scheduled by prepare_decode */
1156                 return;
1157         }
1158
1159         if (!wait_for_completion_timeout(&ctx->completion,
1160                                          msecs_to_jiffies(1000))) {
1161                 dev_err(&dev->plat_dev->dev, "CODA PIC_RUN timeout\n");
1162
1163                 ctx->hold = true;
1164
1165                 coda_hw_reset(ctx);
1166
1167                 if (ctx->ops->run_timeout)
1168                         ctx->ops->run_timeout(ctx);
1169         } else if (!ctx->aborting) {
1170                 ctx->ops->finish_run(ctx);
1171         }
1172
1173         if ((ctx->aborting || (!ctx->streamon_cap && !ctx->streamon_out)) &&
1174             ctx->ops->seq_end_work)
1175                 queue_work(dev->workqueue, &ctx->seq_end_work);
1176
1177         mutex_unlock(&dev->coda_mutex);
1178         mutex_unlock(&ctx->buffer_mutex);
1179
1180         v4l2_m2m_job_finish(ctx->dev->m2m_dev, ctx->fh.m2m_ctx);
1181 }
1182
1183 static int coda_job_ready(void *m2m_priv)
1184 {
1185         struct coda_ctx *ctx = m2m_priv;
1186         int src_bufs = v4l2_m2m_num_src_bufs_ready(ctx->fh.m2m_ctx);
1187
1188         /*
1189          * For both 'P' and 'key' frame cases 1 picture
1190          * and 1 frame are needed. In the decoder case,
1191          * the compressed frame can be in the bitstream.
1192          */
1193         if (!src_bufs && ctx->inst_type != CODA_INST_DECODER) {
1194                 v4l2_dbg(1, coda_debug, &ctx->dev->v4l2_dev,
1195                          "not ready: not enough video buffers.\n");
1196                 return 0;
1197         }
1198
1199         if (!v4l2_m2m_num_dst_bufs_ready(ctx->fh.m2m_ctx)) {
1200                 v4l2_dbg(1, coda_debug, &ctx->dev->v4l2_dev,
1201                          "not ready: not enough video capture buffers.\n");
1202                 return 0;
1203         }
1204
1205         if (ctx->inst_type == CODA_INST_DECODER && ctx->use_bit) {
1206                 bool stream_end = ctx->bit_stream_param &
1207                                   CODA_BIT_STREAM_END_FLAG;
1208                 int num_metas = ctx->num_metas;
1209                 unsigned int count;
1210
1211                 count = hweight32(ctx->frm_dis_flg);
1212                 if (ctx->use_vdoa && count >= (ctx->num_internal_frames - 1)) {
1213                         v4l2_dbg(1, coda_debug, &ctx->dev->v4l2_dev,
1214                                  "%d: not ready: all internal buffers in use: %d/%d (0x%x)",
1215                                  ctx->idx, count, ctx->num_internal_frames,
1216                                  ctx->frm_dis_flg);
1217                         return 0;
1218                 }
1219
1220                 if (ctx->hold && !src_bufs) {
1221                         v4l2_dbg(1, coda_debug, &ctx->dev->v4l2_dev,
1222                                  "%d: not ready: on hold for more buffers.\n",
1223                                  ctx->idx);
1224                         return 0;
1225                 }
1226
1227                 if (!stream_end && (num_metas + src_bufs) < 2) {
1228                         v4l2_dbg(1, coda_debug, &ctx->dev->v4l2_dev,
1229                                  "%d: not ready: need 2 buffers available (%d, %d)\n",
1230                                  ctx->idx, num_metas, src_bufs);
1231                         return 0;
1232                 }
1233
1234
1235                 if (!src_bufs && !stream_end &&
1236                     (coda_get_bitstream_payload(ctx) < 512)) {
1237                         v4l2_dbg(1, coda_debug, &ctx->dev->v4l2_dev,
1238                                  "%d: not ready: not enough bitstream data (%d).\n",
1239                                  ctx->idx, coda_get_bitstream_payload(ctx));
1240                         return 0;
1241                 }
1242         }
1243
1244         if (ctx->aborting) {
1245                 v4l2_dbg(1, coda_debug, &ctx->dev->v4l2_dev,
1246                          "not ready: aborting\n");
1247                 return 0;
1248         }
1249
1250         v4l2_dbg(1, coda_debug, &ctx->dev->v4l2_dev,
1251                         "job ready\n");
1252
1253         return 1;
1254 }
1255
1256 static void coda_job_abort(void *priv)
1257 {
1258         struct coda_ctx *ctx = priv;
1259
1260         ctx->aborting = 1;
1261
1262         v4l2_dbg(1, coda_debug, &ctx->dev->v4l2_dev,
1263                  "Aborting task\n");
1264 }
1265
1266 static void coda_lock(void *m2m_priv)
1267 {
1268         struct coda_ctx *ctx = m2m_priv;
1269         struct coda_dev *pcdev = ctx->dev;
1270
1271         mutex_lock(&pcdev->dev_mutex);
1272 }
1273
1274 static void coda_unlock(void *m2m_priv)
1275 {
1276         struct coda_ctx *ctx = m2m_priv;
1277         struct coda_dev *pcdev = ctx->dev;
1278
1279         mutex_unlock(&pcdev->dev_mutex);
1280 }
1281
1282 static const struct v4l2_m2m_ops coda_m2m_ops = {
1283         .device_run     = coda_device_run,
1284         .job_ready      = coda_job_ready,
1285         .job_abort      = coda_job_abort,
1286         .lock           = coda_lock,
1287         .unlock         = coda_unlock,
1288 };
1289
1290 static void set_default_params(struct coda_ctx *ctx)
1291 {
1292         unsigned int max_w, max_h, usize, csize;
1293
1294         ctx->codec = coda_find_codec(ctx->dev, ctx->cvd->src_formats[0],
1295                                      ctx->cvd->dst_formats[0]);
1296         max_w = min(ctx->codec->max_w, 1920U);
1297         max_h = min(ctx->codec->max_h, 1088U);
1298         usize = max_w * max_h * 3 / 2;
1299         csize = coda_estimate_sizeimage(ctx, usize, max_w, max_h);
1300
1301         ctx->params.codec_mode = ctx->codec->mode;
1302         if (ctx->cvd->src_formats[0] == V4L2_PIX_FMT_JPEG)
1303                 ctx->colorspace = V4L2_COLORSPACE_JPEG;
1304         else
1305                 ctx->colorspace = V4L2_COLORSPACE_REC709;
1306         ctx->xfer_func = V4L2_XFER_FUNC_DEFAULT;
1307         ctx->ycbcr_enc = V4L2_YCBCR_ENC_DEFAULT;
1308         ctx->quantization = V4L2_QUANTIZATION_DEFAULT;
1309         ctx->params.framerate = 30;
1310
1311         /* Default formats for output and input queues */
1312         ctx->q_data[V4L2_M2M_SRC].fourcc = ctx->cvd->src_formats[0];
1313         ctx->q_data[V4L2_M2M_DST].fourcc = ctx->cvd->dst_formats[0];
1314         ctx->q_data[V4L2_M2M_SRC].width = max_w;
1315         ctx->q_data[V4L2_M2M_SRC].height = max_h;
1316         ctx->q_data[V4L2_M2M_DST].width = max_w;
1317         ctx->q_data[V4L2_M2M_DST].height = max_h;
1318         if (ctx->codec->src_fourcc == V4L2_PIX_FMT_YUV420) {
1319                 ctx->q_data[V4L2_M2M_SRC].bytesperline = max_w;
1320                 ctx->q_data[V4L2_M2M_SRC].sizeimage = usize;
1321                 ctx->q_data[V4L2_M2M_DST].bytesperline = 0;
1322                 ctx->q_data[V4L2_M2M_DST].sizeimage = csize;
1323         } else {
1324                 ctx->q_data[V4L2_M2M_SRC].bytesperline = 0;
1325                 ctx->q_data[V4L2_M2M_SRC].sizeimage = csize;
1326                 ctx->q_data[V4L2_M2M_DST].bytesperline = max_w;
1327                 ctx->q_data[V4L2_M2M_DST].sizeimage = usize;
1328         }
1329         ctx->q_data[V4L2_M2M_SRC].rect.width = max_w;
1330         ctx->q_data[V4L2_M2M_SRC].rect.height = max_h;
1331         ctx->q_data[V4L2_M2M_DST].rect.width = max_w;
1332         ctx->q_data[V4L2_M2M_DST].rect.height = max_h;
1333
1334         /*
1335          * Since the RBC2AXI logic only supports a single chroma plane,
1336          * macroblock tiling only works for to NV12 pixel format.
1337          */
1338         ctx->tiled_map_type = GDI_LINEAR_FRAME_MAP;
1339 }
1340
1341 /*
1342  * Queue operations
1343  */
1344 static int coda_queue_setup(struct vb2_queue *vq,
1345                                 unsigned int *nbuffers, unsigned int *nplanes,
1346                                 unsigned int sizes[], struct device *alloc_devs[])
1347 {
1348         struct coda_ctx *ctx = vb2_get_drv_priv(vq);
1349         struct coda_q_data *q_data;
1350         unsigned int size;
1351
1352         q_data = get_q_data(ctx, vq->type);
1353         size = q_data->sizeimage;
1354
1355         *nplanes = 1;
1356         sizes[0] = size;
1357
1358         v4l2_dbg(1, coda_debug, &ctx->dev->v4l2_dev,
1359                  "get %d buffer(s) of size %d each.\n", *nbuffers, size);
1360
1361         return 0;
1362 }
1363
1364 static int coda_buf_prepare(struct vb2_buffer *vb)
1365 {
1366         struct coda_ctx *ctx = vb2_get_drv_priv(vb->vb2_queue);
1367         struct coda_q_data *q_data;
1368
1369         q_data = get_q_data(ctx, vb->vb2_queue->type);
1370
1371         if (vb2_plane_size(vb, 0) < q_data->sizeimage) {
1372                 v4l2_warn(&ctx->dev->v4l2_dev,
1373                           "%s data will not fit into plane (%lu < %lu)\n",
1374                           __func__, vb2_plane_size(vb, 0),
1375                           (long)q_data->sizeimage);
1376                 return -EINVAL;
1377         }
1378
1379         return 0;
1380 }
1381
1382 static void coda_buf_queue(struct vb2_buffer *vb)
1383 {
1384         struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
1385         struct coda_ctx *ctx = vb2_get_drv_priv(vb->vb2_queue);
1386         struct vb2_queue *vq = vb->vb2_queue;
1387         struct coda_q_data *q_data;
1388
1389         q_data = get_q_data(ctx, vb->vb2_queue->type);
1390
1391         /*
1392          * In the decoder case, immediately try to copy the buffer into the
1393          * bitstream ringbuffer and mark it as ready to be dequeued.
1394          */
1395         if (ctx->bitstream.size && vq->type == V4L2_BUF_TYPE_VIDEO_OUTPUT) {
1396                 /*
1397                  * For backwards compatibility, queuing an empty buffer marks
1398                  * the stream end
1399                  */
1400                 if (vb2_get_plane_payload(vb, 0) == 0)
1401                         coda_bit_stream_end_flag(ctx);
1402
1403                 if (q_data->fourcc == V4L2_PIX_FMT_H264) {
1404                         /*
1405                          * Unless already done, try to obtain profile_idc and
1406                          * level_idc from the SPS header. This allows to decide
1407                          * whether to enable reordering during sequence
1408                          * initialization.
1409                          */
1410                         if (!ctx->params.h264_profile_idc)
1411                                 coda_sps_parse_profile(ctx, vb);
1412                 }
1413
1414                 mutex_lock(&ctx->bitstream_mutex);
1415                 v4l2_m2m_buf_queue(ctx->fh.m2m_ctx, vbuf);
1416                 if (vb2_is_streaming(vb->vb2_queue))
1417                         /* This set buf->sequence = ctx->qsequence++ */
1418                         coda_fill_bitstream(ctx, NULL);
1419                 mutex_unlock(&ctx->bitstream_mutex);
1420         } else {
1421                 if (ctx->inst_type == CODA_INST_ENCODER &&
1422                     vq->type == V4L2_BUF_TYPE_VIDEO_OUTPUT)
1423                         vbuf->sequence = ctx->qsequence++;
1424                 v4l2_m2m_buf_queue(ctx->fh.m2m_ctx, vbuf);
1425         }
1426 }
1427
1428 int coda_alloc_aux_buf(struct coda_dev *dev, struct coda_aux_buf *buf,
1429                        size_t size, const char *name, struct dentry *parent)
1430 {
1431         buf->vaddr = dma_alloc_coherent(&dev->plat_dev->dev, size, &buf->paddr,
1432                                         GFP_KERNEL);
1433         if (!buf->vaddr) {
1434                 v4l2_err(&dev->v4l2_dev,
1435                          "Failed to allocate %s buffer of size %zu\n",
1436                          name, size);
1437                 return -ENOMEM;
1438         }
1439
1440         buf->size = size;
1441
1442         if (name && parent) {
1443                 buf->blob.data = buf->vaddr;
1444                 buf->blob.size = size;
1445                 buf->dentry = debugfs_create_blob(name, 0644, parent,
1446                                                   &buf->blob);
1447                 if (!buf->dentry)
1448                         dev_warn(&dev->plat_dev->dev,
1449                                  "failed to create debugfs entry %s\n", name);
1450         }
1451
1452         return 0;
1453 }
1454
1455 void coda_free_aux_buf(struct coda_dev *dev,
1456                        struct coda_aux_buf *buf)
1457 {
1458         if (buf->vaddr) {
1459                 dma_free_coherent(&dev->plat_dev->dev, buf->size,
1460                                   buf->vaddr, buf->paddr);
1461                 buf->vaddr = NULL;
1462                 buf->size = 0;
1463                 debugfs_remove(buf->dentry);
1464                 buf->dentry = NULL;
1465         }
1466 }
1467
1468 static int coda_start_streaming(struct vb2_queue *q, unsigned int count)
1469 {
1470         struct coda_ctx *ctx = vb2_get_drv_priv(q);
1471         struct v4l2_device *v4l2_dev = &ctx->dev->v4l2_dev;
1472         struct coda_q_data *q_data_src, *q_data_dst;
1473         struct v4l2_m2m_buffer *m2m_buf, *tmp;
1474         struct vb2_v4l2_buffer *buf;
1475         struct list_head list;
1476         int ret = 0;
1477
1478         if (count < 1)
1479                 return -EINVAL;
1480
1481         INIT_LIST_HEAD(&list);
1482
1483         q_data_src = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT);
1484         if (q->type == V4L2_BUF_TYPE_VIDEO_OUTPUT) {
1485                 if (ctx->inst_type == CODA_INST_DECODER && ctx->use_bit) {
1486                         /* copy the buffers that were queued before streamon */
1487                         mutex_lock(&ctx->bitstream_mutex);
1488                         coda_fill_bitstream(ctx, &list);
1489                         mutex_unlock(&ctx->bitstream_mutex);
1490
1491                         if (coda_get_bitstream_payload(ctx) < 512) {
1492                                 ret = -EINVAL;
1493                                 goto err;
1494                         }
1495                 }
1496
1497                 ctx->streamon_out = 1;
1498         } else {
1499                 ctx->streamon_cap = 1;
1500         }
1501
1502         /* Don't start the coda unless both queues are on */
1503         if (!(ctx->streamon_out && ctx->streamon_cap))
1504                 goto out;
1505
1506         q_data_dst = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE);
1507         if ((q_data_src->width != q_data_dst->width &&
1508              round_up(q_data_src->width, 16) != q_data_dst->width) ||
1509             (q_data_src->height != q_data_dst->height &&
1510              round_up(q_data_src->height, 16) != q_data_dst->height)) {
1511                 v4l2_err(v4l2_dev, "can't convert %dx%d to %dx%d\n",
1512                          q_data_src->width, q_data_src->height,
1513                          q_data_dst->width, q_data_dst->height);
1514                 ret = -EINVAL;
1515                 goto err;
1516         }
1517
1518         /* Allow BIT decoder device_run with no new buffers queued */
1519         if (ctx->inst_type == CODA_INST_DECODER && ctx->use_bit)
1520                 v4l2_m2m_set_src_buffered(ctx->fh.m2m_ctx, true);
1521
1522         ctx->gopcounter = ctx->params.gop_size - 1;
1523
1524         ctx->codec = coda_find_codec(ctx->dev, q_data_src->fourcc,
1525                                      q_data_dst->fourcc);
1526         if (!ctx->codec) {
1527                 v4l2_err(v4l2_dev, "couldn't tell instance type.\n");
1528                 ret = -EINVAL;
1529                 goto err;
1530         }
1531
1532         if (q_data_dst->fourcc == V4L2_PIX_FMT_JPEG)
1533                 ctx->params.gop_size = 1;
1534         ctx->gopcounter = ctx->params.gop_size - 1;
1535
1536         ret = ctx->ops->start_streaming(ctx);
1537         if (ctx->inst_type == CODA_INST_DECODER) {
1538                 if (ret == -EAGAIN)
1539                         goto out;
1540         }
1541         if (ret < 0)
1542                 goto err;
1543
1544 out:
1545         if (q->type == V4L2_BUF_TYPE_VIDEO_OUTPUT) {
1546                 list_for_each_entry_safe(m2m_buf, tmp, &list, list) {
1547                         list_del(&m2m_buf->list);
1548                         v4l2_m2m_buf_done(&m2m_buf->vb, VB2_BUF_STATE_DONE);
1549                 }
1550         }
1551         return 0;
1552
1553 err:
1554         if (q->type == V4L2_BUF_TYPE_VIDEO_OUTPUT) {
1555                 list_for_each_entry_safe(m2m_buf, tmp, &list, list) {
1556                         list_del(&m2m_buf->list);
1557                         v4l2_m2m_buf_done(&m2m_buf->vb, VB2_BUF_STATE_QUEUED);
1558                 }
1559                 while ((buf = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx)))
1560                         v4l2_m2m_buf_done(buf, VB2_BUF_STATE_QUEUED);
1561         } else {
1562                 while ((buf = v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx)))
1563                         v4l2_m2m_buf_done(buf, VB2_BUF_STATE_QUEUED);
1564         }
1565         return ret;
1566 }
1567
1568 static void coda_stop_streaming(struct vb2_queue *q)
1569 {
1570         struct coda_ctx *ctx = vb2_get_drv_priv(q);
1571         struct coda_dev *dev = ctx->dev;
1572         struct vb2_v4l2_buffer *buf;
1573         unsigned long flags;
1574         bool stop;
1575
1576         stop = ctx->streamon_out && ctx->streamon_cap;
1577
1578         if (q->type == V4L2_BUF_TYPE_VIDEO_OUTPUT) {
1579                 v4l2_dbg(1, coda_debug, &dev->v4l2_dev,
1580                          "%s: output\n", __func__);
1581                 ctx->streamon_out = 0;
1582
1583                 coda_bit_stream_end_flag(ctx);
1584
1585                 ctx->qsequence = 0;
1586
1587                 while ((buf = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx)))
1588                         v4l2_m2m_buf_done(buf, VB2_BUF_STATE_ERROR);
1589         } else {
1590                 v4l2_dbg(1, coda_debug, &dev->v4l2_dev,
1591                          "%s: capture\n", __func__);
1592                 ctx->streamon_cap = 0;
1593
1594                 ctx->osequence = 0;
1595                 ctx->sequence_offset = 0;
1596
1597                 while ((buf = v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx)))
1598                         v4l2_m2m_buf_done(buf, VB2_BUF_STATE_ERROR);
1599         }
1600
1601         if (stop) {
1602                 struct coda_buffer_meta *meta;
1603
1604                 if (ctx->ops->seq_end_work) {
1605                         queue_work(dev->workqueue, &ctx->seq_end_work);
1606                         flush_work(&ctx->seq_end_work);
1607                 }
1608                 spin_lock_irqsave(&ctx->buffer_meta_lock, flags);
1609                 while (!list_empty(&ctx->buffer_meta_list)) {
1610                         meta = list_first_entry(&ctx->buffer_meta_list,
1611                                                 struct coda_buffer_meta, list);
1612                         list_del(&meta->list);
1613                         kfree(meta);
1614                 }
1615                 ctx->num_metas = 0;
1616                 spin_unlock_irqrestore(&ctx->buffer_meta_lock, flags);
1617                 kfifo_init(&ctx->bitstream_fifo,
1618                         ctx->bitstream.vaddr, ctx->bitstream.size);
1619                 ctx->runcounter = 0;
1620                 ctx->aborting = 0;
1621         }
1622
1623         if (!ctx->streamon_out && !ctx->streamon_cap)
1624                 ctx->bit_stream_param &= ~CODA_BIT_STREAM_END_FLAG;
1625 }
1626
1627 static const struct vb2_ops coda_qops = {
1628         .queue_setup            = coda_queue_setup,
1629         .buf_prepare            = coda_buf_prepare,
1630         .buf_queue              = coda_buf_queue,
1631         .start_streaming        = coda_start_streaming,
1632         .stop_streaming         = coda_stop_streaming,
1633         .wait_prepare           = vb2_ops_wait_prepare,
1634         .wait_finish            = vb2_ops_wait_finish,
1635 };
1636
1637 static int coda_s_ctrl(struct v4l2_ctrl *ctrl)
1638 {
1639         struct coda_ctx *ctx =
1640                         container_of(ctrl->handler, struct coda_ctx, ctrls);
1641
1642         v4l2_dbg(1, coda_debug, &ctx->dev->v4l2_dev,
1643                  "s_ctrl: id = %d, val = %d\n", ctrl->id, ctrl->val);
1644
1645         switch (ctrl->id) {
1646         case V4L2_CID_HFLIP:
1647                 if (ctrl->val)
1648                         ctx->params.rot_mode |= CODA_MIR_HOR;
1649                 else
1650                         ctx->params.rot_mode &= ~CODA_MIR_HOR;
1651                 break;
1652         case V4L2_CID_VFLIP:
1653                 if (ctrl->val)
1654                         ctx->params.rot_mode |= CODA_MIR_VER;
1655                 else
1656                         ctx->params.rot_mode &= ~CODA_MIR_VER;
1657                 break;
1658         case V4L2_CID_MPEG_VIDEO_BITRATE:
1659                 ctx->params.bitrate = ctrl->val / 1000;
1660                 break;
1661         case V4L2_CID_MPEG_VIDEO_GOP_SIZE:
1662                 ctx->params.gop_size = ctrl->val;
1663                 break;
1664         case V4L2_CID_MPEG_VIDEO_H264_I_FRAME_QP:
1665                 ctx->params.h264_intra_qp = ctrl->val;
1666                 break;
1667         case V4L2_CID_MPEG_VIDEO_H264_P_FRAME_QP:
1668                 ctx->params.h264_inter_qp = ctrl->val;
1669                 break;
1670         case V4L2_CID_MPEG_VIDEO_H264_MIN_QP:
1671                 ctx->params.h264_min_qp = ctrl->val;
1672                 break;
1673         case V4L2_CID_MPEG_VIDEO_H264_MAX_QP:
1674                 ctx->params.h264_max_qp = ctrl->val;
1675                 break;
1676         case V4L2_CID_MPEG_VIDEO_H264_LOOP_FILTER_ALPHA:
1677                 ctx->params.h264_deblk_alpha = ctrl->val;
1678                 break;
1679         case V4L2_CID_MPEG_VIDEO_H264_LOOP_FILTER_BETA:
1680                 ctx->params.h264_deblk_beta = ctrl->val;
1681                 break;
1682         case V4L2_CID_MPEG_VIDEO_H264_LOOP_FILTER_MODE:
1683                 ctx->params.h264_deblk_enabled = (ctrl->val ==
1684                                 V4L2_MPEG_VIDEO_H264_LOOP_FILTER_MODE_ENABLED);
1685                 break;
1686         case V4L2_CID_MPEG_VIDEO_MPEG4_I_FRAME_QP:
1687                 ctx->params.mpeg4_intra_qp = ctrl->val;
1688                 break;
1689         case V4L2_CID_MPEG_VIDEO_MPEG4_P_FRAME_QP:
1690                 ctx->params.mpeg4_inter_qp = ctrl->val;
1691                 break;
1692         case V4L2_CID_MPEG_VIDEO_MULTI_SLICE_MODE:
1693                 ctx->params.slice_mode = ctrl->val;
1694                 break;
1695         case V4L2_CID_MPEG_VIDEO_MULTI_SLICE_MAX_MB:
1696                 ctx->params.slice_max_mb = ctrl->val;
1697                 break;
1698         case V4L2_CID_MPEG_VIDEO_MULTI_SLICE_MAX_BYTES:
1699                 ctx->params.slice_max_bits = ctrl->val * 8;
1700                 break;
1701         case V4L2_CID_MPEG_VIDEO_HEADER_MODE:
1702                 break;
1703         case V4L2_CID_MPEG_VIDEO_CYCLIC_INTRA_REFRESH_MB:
1704                 ctx->params.intra_refresh = ctrl->val;
1705                 break;
1706         case V4L2_CID_MPEG_VIDEO_FORCE_KEY_FRAME:
1707                 ctx->params.force_ipicture = true;
1708                 break;
1709         case V4L2_CID_JPEG_COMPRESSION_QUALITY:
1710                 coda_set_jpeg_compression_quality(ctx, ctrl->val);
1711                 break;
1712         case V4L2_CID_JPEG_RESTART_INTERVAL:
1713                 ctx->params.jpeg_restart_interval = ctrl->val;
1714                 break;
1715         case V4L2_CID_MPEG_VIDEO_VBV_DELAY:
1716                 ctx->params.vbv_delay = ctrl->val;
1717                 break;
1718         case V4L2_CID_MPEG_VIDEO_VBV_SIZE:
1719                 ctx->params.vbv_size = min(ctrl->val * 8192, 0x7fffffff);
1720                 break;
1721         default:
1722                 v4l2_dbg(1, coda_debug, &ctx->dev->v4l2_dev,
1723                         "Invalid control, id=%d, val=%d\n",
1724                         ctrl->id, ctrl->val);
1725                 return -EINVAL;
1726         }
1727
1728         return 0;
1729 }
1730
1731 static const struct v4l2_ctrl_ops coda_ctrl_ops = {
1732         .s_ctrl = coda_s_ctrl,
1733 };
1734
1735 static void coda_encode_ctrls(struct coda_ctx *ctx)
1736 {
1737         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1738                 V4L2_CID_MPEG_VIDEO_BITRATE, 0, 32767000, 1000, 0);
1739         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1740                 V4L2_CID_MPEG_VIDEO_GOP_SIZE, 1, 60, 1, 16);
1741         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1742                 V4L2_CID_MPEG_VIDEO_H264_I_FRAME_QP, 0, 51, 1, 25);
1743         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1744                 V4L2_CID_MPEG_VIDEO_H264_P_FRAME_QP, 0, 51, 1, 25);
1745         if (ctx->dev->devtype->product != CODA_960) {
1746                 v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1747                         V4L2_CID_MPEG_VIDEO_H264_MIN_QP, 0, 51, 1, 12);
1748         }
1749         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1750                 V4L2_CID_MPEG_VIDEO_H264_MAX_QP, 0, 51, 1, 51);
1751         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1752                 V4L2_CID_MPEG_VIDEO_H264_LOOP_FILTER_ALPHA, 0, 15, 1, 0);
1753         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1754                 V4L2_CID_MPEG_VIDEO_H264_LOOP_FILTER_BETA, 0, 15, 1, 0);
1755         v4l2_ctrl_new_std_menu(&ctx->ctrls, &coda_ctrl_ops,
1756                 V4L2_CID_MPEG_VIDEO_H264_LOOP_FILTER_MODE,
1757                 V4L2_MPEG_VIDEO_H264_LOOP_FILTER_MODE_DISABLED, 0x0,
1758                 V4L2_MPEG_VIDEO_H264_LOOP_FILTER_MODE_ENABLED);
1759         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1760                 V4L2_CID_MPEG_VIDEO_MPEG4_I_FRAME_QP, 1, 31, 1, 2);
1761         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1762                 V4L2_CID_MPEG_VIDEO_MPEG4_P_FRAME_QP, 1, 31, 1, 2);
1763         v4l2_ctrl_new_std_menu(&ctx->ctrls, &coda_ctrl_ops,
1764                 V4L2_CID_MPEG_VIDEO_MULTI_SLICE_MODE,
1765                 V4L2_MPEG_VIDEO_MULTI_SICE_MODE_MAX_BYTES, 0x0,
1766                 V4L2_MPEG_VIDEO_MULTI_SLICE_MODE_SINGLE);
1767         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1768                 V4L2_CID_MPEG_VIDEO_MULTI_SLICE_MAX_MB, 1, 0x3fffffff, 1, 1);
1769         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1770                 V4L2_CID_MPEG_VIDEO_MULTI_SLICE_MAX_BYTES, 1, 0x3fffffff, 1,
1771                 500);
1772         v4l2_ctrl_new_std_menu(&ctx->ctrls, &coda_ctrl_ops,
1773                 V4L2_CID_MPEG_VIDEO_HEADER_MODE,
1774                 V4L2_MPEG_VIDEO_HEADER_MODE_JOINED_WITH_1ST_FRAME,
1775                 (1 << V4L2_MPEG_VIDEO_HEADER_MODE_SEPARATE),
1776                 V4L2_MPEG_VIDEO_HEADER_MODE_JOINED_WITH_1ST_FRAME);
1777         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1778                 V4L2_CID_MPEG_VIDEO_CYCLIC_INTRA_REFRESH_MB, 0,
1779                 1920 * 1088 / 256, 1, 0);
1780         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1781                 V4L2_CID_MPEG_VIDEO_VBV_DELAY, 0, 0x7fff, 1, 0);
1782         /*
1783          * The maximum VBV size value is 0x7fffffff bits,
1784          * one bit less than 262144 KiB
1785          */
1786         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1787                 V4L2_CID_MPEG_VIDEO_VBV_SIZE, 0, 262144, 1, 0);
1788 }
1789
1790 static void coda_jpeg_encode_ctrls(struct coda_ctx *ctx)
1791 {
1792         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1793                 V4L2_CID_JPEG_COMPRESSION_QUALITY, 5, 100, 1, 50);
1794         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1795                 V4L2_CID_JPEG_RESTART_INTERVAL, 0, 100, 1, 0);
1796 }
1797
1798 static int coda_ctrls_setup(struct coda_ctx *ctx)
1799 {
1800         v4l2_ctrl_handler_init(&ctx->ctrls, 2);
1801
1802         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1803                 V4L2_CID_HFLIP, 0, 1, 1, 0);
1804         v4l2_ctrl_new_std(&ctx->ctrls, &coda_ctrl_ops,
1805                 V4L2_CID_VFLIP, 0, 1, 1, 0);
1806         if (ctx->inst_type == CODA_INST_ENCODER) {
1807                 if (ctx->cvd->dst_formats[0] == V4L2_PIX_FMT_JPEG)
1808                         coda_jpeg_encode_ctrls(ctx);
1809                 else
1810                         coda_encode_ctrls(ctx);
1811         }
1812
1813         if (ctx->ctrls.error) {
1814                 v4l2_err(&ctx->dev->v4l2_dev,
1815                         "control initialization error (%d)",
1816                         ctx->ctrls.error);
1817                 return -EINVAL;
1818         }
1819
1820         return v4l2_ctrl_handler_setup(&ctx->ctrls);
1821 }
1822
1823 static int coda_queue_init(struct coda_ctx *ctx, struct vb2_queue *vq)
1824 {
1825         vq->drv_priv = ctx;
1826         vq->ops = &coda_qops;
1827         vq->buf_struct_size = sizeof(struct v4l2_m2m_buffer);
1828         vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY;
1829         vq->lock = &ctx->dev->dev_mutex;
1830         /* One way to indicate end-of-stream for coda is to set the
1831          * bytesused == 0. However by default videobuf2 handles bytesused
1832          * equal to 0 as a special case and changes its value to the size
1833          * of the buffer. Set the allow_zero_bytesused flag, so
1834          * that videobuf2 will keep the value of bytesused intact.
1835          */
1836         vq->allow_zero_bytesused = 1;
1837         vq->dev = &ctx->dev->plat_dev->dev;
1838
1839         return vb2_queue_init(vq);
1840 }
1841
1842 int coda_encoder_queue_init(void *priv, struct vb2_queue *src_vq,
1843                             struct vb2_queue *dst_vq)
1844 {
1845         int ret;
1846
1847         src_vq->type = V4L2_BUF_TYPE_VIDEO_OUTPUT;
1848         src_vq->io_modes = VB2_DMABUF | VB2_MMAP;
1849         src_vq->mem_ops = &vb2_dma_contig_memops;
1850
1851         ret = coda_queue_init(priv, src_vq);
1852         if (ret)
1853                 return ret;
1854
1855         dst_vq->type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
1856         dst_vq->io_modes = VB2_DMABUF | VB2_MMAP;
1857         dst_vq->mem_ops = &vb2_dma_contig_memops;
1858
1859         return coda_queue_init(priv, dst_vq);
1860 }
1861
1862 int coda_decoder_queue_init(void *priv, struct vb2_queue *src_vq,
1863                             struct vb2_queue *dst_vq)
1864 {
1865         int ret;
1866
1867         src_vq->type = V4L2_BUF_TYPE_VIDEO_OUTPUT;
1868         src_vq->io_modes = VB2_DMABUF | VB2_MMAP | VB2_USERPTR;
1869         src_vq->mem_ops = &vb2_vmalloc_memops;
1870
1871         ret = coda_queue_init(priv, src_vq);
1872         if (ret)
1873                 return ret;
1874
1875         dst_vq->type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
1876         dst_vq->io_modes = VB2_DMABUF | VB2_MMAP;
1877         dst_vq->mem_ops = &vb2_dma_contig_memops;
1878
1879         return coda_queue_init(priv, dst_vq);
1880 }
1881
1882 static int coda_next_free_instance(struct coda_dev *dev)
1883 {
1884         int idx = ffz(dev->instance_mask);
1885
1886         if ((idx < 0) ||
1887             (dev->devtype->product == CODA_DX6 && idx > CODADX6_MAX_INSTANCES))
1888                 return -EBUSY;
1889
1890         return idx;
1891 }
1892
1893 /*
1894  * File operations
1895  */
1896
1897 static int coda_open(struct file *file)
1898 {
1899         struct video_device *vdev = video_devdata(file);
1900         struct coda_dev *dev = video_get_drvdata(vdev);
1901         struct coda_ctx *ctx = NULL;
1902         char *name;
1903         int ret;
1904         int idx;
1905
1906         ctx = kzalloc(sizeof(*ctx), GFP_KERNEL);
1907         if (!ctx)
1908                 return -ENOMEM;
1909
1910         idx = coda_next_free_instance(dev);
1911         if (idx < 0) {
1912                 ret = idx;
1913                 goto err_coda_max;
1914         }
1915         set_bit(idx, &dev->instance_mask);
1916
1917         name = kasprintf(GFP_KERNEL, "context%d", idx);
1918         if (!name) {
1919                 ret = -ENOMEM;
1920                 goto err_coda_name_init;
1921         }
1922
1923         ctx->debugfs_entry = debugfs_create_dir(name, dev->debugfs_root);
1924         kfree(name);
1925
1926         ctx->cvd = to_coda_video_device(vdev);
1927         ctx->inst_type = ctx->cvd->type;
1928         ctx->ops = ctx->cvd->ops;
1929         ctx->use_bit = !ctx->cvd->direct;
1930         init_completion(&ctx->completion);
1931         INIT_WORK(&ctx->pic_run_work, coda_pic_run_work);
1932         if (ctx->ops->seq_end_work)
1933                 INIT_WORK(&ctx->seq_end_work, ctx->ops->seq_end_work);
1934         v4l2_fh_init(&ctx->fh, video_devdata(file));
1935         file->private_data = &ctx->fh;
1936         v4l2_fh_add(&ctx->fh);
1937         ctx->dev = dev;
1938         ctx->idx = idx;
1939         switch (dev->devtype->product) {
1940         case CODA_960:
1941                 if (enable_bwb)
1942                         ctx->frame_mem_ctrl = CODA9_FRAME_ENABLE_BWB;
1943                 /* fallthrough */
1944         case CODA_7541:
1945                 ctx->reg_idx = 0;
1946                 break;
1947         default:
1948                 ctx->reg_idx = idx;
1949         }
1950         if (ctx->dev->vdoa && !disable_vdoa) {
1951                 ctx->vdoa = vdoa_context_create(dev->vdoa);
1952                 if (!ctx->vdoa)
1953                         v4l2_warn(&dev->v4l2_dev,
1954                                   "Failed to create vdoa context: not using vdoa");
1955         }
1956         ctx->use_vdoa = false;
1957
1958         /* Power up and upload firmware if necessary */
1959         ret = pm_runtime_get_sync(&dev->plat_dev->dev);
1960         if (ret < 0) {
1961                 v4l2_err(&dev->v4l2_dev, "failed to power up: %d\n", ret);
1962                 goto err_pm_get;
1963         }
1964
1965         ret = clk_prepare_enable(dev->clk_per);
1966         if (ret)
1967                 goto err_clk_per;
1968
1969         ret = clk_prepare_enable(dev->clk_ahb);
1970         if (ret)
1971                 goto err_clk_ahb;
1972
1973         set_default_params(ctx);
1974         ctx->fh.m2m_ctx = v4l2_m2m_ctx_init(dev->m2m_dev, ctx,
1975                                             ctx->ops->queue_init);
1976         if (IS_ERR(ctx->fh.m2m_ctx)) {
1977                 ret = PTR_ERR(ctx->fh.m2m_ctx);
1978
1979                 v4l2_err(&dev->v4l2_dev, "%s return error (%d)\n",
1980                          __func__, ret);
1981                 goto err_ctx_init;
1982         }
1983
1984         ret = coda_ctrls_setup(ctx);
1985         if (ret) {
1986                 v4l2_err(&dev->v4l2_dev, "failed to setup coda controls\n");
1987                 goto err_ctrls_setup;
1988         }
1989
1990         ctx->fh.ctrl_handler = &ctx->ctrls;
1991
1992         mutex_init(&ctx->bitstream_mutex);
1993         mutex_init(&ctx->buffer_mutex);
1994         INIT_LIST_HEAD(&ctx->buffer_meta_list);
1995         spin_lock_init(&ctx->buffer_meta_lock);
1996
1997         coda_lock(ctx);
1998         list_add(&ctx->list, &dev->instances);
1999         coda_unlock(ctx);
2000
2001         v4l2_dbg(1, coda_debug, &dev->v4l2_dev, "Created instance %d (%p)\n",
2002                  ctx->idx, ctx);
2003
2004         return 0;
2005
2006 err_ctrls_setup:
2007         v4l2_m2m_ctx_release(ctx->fh.m2m_ctx);
2008 err_ctx_init:
2009         clk_disable_unprepare(dev->clk_ahb);
2010 err_clk_ahb:
2011         clk_disable_unprepare(dev->clk_per);
2012 err_clk_per:
2013         pm_runtime_put_sync(&dev->plat_dev->dev);
2014 err_pm_get:
2015         v4l2_fh_del(&ctx->fh);
2016         v4l2_fh_exit(&ctx->fh);
2017         clear_bit(ctx->idx, &dev->instance_mask);
2018 err_coda_name_init:
2019 err_coda_max:
2020         kfree(ctx);
2021         return ret;
2022 }
2023
2024 static int coda_release(struct file *file)
2025 {
2026         struct coda_dev *dev = video_drvdata(file);
2027         struct coda_ctx *ctx = fh_to_ctx(file->private_data);
2028
2029         v4l2_dbg(1, coda_debug, &dev->v4l2_dev, "Releasing instance %p\n",
2030                  ctx);
2031
2032         if (ctx->inst_type == CODA_INST_DECODER && ctx->use_bit)
2033                 coda_bit_stream_end_flag(ctx);
2034
2035         /* If this instance is running, call .job_abort and wait for it to end */
2036         v4l2_m2m_ctx_release(ctx->fh.m2m_ctx);
2037
2038         if (ctx->vdoa)
2039                 vdoa_context_destroy(ctx->vdoa);
2040
2041         /* In case the instance was not running, we still need to call SEQ_END */
2042         if (ctx->ops->seq_end_work) {
2043                 queue_work(dev->workqueue, &ctx->seq_end_work);
2044                 flush_work(&ctx->seq_end_work);
2045         }
2046
2047         coda_lock(ctx);
2048         list_del(&ctx->list);
2049         coda_unlock(ctx);
2050
2051         if (ctx->dev->devtype->product == CODA_DX6)
2052                 coda_free_aux_buf(dev, &ctx->workbuf);
2053
2054         v4l2_ctrl_handler_free(&ctx->ctrls);
2055         clk_disable_unprepare(dev->clk_ahb);
2056         clk_disable_unprepare(dev->clk_per);
2057         pm_runtime_put_sync(&dev->plat_dev->dev);
2058         v4l2_fh_del(&ctx->fh);
2059         v4l2_fh_exit(&ctx->fh);
2060         clear_bit(ctx->idx, &dev->instance_mask);
2061         if (ctx->ops->release)
2062                 ctx->ops->release(ctx);
2063         debugfs_remove_recursive(ctx->debugfs_entry);
2064         kfree(ctx);
2065
2066         return 0;
2067 }
2068
2069 static const struct v4l2_file_operations coda_fops = {
2070         .owner          = THIS_MODULE,
2071         .open           = coda_open,
2072         .release        = coda_release,
2073         .poll           = v4l2_m2m_fop_poll,
2074         .unlocked_ioctl = video_ioctl2,
2075         .mmap           = v4l2_m2m_fop_mmap,
2076 };
2077
2078 static int coda_hw_init(struct coda_dev *dev)
2079 {
2080         u32 data;
2081         u16 *p;
2082         int i, ret;
2083
2084         ret = clk_prepare_enable(dev->clk_per);
2085         if (ret)
2086                 goto err_clk_per;
2087
2088         ret = clk_prepare_enable(dev->clk_ahb);
2089         if (ret)
2090                 goto err_clk_ahb;
2091
2092         reset_control_reset(dev->rstc);
2093
2094         /*
2095          * Copy the first CODA_ISRAM_SIZE in the internal SRAM.
2096          * The 16-bit chars in the code buffer are in memory access
2097          * order, re-sort them to CODA order for register download.
2098          * Data in this SRAM survives a reboot.
2099          */
2100         p = (u16 *)dev->codebuf.vaddr;
2101         if (dev->devtype->product == CODA_DX6) {
2102                 for (i = 0; i < (CODA_ISRAM_SIZE / 2); i++)  {
2103                         data = CODA_DOWN_ADDRESS_SET(i) |
2104                                 CODA_DOWN_DATA_SET(p[i ^ 1]);
2105                         coda_write(dev, data, CODA_REG_BIT_CODE_DOWN);
2106                 }
2107         } else {
2108                 for (i = 0; i < (CODA_ISRAM_SIZE / 2); i++) {
2109                         data = CODA_DOWN_ADDRESS_SET(i) |
2110                                 CODA_DOWN_DATA_SET(p[round_down(i, 4) +
2111                                                         3 - (i % 4)]);
2112                         coda_write(dev, data, CODA_REG_BIT_CODE_DOWN);
2113                 }
2114         }
2115
2116         /* Clear registers */
2117         for (i = 0; i < 64; i++)
2118                 coda_write(dev, 0, CODA_REG_BIT_CODE_BUF_ADDR + i * 4);
2119
2120         /* Tell the BIT where to find everything it needs */
2121         if (dev->devtype->product == CODA_960 ||
2122             dev->devtype->product == CODA_7541) {
2123                 coda_write(dev, dev->tempbuf.paddr,
2124                                 CODA_REG_BIT_TEMP_BUF_ADDR);
2125                 coda_write(dev, 0, CODA_REG_BIT_BIT_STREAM_PARAM);
2126         } else {
2127                 coda_write(dev, dev->workbuf.paddr,
2128                               CODA_REG_BIT_WORK_BUF_ADDR);
2129         }
2130         coda_write(dev, dev->codebuf.paddr,
2131                       CODA_REG_BIT_CODE_BUF_ADDR);
2132         coda_write(dev, 0, CODA_REG_BIT_CODE_RUN);
2133
2134         /* Set default values */
2135         switch (dev->devtype->product) {
2136         case CODA_DX6:
2137                 coda_write(dev, CODADX6_STREAM_BUF_PIC_FLUSH,
2138                            CODA_REG_BIT_STREAM_CTRL);
2139                 break;
2140         default:
2141                 coda_write(dev, CODA7_STREAM_BUF_PIC_FLUSH,
2142                            CODA_REG_BIT_STREAM_CTRL);
2143         }
2144         if (dev->devtype->product == CODA_960)
2145                 coda_write(dev, 1 << 12, CODA_REG_BIT_FRAME_MEM_CTRL);
2146         else
2147                 coda_write(dev, 0, CODA_REG_BIT_FRAME_MEM_CTRL);
2148
2149         if (dev->devtype->product != CODA_DX6)
2150                 coda_write(dev, 0, CODA7_REG_BIT_AXI_SRAM_USE);
2151
2152         coda_write(dev, CODA_INT_INTERRUPT_ENABLE,
2153                       CODA_REG_BIT_INT_ENABLE);
2154
2155         /* Reset VPU and start processor */
2156         data = coda_read(dev, CODA_REG_BIT_CODE_RESET);
2157         data |= CODA_REG_RESET_ENABLE;
2158         coda_write(dev, data, CODA_REG_BIT_CODE_RESET);
2159         udelay(10);
2160         data &= ~CODA_REG_RESET_ENABLE;
2161         coda_write(dev, data, CODA_REG_BIT_CODE_RESET);
2162         coda_write(dev, CODA_REG_RUN_ENABLE, CODA_REG_BIT_CODE_RUN);
2163
2164         clk_disable_unprepare(dev->clk_ahb);
2165         clk_disable_unprepare(dev->clk_per);
2166
2167         return 0;
2168
2169 err_clk_ahb:
2170         clk_disable_unprepare(dev->clk_per);
2171 err_clk_per:
2172         return ret;
2173 }
2174
2175 static int coda_register_device(struct coda_dev *dev, int i)
2176 {
2177         struct video_device *vfd = &dev->vfd[i];
2178
2179         if (i >= dev->devtype->num_vdevs)
2180                 return -EINVAL;
2181
2182         strlcpy(vfd->name, dev->devtype->vdevs[i]->name, sizeof(vfd->name));
2183         vfd->fops       = &coda_fops;
2184         vfd->ioctl_ops  = &coda_ioctl_ops;
2185         vfd->release    = video_device_release_empty,
2186         vfd->lock       = &dev->dev_mutex;
2187         vfd->v4l2_dev   = &dev->v4l2_dev;
2188         vfd->vfl_dir    = VFL_DIR_M2M;
2189         video_set_drvdata(vfd, dev);
2190
2191         /* Not applicable, use the selection API instead */
2192         v4l2_disable_ioctl(vfd, VIDIOC_CROPCAP);
2193         v4l2_disable_ioctl(vfd, VIDIOC_G_CROP);
2194         v4l2_disable_ioctl(vfd, VIDIOC_S_CROP);
2195
2196         return video_register_device(vfd, VFL_TYPE_GRABBER, 0);
2197 }
2198
2199 static void coda_copy_firmware(struct coda_dev *dev, const u8 * const buf,
2200                                size_t size)
2201 {
2202         u32 *src = (u32 *)buf;
2203
2204         /* Check if the firmware has a 16-byte Freescale header, skip it */
2205         if (buf[0] == 'M' && buf[1] == 'X')
2206                 src += 4;
2207         /*
2208          * Check whether the firmware is in native order or pre-reordered for
2209          * memory access. The first instruction opcode always is 0xe40e.
2210          */
2211         if (__le16_to_cpup((__le16 *)src) == 0xe40e) {
2212                 u32 *dst = dev->codebuf.vaddr;
2213                 int i;
2214
2215                 /* Firmware in native order, reorder while copying */
2216                 if (dev->devtype->product == CODA_DX6) {
2217                         for (i = 0; i < (size - 16) / 4; i++)
2218                                 dst[i] = (src[i] << 16) | (src[i] >> 16);
2219                 } else {
2220                         for (i = 0; i < (size - 16) / 4; i += 2) {
2221                                 dst[i] = (src[i + 1] << 16) | (src[i + 1] >> 16);
2222                                 dst[i + 1] = (src[i] << 16) | (src[i] >> 16);
2223                         }
2224                 }
2225         } else {
2226                 /* Copy the already reordered firmware image */
2227                 memcpy(dev->codebuf.vaddr, src, size);
2228         }
2229 }
2230
2231 static void coda_fw_callback(const struct firmware *fw, void *context);
2232
2233 static int coda_firmware_request(struct coda_dev *dev)
2234 {
2235         char *fw;
2236
2237         if (dev->firmware >= ARRAY_SIZE(dev->devtype->firmware))
2238                 return -EINVAL;
2239
2240         fw = dev->devtype->firmware[dev->firmware];
2241
2242         dev_dbg(&dev->plat_dev->dev, "requesting firmware '%s' for %s\n", fw,
2243                 coda_product_name(dev->devtype->product));
2244
2245         return request_firmware_nowait(THIS_MODULE, true, fw,
2246                                        &dev->plat_dev->dev, GFP_KERNEL, dev,
2247                                        coda_fw_callback);
2248 }
2249
2250 static void coda_fw_callback(const struct firmware *fw, void *context)
2251 {
2252         struct coda_dev *dev = context;
2253         struct platform_device *pdev = dev->plat_dev;
2254         int i, ret;
2255
2256         if (!fw) {
2257                 dev->firmware++;
2258                 ret = coda_firmware_request(dev);
2259                 if (ret < 0) {
2260                         v4l2_err(&dev->v4l2_dev, "firmware request failed\n");
2261                         goto put_pm;
2262                 }
2263                 return;
2264         }
2265         if (dev->firmware > 0) {
2266                 /*
2267                  * Since we can't suppress warnings for failed asynchronous
2268                  * firmware requests, report that the fallback firmware was
2269                  * found.
2270                  */
2271                 dev_info(&pdev->dev, "Using fallback firmware %s\n",
2272                          dev->devtype->firmware[dev->firmware]);
2273         }
2274
2275         /* allocate auxiliary per-device code buffer for the BIT processor */
2276         ret = coda_alloc_aux_buf(dev, &dev->codebuf, fw->size, "codebuf",
2277                                  dev->debugfs_root);
2278         if (ret < 0)
2279                 goto put_pm;
2280
2281         coda_copy_firmware(dev, fw->data, fw->size);
2282         release_firmware(fw);
2283
2284         ret = coda_hw_init(dev);
2285         if (ret < 0) {
2286                 v4l2_err(&dev->v4l2_dev, "HW initialization failed\n");
2287                 goto put_pm;
2288         }
2289
2290         ret = coda_check_firmware(dev);
2291         if (ret < 0)
2292                 goto put_pm;
2293
2294         dev->m2m_dev = v4l2_m2m_init(&coda_m2m_ops);
2295         if (IS_ERR(dev->m2m_dev)) {
2296                 v4l2_err(&dev->v4l2_dev, "Failed to init mem2mem device\n");
2297                 goto put_pm;
2298         }
2299
2300         for (i = 0; i < dev->devtype->num_vdevs; i++) {
2301                 ret = coda_register_device(dev, i);
2302                 if (ret) {
2303                         v4l2_err(&dev->v4l2_dev,
2304                                  "Failed to register %s video device: %d\n",
2305                                  dev->devtype->vdevs[i]->name, ret);
2306                         goto rel_vfd;
2307                 }
2308         }
2309
2310         v4l2_info(&dev->v4l2_dev, "codec registered as /dev/video[%d-%d]\n",
2311                   dev->vfd[0].num, dev->vfd[i - 1].num);
2312
2313         pm_runtime_put_sync(&pdev->dev);
2314         return;
2315
2316 rel_vfd:
2317         while (--i >= 0)
2318                 video_unregister_device(&dev->vfd[i]);
2319         v4l2_m2m_release(dev->m2m_dev);
2320 put_pm:
2321         pm_runtime_put_sync(&pdev->dev);
2322 }
2323
2324 enum coda_platform {
2325         CODA_IMX27,
2326         CODA_IMX53,
2327         CODA_IMX6Q,
2328         CODA_IMX6DL,
2329 };
2330
2331 static const struct coda_devtype coda_devdata[] = {
2332         [CODA_IMX27] = {
2333                 .firmware     = {
2334                         "vpu_fw_imx27_TO2.bin",
2335                         "vpu/vpu_fw_imx27_TO2.bin",
2336                         "v4l-codadx6-imx27.bin"
2337                 },
2338                 .product      = CODA_DX6,
2339                 .codecs       = codadx6_codecs,
2340                 .num_codecs   = ARRAY_SIZE(codadx6_codecs),
2341                 .vdevs        = codadx6_video_devices,
2342                 .num_vdevs    = ARRAY_SIZE(codadx6_video_devices),
2343                 .workbuf_size = 288 * 1024 + FMO_SLICE_SAVE_BUF_SIZE * 8 * 1024,
2344                 .iram_size    = 0xb000,
2345         },
2346         [CODA_IMX53] = {
2347                 .firmware     = {
2348                         "vpu_fw_imx53.bin",
2349                         "vpu/vpu_fw_imx53.bin",
2350                         "v4l-coda7541-imx53.bin"
2351                 },
2352                 .product      = CODA_7541,
2353                 .codecs       = coda7_codecs,
2354                 .num_codecs   = ARRAY_SIZE(coda7_codecs),
2355                 .vdevs        = coda7_video_devices,
2356                 .num_vdevs    = ARRAY_SIZE(coda7_video_devices),
2357                 .workbuf_size = 128 * 1024,
2358                 .tempbuf_size = 304 * 1024,
2359                 .iram_size    = 0x14000,
2360         },
2361         [CODA_IMX6Q] = {
2362                 .firmware     = {
2363                         "vpu_fw_imx6q.bin",
2364                         "vpu/vpu_fw_imx6q.bin",
2365                         "v4l-coda960-imx6q.bin"
2366                 },
2367                 .product      = CODA_960,
2368                 .codecs       = coda9_codecs,
2369                 .num_codecs   = ARRAY_SIZE(coda9_codecs),
2370                 .vdevs        = coda9_video_devices,
2371                 .num_vdevs    = ARRAY_SIZE(coda9_video_devices),
2372                 .workbuf_size = 80 * 1024,
2373                 .tempbuf_size = 204 * 1024,
2374                 .iram_size    = 0x21000,
2375         },
2376         [CODA_IMX6DL] = {
2377                 .firmware     = {
2378                         "vpu_fw_imx6d.bin",
2379                         "vpu/vpu_fw_imx6d.bin",
2380                         "v4l-coda960-imx6dl.bin"
2381                 },
2382                 .product      = CODA_960,
2383                 .codecs       = coda9_codecs,
2384                 .num_codecs   = ARRAY_SIZE(coda9_codecs),
2385                 .vdevs        = coda9_video_devices,
2386                 .num_vdevs    = ARRAY_SIZE(coda9_video_devices),
2387                 .workbuf_size = 80 * 1024,
2388                 .tempbuf_size = 204 * 1024,
2389                 .iram_size    = 0x20000,
2390         },
2391 };
2392
2393 static struct platform_device_id coda_platform_ids[] = {
2394         { .name = "coda-imx27", .driver_data = CODA_IMX27 },
2395         { /* sentinel */ }
2396 };
2397 MODULE_DEVICE_TABLE(platform, coda_platform_ids);
2398
2399 #ifdef CONFIG_OF
2400 static const struct of_device_id coda_dt_ids[] = {
2401         { .compatible = "fsl,imx27-vpu", .data = &coda_devdata[CODA_IMX27] },
2402         { .compatible = "fsl,imx53-vpu", .data = &coda_devdata[CODA_IMX53] },
2403         { .compatible = "fsl,imx6q-vpu", .data = &coda_devdata[CODA_IMX6Q] },
2404         { .compatible = "fsl,imx6dl-vpu", .data = &coda_devdata[CODA_IMX6DL] },
2405         { /* sentinel */ }
2406 };
2407 MODULE_DEVICE_TABLE(of, coda_dt_ids);
2408 #endif
2409
2410 static int coda_probe(struct platform_device *pdev)
2411 {
2412         const struct of_device_id *of_id =
2413                         of_match_device(of_match_ptr(coda_dt_ids), &pdev->dev);
2414         const struct platform_device_id *pdev_id;
2415         struct coda_platform_data *pdata = pdev->dev.platform_data;
2416         struct device_node *np = pdev->dev.of_node;
2417         struct gen_pool *pool;
2418         struct coda_dev *dev;
2419         struct resource *res;
2420         int ret, irq;
2421
2422         dev = devm_kzalloc(&pdev->dev, sizeof(*dev), GFP_KERNEL);
2423         if (!dev)
2424                 return -ENOMEM;
2425
2426         pdev_id = of_id ? of_id->data : platform_get_device_id(pdev);
2427
2428         if (of_id)
2429                 dev->devtype = of_id->data;
2430         else if (pdev_id)
2431                 dev->devtype = &coda_devdata[pdev_id->driver_data];
2432         else
2433                 return -EINVAL;
2434
2435         spin_lock_init(&dev->irqlock);
2436         INIT_LIST_HEAD(&dev->instances);
2437
2438         dev->plat_dev = pdev;
2439         dev->clk_per = devm_clk_get(&pdev->dev, "per");
2440         if (IS_ERR(dev->clk_per)) {
2441                 dev_err(&pdev->dev, "Could not get per clock\n");
2442                 return PTR_ERR(dev->clk_per);
2443         }
2444
2445         dev->clk_ahb = devm_clk_get(&pdev->dev, "ahb");
2446         if (IS_ERR(dev->clk_ahb)) {
2447                 dev_err(&pdev->dev, "Could not get ahb clock\n");
2448                 return PTR_ERR(dev->clk_ahb);
2449         }
2450
2451         /* Get  memory for physical registers */
2452         res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
2453         dev->regs_base = devm_ioremap_resource(&pdev->dev, res);
2454         if (IS_ERR(dev->regs_base))
2455                 return PTR_ERR(dev->regs_base);
2456
2457         /* IRQ */
2458         irq = platform_get_irq_byname(pdev, "bit");
2459         if (irq < 0)
2460                 irq = platform_get_irq(pdev, 0);
2461         if (irq < 0) {
2462                 dev_err(&pdev->dev, "failed to get irq resource\n");
2463                 return irq;
2464         }
2465
2466         ret = devm_request_threaded_irq(&pdev->dev, irq, NULL, coda_irq_handler,
2467                         IRQF_ONESHOT, dev_name(&pdev->dev), dev);
2468         if (ret < 0) {
2469                 dev_err(&pdev->dev, "failed to request irq: %d\n", ret);
2470                 return ret;
2471         }
2472
2473         dev->rstc = devm_reset_control_get_optional(&pdev->dev, NULL);
2474         if (IS_ERR(dev->rstc)) {
2475                 ret = PTR_ERR(dev->rstc);
2476                 dev_err(&pdev->dev, "failed get reset control: %d\n", ret);
2477                 return ret;
2478         }
2479
2480         /* Get IRAM pool from device tree or platform data */
2481         pool = of_gen_pool_get(np, "iram", 0);
2482         if (!pool && pdata)
2483                 pool = gen_pool_get(pdata->iram_dev, NULL);
2484         if (!pool) {
2485                 dev_err(&pdev->dev, "iram pool not available\n");
2486                 return -ENOMEM;
2487         }
2488         dev->iram_pool = pool;
2489
2490         /* Get vdoa_data if supported by the platform */
2491         dev->vdoa = coda_get_vdoa_data();
2492         if (PTR_ERR(dev->vdoa) == -EPROBE_DEFER)
2493                 return -EPROBE_DEFER;
2494
2495         ret = v4l2_device_register(&pdev->dev, &dev->v4l2_dev);
2496         if (ret)
2497                 return ret;
2498
2499         mutex_init(&dev->dev_mutex);
2500         mutex_init(&dev->coda_mutex);
2501
2502         dev->debugfs_root = debugfs_create_dir("coda", NULL);
2503         if (!dev->debugfs_root)
2504                 dev_warn(&pdev->dev, "failed to create debugfs root\n");
2505
2506         /* allocate auxiliary per-device buffers for the BIT processor */
2507         if (dev->devtype->product == CODA_DX6) {
2508                 ret = coda_alloc_aux_buf(dev, &dev->workbuf,
2509                                          dev->devtype->workbuf_size, "workbuf",
2510                                          dev->debugfs_root);
2511                 if (ret < 0)
2512                         goto err_v4l2_register;
2513         }
2514
2515         if (dev->devtype->tempbuf_size) {
2516                 ret = coda_alloc_aux_buf(dev, &dev->tempbuf,
2517                                          dev->devtype->tempbuf_size, "tempbuf",
2518                                          dev->debugfs_root);
2519                 if (ret < 0)
2520                         goto err_v4l2_register;
2521         }
2522
2523         dev->iram.size = dev->devtype->iram_size;
2524         dev->iram.vaddr = gen_pool_dma_alloc(dev->iram_pool, dev->iram.size,
2525                                              &dev->iram.paddr);
2526         if (!dev->iram.vaddr) {
2527                 dev_warn(&pdev->dev, "unable to alloc iram\n");
2528         } else {
2529                 memset(dev->iram.vaddr, 0, dev->iram.size);
2530                 dev->iram.blob.data = dev->iram.vaddr;
2531                 dev->iram.blob.size = dev->iram.size;
2532                 dev->iram.dentry = debugfs_create_blob("iram", 0644,
2533                                                        dev->debugfs_root,
2534                                                        &dev->iram.blob);
2535         }
2536
2537         dev->workqueue = alloc_workqueue("coda", WQ_UNBOUND | WQ_MEM_RECLAIM, 1);
2538         if (!dev->workqueue) {
2539                 dev_err(&pdev->dev, "unable to alloc workqueue\n");
2540                 ret = -ENOMEM;
2541                 goto err_v4l2_register;
2542         }
2543
2544         platform_set_drvdata(pdev, dev);
2545
2546         /*
2547          * Start activated so we can directly call coda_hw_init in
2548          * coda_fw_callback regardless of whether CONFIG_PM is
2549          * enabled or whether the device is associated with a PM domain.
2550          */
2551         pm_runtime_get_noresume(&pdev->dev);
2552         pm_runtime_set_active(&pdev->dev);
2553         pm_runtime_enable(&pdev->dev);
2554
2555         ret = coda_firmware_request(dev);
2556         if (ret)
2557                 goto err_alloc_workqueue;
2558         return 0;
2559
2560 err_alloc_workqueue:
2561         destroy_workqueue(dev->workqueue);
2562 err_v4l2_register:
2563         v4l2_device_unregister(&dev->v4l2_dev);
2564         return ret;
2565 }
2566
2567 static int coda_remove(struct platform_device *pdev)
2568 {
2569         struct coda_dev *dev = platform_get_drvdata(pdev);
2570         int i;
2571
2572         for (i = 0; i < ARRAY_SIZE(dev->vfd); i++) {
2573                 if (video_get_drvdata(&dev->vfd[i]))
2574                         video_unregister_device(&dev->vfd[i]);
2575         }
2576         if (dev->m2m_dev)
2577                 v4l2_m2m_release(dev->m2m_dev);
2578         pm_runtime_disable(&pdev->dev);
2579         v4l2_device_unregister(&dev->v4l2_dev);
2580         destroy_workqueue(dev->workqueue);
2581         if (dev->iram.vaddr)
2582                 gen_pool_free(dev->iram_pool, (unsigned long)dev->iram.vaddr,
2583                               dev->iram.size);
2584         coda_free_aux_buf(dev, &dev->codebuf);
2585         coda_free_aux_buf(dev, &dev->tempbuf);
2586         coda_free_aux_buf(dev, &dev->workbuf);
2587         debugfs_remove_recursive(dev->debugfs_root);
2588         return 0;
2589 }
2590
2591 #ifdef CONFIG_PM
2592 static int coda_runtime_resume(struct device *dev)
2593 {
2594         struct coda_dev *cdev = dev_get_drvdata(dev);
2595         int ret = 0;
2596
2597         if (dev->pm_domain && cdev->codebuf.vaddr) {
2598                 ret = coda_hw_init(cdev);
2599                 if (ret)
2600                         v4l2_err(&cdev->v4l2_dev, "HW initialization failed\n");
2601         }
2602
2603         return ret;
2604 }
2605 #endif
2606
2607 static const struct dev_pm_ops coda_pm_ops = {
2608         SET_RUNTIME_PM_OPS(NULL, coda_runtime_resume, NULL)
2609 };
2610
2611 static struct platform_driver coda_driver = {
2612         .probe  = coda_probe,
2613         .remove = coda_remove,
2614         .driver = {
2615                 .name   = CODA_NAME,
2616                 .of_match_table = of_match_ptr(coda_dt_ids),
2617                 .pm     = &coda_pm_ops,
2618         },
2619         .id_table = coda_platform_ids,
2620 };
2621
2622 module_platform_driver(coda_driver);
2623
2624 MODULE_LICENSE("GPL");
2625 MODULE_AUTHOR("Javier Martin <javier.martin@vista-silicon.com>");
2626 MODULE_DESCRIPTION("Coda multi-standard codec V4L2 driver");