drivers/media/platform/vivid/vivid-kthread-cap.c

   1 /*
   2  * vivid-kthread-cap.h - video/vbi capture thread support functions.
   3  *
   4  * Copyright 2014 Cisco Systems, Inc. and/or its affiliates. All rights reserved.
   5  *
   6  * This program is free software; you may redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; version 2 of the License.
   9  *
  10  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  11  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  12  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  13  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  14  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  15  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  16  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  17  * SOFTWARE.
  18  */
  19
  20 #include <linux/module.h>
  21 #include <linux/errno.h>
  22 #include <linux/kernel.h>
  23 #include <linux/init.h>
  24 #include <linux/sched.h>
  25 #include <linux/slab.h>
  26 #include <linux/font.h>
  27 #include <linux/mutex.h>
  28 #include <linux/videodev2.h>
  29 #include <linux/kthread.h>
  30 #include <linux/freezer.h>
  31 #include <linux/random.h>
  32 #include <linux/v4l2-dv-timings.h>
  33 #include <asm/div64.h>
  34 #include <media/videobuf2-vmalloc.h>
  35 #include <media/v4l2-dv-timings.h>
  36 #include <media/v4l2-ioctl.h>
  37 #include <media/v4l2-fh.h>
  38 #include <media/v4l2-event.h>
  39 #include <media/v4l2-rect.h>
  40
  41 #include "vivid-core.h"
  42 #include "vivid-vid-common.h"
  43 #include "vivid-vid-cap.h"
  44 #include "vivid-vid-out.h"
  45 #include "vivid-radio-common.h"
  46 #include "vivid-radio-rx.h"
  47 #include "vivid-radio-tx.h"
  48 #include "vivid-sdr-cap.h"
  49 #include "vivid-vbi-cap.h"
  50 #include "vivid-vbi-out.h"
  51 #include "vivid-osd.h"
  52 #include "vivid-ctrls.h"
  53 #include "vivid-kthread-cap.h"
  54
  55 static inline v4l2_std_id vivid_get_std_cap(const struct vivid_dev *dev)
  56 {
  57         if (vivid_is_sdtv_cap(dev))
  58                 return dev->std_cap;
  59         return 0;
  60 }
  61
  62 static void copy_pix(struct vivid_dev *dev, int win_y, int win_x,
  63                         u16 *cap, const u16 *osd)
  64 {
  65         u16 out;
  66         int left = dev->overlay_out_left;
  67         int top = dev->overlay_out_top;
  68         int fb_x = win_x + left;
  69         int fb_y = win_y + top;
  70         int i;
  71
  72         out = *cap;
  73         *cap = *osd;
  74         if (dev->bitmap_out) {
  75                 const u8 *p = dev->bitmap_out;
  76                 unsigned stride = (dev->compose_out.width + 7) / 8;
  77
  78                 win_x -= dev->compose_out.left;
  79                 win_y -= dev->compose_out.top;
  80                 if (!(p[stride * win_y + win_x / 8] & (1 << (win_x & 7))))
  81                         return;
  82         }
  83
  84         for (i = 0; i < dev->clipcount_out; i++) {
  85                 struct v4l2_rect *r = &dev->clips_out[i].c;
  86
  87                 if (fb_y >= r->top && fb_y < r->top + r->height &&
  88                     fb_x >= r->left && fb_x < r->left + r->width)
  89                         return;
  90         }
  91         if ((dev->fbuf_out_flags & V4L2_FBUF_FLAG_CHROMAKEY) &&
  92             *osd != dev->chromakey_out)
  93                 return;
  94         if ((dev->fbuf_out_flags & V4L2_FBUF_FLAG_SRC_CHROMAKEY) &&
  95             out == dev->chromakey_out)
  96                 return;
  97         if (dev->fmt_cap->alpha_mask) {
  98                 if ((dev->fbuf_out_flags & V4L2_FBUF_FLAG_GLOBAL_ALPHA) &&
  99                     dev->global_alpha_out)
 100                         return;
 101                 if ((dev->fbuf_out_flags & V4L2_FBUF_FLAG_LOCAL_ALPHA) &&
 102                     *cap & dev->fmt_cap->alpha_mask)
 103                         return;
 104                 if ((dev->fbuf_out_flags & V4L2_FBUF_FLAG_LOCAL_INV_ALPHA) &&
 105                     !(*cap & dev->fmt_cap->alpha_mask))
 106                         return;
 107         }
 108         *cap = out;
 109 }
 110
 111 static void blend_line(struct vivid_dev *dev, unsigned y_offset, unsigned x_offset,
 112                 u8 *vcapbuf, const u8 *vosdbuf,
 113                 unsigned width, unsigned pixsize)
 114 {
 115         unsigned x;
 116
 117         for (x = 0; x < width; x++, vcapbuf += pixsize, vosdbuf += pixsize) {
 118                 copy_pix(dev, y_offset, x_offset + x,
 119                          (u16 *)vcapbuf, (const u16 *)vosdbuf);
 120         }
 121 }
 122
 123 static void scale_line(const u8 *src, u8 *dst, unsigned srcw, unsigned dstw, unsigned twopixsize)
 124 {
 125         /* Coarse scaling with Bresenham */
 126         unsigned int_part;
 127         unsigned fract_part;
 128         unsigned src_x = 0;
 129         unsigned error = 0;
 130         unsigned x;
 131
 132         /*
 133          * We always combine two pixels to prevent color bleed in the packed
 134          * yuv case.
 135          */
 136         srcw /= 2;
 137         dstw /= 2;
 138         int_part = srcw / dstw;
 139         fract_part = srcw % dstw;
 140         for (x = 0; x < dstw; x++, dst += twopixsize) {
 141                 memcpy(dst, src + src_x * twopixsize, twopixsize);
 142                 src_x += int_part;
 143                 error += fract_part;
 144                 if (error >= dstw) {
 145                         error -= dstw;
 146                         src_x++;
 147                 }
 148         }
 149 }
 150
 151 /*
 152  * Precalculate the rectangles needed to perform video looping:
 153  *
 154  * The nominal pipeline is that the video output buffer is cropped by
 155  * crop_out, scaled to compose_out, overlaid with the output overlay,
 156  * cropped on the capture side by crop_cap and scaled again to the video
 157  * capture buffer using compose_cap.
 158  *
 159  * To keep things efficient we calculate the intersection of compose_out
 160  * and crop_cap (since that's the only part of the video that will
 161  * actually end up in the capture buffer), determine which part of the
 162  * video output buffer that is and which part of the video capture buffer
 163  * so we can scale the video straight from the output buffer to the capture
 164  * buffer without any intermediate steps.
 165  *
 166  * If we need to deal with an output overlay, then there is no choice and
 167  * that intermediate step still has to be taken. For the output overlay
 168  * support we calculate the intersection of the framebuffer and the overlay
 169  * window (which may be partially or wholly outside of the framebuffer
 170  * itself) and the intersection of that with loop_vid_copy (i.e. the part of
 171  * the actual looped video that will be overlaid). The result is calculated
 172  * both in framebuffer coordinates (loop_fb_copy) and compose_out coordinates
 173  * (loop_vid_overlay). Finally calculate the part of the capture buffer that
 174  * will receive that overlaid video.
 175  */
 176 static void vivid_precalc_copy_rects(struct vivid_dev *dev)
 177 {
 178         /* Framebuffer rectangle */
 179         struct v4l2_rect r_fb = {
 180                 0, 0, dev->display_width, dev->display_height
 181         };
 182         /* Overlay window rectangle in framebuffer coordinates */
 183         struct v4l2_rect r_overlay = {
 184                 dev->overlay_out_left, dev->overlay_out_top,
 185                 dev->compose_out.width, dev->compose_out.height
 186         };
 187
 188         v4l2_rect_intersect(&dev->loop_vid_copy, &dev->crop_cap, &dev->compose_out);
 189
 190         dev->loop_vid_out = dev->loop_vid_copy;
 191         v4l2_rect_scale(&dev->loop_vid_out, &dev->compose_out, &dev->crop_out);
 192         dev->loop_vid_out.left += dev->crop_out.left;
 193         dev->loop_vid_out.top += dev->crop_out.top;
 194
 195         dev->loop_vid_cap = dev->loop_vid_copy;
 196         v4l2_rect_scale(&dev->loop_vid_cap, &dev->crop_cap, &dev->compose_cap);
 197
 198         dprintk(dev, 1,
 199                 "loop_vid_copy: %dx%d@%dx%d loop_vid_out: %dx%d@%dx%d loop_vid_cap: %dx%d@%dx%d\n",
 200                 dev->loop_vid_copy.width, dev->loop_vid_copy.height,
 201                 dev->loop_vid_copy.left, dev->loop_vid_copy.top,
 202                 dev->loop_vid_out.width, dev->loop_vid_out.height,
 203                 dev->loop_vid_out.left, dev->loop_vid_out.top,
 204                 dev->loop_vid_cap.width, dev->loop_vid_cap.height,
 205                 dev->loop_vid_cap.left, dev->loop_vid_cap.top);
 206
 207         v4l2_rect_intersect(&r_overlay, &r_fb, &r_overlay);
 208
 209         /* shift r_overlay to the same origin as compose_out */
 210         r_overlay.left += dev->compose_out.left - dev->overlay_out_left;
 211         r_overlay.top += dev->compose_out.top - dev->overlay_out_top;
 212
 213         v4l2_rect_intersect(&dev->loop_vid_overlay, &r_overlay, &dev->loop_vid_copy);
 214         dev->loop_fb_copy = dev->loop_vid_overlay;
 215
 216         /* shift dev->loop_fb_copy back again to the fb origin */
 217         dev->loop_fb_copy.left -= dev->compose_out.left - dev->overlay_out_left;
 218         dev->loop_fb_copy.top -= dev->compose_out.top - dev->overlay_out_top;
 219
 220         dev->loop_vid_overlay_cap = dev->loop_vid_overlay;
 221         v4l2_rect_scale(&dev->loop_vid_overlay_cap, &dev->crop_cap, &dev->compose_cap);
 222
 223         dprintk(dev, 1,
 224                 "loop_fb_copy: %dx%d@%dx%d loop_vid_overlay: %dx%d@%dx%d loop_vid_overlay_cap: %dx%d@%dx%d\n",
 225                 dev->loop_fb_copy.width, dev->loop_fb_copy.height,
 226                 dev->loop_fb_copy.left, dev->loop_fb_copy.top,
 227                 dev->loop_vid_overlay.width, dev->loop_vid_overlay.height,
 228                 dev->loop_vid_overlay.left, dev->loop_vid_overlay.top,
 229                 dev->loop_vid_overlay_cap.width, dev->loop_vid_overlay_cap.height,
 230                 dev->loop_vid_overlay_cap.left, dev->loop_vid_overlay_cap.top);
 231 }
 232
 233 static void *plane_vaddr(struct tpg_data *tpg, struct vivid_buffer *buf,
 234                          unsigned p, unsigned bpl[TPG_MAX_PLANES], unsigned h)
 235 {
 236         unsigned i;
 237         void *vbuf;
 238
 239         if (p == 0 || tpg_g_buffers(tpg) > 1)
 240                 return vb2_plane_vaddr(&buf->vb.vb2_buf, p);
 241         vbuf = vb2_plane_vaddr(&buf->vb.vb2_buf, 0);
 242         for (i = 0; i < p; i++)
 243                 vbuf += bpl[i] * h / tpg->vdownsampling[i];
 244         return vbuf;
 245 }
 246
 247 static int vivid_copy_buffer(struct vivid_dev *dev, unsigned p, u8 *vcapbuf,
 248                 struct vivid_buffer *vid_cap_buf)
 249 {
 250         bool blank = dev->must_blank[vid_cap_buf->vb.vb2_buf.index];
 251         struct tpg_data *tpg = &dev->tpg;
 252         struct vivid_buffer *vid_out_buf = NULL;
 253         unsigned vdiv = dev->fmt_out->vdownsampling[p];
 254         unsigned twopixsize = tpg_g_twopixelsize(tpg, p);
 255         unsigned img_width = tpg_hdiv(tpg, p, dev->compose_cap.width);
 256         unsigned img_height = dev->compose_cap.height;
 257         unsigned stride_cap = tpg->bytesperline[p];
 258         unsigned stride_out = dev->bytesperline_out[p];
 259         unsigned stride_osd = dev->display_byte_stride;
 260         unsigned hmax = (img_height * tpg->perc_fill) / 100;
 261         u8 *voutbuf;
 262         u8 *vosdbuf = NULL;
 263         unsigned y;
 264         bool blend = dev->bitmap_out || dev->clipcount_out || dev->fbuf_out_flags;
 265         /* Coarse scaling with Bresenham */
 266         unsigned vid_out_int_part;
 267         unsigned vid_out_fract_part;
 268         unsigned vid_out_y = 0;
 269         unsigned vid_out_error = 0;
 270         unsigned vid_overlay_int_part = 0;
 271         unsigned vid_overlay_fract_part = 0;
 272         unsigned vid_overlay_y = 0;
 273         unsigned vid_overlay_error = 0;
 274         unsigned vid_cap_left = tpg_hdiv(tpg, p, dev->loop_vid_cap.left);
 275         unsigned vid_cap_right;
 276         bool quick;
 277
 278         vid_out_int_part = dev->loop_vid_out.height / dev->loop_vid_cap.height;
 279         vid_out_fract_part = dev->loop_vid_out.height % dev->loop_vid_cap.height;
 280
 281         if (!list_empty(&dev->vid_out_active))
 282                 vid_out_buf = list_entry(dev->vid_out_active.next,
 283                                          struct vivid_buffer, list);
 284         if (vid_out_buf == NULL)
 285                 return -ENODATA;
 286
 287         vid_cap_buf->vb.field = vid_out_buf->vb.field;
 288
 289         voutbuf = plane_vaddr(tpg, vid_out_buf, p,
 290                               dev->bytesperline_out, dev->fmt_out_rect.height);
 291         if (p < dev->fmt_out->buffers)
 292                 voutbuf += vid_out_buf->vb.vb2_buf.planes[p].data_offset;
 293         voutbuf += tpg_hdiv(tpg, p, dev->loop_vid_out.left) +
 294                 (dev->loop_vid_out.top / vdiv) * stride_out;
 295         vcapbuf += tpg_hdiv(tpg, p, dev->compose_cap.left) +
 296                 (dev->compose_cap.top / vdiv) * stride_cap;
 297
 298         if (dev->loop_vid_copy.width == 0 || dev->loop_vid_copy.height == 0) {
 299                 /*
 300                  * If there is nothing to copy, then just fill the capture window
 301                  * with black.
 302                  */
 303                 for (y = 0; y < hmax / vdiv; y++, vcapbuf += stride_cap)
 304                         memcpy(vcapbuf, tpg->black_line[p], img_width);
 305                 return 0;
 306         }
 307
 308         if (dev->overlay_out_enabled &&
 309             dev->loop_vid_overlay.width && dev->loop_vid_overlay.height) {
 310                 vosdbuf = dev->video_vbase;
 311                 vosdbuf += (dev->loop_fb_copy.left * twopixsize) / 2 +
 312                            dev->loop_fb_copy.top * stride_osd;
 313                 vid_overlay_int_part = dev->loop_vid_overlay.height /
 314                                        dev->loop_vid_overlay_cap.height;
 315                 vid_overlay_fract_part = dev->loop_vid_overlay.height %
 316                                          dev->loop_vid_overlay_cap.height;
 317         }
 318
 319         vid_cap_right = tpg_hdiv(tpg, p, dev->loop_vid_cap.left + dev->loop_vid_cap.width);
 320         /* quick is true if no video scaling is needed */
 321         quick = dev->loop_vid_out.width == dev->loop_vid_cap.width;
 322
 323         dev->cur_scaled_line = dev->loop_vid_out.height;
 324         for (y = 0; y < hmax; y += vdiv, vcapbuf += stride_cap) {
 325                 /* osdline is true if this line requires overlay blending */
 326                 bool osdline = vosdbuf && y >= dev->loop_vid_overlay_cap.top &&
 327                           y < dev->loop_vid_overlay_cap.top + dev->loop_vid_overlay_cap.height;
 328
 329                 /*
 330                  * If this line of the capture buffer doesn't get any video, then
 331                  * just fill with black.
 332                  */
 333                 if (y < dev->loop_vid_cap.top ||
 334                     y >= dev->loop_vid_cap.top + dev->loop_vid_cap.height) {
 335                         memcpy(vcapbuf, tpg->black_line[p], img_width);
 336                         continue;
 337                 }
 338
 339                 /* fill the left border with black */
 340                 if (dev->loop_vid_cap.left)
 341                         memcpy(vcapbuf, tpg->black_line[p], vid_cap_left);
 342
 343                 /* fill the right border with black */
 344                 if (vid_cap_right < img_width)
 345                         memcpy(vcapbuf + vid_cap_right, tpg->black_line[p],
 346                                 img_width - vid_cap_right);
 347
 348                 if (quick && !osdline) {
 349                         memcpy(vcapbuf + vid_cap_left,
 350                                voutbuf + vid_out_y * stride_out,
 351                                tpg_hdiv(tpg, p, dev->loop_vid_cap.width));
 352                         goto update_vid_out_y;
 353                 }
 354                 if (dev->cur_scaled_line == vid_out_y) {
 355                         memcpy(vcapbuf + vid_cap_left, dev->scaled_line,
 356                                tpg_hdiv(tpg, p, dev->loop_vid_cap.width));
 357                         goto update_vid_out_y;
 358                 }
 359                 if (!osdline) {
 360                         scale_line(voutbuf + vid_out_y * stride_out, dev->scaled_line,
 361                                 tpg_hdiv(tpg, p, dev->loop_vid_out.width),
 362                                 tpg_hdiv(tpg, p, dev->loop_vid_cap.width),
 363                                 tpg_g_twopixelsize(tpg, p));
 364                 } else {
 365                         /*
 366                          * Offset in bytes within loop_vid_copy to the start of the
 367                          * loop_vid_overlay rectangle.
 368                          */
 369                         unsigned offset =
 370                                 ((dev->loop_vid_overlay.left - dev->loop_vid_copy.left) *
 371                                  twopixsize) / 2;
 372                         u8 *osd = vosdbuf + vid_overlay_y * stride_osd;
 373
 374                         scale_line(voutbuf + vid_out_y * stride_out, dev->blended_line,
 375                                 dev->loop_vid_out.width, dev->loop_vid_copy.width,
 376                                 tpg_g_twopixelsize(tpg, p));
 377                         if (blend)
 378                                 blend_line(dev, vid_overlay_y + dev->loop_vid_overlay.top,
 379                                            dev->loop_vid_overlay.left,
 380                                            dev->blended_line + offset, osd,
 381                                            dev->loop_vid_overlay.width, twopixsize / 2);
 382                         else
 383                                 memcpy(dev->blended_line + offset,
 384                                        osd, (dev->loop_vid_overlay.width * twopixsize) / 2);
 385                         scale_line(dev->blended_line, dev->scaled_line,
 386                                         dev->loop_vid_copy.width, dev->loop_vid_cap.width,
 387                                         tpg_g_twopixelsize(tpg, p));
 388                 }
 389                 dev->cur_scaled_line = vid_out_y;
 390                 memcpy(vcapbuf + vid_cap_left, dev->scaled_line,
 391                        tpg_hdiv(tpg, p, dev->loop_vid_cap.width));
 392
 393 update_vid_out_y:
 394                 if (osdline) {
 395                         vid_overlay_y += vid_overlay_int_part;
 396                         vid_overlay_error += vid_overlay_fract_part;
 397                         if (vid_overlay_error >= dev->loop_vid_overlay_cap.height) {
 398                                 vid_overlay_error -= dev->loop_vid_overlay_cap.height;
 399                                 vid_overlay_y++;
 400                         }
 401                 }
 402                 vid_out_y += vid_out_int_part;
 403                 vid_out_error += vid_out_fract_part;
 404                 if (vid_out_error >= dev->loop_vid_cap.height / vdiv) {
 405                         vid_out_error -= dev->loop_vid_cap.height / vdiv;
 406                         vid_out_y++;
 407                 }
 408         }
 409
 410         if (!blank)
 411                 return 0;
 412         for (; y < img_height; y += vdiv, vcapbuf += stride_cap)
 413                 memcpy(vcapbuf, tpg->contrast_line[p], img_width);
 414         return 0;
 415 }
 416
 417 static void vivid_fillbuff(struct vivid_dev *dev, struct vivid_buffer *buf)
 418 {
 419         struct tpg_data *tpg = &dev->tpg;
 420         unsigned factor = V4L2_FIELD_HAS_T_OR_B(dev->field_cap) ? 2 : 1;
 421         unsigned line_height = 16 / factor;
 422         bool is_tv = vivid_is_sdtv_cap(dev);
 423         bool is_60hz = is_tv && (dev->std_cap & V4L2_STD_525_60);
 424         unsigned p;
 425         int line = 1;
 426         u8 *basep[TPG_MAX_PLANES][2];
 427         unsigned ms;
 428         char str[100];
 429         s32 gain;
 430         bool is_loop = false;
 431
 432         if (dev->loop_video && dev->can_loop_video &&
 433                 ((vivid_is_svid_cap(dev) &&
 434                 !VIVID_INVALID_SIGNAL(dev->std_signal_mode)) ||
 435                 (vivid_is_hdmi_cap(dev) &&
 436                 !VIVID_INVALID_SIGNAL(dev->dv_timings_signal_mode))))
 437                 is_loop = true;
 438
 439         buf->vb.sequence = dev->vid_cap_seq_count;
 440         /*
 441          * Take the timestamp now if the timestamp source is set to
 442          * "Start of Exposure".
 443          */
 444         if (dev->tstamp_src_is_soe)
 445                 buf->vb.vb2_buf.timestamp = ktime_get_ns();
 446         if (dev->field_cap == V4L2_FIELD_ALTERNATE) {
 447                 /*
 448                  * 60 Hz standards start with the bottom field, 50 Hz standards
 449                  * with the top field. So if the 0-based seq_count is even,
 450                  * then the field is TOP for 50 Hz and BOTTOM for 60 Hz
 451                  * standards.
 452                  */
 453                 buf->vb.field = ((dev->vid_cap_seq_count & 1) ^ is_60hz) ?
 454                         V4L2_FIELD_BOTTOM : V4L2_FIELD_TOP;
 455                 /*
 456                  * The sequence counter counts frames, not fields. So divide
 457                  * by two.
 458                  */
 459                 buf->vb.sequence /= 2;
 460         } else {
 461                 buf->vb.field = dev->field_cap;
 462         }
 463         tpg_s_field(tpg, buf->vb.field,
 464                     dev->field_cap == V4L2_FIELD_ALTERNATE);
 465         tpg_s_perc_fill_blank(tpg, dev->must_blank[buf->vb.vb2_buf.index]);
 466
 467         vivid_precalc_copy_rects(dev);
 468
 469         for (p = 0; p < tpg_g_planes(tpg); p++) {
 470                 void *vbuf = plane_vaddr(tpg, buf, p,
 471                                          tpg->bytesperline, tpg->buf_height);
 472
 473                 /*
 474                  * The first plane of a multiplanar format has a non-zero
 475                  * data_offset. This helps testing whether the application
 476                  * correctly supports non-zero data offsets.
 477                  */
 478                 if (p < tpg_g_buffers(tpg) && dev->fmt_cap->data_offset[p]) {
 479                         memset(vbuf, dev->fmt_cap->data_offset[p] & 0xff,
 480                                dev->fmt_cap->data_offset[p]);
 481                         vbuf += dev->fmt_cap->data_offset[p];
 482                 }
 483                 tpg_calc_text_basep(tpg, basep, p, vbuf);
 484                 if (!is_loop || vivid_copy_buffer(dev, p, vbuf, buf))
 485                         tpg_fill_plane_buffer(tpg, vivid_get_std_cap(dev),
 486                                         p, vbuf);
 487         }
 488         dev->must_blank[buf->vb.vb2_buf.index] = false;
 489
 490         /* Updates stream time, only update at the start of a new frame. */
 491         if (dev->field_cap != V4L2_FIELD_ALTERNATE ||
 492                         (buf->vb.sequence & 1) == 0)
 493                 dev->ms_vid_cap =
 494                         jiffies_to_msecs(jiffies - dev->jiffies_vid_cap);
 495
 496         ms = dev->ms_vid_cap;
 497         if (dev->osd_mode <= 1) {
 498                 snprintf(str, sizeof(str), " %02d:%02d:%02d:%03d %u%s",
 499                                 (ms / (60 * 60 * 1000)) % 24,
 500                                 (ms / (60 * 1000)) % 60,
 501                                 (ms / 1000) % 60,
 502                                 ms % 1000,
 503                                 buf->vb.sequence,
 504                                 (dev->field_cap == V4L2_FIELD_ALTERNATE) ?
 505                                         (buf->vb.field == V4L2_FIELD_TOP ?
 506                                          " top" : " bottom") : "");
 507                 tpg_gen_text(tpg, basep, line++ * line_height, 16, str);
 508         }
 509         if (dev->osd_mode == 0) {
 510                 snprintf(str, sizeof(str), " %dx%d, input %d ",
 511                                 dev->src_rect.width, dev->src_rect.height, dev->input);
 512                 tpg_gen_text(tpg, basep, line++ * line_height, 16, str);
 513
 514                 gain = v4l2_ctrl_g_ctrl(dev->gain);
 515                 mutex_lock(dev->ctrl_hdl_user_vid.lock);
 516                 snprintf(str, sizeof(str),
 517                         " brightness %3d, contrast %3d, saturation %3d, hue %d ",
 518                         dev->brightness->cur.val,
 519                         dev->contrast->cur.val,
 520                         dev->saturation->cur.val,
 521                         dev->hue->cur.val);
 522                 tpg_gen_text(tpg, basep, line++ * line_height, 16, str);
 523                 snprintf(str, sizeof(str),
 524                         " autogain %d, gain %3d, alpha 0x%02x ",
 525                         dev->autogain->cur.val, gain, dev->alpha->cur.val);
 526                 mutex_unlock(dev->ctrl_hdl_user_vid.lock);
 527                 tpg_gen_text(tpg, basep, line++ * line_height, 16, str);
 528                 mutex_lock(dev->ctrl_hdl_user_aud.lock);
 529                 snprintf(str, sizeof(str),
 530                         " volume %3d, mute %d ",
 531                         dev->volume->cur.val, dev->mute->cur.val);
 532                 mutex_unlock(dev->ctrl_hdl_user_aud.lock);
 533                 tpg_gen_text(tpg, basep, line++ * line_height, 16, str);
 534                 mutex_lock(dev->ctrl_hdl_user_gen.lock);
 535                 snprintf(str, sizeof(str), " int32 %d, int64 %lld, bitmask %08x ",
 536                         dev->int32->cur.val,
 537                         *dev->int64->p_cur.p_s64,
 538                         dev->bitmask->cur.val);
 539                 tpg_gen_text(tpg, basep, line++ * line_height, 16, str);
 540                 snprintf(str, sizeof(str), " boolean %d, menu %s, string \"%s\" ",
 541                         dev->boolean->cur.val,
 542                         dev->menu->qmenu[dev->menu->cur.val],
 543                         dev->string->p_cur.p_char);
 544                 tpg_gen_text(tpg, basep, line++ * line_height, 16, str);
 545                 snprintf(str, sizeof(str), " integer_menu %lld, value %d ",
 546                         dev->int_menu->qmenu_int[dev->int_menu->cur.val],
 547                         dev->int_menu->cur.val);
 548                 mutex_unlock(dev->ctrl_hdl_user_gen.lock);
 549                 tpg_gen_text(tpg, basep, line++ * line_height, 16, str);
 550                 if (dev->button_pressed) {
 551                         dev->button_pressed--;
 552                         snprintf(str, sizeof(str), " button pressed!");
 553                         tpg_gen_text(tpg, basep, line++ * line_height, 16, str);
 554                 }
 555                 if (dev->osd[0]) {
 556                         if (vivid_is_hdmi_cap(dev)) {
 557                                 snprintf(str, sizeof(str),
 558                                          " OSD \"%s\"", dev->osd);
 559                                 tpg_gen_text(tpg, basep, line++ * line_height,
 560                                              16, str);
 561                         }
 562                         if (dev->osd_jiffies &&
 563                             time_is_before_jiffies(dev->osd_jiffies + 5 * HZ)) {
 564                                 dev->osd[0] = 0;
 565                                 dev->osd_jiffies = 0;
 566                         }
 567                 }
 568         }
 569
 570         /*
 571          * If "End of Frame" is specified at the timestamp source, then take
 572          * the timestamp now.
 573          */
 574         if (!dev->tstamp_src_is_soe)
 575                 buf->vb.vb2_buf.timestamp = ktime_get_ns();
 576         buf->vb.vb2_buf.timestamp += dev->time_wrap_offset;
 577 }
 578
 579 /*
 580  * Return true if this pixel coordinate is a valid video pixel.
 581  */
 582 static bool valid_pix(struct vivid_dev *dev, int win_y, int win_x, int fb_y, int fb_x)
 583 {
 584         int i;
 585
 586         if (dev->bitmap_cap) {
 587                 /*
 588                  * Only if the corresponding bit in the bitmap is set can
 589                  * the video pixel be shown. Coordinates are relative to
 590                  * the overlay window set by VIDIOC_S_FMT.
 591                  */
 592                 const u8 *p = dev->bitmap_cap;
 593                 unsigned stride = (dev->compose_cap.width + 7) / 8;
 594
 595                 if (!(p[stride * win_y + win_x / 8] & (1 << (win_x & 7))))
 596                         return false;
 597         }
 598
 599         for (i = 0; i < dev->clipcount_cap; i++) {
 600                 /*
 601                  * Only if the framebuffer coordinate is not in any of the
 602                  * clip rectangles will be video pixel be shown.
 603                  */
 604                 struct v4l2_rect *r = &dev->clips_cap[i].c;
 605
 606                 if (fb_y >= r->top && fb_y < r->top + r->height &&
 607                     fb_x >= r->left && fb_x < r->left + r->width)
 608                         return false;
 609         }
 610         return true;
 611 }
 612
 613 /*
 614  * Draw the image into the overlay buffer.
 615  * Note that the combination of overlay and multiplanar is not supported.
 616  */
 617 static void vivid_overlay(struct vivid_dev *dev, struct vivid_buffer *buf)
 618 {
 619         struct tpg_data *tpg = &dev->tpg;
 620         unsigned pixsize = tpg_g_twopixelsize(tpg, 0) / 2;
 621         void *vbase = dev->fb_vbase_cap;
 622         void *vbuf = vb2_plane_vaddr(&buf->vb.vb2_buf, 0);
 623         unsigned img_width = dev->compose_cap.width;
 624         unsigned img_height = dev->compose_cap.height;
 625         unsigned stride = tpg->bytesperline[0];
 626         /* if quick is true, then valid_pix() doesn't have to be called */
 627         bool quick = dev->bitmap_cap == NULL && dev->clipcount_cap == 0;
 628         int x, y, w, out_x = 0;
 629
 630         /*
 631          * Overlay support is only supported for formats that have a twopixelsize
 632          * that's >= 2. Warn and bail out if that's not the case.
 633          */
 634         if (WARN_ON(pixsize == 0))
 635                 return;
 636         if ((dev->overlay_cap_field == V4L2_FIELD_TOP ||
 637              dev->overlay_cap_field == V4L2_FIELD_BOTTOM) &&
 638             dev->overlay_cap_field != buf->vb.field)
 639                 return;
 640
 641         vbuf += dev->compose_cap.left * pixsize + dev->compose_cap.top * stride;
 642         x = dev->overlay_cap_left;
 643         w = img_width;
 644         if (x < 0) {
 645                 out_x = -x;
 646                 w = w - out_x;
 647                 x = 0;
 648         } else {
 649                 w = dev->fb_cap.fmt.width - x;
 650                 if (w > img_width)
 651                         w = img_width;
 652         }
 653         if (w <= 0)
 654                 return;
 655         if (dev->overlay_cap_top >= 0)
 656                 vbase += dev->overlay_cap_top * dev->fb_cap.fmt.bytesperline;
 657         for (y = dev->overlay_cap_top;
 658              y < dev->overlay_cap_top + (int)img_height;
 659              y++, vbuf += stride) {
 660                 int px;
 661
 662                 if (y < 0 || y > dev->fb_cap.fmt.height)
 663                         continue;
 664                 if (quick) {
 665                         memcpy(vbase + x * pixsize,
 666                                vbuf + out_x * pixsize, w * pixsize);
 667                         vbase += dev->fb_cap.fmt.bytesperline;
 668                         continue;
 669                 }
 670                 for (px = 0; px < w; px++) {
 671                         if (!valid_pix(dev, y - dev->overlay_cap_top,
 672                                        px + out_x, y, px + x))
 673                                 continue;
 674                         memcpy(vbase + (px + x) * pixsize,
 675                                vbuf + (px + out_x) * pixsize,
 676                                pixsize);
 677                 }
 678                 vbase += dev->fb_cap.fmt.bytesperline;
 679         }
 680 }
 681
 682 static void vivid_thread_vid_cap_tick(struct vivid_dev *dev, int dropped_bufs)
 683 {
 684         struct vivid_buffer *vid_cap_buf = NULL;
 685         struct vivid_buffer *vbi_cap_buf = NULL;
 686
 687         dprintk(dev, 1, "Video Capture Thread Tick\n");
 688
 689         while (dropped_bufs-- > 1)
 690                 tpg_update_mv_count(&dev->tpg,
 691                                 dev->field_cap == V4L2_FIELD_NONE ||
 692                                 dev->field_cap == V4L2_FIELD_ALTERNATE);
 693
 694         /* Drop a certain percentage of buffers. */
 695         if (dev->perc_dropped_buffers &&
 696             prandom_u32_max(100) < dev->perc_dropped_buffers)
 697                 goto update_mv;
 698
 699         spin_lock(&dev->slock);
 700         if (!list_empty(&dev->vid_cap_active)) {
 701                 vid_cap_buf = list_entry(dev->vid_cap_active.next, struct vivid_buffer, list);
 702                 list_del(&vid_cap_buf->list);
 703         }
 704         if (!list_empty(&dev->vbi_cap_active)) {
 705                 if (dev->field_cap != V4L2_FIELD_ALTERNATE ||
 706                     (dev->vbi_cap_seq_count & 1)) {
 707                         vbi_cap_buf = list_entry(dev->vbi_cap_active.next,
 708                                                  struct vivid_buffer, list);
 709                         list_del(&vbi_cap_buf->list);
 710                 }
 711         }
 712         spin_unlock(&dev->slock);
 713
 714         if (!vid_cap_buf && !vbi_cap_buf)
 715                 goto update_mv;
 716
 717         if (vid_cap_buf) {
 718                 /* Fill buffer */
 719                 vivid_fillbuff(dev, vid_cap_buf);
 720                 dprintk(dev, 1, "filled buffer %d\n",
 721                         vid_cap_buf->vb.vb2_buf.index);
 722
 723                 /* Handle overlay */
 724                 if (dev->overlay_cap_owner && dev->fb_cap.base &&
 725                         dev->fb_cap.fmt.pixelformat == dev->fmt_cap->fourcc)
 726                         vivid_overlay(dev, vid_cap_buf);
 727
 728                 vb2_buffer_done(&vid_cap_buf->vb.vb2_buf, dev->dqbuf_error ?
 729                                 VB2_BUF_STATE_ERROR : VB2_BUF_STATE_DONE);
 730                 dprintk(dev, 2, "vid_cap buffer %d done\n",
 731                                 vid_cap_buf->vb.vb2_buf.index);
 732         }
 733
 734         if (vbi_cap_buf) {
 735                 if (dev->stream_sliced_vbi_cap)
 736                         vivid_sliced_vbi_cap_process(dev, vbi_cap_buf);
 737                 else
 738                         vivid_raw_vbi_cap_process(dev, vbi_cap_buf);
 739                 vb2_buffer_done(&vbi_cap_buf->vb.vb2_buf, dev->dqbuf_error ?
 740                                 VB2_BUF_STATE_ERROR : VB2_BUF_STATE_DONE);
 741                 dprintk(dev, 2, "vbi_cap %d done\n",
 742                                 vbi_cap_buf->vb.vb2_buf.index);
 743         }
 744         dev->dqbuf_error = false;
 745
 746 update_mv:
 747         /* Update the test pattern movement counters */
 748         tpg_update_mv_count(&dev->tpg, dev->field_cap == V4L2_FIELD_NONE ||
 749                                        dev->field_cap == V4L2_FIELD_ALTERNATE);
 750 }
 751
 752 static int vivid_thread_vid_cap(void *data)
 753 {
 754         struct vivid_dev *dev = data;
 755         u64 numerators_since_start;
 756         u64 buffers_since_start;
 757         u64 next_jiffies_since_start;
 758         unsigned long jiffies_since_start;
 759         unsigned long cur_jiffies;
 760         unsigned wait_jiffies;
 761         unsigned numerator;
 762         unsigned denominator;
 763         int dropped_bufs;
 764
 765         dprintk(dev, 1, "Video Capture Thread Start\n");
 766
 767         set_freezable();
 768
 769         /* Resets frame counters */
 770         dev->cap_seq_offset = 0;
 771         dev->cap_seq_count = 0;
 772         dev->cap_seq_resync = false;
 773         dev->jiffies_vid_cap = jiffies;
 774
 775         for (;;) {
 776                 try_to_freeze();
 777                 if (kthread_should_stop())
 778                         break;
 779
 780                 if (!mutex_trylock(&dev->mutex)) {
 781                         schedule_timeout_uninterruptible(1);
 782                         continue;
 783                 }
 784
 785                 cur_jiffies = jiffies;
 786                 if (dev->cap_seq_resync) {
 787                         dev->jiffies_vid_cap = cur_jiffies;
 788                         dev->cap_seq_offset = dev->cap_seq_count + 1;
 789                         dev->cap_seq_count = 0;
 790                         dev->cap_seq_resync = false;
 791                 }
 792                 numerator = dev->timeperframe_vid_cap.numerator;
 793                 denominator = dev->timeperframe_vid_cap.denominator;
 794
 795                 if (dev->field_cap == V4L2_FIELD_ALTERNATE)
 796                         denominator *= 2;
 797
 798                 /* Calculate the number of jiffies since we started streaming */
 799                 jiffies_since_start = cur_jiffies - dev->jiffies_vid_cap;
 800                 /* Get the number of buffers streamed since the start */
 801                 buffers_since_start = (u64)jiffies_since_start * denominator +
 802                                       (HZ * numerator) / 2;
 803                 do_div(buffers_since_start, HZ * numerator);
 804
 805                 /*
 806                  * After more than 0xf0000000 (rounded down to a multiple of
 807                  * 'jiffies-per-day' to ease jiffies_to_msecs calculation)
 808                  * jiffies have passed since we started streaming reset the
 809                  * counters and keep track of the sequence offset.
 810                  */
 811                 if (jiffies_since_start > JIFFIES_RESYNC) {
 812                         dev->jiffies_vid_cap = cur_jiffies;
 813                         dev->cap_seq_offset = buffers_since_start;
 814                         buffers_since_start = 0;
 815                 }
 816                 dropped_bufs = buffers_since_start + dev->cap_seq_offset - dev->cap_seq_count;
 817                 dev->cap_seq_count = buffers_since_start + dev->cap_seq_offset;
 818                 dev->vid_cap_seq_count = dev->cap_seq_count - dev->vid_cap_seq_start;
 819                 dev->vbi_cap_seq_count = dev->cap_seq_count - dev->vbi_cap_seq_start;
 820
 821                 vivid_thread_vid_cap_tick(dev, dropped_bufs);
 822
 823                 /*
 824                  * Calculate the number of 'numerators' streamed since we started,
 825                  * including the current buffer.
 826                  */
 827                 numerators_since_start = ++buffers_since_start * numerator;
 828
 829                 /* And the number of jiffies since we started */
 830                 jiffies_since_start = jiffies - dev->jiffies_vid_cap;
 831
 832                 mutex_unlock(&dev->mutex);
 833
 834                 /*
 835                  * Calculate when that next buffer is supposed to start
 836                  * in jiffies since we started streaming.
 837                  */
 838                 next_jiffies_since_start = numerators_since_start * HZ +
 839                                            denominator / 2;
 840                 do_div(next_jiffies_since_start, denominator);
 841                 /* If it is in the past, then just schedule asap */
 842                 if (next_jiffies_since_start < jiffies_since_start)
 843                         next_jiffies_since_start = jiffies_since_start;
 844
 845                 wait_jiffies = next_jiffies_since_start - jiffies_since_start;
 846                 schedule_timeout_interruptible(wait_jiffies ? wait_jiffies : 1);
 847         }
 848         dprintk(dev, 1, "Video Capture Thread End\n");
 849         return 0;
 850 }
 851
 852 static void vivid_grab_controls(struct vivid_dev *dev, bool grab)
 853 {
 854         v4l2_ctrl_grab(dev->ctrl_has_crop_cap, grab);
 855         v4l2_ctrl_grab(dev->ctrl_has_compose_cap, grab);
 856         v4l2_ctrl_grab(dev->ctrl_has_scaler_cap, grab);
 857 }
 858
 859 int vivid_start_generating_vid_cap(struct vivid_dev *dev, bool *pstreaming)
 860 {
 861         dprintk(dev, 1, "%s\n", __func__);
 862
 863         if (dev->kthread_vid_cap) {
 864                 u32 seq_count = dev->cap_seq_count + dev->seq_wrap * 128;
 865
 866                 if (pstreaming == &dev->vid_cap_streaming)
 867                         dev->vid_cap_seq_start = seq_count;
 868                 else
 869                         dev->vbi_cap_seq_start = seq_count;
 870                 *pstreaming = true;
 871                 return 0;
 872         }
 873
 874         /* Resets frame counters */
 875         tpg_init_mv_count(&dev->tpg);
 876
 877         dev->vid_cap_seq_start = dev->seq_wrap * 128;
 878         dev->vbi_cap_seq_start = dev->seq_wrap * 128;
 879
 880         dev->kthread_vid_cap = kthread_run(vivid_thread_vid_cap, dev,
 881                         "%s-vid-cap", dev->v4l2_dev.name);
 882
 883         if (IS_ERR(dev->kthread_vid_cap)) {
 884                 int err = PTR_ERR(dev->kthread_vid_cap);
 885
 886                 dev->kthread_vid_cap = NULL;
 887                 v4l2_err(&dev->v4l2_dev, "kernel_thread() failed\n");
 888                 return err;
 889         }
 890         *pstreaming = true;
 891         vivid_grab_controls(dev, true);
 892
 893         dprintk(dev, 1, "returning from %s\n", __func__);
 894         return 0;
 895 }
 896
 897 void vivid_stop_generating_vid_cap(struct vivid_dev *dev, bool *pstreaming)
 898 {
 899         dprintk(dev, 1, "%s\n", __func__);
 900
 901         if (dev->kthread_vid_cap == NULL)
 902                 return;
 903
 904         *pstreaming = false;
 905         if (pstreaming == &dev->vid_cap_streaming) {
 906                 /* Release all active buffers */
 907                 while (!list_empty(&dev->vid_cap_active)) {
 908                         struct vivid_buffer *buf;
 909
 910                         buf = list_entry(dev->vid_cap_active.next,
 911                                          struct vivid_buffer, list);
 912                         list_del(&buf->list);
 913                         vb2_buffer_done(&buf->vb.vb2_buf, VB2_BUF_STATE_ERROR);
 914                         dprintk(dev, 2, "vid_cap buffer %d done\n",
 915                                 buf->vb.vb2_buf.index);
 916                 }
 917         }
 918
 919         if (pstreaming == &dev->vbi_cap_streaming) {
 920                 while (!list_empty(&dev->vbi_cap_active)) {
 921                         struct vivid_buffer *buf;
 922
 923                         buf = list_entry(dev->vbi_cap_active.next,
 924                                          struct vivid_buffer, list);
 925                         list_del(&buf->list);
 926                         vb2_buffer_done(&buf->vb.vb2_buf, VB2_BUF_STATE_ERROR);
 927                         dprintk(dev, 2, "vbi_cap buffer %d done\n",
 928                                 buf->vb.vb2_buf.index);
 929                 }
 930         }
 931
 932         if (dev->vid_cap_streaming || dev->vbi_cap_streaming)
 933                 return;
 934
 935         /* shutdown control thread */
 936         vivid_grab_controls(dev, false);
 937         kthread_stop(dev->kthread_vid_cap);
 938         dev->kthread_vid_cap = NULL;
 939 }