/*
 * Copyright (C) 2010 Texas Instruments Incorporated - http://www.ti.com/
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation version 2.
 *
 * This program is distributed WITHOUT ANY WARRANTY of any
 * kind, whether express or implied; without even the implied warranty
 * of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 */
#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/module.h>
#include <linux/errno.h>
#include <linux/interrupt.h>
#include <linux/string.h>
#include <linux/wait.h>
#include <linux/time.h>
#include <linux/platform_device.h>
#include <linux/irq.h>
#include <linux/mm.h>
#include <linux/mutex.h>
#include <linux/videodev2.h>
#include <linux/slab.h>

#include <asm/pgtable.h>
#include <mach/cputype.h>

#include <media/v4l2-dev.h>
#include <media/v4l2-common.h>
#include <media/v4l2-ioctl.h>
#include <media/v4l2-device.h>
#include <media/davinci/vpbe_display.h>
#include <media/davinci/vpbe_types.h>
#include <media/davinci/vpbe.h>
#include <media/davinci/vpbe_venc.h>
#include <media/davinci/vpbe_osd.h>
#include "vpbe_venc_regs.h"

#define VPBE_DISPLAY_DRIVER "vpbe-v4l2"

static int debug;

#define VPBE_DEFAULT_NUM_BUFS 3

module_param(debug, int, 0644);

static int vpbe_set_osd_display_params(struct vpbe_display *disp_dev,
                        struct vpbe_layer *layer);

static int venc_is_second_field(struct vpbe_display *disp_dev)
{
        struct vpbe_device *vpbe_dev = disp_dev->vpbe_dev;
        int ret;
        int val;

        ret = v4l2_subdev_call(vpbe_dev->venc,
                               core,
                               ioctl,
                               VENC_GET_FLD,
                               &val);
        if (ret < 0) {
                v4l2_err(&vpbe_dev->v4l2_dev,
                         "Error in getting Field ID 0\n");
        }
        return val;
}

static void vpbe_isr_even_field(struct vpbe_display *disp_obj,
                                struct vpbe_layer *layer)
{
        struct timespec timevalue;

        if (layer->cur_frm == layer->next_frm)
                return;
        ktime_get_ts(&timevalue);
        layer->cur_frm->vb.v4l2_buf.timestamp.tv_sec =
                timevalue.tv_sec;
        layer->cur_frm->vb.v4l2_buf.timestamp.tv_usec =
                timevalue.tv_nsec / NSEC_PER_USEC;
        vb2_buffer_done(&layer->cur_frm->vb, VB2_BUF_STATE_DONE);
        /* Make cur_frm pointing to next_frm */
        layer->cur_frm = layer->next_frm;
}

static void vpbe_isr_odd_field(struct vpbe_display *disp_obj,
                                struct vpbe_layer *layer)
{
        struct osd_state *osd_device = disp_obj->osd_device;
        unsigned long addr;

        spin_lock(&disp_obj->dma_queue_lock);
        if (list_empty(&layer->dma_queue) ||
                (layer->cur_frm != layer->next_frm)) {
                spin_unlock(&disp_obj->dma_queue_lock);
                return;
        }
        /*
         * one field is displayed configure
         * the next frame if it is available
         * otherwise hold on current frame
         * Get next from the buffer queue
         */
        layer->next_frm = list_entry(layer->dma_queue.next,
                          struct  vpbe_disp_buffer, list);
        /* Remove that from the buffer queue */
        list_del(&layer->next_frm->list);
        spin_unlock(&disp_obj->dma_queue_lock);
        /* Mark state of the frame to active */
        layer->next_frm->vb.state = VB2_BUF_STATE_ACTIVE;
        addr = vb2_dma_contig_plane_dma_addr(&layer->next_frm->vb, 0);
        osd_device->ops.start_layer(osd_device,
                        layer->layer_info.id,
                        addr,
                        disp_obj->cbcr_ofst);
}

/* interrupt service routine */
static irqreturn_t venc_isr(int irq, void *arg)
{
        struct vpbe_display *disp_dev = (struct vpbe_display *)arg;
        struct vpbe_layer *layer;
        static unsigned last_event;
        unsigned event = 0;
        int fid;
        int i;

        if ((NULL == arg) || (NULL == disp_dev->dev[0]))
                return IRQ_HANDLED;

        if (venc_is_second_field(disp_dev))
                event |= VENC_SECOND_FIELD;
        else
                event |= VENC_FIRST_FIELD;

        if (event == (last_event & ~VENC_END_OF_FRAME)) {
                /*
                * If the display is non-interlaced, then we need to flag the
                * end-of-frame event at every interrupt regardless of the
                * value of the FIDST bit.  We can conclude that the display is
                * non-interlaced if the value of the FIDST bit is unchanged
                * from the previous interrupt.
                */
                event |= VENC_END_OF_FRAME;
        } else if (event == VENC_SECOND_FIELD) {
                /* end-of-frame for interlaced display */
                event |= VENC_END_OF_FRAME;
        }
        last_event = event;

        for (i = 0; i < VPBE_DISPLAY_MAX_DEVICES; i++) {
                layer = disp_dev->dev[i];
                /* If streaming is started in this layer */
                if (!layer->started)
                        continue;

                if (layer->layer_first_int) {
                        layer->layer_first_int = 0;
                        continue;
                }
                /* Check the field format */
                if ((V4L2_FIELD_NONE == layer->pix_fmt.field) &&
                        (event & VENC_END_OF_FRAME)) {
                        /* Progressive mode */

                        vpbe_isr_even_field(disp_dev, layer);
                        vpbe_isr_odd_field(disp_dev, layer);
                } else {
                /* Interlaced mode */

                        layer->field_id ^= 1;
                        if (event & VENC_FIRST_FIELD)
                                fid = 0;
                        else
                                fid = 1;

                        /*
                        * If field id does not match with store
                        * field id
                        */
                        if (fid != layer->field_id) {
                                /* Make them in sync */
                                layer->field_id = fid;
                                continue;
                        }
                        /*
                        * device field id and local field id are
                        * in sync. If this is even field
                        */
                        if (0 == fid)
                                vpbe_isr_even_field(disp_dev, layer);
                        else  /* odd field */
                                vpbe_isr_odd_field(disp_dev, layer);
                }
        }

        return IRQ_HANDLED;
}

/*
 * vpbe_buffer_prepare()
 * This is the callback function called from vb2_qbuf() function
 * the buffer is prepared and user space virtual address is converted into
 * physical address
 */
static int vpbe_buffer_prepare(struct vb2_buffer *vb)
{
        struct vpbe_fh *fh = vb2_get_drv_priv(vb->vb2_queue);
        struct vb2_queue *q = vb->vb2_queue;
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        unsigned long addr;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev,
                                "vpbe_buffer_prepare\n");

        if (vb->state != VB2_BUF_STATE_ACTIVE &&
                vb->state != VB2_BUF_STATE_PREPARED) {
                vb2_set_plane_payload(vb, 0, layer->pix_fmt.sizeimage);
                if (vb2_plane_vaddr(vb, 0) &&
                vb2_get_plane_payload(vb, 0) > vb2_plane_size(vb, 0))
                        return -EINVAL;

                addr = vb2_dma_contig_plane_dma_addr(vb, 0);
                if (q->streaming) {
                        if (!IS_ALIGNED(addr, 8)) {
                                v4l2_err(&vpbe_dev->v4l2_dev,
                                        "buffer_prepare:offset is \
                                        not aligned to 32 bytes\n");
                                return -EINVAL;
                        }
                }
        }
        return 0;
}

/*
 * vpbe_buffer_setup()
 * This function allocates memory for the buffers
 */
static int
vpbe_buffer_queue_setup(struct vb2_queue *vq, const struct v4l2_format *fmt,
                        unsigned int *nbuffers, unsigned int *nplanes,
                        unsigned int sizes[], void *alloc_ctxs[])

