drivers/staging/media/solo6x10/solo6x10-v4l2-enc.c

   1 /*
   2  * Copyright (C) 2010-2013 Bluecherry, LLC <http://www.bluecherrydvr.com>
   3  *
   4  * Original author:
   5  * Ben Collins <bcollins@ubuntu.com>
   6  *
   7  * Additional work by:
   8  * John Brooks <john.brooks@bluecherry.net>
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  23  */
  24
  25 #include <linux/kernel.h>
  26 #include <linux/module.h>
  27 #include <linux/kthread.h>
  28 #include <linux/freezer.h>
  29
  30 #include <media/v4l2-ioctl.h>
  31 #include <media/v4l2-common.h>
  32 #include <media/v4l2-event.h>
  33 #include <media/videobuf2-dma-sg.h>
  34
  35 #include "solo6x10.h"
  36 #include "solo6x10-tw28.h"
  37 #include "solo6x10-jpeg.h"
  38
  39 #define MIN_VID_BUFFERS         2
  40 #define FRAME_BUF_SIZE          (196 * 1024)
  41 #define MP4_QS                  16
  42 #define DMA_ALIGN               4096
  43
  44 /* 6010 M4V */
  45 static unsigned char vop_6010_ntsc_d1[] = {
  46         0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x20,
  47         0x02, 0x48, 0x1d, 0xc0, 0x00, 0x40, 0x00, 0x40,
  48         0x00, 0x40, 0x00, 0x80, 0x00, 0x97, 0x53, 0x04,
  49         0x1f, 0x4c, 0x58, 0x10, 0xf0, 0x71, 0x18, 0x3f,
  50 };
  51
  52 static unsigned char vop_6010_ntsc_cif[] = {
  53         0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x20,
  54         0x02, 0x48, 0x1d, 0xc0, 0x00, 0x40, 0x00, 0x40,
  55         0x00, 0x40, 0x00, 0x80, 0x00, 0x97, 0x53, 0x04,
  56         0x1f, 0x4c, 0x2c, 0x10, 0x78, 0x51, 0x18, 0x3f,
  57 };
  58
  59 static unsigned char vop_6010_pal_d1[] = {
  60         0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x20,
  61         0x02, 0x48, 0x15, 0xc0, 0x00, 0x40, 0x00, 0x40,
  62         0x00, 0x40, 0x00, 0x80, 0x00, 0x97, 0x53, 0x04,
  63         0x1f, 0x4c, 0x58, 0x11, 0x20, 0x71, 0x18, 0x3f,
  64 };
  65
  66 static unsigned char vop_6010_pal_cif[] = {
  67         0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01, 0x20,
  68         0x02, 0x48, 0x15, 0xc0, 0x00, 0x40, 0x00, 0x40,
  69         0x00, 0x40, 0x00, 0x80, 0x00, 0x97, 0x53, 0x04,
  70         0x1f, 0x4c, 0x2c, 0x10, 0x90, 0x51, 0x18, 0x3f,
  71 };
  72
  73 /* 6110 h.264 */
  74 static unsigned char vop_6110_ntsc_d1[] = {
  75         0x00, 0x00, 0x00, 0x01, 0x67, 0x42, 0x00, 0x1e,
  76         0x9a, 0x74, 0x05, 0x81, 0xec, 0x80, 0x00, 0x00,
  77         0x00, 0x01, 0x68, 0xce, 0x32, 0x28, 0x00, 0x00,
  78 };
  79
  80 static unsigned char vop_6110_ntsc_cif[] = {
  81         0x00, 0x00, 0x00, 0x01, 0x67, 0x42, 0x00, 0x1e,
  82         0x9a, 0x74, 0x0b, 0x0f, 0xc8, 0x00, 0x00, 0x00,
  83         0x01, 0x68, 0xce, 0x32, 0x28, 0x00, 0x00, 0x00,
  84 };
  85
  86 static unsigned char vop_6110_pal_d1[] = {
  87         0x00, 0x00, 0x00, 0x01, 0x67, 0x42, 0x00, 0x1e,
  88         0x9a, 0x74, 0x05, 0x80, 0x93, 0x20, 0x00, 0x00,
  89         0x00, 0x01, 0x68, 0xce, 0x32, 0x28, 0x00, 0x00,
  90 };
  91
  92 static unsigned char vop_6110_pal_cif[] = {
  93         0x00, 0x00, 0x00, 0x01, 0x67, 0x42, 0x00, 0x1e,
  94         0x9a, 0x74, 0x0b, 0x04, 0xb2, 0x00, 0x00, 0x00,
  95         0x01, 0x68, 0xce, 0x32, 0x28, 0x00, 0x00, 0x00,
  96 };
  97
  98 struct vop_header {
  99         /* VE_STATUS0 */
 100         u32 mpeg_size:20, sad_motion_flag:1, video_motion_flag:1, vop_type:2,
 101                 channel:5, source_fl:1, interlace:1, progressive:1;
 102
 103         /* VE_STATUS1 */
 104         u32 vsize:8, hsize:8, last_queue:4, nop0:8, scale:4;
 105
 106         /* VE_STATUS2 */
 107         u32 mpeg_off;
 108
 109         /* VE_STATUS3 */
 110         u32 jpeg_off;
 111
 112         /* VE_STATUS4 */
 113         u32 jpeg_size:20, interval:10, nop1:2;
 114
 115         /* VE_STATUS5/6 */
 116         u32 sec, usec;
 117
 118         /* VE_STATUS7/8/9 */
 119         u32 nop2[3];
 120
 121         /* VE_STATUS10 */
 122         u32 mpeg_size_alt:20, nop3:12;
 123
 124         u32 end_nops[5];
 125 } __packed;
 126
 127 struct solo_enc_buf {
 128         enum solo_enc_types     type;
 129         struct vop_header       *vh;
 130         int                     motion;
 131 };
 132
 133 static int solo_is_motion_on(struct solo_enc_dev *solo_enc)
 134 {
 135         struct solo_dev *solo_dev = solo_enc->solo_dev;
 136
 137         return (solo_dev->motion_mask >> solo_enc->ch) & 1;
 138 }
 139
 140 static int solo_motion_detected(struct solo_enc_dev *solo_enc)
 141 {
 142         struct solo_dev *solo_dev = solo_enc->solo_dev;
 143         unsigned long flags;
 144         u32 ch_mask = 1 << solo_enc->ch;
 145         int ret = 0;
 146
 147         spin_lock_irqsave(&solo_enc->motion_lock, flags);
 148         if (solo_reg_read(solo_dev, SOLO_VI_MOT_STATUS) & ch_mask) {
 149                 solo_reg_write(solo_dev, SOLO_VI_MOT_CLEAR, ch_mask);
 150                 ret = 1;
 151         }
 152         spin_unlock_irqrestore(&solo_enc->motion_lock, flags);
 153
 154         return ret;
 155 }
 156
 157 static void solo_motion_toggle(struct solo_enc_dev *solo_enc, int on)
 158 {
 159         struct solo_dev *solo_dev = solo_enc->solo_dev;
 160         u32 mask = 1 << solo_enc->ch;
 161         unsigned long flags;
 162
 163         spin_lock_irqsave(&solo_enc->motion_lock, flags);
 164
 165         if (on)
 166                 solo_dev->motion_mask |= mask;
 167         else
 168                 solo_dev->motion_mask &= ~mask;
 169
 170         solo_reg_write(solo_dev, SOLO_VI_MOT_CLEAR, mask);
 171
 172         solo_reg_write(solo_dev, SOLO_VI_MOT_ADR,
 173                        SOLO_VI_MOTION_EN(solo_dev->motion_mask) |
 174                        (SOLO_MOTION_EXT_ADDR(solo_dev) >> 16));
 175
 176         spin_unlock_irqrestore(&solo_enc->motion_lock, flags);
 177 }
 178
 179 void solo_update_mode(struct solo_enc_dev *solo_enc)
 180 {
 181         struct solo_dev *solo_dev = solo_enc->solo_dev;
 182         int vop_len;
 183         unsigned char *vop;
 184
 185         solo_enc->interlaced = (solo_enc->mode & 0x08) ? 1 : 0;
 186         solo_enc->bw_weight = max(solo_dev->fps / solo_enc->interval, 1);
 187
 188         if (solo_enc->mode == SOLO_ENC_MODE_CIF) {
 189                 solo_enc->width = solo_dev->video_hsize >> 1;
 190                 solo_enc->height = solo_dev->video_vsize;
 191                 if (solo_dev->type == SOLO_DEV_6110) {
 192                         if (solo_dev->video_type == SOLO_VO_FMT_TYPE_NTSC) {
 193                                 vop = vop_6110_ntsc_cif;
 194                                 vop_len = sizeof(vop_6110_ntsc_cif);
 195                         } else {
 196                                 vop = vop_6110_pal_cif;
 197                                 vop_len = sizeof(vop_6110_pal_cif);
 198                         }
 199                 } else {
 200                         if (solo_dev->video_type == SOLO_VO_FMT_TYPE_NTSC) {
 201                                 vop = vop_6010_ntsc_cif;
 202                                 vop_len = sizeof(vop_6010_ntsc_cif);
 203                         } else {
 204                                 vop = vop_6010_pal_cif;
 205                                 vop_len = sizeof(vop_6010_pal_cif);
 206                         }
 207                 }
 208         } else {
 209                 solo_enc->width = solo_dev->video_hsize;
 210                 solo_enc->height = solo_dev->video_vsize << 1;
 211                 solo_enc->bw_weight <<= 2;
 212                 if (solo_dev->type == SOLO_DEV_6110) {
 213                         if (solo_dev->video_type == SOLO_VO_FMT_TYPE_NTSC) {
 214                                 vop = vop_6110_ntsc_d1;
 215                                 vop_len = sizeof(vop_6110_ntsc_d1);
 216                         } else {
 217                                 vop = vop_6110_pal_d1;
 218                                 vop_len = sizeof(vop_6110_pal_d1);
 219                         }
 220                 } else {
 221                         if (solo_dev->video_type == SOLO_VO_FMT_TYPE_NTSC) {
 222                                 vop = vop_6010_ntsc_d1;
 223                                 vop_len = sizeof(vop_6010_ntsc_d1);
 224                         } else {
 225                                 vop = vop_6010_pal_d1;
 226                                 vop_len = sizeof(vop_6010_pal_d1);
 227                         }
 228                 }
 229         }
 230
 231         memcpy(solo_enc->vop, vop, vop_len);
 232
 233         /* Some fixups for 6010/M4V */
 234         if (solo_dev->type == SOLO_DEV_6010) {
 235                 u16 fps = solo_dev->fps * 1000;
 236                 u16 interval = solo_enc->interval * 1000;
 237
 238                 vop = solo_enc->vop;
 239
 240                 /* Frame rate and interval */
 241                 vop[22] = fps >> 4;
 242                 vop[23] = ((fps << 4) & 0xf0) | 0x0c
 243                         | ((interval >> 13) & 0x3);
 244                 vop[24] = (interval >> 5) & 0xff;
 245                 vop[25] = ((interval << 3) & 0xf8) | 0x04;
 246         }
 247
 248         solo_enc->vop_len = vop_len;
 249
 250         /* Now handle the jpeg header */
 251         vop = solo_enc->jpeg_header;
 252         vop[SOF0_START + 5] = 0xff & (solo_enc->height >> 8);
 253         vop[SOF0_START + 6] = 0xff & solo_enc->height;
 254         vop[SOF0_START + 7] = 0xff & (solo_enc->width >> 8);
 255         vop[SOF0_START + 8] = 0xff & solo_enc->width;
 256
 257         memcpy(vop + DQT_START,
 258                jpeg_dqt[solo_g_jpeg_qp(solo_dev, solo_enc->ch)], DQT_LEN);
 259 }
 260
 261 static int solo_enc_on(struct solo_enc_dev *solo_enc)
 262 {
 263         u8 ch = solo_enc->ch;
 264         struct solo_dev *solo_dev = solo_enc->solo_dev;
 265         u8 interval;
 266
 267         solo_update_mode(solo_enc);
 268
 269         /* Make sure to do a bandwidth check */
 270         if (solo_enc->bw_weight > solo_dev->enc_bw_remain)
 271                 return -EBUSY;
 272         solo_enc->sequence = 0;
 273         solo_dev->enc_bw_remain -= solo_enc->bw_weight;
 274
 275         if (solo_enc->type == SOLO_ENC_TYPE_EXT)
 276                 solo_reg_write(solo_dev, SOLO_CAP_CH_COMP_ENA_E(ch), 1);
 277
 278         /* Disable all encoding for this channel */
 279         solo_reg_write(solo_dev, SOLO_CAP_CH_SCALE(ch), 0);
 280
 281         /* Common for both std and ext encoding */
 282         solo_reg_write(solo_dev, SOLO_VE_CH_INTL(ch),
 283                        solo_enc->interlaced ? 1 : 0);
 284
 285         if (solo_enc->interlaced)
 286                 interval = solo_enc->interval - 1;
 287         else
 288                 interval = solo_enc->interval;
 289
 290         /* Standard encoding only */
 291         solo_reg_write(solo_dev, SOLO_VE_CH_GOP(ch), solo_enc->gop);
 292         solo_reg_write(solo_dev, SOLO_VE_CH_QP(ch), solo_enc->qp);
 293         solo_reg_write(solo_dev, SOLO_CAP_CH_INTV(ch), interval);
 294
 295         /* Extended encoding only */
 296         solo_reg_write(solo_dev, SOLO_VE_CH_GOP_E(ch), solo_enc->gop);
 297         solo_reg_write(solo_dev, SOLO_VE_CH_QP_E(ch), solo_enc->qp);
 298         solo_reg_write(solo_dev, SOLO_CAP_CH_INTV_E(ch), interval);
 299
 300         /* Enables the standard encoder */
 301         solo_reg_write(solo_dev, SOLO_CAP_CH_SCALE(ch), solo_enc->mode);
 302
 303         return 0;
 304 }
 305
 306 static void solo_enc_off(struct solo_enc_dev *solo_enc)
 307 {
 308         struct solo_dev *solo_dev = solo_enc->solo_dev;
 309
 310         solo_dev->enc_bw_remain += solo_enc->bw_weight;
 311
 312         solo_reg_write(solo_dev, SOLO_CAP_CH_SCALE(solo_enc->ch), 0);
 313         solo_reg_write(solo_dev, SOLO_CAP_CH_COMP_ENA_E(solo_enc->ch), 0);
 314 }
 315
 316 static int enc_get_mpeg_dma(struct solo_dev *solo_dev, dma_addr_t dma,
 317                               unsigned int off, unsigned int size)
 318 {
 319         int ret;
 320
 321         if (off > SOLO_MP4E_EXT_SIZE(solo_dev))
 322                 return -EINVAL;
 323
 324         /* Single shot */
 325         if (off + size <= SOLO_MP4E_EXT_SIZE(solo_dev)) {
 326                 return solo_p2m_dma_t(solo_dev, 0, dma,
 327                                       SOLO_MP4E_EXT_ADDR(solo_dev) + off, size,
 328                                       0, 0);
 329         }
 330
 331         /* Buffer wrap */
 332         ret = solo_p2m_dma_t(solo_dev, 0, dma,
 333                              SOLO_MP4E_EXT_ADDR(solo_dev) + off,
 334                              SOLO_MP4E_EXT_SIZE(solo_dev) - off, 0, 0);
 335
 336         if (!ret) {
 337                 ret = solo_p2m_dma_t(solo_dev, 0,
 338                              dma + SOLO_MP4E_EXT_SIZE(solo_dev) - off,
 339                              SOLO_MP4E_EXT_ADDR(solo_dev),
 340                              size + off - SOLO_MP4E_EXT_SIZE(solo_dev), 0, 0);
 341         }
 342
 343         return ret;
 344 }
 345
 346 /* Build a descriptor queue out of an SG list and send it to the P2M for
 347  * processing. */
 348 static int solo_send_desc(struct solo_enc_dev *solo_enc, int skip,
 349                           struct vb2_dma_sg_desc *vbuf, int off, int size,
 350                           unsigned int base, unsigned int base_size)
 351 {
 352         struct solo_dev *solo_dev = solo_enc->solo_dev;
 353         struct scatterlist *sg;
 354         int i;
 355         int ret;
 356
 357         if (WARN_ON_ONCE(size > FRAME_BUF_SIZE))
 358                 return -EINVAL;
 359
 360         solo_enc->desc_count = 1;
 361
 362         for_each_sg(vbuf->sglist, sg, vbuf->num_pages, i) {
 363                 struct solo_p2m_desc *desc;
 364                 dma_addr_t dma;
 365                 int len;
 366                 int left = base_size - off;
 367
 368                 desc = &solo_enc->desc_items[solo_enc->desc_count++];
 369                 dma = sg_dma_address(sg);
 370                 len = sg_dma_len(sg);
 371
 372                 /* We assume this is smaller than the scatter size */
 373                 BUG_ON(skip >= len);
 374                 if (skip) {
 375                         len -= skip;
 376                         dma += skip;
 377                         size -= skip;
 378                         skip = 0;
 379                 }
 380
 381                 len = min(len, size);
 382
 383                 if (len <= left) {
 384                         /* Single descriptor */
 385                         solo_p2m_fill_desc(desc, 0, dma, base + off,
 386                                            len, 0, 0);
 387                 } else {
 388                         /* Buffer wrap */
 389                         /* XXX: Do these as separate DMA requests, to avoid
 390                            timeout errors triggered by awkwardly sized
 391                            descriptors. See
 392                            <https://github.com/bluecherrydvr/solo6x10/issues/8>
 393                          */
 394                         ret = solo_p2m_dma_t(solo_dev, 0, dma, base + off,
 395                                              left, 0, 0);
 396                         if (ret)
 397                                 return ret;
 398
 399                         ret = solo_p2m_dma_t(solo_dev, 0, dma + left, base,
 400                                              len - left, 0, 0);
 401                         if (ret)
 402                                 return ret;
 403
 404                         solo_enc->desc_count--;
 405                 }
 406
 407                 size -= len;
 408                 if (size <= 0)
 409                         break;
 410
 411                 off += len;
 412                 if (off >= base_size)
 413                         off -= base_size;
 414
 415                 /* Because we may use two descriptors per loop */
 416                 if (solo_enc->desc_count >= (solo_enc->desc_nelts - 1)) {
 417                         ret = solo_p2m_dma_desc(solo_dev, solo_enc->desc_items,
 418                                                 solo_enc->desc_dma,
 419                                                 solo_enc->desc_count - 1);
 420                         if (ret)
 421                                 return ret;
 422                         solo_enc->desc_count = 1;
 423                 }
 424         }
 425
 426         if (solo_enc->desc_count <= 1)
 427                 return 0;
 428
 429         return solo_p2m_dma_desc(solo_dev, solo_enc->desc_items, solo_enc->desc_dma,
 430                                  solo_enc->desc_count - 1);
 431 }
 432
 433 static int solo_fill_jpeg(struct solo_enc_dev *solo_enc,
 434                 struct vb2_buffer *vb, struct vop_header *vh)
 435 {
 436         struct solo_dev *solo_dev = solo_enc->solo_dev;
 437         struct vb2_dma_sg_desc *vbuf = vb2_dma_sg_plane_desc(vb, 0);
 438         int frame_size;
 439         int ret;
 440
 441         vb->v4l2_buf.flags |= V4L2_BUF_FLAG_KEYFRAME;
 442
 443         if (vb2_plane_size(vb, 0) < vh->jpeg_size + solo_enc->jpeg_len)
 444                 return -EIO;
 445
 446         sg_copy_from_buffer(vbuf->sglist, vbuf->num_pages,
 447                         solo_enc->jpeg_header,
 448                         solo_enc->jpeg_len);
 449
 450         frame_size = (vh->jpeg_size + solo_enc->jpeg_len + (DMA_ALIGN - 1))
 451                 & ~(DMA_ALIGN - 1);
 452         vb2_set_plane_payload(vb, 0, vh->jpeg_size + solo_enc->jpeg_len);
 453
 454         dma_map_sg(&solo_dev->pdev->dev, vbuf->sglist, vbuf->num_pages,
 455                         DMA_FROM_DEVICE);
 456         ret = solo_send_desc(solo_enc, solo_enc->jpeg_len, vbuf, vh->jpeg_off,
 457                         frame_size, SOLO_JPEG_EXT_ADDR(solo_dev),
 458                         SOLO_JPEG_EXT_SIZE(solo_dev));
 459         dma_unmap_sg(&solo_dev->pdev->dev, vbuf->sglist, vbuf->num_pages,
 460                         DMA_FROM_DEVICE);
 461         return ret;
 462 }
 463
 464 static int solo_fill_mpeg(struct solo_enc_dev *solo_enc,
 465                 struct vb2_buffer *vb, struct vop_header *vh)
 466 {
 467         struct solo_dev *solo_dev = solo_enc->solo_dev;
 468         struct vb2_dma_sg_desc *vbuf = vb2_dma_sg_plane_desc(vb, 0);
 469         int frame_off, frame_size;
 470         int skip = 0;
 471         int ret;
 472
 473         if (vb2_plane_size(vb, 0) < vh->mpeg_size)
 474                 return -EIO;
 475
 476         /* If this is a key frame, add extra header */
 477         if (!vh->vop_type) {
 478                 sg_copy_from_buffer(vbuf->sglist, vbuf->num_pages,
 479                                 solo_enc->vop,
 480                                 solo_enc->vop_len);
 481
 482                 skip = solo_enc->vop_len;
 483
 484                 vb->v4l2_buf.flags |= V4L2_BUF_FLAG_KEYFRAME;
 485                 vb2_set_plane_payload(vb, 0, vh->mpeg_size + solo_enc->vop_len);
 486         } else {
 487                 vb->v4l2_buf.flags |= V4L2_BUF_FLAG_PFRAME;
 488                 vb2_set_plane_payload(vb, 0, vh->mpeg_size);
 489         }
 490
 491         /* Now get the actual mpeg payload */
 492         frame_off = (vh->mpeg_off + sizeof(*vh))
 493                 % SOLO_MP4E_EXT_SIZE(solo_dev);
 494         frame_size = (vh->mpeg_size + skip + (DMA_ALIGN - 1))
 495                 & ~(DMA_ALIGN - 1);
 496
 497         dma_map_sg(&solo_dev->pdev->dev, vbuf->sglist, vbuf->num_pages,
 498                         DMA_FROM_DEVICE);
 499         ret = solo_send_desc(solo_enc, skip, vbuf, frame_off, frame_size,
 500                         SOLO_MP4E_EXT_ADDR(solo_dev),
 501                         SOLO_MP4E_EXT_SIZE(solo_dev));
 502         dma_unmap_sg(&solo_dev->pdev->dev, vbuf->sglist, vbuf->num_pages,
 503                         DMA_FROM_DEVICE);
 504         return ret;
 505 }
 506
 507 static int solo_enc_fillbuf(struct solo_enc_dev *solo_enc,
 508                             struct vb2_buffer *vb, struct solo_enc_buf *enc_buf)
 509 {
 510         struct vop_header *vh = enc_buf->vh;
 511         int ret;
 512
 513         /* Check for motion flags */
 514         vb->v4l2_buf.flags &= ~(V4L2_BUF_FLAG_MOTION_ON |
 515                                 V4L2_BUF_FLAG_MOTION_DETECTED);
 516         if (solo_is_motion_on(solo_enc)) {
 517                 vb->v4l2_buf.flags |= V4L2_BUF_FLAG_MOTION_ON;
 518                 if (enc_buf->motion)
 519                         vb->v4l2_buf.flags |= V4L2_BUF_FLAG_MOTION_DETECTED;
 520         }
 521
 522         switch (solo_enc->fmt) {
 523         case V4L2_PIX_FMT_MPEG4:
 524         case V4L2_PIX_FMT_H264:
 525                 ret = solo_fill_mpeg(solo_enc, vb, vh);
 526                 break;
 527         default: /* V4L2_PIX_FMT_MJPEG */
 528                 ret = solo_fill_jpeg(solo_enc, vb, vh);
 529                 break;
 530         }
 531
 532         if (!ret) {
 533                 vb->v4l2_buf.sequence = solo_enc->sequence++;
 534                 vb->v4l2_buf.timestamp.tv_sec = vh->sec;
 535                 vb->v4l2_buf.timestamp.tv_usec = vh->usec;
 536         }
 537
 538         vb2_buffer_done(vb, ret ? VB2_BUF_STATE_ERROR : VB2_BUF_STATE_DONE);
 539
 540         return ret;
 541 }
 542
 543 static void solo_enc_handle_one(struct solo_enc_dev *solo_enc,
 544                                 struct solo_enc_buf *enc_buf)
 545 {
 546         struct solo_vb2_buf *vb;
 547         unsigned long flags;
 548
 549         mutex_lock(&solo_enc->lock);
 550         if (solo_enc->type != enc_buf->type)
 551                 goto unlock;
 552
 553         spin_lock_irqsave(&solo_enc->av_lock, flags);
 554         if (list_empty(&solo_enc->vidq_active)) {
 555                 spin_unlock_irqrestore(&solo_enc->av_lock, flags);
 556                 goto unlock;
 557         }
 558         vb = list_first_entry(&solo_enc->vidq_active, struct solo_vb2_buf, list);
 559         list_del(&vb->list);
 560         spin_unlock_irqrestore(&solo_enc->av_lock, flags);
 561
 562         solo_enc_fillbuf(solo_enc, &vb->vb, enc_buf);
 563 unlock:
 564         mutex_unlock(&solo_enc->lock);
 565 }
 566
 567 void solo_enc_v4l2_isr(struct solo_dev *solo_dev)
 568 {
 569         wake_up_interruptible_all(&solo_dev->ring_thread_wait);
 570 }
 571
 572 static void solo_handle_ring(struct solo_dev *solo_dev)
 573 {
 574         for (;;) {
 575                 struct solo_enc_dev *solo_enc;
 576                 struct solo_enc_buf enc_buf;
 577                 u32 mpeg_current, off;
 578                 u8 ch;
 579                 u8 cur_q;
 580
 581                 /* Check if the hardware has any new ones in the queue */
 582                 cur_q = solo_reg_read(solo_dev, SOLO_VE_STATE(11)) & 0xff;
 583                 if (cur_q == solo_dev->enc_idx)
 584                         break;
 585
 586                 mpeg_current = solo_reg_read(solo_dev,
 587                                         SOLO_VE_MPEG4_QUE(solo_dev->enc_idx));
 588                 solo_dev->enc_idx = (solo_dev->enc_idx + 1) % MP4_QS;
 589
 590                 ch = (mpeg_current >> 24) & 0x1f;
 591                 off = mpeg_current & 0x00ffffff;
 592
 593                 if (ch >= SOLO_MAX_CHANNELS) {
 594                         ch -= SOLO_MAX_CHANNELS;
 595                         enc_buf.type = SOLO_ENC_TYPE_EXT;
 596                 } else
 597                         enc_buf.type = SOLO_ENC_TYPE_STD;
 598
 599                 solo_enc = solo_dev->v4l2_enc[ch];
 600                 if (solo_enc == NULL) {
 601                         dev_err(&solo_dev->pdev->dev,
 602                                 "Got spurious packet for channel %d\n", ch);
 603                         continue;
 604                 }
 605
 606                 /* FAIL... */
 607                 if (enc_get_mpeg_dma(solo_dev, solo_dev->vh_dma, off,
 608                                      sizeof(struct vop_header)))
 609                         continue;
 610
 611                 enc_buf.vh = (struct vop_header *)solo_dev->vh_buf;
 612                 enc_buf.vh->mpeg_off -= SOLO_MP4E_EXT_ADDR(solo_dev);
 613                 enc_buf.vh->jpeg_off -= SOLO_JPEG_EXT_ADDR(solo_dev);
 614
 615                 /* Sanity check */
 616                 if (enc_buf.vh->mpeg_off != off)
 617                         continue;
 618
 619                 if (solo_motion_detected(solo_enc))
 620                         enc_buf.motion = 1;
 621                 else
 622                         enc_buf.motion = 0;
 623
 624                 solo_enc_handle_one(solo_enc, &enc_buf);
 625         }
 626 }
 627
 628 static int solo_ring_thread(void *data)
 629 {
 630         struct solo_dev *solo_dev = data;
 631         DECLARE_WAITQUEUE(wait, current);
 632
 633         set_freezable();
 634         add_wait_queue(&solo_dev->ring_thread_wait, &wait);
 635
 636         for (;;) {
 637                 long timeout = schedule_timeout_interruptible(HZ);
 638                 if (timeout == -ERESTARTSYS || kthread_should_stop())
 639                         break;
 640                 solo_irq_off(solo_dev, SOLO_IRQ_ENCODER);
 641                 solo_handle_ring(solo_dev);
 642                 solo_irq_on(solo_dev, SOLO_IRQ_ENCODER);
 643                 try_to_freeze();
 644         }
 645
 646         remove_wait_queue(&solo_dev->ring_thread_wait, &wait);
 647
 648         return 0;
 649 }
 650
 651 static int solo_enc_queue_setup(struct vb2_queue *q, const struct v4l2_format *fmt,
 652                            unsigned int *num_buffers, unsigned int *num_planes,
 653                            unsigned int sizes[], void *alloc_ctxs[])
 654 {
 655         sizes[0] = FRAME_BUF_SIZE;
 656         *num_planes = 1;
 657
 658         if (*num_buffers < MIN_VID_BUFFERS)
 659                 *num_buffers = MIN_VID_BUFFERS;
 660
 661         return 0;
 662 }
 663
 664 static void solo_enc_buf_queue(struct vb2_buffer *vb)
 665 {
 666         struct vb2_queue *vq = vb->vb2_queue;
 667         struct solo_enc_dev *solo_enc = vb2_get_drv_priv(vq);
 668         struct solo_vb2_buf *solo_vb =
 669                 container_of(vb, struct solo_vb2_buf, vb);
 670
 671         spin_lock(&solo_enc->av_lock);
 672         list_add_tail(&solo_vb->list, &solo_enc->vidq_active);
 673         spin_unlock(&solo_enc->av_lock);
 674 }
 675
 676 static int solo_ring_start(struct solo_dev *solo_dev)
 677 {
 678         solo_dev->ring_thread = kthread_run(solo_ring_thread, solo_dev,
 679                                             SOLO6X10_NAME "_ring");
 680         if (IS_ERR(solo_dev->ring_thread)) {
 681                 int err = PTR_ERR(solo_dev->ring_thread);
 682                 solo_dev->ring_thread = NULL;
 683                 return err;
 684         }
 685
 686         solo_irq_on(solo_dev, SOLO_IRQ_ENCODER);
 687
 688         return 0;
 689 }
 690
 691 static void solo_ring_stop(struct solo_dev *solo_dev)
 692 {
 693         if (solo_dev->ring_thread) {
 694                 kthread_stop(solo_dev->ring_thread);
 695                 solo_dev->ring_thread = NULL;
 696         }
 697
 698         solo_irq_off(solo_dev, SOLO_IRQ_ENCODER);
 699 }
 700
 701 static int solo_enc_start_streaming(struct vb2_queue *q, unsigned int count)
 702 {
 703         struct solo_enc_dev *solo_enc = vb2_get_drv_priv(q);
 704         int ret;
 705
 706         ret = solo_enc_on(solo_enc);
 707         if (ret)
 708                 return ret;
 709         return solo_ring_start(solo_enc->solo_dev);
 710 }
 711
 712 static int solo_enc_stop_streaming(struct vb2_queue *q)
 713 {
 714         struct solo_enc_dev *solo_enc = vb2_get_drv_priv(q);
 715
 716         solo_enc_off(solo_enc);
 717         INIT_LIST_HEAD(&solo_enc->vidq_active);
 718         solo_ring_stop(solo_enc->solo_dev);
 719         return 0;
 720 }
 721
 722 static struct vb2_ops solo_enc_video_qops = {
 723         .queue_setup    = solo_enc_queue_setup,
 724         .buf_queue      = solo_enc_buf_queue,
 725         .start_streaming = solo_enc_start_streaming,
 726         .stop_streaming = solo_enc_stop_streaming,
 727         .wait_prepare   = vb2_ops_wait_prepare,
 728         .wait_finish    = vb2_ops_wait_finish,
 729 };
 730
 731 static int solo_enc_querycap(struct file *file, void  *priv,
 732                              struct v4l2_capability *cap)
 733 {
 734         struct solo_enc_dev *solo_enc = video_drvdata(file);
 735         struct solo_dev *solo_dev = solo_enc->solo_dev;
 736
 737         strcpy(cap->driver, SOLO6X10_NAME);
 738         snprintf(cap->card, sizeof(cap->card), "Softlogic 6x10 Enc %d",
 739                  solo_enc->ch);
 740         snprintf(cap->bus_info, sizeof(cap->bus_info), "PCI:%s",
 741                  pci_name(solo_dev->pdev));
 742         cap->device_caps = V4L2_CAP_VIDEO_CAPTURE |
 743                         V4L2_CAP_READWRITE | V4L2_CAP_STREAMING;
 744         cap->capabilities = cap->device_caps | V4L2_CAP_DEVICE_CAPS;
 745         return 0;
 746 }
 747
 748 static int solo_enc_enum_input(struct file *file, void *priv,
 749                                struct v4l2_input *input)
 750 {
 751         struct solo_enc_dev *solo_enc = video_drvdata(file);
 752         struct solo_dev *solo_dev = solo_enc->solo_dev;
 753
 754         if (input->index)
 755                 return -EINVAL;
 756
 757         snprintf(input->name, sizeof(input->name), "Encoder %d",
 758                  solo_enc->ch + 1);
 759         input->type = V4L2_INPUT_TYPE_CAMERA;
 760         input->std = solo_enc->vfd->tvnorms;
 761
 762         if (!tw28_get_video_status(solo_dev, solo_enc->ch))
 763                 input->status = V4L2_IN_ST_NO_SIGNAL;
 764
 765         return 0;
 766 }
 767
 768 static int solo_enc_set_input(struct file *file, void *priv,
 769                               unsigned int index)
 770 {
 771         if (index)
 772                 return -EINVAL;
 773
 774         return 0;
 775 }
 776
 777 static int solo_enc_get_input(struct file *file, void *priv,
 778                               unsigned int *index)
 779 {
 780         *index = 0;
 781
 782         return 0;
 783 }
 784
 785 static int solo_enc_enum_fmt_cap(struct file *file, void *priv,
 786                                  struct v4l2_fmtdesc *f)
 787 {
 788         struct solo_enc_dev *solo_enc = video_drvdata(file);
 789         int dev_type = solo_enc->solo_dev->type;
 790
 791         switch (f->index) {
 792         case 0:
 793                 switch (dev_type) {
 794                 case SOLO_DEV_6010:
 795                         f->pixelformat = V4L2_PIX_FMT_MPEG4;
 796                         strcpy(f->description, "MPEG-4 part 2");
 797                         break;
 798                 case SOLO_DEV_6110:
 799                         f->pixelformat = V4L2_PIX_FMT_H264;
 800                         strcpy(f->description, "H.264");
 801                         break;
 802                 }
 803                 break;
 804         case 1:
 805                 f->pixelformat = V4L2_PIX_FMT_MJPEG;
 806                 strcpy(f->description, "MJPEG");
 807                 break;
 808         default:
 809                 return -EINVAL;
 810         }
 811
 812         f->flags = V4L2_FMT_FLAG_COMPRESSED;
 813
 814         return 0;
 815 }
 816
 817 static inline int solo_valid_pixfmt(u32 pixfmt, int dev_type)
 818 {
 819         return (pixfmt == V4L2_PIX_FMT_H264 && dev_type == SOLO_DEV_6110)
 820                 || (pixfmt == V4L2_PIX_FMT_MPEG4 && dev_type == SOLO_DEV_6010)
 821                 || pixfmt == V4L2_PIX_FMT_MJPEG ? 0 : -EINVAL;
 822 }
 823
 824 static int solo_enc_try_fmt_cap(struct file *file, void *priv,
 825                             struct v4l2_format *f)
 826 {
 827         struct solo_enc_dev *solo_enc = video_drvdata(file);
 828         struct solo_dev *solo_dev = solo_enc->solo_dev;
 829         struct v4l2_pix_format *pix = &f->fmt.pix;
 830
 831         if (solo_valid_pixfmt(pix->pixelformat, solo_dev->type))
 832                 return -EINVAL;
 833
 834         if (pix->width < solo_dev->video_hsize ||
 835             pix->height < solo_dev->video_vsize << 1) {
 836                 /* Default to CIF 1/2 size */
 837                 pix->width = solo_dev->video_hsize >> 1;
 838                 pix->height = solo_dev->video_vsize;
 839         } else {
 840                 /* Full frame */
 841                 pix->width = solo_dev->video_hsize;
 842                 pix->height = solo_dev->video_vsize << 1;
 843         }
 844
 845         switch (pix->field) {
 846         case V4L2_FIELD_NONE:
 847         case V4L2_FIELD_INTERLACED:
 848                 break;
 849         case V4L2_FIELD_ANY:
 850         default:
 851                 pix->field = V4L2_FIELD_INTERLACED;
 852                 break;
 853         }
 854
 855         /* Just set these */
 856         pix->colorspace = V4L2_COLORSPACE_SMPTE170M;
 857         pix->sizeimage = FRAME_BUF_SIZE;
 858         pix->bytesperline = 0;
 859         pix->priv = 0;
 860
 861         return 0;
 862 }
 863
 864 static int solo_enc_set_fmt_cap(struct file *file, void *priv,
 865                                 struct v4l2_format *f)
 866 {
 867         struct solo_enc_dev *solo_enc = video_drvdata(file);
 868         struct solo_dev *solo_dev = solo_enc->solo_dev;
 869         struct v4l2_pix_format *pix = &f->fmt.pix;
 870         int ret;
 871
 872         if (vb2_is_busy(&solo_enc->vidq))
 873                 return -EBUSY;
 874
 875         ret = solo_enc_try_fmt_cap(file, priv, f);
 876         if (ret)
 877                 return ret;
 878
 879         if (pix->width == solo_dev->video_hsize)
 880                 solo_enc->mode = SOLO_ENC_MODE_D1;
 881         else
 882                 solo_enc->mode = SOLO_ENC_MODE_CIF;
 883
 884         /* This does not change the encoder at all */
 885         solo_enc->fmt = pix->pixelformat;
 886
 887         /*
 888          * More information is needed about these 'extended' types. As far
 889          * as I can tell these are basically additional video streams with
 890          * different MPEG encoding attributes that can run in parallel with
 891          * the main stream. If so, then this should be implemented as a
 892          * second video node. Abusing priv like this is certainly not the
 893          * right approach.
 894         if (pix->priv)
 895                 solo_enc->type = SOLO_ENC_TYPE_EXT;
 896          */
 897         solo_update_mode(solo_enc);
 898         return 0;
 899 }
 900
 901 static int solo_enc_get_fmt_cap(struct file *file, void *priv,
 902                                 struct v4l2_format *f)
 903 {
 904         struct solo_enc_dev *solo_enc = video_drvdata(file);
 905         struct v4l2_pix_format *pix = &f->fmt.pix;
 906
 907         pix->width = solo_enc->width;
 908         pix->height = solo_enc->height;
 909         pix->pixelformat = solo_enc->fmt;
 910         pix->field = solo_enc->interlaced ? V4L2_FIELD_INTERLACED :
 911                      V4L2_FIELD_NONE;
 912         pix->sizeimage = FRAME_BUF_SIZE;
 913         pix->colorspace = V4L2_COLORSPACE_SMPTE170M;
 914         pix->priv = 0;
 915
 916         return 0;
 917 }
 918
 919 static int solo_enc_g_std(struct file *file, void *priv, v4l2_std_id *i)
 920 {
 921         struct solo_enc_dev *solo_enc = video_drvdata(file);
 922         struct solo_dev *solo_dev = solo_enc->solo_dev;
 923
 924         if (solo_dev->video_type == SOLO_VO_FMT_TYPE_NTSC)
 925                 *i = V4L2_STD_NTSC_M;
 926         else
 927                 *i = V4L2_STD_PAL;
 928         return 0;
 929 }
 930
 931 static int solo_enc_s_std(struct file *file, void *priv, v4l2_std_id std)
 932 {
 933         struct solo_enc_dev *solo_enc = video_drvdata(file);
 934
 935         return solo_set_video_type(solo_enc->solo_dev, std & V4L2_STD_PAL);
 936 }
 937
 938 static int solo_enum_framesizes(struct file *file, void *priv,
 939                                 struct v4l2_frmsizeenum *fsize)
 940 {
 941         struct solo_enc_dev *solo_enc = video_drvdata(file);
 942         struct solo_dev *solo_dev = solo_enc->solo_dev;
 943
 944         if (solo_valid_pixfmt(fsize->pixel_format, solo_dev->type))
 945                 return -EINVAL;
 946
 947         switch (fsize->index) {
 948         case 0:
 949                 fsize->discrete.width = solo_dev->video_hsize >> 1;
 950                 fsize->discrete.height = solo_dev->video_vsize;
 951                 break;
 952         case 1:
 953                 fsize->discrete.width = solo_dev->video_hsize;
 954                 fsize->discrete.height = solo_dev->video_vsize << 1;
 955                 break;
 956         default:
 957                 return -EINVAL;
 958         }
 959
 960         fsize->type = V4L2_FRMSIZE_TYPE_DISCRETE;
 961
 962         return 0;
 963 }
 964
 965 static int solo_enum_frameintervals(struct file *file, void *priv,
 966                                     struct v4l2_frmivalenum *fintv)
 967 {
 968         struct solo_enc_dev *solo_enc = video_drvdata(file);
 969         struct solo_dev *solo_dev = solo_enc->solo_dev;
 970
 971         if (solo_valid_pixfmt(fintv->pixel_format, solo_dev->type))
 972                 return -EINVAL;
 973         if (fintv->index)
 974                 return -EINVAL;
 975         if ((fintv->width != solo_dev->video_hsize >> 1 ||
 976              fintv->height != solo_dev->video_vsize) &&
 977             (fintv->width != solo_dev->video_hsize ||
 978              fintv->height != solo_dev->video_vsize << 1))
 979                 return -EINVAL;
 980
 981         fintv->type = V4L2_FRMIVAL_TYPE_STEPWISE;
 982
 983         fintv->stepwise.min.numerator = 1;
 984         fintv->stepwise.min.denominator = solo_dev->fps;
 985
 986         fintv->stepwise.max.numerator = 15;
 987         fintv->stepwise.max.denominator = solo_dev->fps;
 988
 989         fintv->stepwise.step.numerator = 1;
 990         fintv->stepwise.step.denominator = solo_dev->fps;
 991
 992         return 0;
 993 }
 994
 995 static int solo_g_parm(struct file *file, void *priv,
 996                        struct v4l2_streamparm *sp)
 997 {
 998         struct solo_enc_dev *solo_enc = video_drvdata(file);
 999         struct v4l2_captureparm *cp = &sp->parm.capture;
1000
1001         cp->capability = V4L2_CAP_TIMEPERFRAME;
1002         cp->timeperframe.numerator = solo_enc->interval;
1003         cp->timeperframe.denominator = solo_enc->solo_dev->fps;
1004         cp->capturemode = 0;
1005         /* XXX: Shouldn't we be able to get/set this from videobuf? */
1006         cp->readbuffers = 2;
1007
1008         return 0;
1009 }
1010
1011 static inline int calc_interval(u8 fps, u32 n, u32 d)
1012 {
1013         if (!n || !d)
1014                 return 1;
1015         if (d == fps)
1016                 return n;
1017         n *= fps;
1018         return min(15U, n / d + (n % d >= (fps >> 1)));
1019 }
1020
1021 static int solo_s_parm(struct file *file, void *priv,
1022                        struct v4l2_streamparm *sp)
1023 {
1024         struct solo_enc_dev *solo_enc = video_drvdata(file);
1025         struct v4l2_fract *t = &sp->parm.capture.timeperframe;
1026         u8 fps = solo_enc->solo_dev->fps;
1027
1028         if (vb2_is_streaming(&solo_enc->vidq))
1029                 return -EBUSY;
1030
1031         solo_enc->interval = calc_interval(fps, t->numerator, t->denominator);
1032         solo_update_mode(solo_enc);
1033         return solo_g_parm(file, priv, sp);
1034 }
1035
1036 static long solo_enc_default(struct file *file, void *fh,
1037                         bool valid_prio, unsigned int cmd, void *arg)
1038 {
1039         struct solo_enc_dev *solo_enc = video_drvdata(file);
1040         struct solo_dev *solo_dev = solo_enc->solo_dev;
1041         struct solo_motion_thresholds *thresholds = arg;
1042
1043         switch (cmd) {
1044         case SOLO_IOC_G_MOTION_THRESHOLDS:
1045                 *thresholds = solo_enc->motion_thresholds;
1046                 return 0;
1047
1048         case SOLO_IOC_S_MOTION_THRESHOLDS:
1049                 if (!valid_prio)
1050                         return -EBUSY;
1051                 solo_enc->motion_thresholds = *thresholds;
1052                 if (solo_enc->motion_enabled && !solo_enc->motion_global)
1053                         return solo_set_motion_block(solo_dev, solo_enc->ch,
1054                                                 &solo_enc->motion_thresholds);
1055                 return 0;
1056         default:
1057                 return -ENOTTY;
1058         }
1059 }
1060
1061 static int solo_s_ctrl(struct v4l2_ctrl *ctrl)
1062 {
1063         struct solo_enc_dev *solo_enc =
1064                 container_of(ctrl->handler, struct solo_enc_dev, hdl);
1065         struct solo_dev *solo_dev = solo_enc->solo_dev;
1066         int err;
1067
1068         switch (ctrl->id) {
1069         case V4L2_CID_BRIGHTNESS:
1070         case V4L2_CID_CONTRAST:
1071         case V4L2_CID_SATURATION:
1072         case V4L2_CID_HUE:
1073         case V4L2_CID_SHARPNESS:
1074                 return tw28_set_ctrl_val(solo_dev, ctrl->id, solo_enc->ch,
1075                                          ctrl->val);
1076         case V4L2_CID_MPEG_VIDEO_GOP_SIZE:
1077                 solo_enc->gop = ctrl->val;
1078                 return 0;
1079         case V4L2_CID_MOTION_THRESHOLD:
1080                 solo_enc->motion_thresh = ctrl->val;
1081                 if (!solo_enc->motion_global || !solo_enc->motion_enabled)
1082                         return 0;
1083                 return solo_set_motion_threshold(solo_dev, solo_enc->ch, ctrl->val);
1084         case V4L2_CID_MOTION_MODE:
1085                 solo_enc->motion_global = ctrl->val == 1;
1086                 solo_enc->motion_enabled = ctrl->val > 0;
1087                 if (ctrl->val) {
1088                         if (solo_enc->motion_global)
1089                                 solo_set_motion_threshold(solo_dev, solo_enc->ch,
1090                                                 solo_enc->motion_thresh);
1091                         else
1092                                 solo_set_motion_block(solo_dev, solo_enc->ch,
1093                                                 &solo_enc->motion_thresholds);
1094                 }
1095                 solo_motion_toggle(solo_enc, ctrl->val);
1096                 return 0;
1097         case V4L2_CID_OSD_TEXT:
1098                 strcpy(solo_enc->osd_text, ctrl->string);
1099                 err = solo_osd_print(solo_enc);
1100                 return err;
1101         default:
1102                 return -EINVAL;
1103         }
1104
1105         return 0;
1106 }
1107
1108 static const struct v4l2_file_operations solo_enc_fops = {
1109         .owner                  = THIS_MODULE,
1110         .open                   = v4l2_fh_open,
1111         .release                = vb2_fop_release,
1112         .read                   = vb2_fop_read,
1113         .poll                   = vb2_fop_poll,
1114         .mmap                   = vb2_fop_mmap,
1115         .unlocked_ioctl         = video_ioctl2,
1116 };
1117
1118 static const struct v4l2_ioctl_ops solo_enc_ioctl_ops = {
1119         .vidioc_querycap                = solo_enc_querycap,
1120         .vidioc_s_std                   = solo_enc_s_std,
1121         .vidioc_g_std                   = solo_enc_g_std,
1122         /* Input callbacks */
1123         .vidioc_enum_input              = solo_enc_enum_input,
1124         .vidioc_s_input                 = solo_enc_set_input,
1125         .vidioc_g_input                 = solo_enc_get_input,
1126         /* Video capture format callbacks */
1127         .vidioc_enum_fmt_vid_cap        = solo_enc_enum_fmt_cap,
1128         .vidioc_try_fmt_vid_cap         = solo_enc_try_fmt_cap,
1129         .vidioc_s_fmt_vid_cap           = solo_enc_set_fmt_cap,
1130         .vidioc_g_fmt_vid_cap           = solo_enc_get_fmt_cap,
1131         /* Streaming I/O */
1132         .vidioc_reqbufs                 = vb2_ioctl_reqbufs,
1133         .vidioc_querybuf                = vb2_ioctl_querybuf,
1134         .vidioc_qbuf                    = vb2_ioctl_qbuf,
1135         .vidioc_dqbuf                   = vb2_ioctl_dqbuf,
1136         .vidioc_streamon                = vb2_ioctl_streamon,
1137         .vidioc_streamoff               = vb2_ioctl_streamoff,
1138         /* Frame size and interval */
1139         .vidioc_enum_framesizes         = solo_enum_framesizes,
1140         .vidioc_enum_frameintervals     = solo_enum_frameintervals,
1141         /* Video capture parameters */
1142         .vidioc_s_parm                  = solo_s_parm,
1143         .vidioc_g_parm                  = solo_g_parm,
1144         /* Logging and events */
1145         .vidioc_log_status              = v4l2_ctrl_log_status,
1146         .vidioc_subscribe_event         = v4l2_ctrl_subscribe_event,
1147         .vidioc_unsubscribe_event       = v4l2_event_unsubscribe,
1148         .vidioc_default                 = solo_enc_default,
1149 };
1150
1151 static const struct video_device solo_enc_template = {
1152         .name                   = SOLO6X10_NAME,
1153         .fops                   = &solo_enc_fops,
1154         .ioctl_ops              = &solo_enc_ioctl_ops,
1155         .minor                  = -1,
1156         .release                = video_device_release,
1157         .tvnorms                = V4L2_STD_NTSC_M | V4L2_STD_PAL,
1158 };
1159
1160 static const struct v4l2_ctrl_ops solo_ctrl_ops = {
1161         .s_ctrl = solo_s_ctrl,
1162 };
1163
1164 static const struct v4l2_ctrl_config solo_motion_threshold_ctrl = {
1165         .ops = &solo_ctrl_ops,
1166         .id = V4L2_CID_MOTION_THRESHOLD,
1167         .name = "Motion Detection Threshold",
1168         .type = V4L2_CTRL_TYPE_INTEGER,
1169         .max = 0xffff,
1170         .def = SOLO_DEF_MOT_THRESH,
1171         .step = 1,
1172         .flags = V4L2_CTRL_FLAG_SLIDER,
1173 };
1174
1175 static const char * const solo_motion_mode_menu[] = {
1176         "Disabled",
1177         "Global Threshold",
1178         "Regional Threshold",
1179         NULL
1180 };
1181
1182 static const struct v4l2_ctrl_config solo_motion_enable_ctrl = {
1183         .ops = &solo_ctrl_ops,
1184         .id = V4L2_CID_MOTION_MODE,
1185         .name = "Motion Detection Mode",
1186         .type = V4L2_CTRL_TYPE_MENU,
1187         .qmenu = solo_motion_mode_menu,
1188         .max = 2,
1189 };
1190
1191 static const struct v4l2_ctrl_config solo_osd_text_ctrl = {
1192         .ops = &solo_ctrl_ops,
1193         .id = V4L2_CID_OSD_TEXT,
1194         .name = "OSD Text",
1195         .type = V4L2_CTRL_TYPE_STRING,
1196         .max = OSD_TEXT_MAX,
1197         .step = 1,
1198 };
1199
1200 static struct solo_enc_dev *solo_enc_alloc(struct solo_dev *solo_dev,
1201                                            u8 ch, unsigned nr)
1202 {
1203         struct solo_enc_dev *solo_enc;
1204         struct v4l2_ctrl_handler *hdl;
1205         int ret;
1206         int x, y;
1207
1208         solo_enc = kzalloc(sizeof(*solo_enc), GFP_KERNEL);
1209         if (!solo_enc)
1210                 return ERR_PTR(-ENOMEM);
1211
1212         hdl = &solo_enc->hdl;
1213         v4l2_ctrl_handler_init(hdl, 10);
1214         v4l2_ctrl_new_std(hdl, &solo_ctrl_ops,
1215                         V4L2_CID_BRIGHTNESS, 0, 255, 1, 128);
1216         v4l2_ctrl_new_std(hdl, &solo_ctrl_ops,
1217                         V4L2_CID_CONTRAST, 0, 255, 1, 128);
1218         v4l2_ctrl_new_std(hdl, &solo_ctrl_ops,
1219                         V4L2_CID_SATURATION, 0, 255, 1, 128);
1220         v4l2_ctrl_new_std(hdl, &solo_ctrl_ops,
1221                         V4L2_CID_HUE, 0, 255, 1, 128);
1222         if (tw28_has_sharpness(solo_dev, ch))
1223                 v4l2_ctrl_new_std(hdl, &solo_ctrl_ops,
1224                         V4L2_CID_SHARPNESS, 0, 15, 1, 0);
1225         v4l2_ctrl_new_std(hdl, &solo_ctrl_ops,
1226                         V4L2_CID_MPEG_VIDEO_GOP_SIZE, 1, 255, 1, solo_dev->fps);
1227         v4l2_ctrl_new_custom(hdl, &solo_motion_threshold_ctrl, NULL);
1228         v4l2_ctrl_new_custom(hdl, &solo_motion_enable_ctrl, NULL);
1229         v4l2_ctrl_new_custom(hdl, &solo_osd_text_ctrl, NULL);
1230         if (hdl->error) {
1231                 ret = hdl->error;
1232                 goto hdl_free;
1233         }
1234
1235         solo_enc->solo_dev = solo_dev;
1236         solo_enc->ch = ch;
1237         mutex_init(&solo_enc->lock);
1238         spin_lock_init(&solo_enc->av_lock);
1239         INIT_LIST_HEAD(&solo_enc->vidq_active);
1240         solo_enc->fmt = (solo_dev->type == SOLO_DEV_6010) ?
1241                 V4L2_PIX_FMT_MPEG4 : V4L2_PIX_FMT_H264;
1242         solo_enc->type = SOLO_ENC_TYPE_STD;
1243
1244         solo_enc->qp = SOLO_DEFAULT_QP;
1245         solo_enc->gop = solo_dev->fps;
1246         solo_enc->interval = 1;
1247         solo_enc->mode = SOLO_ENC_MODE_CIF;
1248         solo_enc->motion_global = true;
1249         solo_enc->motion_thresh = SOLO_DEF_MOT_THRESH;
1250         for (y = 0; y < SOLO_MOTION_SZ; y++)
1251                 for (x = 0; x < SOLO_MOTION_SZ; x++)
1252                         solo_enc->motion_thresholds.thresholds[y][x] =
1253                                                         SOLO_DEF_MOT_THRESH;
1254
1255         solo_enc->vidq.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
1256         solo_enc->vidq.io_modes = VB2_MMAP | VB2_USERPTR | VB2_READ;
1257         solo_enc->vidq.ops = &solo_enc_video_qops;
1258         solo_enc->vidq.mem_ops = &vb2_dma_sg_memops;
1259         solo_enc->vidq.drv_priv = solo_enc;
1260         solo_enc->vidq.gfp_flags = __GFP_DMA32;
1261         solo_enc->vidq.timestamp_type = V4L2_BUF_FLAG_TIMESTAMP_MONOTONIC;
1262         solo_enc->vidq.buf_struct_size = sizeof(struct solo_vb2_buf);
1263         solo_enc->vidq.lock = &solo_enc->lock;
1264         ret = vb2_queue_init(&solo_enc->vidq);
1265         if (ret)
1266                 goto hdl_free;
1267         solo_update_mode(solo_enc);
1268
1269         spin_lock_init(&solo_enc->motion_lock);
1270
1271         /* Initialize this per encoder */
1272         solo_enc->jpeg_len = sizeof(jpeg_header);
1273         memcpy(solo_enc->jpeg_header, jpeg_header, solo_enc->jpeg_len);
1274
1275         solo_enc->desc_nelts = 32;
1276         solo_enc->desc_items = pci_alloc_consistent(solo_dev->pdev,
1277                                       sizeof(struct solo_p2m_desc) *
1278                                       solo_enc->desc_nelts, &solo_enc->desc_dma);
1279         ret = -ENOMEM;
1280         if (solo_enc->desc_items == NULL)
1281                 goto hdl_free;
1282
1283         solo_enc->vfd = video_device_alloc();
1284         if (!solo_enc->vfd)
1285                 goto pci_free;
1286
1287         *solo_enc->vfd = solo_enc_template;
1288         solo_enc->vfd->v4l2_dev = &solo_dev->v4l2_dev;
1289         solo_enc->vfd->ctrl_handler = hdl;
1290         solo_enc->vfd->queue = &solo_enc->vidq;
1291         solo_enc->vfd->lock = &solo_enc->lock;
1292         set_bit(V4L2_FL_USE_FH_PRIO, &solo_enc->vfd->flags);
1293         video_set_drvdata(solo_enc->vfd, solo_enc);
1294         ret = video_register_device(solo_enc->vfd, VFL_TYPE_GRABBER, nr);
1295         if (ret < 0)
1296                 goto vdev_release;
1297
1298         snprintf(solo_enc->vfd->name, sizeof(solo_enc->vfd->name),
1299                  "%s-enc (%i/%i)", SOLO6X10_NAME, solo_dev->vfd->num,
1300                  solo_enc->vfd->num);
1301
1302         return solo_enc;
1303
1304 vdev_release:
1305         video_device_release(solo_enc->vfd);
1306 pci_free:
1307         pci_free_consistent(solo_enc->solo_dev->pdev,
1308                         sizeof(struct solo_p2m_desc) * solo_enc->desc_nelts,
1309                         solo_enc->desc_items, solo_enc->desc_dma);
1310 hdl_free:
1311         v4l2_ctrl_handler_free(hdl);
1312         kfree(solo_enc);
1313         return ERR_PTR(ret);
1314 }
1315
1316 static void solo_enc_free(struct solo_enc_dev *solo_enc)
1317 {
1318         if (solo_enc == NULL)
1319                 return;
1320
1321         video_unregister_device(solo_enc->vfd);
1322         v4l2_ctrl_handler_free(&solo_enc->hdl);
1323         kfree(solo_enc);
1324 }
1325
1326 int solo_enc_v4l2_init(struct solo_dev *solo_dev, unsigned nr)
1327 {
1328         int i;
1329
1330         init_waitqueue_head(&solo_dev->ring_thread_wait);
1331
1332         solo_dev->vh_size = sizeof(struct vop_header);
1333         solo_dev->vh_buf = pci_alloc_consistent(solo_dev->pdev,
1334                                                 solo_dev->vh_size,
1335                                                 &solo_dev->vh_dma);
1336         if (solo_dev->vh_buf == NULL)
1337                 return -ENOMEM;
1338
1339         for (i = 0; i < solo_dev->nr_chans; i++) {
1340                 solo_dev->v4l2_enc[i] = solo_enc_alloc(solo_dev, i, nr);
1341                 if (IS_ERR(solo_dev->v4l2_enc[i]))
1342                         break;
1343         }
1344
1345         if (i != solo_dev->nr_chans) {
1346                 int ret = PTR_ERR(solo_dev->v4l2_enc[i]);
1347                 while (i--)
1348                         solo_enc_free(solo_dev->v4l2_enc[i]);
1349                 pci_free_consistent(solo_dev->pdev, solo_dev->vh_size,
1350                                     solo_dev->vh_buf, solo_dev->vh_dma);
1351                 solo_dev->vh_buf = NULL;
1352                 return ret;
1353         }
1354
1355         if (solo_dev->type == SOLO_DEV_6010)
1356                 solo_dev->enc_bw_remain = solo_dev->fps * 4 * 4;
1357         else
1358                 solo_dev->enc_bw_remain = solo_dev->fps * 4 * 5;
1359
1360         dev_info(&solo_dev->pdev->dev, "Encoders as /dev/video%d-%d\n",
1361                  solo_dev->v4l2_enc[0]->vfd->num,
1362                  solo_dev->v4l2_enc[solo_dev->nr_chans - 1]->vfd->num);
1363
1364         return 0;
1365 }
1366
1367 void solo_enc_v4l2_exit(struct solo_dev *solo_dev)
1368 {
1369         int i;
1370
1371         for (i = 0; i < solo_dev->nr_chans; i++)
1372                 solo_enc_free(solo_dev->v4l2_enc[i]);
1373
1374         if (solo_dev->vh_buf)
1375                 pci_free_consistent(solo_dev->pdev, solo_dev->vh_size,
1376                             solo_dev->vh_buf, solo_dev->vh_dma);
1377 }