vivi-lite：new simple vivi

        vivi驱动是学习v4l2的敲门砖，但内核从3.18开始使用vivid取代了vivi。vivid包含模块比较多，学习起来有难度。内核从4.4开始使用了新的vb2_v4l2_buffer取代了vb2_buffer。

        该驱动配合yavta应用，非常适合新手学习。mplane暂时未作支持，不过最新的yavta已经支持了mplane。

/*
 * Virtual Video driver - This code emulates a real video device with v4l2 api
 *
 * Copyright (c) 2006 by:
 *      Mauro Carvalho Chehab <mchehab--a.t--infradead.org>
 *      Ted Walther <ted--a.t--enumera.com>
 *      John Sokol <sokol--a.t--videotechnology.com>
 *      http://v4l.videotechnology.com/
 *
 *      Conversion to videobuf2 by Pawel Osciak & Marek Szyprowski
 *      Copyright (c) 2010 Samsung Electronics
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the BSD Licence, GNU General Public License
 * as published by the Free Software Foundation; either version 2 of the
 * License, or (at your option) any later version
 */
#include <linux/module.h>
#include <linux/errno.h>
#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/sched.h>
#include <linux/slab.h>
#include <linux/mutex.h>
#include <linux/videodev2.h>
#include <linux/kthread.h>
#include <linux/freezer.h>
#include <linux/ktime.h>

#include <media/videobuf2-vmalloc.h>
#include <media/videobuf2-dma-contig.h>

#include <media/v4l2-device.h>
#include <media/v4l2-ioctl.h>
#include <media/v4l2-ctrls.h>
#include <media/v4l2-fh.h>
#include <media/v4l2-event.h>
#include <media/v4l2-common.h>
#include <media/v4l2-subdev.h>

#include <linux/platform_device.h>

#include "font_8x16.c"


struct vivi_dummy_buffer {
    void *vaddr;
    dma_addr_t dma_addr;
    u32 size;
};


#define VIVI_MODULE_NAME "vivi"

/* Wake up at about 30 fps */
#define WAKE_NUMERATOR 30
#define WAKE_DENOMINATOR 1001
#define BUFFER_TIMEOUT     msecs_to_jiffies(500)  /* 0.5 seconds */

#define MAX_WIDTH 1920
#define MAX_HEIGHT 1200

#define VIVI_VERSION "0.8.1"

MODULE_DESCRIPTION("Video Technology Magazine Virtual Video Capture Board");
MODULE_AUTHOR("Mauro Carvalho Chehab, Ted Walther and John Sokol");
MODULE_LICENSE("Dual BSD/GPL");
MODULE_VERSION(VIVI_VERSION);

static unsigned video_nr = -1;
module_param(video_nr, uint, 0644);
MODULE_PARM_DESC(video_nr, "videoX start number, -1 is autodetect");

static unsigned n_devs = 1;
module_param(n_devs, uint, 0644);
MODULE_PARM_DESC(n_devs, "number of video devices to create");

static unsigned debug;
module_param(debug, uint, 0644);
MODULE_PARM_DESC(debug, "activates debug info");

static unsigned int vid_limit = 16;
module_param(vid_limit, uint, 0644);
MODULE_PARM_DESC(vid_limit, "capture memory limit in megabytes");

static unsigned memops = 0;
module_param(memops, uint, 0644);
MODULE_PARM_DESC(memops, "default vb2_dma_contig_memops");

/* Global font descriptor */
static const u8 *font8x16;