{
        /* Get the file handle object and layer object */
        struct vpbe_fh *fh = vb2_get_drv_priv(vq);
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "vpbe_buffer_setup\n");

        /* Store number of buffers allocated in numbuffer member */
        if (*nbuffers < VPBE_DEFAULT_NUM_BUFS)
                *nbuffers = layer->numbuffers = VPBE_DEFAULT_NUM_BUFS;

        *nplanes = 1;
        sizes[0] = layer->pix_fmt.sizeimage;
        alloc_ctxs[0] = layer->alloc_ctx;

        return 0;
}

/*
 * vpbe_buffer_queue()
 * This function adds the buffer to DMA queue
 */
static void vpbe_buffer_queue(struct vb2_buffer *vb)
{
        /* Get the file handle object and layer object */
        struct vpbe_fh *fh = vb2_get_drv_priv(vb->vb2_queue);
        struct vpbe_disp_buffer *buf = container_of(vb,
                                struct vpbe_disp_buffer, vb);
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_display *disp = fh->disp_dev;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        unsigned long flags;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev,
                        "vpbe_buffer_queue\n");

        /* add the buffer to the DMA queue */
        spin_lock_irqsave(&disp->dma_queue_lock, flags);
        list_add_tail(&buf->list, &layer->dma_queue);
        spin_unlock_irqrestore(&disp->dma_queue_lock, flags);
}

/*
 * vpbe_buf_cleanup()
 * This function is called from the vb2 layer to free memory allocated to
 * the buffers
 */
static void vpbe_buf_cleanup(struct vb2_buffer *vb)
{
        /* Get the file handle object and layer object */
        struct vpbe_fh *fh = vb2_get_drv_priv(vb->vb2_queue);
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        struct vpbe_disp_buffer *buf = container_of(vb,
                                        struct vpbe_disp_buffer, vb);
        unsigned long flags;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev,
                        "vpbe_buf_cleanup\n");

        spin_lock_irqsave(&layer->irqlock, flags);
        if (vb->state == VB2_BUF_STATE_ACTIVE)
                list_del_init(&buf->list);
        spin_unlock_irqrestore(&layer->irqlock, flags);
}

static void vpbe_wait_prepare(struct vb2_queue *vq)
{
        struct vpbe_fh *fh = vb2_get_drv_priv(vq);
        struct vpbe_layer *layer = fh->layer;

        mutex_unlock(&layer->opslock);
}

static void vpbe_wait_finish(struct vb2_queue *vq)
{
        struct vpbe_fh *fh = vb2_get_drv_priv(vq);
        struct vpbe_layer *layer = fh->layer;

        mutex_lock(&layer->opslock);
}

static int vpbe_buffer_init(struct vb2_buffer *vb)
{
        struct vpbe_disp_buffer *buf = container_of(vb,
                                        struct vpbe_disp_buffer, vb);

        INIT_LIST_HEAD(&buf->list);
        return 0;
}

static int vpbe_start_streaming(struct vb2_queue *vq, unsigned int count)
{
        struct vpbe_fh *fh = vb2_get_drv_priv(vq);
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        int ret;

        /* If buffer queue is empty, return error */
        if (list_empty(&layer->dma_queue)) {
                v4l2_err(&vpbe_dev->v4l2_dev, "buffer queue is empty\n");
                return -EINVAL;
        }
        /* Get the next frame from the buffer queue */
        layer->next_frm = layer->cur_frm = list_entry(layer->dma_queue.next,
                                struct vpbe_disp_buffer, list);
        /* Remove buffer from the buffer queue */
        list_del(&layer->cur_frm->list);
        /* Mark state of the current frame to active */
        layer->cur_frm->vb.state = VB2_BUF_STATE_ACTIVE;
        /* Initialize field_id and started member */
        layer->field_id = 0;

        /* Set parameters in OSD and VENC */
        ret = vpbe_set_osd_display_params(fh->disp_dev, layer);
        if (ret < 0)
                return ret;

        /*
         * if request format is yuv420 semiplanar, need to
         * enable both video windows
         */
        layer->started = 1;
        layer->layer_first_int = 1;

        return ret;
}

static int vpbe_stop_streaming(struct vb2_queue *vq)
{
        struct vpbe_fh *fh = vb2_get_drv_priv(vq);
        struct vpbe_layer *layer = fh->layer;

        if (!vb2_is_streaming(vq))
                return 0;

        /* release all active buffers */
        while (!list_empty(&layer->dma_queue)) {
                layer->next_frm = list_entry(layer->dma_queue.next,
                                                struct vpbe_disp_buffer, list);
                list_del(&layer->next_frm->list);
                vb2_buffer_done(&layer->next_frm->vb, VB2_BUF_STATE_ERROR);
        }

        return 0;
}

static struct vb2_ops video_qops = {
        .queue_setup = vpbe_buffer_queue_setup,
        .wait_prepare = vpbe_wait_prepare,
        .wait_finish = vpbe_wait_finish,
        .buf_init = vpbe_buffer_init,
        .buf_prepare = vpbe_buffer_prepare,
        .start_streaming = vpbe_start_streaming,
        .stop_streaming = vpbe_stop_streaming,
        .buf_cleanup = vpbe_buf_cleanup,
        .buf_queue = vpbe_buffer_queue,
};

static
struct vpbe_layer*
_vpbe_display_get_other_win_layer(struct vpbe_display *disp_dev,
                        struct vpbe_layer *layer)
{
        enum vpbe_display_device_id thiswin, otherwin;
        thiswin = layer->device_id;

        otherwin = (thiswin == VPBE_DISPLAY_DEVICE_0) ?
        VPBE_DISPLAY_DEVICE_1 : VPBE_DISPLAY_DEVICE_0;
        return disp_dev->dev[otherwin];
}

static int vpbe_set_osd_display_params(struct vpbe_display *disp_dev,
                        struct vpbe_layer *layer)
{
        struct osd_layer_config *cfg  = &layer->layer_info.config;
        struct osd_state *osd_device = disp_dev->osd_device;
        struct vpbe_device *vpbe_dev = disp_dev->vpbe_dev;
        unsigned long addr;
        int ret;

        addr = vb2_dma_contig_plane_dma_addr(&layer->cur_frm->vb, 0);
        /* Set address in the display registers */
        osd_device->ops.start_layer(osd_device,
                                    layer->layer_info.id,
                                    addr,
                                    disp_dev->cbcr_ofst);

        ret = osd_device->ops.enable_layer(osd_device,
                                layer->layer_info.id, 0);
        if (ret < 0) {
                v4l2_err(&vpbe_dev->v4l2_dev,
                        "Error in enabling osd window layer 0\n");
                return -1;
        }

        /* Enable the window */
        layer->layer_info.enable = 1;
        if (cfg->pixfmt == PIXFMT_NV12) {
                struct vpbe_layer *otherlayer =
                        _vpbe_display_get_other_win_layer(disp_dev, layer);

                ret = osd_device->ops.enable_layer(osd_device,
                                otherlayer->layer_info.id, 1);
                if (ret < 0) {
                        v4l2_err(&vpbe_dev->v4l2_dev,
                                "Error in enabling osd window layer 1\n");
                        return -1;
                }
                otherlayer->layer_info.enable = 1;
        }
        return 0;
}

