git.oblomov.eu Git - linux-2.6/blob - drivers/media/video/vivi.c

   1 /*
   2  * Virtual Video driver - This code emulates a real video device with v4l2 api
   3  *
   4  * Copyright (c) 2006 by:
   5  *      Mauro Carvalho Chehab <mchehab--a.t--infradead.org>
   6  *      Ted Walther <ted--a.t--enumera.com>
   7  *      John Sokol <sokol--a.t--videotechnology.com>
   8  *      http://v4l.videotechnology.com/
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the BSD Licence, GNU General Public License
  12  * as published by the Free Software Foundation; either version 2 of the
  13  * License, or (at your option) any later version
  14  */
  15 #include <linux/module.h>
  16 #include <linux/delay.h>
  17 #include <linux/errno.h>
  18 #include <linux/fs.h>
  19 #include <linux/kernel.h>
  20 #include <linux/slab.h>
  21 #include <linux/mm.h>
  22 #include <linux/ioport.h>
  23 #include <linux/init.h>
  24 #include <linux/sched.h>
  25 #include <linux/pci.h>
  26 #include <linux/random.h>
  27 #include <linux/version.h>
  28 #include <linux/mutex.h>
  29 #include <linux/videodev2.h>
  30 #include <linux/dma-mapping.h>
  31 #ifdef CONFIG_VIDEO_V4L1_COMPAT
  32 /* Include V4L1 specific functions. Should be removed soon */
  33 #include <linux/videodev.h>
  34 #endif
  35 #include <linux/interrupt.h>
  36 #include <media/videobuf-vmalloc.h>
  37 #include <media/v4l2-common.h>
  38 #include <linux/kthread.h>
  39 #include <linux/highmem.h>
  40 #include <linux/freezer.h>
  41
  42 /* Wake up at about 30 fps */
  43 #define WAKE_NUMERATOR 30
  44 #define WAKE_DENOMINATOR 1001
  45 #define BUFFER_TIMEOUT     msecs_to_jiffies(500)  /* 0.5 seconds */
  46
  47 #include "font.h"
  48
  49 #define VIVI_MAJOR_VERSION 0
  50 #define VIVI_MINOR_VERSION 4
  51 #define VIVI_RELEASE 0
  52 #define VIVI_VERSION \
  53         KERNEL_VERSION(VIVI_MAJOR_VERSION, VIVI_MINOR_VERSION, VIVI_RELEASE)
  54
  55 /* Declare static vars that will be used as parameters */
  56 static unsigned int vid_limit = 16;     /* Video memory limit, in Mb */
  57 static int video_nr = -1;               /* /dev/videoN, -1 for autodetect */
  58 static int n_devs = 1;                  /* Number of virtual devices */
  59
  60 /* supported controls */
  61 static struct v4l2_queryctrl vivi_qctrl[] = {
  62         {
  63                 .id            = V4L2_CID_AUDIO_VOLUME,
  64                 .name          = "Volume",
  65                 .minimum       = 0,
  66                 .maximum       = 65535,
  67                 .step          = 65535/100,
  68                 .default_value = 65535,
  69                 .flags         = 0,
  70                 .type          = V4L2_CTRL_TYPE_INTEGER,
  71         }, {
  72                 .id            = V4L2_CID_BRIGHTNESS,
  73                 .type          = V4L2_CTRL_TYPE_INTEGER,
  74                 .name          = "Brightness",
  75                 .minimum       = 0,
  76                 .maximum       = 255,
  77                 .step          = 1,
  78                 .default_value = 127,
  79                 .flags         = 0,
  80         }, {
  81                 .id            = V4L2_CID_CONTRAST,
  82                 .type          = V4L2_CTRL_TYPE_INTEGER,
  83                 .name          = "Contrast",
  84                 .minimum       = 0,
  85                 .maximum       = 255,
  86                 .step          = 0x1,
  87                 .default_value = 0x10,
  88                 .flags         = 0,
  89         }, {
  90                 .id            = V4L2_CID_SATURATION,
  91                 .type          = V4L2_CTRL_TYPE_INTEGER,
  92                 .name          = "Saturation",
  93                 .minimum       = 0,
  94                 .maximum       = 255,
  95                 .step          = 0x1,
  96                 .default_value = 127,
  97                 .flags         = 0,
  98         }, {
  99                 .id            = V4L2_CID_HUE,
 100                 .type          = V4L2_CTRL_TYPE_INTEGER,
 101                 .name          = "Hue",
 102                 .minimum       = -128,
 103                 .maximum       = 127,
 104                 .step          = 0x1,
 105                 .default_value = 0,
 106                 .flags         = 0,
 107         }
 108 };
 109
 110 static int qctl_regs[ARRAY_SIZE(vivi_qctrl)];
 111
 112 #define dprintk(dev, level, fmt, arg...)                                \
 113         do {                                                            \
 114                 if (dev->vfd->debug >= (level))                         \
 115                         printk(KERN_DEBUG "vivi: " fmt , ## arg);       \
 116         } while (0)
 117
 118 /* ------------------------------------------------------------------
 119         Basic structures
 120    ------------------------------------------------------------------*/
 121
 122 struct vivi_fmt {
 123         char  *name;
 124         u32   fourcc;          /* v4l2 format id */
 125         int   depth;
 126 };
 127
 128 static struct vivi_fmt format = {
 129         .name     = "4:2:2, packed, YUYV",
 130         .fourcc   = V4L2_PIX_FMT_YUYV,
 131         .depth    = 16,
 132 };
 133
 134 struct sg_to_addr {
 135         int pos;
 136         struct scatterlist *sg;
 137 };
 138
 139 /* buffer for one video frame */
 140 struct vivi_buffer {
 141         /* common v4l buffer stuff -- must be first */
 142         struct videobuf_buffer vb;
 143
 144         struct vivi_fmt        *fmt;
 145 };
 146
 147 struct vivi_dmaqueue {
 148         struct list_head       active;
 149         struct list_head       queued;
 150         struct timer_list      timeout;
 151
 152         /* thread for generating video stream*/
 153         struct task_struct         *kthread;
 154         wait_queue_head_t          wq;
 155         /* Counters to control fps rate */
 156         int                        frame;
 157         int                        ini_jiffies;
 158 };
 159
 160 static LIST_HEAD(vivi_devlist);
 161
 162 struct vivi_dev {
 163         struct list_head           vivi_devlist;
 164
 165         struct mutex               lock;
 166         spinlock_t                 slock;
 167
 168         int                        users;
 169
 170         /* various device info */
 171         struct video_device        *vfd;
 172
 173         struct vivi_dmaqueue       vidq;
 174
 175         /* Several counters */
 176         int                        h, m, s, us, jiffies;
 177         char                       timestr[13];
 178
 179         int                        mv_count;    /* Controls bars movement */
 180 };
 181
 182 struct vivi_fh {
 183         struct vivi_dev            *dev;
 184
 185         /* video capture */
 186         struct vivi_fmt            *fmt;
 187         unsigned int               width, height;
 188         struct videobuf_queue      vb_vidq;
 189
 190         enum v4l2_buf_type         type;
 191 };
 192
 193 /* ------------------------------------------------------------------
 194         DMA and thread functions
 195    ------------------------------------------------------------------*/
 196
 197 /* Bars and Colors should match positions */
 198
 199 enum colors {
 200         WHITE,
 201         AMBAR,
 202         CYAN,
 203         GREEN,
 204         MAGENTA,
 205         RED,
 206         BLUE,
 207         BLACK,
 208 };
 209
 210 static u8 bars[8][3] = {
 211         /* R   G   B */
 212         {204, 204, 204},  /* white */
 213         {208, 208,   0},  /* ambar */
 214         {  0, 206, 206},  /* cyan */
 215         {  0, 239,   0},  /* green */
 216         {239,   0, 239},  /* magenta */
 217         {205,   0,   0},  /* red */
 218         {  0,   0, 255},  /* blue */
 219         {  0,   0,   0},  /* black */
 220 };
 221
 222 #define TO_Y(r, g, b) \
 223         (((16829 * r + 33039 * g + 6416 * b  + 32768) >> 16) + 16)
 224 /* RGB to  V(Cr) Color transform */
 225 #define TO_V(r, g, b) \
 226         (((28784 * r - 24103 * g - 4681 * b  + 32768) >> 16) + 128)
 227 /* RGB to  U(Cb) Color transform */
 228 #define TO_U(r, g, b) \
 229         (((-9714 * r - 19070 * g + 28784 * b + 32768) >> 16) + 128)
 230
 231 #define TSTAMP_MIN_Y 24
 232 #define TSTAMP_MAX_Y TSTAMP_MIN_Y+15
 233 #define TSTAMP_MIN_X 64
 234
 235 static void gen_line(char *basep, int inipos, int wmax,
 236                 int hmax, int line, int count, char *timestr)
 237 {
 238         int  w, i, j, y;
 239         int pos = inipos;
 240         char *p, *s;
 241         u8   chr, r, g, b, color;
 242
 243         /* We will just duplicate the second pixel at the packet */
 244         wmax /= 2;
 245
 246         /* Generate a standard color bar pattern */
 247         for (w = 0; w < wmax; w++) {
 248                 int colorpos = ((w + count) * 8/(wmax + 1)) % 8;
 249                 r = bars[colorpos][0];
 250                 g = bars[colorpos][1];
 251                 b = bars[colorpos][2];
 252
 253                 for (color = 0; color < 4; color++) {
 254                         p = basep + pos;
 255
 256                         switch (color) {
 257                         case 0:
 258                         case 2:
 259                                 *p = TO_Y(r, g, b);     /* Luma */
 260                                 break;
 261                         case 1:
 262                                 *p = TO_U(r, g, b);     /* Cb */
 263                                 break;
 264                         case 3:
 265                                 *p = TO_V(r, g, b);     /* Cr */
 266                                 break;
 267                         }
 268                         pos++;
 269                 }
 270         }
 271
 272         /* Checks if it is possible to show timestamp */
 273         if (TSTAMP_MAX_Y >= hmax)
 274                 goto end;
 275         if (TSTAMP_MIN_X + strlen(timestr) >= wmax)
 276                 goto end;
 277
 278         /* Print stream time */
 279         if (line >= TSTAMP_MIN_Y && line <= TSTAMP_MAX_Y) {
 280                 j = TSTAMP_MIN_X;
 281                 for (s = timestr; *s; s++) {
 282                         chr = rom8x16_bits[(*s-0x30)*16+line-TSTAMP_MIN_Y];
 283                         for (i = 0; i < 7; i++) {
 284                                 if (chr & 1 << (7 - i)) {
 285                                         /* Font color*/
 286                                         r = 0;
 287                                         g = 198;
 288                                         b = 0;
 289                                 } else {
 290                                         /* Background color */
 291                                         r = bars[BLACK][0];
 292                                         g = bars[BLACK][1];
 293                                         b = bars[BLACK][2];
 294                                 }
 295
 296                                 pos = inipos + j * 2;
 297                                 for (color = 0; color < 4; color++) {
 298                                         p = basep + pos;
 299
 300                                         y = TO_Y(r, g, b);
 301
 302                                         switch (color) {
 303                                         case 0:
 304                                         case 2:
 305                                                 *p = TO_Y(r, g, b); /* Luma */
 306                                                 break;
 307                                         case 1:
 308                                                 *p = TO_U(r, g, b); /* Cb */
 309                                                 break;
 310                                         case 3:
 311                                                 *p = TO_V(r, g, b); /* Cr */
 312                                                 break;
 313                                         }
 314                                         pos++;
 315                                 }
 316                                 j++;
 317                         }
 318                 }
 319         }
 320
 321 end:
 322         return;
 323 }
 324 static void vivi_fillbuff(struct vivi_dev *dev, struct vivi_buffer *buf)
 325 {
 326         int h , pos = 0;
 327         int hmax  = buf->vb.height;
 328         int wmax  = buf->vb.width;
 329         struct timeval ts;
 330         char *tmpbuf = kmalloc(wmax * 2, GFP_KERNEL);
 331         void *vbuf = videobuf_to_vmalloc(&buf->vb);
 332
 333         if (!tmpbuf)
 334                 return;
 335
 336         for (h = 0; h < hmax; h++) {
 337                 gen_line(tmpbuf, 0, wmax, hmax, h, dev->mv_count,
 338                          dev->timestr);
 339                 /* FIXME: replacing to __copy_to_user */
 340                 if (copy_to_user(vbuf + pos, tmpbuf, wmax * 2) != 0)
 341                         dprintk(dev, 2, "vivifill copy_to_user failed.\n");
 342                 pos += wmax*2;
 343         }
 344
 345         dev->mv_count++;
 346
 347         kfree(tmpbuf);
 348
 349         /* Updates stream time */
 350
 351         dev->us += jiffies_to_usecs(jiffies-dev->jiffies);
 352         dev->jiffies = jiffies;
 353         if (dev->us >= 1000000) {
 354                 dev->us -= 1000000;
 355                 dev->s++;
 356                 if (dev->s >= 60) {
 357                         dev->s -= 60;
 358                         dev->m++;
 359                         if (dev->m > 60) {
 360                                 dev->m -= 60;
 361                                 dev->h++;
 362                                 if (dev->h > 24)
 363                                         dev->h -= 24;
 364                         }
 365                 }
 366         }
 367         sprintf(dev->timestr, "%02d:%02d:%02d:%03d",
 368                         dev->h, dev->m, dev->s,  (dev->us + 500) / 1000);
 369
 370         dprintk(dev, 2, "vivifill at %s: Buffer 0x%08lx size= %d\n",
 371                         dev->timestr, (unsigned long)tmpbuf, pos);
 372
 373         /* Advice that buffer was filled */
 374         buf->vb.state = VIDEOBUF_DONE;
 375         buf->vb.field_count++;
 376         do_gettimeofday(&ts);
 377         buf->vb.ts = ts;
 378
 379         list_del(&buf->vb.queue);
 380         wake_up(&buf->vb.done);
 381 }
 382
 383 static int restart_video_queue(struct vivi_dmaqueue *dma_q);
 384
 385 static void vivi_thread_tick(struct vivi_dmaqueue  *dma_q)
 386 {
 387         struct vivi_buffer    *buf;
 388         struct vivi_dev *dev = container_of(dma_q, struct vivi_dev, vidq);
 389
 390         int bc;
 391
 392         spin_lock(&dev->slock);
 393         /* Announces videobuf that all went ok */
 394         for (bc = 0;; bc++) {
 395                 if (list_empty(&dma_q->active)) {
 396                         dprintk(dev, 1, "No active queue to serve\n");
 397                         break;
 398                 }
 399
 400                 buf = list_entry(dma_q->active.next,
 401                                  struct vivi_buffer, vb.queue);
 402
 403                 /* Nobody is waiting something to be done, just return */
 404                 if (!waitqueue_active(&buf->vb.done)) {
 405                         mod_timer(&dma_q->timeout, jiffies+BUFFER_TIMEOUT);
 406                         spin_unlock(&dev->slock);
 407                         return;
 408                 }
 409
 410                 do_gettimeofday(&buf->vb.ts);
 411                 dprintk(dev, 2, "[%p/%d] wakeup\n", buf, buf->vb. i);
 412
 413                 /* Fill buffer */
 414                 vivi_fillbuff(dev, buf);
 415
 416                 if (list_empty(&dma_q->active)) {
 417                         del_timer(&dma_q->timeout);
 418                 } else {
 419                         mod_timer(&dma_q->timeout, jiffies + BUFFER_TIMEOUT);
 420                 }
 421         }
 422         if (bc != 1)
 423                 dprintk(dev, 1, "%s: %d buffers handled (should be 1)\n",
 424                         __FUNCTION__, bc);
 425         spin_unlock(&dev->slock);
 426 }
 427
 428 static void vivi_sleep(struct vivi_dmaqueue  *dma_q)
 429 {
 430         struct vivi_dev *dev = container_of(dma_q, struct vivi_dev, vidq);
 431         int timeout;
 432         DECLARE_WAITQUEUE(wait, current);
 433
 434         dprintk(dev, 1, "%s dma_q=0x%08lx\n", __FUNCTION__,
 435                 (unsigned long)dma_q);
 436
 437         add_wait_queue(&dma_q->wq, &wait);
 438         if (!kthread_should_stop()) {
 439                 dma_q->frame++;
 440
 441                 /* Calculate time to wake up */
 442                 timeout = dma_q->ini_jiffies+
 443                           msecs_to_jiffies((dma_q->frame*WAKE_NUMERATOR * 1000)
 444                                            / WAKE_DENOMINATOR) - jiffies;
 445
 446                 if (timeout <= 0) {
 447                         int old = dma_q->frame;
 448                         dma_q->frame = (jiffies_to_msecs(jiffies -
 449                                         dma_q->ini_jiffies) *
 450                                         WAKE_DENOMINATOR) /
 451                                         (WAKE_NUMERATOR * 1000) + 1;
 452
 453                         timeout = dma_q->ini_jiffies+
 454                                 msecs_to_jiffies((dma_q->frame *
 455                                                   WAKE_NUMERATOR * 1000)
 456                                                   / WAKE_DENOMINATOR) - jiffies;
 457
 458                         dprintk(dev, 1, "underrun, losed %d frames. "
 459                                    "Now, frame is %d. Waking on %d jiffies\n",
 460                                    dma_q->frame-old, dma_q->frame, timeout);
 461                 } else
 462                         dprintk(dev, 1, "will sleep for %i jiffies\n",
 463                                 timeout);
 464
 465                 vivi_thread_tick(dma_q);
 466
 467                 schedule_timeout_interruptible(timeout);
 468         }
 469
 470         remove_wait_queue(&dma_q->wq, &wait);
 471         try_to_freeze();
 472 }
 473
 474 static int vivi_thread(void *data)
 475 {
 476         struct vivi_dmaqueue  *dma_q = data;
 477         struct vivi_dev *dev = container_of(dma_q, struct vivi_dev, vidq);
 478
 479         dprintk(dev, 1, "thread started\n");
 480
 481         mod_timer(&dma_q->timeout, jiffies+BUFFER_TIMEOUT);
 482         set_freezable();
 483
 484         for (;;) {
 485                 vivi_sleep(dma_q);
 486
 487                 if (kthread_should_stop())
 488                         break;
 489         }
 490         dprintk(dev, 1, "thread: exit\n");
 491         return 0;
 492 }
 493
 494 static int vivi_start_thread(struct vivi_dmaqueue  *dma_q)
 495 {
 496         struct vivi_dev *dev = container_of(dma_q, struct vivi_dev, vidq);
 497
 498         dma_q->frame = 0;
 499         dma_q->ini_jiffies = jiffies;
 500
 501         dprintk(dev, 1, "%s\n", __FUNCTION__);
 502
 503         dma_q->kthread = kthread_run(vivi_thread, dma_q, "vivi");
 504
 505         if (IS_ERR(dma_q->kthread)) {
 506                 printk(KERN_ERR "vivi: kernel_thread() failed\n");
 507                 return PTR_ERR(dma_q->kthread);
 508         }
 509         /* Wakes thread */
 510         wake_up_interruptible(&dma_q->wq);
 511
 512         dprintk(dev, 1, "returning from %s\n", __FUNCTION__);
 513         return 0;
 514 }
 515
 516 static void vivi_stop_thread(struct vivi_dmaqueue  *dma_q)
 517 {
 518         struct vivi_dev *dev = container_of(dma_q, struct vivi_dev, vidq);
 519
 520         dprintk(dev, 1, "%s\n", __FUNCTION__);
 521         /* shutdown control thread */
 522         if (dma_q->kthread) {
 523                 kthread_stop(dma_q->kthread);
 524                 dma_q->kthread = NULL;
 525         }
 526 }
 527
 528 static int restart_video_queue(struct vivi_dmaqueue *dma_q)
 529 {
 530         struct vivi_dev *dev = container_of(dma_q, struct vivi_dev, vidq);
 531         struct vivi_buffer *buf, *prev;
 532
 533         dprintk(dev, 1, "%s dma_q=0x%08lx\n", __FUNCTION__,
 534                 (unsigned long)dma_q);
 535
 536         if (!list_empty(&dma_q->active)) {
 537                 buf = list_entry(dma_q->active.next,
 538                                  struct vivi_buffer, vb.queue);
 539                 dprintk(dev, 2, "restart_queue [%p/%d]: restart dma\n",
 540                         buf, buf->vb.i);
 541
 542                 dprintk(dev, 1, "Restarting video dma\n");
 543                 vivi_stop_thread(dma_q);
 544
 545                 /* cancel all outstanding capture / vbi requests */
 546                 list_for_each_entry_safe(buf, prev, &dma_q->active, vb.queue) {
 547                         list_del(&buf->vb.queue);
 548                         buf->vb.state = VIDEOBUF_ERROR;
 549                         wake_up(&buf->vb.done);
 550                 }
 551                 mod_timer(&dma_q->timeout, jiffies+BUFFER_TIMEOUT);
 552
 553                 return 0;
 554         }
 555
 556         prev = NULL;
 557         for (;;) {
 558                 if (list_empty(&dma_q->queued))
 559                         return 0;
 560                 buf = list_entry(dma_q->queued.next,
 561                                  struct vivi_buffer, vb.queue);
 562                 if (NULL == prev) {
 563                         list_del(&buf->vb.queue);
 564                         list_add_tail(&buf->vb.queue, &dma_q->active);
 565
 566                         dprintk(dev, 1, "Restarting video dma\n");
 567                         vivi_stop_thread(dma_q);
 568                         vivi_start_thread(dma_q);
 569
 570                         buf->vb.state = VIDEOBUF_ACTIVE;
 571                         mod_timer(&dma_q->timeout, jiffies+BUFFER_TIMEOUT);
 572                         dprintk(dev, 2,
 573                                 "[%p/%d] restart_queue - first active\n",
 574                                 buf, buf->vb.i);
 575
 576                 } else if (prev->vb.width  == buf->vb.width  &&
 577                            prev->vb.height == buf->vb.height &&
 578                            prev->fmt       == buf->fmt) {
 579                         list_del(&buf->vb.queue);
 580                         list_add_tail(&buf->vb.queue, &dma_q->active);
 581                         buf->vb.state = VIDEOBUF_ACTIVE;
 582                         dprintk(dev, 2,
 583                                 "[%p/%d] restart_queue - move to active\n",
 584                                 buf, buf->vb.i);
 585                 } else {
 586                         return 0;
 587                 }
 588                 prev = buf;
 589         }
 590 }
 591
 592 static void vivi_vid_timeout(unsigned long data)
 593 {
 594         struct vivi_dev      *dev  = (struct vivi_dev *)data;
 595         struct vivi_dmaqueue *vidq = &dev->vidq;
 596         struct vivi_buffer   *buf;
 597
 598         spin_lock(&dev->slock);
 599
 600         while (!list_empty(&vidq->active)) {
 601                 buf = list_entry(vidq->active.next,
 602                                  struct vivi_buffer, vb.queue);
 603                 list_del(&buf->vb.queue);
 604                 buf->vb.state = VIDEOBUF_ERROR;
 605                 wake_up(&buf->vb.done);
 606                 printk(KERN_INFO "vivi/0: [%p/%d] timeout\n", buf, buf->vb.i);
 607         }
 608         restart_video_queue(vidq);
 609
 610         spin_unlock(&dev->slock);
 611 }
 612
 613 /* ------------------------------------------------------------------
 614         Videobuf operations
 615    ------------------------------------------------------------------*/
 616 static int
 617 buffer_setup(struct videobuf_queue *vq, unsigned int *count, unsigned int *size)
 618 {
 619         struct vivi_fh  *fh = vq->priv_data;
 620         struct vivi_dev *dev  = fh->dev;
 621
 622         *size = fh->width*fh->height*2;
 623
 624         if (0 == *count)
 625                 *count = 32;
 626
 627         while (*size * *count > vid_limit * 1024 * 1024)
 628                 (*count)--;
 629
 630         dprintk(dev, 1, "%s, count=%d, size=%d\n", __FUNCTION__,
 631                 *count, *size);
 632
 633         return 0;
 634 }
 635
 636 static void free_buffer(struct videobuf_queue *vq, struct vivi_buffer *buf)
 637 {
 638         struct vivi_fh  *fh = vq->priv_data;
 639         struct vivi_dev *dev  = fh->dev;
 640
 641         dprintk(dev, 1, "%s\n", __FUNCTION__);
 642
 643         if (in_interrupt())
 644                 BUG();
 645
 646         videobuf_waiton(&buf->vb, 0, 0);
 647         videobuf_vmalloc_free(&buf->vb);
 648         buf->vb.state = VIDEOBUF_NEEDS_INIT;
 649 }
 650
 651 #define norm_maxw() 1024
 652 #define norm_maxh() 768
 653 static int
 654 buffer_prepare(struct videobuf_queue *vq, struct videobuf_buffer *vb,
 655                                                 enum v4l2_field field)
 656 {
 657         struct vivi_fh     *fh  = vq->priv_data;
 658         struct vivi_dev    *dev = fh->dev;
 659         struct vivi_buffer *buf = container_of(vb, struct vivi_buffer, vb);
 660         int rc, init_buffer = 0;
 661
 662         dprintk(dev, 1, "%s, field=%d\n", __FUNCTION__, field);
 663
 664         BUG_ON(NULL == fh->fmt);
 665         if (fh->width  < 48 || fh->width  > norm_maxw() ||
 666             fh->height < 32 || fh->height > norm_maxh())
 667                 return -EINVAL;
 668         buf->vb.size = fh->width*fh->height*2;
 669         if (0 != buf->vb.baddr  &&  buf->vb.bsize < buf->vb.size)
 670                 return -EINVAL;
 671
 672         if (buf->fmt       != fh->fmt    ||
 673             buf->vb.width  != fh->width  ||
 674             buf->vb.height != fh->height ||
 675         buf->vb.field  != field) {
 676                 buf->fmt       = fh->fmt;
 677                 buf->vb.width  = fh->width;
 678                 buf->vb.height = fh->height;
 679                 buf->vb.field  = field;
 680                 init_buffer = 1;
 681         }
 682
 683         if (VIDEOBUF_NEEDS_INIT == buf->vb.state) {
 684                 rc = videobuf_iolock(vq, &buf->vb, NULL);
 685                 if (rc < 0)
 686                         goto fail;
 687         }
 688
 689         buf->vb.state = VIDEOBUF_PREPARED;
 690
 691         return 0;
 692
 693 fail:
 694         free_buffer(vq, buf);
 695         return rc;
 696 }
 697
 698 static void
 699 buffer_queue(struct videobuf_queue *vq, struct videobuf_buffer *vb)
 700 {
 701         struct vivi_buffer    *buf  = container_of(vb, struct vivi_buffer, vb);
 702         struct vivi_fh        *fh   = vq->priv_data;
 703         struct vivi_dev       *dev  = fh->dev;
 704         struct vivi_dmaqueue  *vidq = &dev->vidq;
 705         struct vivi_buffer    *prev;
 706
 707         if (!list_empty(&vidq->queued)) {
 708                 dprintk(dev, 1, "adding vb queue=0x%08lx\n",
 709                         (unsigned long)&buf->vb.queue);
 710                 list_add_tail(&buf->vb.queue, &vidq->queued);
 711                 buf->vb.state = VIDEOBUF_QUEUED;
 712                 dprintk(dev, 2, "[%p/%d] buffer_queue - append to queued\n",
 713                         buf, buf->vb.i);
 714         } else if (list_empty(&vidq->active)) {
 715                 list_add_tail(&buf->vb.queue, &vidq->active);
 716
 717                 buf->vb.state = VIDEOBUF_ACTIVE;
 718                 mod_timer(&vidq->timeout, jiffies+BUFFER_TIMEOUT);
 719                 dprintk(dev, 2, "[%p/%d] buffer_queue - first active\n",
 720                         buf, buf->vb.i);
 721
 722                 vivi_start_thread(vidq);
 723         } else {
 724                 prev = list_entry(vidq->active.prev,
 725                                   struct vivi_buffer, vb.queue);
 726                 if (prev->vb.width  == buf->vb.width  &&
 727                     prev->vb.height == buf->vb.height &&
 728                     prev->fmt       == buf->fmt) {
 729                         list_add_tail(&buf->vb.queue, &vidq->active);
 730                         buf->vb.state = VIDEOBUF_ACTIVE;
 731                         dprintk(dev, 2,
 732                                 "[%p/%d] buffer_queue - append to active\n",
 733                                 buf, buf->vb.i);
 734
 735                 } else {
 736                         list_add_tail(&buf->vb.queue, &vidq->queued);
 737                         buf->vb.state = VIDEOBUF_QUEUED;
 738                         dprintk(dev, 2,
 739                                 "[%p/%d] buffer_queue - first queued\n",
 740                                 buf, buf->vb.i);
 741                 }
 742         }
 743 }
 744
 745 static void buffer_release(struct videobuf_queue *vq,
 746                            struct videobuf_buffer *vb)
 747 {
 748         struct vivi_buffer   *buf  = container_of(vb, struct vivi_buffer, vb);
 749         struct vivi_fh       *fh   = vq->priv_data;
 750         struct vivi_dev      *dev  = (struct vivi_dev *)fh->dev;
 751         struct vivi_dmaqueue *vidq = &dev->vidq;
 752
 753         dprintk(dev, 1, "%s\n", __FUNCTION__);
 754
 755         vivi_stop_thread(vidq);
 756
 757         free_buffer(vq, buf);
 758 }
 759
 760 static struct videobuf_queue_ops vivi_video_qops = {
 761         .buf_setup      = buffer_setup,
 762         .buf_prepare    = buffer_prepare,
 763         .buf_queue      = buffer_queue,
 764         .buf_release    = buffer_release,
 765 };
 766
 767 /* ------------------------------------------------------------------
 768         IOCTL vidioc handling
 769    ------------------------------------------------------------------*/
 770 static int vidioc_querycap(struct file *file, void  *priv,
 771                                         struct v4l2_capability *cap)
 772 {
 773         strcpy(cap->driver, "vivi");
 774         strcpy(cap->card, "vivi");
 775         cap->version = VIVI_VERSION;
 776         cap->capabilities =     V4L2_CAP_VIDEO_CAPTURE |
 777                                 V4L2_CAP_STREAMING     |
 778                                 V4L2_CAP_READWRITE;
 779         return 0;
 780 }
 781
 782 static int vidioc_enum_fmt_cap(struct file *file, void  *priv,
 783                                         struct v4l2_fmtdesc *f)
 784 {
 785         if (f->index > 0)
 786                 return -EINVAL;
 787
 788         strlcpy(f->description, format.name, sizeof(f->description));
 789         f->pixelformat = format.fourcc;
 790         return 0;
 791 }
 792
 793 static int vidioc_g_fmt_cap(struct file *file, void *priv,
 794                                         struct v4l2_format *f)
 795 {
 796         struct vivi_fh *fh = priv;
 797
 798         f->fmt.pix.width        = fh->width;
 799         f->fmt.pix.height       = fh->height;
 800         f->fmt.pix.field        = fh->vb_vidq.field;
 801         f->fmt.pix.pixelformat  = fh->fmt->fourcc;
 802         f->fmt.pix.bytesperline =
 803                 (f->fmt.pix.width * fh->fmt->depth) >> 3;
 804         f->fmt.pix.sizeimage =
 805                 f->fmt.pix.height * f->fmt.pix.bytesperline;
 806
 807         return (0);
 808 }
 809
 810 static int vidioc_try_fmt_cap(struct file *file, void *priv,
 811                         struct v4l2_format *f)
 812 {
 813         struct vivi_fh  *fh  = priv;
 814         struct vivi_dev *dev = fh->dev;
 815         struct vivi_fmt *fmt;
 816         enum v4l2_field field;
 817         unsigned int maxw, maxh;
 818
 819         if (format.fourcc != f->fmt.pix.pixelformat) {
 820                 dprintk(dev, 1, "Fourcc format (0x%08x) invalid. "
 821                         "Driver accepts only 0x%08x\n",
 822                         f->fmt.pix.pixelformat, format.fourcc);
 823                 return -EINVAL;
 824         }
 825         fmt = &format;
 826
 827         field = f->fmt.pix.field;
 828
 829         if (field == V4L2_FIELD_ANY) {
 830                 field = V4L2_FIELD_INTERLACED;
 831         } else if (V4L2_FIELD_INTERLACED != field) {
 832                 dprintk(dev, 1, "Field type invalid.\n");
 833                 return -EINVAL;
 834         }
 835
 836         maxw  = norm_maxw();
 837         maxh  = norm_maxh();
 838
 839         f->fmt.pix.field = field;
 840         if (f->fmt.pix.height < 32)
 841                 f->fmt.pix.height = 32;
 842         if (f->fmt.pix.height > maxh)
 843                 f->fmt.pix.height = maxh;
 844         if (f->fmt.pix.width < 48)
 845                 f->fmt.pix.width = 48;
 846         if (f->fmt.pix.width > maxw)
 847                 f->fmt.pix.width = maxw;
 848         f->fmt.pix.width &= ~0x03;
 849         f->fmt.pix.bytesperline =
 850                 (f->fmt.pix.width * fmt->depth) >> 3;
 851         f->fmt.pix.sizeimage =
 852                 f->fmt.pix.height * f->fmt.pix.bytesperline;
 853
 854         return 0;
 855 }
 856
 857 /*FIXME: This seems to be generic enough to be at videodev2 */
 858 static int vidioc_s_fmt_cap(struct file *file, void *priv,
 859                                         struct v4l2_format *f)
 860 {
 861         struct vivi_fh  *fh = priv;
 862         int ret = vidioc_try_fmt_cap(file, fh, f);
 863         if (ret < 0)
 864                 return (ret);
 865
 866         fh->fmt           = &format;
 867         fh->width         = f->fmt.pix.width;
 868         fh->height        = f->fmt.pix.height;
 869         fh->vb_vidq.field = f->fmt.pix.field;
 870         fh->type          = f->type;
 871
 872         return (0);
 873 }
 874
 875 static int vidioc_reqbufs(struct file *file, void *priv,
 876                           struct v4l2_requestbuffers *p)
 877 {
 878         struct vivi_fh  *fh = priv;
 879
 880         return (videobuf_reqbufs(&fh->vb_vidq, p));
 881 }
 882
 883 static int vidioc_querybuf(struct file *file, void *priv, struct v4l2_buffer *p)
 884 {
 885         struct vivi_fh  *fh = priv;
 886
 887         return (videobuf_querybuf(&fh->vb_vidq, p));
 888 }
 889
 890 static int vidioc_qbuf(struct file *file, void *priv, struct v4l2_buffer *p)
 891 {
 892         struct vivi_fh *fh = priv;
 893
 894         return (videobuf_qbuf(&fh->vb_vidq, p));
 895 }
 896
 897 static int vidioc_dqbuf(struct file *file, void *priv, struct v4l2_buffer *p)
 898 {
 899         struct vivi_fh  *fh = priv;
 900
 901         return (videobuf_dqbuf(&fh->vb_vidq, p,
 902                                 file->f_flags & O_NONBLOCK));
 903 }
 904
 905 #ifdef CONFIG_VIDEO_V4L1_COMPAT
 906 static int vidiocgmbuf(struct file *file, void *priv, struct video_mbuf *mbuf)
 907 {
 908         struct vivi_fh  *fh = priv;
 909
 910         return videobuf_cgmbuf(&fh->vb_vidq, mbuf, 8);
 911 }
 912 #endif
 913
 914 static int vidioc_streamon(struct file *file, void *priv, enum v4l2_buf_type i)
 915 {
 916         struct vivi_fh  *fh = priv;
 917
 918         if (fh->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
 919                 return -EINVAL;
 920         if (i != fh->type)
 921                 return -EINVAL;
 922
 923         return videobuf_streamon(&fh->vb_vidq);
 924 }
 925
 926 static int vidioc_streamoff(struct file *file, void *priv, enum v4l2_buf_type i)
 927 {
 928         struct vivi_fh  *fh = priv;
 929
 930         if (fh->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
 931                 return -EINVAL;
 932         if (i != fh->type)
 933                 return -EINVAL;
 934
 935         return videobuf_streamoff(&fh->vb_vidq);
 936 }
 937
 938 static int vidioc_s_std(struct file *file, void *priv, v4l2_std_id *i)
 939 {
 940         return 0;
 941 }
 942
 943 /* only one input in this sample driver */
 944 static int vidioc_enum_input(struct file *file, void *priv,
 945                                 struct v4l2_input *inp)
 946 {
 947         if (inp->index != 0)
 948                 return -EINVAL;
 949
 950         inp->type = V4L2_INPUT_TYPE_CAMERA;
 951         inp->std = V4L2_STD_525_60;
 952         strcpy(inp->name, "Camera");
 953
 954         return (0);
 955 }
 956
 957 static int vidioc_g_input(struct file *file, void *priv, unsigned int *i)
 958 {
 959         *i = 0;
 960
 961         return (0);
 962 }
 963 static int vidioc_s_input(struct file *file, void *priv, unsigned int i)
 964 {
 965         if (i > 0)
 966                 return -EINVAL;
 967
 968         return (0);
 969 }
 970
 971         /* --- controls ---------------------------------------------- */
 972 static int vidioc_queryctrl(struct file *file, void *priv,
 973                             struct v4l2_queryctrl *qc)
 974 {
 975         int i;
 976
 977         for (i = 0; i < ARRAY_SIZE(vivi_qctrl); i++)
 978                 if (qc->id && qc->id == vivi_qctrl[i].id) {
 979                         memcpy(qc, &(vivi_qctrl[i]),
 980                                 sizeof(*qc));
 981                         return (0);
 982                 }
 983
 984         return -EINVAL;
 985 }
 986
 987 static int vidioc_g_ctrl(struct file *file, void *priv,
 988                          struct v4l2_control *ctrl)
 989 {
 990         int i;
 991
 992         for (i = 0; i < ARRAY_SIZE(vivi_qctrl); i++)
 993                 if (ctrl->id == vivi_qctrl[i].id) {
 994                         ctrl->value = qctl_regs[i];
 995                         return (0);
 996                 }
 997
 998         return -EINVAL;
 999 }
1000 static int vidioc_s_ctrl(struct file *file, void *priv,
1001                                 struct v4l2_control *ctrl)
1002 {
1003         int i;
1004
1005         for (i = 0; i < ARRAY_SIZE(vivi_qctrl); i++)
1006                 if (ctrl->id == vivi_qctrl[i].id) {
1007                         if (ctrl->value < vivi_qctrl[i].minimum
1008                             || ctrl->value > vivi_qctrl[i].maximum) {
1009                                         return (-ERANGE);
1010                                 }
1011                         qctl_regs[i] = ctrl->value;
1012                         return (0);
1013                 }
1014         return -EINVAL;
1015 }
1016
1017 /* ------------------------------------------------------------------
1018         File operations for the device
1019    ------------------------------------------------------------------*/
1020
1021 #define line_buf_size(norm) (norm_maxw(norm)*(format.depth+7)/8)
1022
1023 static int vivi_open(struct inode *inode, struct file *file)
1024 {
1025         int minor = iminor(inode);
1026         struct vivi_dev *dev;
1027         struct vivi_fh *fh;
1028         int i;
1029
1030         printk(KERN_DEBUG "vivi: open called (minor=%d)\n", minor);
1031
1032         list_for_each_entry(dev, &vivi_devlist, vivi_devlist)
1033                 if (dev->vfd->minor == minor)
1034                         goto found;
1035         return -ENODEV;
1036
1037 found:
1038         /* If more than one user, mutex should be added */
1039         dev->users++;
1040
1041         dprintk(dev, 1, "open minor=%d type=%s users=%d\n", minor,
1042                 v4l2_type_names[V4L2_BUF_TYPE_VIDEO_CAPTURE], dev->users);
1043
1044         /* allocate + initialize per filehandle data */
1045         fh = kzalloc(sizeof(*fh), GFP_KERNEL);
1046         if (NULL == fh) {
1047                 dev->users--;
1048                 return -ENOMEM;
1049         }
1050
1051         file->private_data = fh;
1052         fh->dev      = dev;
1053
1054         fh->type     = V4L2_BUF_TYPE_VIDEO_CAPTURE;
1055         fh->fmt      = &format;
1056         fh->width    = 640;
1057         fh->height   = 480;
1058
1059         /* Put all controls at a sane state */
1060         for (i = 0; i < ARRAY_SIZE(vivi_qctrl); i++)
1061                 qctl_regs[i] = vivi_qctrl[i].default_value;
1062
1063         /* Resets frame counters */
1064         dev->h = 0;
1065         dev->m = 0;
1066         dev->s = 0;
1067         dev->us = 0;
1068         dev->mv_count = 0;
1069         dev->jiffies = jiffies;
1070         sprintf(dev->timestr, "%02d:%02d:%02d:%03d",
1071                         dev->h, dev->m, dev->s, (dev->us + 500) / 1000);
1072
1073         videobuf_queue_vmalloc_init(&fh->vb_vidq, &vivi_video_qops,
1074                         NULL, &dev->slock, fh->type, V4L2_FIELD_INTERLACED,
1075                         sizeof(struct vivi_buffer), fh);
1076
1077         return 0;
1078 }
1079
1080 static ssize_t
1081 vivi_read(struct file *file, char __user *data, size_t count, loff_t *ppos)
1082 {
1083         struct vivi_fh *fh = file->private_data;
1084
1085         if (fh->type == V4L2_BUF_TYPE_VIDEO_CAPTURE) {
1086                 return videobuf_read_stream(&fh->vb_vidq, data, count, ppos, 0,
1087                                         file->f_flags & O_NONBLOCK);
1088         }
1089         return 0;
1090 }
1091
1092 static unsigned int
1093 vivi_poll(struct file *file, struct poll_table_struct *wait)
1094 {
1095         struct vivi_fh        *fh = file->private_data;
1096         struct vivi_dev       *dev = fh->dev;
1097         struct videobuf_queue *q = &fh->vb_vidq;
1098
1099         dprintk(dev, 1, "%s\n", __FUNCTION__);
1100
1101         if (V4L2_BUF_TYPE_VIDEO_CAPTURE != fh->type)
1102                 return POLLERR;
1103
1104         return videobuf_poll_stream(file, q, wait);
1105 }
1106
1107 static int vivi_close(struct inode *inode, struct file *file)
1108 {
1109         struct vivi_fh         *fh = file->private_data;
1110         struct vivi_dev *dev       = fh->dev;
1111         struct vivi_dmaqueue *vidq = &dev->vidq;
1112
1113         int minor = iminor(inode);
1114
1115         vivi_stop_thread(vidq);
1116         videobuf_stop(&fh->vb_vidq);
1117         videobuf_mmap_free(&fh->vb_vidq);
1118
1119         kfree(fh);
1120
1121         dev->users--;
1122
1123         dprintk(dev, 1, "close called (minor=%d, users=%d)\n",
1124                 minor, dev->users);
1125
1126         return 0;
1127 }
1128
1129 static int vivi_release(void)
1130 {
1131         struct vivi_dev *dev;
1132         struct list_head *list;
1133
1134         while (!list_empty(&vivi_devlist)) {
1135                 list = vivi_devlist.next;
1136                 list_del(list);
1137                 dev = list_entry(list, struct vivi_dev, vivi_devlist);
1138
1139                 if (-1 != dev->vfd->minor)
1140                         video_unregister_device(dev->vfd);
1141                 else
1142                         video_device_release(dev->vfd);
1143
1144                 kfree(dev);
1145         }
1146
1147         return 0;
1148 }
1149
1150 static int vivi_mmap(struct file *file, struct vm_area_struct *vma)
1151 {
1152         struct vivi_fh  *fh = file->private_data;
1153         struct vivi_dev *dev = fh->dev;
1154         int ret;
1155
1156         dprintk(dev, 1, "mmap called, vma=0x%08lx\n", (unsigned long)vma);
1157
1158         ret = videobuf_mmap_mapper(&fh->vb_vidq, vma);
1159
1160         dprintk(dev, 1, "vma start=0x%08lx, size=%ld, ret=%d\n",
1161                 (unsigned long)vma->vm_start,
1162                 (unsigned long)vma->vm_end-(unsigned long)vma->vm_start,
1163                 ret);
1164
1165         return ret;
1166 }
1167
1168 static const struct file_operations vivi_fops = {
1169         .owner          = THIS_MODULE,
1170         .open           = vivi_open,
1171         .release        = vivi_close,
1172         .read           = vivi_read,
1173         .poll           = vivi_poll,
1174         .ioctl          = video_ioctl2, /* V4L2 ioctl handler */
1175         .mmap           = vivi_mmap,
1176         .llseek         = no_llseek,
1177 };
1178
1179 static struct video_device vivi_template = {
1180         .name           = "vivi",
1181         .type           = VID_TYPE_CAPTURE,
1182         .fops           = &vivi_fops,
1183         .minor          = -1,
1184         .release        = video_device_release,
1185
1186         .vidioc_querycap      = vidioc_querycap,
1187         .vidioc_enum_fmt_cap  = vidioc_enum_fmt_cap,
1188         .vidioc_g_fmt_cap     = vidioc_g_fmt_cap,
1189         .vidioc_try_fmt_cap   = vidioc_try_fmt_cap,
1190         .vidioc_s_fmt_cap     = vidioc_s_fmt_cap,
1191         .vidioc_reqbufs       = vidioc_reqbufs,
1192         .vidioc_querybuf      = vidioc_querybuf,
1193         .vidioc_qbuf          = vidioc_qbuf,
1194         .vidioc_dqbuf         = vidioc_dqbuf,
1195         .vidioc_s_std         = vidioc_s_std,
1196         .vidioc_enum_input    = vidioc_enum_input,
1197         .vidioc_g_input       = vidioc_g_input,
1198         .vidioc_s_input       = vidioc_s_input,
1199         .vidioc_queryctrl     = vidioc_queryctrl,
1200         .vidioc_g_ctrl        = vidioc_g_ctrl,
1201         .vidioc_s_ctrl        = vidioc_s_ctrl,
1202         .vidioc_streamon      = vidioc_streamon,
1203         .vidioc_streamoff     = vidioc_streamoff,
1204 #ifdef CONFIG_VIDEO_V4L1_COMPAT
1205         .vidiocgmbuf          = vidiocgmbuf,
1206 #endif
1207         .tvnorms              = V4L2_STD_525_60,
1208         .current_norm         = V4L2_STD_NTSC_M,
1209 };
1210 /* -----------------------------------------------------------------
1211         Initialization and module stuff
1212    ------------------------------------------------------------------*/
1213
1214 static int __init vivi_init(void)
1215 {
1216         int ret = -ENOMEM, i;
1217         struct vivi_dev *dev;
1218         struct video_device *vfd;
1219
1220         for (i = 0; i < n_devs; i++) {
1221                 dev = kzalloc(sizeof(*dev), GFP_KERNEL);
1222                 if (NULL == dev)
1223                         break;
1224
1225                 list_add_tail(&dev->vivi_devlist, &vivi_devlist);
1226
1227                 /* init video dma queues */
1228                 INIT_LIST_HEAD(&dev->vidq.active);
1229                 INIT_LIST_HEAD(&dev->vidq.queued);
1230                 init_waitqueue_head(&dev->vidq.wq);
1231
1232                 /* initialize locks */
1233                 mutex_init(&dev->lock);
1234                 spin_lock_init(&dev->slock);
1235
1236                 dev->vidq.timeout.function = vivi_vid_timeout;
1237                 dev->vidq.timeout.data     = (unsigned long)dev;
1238                 init_timer(&dev->vidq.timeout);
1239
1240                 vfd = video_device_alloc();
1241                 if (NULL == vfd)
1242                         break;
1243
1244                 *vfd = vivi_template;
1245
1246                 ret = video_register_device(vfd, VFL_TYPE_GRABBER, video_nr);
1247                 if (ret < 0)
1248                         break;
1249
1250                 snprintf(vfd->name, sizeof(vfd->name), "%s (%i)",
1251                          vivi_template.name, vfd->minor);
1252
1253                 if (video_nr >= 0)
1254                         video_nr++;
1255
1256                 dev->vfd = vfd;
1257         }
1258
1259         if (ret < 0) {
1260                 vivi_release();
1261                 printk(KERN_INFO "Error %d while loading vivi driver\n", ret);
1262         } else
1263                 printk(KERN_INFO "Video Technology Magazine Virtual Video "
1264                                  "Capture Board successfully loaded.\n");
1265         return ret;
1266 }
1267
1268 static void __exit vivi_exit(void)
1269 {
1270         vivi_release();
1271 }
1272
1273 module_init(vivi_init);
1274 module_exit(vivi_exit);
1275
1276 MODULE_DESCRIPTION("Video Technology Magazine Virtual Video Capture Board");
1277 MODULE_AUTHOR("Mauro Carvalho Chehab, Ted Walther and John Sokol");
1278 MODULE_LICENSE("Dual BSD/GPL");
1279
1280 module_param(video_nr, int, 0);
1281 MODULE_PARM_DESC(video_nr, "video iminor start number");
1282
1283 module_param(n_devs, int, 0);
1284 MODULE_PARM_DESC(n_devs, "number of video devices to create");
1285
1286 module_param_named(debug, vivi_template.debug, int, 0444);
1287 MODULE_PARM_DESC(debug, "activates debug info");
1288
1289 module_param(vid_limit, int, 0644);
1290 MODULE_PARM_DESC(vid_limit, "capture memory limit in megabytes");