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

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