static void
vpbe_disp_calculate_scale_factor(struct vpbe_display *disp_dev,
                        struct vpbe_layer *layer,
                        int expected_xsize, int expected_ysize)
{
        struct display_layer_info *layer_info = &layer->layer_info;
        struct v4l2_pix_format *pixfmt = &layer->pix_fmt;
        struct osd_layer_config *cfg  = &layer->layer_info.config;
        struct vpbe_device *vpbe_dev = disp_dev->vpbe_dev;
        int calculated_xsize;
        int h_exp = 0;
        int v_exp = 0;
        int h_scale;
        int v_scale;

        v4l2_std_id standard_id = vpbe_dev->current_timings.std_id;

        /*
         * Application initially set the image format. Current display
         * size is obtained from the vpbe display controller. expected_xsize
         * and expected_ysize are set through S_CROP ioctl. Based on this,
         * driver will calculate the scale factors for vertical and
         * horizontal direction so that the image is displayed scaled
         * and expanded. Application uses expansion to display the image
         * in a square pixel. Otherwise it is displayed using displays
         * pixel aspect ratio.It is expected that application chooses
         * the crop coordinates for cropped or scaled display. if crop
         * size is less than the image size, it is displayed cropped or
         * it is displayed scaled and/or expanded.
         *
         * to begin with, set the crop window same as expected. Later we
         * will override with scaled window size
         */

        cfg->xsize = pixfmt->width;
        cfg->ysize = pixfmt->height;
        layer_info->h_zoom = ZOOM_X1;   /* no horizontal zoom */
        layer_info->v_zoom = ZOOM_X1;   /* no horizontal zoom */
        layer_info->h_exp = H_EXP_OFF;  /* no horizontal zoom */
        layer_info->v_exp = V_EXP_OFF;  /* no horizontal zoom */

        if (pixfmt->width < expected_xsize) {
                h_scale = vpbe_dev->current_timings.xres / pixfmt->width;
                if (h_scale < 2)
                        h_scale = 1;
                else if (h_scale >= 4)
                        h_scale = 4;
                else
                        h_scale = 2;
                cfg->xsize *= h_scale;
                if (cfg->xsize < expected_xsize) {
                        if ((standard_id & V4L2_STD_525_60) ||
                        (standard_id & V4L2_STD_625_50)) {
                                calculated_xsize = (cfg->xsize *
                                        VPBE_DISPLAY_H_EXP_RATIO_N) /
                                        VPBE_DISPLAY_H_EXP_RATIO_D;
                                if (calculated_xsize <= expected_xsize) {
                                        h_exp = 1;
                                        cfg->xsize = calculated_xsize;
                                }
                        }
                }
                if (h_scale == 2)
                        layer_info->h_zoom = ZOOM_X2;
                else if (h_scale == 4)
                        layer_info->h_zoom = ZOOM_X4;
                if (h_exp)
                        layer_info->h_exp = H_EXP_9_OVER_8;
        } else {
                /* no scaling, only cropping. Set display area to crop area */
                cfg->xsize = expected_xsize;
        }

        if (pixfmt->height < expected_ysize) {
                v_scale = expected_ysize / pixfmt->height;
                if (v_scale < 2)
                        v_scale = 1;
                else if (v_scale >= 4)
                        v_scale = 4;
                else
                        v_scale = 2;
                cfg->ysize *= v_scale;
                if (cfg->ysize < expected_ysize) {
                        if ((standard_id & V4L2_STD_625_50)) {
                                calculated_xsize = (cfg->ysize *
                                        VPBE_DISPLAY_V_EXP_RATIO_N) /
                                        VPBE_DISPLAY_V_EXP_RATIO_D;
                                if (calculated_xsize <= expected_ysize) {
                                        v_exp = 1;
                                        cfg->ysize = calculated_xsize;
                                }
                        }
                }
                if (v_scale == 2)
                        layer_info->v_zoom = ZOOM_X2;
                else if (v_scale == 4)
                        layer_info->v_zoom = ZOOM_X4;
                if (v_exp)
                        layer_info->h_exp = V_EXP_6_OVER_5;
        } else {
                /* no scaling, only cropping. Set display area to crop area */
                cfg->ysize = expected_ysize;
        }
        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev,
                "crop display xsize = %d, ysize = %d\n",
                cfg->xsize, cfg->ysize);
}

static void vpbe_disp_adj_position(struct vpbe_display *disp_dev,
                        struct vpbe_layer *layer,
                        int top, int left)
{
        struct osd_layer_config *cfg = &layer->layer_info.config;
        struct vpbe_device *vpbe_dev = disp_dev->vpbe_dev;

        cfg->xpos = min((unsigned int)left,
                        vpbe_dev->current_timings.xres - cfg->xsize);
        cfg->ypos = min((unsigned int)top,
                        vpbe_dev->current_timings.yres - cfg->ysize);

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev,
                "new xpos = %d, ypos = %d\n",
                cfg->xpos, cfg->ypos);
}

static void vpbe_disp_check_window_params(struct vpbe_display *disp_dev,
                        struct v4l2_rect *c)
{
        struct vpbe_device *vpbe_dev = disp_dev->vpbe_dev;

        if ((c->width == 0) ||
          ((c->width + c->left) > vpbe_dev->current_timings.xres))
                c->width = vpbe_dev->current_timings.xres - c->left;

        if ((c->height == 0) || ((c->height + c->top) >
          vpbe_dev->current_timings.yres))
                c->height = vpbe_dev->current_timings.yres - c->top;

        /* window height must be even for interlaced display */
        if (vpbe_dev->current_timings.interlaced)
                c->height &= (~0x01);

}

/**
 * vpbe_try_format()
 * If user application provides width and height, and have bytesperline set
 * to zero, driver calculates bytesperline and sizeimage based on hardware
 * limits.
 */
static int vpbe_try_format(struct vpbe_display *disp_dev,
                        struct v4l2_pix_format *pixfmt, int check)
{
        struct vpbe_device *vpbe_dev = disp_dev->vpbe_dev;
        int min_height = 1;
        int min_width = 32;
        int max_height;
        int max_width;
        int bpp;

        if ((pixfmt->pixelformat != V4L2_PIX_FMT_UYVY) &&
            (pixfmt->pixelformat != V4L2_PIX_FMT_NV12))
                /* choose default as V4L2_PIX_FMT_UYVY */
                pixfmt->pixelformat = V4L2_PIX_FMT_UYVY;

        /* Check the field format */
        if ((pixfmt->field != V4L2_FIELD_INTERLACED) &&
                (pixfmt->field != V4L2_FIELD_NONE)) {
                if (vpbe_dev->current_timings.interlaced)
                        pixfmt->field = V4L2_FIELD_INTERLACED;
                else
                        pixfmt->field = V4L2_FIELD_NONE;
        }

        if (pixfmt->field == V4L2_FIELD_INTERLACED)
                min_height = 2;

        if (pixfmt->pixelformat == V4L2_PIX_FMT_NV12)
                bpp = 1;
        else
                bpp = 2;

        max_width = vpbe_dev->current_timings.xres;
        max_height = vpbe_dev->current_timings.yres;

        min_width /= bpp;

        if (!pixfmt->width || (pixfmt->width < min_width) ||
                (pixfmt->width > max_width)) {
                pixfmt->width = vpbe_dev->current_timings.xres;
        }

        if (!pixfmt->height || (pixfmt->height  < min_height) ||
                (pixfmt->height  > max_height)) {
                pixfmt->height = vpbe_dev->current_timings.yres;
        }

        if (pixfmt->bytesperline < (pixfmt->width * bpp))
                pixfmt->bytesperline = pixfmt->width * bpp;

        /* Make the bytesperline 32 byte aligned */
        pixfmt->bytesperline = ((pixfmt->width * bpp + 31) & ~31);

        if (pixfmt->pixelformat == V4L2_PIX_FMT_NV12)
                pixfmt->sizeimage = pixfmt->bytesperline * pixfmt->height +
                                (pixfmt->bytesperline * pixfmt->height >> 1);
        else
                pixfmt->sizeimage = pixfmt->bytesperline * pixfmt->height;

        return 0;
}

static int vpbe_display_g_priority(struct file *file, void *priv,
                                enum v4l2_priority *p)
{
        struct vpbe_fh *fh = file->private_data;
        struct vpbe_layer *layer = fh->layer;

        *p = v4l2_prio_max(&layer->prio);

        return 0;
}

static int vpbe_display_s_priority(struct file *file, void *priv,
                                enum v4l2_priority p)
{
        struct vpbe_fh *fh = file->private_data;
        struct vpbe_layer *layer = fh->layer;
        int ret;

        ret = v4l2_prio_change(&layer->prio, &fh->prio, p);

        return ret;
}