#define dprintk(dev, level, fmt, arg...) \
    v4l2_dbg(level, debug, &dev->v4l2_dev, fmt, ## arg)

/* ------------------------------------------------------------------
    Basic structures
   ------------------------------------------------------------------*/

struct vivi_fmt {
    char  *name;
    u32   fourcc;          /* v4l2 format id */
    int   depth;
};

static struct vivi_fmt formats[] = {
    {
        .name     = "4:2:2, packed, YUYV",
        .fourcc   = V4L2_PIX_FMT_YUYV,
        .depth    = 16,
    },
    {
        .name     = "4:2:2, packed, UYVY",
        .fourcc   = V4L2_PIX_FMT_UYVY,
        .depth    = 16,
    },
    {
        .name     = "4:2:2, packed, YVYU",
        .fourcc   = V4L2_PIX_FMT_YVYU,
        .depth    = 16,
    },
    {
        .name     = "4:2:2, packed, VYUY",
        .fourcc   = V4L2_PIX_FMT_VYUY,
        .depth    = 16,
    },
    {
        .name     = "RGB24 (LE)",
        .fourcc   = V4L2_PIX_FMT_RGB24, /* rgb */
        .depth    = 24,
    },
    {
        .name     = "RGB24 (BE)",
        .fourcc   = V4L2_PIX_FMT_BGR24, /* bgr */
        .depth    = 24,
    },
    {
        .name     = "RGB32 (LE)",
        .fourcc   = V4L2_PIX_FMT_RGB32, /* argb */
        .depth    = 32,
    },
    {
        .name     = "RGB32 (BE)",
        .fourcc   = V4L2_PIX_FMT_BGR32, /* bgra */
        .depth    = 32,
    },
};

static struct vivi_fmt *get_format(struct v4l2_format *f)
{
    struct vivi_fmt *fmt;
    unsigned int k;

    for (k = 0; k < ARRAY_SIZE(formats); k++) {
        fmt = &formats[k];
        if (fmt->fourcc == f->fmt.pix.pixelformat)
            break;
    }

    if (k == ARRAY_SIZE(formats))
        return NULL;

    return &formats[k];
}

/* buffer for one video frame */
struct vivi_buffer {
    /* common v4l buffer stuff -- must be first */
    struct vb2_v4l2_buffer  vb;
    struct list_head    list;
    struct vivi_fmt        *fmt;
};

static inline struct vivi_buffer *to_vivi_buffer(struct vb2_v4l2_buffer *vb)
{
    return container_of(vb, struct vivi_buffer, vb);
}

struct vivi_dmaqueue {
    struct list_head       active;

    /* thread for generating video stream*/
    struct task_struct         *kthread;
    wait_queue_head_t          wq;
    /* Counters to control fps rate */
    int                        frame;
    int                        ini_jiffies;
};

static LIST_HEAD(vivi_devlist);

struct vivi_dev {
    struct list_head           vivi_devlist;
    struct v4l2_device     v4l2_dev;
    struct video_device    vdev;

    spinlock_t                 slock;
    struct mutex           mutex;

    struct vivi_dmaqueue       vidq;

    /* Several counters */
    unsigned           ms;
    unsigned long              jiffies;

    int            mv_count;    /* Controls bars movement */

    /* Input Number */
    int            input;

    /* video capture */
    struct vivi_fmt            *fmt;
    unsigned int               width, height;
    struct vb2_queue       vb_vidq;
    unsigned int           field_count;

    u8             bars[9][3];
    u8             line[MAX_WIDTH * 8];
    unsigned int           pixelsize;

    struct vivi_dummy_buffer dummy_buf;
};

#if 0
static int vivi_create_dummy_buf(struct vivi_dev *dev, int size)
{
    struct vivi_dummy_buffer *dummy_buf = &dev->dummy_buf;

    dummy_buf->size = size;
    dummy_buf->vaddr = dma_alloc_coherent(dev->v4l2_dev.dev, dummy_buf->size, &dummy_buf->dma_addr, GFP_KERNEL);
    if (!dummy_buf->vaddr) {
        v4l2_err(&dev->v4l2_dev, "Failed to allocate the memory for dummy buffer\n");
        return -ENOMEM;
    }
    return 0;
}
static void vivi_destroy_dummy_buf(struct vivi_dev *dev)
{
    struct vivi_dummy_buffer *dummy_buf = &dev->dummy_buf;
    if (dummy_buf->vaddr) {
        dma_free_coherent(dev->v4l2_dev.dev, dummy_buf->size, dummy_buf->vaddr, dummy_buf->dma_addr);
    }
}
#endif


/* ------------------------------------------------------------------
    DMA and thread functions
   ------------------------------------------------------------------*/

/* Bars and Colors should match positions */

enum colors {
    WHITE,
    AMBER,
    CYAN,
    GREEN,
    MAGENTA,
    RED,
    BLUE,
    BLACK,
    TEXT_BLACK,
};

/* R   G   B */
#define COLOR_WHITE {204, 204, 204}
#define COLOR_AMBER {208, 208,   0}
#define COLOR_CYAN  {  0, 206, 206}
#define COLOR_GREEN {  0, 239,   0}
#define COLOR_MAGENTA   {239,   0, 239}
#define COLOR_RED   {205,   0,   0}
#define COLOR_BLUE  {  0,   0, 255}
#define COLOR_BLACK {  0,   0,   0}

struct bar_std {
    u8 bar[9][3];
};

/* Maximum number of bars are 10 - otherwise, the input print code
   should be modified */
static struct bar_std bars[] = {
    {   /* Standard ITU-R color bar sequence */
        { COLOR_WHITE, COLOR_AMBER, COLOR_CYAN, COLOR_GREEN,
          COLOR_MAGENTA, COLOR_RED, COLOR_BLUE, COLOR_BLACK, COLOR_BLACK }
    }, {
        { COLOR_WHITE, COLOR_AMBER, COLOR_BLACK, COLOR_WHITE,
          COLOR_AMBER, COLOR_BLACK, COLOR_WHITE, COLOR_AMBER, COLOR_BLACK }
    }, {
        { COLOR_WHITE, COLOR_CYAN, COLOR_BLACK, COLOR_WHITE,
          COLOR_CYAN, COLOR_BLACK, COLOR_WHITE, COLOR_CYAN, COLOR_BLACK }
    }, {
        { COLOR_WHITE, COLOR_GREEN, COLOR_BLACK, COLOR_WHITE,
          COLOR_GREEN, COLOR_BLACK, COLOR_WHITE, COLOR_GREEN, COLOR_BLACK }
    },
};

#define NUM_INPUTS ARRAY_SIZE(bars)

#define TO_Y(r, g, b) \
    (((16829 * r + 33039 * g + 6416 * b  + 32768) >> 16) + 16)
/* RGB to  V(Cr) Color transform */
#define TO_V(r, g, b) \
    (((28784 * r - 24103 * g - 4681 * b  + 32768) >> 16) + 128)
/* RGB to  U(Cb) Color transform */
#define TO_U(r, g, b) \
    (((-9714 * r - 19070 * g + 28784 * b + 32768) >> 16) + 128)

/* precalculate color bar values to speed up rendering */
static void precalculate_bars(struct vivi_dev *dev)
{
    u8 r, g, b;
    int k, is_yuv;

    for (k = 0; k < 9; k++) {
        r = bars[dev->input].bar[k][0];
        g = bars[dev->input].bar[k][1];
        b = bars[dev->input].bar[k][2];
        is_yuv = 0;

        switch (dev->fmt->fourcc) {
        case V4L2_PIX_FMT_YUYV:
        case V4L2_PIX_FMT_UYVY:
        case V4L2_PIX_FMT_YVYU:
        case V4L2_PIX_FMT_VYUY:
            is_yuv = 1;
            break;
        case V4L2_PIX_FMT_RGB24:
        case V4L2_PIX_FMT_BGR24:
        case V4L2_PIX_FMT_RGB32:
        case V4L2_PIX_FMT_BGR32:
            break;
        }

        if (is_yuv) {
            dev->bars[k][0] = TO_Y(r, g, b);    /* Luma */
            dev->bars[k][1] = TO_U(r, g, b);    /* Cb */
            dev->bars[k][2] = TO_V(r, g, b);    /* Cr */
        } else {
            dev->bars[k][0] = r;
            dev->bars[k][1] = g;
            dev->bars[k][2] = b;
        }
    }
}

#define TSTAMP_MIN_Y    24
#define TSTAMP_MAX_Y    (TSTAMP_MIN_Y + 15)
#define TSTAMP_INPUT_X  10
#define TSTAMP_MIN_X    (54 + TSTAMP_INPUT_X)

/* 'odd' is true for pixels 1, 3, 5, etc. and false for pixels 0, 2, 4, etc. */
static void gen_twopix(struct vivi_dev *dev, u8 *buf, int colorpos, bool odd)
{
    u8 r_y, g_u, b_v;
    u8 alpha = 0;
    int color;
    u8 *p;

    r_y = dev->bars[colorpos][0]; /* R or precalculated Y */
    g_u = dev->bars[colorpos][1]; /* G or precalculated U */
    b_v = dev->bars[colorpos][2]; /* B or precalculated V */

    for (color = 0; color < dev->pixelsize; color++) {
        p = buf + color;

        switch (dev->fmt->fourcc) {
        case V4L2_PIX_FMT_YUYV:
            switch (color) {
            case 0:
                *p = r_y;
                break;
            case 1:
                *p = odd ? b_v : g_u;
                break;
            }
            break;
        case V4L2_PIX_FMT_UYVY:
            switch (color) {
            case 0:
                *p = odd ? b_v : g_u;
                break;
            case 1:
                *p = r_y;
                break;
            }
            break;
        case V4L2_PIX_FMT_YVYU:
            switch (color) {
            case 0:
                *p = r_y;
                break;
            case 1:
                *p = odd ? g_u : b_v;
                break;
            }
            break;
        case V4L2_PIX_FMT_VYUY:
            switch (color) {
            case 0:
                *p = odd ? g_u : b_v;
                break;
            case 1:
                *p = r_y;
                break;
            }
            break;
        case V4L2_PIX_FMT_RGB24:
            switch (color) {
            case 0:
                *p = r_y;
                break;
            case 1:
                *p = g_u;
                break;
            case 2:
                *p = b_v;
                break;
            }
            break;
        case V4L2_PIX_FMT_BGR24:
            switch (color) {
            case 0:
                *p = b_v;
                break;
            case 1:
                *p = g_u;
                break;
            case 2:
                *p = r_y;
                break;
            }
            break;
        case V4L2_PIX_FMT_RGB32:
            switch (color) {
            case 0:
                *p = alpha;
                break;
            case 1:
                *p = r_y;
                break;
            case 2:
                *p = g_u;
                break;
            case 3:
                *p = b_v;
                break;
            }
            break;
        case V4L2_PIX_FMT_BGR32:
            switch (color) {
            case 0:
                *p = b_v;
                break;
            case 1:
                *p = g_u;
                break;
            case 2:
                *p = r_y;
                break;
            case 3:
                *p = alpha;
                break;
            }
            break;
        }
    }
}

static void precalculate_line(struct vivi_dev *dev)
{
    int w;

    for (w = 0; w < dev->width * 2; w++) {
        int colorpos = w / (dev->width / 8) % 8;

        gen_twopix(dev, dev->line + w * dev->pixelsize, colorpos, w & 1);
    }
}

static void gen_text(struct vivi_dev *dev, char *basep,
                    int y, int x, char *text)
{
    int line;

    /* Checks if it is possible to show string */
    if (y + 16 >= dev->height || x + strlen(text) * 8 >= dev->width)
        return;

    /* Print stream time */
    for (line = y; line < y + 16; line++) {
        int j = 0;
        char *pos = basep + line * dev->width * dev->pixelsize + x * dev->pixelsize;
        char *s;

        for (s = text; *s; s++) {
            u8 chr = font8x16[*s * 16 + line - y];
            int i;

            for (i = 0; i < 7; i++, j++) {
                /* Draw white font on black background */
                if (chr & (1 << (7 - i)))
                    gen_twopix(dev, pos + j * dev->pixelsize, WHITE, (x+y) & 1);
                else
                    gen_twopix(dev, pos + j * dev->pixelsize, TEXT_BLACK, (x+y) & 1);
            }
        }
    }
}

static void vivi_fillbuff(struct vivi_dev *dev, struct vivi_buffer *buf)
{
    int wmax = dev->width;
    int hmax = dev->height;
    void *vbuf = vb2_plane_vaddr(&buf->vb.vb2_buf, 0);
    unsigned ms;
    char str[100];
    int h, line = 1;

    if (!vbuf)
        return;

#if 0
    /* move to buf_prepare, only fill text */
    for (h = 0; h < hmax; h++)
        memcpy(vbuf + h * wmax * dev->pixelsize,
               dev->line + (dev->mv_count % wmax) * dev->pixelsize,
               wmax * dev->pixelsize);
#endif

    /* Updates stream time */

    dev->ms += jiffies_to_msecs(jiffies - dev->jiffies);
    dev->jiffies = jiffies;
    ms = dev->ms;

    snprintf(str, sizeof(str), " %dx%d, input %d ",
            dev->width, dev->height, dev->input);
    gen_text(dev, vbuf, line++ * 16, 16, str);

    snprintf(str, sizeof(str), " %02d:%02d:%02d:%03d ",
            (ms / (60 * 60 * 1000)) % 24,
            (ms / (60 * 1000)) % 60,
            (ms / 1000) % 60,
            ms % 1000);
    gen_text(dev, vbuf, line++ * 16, 16, str);

    snprintf(str, sizeof(str), " sequence %05d ", dev->field_count);
    gen_text(dev, vbuf, line++ * 16, 16, str);

    dev->mv_count += 2;
    dev->field_count++;
    buf->vb.sequence = dev->field_count;
    buf->vb.vb2_buf.timestamp = ktime_get_ns();
}

static void vivi_thread_tick(struct vivi_dev *dev)
{
    struct vivi_dmaqueue *dma_q = &dev->vidq;
    struct vivi_buffer *buf;
    unsigned long flags = 0;

    dprintk(dev, 1, "Thread tick\n");

    spin_lock_irqsave(&dev->slock, flags);
    if (list_empty(&dma_q->active)) {
        dprintk(dev, 1, "No active queue to serve\n");
        spin_unlock_irqrestore(&dev->slock, flags);
        return;
    }

    buf = list_entry(dma_q->active.next, struct vivi_buffer, list);
    list_del(&buf->list);
    spin_unlock_irqrestore(&dev->slock, flags);

    buf->vb.vb2_buf.timestamp = ktime_get_ns();

    /* Fill buffer */
    vivi_fillbuff(dev, buf);
    dprintk(dev, 1, "filled buffer %p\n", buf);

    vb2_buffer_done(&buf->vb.vb2_buf, VB2_BUF_STATE_DONE);
    dprintk(dev, 2, "[%p/%d] done\n", buf, buf->vb.vb2_buf.index);
}

#define frames_to_ms(frames)                    \
    ((frames * WAKE_NUMERATOR * 1000) / WAKE_DENOMINATOR)

static void vivi_sleep(struct vivi_dev *dev)
{
    struct vivi_dmaqueue *dma_q = &dev->vidq;
    int timeout;
    DECLARE_WAITQUEUE(wait, current);

    dprintk(dev, 1, "%s dma_q=0x%08lx\n", __func__,
        (unsigned long)dma_q);

    add_wait_queue(&dma_q->wq, &wait);
    if (kthread_should_stop())
        goto stop_task;

    /* Calculate time to wake up */
    timeout = msecs_to_jiffies(frames_to_ms(1));

    vivi_thread_tick(dev);

    schedule_timeout_interruptible(timeout);

stop_task:
    remove_wait_queue(&dma_q->wq, &wait);
    try_to_freeze();
}

static int vivi_thread(void *data)
{
    struct vivi_dev *dev = data;

    dprintk(dev, 1, "thread started\n");

    set_freezable();

    for (;;) {
        vivi_sleep(dev);

        if (kthread_should_stop())
            break;
    }
    dprintk(dev, 1, "thread: exit\n");
    return 0;
}

static int vivi_start_generating(struct vivi_dev *dev)
{
    struct vivi_dmaqueue *dma_q = &dev->vidq;

    dprintk(dev, 1, "%s\n", __func__);

    /* Resets frame counters */
    dev->ms = 0;
    dev->mv_count = 0;
    dev->jiffies = jiffies;

    dma_q->frame = 0;
    dma_q->ini_jiffies = jiffies;
    dma_q->kthread = kthread_run(vivi_thread, dev, dev->v4l2_dev.name);

    if (IS_ERR(dma_q->kthread)) {
        v4l2_err(&dev->v4l2_dev, "kernel_thread() failed\n");
        return PTR_ERR(dma_q->kthread);
    }
    /* Wakes thread */
    wake_up_interruptible(&dma_q->wq);

    dprintk(dev, 1, "returning from %s\n", __func__);
    return 0;
}

static void vivi_stop_generating(struct vivi_dev *dev)
{
    struct vivi_dmaqueue *dma_q = &dev->vidq;

    dprintk(dev, 1, "%s\n", __func__);

    /* shutdown control thread */
    if (dma_q->kthread) {
        kthread_stop(dma_q->kthread);
        dma_q->kthread = NULL;
    }

    /*
     * Typical driver might need to wait here until dma engine stops.
     * In this case we can abort imiedetly, so it's just a noop.
     */

    /* Release all active buffers */
    while (!list_empty(&dma_q->active)) {
        struct vivi_buffer *buf;
        buf = list_entry(dma_q->active.next, struct vivi_buffer, list);
        list_del(&buf->list);
        vb2_buffer_done(&buf->vb.vb2_buf, VB2_BUF_STATE_ERROR);
        dprintk(dev, 2, "[%p/%d] done\n", buf, buf->vb.vb2_buf.index);
    }
}
/* ------------------------------------------------------------------
    Videobuf operations
   ------------------------------------------------------------------*/
static int queue_setup(struct vb2_queue *vq,
                unsigned int *nbuffers, unsigned int *nplanes,
                unsigned int sizes[], struct device  *alloc_ctxs[])
{
    struct vivi_dev *dev = vb2_get_drv_priv(vq);
    unsigned long size;

    size = dev->width * dev->height * dev->pixelsize;

    if (size == 0)
        return -EINVAL;

    if (0 == *nbuffers)
        *nbuffers = 32;

    while (size * *nbuffers > vid_limit * 1024 * 1024)
        (*nbuffers)--;

    *nplanes = 1;

    sizes[0] = size;

    /*
     * videobuf2-vmalloc allocator is context-less so no need to set
     * alloc_ctxs array.
     */

    dprintk(dev, 1, "%s, count=%d, size=%ld\n", __func__,
        *nbuffers, size);

    return 0;
}

static int buffer_prepare(struct vb2_buffer *vb)
{
    struct vivi_dev *dev = vb2_get_drv_priv(vb->vb2_queue);
    struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
    struct vivi_buffer *buf = to_vivi_buffer(vbuf);
    unsigned long size;

    BUG_ON(NULL == dev->fmt);

    /*
     * Theses properties only change when queue is idle, see s_fmt.
     * The below checks should not be performed here, on each
     * buffer_prepare (i.e. on each qbuf). Most of the code in this function
     * should thus be moved to buffer_init and s_fmt.
     */
    if (dev->width  < 48 || dev->width  > MAX_WIDTH ||
        dev->height < 32 || dev->height > MAX_HEIGHT)
        return -EINVAL;

    size = dev->width * dev->height * dev->pixelsize;
    if (vb2_plane_size(vb, 0) < size) {
        dprintk(dev, 1, "%s data will not fit into plane (%lu < %lu)\n",
                __func__, vb2_plane_size(vb, 0), size);
        return -EINVAL;
    }

    vb2_set_plane_payload(&buf->vb.vb2_buf, 0, size);

    buf->fmt = dev->fmt;

    precalculate_bars(dev);
    precalculate_line(dev);

    {
        /* draw colorbar only once */
        int h;
        void *vaddr = vb2_plane_vaddr(&buf->vb.vb2_buf, 0);

        if (!vaddr)
            return 0;

        for (h = 0; h < dev->height; h++)
            memcpy(vaddr + h * dev->width * dev->pixelsize,
                   dev->line, dev->width * dev->pixelsize);
    }

    return 0;
}

static void buffer_queue(struct vb2_buffer *vb)
{
    struct vivi_dev *dev = vb2_get_drv_priv(vb->vb2_queue);
    struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
    struct vivi_buffer *buf = to_vivi_buffer(vbuf);
    struct vivi_dmaqueue *vidq = &dev->vidq;
    unsigned long flags = 0;

    dprintk(dev, 1, "%s\n", __func__);

    spin_lock_irqsave(&dev->slock, flags);
    list_add_tail(&buf->list, &vidq->active);
    spin_unlock_irqrestore(&dev->slock, flags);
}

static int start_streaming(struct vb2_queue *vq, unsigned int count)
{
    struct vivi_dev *dev = vb2_get_drv_priv(vq);
    dprintk(dev, 1, "%s\n", __func__);
    return vivi_start_generating(dev);
}

/* abort streaming and wait for last buffer */
static void stop_streaming(struct vb2_queue *vq)
{
    struct vivi_dev *dev = vb2_get_drv_priv(vq);
    dprintk(dev, 1, "%s\n", __func__);
    vivi_stop_generating(dev);
}

#if 0
static void vivi_lock(struct vb2_queue *vq)
{
    struct vivi_dev *dev = vb2_get_drv_priv(vq);
    mutex_lock(&dev->mutex);
}

static void vivi_unlock(struct vb2_queue *vq)
{
    struct vivi_dev *dev = vb2_get_drv_priv(vq);
    mutex_unlock(&dev->mutex);
}
#endif

static struct vb2_ops vivi_video_qops = {
    .queue_setup        = queue_setup,
    .buf_prepare        = buffer_prepare,
    .buf_queue          = buffer_queue,

    .start_streaming    = start_streaming,
    .stop_streaming     = stop_streaming,

#if 0
    .wait_prepare       = vivi_unlock,
    .wait_finish        = vivi_lock,
#else
    .wait_prepare = vb2_ops_wait_prepare,
    .wait_finish = vb2_ops_wait_finish,
#endif
};

/* ------------------------------------------------------------------
    IOCTL vidioc handling
   ------------------------------------------------------------------*/
static int vidioc_querycap(struct file *file, void  *priv,
                    struct v4l2_capability *cap)
{
    struct vivi_dev *dev = video_drvdata(file);

    strcpy(cap->driver, "vivi");
    strcpy(cap->card, "vivi");
    strlcpy(cap->bus_info, dev->v4l2_dev.name, sizeof(cap->bus_info));
#if 0
    cap->device_caps = V4L2_CAP_VIDEO_CAPTURE | V4L2_CAP_STREAMING |
                V4L2_CAP_READWRITE;
    cap->capabilities = cap->device_caps | V4L2_CAP_DEVICE_CAPS;
#endif
    return 0;
}


#ifdef MPLANES

static int vivi_try_fmt_vid_cap_mplane(struct file *file, void *priv, struct v4l2_format *f)
{
    struct rkisp1_stream *stream = video_drvdata(file);

    return rkisp1_set_fmt(stream, &f->fmt.pix_mp, true);
}

static int vivi_enum_fmt_vid_cap_mplane(struct file *file, void *priv, struct v4l2_fmtdesc *f)
{
    struct rkisp1_stream *stream = video_drvdata(file);
    const struct capture_fmt *fmt = NULL;

    if (f->index >= stream->config->fmt_size)
        return -EINVAL;

    fmt = &stream->config->fmts[f->index];
    f->pixelformat = fmt->fourcc;

    return 0;
}

static int vivi_s_fmt_vid_cap_mplane(struct file *file, void *priv, struct v4l2_format *f)
{
    return 0;
}

static int vivi_g_fmt_vid_cap_mplane(struct file *file, void *priv, struct v4l2_format *f)
{
    struct rkisp1_stream *stream = video_drvdata(file);

    f->fmt.pix_mp = stream->out_fmt;

    return 0;
}

#else

static int vidioc_enum_fmt_vid_cap(struct file *file, void  *priv,
                    struct v4l2_fmtdesc *f)
{
    struct vivi_fmt *fmt;

    if (f->index >= ARRAY_SIZE(formats))
        return -EINVAL;

    fmt = &formats[f->index];
    f->pixelformat = fmt->fourcc;
//    strlcpy(f->description, fmt->name, sizeof(f->description));

    return 0;
}

static int vidioc_g_fmt_vid_cap(struct file *file, void *priv,
                    struct v4l2_format *f)
{
    struct vivi_dev *dev = video_drvdata(file);

    f->fmt.pix.width        = dev->width;
    f->fmt.pix.height       = dev->height;
    f->fmt.pix.field        = V4L2_FIELD_INTERLACED;
    f->fmt.pix.pixelformat  = dev->fmt->fourcc;
    f->fmt.pix.bytesperline =
        (f->fmt.pix.width * dev->fmt->depth) >> 3;
    f->fmt.pix.sizeimage =
        f->fmt.pix.height * f->fmt.pix.bytesperline;
    if (dev->fmt->fourcc == V4L2_PIX_FMT_YUYV ||
        dev->fmt->fourcc == V4L2_PIX_FMT_UYVY)
        f->fmt.pix.colorspace = V4L2_COLORSPACE_SMPTE170M;
    else
        f->fmt.pix.colorspace = V4L2_COLORSPACE_SRGB;
    return 0;
}

static int vidioc_try_fmt_vid_cap(struct file *file, void *priv,
            struct v4l2_format *f)
{
    struct vivi_dev *dev = video_drvdata(file);
    struct vivi_fmt *fmt;

    fmt = get_format(f);
    if (!fmt) {
        dprintk(dev, 1, "Fourcc format (0x%08x) unknown.\n",
            f->fmt.pix.pixelformat);
        f->fmt.pix.pixelformat = V4L2_PIX_FMT_YUYV;
        fmt = get_format(f);
    }

    f->fmt.pix.field = V4L2_FIELD_INTERLACED;
    v4l_bound_align_image(&f->fmt.pix.width, 48, MAX_WIDTH, 2,
                  &f->fmt.pix.height, 32, MAX_HEIGHT, 0, 0);
    f->fmt.pix.bytesperline =
        (f->fmt.pix.width * fmt->depth) >> 3;
    f->fmt.pix.sizeimage =
        f->fmt.pix.height * f->fmt.pix.bytesperline;
    if (fmt->fourcc == V4L2_PIX_FMT_YUYV ||
        fmt->fourcc == V4L2_PIX_FMT_UYVY)
        f->fmt.pix.colorspace = V4L2_COLORSPACE_SMPTE170M;
    else
        f->fmt.pix.colorspace = V4L2_COLORSPACE_SRGB;
    return 0;
}

static int vidioc_s_fmt_vid_cap(struct file *file, void *priv,
                    struct v4l2_format *f)
{
    struct vivi_dev *dev = video_drvdata(file);
    struct vb2_queue *q = &dev->vb_vidq;

    int ret = vidioc_try_fmt_vid_cap(file, priv, f);
    if (ret < 0)
        return ret;

    if (vb2_is_busy(q)) {
        dprintk(dev, 1, "%s device busy\n", __func__);
        return -EBUSY;
    }

    dev->fmt = get_format(f);
    dev->pixelsize = dev->fmt->depth / 8;
    dev->width = f->fmt.pix.width;
    dev->height = f->fmt.pix.height;

    return 0;
}
#endif

/* only one input in this sample driver */
static int vidioc_enum_input(struct file *file, void *priv,
                struct v4l2_input *inp)
{
    if (inp->index >= NUM_INPUTS)
        return -EINVAL;

    inp->type = V4L2_INPUT_TYPE_CAMERA;
    sprintf(inp->name, "Camera %u", inp->index);
    return 0;
}

static int vidioc_g_input(struct file *file, void *priv, unsigned int *i)
{
    struct vivi_dev *dev = video_drvdata(file);

    *i = dev->input;
    return 0;
}

static int vidioc_s_input(struct file *file, void *priv, unsigned int i)
{
    struct vivi_dev *dev = video_drvdata(file);

    if (i >= NUM_INPUTS)
        return -EINVAL;

    if (i == dev->input)
        return 0;

    dev->input = i;
    precalculate_bars(dev);
    precalculate_line(dev);
    return 0;
}

/* ------------------------------------------------------------------
    File operations for the device
   ------------------------------------------------------------------*/

static const struct v4l2_file_operations vivi_fops = {
    .owner          = THIS_MODULE,
    .open           = v4l2_fh_open,
    .release        = vb2_fop_release,
    .read           = vb2_fop_read,
    .poll           = vb2_fop_poll,
    .unlocked_ioctl = video_ioctl2, /* V4L2 ioctl handler */
    .mmap           = vb2_fop_mmap,
};

static const struct v4l2_ioctl_ops vivi_ioctl_ops = {
    .vidioc_querycap      = vidioc_querycap,

#ifdef MPLANES
    .vidioc_try_fmt_vid_cap_mplane  = vivi_try_fmt_vid_cap_mplane,
    .vidioc_enum_fmt_vid_cap_mplane = vivi_enum_fmt_vid_cap_mplane,
    .vidioc_s_fmt_vid_cap_mplane    = vivi_s_fmt_vid_cap_mplane,
    .vidioc_g_fmt_vid_cap_mplane    = vivi_g_fmt_vid_cap_mplane,

#else
    .vidioc_enum_fmt_vid_cap  = vidioc_enum_fmt_vid_cap,
    .vidioc_g_fmt_vid_cap     = vidioc_g_fmt_vid_cap,
    .vidioc_try_fmt_vid_cap   = vidioc_try_fmt_vid_cap,
    .vidioc_s_fmt_vid_cap     = vidioc_s_fmt_vid_cap,
#endif

    .vidioc_reqbufs       = vb2_ioctl_reqbufs,
    .vidioc_create_bufs   = vb2_ioctl_create_bufs,
    .vidioc_prepare_buf   = vb2_ioctl_prepare_buf,
    .vidioc_querybuf      = vb2_ioctl_querybuf,
    .vidioc_qbuf          = vb2_ioctl_qbuf,
    .vidioc_dqbuf         = vb2_ioctl_dqbuf,
    .vidioc_streamon      = vb2_ioctl_streamon,
    .vidioc_streamoff     = vb2_ioctl_streamoff,
    .vidioc_log_status    = v4l2_ctrl_log_status,
    .vidioc_subscribe_event = v4l2_ctrl_subscribe_event,
    .vidioc_unsubscribe_event = v4l2_event_unsubscribe,

    .vidioc_enum_input    = vidioc_enum_input,
    .vidioc_g_input       = vidioc_g_input,
    .vidioc_s_input       = vidioc_s_input,
};

static struct video_device vivi_template = {
    .name       = "vivi-capture",
    .fops       = &vivi_fops,
    .ioctl_ops  = &vivi_ioctl_ops,
    .release    = video_device_release_empty,
};

/* -----------------------------------------------------------------
    Initialization and module stuff
   ------------------------------------------------------------------*/
static int vivi_release(struct platform_device *pdev)
{
    struct vivi_dev *dev;
    struct list_head *list;

    while (!list_empty(&vivi_devlist)) {

        list = vivi_devlist.next;
        list_del(list);
        dev = list_entry(list, struct vivi_dev, vivi_devlist);

        v4l2_info(&dev->v4l2_dev, "unregistering %s\n",
            video_device_node_name(&dev->vdev));
        video_unregister_device(&dev->vdev);
        v4l2_device_unregister(&dev->v4l2_dev);
        kfree(dev);
    }

    return 0;
}

static int vivi_create_instance(int inst, struct device *device)
{
    struct vivi_dev *dev;
    struct video_device *vfd;
    struct vb2_queue *q;
    int ret;

    dev = kzalloc(sizeof(*dev), GFP_KERNEL);
    if (!dev)
        return -ENOMEM;

    snprintf(dev->v4l2_dev.name, sizeof(dev->v4l2_dev.name),
            "%s-%03d", VIVI_MODULE_NAME, inst);
    ret = v4l2_device_register(device, &dev->v4l2_dev);
    if (ret)
        goto free_dev;

    dev->fmt = &formats[0];
    dev->width = 640;
    dev->height = 480;
    dev->pixelsize = dev->fmt->depth / 8;

    /* initialize locks */
    spin_lock_init(&dev->slock);

    /* initialize queue */
    q = &dev->vb_vidq;
    memset(q, 0, sizeof(dev->vb_vidq));
#ifdef MPLANES
    q->type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE;
#else
    q->type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
#endif
    q->io_modes = VB2_MMAP | VB2_USERPTR | VB2_DMABUF;
    q->drv_priv = dev;
    q->buf_struct_size = sizeof(struct vivi_buffer);
    q->ops = &vivi_video_qops;

    if (memops)
        q->mem_ops = &vb2_vmalloc_memops;
    else {
        q->mem_ops = &vb2_dma_contig_memops;
        q->dev = device;
    }

#if 1
    q->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_MONOTONIC;
    q->lock = &dev->mutex;
#endif

    vb2_queue_init(q);

    mutex_init(&dev->mutex);

    /* init video dma queues */
    INIT_LIST_HEAD(&dev->vidq.active);
    init_waitqueue_head(&dev->vidq.wq);

    vfd = &dev->vdev;
    *vfd = vivi_template;
    vfd->v4l2_dev = &dev->v4l2_dev;
    vfd->queue = q;
#if 0
    vfd->debug = debug;
    set_bit(V4L2_FL_USE_FH_PRIO, &vfd->flags);
#endif

    vfd->vfl_dir = VFL_DIR_RX;
#ifdef MPLANES
    vfd->device_caps = V4L2_CAP_VIDEO_CAPTURE_MPLANE | V4L2_CAP_STREAMING;
#else
    vfd->device_caps = V4L2_CAP_VIDEO_CAPTURE | V4L2_CAP_STREAMING;
#endif
    /*
     * Provide a mutex to v4l2 core. It will be used to protect
     * all fops and v4l2 ioctls.
     */
    vfd->lock = &dev->mutex;
    video_set_drvdata(vfd, dev);

    ret = video_register_device(vfd, VFL_TYPE_GRABBER, video_nr);
    if (ret < 0)
        goto unreg_dev;

    /* Now that everything is fine, let's add it to device list */
    list_add_tail(&dev->vivi_devlist, &vivi_devlist);

    v4l2_info(&dev->v4l2_dev, "V4L2 device registered as %s\n",
          video_device_node_name(vfd));
    return 0;

unreg_dev:
    v4l2_device_unregister(&dev->v4l2_dev);
free_dev:
    kfree(dev);
    return ret;
}

/* This routine allocates from 1 to n_devs virtual drivers.

   The real maximum number of virtual drivers will depend on how many drivers
   will succeed. This is limited to the maximum number of devices that
   videodev supports, which is equal to VIDEO_NUM_DEVICES.
 */
static int vivi_probe(struct platform_device *pdev)
{
    int ret = 0, i;

    font8x16 = fontdata_8x16;

    if (n_devs <= 0)
        n_devs = 1;

    for (i = 0; i < n_devs; i++) {
        ret = vivi_create_instance(i, &pdev->dev);
        if (ret) {
            /* If some instantiations succeeded, keep driver */
            if (i)
                ret = 0;
            break;
        }
    }

    if (ret < 0) {
        printk(KERN_ERR "vivi: error %d while loading driver\n", ret);
        return ret;
    }

    printk(KERN_INFO "Video Technology Magazine Virtual Video "
            "Capture Board ver %s successfully loaded.\n",
            VIVI_VERSION);

    /* n_devs will reflect the actual number of allocated devices */
    n_devs = i;

    return ret;
}


static u64 vivi_dma_mask = ~(u64)0;

static void device_release(struct device *dev) { /* dismiss warn */ }

struct platform_device vivi_device = {
   .name = "vivi-mod",
   .id   = -1,
   .dev  = {
       .dma_mask = &vivi_dma_mask,
       .coherent_dma_mask = DMA_BIT_MASK(32),
       .release = device_release,
   },
};

static struct platform_driver vivi_driver = {
   .driver = {
       .name = "vivi-mod",
   },
   .probe    = vivi_probe,
   .remove   = vivi_release,
};


static int vivi_init(void)
{
    int ret = 0;
    ret = platform_device_register(&vivi_device);
    ret = platform_driver_register(&vivi_driver);
    return ret;
}

static void vivi_exit(void)
{
    platform_driver_unregister(&vivi_driver);
    platform_device_unregister(&vivi_device);
}

module_init(vivi_init);
module_exit(vivi_exit);