static int vpbe_display_querycap(struct file *file, void  *priv,
                               struct v4l2_capability *cap)
{
        struct vpbe_fh *fh = file->private_data;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;

        cap->version = VPBE_DISPLAY_VERSION_CODE;
        cap->device_caps = V4L2_CAP_VIDEO_OUTPUT | V4L2_CAP_STREAMING;
        cap->capabilities = cap->device_caps | V4L2_CAP_DEVICE_CAPS;
        snprintf(cap->driver, sizeof(cap->driver), "%s",
                dev_name(vpbe_dev->pdev));
        snprintf(cap->bus_info, sizeof(cap->bus_info), "platform:%s",
                 dev_name(vpbe_dev->pdev));
        strlcpy(cap->card, vpbe_dev->cfg->module_name, sizeof(cap->card));

        return 0;
}

static int vpbe_display_s_crop(struct file *file, void *priv,
                             const struct v4l2_crop *crop)
{
        struct vpbe_fh *fh = file->private_data;
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_display *disp_dev = fh->disp_dev;
        struct vpbe_device *vpbe_dev = disp_dev->vpbe_dev;
        struct osd_layer_config *cfg = &layer->layer_info.config;
        struct osd_state *osd_device = disp_dev->osd_device;
        struct v4l2_rect rect = crop->c;
        int ret;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev,
                "VIDIOC_S_CROP, layer id = %d\n", layer->device_id);

        if (crop->type != V4L2_BUF_TYPE_VIDEO_OUTPUT) {
                v4l2_err(&vpbe_dev->v4l2_dev, "Invalid buf type\n");
                return -EINVAL;
        }

        if (rect.top < 0)
                rect.top = 0;
        if (rect.left < 0)
                rect.left = 0;

        vpbe_disp_check_window_params(disp_dev, &rect);

        osd_device->ops.get_layer_config(osd_device,
                        layer->layer_info.id, cfg);

        vpbe_disp_calculate_scale_factor(disp_dev, layer,
                                        rect.width,
                                        rect.height);
        vpbe_disp_adj_position(disp_dev, layer, rect.top,
                                        rect.left);
        ret = osd_device->ops.set_layer_config(osd_device,
                                layer->layer_info.id, cfg);
        if (ret < 0) {
                v4l2_err(&vpbe_dev->v4l2_dev,
                        "Error in set layer config:\n");
                return -EINVAL;
        }

        /* apply zooming and h or v expansion */
        osd_device->ops.set_zoom(osd_device,
                        layer->layer_info.id,
                        layer->layer_info.h_zoom,
                        layer->layer_info.v_zoom);
        ret = osd_device->ops.set_vid_expansion(osd_device,
                        layer->layer_info.h_exp,
                        layer->layer_info.v_exp);
        if (ret < 0) {
                v4l2_err(&vpbe_dev->v4l2_dev,
                "Error in set vid expansion:\n");
                return -EINVAL;
        }

        if ((layer->layer_info.h_zoom != ZOOM_X1) ||
                (layer->layer_info.v_zoom != ZOOM_X1) ||
                (layer->layer_info.h_exp != H_EXP_OFF) ||
                (layer->layer_info.v_exp != V_EXP_OFF))
                /* Enable expansion filter */
                osd_device->ops.set_interpolation_filter(osd_device, 1);
        else
                osd_device->ops.set_interpolation_filter(osd_device, 0);

        return 0;
}

static int vpbe_display_g_crop(struct file *file, void *priv,
                             struct v4l2_crop *crop)
{
        struct vpbe_fh *fh = file->private_data;
        struct vpbe_layer *layer = fh->layer;
        struct osd_layer_config *cfg = &layer->layer_info.config;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        struct osd_state *osd_device = fh->disp_dev->osd_device;
        struct v4l2_rect *rect = &crop->c;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev,
                        "VIDIOC_G_CROP, layer id = %d\n",
                        layer->device_id);

        if (crop->type != V4L2_BUF_TYPE_VIDEO_OUTPUT) {
                v4l2_err(&vpbe_dev->v4l2_dev, "Invalid buf type\n");
                return -EINVAL;
        }
        osd_device->ops.get_layer_config(osd_device,
                                layer->layer_info.id, cfg);
        rect->top = cfg->ypos;
        rect->left = cfg->xpos;
        rect->width = cfg->xsize;
        rect->height = cfg->ysize;

        return 0;
}

static int vpbe_display_cropcap(struct file *file, void *priv,
                              struct v4l2_cropcap *cropcap)
{
        struct vpbe_fh *fh = file->private_data;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "VIDIOC_CROPCAP ioctl\n");

        cropcap->type = V4L2_BUF_TYPE_VIDEO_OUTPUT;
        cropcap->bounds.left = 0;
        cropcap->bounds.top = 0;
        cropcap->bounds.width = vpbe_dev->current_timings.xres;
        cropcap->bounds.height = vpbe_dev->current_timings.yres;
        cropcap->pixelaspect = vpbe_dev->current_timings.aspect;
        cropcap->defrect = cropcap->bounds;
        return 0;
}

static int vpbe_display_g_fmt(struct file *file, void *priv,
                                struct v4l2_format *fmt)
{
        struct vpbe_fh *fh = file->private_data;
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev,
                        "VIDIOC_G_FMT, layer id = %d\n",
                        layer->device_id);

        /* If buffer type is video output */
        if (V4L2_BUF_TYPE_VIDEO_OUTPUT != fmt->type) {
                v4l2_err(&vpbe_dev->v4l2_dev, "invalid type\n");
                return -EINVAL;
        }
        /* Fill in the information about format */
        fmt->fmt.pix = layer->pix_fmt;

        return 0;
}

static int vpbe_display_enum_fmt(struct file *file, void  *priv,
                                   struct v4l2_fmtdesc *fmt)
{
        struct vpbe_fh *fh = file->private_data;
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        unsigned int index = 0;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev,
                                "VIDIOC_ENUM_FMT, layer id = %d\n",
                                layer->device_id);
        if (fmt->index > 1) {
                v4l2_err(&vpbe_dev->v4l2_dev, "Invalid format index\n");
                return -EINVAL;
        }

        /* Fill in the information about format */
        index = fmt->index;
        memset(fmt, 0, sizeof(*fmt));
        fmt->index = index;
        fmt->type = V4L2_BUF_TYPE_VIDEO_OUTPUT;
        if (index == 0) {
                strcpy(fmt->description, "YUV 4:2:2 - UYVY");
                fmt->pixelformat = V4L2_PIX_FMT_UYVY;
        } else {
                strcpy(fmt->description, "Y/CbCr 4:2:0");
                fmt->pixelformat = V4L2_PIX_FMT_NV12;
        }

        return 0;
}

static int vpbe_display_s_fmt(struct file *file, void *priv,
                                struct v4l2_format *fmt)
{
        struct vpbe_fh *fh = file->private_data;
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_display *disp_dev = fh->disp_dev;
        struct vpbe_device *vpbe_dev = disp_dev->vpbe_dev;
        struct osd_layer_config *cfg  = &layer->layer_info.config;
        struct v4l2_pix_format *pixfmt = &fmt->fmt.pix;
        struct osd_state *osd_device = disp_dev->osd_device;
        int ret;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev,
                        "VIDIOC_S_FMT, layer id = %d\n",
                        layer->device_id);

        /* If streaming is started, return error */
        if (layer->started) {
                v4l2_err(&vpbe_dev->v4l2_dev, "Streaming is started\n");
                return -EBUSY;
        }
        if (V4L2_BUF_TYPE_VIDEO_OUTPUT != fmt->type) {
                v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "invalid type\n");
                return -EINVAL;
        }
        /* Check for valid pixel format */
        ret = vpbe_try_format(disp_dev, pixfmt, 1);
        if (ret)
                return ret;

        /* YUV420 is requested, check availability of the
        other video window */

        layer->pix_fmt = *pixfmt;
        if (pixfmt->pixelformat == V4L2_PIX_FMT_NV12) {
                struct vpbe_layer *otherlayer;

                otherlayer = _vpbe_display_get_other_win_layer(disp_dev, layer);
                /* if other layer is available, only
                 * claim it, do not configure it
                 */
                ret = osd_device->ops.request_layer(osd_device,
                                                    otherlayer->layer_info.id);
                if (ret < 0) {
                        v4l2_err(&vpbe_dev->v4l2_dev,
                                 "Display Manager failed to allocate layer\n");
                        return -EBUSY;
                }
        }

        /* Get osd layer config */
        osd_device->ops.get_layer_config(osd_device,
                        layer->layer_info.id, cfg);
        /* Store the pixel format in the layer object */
        cfg->xsize = pixfmt->width;
        cfg->ysize = pixfmt->height;
        cfg->line_length = pixfmt->bytesperline;
        cfg->ypos = 0;
        cfg->xpos = 0;
        cfg->interlaced = vpbe_dev->current_timings.interlaced;

        if (V4L2_PIX_FMT_UYVY == pixfmt->pixelformat)
                cfg->pixfmt = PIXFMT_YCBCRI;

        /* Change of the default pixel format for both video windows */
        if (V4L2_PIX_FMT_NV12 == pixfmt->pixelformat) {
                struct vpbe_layer *otherlayer;
                cfg->pixfmt = PIXFMT_NV12;
                otherlayer = _vpbe_display_get_other_win_layer(disp_dev,
                                                                layer);
                otherlayer->layer_info.config.pixfmt = PIXFMT_NV12;
        }

        /* Set the layer config in the osd window */
        ret = osd_device->ops.set_layer_config(osd_device,
                                layer->layer_info.id, cfg);
        if (ret < 0) {
                v4l2_err(&vpbe_dev->v4l2_dev,
                                "Error in S_FMT params:\n");
                return -EINVAL;
        }

        /* Readback and fill the local copy of current pix format */
        osd_device->ops.get_layer_config(osd_device,
                        layer->layer_info.id, cfg);

        return 0;
}

static int vpbe_display_try_fmt(struct file *file, void *priv,
                                  struct v4l2_format *fmt)
{
        struct vpbe_fh *fh = file->private_data;
        struct vpbe_display *disp_dev = fh->disp_dev;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        struct v4l2_pix_format *pixfmt = &fmt->fmt.pix;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "VIDIOC_TRY_FMT\n");

        if (V4L2_BUF_TYPE_VIDEO_OUTPUT != fmt->type) {
                v4l2_err(&vpbe_dev->v4l2_dev, "invalid type\n");
                return -EINVAL;
        }

        /* Check for valid field format */
        return  vpbe_try_format(disp_dev, pixfmt, 0);

}

/**
 * vpbe_display_s_std - Set the given standard in the encoder
 *
 * Sets the standard if supported by the current encoder. Return the status.
 * 0 - success & -EINVAL on error
 */
static int vpbe_display_s_std(struct file *file, void *priv,

                                v4l2_std_id std_id)
{
        struct vpbe_fh *fh = priv;
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        int ret;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "VIDIOC_S_STD\n");

        /* If streaming is started, return error */
        if (layer->started) {
                v4l2_err(&vpbe_dev->v4l2_dev, "Streaming is started\n");
                return -EBUSY;
        }
        if (NULL != vpbe_dev->ops.s_std) {
                ret = vpbe_dev->ops.s_std(vpbe_dev, std_id);
                if (ret) {
                        v4l2_err(&vpbe_dev->v4l2_dev,
                        "Failed to set standard for sub devices\n");
                        return -EINVAL;
                }
        } else {
                return -EINVAL;
        }

        return 0;
}

/**
 * vpbe_display_g_std - Get the standard in the current encoder
 *
 * Get the standard in the current encoder. Return the status. 0 - success
 * -EINVAL on error
 */
static int vpbe_display_g_std(struct file *file, void *priv,
                                v4l2_std_id *std_id)
{
        struct vpbe_fh *fh = priv;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "VIDIOC_G_STD\n");

        /* Get the standard from the current encoder */
        if (vpbe_dev->current_timings.timings_type & VPBE_ENC_STD) {
                *std_id = vpbe_dev->current_timings.std_id;
                return 0;
        }

        return -EINVAL;
}

/**
 * vpbe_display_enum_output - enumerate outputs
 *
 * Enumerates the outputs available at the vpbe display
 * returns the status, -EINVAL if end of output list
 */
static int vpbe_display_enum_output(struct file *file, void *priv,
                                    struct v4l2_output *output)
{
        struct vpbe_fh *fh = priv;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        int ret;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "VIDIOC_ENUM_OUTPUT\n");

        /* Enumerate outputs */

        if (NULL == vpbe_dev->ops.enum_outputs)
                return -EINVAL;

        ret = vpbe_dev->ops.enum_outputs(vpbe_dev, output);
        if (ret) {
                v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev,
                        "Failed to enumerate outputs\n");
                return -EINVAL;
        }

        return 0;
}

/**
 * vpbe_display_s_output - Set output to
 * the output specified by the index
 */
static int vpbe_display_s_output(struct file *file, void *priv,
                                unsigned int i)
{
        struct vpbe_fh *fh = priv;
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        int ret;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "VIDIOC_S_OUTPUT\n");
        /* If streaming is started, return error */
        if (layer->started) {
                v4l2_err(&vpbe_dev->v4l2_dev, "Streaming is started\n");
                return -EBUSY;
        }
        if (NULL == vpbe_dev->ops.set_output)
                return -EINVAL;

        ret = vpbe_dev->ops.set_output(vpbe_dev, i);
        if (ret) {
                v4l2_err(&vpbe_dev->v4l2_dev,
                        "Failed to set output for sub devices\n");
                return -EINVAL;
        }

        return 0;
}

/**
 * vpbe_display_g_output - Get output from subdevice
 * for a given by the index
 */
static int vpbe_display_g_output(struct file *file, void *priv,
                                unsigned int *i)
{
        struct vpbe_fh *fh = priv;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "VIDIOC_G_OUTPUT\n");
        /* Get the standard from the current encoder */
        *i = vpbe_dev->current_out_index;

        return 0;
}

/**
 * vpbe_display_enum_dv_timings - Enumerate the dv timings
 *
 * enum the timings in the current encoder. Return the status. 0 - success
 * -EINVAL on error
 */
static int
vpbe_display_enum_dv_timings(struct file *file, void *priv,
                        struct v4l2_enum_dv_timings *timings)
{
        struct vpbe_fh *fh = priv;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        int ret;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "VIDIOC_ENUM_DV_TIMINGS\n");

        /* Enumerate outputs */
        if (NULL == vpbe_dev->ops.enum_dv_timings)
                return -EINVAL;

        ret = vpbe_dev->ops.enum_dv_timings(vpbe_dev, timings);
        if (ret) {
                v4l2_err(&vpbe_dev->v4l2_dev,
                        "Failed to enumerate dv timings info\n");
                return -EINVAL;
        }

        return 0;
}

/**
 * vpbe_display_s_dv_timings - Set the dv timings
 *
 * Set the timings in the current encoder. Return the status. 0 - success
 * -EINVAL on error
 */
static int
vpbe_display_s_dv_timings(struct file *file, void *priv,
                                struct v4l2_dv_timings *timings)
{
        struct vpbe_fh *fh = priv;
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        int ret;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "VIDIOC_S_DV_TIMINGS\n");


        /* If streaming is started, return error */
        if (layer->started) {
                v4l2_err(&vpbe_dev->v4l2_dev, "Streaming is started\n");
                return -EBUSY;
        }

        /* Set the given standard in the encoder */
        if (!vpbe_dev->ops.s_dv_timings)
                return -EINVAL;

        ret = vpbe_dev->ops.s_dv_timings(vpbe_dev, timings);
        if (ret) {
                v4l2_err(&vpbe_dev->v4l2_dev,
                        "Failed to set the dv timings info\n");
                return -EINVAL;
        }

        return 0;
}

/**
 * vpbe_display_g_dv_timings - Set the dv timings
 *
 * Get the timings in the current encoder. Return the status. 0 - success
 * -EINVAL on error
 */
static int
vpbe_display_g_dv_timings(struct file *file, void *priv,
                                struct v4l2_dv_timings *dv_timings)
{
        struct vpbe_fh *fh = priv;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "VIDIOC_G_DV_TIMINGS\n");

        /* Get the given standard in the encoder */

        if (vpbe_dev->current_timings.timings_type &
                                VPBE_ENC_DV_TIMINGS) {
                *dv_timings = vpbe_dev->current_timings.dv_timings;
        } else {
                return -EINVAL;
        }

        return 0;
}

static int vpbe_display_streamoff(struct file *file, void *priv,
                                enum v4l2_buf_type buf_type)
{
        struct vpbe_fh *fh = file->private_data;
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        struct osd_state *osd_device = fh->disp_dev->osd_device;
        int ret;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev,
                        "VIDIOC_STREAMOFF,layer id = %d\n",
                        layer->device_id);

        if (V4L2_BUF_TYPE_VIDEO_OUTPUT != buf_type) {
                v4l2_err(&vpbe_dev->v4l2_dev, "Invalid buffer type\n");
                return -EINVAL;
        }

        /* If io is allowed for this file handle, return error */
        if (!fh->io_allowed) {
                v4l2_err(&vpbe_dev->v4l2_dev, "No io_allowed\n");
                return -EACCES;
        }

        /* If streaming is not started, return error */
        if (!layer->started) {
                v4l2_err(&vpbe_dev->v4l2_dev, "streaming not started in layer"
                        " id = %d\n", layer->device_id);
                return -EINVAL;
        }

        osd_device->ops.disable_layer(osd_device,
                        layer->layer_info.id);
        layer->started = 0;
        ret = vb2_streamoff(&layer->buffer_queue, buf_type);

        return ret;
}

static int vpbe_display_streamon(struct file *file, void *priv,
                         enum v4l2_buf_type buf_type)
{
        struct vpbe_fh *fh = file->private_data;
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_display *disp_dev = fh->disp_dev;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        struct osd_state *osd_device = disp_dev->osd_device;
        int ret;

        osd_device->ops.disable_layer(osd_device,
                        layer->layer_info.id);

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "VIDIOC_STREAMON, layerid=%d\n",
                                                layer->device_id);

        if (V4L2_BUF_TYPE_VIDEO_OUTPUT != buf_type) {
                v4l2_err(&vpbe_dev->v4l2_dev, "Invalid buffer type\n");
                return -EINVAL;
        }

        /* If file handle is not allowed IO, return error */
        if (!fh->io_allowed) {
                v4l2_err(&vpbe_dev->v4l2_dev, "No io_allowed\n");
                return -EACCES;
        }
        /* If Streaming is already started, return error */
        if (layer->started) {
                v4l2_err(&vpbe_dev->v4l2_dev, "layer is already streaming\n");
                return -EBUSY;
        }

        /*
         * Call vb2_streamon to start streaming
         * in videobuf
         */
        ret = vb2_streamon(&layer->buffer_queue, buf_type);
        if (ret) {
                v4l2_err(&vpbe_dev->v4l2_dev,
                "error in vb2_streamon\n");
                return ret;
        }
        return ret;
}

static int vpbe_display_dqbuf(struct file *file, void *priv,
                      struct v4l2_buffer *buf)
{
        struct vpbe_fh *fh = file->private_data;
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        int ret;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev,
                "VIDIOC_DQBUF, layer id = %d\n",
                layer->device_id);

        if (V4L2_BUF_TYPE_VIDEO_OUTPUT != buf->type) {
                v4l2_err(&vpbe_dev->v4l2_dev, "Invalid buffer type\n");
                return -EINVAL;
        }
        /* If this file handle is not allowed to do IO, return error */
        if (!fh->io_allowed) {
                v4l2_err(&vpbe_dev->v4l2_dev, "No io_allowed\n");
                return -EACCES;
        }
        if (file->f_flags & O_NONBLOCK)
                /* Call videobuf_dqbuf for non blocking mode */
                ret = vb2_dqbuf(&layer->buffer_queue, buf, 1);
        else
                /* Call videobuf_dqbuf for blocking mode */
                ret = vb2_dqbuf(&layer->buffer_queue, buf, 0);

        return ret;
}

static int vpbe_display_qbuf(struct file *file, void *priv,
                     struct v4l2_buffer *p)
{
        struct vpbe_fh *fh = file->private_data;
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev,
                "VIDIOC_QBUF, layer id = %d\n",
                layer->device_id);

        if (V4L2_BUF_TYPE_VIDEO_OUTPUT != p->type) {
                v4l2_err(&vpbe_dev->v4l2_dev, "Invalid buffer type\n");
                return -EINVAL;
        }

        /* If this file handle is not allowed to do IO, return error */
        if (!fh->io_allowed) {
                v4l2_err(&vpbe_dev->v4l2_dev, "No io_allowed\n");
                return -EACCES;
        }

        return vb2_qbuf(&layer->buffer_queue, p);
}

static int vpbe_display_querybuf(struct file *file, void *priv,
                         struct v4l2_buffer *buf)
{
        struct vpbe_fh *fh = file->private_data;
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev,
                "VIDIOC_QUERYBUF, layer id = %d\n",
                layer->device_id);

        if (V4L2_BUF_TYPE_VIDEO_OUTPUT != buf->type) {
                v4l2_err(&vpbe_dev->v4l2_dev, "Invalid buffer type\n");
                return -EINVAL;
        }
        /* Call vb2_querybuf to get information */
        return vb2_querybuf(&layer->buffer_queue, buf);
}

static int vpbe_display_reqbufs(struct file *file, void *priv,
                        struct v4l2_requestbuffers *req_buf)
{
        struct vpbe_fh *fh = file->private_data;
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        struct vb2_queue *q;
        int ret;
        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "vpbe_display_reqbufs\n");

        if (V4L2_BUF_TYPE_VIDEO_OUTPUT != req_buf->type) {
                v4l2_err(&vpbe_dev->v4l2_dev, "Invalid buffer type\n");
                return -EINVAL;
        }

        /* If io users of the layer is not zero, return error */
        if (0 != layer->io_usrs) {
                v4l2_err(&vpbe_dev->v4l2_dev, "not IO user\n");
                return -EBUSY;
        }
        /* Initialize videobuf queue as per the buffer type */
        layer->alloc_ctx = vb2_dma_contig_init_ctx(vpbe_dev->pdev);
        if (IS_ERR(layer->alloc_ctx)) {
                v4l2_err(&vpbe_dev->v4l2_dev, "Failed to get the context\n");
                return PTR_ERR(layer->alloc_ctx);
        }
        q = &layer->buffer_queue;
        memset(q, 0, sizeof(*q));
        q->type = V4L2_BUF_TYPE_VIDEO_OUTPUT;
        q->io_modes = VB2_MMAP | VB2_USERPTR;
        q->drv_priv = fh;
        q->ops = &video_qops;
        q->mem_ops = &vb2_dma_contig_memops;
        q->buf_struct_size = sizeof(struct vpbe_disp_buffer);
        q->timestamp_type = V4L2_BUF_FLAG_TIMESTAMP_MONOTONIC;

        ret = vb2_queue_init(q);
        if (ret) {
                v4l2_err(&vpbe_dev->v4l2_dev, "vb2_queue_init() failed\n");
                vb2_dma_contig_cleanup_ctx(layer->alloc_ctx);
                return ret;
        }
        /* Set io allowed member of file handle to TRUE */
        fh->io_allowed = 1;
        /* Increment io usrs member of layer object to 1 */
        layer->io_usrs = 1;
        /* Store type of memory requested in layer object */
        layer->memory = req_buf->memory;
        /* Initialize buffer queue */
        INIT_LIST_HEAD(&layer->dma_queue);
        /* Allocate buffers */
        return vb2_reqbufs(q, req_buf);
}

/*
 * vpbe_display_mmap()
 * It is used to map kernel space buffers into user spaces
 */
static int vpbe_display_mmap(struct file *filep, struct vm_area_struct *vma)
{
        /* Get the layer object and file handle object */
        struct vpbe_fh *fh = filep->private_data;
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        int ret;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "vpbe_display_mmap\n");

        if (mutex_lock_interruptible(&layer->opslock))
                return -ERESTARTSYS;
        ret = vb2_mmap(&layer->buffer_queue, vma);
        mutex_unlock(&layer->opslock);
        return ret;
}

/* vpbe_display_poll(): It is used for select/poll system call
 */
static unsigned int vpbe_display_poll(struct file *filep, poll_table *wait)
{
        struct vpbe_fh *fh = filep->private_data;
        struct vpbe_layer *layer = fh->layer;
        struct vpbe_device *vpbe_dev = fh->disp_dev->vpbe_dev;
        unsigned int err = 0;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "vpbe_display_poll\n");
        if (layer->started) {
                mutex_lock(&layer->opslock);
                err = vb2_poll(&layer->buffer_queue, filep, wait);
                mutex_unlock(&layer->opslock);
        }
        return err;
}

/*
 * vpbe_display_open()
 * It creates object of file handle structure and stores it in private_data
 * member of filepointer
 */
static int vpbe_display_open(struct file *file)
{
        struct vpbe_fh *fh = NULL;
        struct vpbe_layer *layer = video_drvdata(file);
        struct vpbe_display *disp_dev = layer->disp_dev;
        struct vpbe_device *vpbe_dev = disp_dev->vpbe_dev;
        struct osd_state *osd_device = disp_dev->osd_device;
        int err;

        /* Allocate memory for the file handle object */
        fh = kmalloc(sizeof(struct vpbe_fh), GFP_KERNEL);
        if (fh == NULL) {
                v4l2_err(&vpbe_dev->v4l2_dev,
                        "unable to allocate memory for file handle object\n");
                return -ENOMEM;
        }
        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev,
                        "vpbe display open plane = %d\n",
                        layer->device_id);

        /* store pointer to fh in private_data member of filep */
        file->private_data = fh;
        fh->layer = layer;
        fh->disp_dev = disp_dev;

        if (!layer->usrs) {
                if (mutex_lock_interruptible(&layer->opslock))
                        return -ERESTARTSYS;
                /* First claim the layer for this device */
                err = osd_device->ops.request_layer(osd_device,
                                                layer->layer_info.id);
                mutex_unlock(&layer->opslock);
                if (err < 0) {
                        /* Couldn't get layer */
                        v4l2_err(&vpbe_dev->v4l2_dev,
                                "Display Manager failed to allocate layer\n");
                        kfree(fh);
                        return -EINVAL;
                }
        }
        /* Increment layer usrs counter */
        layer->usrs++;
        /* Set io_allowed member to false */
        fh->io_allowed = 0;
        /* Initialize priority of this instance to default priority */
        fh->prio = V4L2_PRIORITY_UNSET;
        v4l2_prio_open(&layer->prio, &fh->prio);
        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev,
                        "vpbe display device opened successfully\n");
        return 0;
}

/*
 * vpbe_display_release()
 * This function deletes buffer queue, frees the buffers and the davinci
 * display file * handle
 */
static int vpbe_display_release(struct file *file)
{
        /* Get the layer object and file handle object */
        struct vpbe_fh *fh = file->private_data;
        struct vpbe_layer *layer = fh->layer;
        struct osd_layer_config *cfg  = &layer->layer_info.config;
        struct vpbe_display *disp_dev = fh->disp_dev;
        struct vpbe_device *vpbe_dev = disp_dev->vpbe_dev;
        struct osd_state *osd_device = disp_dev->osd_device;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "vpbe_display_release\n");

        mutex_lock(&layer->opslock);
        /* if this instance is doing IO */
        if (fh->io_allowed) {
                /* Reset io_usrs member of layer object */
                layer->io_usrs = 0;

                osd_device->ops.disable_layer(osd_device,
                                layer->layer_info.id);
                layer->started = 0;
                /* Free buffers allocated */
                vb2_queue_release(&layer->buffer_queue);
                vb2_dma_contig_cleanup_ctx(&layer->buffer_queue);
        }

        /* Decrement layer usrs counter */
        layer->usrs--;
        /* If this file handle has initialize encoder device, reset it */
        if (!layer->usrs) {
                if (cfg->pixfmt == PIXFMT_NV12) {
                        struct vpbe_layer *otherlayer;
                        otherlayer =
                        _vpbe_display_get_other_win_layer(disp_dev, layer);
                        osd_device->ops.disable_layer(osd_device,
                                        otherlayer->layer_info.id);
                        osd_device->ops.release_layer(osd_device,
                                        otherlayer->layer_info.id);
                }
                osd_device->ops.disable_layer(osd_device,
                                layer->layer_info.id);
                osd_device->ops.release_layer(osd_device,
                                layer->layer_info.id);
        }
        /* Close the priority */
        v4l2_prio_close(&layer->prio, fh->prio);
        file->private_data = NULL;
        mutex_unlock(&layer->opslock);

        /* Free memory allocated to file handle object */
        kfree(fh);

        disp_dev->cbcr_ofst = 0;

        return 0;
}

/* vpbe capture ioctl operations */
static const struct v4l2_ioctl_ops vpbe_ioctl_ops = {
        .vidioc_querycap         = vpbe_display_querycap,
        .vidioc_g_fmt_vid_out    = vpbe_display_g_fmt,
        .vidioc_enum_fmt_vid_out = vpbe_display_enum_fmt,
        .vidioc_s_fmt_vid_out    = vpbe_display_s_fmt,
        .vidioc_try_fmt_vid_out  = vpbe_display_try_fmt,
        .vidioc_reqbufs          = vpbe_display_reqbufs,
        .vidioc_querybuf         = vpbe_display_querybuf,
        .vidioc_qbuf             = vpbe_display_qbuf,
        .vidioc_dqbuf            = vpbe_display_dqbuf,
        .vidioc_streamon         = vpbe_display_streamon,
        .vidioc_streamoff        = vpbe_display_streamoff,
        .vidioc_cropcap          = vpbe_display_cropcap,
        .vidioc_g_crop           = vpbe_display_g_crop,
        .vidioc_s_crop           = vpbe_display_s_crop,
        .vidioc_g_priority       = vpbe_display_g_priority,
        .vidioc_s_priority       = vpbe_display_s_priority,
        .vidioc_s_std            = vpbe_display_s_std,
        .vidioc_g_std            = vpbe_display_g_std,
        .vidioc_enum_output      = vpbe_display_enum_output,
        .vidioc_s_output         = vpbe_display_s_output,
        .vidioc_g_output         = vpbe_display_g_output,
        .vidioc_s_dv_timings     = vpbe_display_s_dv_timings,
        .vidioc_g_dv_timings     = vpbe_display_g_dv_timings,
        .vidioc_enum_dv_timings  = vpbe_display_enum_dv_timings,
};

static struct v4l2_file_operations vpbe_fops = {
        .owner = THIS_MODULE,
        .open = vpbe_display_open,
        .release = vpbe_display_release,
        .unlocked_ioctl = video_ioctl2,
        .mmap = vpbe_display_mmap,
        .poll = vpbe_display_poll
};

static int vpbe_device_get(struct device *dev, void *data)
{
        struct platform_device *pdev = to_platform_device(dev);
        struct vpbe_display *vpbe_disp  = data;

        if (strcmp("vpbe_controller", pdev->name) == 0)
                vpbe_disp->vpbe_dev = platform_get_drvdata(pdev);

        if (strstr(pdev->name, "vpbe-osd") != NULL)
                vpbe_disp->osd_device = platform_get_drvdata(pdev);

        return 0;
}

static int init_vpbe_layer(int i, struct vpbe_display *disp_dev,
                           struct platform_device *pdev)
{
        struct vpbe_layer *vpbe_display_layer = NULL;
        struct video_device *vbd = NULL;

        /* Allocate memory for four plane display objects */

        disp_dev->dev[i] =
                kzalloc(sizeof(struct vpbe_layer), GFP_KERNEL);

        /* If memory allocation fails, return error */
        if (!disp_dev->dev[i]) {
                printk(KERN_ERR "ran out of memory\n");
                return  -ENOMEM;
        }
        spin_lock_init(&disp_dev->dev[i]->irqlock);
        mutex_init(&disp_dev->dev[i]->opslock);

        /* Get the pointer to the layer object */
        vpbe_display_layer = disp_dev->dev[i];
        vbd = &vpbe_display_layer->video_dev;
        /* Initialize field of video device */
        vbd->release    = video_device_release_empty;
        vbd->fops       = &vpbe_fops;
        vbd->ioctl_ops  = &vpbe_ioctl_ops;
        vbd->minor      = -1;
        vbd->v4l2_dev   = &disp_dev->vpbe_dev->v4l2_dev;
        vbd->lock       = &vpbe_display_layer->opslock;
        vbd->vfl_dir    = VFL_DIR_TX;

        if (disp_dev->vpbe_dev->current_timings.timings_type &
                        VPBE_ENC_STD)
                vbd->tvnorms = (V4L2_STD_525_60 | V4L2_STD_625_50);

        snprintf(vbd->name, sizeof(vbd->name),
                        "DaVinci_VPBE Display_DRIVER_V%d.%d.%d",
                        (VPBE_DISPLAY_VERSION_CODE >> 16) & 0xff,
                        (VPBE_DISPLAY_VERSION_CODE >> 8) & 0xff,
                        (VPBE_DISPLAY_VERSION_CODE) & 0xff);

        vpbe_display_layer->device_id = i;

        vpbe_display_layer->layer_info.id =
                ((i == VPBE_DISPLAY_DEVICE_0) ? WIN_VID0 : WIN_VID1);

        /* Initialize prio member of layer object */
        v4l2_prio_init(&vpbe_display_layer->prio);

        return 0;
}

static int register_device(struct vpbe_layer *vpbe_display_layer,
                           struct vpbe_display *disp_dev,
                           struct platform_device *pdev)
{
        int err;

        v4l2_info(&disp_dev->vpbe_dev->v4l2_dev,
                  "Trying to register VPBE display device.\n");
        v4l2_info(&disp_dev->vpbe_dev->v4l2_dev,
                  "layer=%x,layer->video_dev=%x\n",
                  (int)vpbe_display_layer,
                  (int)&vpbe_display_layer->video_dev);

        err = video_register_device(&vpbe_display_layer->video_dev,
                                    VFL_TYPE_GRABBER,
                                    -1);
        if (err)
                return -ENODEV;

        vpbe_display_layer->disp_dev = disp_dev;
        /* set the driver data in platform device */
        platform_set_drvdata(pdev, disp_dev);
        video_set_drvdata(&vpbe_display_layer->video_dev,
                          vpbe_display_layer);

        return 0;
}



/*
 * vpbe_display_probe()
 * This function creates device entries by register itself to the V4L2 driver
 * and initializes fields of each layer objects
 */
static int vpbe_display_probe(struct platform_device *pdev)
{
        struct vpbe_layer *vpbe_display_layer;
        struct vpbe_display *disp_dev;
        struct resource *res = NULL;
        int k;
        int i;
        int err;
        int irq;

        printk(KERN_DEBUG "vpbe_display_probe\n");
        /* Allocate memory for vpbe_display */
        disp_dev = devm_kzalloc(&pdev->dev, sizeof(struct vpbe_display),
                                GFP_KERNEL);
        if (!disp_dev)
                return -ENOMEM;

        spin_lock_init(&disp_dev->dma_queue_lock);
        /*
         * Scan all the platform devices to find the vpbe
         * controller device and get the vpbe_dev object
         */
        err = bus_for_each_dev(&platform_bus_type, NULL, disp_dev,
                        vpbe_device_get);
        if (err < 0)
                return err;
        /* Initialize the vpbe display controller */
        if (NULL != disp_dev->vpbe_dev->ops.initialize) {
                err = disp_dev->vpbe_dev->ops.initialize(&pdev->dev,
                                                         disp_dev->vpbe_dev);
                if (err) {
                        v4l2_err(&disp_dev->vpbe_dev->v4l2_dev,
                                        "Error initing vpbe\n");
                        err = -ENOMEM;
                        goto probe_out;
                }
        }

        for (i = 0; i < VPBE_DISPLAY_MAX_DEVICES; i++) {
                if (init_vpbe_layer(i, disp_dev, pdev)) {
                        err = -ENODEV;
                        goto probe_out;
                }
        }

        res = platform_get_resource(pdev, IORESOURCE_IRQ, 0);
        if (!res) {
                v4l2_err(&disp_dev->vpbe_dev->v4l2_dev,
                         "Unable to get VENC interrupt resource\n");
                err = -ENODEV;
                goto probe_out;
        }

        irq = res->start;
        err = devm_request_irq(&pdev->dev, irq, venc_isr, IRQF_DISABLED,
                               VPBE_DISPLAY_DRIVER, disp_dev);
        if (err) {
                v4l2_err(&disp_dev->vpbe_dev->v4l2_dev,
                                "Unable to request interrupt\n");
                goto probe_out;
        }

        for (i = 0; i < VPBE_DISPLAY_MAX_DEVICES; i++) {
                if (register_device(disp_dev->dev[i], disp_dev, pdev)) {
                        err = -ENODEV;
                        goto probe_out;
                }
        }

        printk(KERN_DEBUG "Successfully completed the probing of vpbe v4l2 device\n");
        return 0;

probe_out:
        for (k = 0; k < VPBE_DISPLAY_MAX_DEVICES; k++) {
                /* Get the pointer to the layer object */
                vpbe_display_layer = disp_dev->dev[k];
                /* Unregister video device */
                if (vpbe_display_layer) {
                        video_unregister_device(
                                &vpbe_display_layer->video_dev);
                                kfree(disp_dev->dev[k]);
                }
        }
        return err;
}

/*
 * vpbe_display_remove()
 * It un-register hardware layer from V4L2 driver
 */
static int vpbe_display_remove(struct platform_device *pdev)
{
        struct vpbe_layer *vpbe_display_layer;
        struct vpbe_display *disp_dev = platform_get_drvdata(pdev);
        struct vpbe_device *vpbe_dev = disp_dev->vpbe_dev;
        int i;

        v4l2_dbg(1, debug, &vpbe_dev->v4l2_dev, "vpbe_display_remove\n");

        /* deinitialize the vpbe display controller */
        if (NULL != vpbe_dev->ops.deinitialize)
                vpbe_dev->ops.deinitialize(&pdev->dev, vpbe_dev);
        /* un-register device */
        for (i = 0; i < VPBE_DISPLAY_MAX_DEVICES; i++) {
                /* Get the pointer to the layer object */
                vpbe_display_layer = disp_dev->dev[i];
                /* Unregister video device */
                video_unregister_device(&vpbe_display_layer->video_dev);

        }
        for (i = 0; i < VPBE_DISPLAY_MAX_DEVICES; i++) {
                kfree(disp_dev->dev[i]);
                disp_dev->dev[i] = NULL;
        }

        return 0;
}

static struct platform_driver vpbe_display_driver = {
        .driver = {
                .name = VPBE_DISPLAY_DRIVER,
                .owner = THIS_MODULE,
                .bus = &platform_bus_type,
        },
        .probe = vpbe_display_probe,
        .remove = vpbe_display_remove,
};

module_platform_driver(vpbe_display_driver);

MODULE_DESCRIPTION("TI DM644x/DM355/DM365 VPBE Display controller");
MODULE_LICENSE("GPL");
MODULE_AUTHOR("Texas Instruments");