/*
 * Samsung S5P/EXYNOS4 SoC series camera interface (camera capture) driver
 *
 * Copyright (C) 2010 - 2012 Samsung Electronics Co., Ltd.
 * Sylwester Nawrocki <s.nawrocki@samsung.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */

#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/types.h>
#include <linux/errno.h>
#include <linux/bug.h>
#include <linux/interrupt.h>
#include <linux/device.h>
#include <linux/pm_runtime.h>
#include <linux/list.h>
#include <linux/slab.h>

#include <linux/videodev2.h>
#include <media/v4l2-device.h>
#include <media/v4l2-ioctl.h>
#include <media/v4l2-mem2mem.h>
#include <media/videobuf2-core.h>
#include <media/videobuf2-dma-contig.h>

#include "media-dev.h"
#include "fimc-core.h"
#include "fimc-reg.h"

static int fimc_capture_hw_init(struct fimc_dev *fimc)
{
        struct fimc_source_info *si = &fimc->vid_cap.source_config;
        struct fimc_ctx *ctx = fimc->vid_cap.ctx;
        int ret;
        unsigned long flags;

        if (ctx == NULL || ctx->s_frame.fmt == NULL)
                return -EINVAL;

        if (si->fimc_bus_type == FIMC_BUS_TYPE_ISP_WRITEBACK) {
                ret = fimc_hw_camblk_cfg_writeback(fimc);
                if (ret < 0)
                        return ret;
        }

        spin_lock_irqsave(&fimc->slock, flags);
        fimc_prepare_dma_offset(ctx, &ctx->d_frame);
        fimc_set_yuv_order(ctx);

        fimc_hw_set_camera_polarity(fimc, si);
        fimc_hw_set_camera_type(fimc, si);
        fimc_hw_set_camera_source(fimc, si);
        fimc_hw_set_camera_offset(fimc, &ctx->s_frame);

        ret = fimc_set_scaler_info(ctx);
        if (!ret) {
                fimc_hw_set_input_path(ctx);
                fimc_hw_set_prescaler(ctx);
                fimc_hw_set_mainscaler(ctx);
                fimc_hw_set_target_format(ctx);
                fimc_hw_set_rotation(ctx);
                fimc_hw_set_effect(ctx);
                fimc_hw_set_output_path(ctx);
                fimc_hw_set_out_dma(ctx);
                if (fimc->drv_data->alpha_color)
                        fimc_hw_set_rgb_alpha(ctx);
                clear_bit(ST_CAPT_APPLY_CFG, &fimc->state);
        }
        spin_unlock_irqrestore(&fimc->slock, flags);
        return ret;
}

/*
 * Reinitialize the driver so it is ready to start the streaming again.
 * Set fimc->state to indicate stream off and the hardware shut down state.
 * If not suspending (@suspend is false), return any buffers to videobuf2.
 * Otherwise put any owned buffers onto the pending buffers queue, so they
 * can be re-spun when the device is being resumed. Also perform FIMC
 * software reset and disable streaming on the whole pipeline if required.
 */
static int fimc_capture_state_cleanup(struct fimc_dev *fimc, bool suspend)
{
        struct fimc_vid_cap *cap = &fimc->vid_cap;
        struct fimc_vid_buffer *buf;
        unsigned long flags;
        bool streaming;

        spin_lock_irqsave(&fimc->slock, flags);
        streaming = fimc->state & (1 << ST_CAPT_ISP_STREAM);

        fimc->state &= ~(1 << ST_CAPT_RUN | 1 << ST_CAPT_SHUT |
                         1 << ST_CAPT_STREAM | 1 << ST_CAPT_ISP_STREAM);
        if (suspend)
                fimc->state |= (1 << ST_CAPT_SUSPENDED);
        else
                fimc->state &= ~(1 << ST_CAPT_PEND | 1 << ST_CAPT_SUSPENDED);

        /* Release unused buffers */
        while (!suspend && !list_empty(&cap->pending_buf_q)) {
                buf = fimc_pending_queue_pop(cap);
                vb2_buffer_done(&buf->vb, VB2_BUF_STATE_ERROR);
        }
        /* If suspending put unused buffers onto pending queue */
        while (!list_empty(&cap->active_buf_q)) {
                buf = fimc_active_queue_pop(cap);
                if (suspend)
                        fimc_pending_queue_add(cap, buf);
                else
                        vb2_buffer_done(&buf->vb, VB2_BUF_STATE_ERROR);
        }

        fimc_hw_reset(fimc);
        cap->buf_index = 0;

        spin_unlock_irqrestore(&fimc->slock, flags);

        if (streaming)
                return fimc_pipeline_call(fimc, set_stream,
                                          &fimc->pipeline, 0);
        else
                return 0;
}

static int fimc_stop_capture(struct fimc_dev *fimc, bool suspend)
{
        unsigned long flags;

        if (!fimc_capture_active(fimc))
                return 0;

        spin_lock_irqsave(&fimc->slock, flags);
        set_bit(ST_CAPT_SHUT, &fimc->state);
        fimc_deactivate_capture(fimc);
        spin_unlock_irqrestore(&fimc->slock, flags);

        wait_event_timeout(fimc->irq_queue,
                           !test_bit(ST_CAPT_SHUT, &fimc->state),
                           (2*HZ/10)); /* 200 ms */

        return fimc_capture_state_cleanup(fimc, suspend);
}

/**
 * fimc_capture_config_update - apply the camera interface configuration
 *
 * To be called from within the interrupt handler with fimc.slock
 * spinlock held. It updates the camera pixel crop, rotation and
 * image flip in H/W.
 */
static int fimc_capture_config_update(struct fimc_ctx *ctx)
{
        struct fimc_dev *fimc = ctx->fimc_dev;
        int ret;

        fimc_hw_set_camera_offset(fimc, &ctx->s_frame);

        ret = fimc_set_scaler_info(ctx);
        if (ret)
                return ret;

        fimc_hw_set_prescaler(ctx);
        fimc_hw_set_mainscaler(ctx);
        fimc_hw_set_target_format(ctx);
        fimc_hw_set_rotation(ctx);
        fimc_hw_set_effect(ctx);
        fimc_prepare_dma_offset(ctx, &ctx->d_frame);
        fimc_hw_set_out_dma(ctx);
        if (fimc->drv_data->alpha_color)
                fimc_hw_set_rgb_alpha(ctx);

        clear_bit(ST_CAPT_APPLY_CFG, &fimc->state);
        return ret;
}

void fimc_capture_irq_handler(struct fimc_dev *fimc, int deq_buf)
{
        struct v4l2_subdev *csis = fimc->pipeline.subdevs[IDX_CSIS];
        struct fimc_vid_cap *cap = &fimc->vid_cap;
        struct fimc_frame *f = &cap->ctx->d_frame;
        struct fimc_vid_buffer *v_buf;
        struct timeval *tv;
        struct timespec ts;

        if (test_and_clear_bit(ST_CAPT_SHUT, &fimc->state)) {
                wake_up(&fimc->irq_queue);
                goto done;
        }

        if (!list_empty(&cap->active_buf_q) &&
            test_bit(ST_CAPT_RUN, &fimc->state) && deq_buf) {
                ktime_get_real_ts(&ts);

                v_buf = fimc_active_queue_pop(cap);

                tv = &v_buf->vb.v4l2_buf.timestamp;
                tv->tv_sec = ts.tv_sec;
                tv->tv_usec = ts.tv_nsec / NSEC_PER_USEC;
                v_buf->vb.v4l2_buf.sequence = cap->frame_count++;

                vb2_buffer_done(&v_buf->vb, VB2_BUF_STATE_DONE);
        }

        if (!list_empty(&cap->pending_buf_q)) {

                v_buf = fimc_pending_queue_pop(cap);
                fimc_hw_set_output_addr(fimc, &v_buf->paddr, cap->buf_index);
                v_buf->index = cap->buf_index;

                /* Move the buffer to the capture active queue */
                fimc_active_queue_add(cap, v_buf);

                dbg("next frame: %d, done frame: %d",
                    fimc_hw_get_frame_index(fimc), v_buf->index);

                if (++cap->buf_index >= FIMC_MAX_OUT_BUFS)
                        cap->buf_index = 0;
        }
        /*
         * Set up a buffer at MIPI-CSIS if current image format
         * requires the frame embedded data capture.
         */
        if (f->fmt->mdataplanes && !list_empty(&cap->active_buf_q)) {
                unsigned int plane = ffs(f->fmt->mdataplanes) - 1;
                unsigned int size = f->payload[plane];
                s32 index = fimc_hw_get_frame_index(fimc);
                void *vaddr;

                list_for_each_entry(v_buf, &cap->active_buf_q, list) {
                        if (v_buf->index != index)
                                continue;
                        vaddr = vb2_plane_vaddr(&v_buf->vb, plane);
                        v4l2_subdev_call(csis, video, s_rx_buffer,
                                         vaddr, &size);
                        break;
                }
        }

        if (cap->active_buf_cnt == 0) {
                if (deq_buf)
                        clear_bit(ST_CAPT_RUN, &fimc->state);

                if (++cap->buf_index >= FIMC_MAX_OUT_BUFS)
                        cap->buf_index = 0;
        } else {
                set_bit(ST_CAPT_RUN, &fimc->state);
        }

        if (test_bit(ST_CAPT_APPLY_CFG, &fimc->state))
                fimc_capture_config_update(cap->ctx);
done:
        if (cap->active_buf_cnt == 1) {
                fimc_deactivate_capture(fimc);
                clear_bit(ST_CAPT_STREAM, &fimc->state);
        }

        dbg("frame: %d, active_buf_cnt: %d",
            fimc_hw_get_frame_index(fimc), cap->active_buf_cnt);
}


static int start_streaming(struct vb2_queue *q, unsigned int count)
{
        struct fimc_ctx *ctx = q->drv_priv;
        struct fimc_dev *fimc = ctx->fimc_dev;
        struct fimc_vid_cap *vid_cap = &fimc->vid_cap;
        int min_bufs;
        int ret;

        vid_cap->frame_count = 0;

        ret = fimc_capture_hw_init(fimc);
        if (ret) {
                fimc_capture_state_cleanup(fimc, false);
                return ret;
        }

        set_bit(ST_CAPT_PEND, &fimc->state);

        min_bufs = fimc->vid_cap.reqbufs_count > 1 ? 2 : 1;

        if (vid_cap->active_buf_cnt >= min_bufs &&
            !test_and_set_bit(ST_CAPT_STREAM, &fimc->state)) {
                fimc_activate_capture(ctx);

                if (!test_and_set_bit(ST_CAPT_ISP_STREAM, &fimc->state))
                        return fimc_pipeline_call(fimc, set_stream,
                                                  &fimc->pipeline, 1);
        }

        return 0;
}

static int stop_streaming(struct vb2_queue *q)
{
        struct fimc_ctx *ctx = q->drv_priv;
        struct fimc_dev *fimc = ctx->fimc_dev;

        if (!fimc_capture_active(fimc))
                return -EINVAL;

        return fimc_stop_capture(fimc, false);
}

int fimc_capture_suspend(struct fimc_dev *fimc)
{
        bool suspend = fimc_capture_busy(fimc);

        int ret = fimc_stop_capture(fimc, suspend);
        if (ret)
                return ret;
        return fimc_pipeline_call(fimc, close, &fimc->pipeline);
}

static void buffer_queue(struct vb2_buffer *vb);

int fimc_capture_resume(struct fimc_dev *fimc)
{
        struct fimc_vid_cap *vid_cap = &fimc->vid_cap;
        struct fimc_vid_buffer *buf;
        int i;

        if (!test_and_clear_bit(ST_CAPT_SUSPENDED, &fimc->state))
                return 0;

        INIT_LIST_HEAD(&fimc->vid_cap.active_buf_q);
        vid_cap->buf_index = 0;
        fimc_pipeline_call(fimc, open, &fimc->pipeline,
                           &vid_cap->vfd.entity, false);
        fimc_capture_hw_init(fimc);

        clear_bit(ST_CAPT_SUSPENDED, &fimc->state);

        for (i = 0; i < vid_cap->reqbufs_count; i++) {
                if (list_empty(&vid_cap->pending_buf_q))
                        break;
                buf = fimc_pending_queue_pop(vid_cap);
                buffer_queue(&buf->vb);
        }
        return 0;

}

static int queue_setup(struct vb2_queue *vq, const struct v4l2_format *pfmt,
                       unsigned int *num_buffers, unsigned int *num_planes,
                       unsigned int sizes[], void *allocators[])
{
        const struct v4l2_pix_format_mplane *pixm = NULL;
        struct fimc_ctx *ctx = vq->drv_priv;
        struct fimc_frame *frame = &ctx->d_frame;
        struct fimc_fmt *fmt = frame->fmt;
        unsigned long wh;
        int i;

        if (pfmt) {
                pixm = &pfmt->fmt.pix_mp;
                fmt = fimc_find_format(&pixm->pixelformat, NULL,
                                       FMT_FLAGS_CAM | FMT_FLAGS_M2M, -1);
                wh = pixm->width * pixm->height;
        } else {
                wh = frame->f_width * frame->f_height;
        }

        if (fmt == NULL)
                return -EINVAL;

        *num_planes = fmt->memplanes;

        for (i = 0; i < fmt->memplanes; i++) {
                unsigned int size = (wh * fmt->depth[i]) / 8;
                if (pixm)
                        sizes[i] = max(size, pixm->plane_fmt[i].sizeimage);
                else if (fimc_fmt_is_user_defined(fmt->color))
                        sizes[i] = frame->payload[i];
                else
                        sizes[i] = max_t(u32, size, frame->payload[i]);

                allocators[i] = ctx->fimc_dev->alloc_ctx;
        }

        return 0;
}

static int buffer_prepare(struct vb2_buffer *vb)
{
        struct vb2_queue *vq = vb->vb2_queue;
        struct fimc_ctx *ctx = vq->drv_priv;
        int i;

        if (ctx->d_frame.fmt == NULL)
                return -EINVAL;

        for (i = 0; i < ctx->d_frame.fmt->memplanes; i++) {
                unsigned long size = ctx->d_frame.payload[i];

                if (vb2_plane_size(vb, i) < size) {
                        v4l2_err(&ctx->fimc_dev->vid_cap.vfd,
                                 "User buffer too small (%ld < %ld)\n",
                                 vb2_plane_size(vb, i), size);
                        return -EINVAL;
                }
                vb2_set_plane_payload(vb, i, size);
        }

        return 0;
}

static void buffer_queue(struct vb2_buffer *vb)
{
        struct fimc_vid_buffer *buf
                = container_of(vb, struct fimc_vid_buffer, vb);
        struct fimc_ctx *ctx = vb2_get_drv_priv(vb->vb2_queue);
        struct fimc_dev *fimc = ctx->fimc_dev;
        struct fimc_vid_cap *vid_cap = &fimc->vid_cap;
        unsigned long flags;
        int min_bufs;

        spin_lock_irqsave(&fimc->slock, flags);
        fimc_prepare_addr(ctx, &buf->vb, &ctx->d_frame, &buf->paddr);

        if (!test_bit(ST_CAPT_SUSPENDED, &fimc->state) &&
            !test_bit(ST_CAPT_STREAM, &fimc->state) &&
            vid_cap->active_buf_cnt < FIMC_MAX_OUT_BUFS) {
                /* Setup the buffer directly for processing. */
                int buf_id = (vid_cap->reqbufs_count == 1) ? -1 :
                                vid_cap->buf_index;

                fimc_hw_set_output_addr(fimc, &buf->paddr, buf_id);
                buf->index = vid_cap->buf_index;
                fimc_active_queue_add(vid_cap, buf);

                if (++vid_cap->buf_index >= FIMC_MAX_OUT_BUFS)
                        vid_cap->buf_index = 0;
        } else {
                fimc_pending_queue_add(vid_cap, buf);
        }

        min_bufs = vid_cap->reqbufs_count > 1 ? 2 : 1;


        if (vb2_is_streaming(&vid_cap->vbq) &&
            vid_cap->active_buf_cnt >= min_bufs &&
            !test_and_set_bit(ST_CAPT_STREAM, &fimc->state)) {
                int ret;

                fimc_activate_capture(ctx);
                spin_unlock_irqrestore(&fimc->slock, flags);

                if (test_and_set_bit(ST_CAPT_ISP_STREAM, &fimc->state))
                        return;

                ret = fimc_pipeline_call(fimc, set_stream, &fimc->pipeline, 1);
                if (ret < 0)
                        v4l2_err(&vid_cap->vfd, "stream on failed: %d\n", ret);
                return;
        }
        spin_unlock_irqrestore(&fimc->slock, flags);
}

static struct vb2_ops fimc_capture_qops = {
        .queue_setup            = queue_setup,
        .buf_prepare            = buffer_prepare,
        .buf_queue              = buffer_queue,
        .wait_prepare           = vb2_ops_wait_prepare,
        .wait_finish            = vb2_ops_wait_finish,
        .start_streaming        = start_streaming,
        .stop_streaming         = stop_streaming,
};

/**
 * fimc_capture_ctrls_create - initialize the control handler
 * Initialize the capture video node control handler and fill it
 * with the FIMC controls. Inherit any sensor's controls if the
 * 'user_subdev_api' flag is false (default behaviour).
 * This function need to be called with the graph mutex held.
 */
int fimc_capture_ctrls_create(struct fimc_dev *fimc)
{
        struct fimc_vid_cap *vid_cap = &fimc->vid_cap;
        struct v4l2_subdev *sensor = fimc->pipeline.subdevs[IDX_SENSOR];
        int ret;

        if (WARN_ON(vid_cap->ctx == NULL))
                return -ENXIO;
        if (vid_cap->ctx->ctrls.ready)
                return 0;

        ret = fimc_ctrls_create(vid_cap->ctx);

        if (ret || vid_cap->user_subdev_api || !sensor ||
            !vid_cap->ctx->ctrls.ready)
                return ret;

        return v4l2_ctrl_add_handler(&vid_cap->ctx->ctrls.handler,
                                     sensor->ctrl_handler, NULL);
}

static int fimc_capture_set_default_format(struct fimc_dev *fimc);

static int fimc_capture_open(struct file *file)
{
        struct fimc_dev *fimc = video_drvdata(file);
        int ret = -EBUSY;

        dbg("pid: %d, state: 0x%lx", task_pid_nr(current), fimc->state);

        fimc_md_graph_lock(fimc);
        mutex_lock(&fimc->lock);

        if (fimc_m2m_active(fimc))
                goto unlock;

        set_bit(ST_CAPT_BUSY, &fimc->state);
        ret = pm_runtime_get_sync(&fimc->pdev->dev);
        if (ret < 0)
                goto unlock;

        ret = v4l2_fh_open(file);
        if (ret) {
                pm_runtime_put(&fimc->pdev->dev);
                goto unlock;
        }

        if (v4l2_fh_is_singular_file(file)) {
                ret = fimc_pipeline_call(fimc, open, &fimc->pipeline,
                                         &fimc->vid_cap.vfd.entity, true);

                if (!ret && !fimc->vid_cap.user_subdev_api)
                        ret = fimc_capture_set_default_format(fimc);

                if (!ret)
                        ret = fimc_capture_ctrls_create(fimc);

                if (ret < 0) {
                        clear_bit(ST_CAPT_BUSY, &fimc->state);
                        pm_runtime_put_sync(&fimc->pdev->dev);
                        v4l2_fh_release(file);
                } else {
                        fimc->vid_cap.refcnt++;
                }
        }
unlock:
        mutex_unlock(&fimc->lock);
        fimc_md_graph_unlock(fimc);
        return ret;
}

static int fimc_capture_release(struct file *file)
{
        struct fimc_dev *fimc = video_drvdata(file);
        struct fimc_vid_cap *vc = &fimc->vid_cap;
        int ret;

        dbg("pid: %d, state: 0x%lx", task_pid_nr(current), fimc->state);

        mutex_lock(&fimc->lock);

        if (v4l2_fh_is_singular_file(file)) {
                if (vc->streaming) {
                        media_entity_pipeline_stop(&vc->vfd.entity);
                        vc->streaming = false;
                }
                clear_bit(ST_CAPT_BUSY, &fimc->state);
                fimc_stop_capture(fimc, false);
                fimc_pipeline_call(fimc, close, &fimc->pipeline);
                clear_bit(ST_CAPT_SUSPENDED, &fimc->state);
                fimc->vid_cap.refcnt--;
        }

        pm_runtime_put(&fimc->pdev->dev);

        if (v4l2_fh_is_singular_file(file))
                fimc_ctrls_delete(fimc->vid_cap.ctx);

        ret = vb2_fop_release(file);
        mutex_unlock(&fimc->lock);

        return ret;
}

static const struct v4l2_file_operations fimc_capture_fops = {
        .owner          = THIS_MODULE,
        .open           = fimc_capture_open,
        .release        = fimc_capture_release,
        .poll           = vb2_fop_poll,
        .unlocked_ioctl = video_ioctl2,
        .mmap           = vb2_fop_mmap,
};

/*
 * Format and crop negotiation helpers
 */

static struct fimc_fmt *fimc_capture_try_format(struct fimc_ctx *ctx,
                                                u32 *width, u32 *height,
                                                u32 *code, u32 *fourcc, int pad)
{
        bool rotation = ctx->rotation == 90 || ctx->rotation == 270;
        struct fimc_dev *fimc = ctx->fimc_dev;
        const struct fimc_variant *var = fimc->variant;
        const struct fimc_pix_limit *pl = var->pix_limit;
        struct fimc_frame *dst = &ctx->d_frame;
        u32 depth, min_w, max_w, min_h, align_h = 3;
        u32 mask = FMT_FLAGS_CAM;
        struct fimc_fmt *ffmt;

        /* Conversion from/to JPEG or User Defined format is not supported */
        if (code && ctx->s_frame.fmt && pad == FIMC_SD_PAD_SOURCE &&
            fimc_fmt_is_user_defined(ctx->s_frame.fmt->color))
                *code = ctx->s_frame.fmt->mbus_code;

        if (fourcc && *fourcc != V4L2_PIX_FMT_JPEG && pad == FIMC_SD_PAD_SOURCE)
                mask |= FMT_FLAGS_M2M;

        if (pad == FIMC_SD_PAD_SINK_FIFO)
                mask = FMT_FLAGS_WRITEBACK;

        ffmt = fimc_find_format(fourcc, code, mask, 0);
        if (WARN_ON(!ffmt))
                return NULL;

        if (code)
                *code = ffmt->mbus_code;
        if (fourcc)
                *fourcc = ffmt->fourcc;

        if (pad != FIMC_SD_PAD_SOURCE) {
                max_w = fimc_fmt_is_user_defined(ffmt->color) ?
                        pl->scaler_dis_w : pl->scaler_en_w;
                /* Apply the camera input interface pixel constraints */
                v4l_bound_align_image(width, max_t(u32, *width, 32), max_w, 4,
                                      height, max_t(u32, *height, 32),
                                      FIMC_CAMIF_MAX_HEIGHT,
                                      fimc_fmt_is_user_defined(ffmt->color) ?
                                      3 : 1,
                                      0);
                return ffmt;
        }
        /* Can't scale or crop in transparent (JPEG) transfer mode */
        if (fimc_fmt_is_user_defined(ffmt->color)) {
                *width  = ctx->s_frame.f_width;
                *height = ctx->s_frame.f_height;
                return ffmt;
        }
        /* Apply the scaler and the output DMA constraints */
        max_w = rotation ? pl->out_rot_en_w : pl->out_rot_dis_w;
        if (ctx->state & FIMC_COMPOSE) {
                min_w = dst->offs_h + dst->width;
                min_h = dst->offs_v + dst->height;
        } else {
                min_w = var->min_out_pixsize;
                min_h = var->min_out_pixsize;
        }
        if (var->min_vsize_align == 1 && !rotation)
                align_h = fimc_fmt_is_rgb(ffmt->color) ? 0 : 1;

        depth = fimc_get_format_depth(ffmt);
        v4l_bound_align_image(width, min_w, max_w,
                              ffs(var->min_out_pixsize) - 1,
                              height, min_h, FIMC_CAMIF_MAX_HEIGHT,
                              align_h,
                              64/(ALIGN(depth, 8)));

        dbg("pad%d: code: 0x%x, %dx%d. dst fmt: %dx%d",
            pad, code ? *code : 0, *width, *height,
            dst->f_width, dst->f_height);

        return ffmt;
}

static void fimc_capture_try_selection(struct fimc_ctx *ctx,
                                       struct v4l2_rect *r,
                                       int target)
{
        bool rotate = ctx->rotation == 90 || ctx->rotation == 270;
        struct fimc_dev *fimc = ctx->fimc_dev;
        const struct fimc_variant *var = fimc->variant;
        const struct fimc_pix_limit *pl = var->pix_limit;
        struct fimc_frame *sink = &ctx->s_frame;
        u32 max_w, max_h, min_w = 0, min_h = 0, min_sz;
        u32 align_sz = 0, align_h = 4;
        u32 max_sc_h, max_sc_v;

        /* In JPEG transparent transfer mode cropping is not supported */
        if (fimc_fmt_is_user_defined(ctx->d_frame.fmt->color)) {
                r->width  = sink->f_width;
                r->height = sink->f_height;
                r->left   = r->top = 0;
                return;
        }
        if (target == V4L2_SEL_TGT_COMPOSE) {
                if (ctx->rotation != 90 && ctx->rotation != 270)
                        align_h = 1;
                max_sc_h = min(SCALER_MAX_HRATIO, 1 << (ffs(sink->width) - 3));
                max_sc_v = min(SCALER_MAX_VRATIO, 1 << (ffs(sink->height) - 1));
                min_sz = var->min_out_pixsize;
        } else {
                u32 depth = fimc_get_format_depth(sink->fmt);
                align_sz = 64/ALIGN(depth, 8);
                min_sz = var->min_inp_pixsize;
                min_w = min_h = min_sz;
                max_sc_h = max_sc_v = 1;
        }
        /*
         * For the compose rectangle the following constraints must be met:
         * - it must fit in the sink pad format rectangle (f_width/f_height);
         * - maximum downscaling ratio is 64;
         * - maximum crop size depends if the rotator is used or not;
         * - the sink pad format width/height must be 4 multiple of the
         *   prescaler ratios determined by sink pad size and source pad crop,
         *   the prescaler ratio is returned by fimc_get_scaler_factor().
         */
        max_w = min_t(u32,
                      rotate ? pl->out_rot_en_w : pl->out_rot_dis_w,
                      rotate ? sink->f_height : sink->f_width);
        max_h = min_t(u32, FIMC_CAMIF_MAX_HEIGHT, sink->f_height);

        if (target == V4L2_SEL_TGT_COMPOSE) {
                min_w = min_t(u32, max_w, sink->f_width / max_sc_h);
                min_h = min_t(u32, max_h, sink->f_height / max_sc_v);
                if (rotate) {
                        swap(max_sc_h, max_sc_v);
                        swap(min_w, min_h);
                }
        }
        v4l_bound_align_image(&r->width, min_w, max_w, ffs(min_sz) - 1,
                              &r->height, min_h, max_h, align_h,
                              align_sz);
        /* Adjust left/top if crop/compose rectangle is out of bounds */
        r->left = clamp_t(u32, r->left, 0, sink->f_width - r->width);
        r->top  = clamp_t(u32, r->top, 0, sink->f_height - r->height);
        r->left = round_down(r->left, var->hor_offs_align);

        dbg("target %#x: (%d,%d)/%dx%d, sink fmt: %dx%d",
            target, r->left, r->top, r->width, r->height,
            sink->f_width, sink->f_height);
}

/*
 * The video node ioctl operations
 */
static int fimc_cap_querycap(struct file *file, void *priv,
                                        struct v4l2_capability *cap)
{
        struct fimc_dev *fimc = video_drvdata(file);

        __fimc_vidioc_querycap(&fimc->pdev->dev, cap, V4L2_CAP_STREAMING |
                                        V4L2_CAP_VIDEO_CAPTURE_MPLANE);
        return 0;
}

static int fimc_cap_enum_fmt_mplane(struct file *file, void *priv,
                                    struct v4l2_fmtdesc *f)
{
        struct fimc_fmt *fmt;

        fmt = fimc_find_format(NULL, NULL, FMT_FLAGS_CAM | FMT_FLAGS_M2M,
                               f->index);
        if (!fmt)
                return -EINVAL;
        strncpy(f->description, fmt->name, sizeof(f->description) - 1);
        f->pixelformat = fmt->fourcc;
        if (fmt->fourcc == V4L2_MBUS_FMT_JPEG_1X8)
                f->flags |= V4L2_FMT_FLAG_COMPRESSED;
        return 0;
}

static struct media_entity *fimc_pipeline_get_head(struct media_entity *me)
{
        struct media_pad *pad = &me->pads[0];

        while (!(pad->flags & MEDIA_PAD_FL_SOURCE)) {
                pad = media_entity_remote_source(pad);
                if (!pad)
                        break;
                me = pad->entity;
                pad = &me->pads[0];
        }

        return me;
}

/**
 * fimc_pipeline_try_format - negotiate and/or set formats at pipeline
 *                            elements
 * @ctx: FIMC capture context
 * @tfmt: media bus format to try/set on subdevs
 * @fmt_id: fimc pixel format id corresponding to returned @tfmt (output)
 * @set: true to set format on subdevs, false to try only
 */
static int fimc_pipeline_try_format(struct fimc_ctx *ctx,
                                    struct v4l2_mbus_framefmt *tfmt,
                                    struct fimc_fmt **fmt_id,
                                    bool set)
{
        struct fimc_dev *fimc = ctx->fimc_dev;
        struct v4l2_subdev *sd = fimc->pipeline.subdevs[IDX_SENSOR];
        struct v4l2_subdev_format sfmt;
        struct v4l2_mbus_framefmt *mf = &sfmt.format;
        struct media_entity *me;
        struct fimc_fmt *ffmt;
        struct media_pad *pad;
        int ret, i = 1;
        u32 fcc;

        if (WARN_ON(!sd || !tfmt))
                return -EINVAL;

        memset(&sfmt, 0, sizeof(sfmt));
        sfmt.format = *tfmt;
        sfmt.which = set ? V4L2_SUBDEV_FORMAT_ACTIVE : V4L2_SUBDEV_FORMAT_TRY;

        me = fimc_pipeline_get_head(&sd->entity);

        while (1) {
                ffmt = fimc_find_format(NULL, mf->code != 0 ? &mf->code : NULL,
                                        FMT_FLAGS_CAM, i++);
                if (ffmt == NULL) {
                        /*
                         * Notify user-space if common pixel code for
                         * host and sensor does not exist.
                         */
                        return -EINVAL;
                }
                mf->code = tfmt->code = ffmt->mbus_code;

                /* set format on all pipeline subdevs */
                while (me != &fimc->vid_cap.subdev.entity) {
                        sd = media_entity_to_v4l2_subdev(me);

                        sfmt.pad = 0;
                        ret = v4l2_subdev_call(sd, pad, set_fmt, NULL, &sfmt);
                        if (ret)
                                return ret;

                        if (me->pads[0].flags & MEDIA_PAD_FL_SINK) {
                                sfmt.pad = me->num_pads - 1;
                                mf->code = tfmt->code;
                                ret = v4l2_subdev_call(sd, pad, set_fmt, NULL,
                                                                        &sfmt);
                                if (ret)
                                        return ret;
                        }

                        pad = media_entity_remote_source(&me->pads[sfmt.pad]);
                        if (!pad)
                                return -EINVAL;
                        me = pad->entity;
                }

                if (mf->code != tfmt->code)
                        continue;

                fcc = ffmt->fourcc;
                tfmt->width  = mf->width;
                tfmt->height = mf->height;
                ffmt = fimc_capture_try_format(ctx, &tfmt->width, &tfmt->height,
                                        NULL, &fcc, FIMC_SD_PAD_SINK_CAM);
                ffmt = fimc_capture_try_format(ctx, &tfmt->width, &tfmt->height,
                                        NULL, &fcc, FIMC_SD_PAD_SOURCE);
                if (ffmt && ffmt->mbus_code)
                        mf->code = ffmt->mbus_code;
                if (mf->width != tfmt->width || mf->height != tfmt->height)
                        continue;
                tfmt->code = mf->code;
                break;
        }

        if (fmt_id && ffmt)
                *fmt_id = ffmt;
        *tfmt = *mf;

        return 0;
}

/**
 * fimc_get_sensor_frame_desc - query the sensor for media bus frame parameters
 * @sensor: pointer to the sensor subdev
 * @plane_fmt: provides plane sizes corresponding to the frame layout entries
 * @try: true to set the frame parameters, false to query only
 *
 * This function is used by this driver only for compressed/blob data formats.
 */
static int fimc_get_sensor_frame_desc(struct v4l2_subdev *sensor,
                                      struct v4l2_plane_pix_format *plane_fmt,
                                      unsigned int num_planes, bool try)
{
        struct v4l2_mbus_frame_desc fd;
        int i, ret;
        int pad;

        for (i = 0; i < num_planes; i++)
                fd.entry[i].length = plane_fmt[i].sizeimage;

        pad = sensor->entity.num_pads - 1;
        if (try)
                ret = v4l2_subdev_call(sensor, pad, set_frame_desc, pad, &fd);
        else
                ret = v4l2_subdev_call(sensor, pad, get_frame_desc, pad, &fd);

        if (ret < 0)
                return ret;

        if (num_planes != fd.num_entries)
                return -EINVAL;

        for (i = 0; i < num_planes; i++)
                plane_fmt[i].sizeimage = fd.entry[i].length;

        if (fd.entry[0].length > FIMC_MAX_JPEG_BUF_SIZE) {
                v4l2_err(sensor->v4l2_dev,  "Unsupported buffer size: %u\n",
                         fd.entry[0].length);

                return -EINVAL;
        }

        return 0;
}

static int fimc_cap_g_fmt_mplane(struct file *file, void *fh,
                                 struct v4l2_format *f)
{
        struct fimc_dev *fimc = video_drvdata(file);

        __fimc_get_format(&fimc->vid_cap.ctx->d_frame, f);
        return 0;
}

static int fimc_cap_try_fmt_mplane(struct file *file, void *fh,
                                   struct v4l2_format *f)
{
        struct v4l2_pix_format_mplane *pix = &f->fmt.pix_mp;
        struct fimc_dev *fimc = video_drvdata(file);
        struct fimc_ctx *ctx = fimc->vid_cap.ctx;
        struct v4l2_mbus_framefmt mf;
        struct fimc_fmt *ffmt = NULL;
        int ret = 0;

        fimc_md_graph_lock(fimc);
        mutex_lock(&fimc->lock);

        if (fimc_jpeg_fourcc(pix->pixelformat)) {
                fimc_capture_try_format(ctx, &pix->width, &pix->height,
                                        NULL, &pix->pixelformat,
                                        FIMC_SD_PAD_SINK_CAM);
                ctx->s_frame.f_width  = pix->width;
                ctx->s_frame.f_height = pix->height;
        }
        ffmt = fimc_capture_try_format(ctx, &pix->width, &pix->height,
                                       NULL, &pix->pixelformat,
                                       FIMC_SD_PAD_SOURCE);
        if (!ffmt) {
                ret = -EINVAL;
                goto unlock;
        }

        if (!fimc->vid_cap.user_subdev_api) {
                mf.width = pix->width;
                mf.height = pix->height;
                mf.code = ffmt->mbus_code;
                fimc_pipeline_try_format(ctx, &mf, &ffmt, false);
                pix->width = mf.width;
                pix->height = mf.height;
                if (ffmt)
                        pix->pixelformat = ffmt->fourcc;
        }

        fimc_adjust_mplane_format(ffmt, pix->width, pix->height, pix);

        if (ffmt->flags & FMT_FLAGS_COMPRESSED)
                fimc_get_sensor_frame_desc(fimc->pipeline.subdevs[IDX_SENSOR],
                                        pix->plane_fmt, ffmt->memplanes, true);
unlock:
        mutex_unlock(&fimc->lock);
        fimc_md_graph_unlock(fimc);

        return ret;
}

static void fimc_capture_mark_jpeg_xfer(struct fimc_ctx *ctx,
                                        enum fimc_color_fmt color)
{
        bool jpeg = fimc_fmt_is_user_defined(color);

        ctx->scaler.enabled = !jpeg;
        fimc_ctrls_activate(ctx, !jpeg);

        if (jpeg)
                set_bit(ST_CAPT_JPEG, &ctx->fimc_dev->state);
        else
                clear_bit(ST_CAPT_JPEG, &ctx->fimc_dev->state);
}

static int __fimc_capture_set_format(struct fimc_dev *fimc,
                                     struct v4l2_format *f)
{
        struct fimc_ctx *ctx = fimc->vid_cap.ctx;

        struct v4l2_pix_format_mplane *pix = &f->fmt.pix_mp;
        struct v4l2_mbus_framefmt *mf = &fimc->vid_cap.ci_fmt;
        struct fimc_frame *ff = &ctx->d_frame;
        struct fimc_fmt *s_fmt = NULL;
        int ret, i;

        if (vb2_is_busy(&fimc->vid_cap.vbq))
                return -EBUSY;

        /* Pre-configure format at camera interface input, for JPEG only */
        if (fimc_jpeg_fourcc(pix->pixelformat)) {
                fimc_capture_try_format(ctx, &pix->width, &pix->height,
                                        NULL, &pix->pixelformat,
                                        FIMC_SD_PAD_SINK_CAM);
                ctx->s_frame.f_width  = pix->width;
                ctx->s_frame.f_height = pix->height;
        }
        /* Try the format at the scaler and the DMA output */
        ff->fmt = fimc_capture_try_format(ctx, &pix->width, &pix->height,
                                          NULL, &pix->pixelformat,
                                          FIMC_SD_PAD_SOURCE);
        if (!ff->fmt)
                return -EINVAL;

        /* Update RGB Alpha control state and value range */
        fimc_alpha_ctrl_update(ctx);

        /* Try to match format at the host and the sensor */
        if (!fimc->vid_cap.user_subdev_api) {
                mf->code   = ff->fmt->mbus_code;
                mf->width  = pix->width;
                mf->height = pix->height;
                ret = fimc_pipeline_try_format(ctx, mf, &s_fmt, true);
                if (ret)
                        return ret;

                pix->width  = mf->width;
                pix->height = mf->height;
        }

        fimc_adjust_mplane_format(ff->fmt, pix->width, pix->height, pix);

        if (ff->fmt->flags & FMT_FLAGS_COMPRESSED) {
                ret = fimc_get_sensor_frame_desc(fimc->pipeline.subdevs[IDX_SENSOR],
                                        pix->plane_fmt, ff->fmt->memplanes,
                                        true);
                if (ret < 0)
                        return ret;
        }

        for (i = 0; i < ff->fmt->memplanes; i++) {
                ff->bytesperline[i] = pix->plane_fmt[i].bytesperline;
                ff->payload[i] = pix->plane_fmt[i].sizeimage;
        }

        set_frame_bounds(ff, pix->width, pix->height);
        /* Reset the composition rectangle if not yet configured */
        if (!(ctx->state & FIMC_COMPOSE))
                set_frame_crop(ff, 0, 0, pix->width, pix->height);

        fimc_capture_mark_jpeg_xfer(ctx, ff->fmt->color);

        /* Reset cropping and set format at the camera interface input */
        if (!fimc->vid_cap.user_subdev_api) {
                ctx->s_frame.fmt = s_fmt;
                set_frame_bounds(&ctx->s_frame, pix->width, pix->height);
                set_frame_crop(&ctx->s_frame, 0, 0, pix->width, pix->height);
        }

        return ret;
}

static int fimc_cap_s_fmt_mplane(struct file *file, void *priv,
                                 struct v4l2_format *f)
{
        struct fimc_dev *fimc = video_drvdata(file);
        int ret;

        fimc_md_graph_lock(fimc);
        mutex_lock(&fimc->lock);
        /*
         * The graph is walked within __fimc_capture_set_format() to set
         * the format at subdevs thus the graph mutex needs to be held at
         * this point and acquired before the video mutex, to avoid  AB-BA
         * deadlock when fimc_md_link_notify() is called by other thread.
         * Ideally the graph walking and setting format at the whole pipeline
         * should be removed from this driver and handled in userspace only.
         */
        ret = __fimc_capture_set_format(fimc, f);

        mutex_unlock(&fimc->lock);
        fimc_md_graph_unlock(fimc);
        return ret;
}

static int fimc_cap_enum_input(struct file *file, void *priv,
                               struct v4l2_input *i)
{
        struct fimc_dev *fimc = video_drvdata(file);
        struct v4l2_subdev *sd = fimc->pipeline.subdevs[IDX_SENSOR];

        if (i->index != 0)
                return -EINVAL;

        i->type = V4L2_INPUT_TYPE_CAMERA;
        if (sd)
                strlcpy(i->name, sd->name, sizeof(i->name));
        return 0;
}

static int fimc_cap_s_input(struct file *file, void *priv, unsigned int i)
{
        return i == 0 ? i : -EINVAL;
}

static int fimc_cap_g_input(struct file *file, void *priv, unsigned int *i)
{
        *i = 0;
        return 0;
}

/**
 * fimc_pipeline_validate - check for formats inconsistencies
 *                          between source and sink pad of each link
 *
 * Return 0 if all formats match or -EPIPE otherwise.
 */
static int fimc_pipeline_validate(struct fimc_dev *fimc)
{
        struct v4l2_subdev_format sink_fmt, src_fmt;
        struct fimc_vid_cap *vc = &fimc->vid_cap;
        struct v4l2_subdev *sd = &vc->subdev;
        struct media_pad *sink_pad, *src_pad;
        int i, ret;

        while (1) {
                /*
                 * Find current entity sink pad and any remote sink pad linked
                 * to it. We stop if there is no sink pad in current entity or
                 * it is not linked to any other remote entity.
                 */
                src_pad = NULL;

                for (i = 0; i < sd->entity.num_pads; i++) {
                        struct media_pad *p = &sd->entity.pads[i];

                        if (p->flags & MEDIA_PAD_FL_SINK) {
                                sink_pad = p;
                                src_pad = media_entity_remote_source(sink_pad);
                                if (src_pad)
                                        break;
                        }
                }

                if (src_pad == NULL ||
                    media_entity_type(src_pad->entity) != MEDIA_ENT_T_V4L2_SUBDEV)
                        break;

                /* Don't call FIMC subdev operation to avoid nested locking */
                if (sd == &vc->subdev) {
                        struct fimc_frame *ff = &vc->ctx->s_frame;
                        sink_fmt.format.width = ff->f_width;
                        sink_fmt.format.height = ff->f_height;
                        sink_fmt.format.code = ff->fmt ? ff->fmt->mbus_code : 0;
                } else {
                        sink_fmt.pad = sink_pad->index;
                        sink_fmt.which = V4L2_SUBDEV_FORMAT_ACTIVE;
                        ret = v4l2_subdev_call(sd, pad, get_fmt, NULL, &sink_fmt);
                        if (ret < 0 && ret != -ENOIOCTLCMD)
                                return -EPIPE;
                }

                /* Retrieve format at the source pad */
                sd = media_entity_to_v4l2_subdev(src_pad->entity);
                src_fmt.pad = src_pad->index;
                src_fmt.which = V4L2_SUBDEV_FORMAT_ACTIVE;
                ret = v4l2_subdev_call(sd, pad, get_fmt, NULL, &src_fmt);
                if (ret < 0 && ret != -ENOIOCTLCMD)
                        return -EPIPE;

                if (src_fmt.format.width != sink_fmt.format.width ||
                    src_fmt.format.height != sink_fmt.format.height ||
                    src_fmt.format.code != sink_fmt.format.code)
                        return -EPIPE;

                if (sd == fimc->pipeline.subdevs[IDX_SENSOR] &&
                    fimc_user_defined_mbus_fmt(src_fmt.format.code)) {
                        struct v4l2_plane_pix_format plane_fmt[FIMC_MAX_PLANES];
                        struct fimc_frame *frame = &vc->ctx->d_frame;
                        unsigned int i;

                        ret = fimc_get_sensor_frame_desc(sd, plane_fmt,
                                                         frame->fmt->memplanes,
                                                         false);
                        if (ret < 0)
                                return -EPIPE;

                        for (i = 0; i < frame->fmt->memplanes; i++)
                                if (frame->payload[i] < plane_fmt[i].sizeimage)
                                        return -EPIPE;
                }
        }
        return 0;
}

static int fimc_cap_streamon(struct file *file, void *priv,
                             enum v4l2_buf_type type)
{
        struct fimc_dev *fimc = video_drvdata(file);
        struct fimc_pipeline *p = &fimc->pipeline;
        struct fimc_vid_cap *vc = &fimc->vid_cap;
        struct media_entity *entity = &vc->vfd.entity;
        struct fimc_source_info *si = NULL;
        struct v4l2_subdev *sd;
        int ret;

        if (fimc_capture_active(fimc))
                return -EBUSY;

        ret = media_entity_pipeline_start(entity, p->m_pipeline);
        if (ret < 0)
                return ret;

        sd = p->subdevs[IDX_SENSOR];
        if (sd)
                si = v4l2_get_subdev_hostdata(sd);

        if (si == NULL) {
                ret = -EPIPE;
                goto err_p_stop;
        }
        /*
         * Save configuration data related to currently attached image
         * sensor or other data source, e.g. FIMC-IS.
         */
        vc->source_config = *si;

        if (vc->input == GRP_ID_FIMC_IS)
                vc->source_config.fimc_bus_type = FIMC_BUS_TYPE_ISP_WRITEBACK;

        if (vc->user_subdev_api) {
                ret = fimc_pipeline_validate(fimc);
                if (ret < 0)
                        goto err_p_stop;
        }

        ret = vb2_ioctl_streamon(file, priv, type);
        if (!ret) {
                vc->streaming = true;
                return ret;
        }

err_p_stop:
        media_entity_pipeline_stop(entity);
        return ret;
}

static int fimc_cap_streamoff(struct file *file, void *priv,
                            enum v4l2_buf_type type)
{
        struct fimc_dev *fimc = video_drvdata(file);
        int ret;

        ret = vb2_ioctl_streamoff(file, priv, type);
        if (ret < 0)
                return ret;

        media_entity_pipeline_stop(&fimc->vid_cap.vfd.entity);
        fimc->vid_cap.streaming = false;
        return 0;
}

static int fimc_cap_reqbufs(struct file *file, void *priv,
                            struct v4l2_requestbuffers *reqbufs)
{
        struct fimc_dev *fimc = video_drvdata(file);
        int ret;

        ret = vb2_ioctl_reqbufs(file, priv, reqbufs);

        if (!ret)
                fimc->vid_cap.reqbufs_count = reqbufs->count;

        return ret;
}

static int fimc_cap_g_selection(struct file *file, void *fh,
                                struct v4l2_selection *s)
{
        struct fimc_dev *fimc = video_drvdata(file);
        struct fimc_ctx *ctx = fimc->vid_cap.ctx;
        struct fimc_frame *f = &ctx->s_frame;

        if (s->type != V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE)
                return -EINVAL;

        switch (s->target) {
        case V4L2_SEL_TGT_COMPOSE_DEFAULT:
        case V4L2_SEL_TGT_COMPOSE_BOUNDS:
                f = &ctx->d_frame;
        case V4L2_SEL_TGT_CROP_BOUNDS:
        case V4L2_SEL_TGT_CROP_DEFAULT:
                s->r.left = 0;
                s->r.top = 0;
                s->r.width = f->o_width;
                s->r.height = f->o_height;
                return 0;

        case V4L2_SEL_TGT_COMPOSE:
                f = &ctx->d_frame;
        case V4L2_SEL_TGT_CROP:
                s->r.left = f->offs_h;
                s->r.top = f->offs_v;
                s->r.width = f->width;
                s->r.height = f->height;
                return 0;
        }

        return -EINVAL;
}

/* Return 1 if rectangle a is enclosed in rectangle b, or 0 otherwise. */
static int enclosed_rectangle(struct v4l2_rect *a, struct v4l2_rect *b)
{
        if (a->left < b->left || a->top < b->top)
                return 0;
        if (a->left + a->width > b->left + b->width)
                return 0;
        if (a->top + a->height > b->top + b->height)
                return 0;

        return 1;
}

static int fimc_cap_s_selection(struct file *file, void *fh,
                                struct v4l2_selection *s)
{
        struct fimc_dev *fimc = video_drvdata(file);
        struct fimc_ctx *ctx = fimc->vid_cap.ctx;
        struct v4l2_rect rect = s->r;
        struct fimc_frame *f;
        unsigned long flags;

        if (s->type != V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE)
                return -EINVAL;

        if (s->target == V4L2_SEL_TGT_COMPOSE)
                f = &ctx->d_frame;
        else if (s->target == V4L2_SEL_TGT_CROP)
                f = &ctx->s_frame;
        else
                return -EINVAL;

        fimc_capture_try_selection(ctx, &rect, s->target);

        if (s->flags & V4L2_SEL_FLAG_LE &&
            !enclosed_rectangle(&rect, &s->r))
                return -ERANGE;

        if (s->flags & V4L2_SEL_FLAG_GE &&
            !enclosed_rectangle(&s->r, &rect))
                return -ERANGE;

        s->r = rect;
        spin_lock_irqsave(&fimc->slock, flags);
        set_frame_crop(f, s->r.left, s->r.top, s->r.width,
                       s->r.height);
        spin_unlock_irqrestore(&fimc->slock, flags);

        set_bit(ST_CAPT_APPLY_CFG, &fimc->state);
        return 0;
}

static const struct v4l2_ioctl_ops fimc_capture_ioctl_ops = {
        .vidioc_querycap                = fimc_cap_querycap,

        .vidioc_enum_fmt_vid_cap_mplane = fimc_cap_enum_fmt_mplane,
        .vidioc_try_fmt_vid_cap_mplane  = fimc_cap_try_fmt_mplane,
        .vidioc_s_fmt_vid_cap_mplane    = fimc_cap_s_fmt_mplane,
        .vidioc_g_fmt_vid_cap_mplane    = fimc_cap_g_fmt_mplane,

        .vidioc_reqbufs                 = fimc_cap_reqbufs,
        .vidioc_querybuf                = vb2_ioctl_querybuf,
        .vidioc_qbuf                    = vb2_ioctl_qbuf,
        .vidioc_dqbuf                   = vb2_ioctl_dqbuf,
        .vidioc_expbuf                  = vb2_ioctl_expbuf,
        .vidioc_prepare_buf             = vb2_ioctl_prepare_buf,
        .vidioc_create_bufs             = vb2_ioctl_create_bufs,

        .vidioc_streamon                = fimc_cap_streamon,
        .vidioc_streamoff               = fimc_cap_streamoff,

        .vidioc_g_selection             = fimc_cap_g_selection,
        .vidioc_s_selection             = fimc_cap_s_selection,

        .vidioc_enum_input              = fimc_cap_enum_input,
        .vidioc_s_input                 = fimc_cap_s_input,
        .vidioc_g_input                 = fimc_cap_g_input,
};

/* Capture subdev media entity operations */
static int fimc_link_setup(struct media_entity *entity,
                           const struct media_pad *local,
                           const struct media_pad *remote, u32 flags)
{
        struct v4l2_subdev *sd = media_entity_to_v4l2_subdev(entity);
        struct fimc_dev *fimc = v4l2_get_subdevdata(sd);

        if (media_entity_type(remote->entity) != MEDIA_ENT_T_V4L2_SUBDEV)
                return -EINVAL;

        if (WARN_ON(fimc == NULL))
                return 0;

        dbg("%s --> %s, flags: 0x%x. input: 0x%x",
            local->entity->name, remote->entity->name, flags,
            fimc->vid_cap.input);

        if (flags & MEDIA_LNK_FL_ENABLED) {
                if (fimc->vid_cap.input != 0)
                        return -EBUSY;
                fimc->vid_cap.input = sd->grp_id;
                return 0;
        }

        fimc->vid_cap.input = 0;
        return 0;
}

static const struct media_entity_operations fimc_sd_media_ops = {
        .link_setup = fimc_link_setup,
};

/**
 * fimc_sensor_notify - v4l2_device notification from a sensor subdev
 * @sd: pointer to a subdev generating the notification
 * @notification: the notification type, must be S5P_FIMC_TX_END_NOTIFY
 * @arg: pointer to an u32 type integer that stores the frame payload value
 *
 * The End Of Frame notification sent by sensor subdev in its still capture
 * mode. If there is only a single VSYNC generated by the sensor at the
 * beginning of a frame transmission, FIMC does not issue the LastIrq
 * (end of frame) interrupt. And this notification is used to complete the
 * frame capture and returning a buffer to user-space. Subdev drivers should
 * call this notification from their last 'End of frame capture' interrupt.
 */
void fimc_sensor_notify(struct v4l2_subdev *sd, unsigned int notification,
                        void *arg)
{
        struct fimc_source_info *si;
        struct fimc_vid_buffer *buf;
        struct fimc_md *fmd;
        struct fimc_dev *fimc;
        unsigned long flags;

        if (sd == NULL)
                return;

        si = v4l2_get_subdev_hostdata(sd);
        fmd = entity_to_fimc_mdev(&sd->entity);

        spin_lock_irqsave(&fmd->slock, flags);

        fimc = si ? source_to_sensor_info(si)->host : NULL;

        if (fimc && arg && notification == S5P_FIMC_TX_END_NOTIFY &&
            test_bit(ST_CAPT_PEND, &fimc->state)) {
                unsigned long irq_flags;
                spin_lock_irqsave(&fimc->slock, irq_flags);
                if (!list_empty(&fimc->vid_cap.active_buf_q)) {
                        buf = list_entry(fimc->vid_cap.active_buf_q.next,
                                         struct fimc_vid_buffer, list);
                        vb2_set_plane_payload(&buf->vb, 0, *((u32 *)arg));
                }
                fimc_capture_irq_handler(fimc, 1);
                fimc_deactivate_capture(fimc);
                spin_unlock_irqrestore(&fimc->slock, irq_flags);
        }
        spin_unlock_irqrestore(&fmd->slock, flags);
}

static int fimc_subdev_enum_mbus_code(struct v4l2_subdev *sd,
                                      struct v4l2_subdev_fh *fh,
                                      struct v4l2_subdev_mbus_code_enum *code)
{
        struct fimc_fmt *fmt;

        fmt = fimc_find_format(NULL, NULL, FMT_FLAGS_CAM, code->index);
        if (!fmt)
                return -EINVAL;
        code->code = fmt->mbus_code;
        return 0;
}

static int fimc_subdev_get_fmt(struct v4l2_subdev *sd,
                               struct v4l2_subdev_fh *fh,
                               struct v4l2_subdev_format *fmt)
{
        struct fimc_dev *fimc = v4l2_get_subdevdata(sd);
        struct fimc_ctx *ctx = fimc->vid_cap.ctx;
        struct fimc_frame *ff = &ctx->s_frame;
        struct v4l2_mbus_framefmt *mf;

        if (fmt->which == V4L2_SUBDEV_FORMAT_TRY) {
                mf = v4l2_subdev_get_try_format(fh, fmt->pad);
                fmt->format = *mf;
                return 0;
        }

        mf = &fmt->format;
        mutex_lock(&fimc->lock);

        switch (fmt->pad) {
        case FIMC_SD_PAD_SOURCE:
                if (!WARN_ON(ff->fmt == NULL))
                        mf->code = ff->fmt->mbus_code;
                /* Sink pads crop rectangle size */
                mf->width = ff->width;
                mf->height = ff->height;
                break;
        case FIMC_SD_PAD_SINK_FIFO:
                *mf = fimc->vid_cap.wb_fmt;
                break;
        case FIMC_SD_PAD_SINK_CAM:
        default:
                *mf = fimc->vid_cap.ci_fmt;
                break;
        }

        mutex_unlock(&fimc->lock);
        mf->colorspace = V4L2_COLORSPACE_JPEG;

        return 0;
}

static int fimc_subdev_set_fmt(struct v4l2_subdev *sd,
                               struct v4l2_subdev_fh *fh,
                               struct v4l2_subdev_format *fmt)
{
        struct fimc_dev *fimc = v4l2_get_subdevdata(sd);
        struct v4l2_mbus_framefmt *mf = &fmt->format;
        struct fimc_vid_cap *vc = &fimc->vid_cap;
        struct fimc_ctx *ctx = vc->ctx;
        struct fimc_frame *ff;
        struct fimc_fmt *ffmt;

        dbg("pad%d: code: 0x%x, %dx%d",
            fmt->pad, mf->code, mf->width, mf->height);

        if (fmt->pad == FIMC_SD_PAD_SOURCE && vb2_is_busy(&vc->vbq))
                return -EBUSY;

        mutex_lock(&fimc->lock);
        ffmt = fimc_capture_try_format(ctx, &mf->width, &mf->height,
                                       &mf->code, NULL, fmt->pad);
        mutex_unlock(&fimc->lock);
        mf->colorspace = V4L2_COLORSPACE_JPEG;

        if (fmt->which == V4L2_SUBDEV_FORMAT_TRY) {
                mf = v4l2_subdev_get_try_format(fh, fmt->pad);
                *mf = fmt->format;
                return 0;
        }
        /* There must be a bug in the driver if this happens */
        if (WARN_ON(ffmt == NULL))
                return -EINVAL;

        /* Update RGB Alpha control state and value range */
        fimc_alpha_ctrl_update(ctx);

        fimc_capture_mark_jpeg_xfer(ctx, ffmt->color);
        if (fmt->pad == FIMC_SD_PAD_SOURCE) {
                ff = &ctx->d_frame;
                /* Sink pads crop rectangle size */
                mf->width = ctx->s_frame.width;
                mf->height = ctx->s_frame.height;
        } else {
                ff = &ctx->s_frame;
        }

        mutex_lock(&fimc->lock);
        set_frame_bounds(ff, mf->width, mf->height);

        if (fmt->pad == FIMC_SD_PAD_SINK_FIFO)
                vc->wb_fmt = *mf;
        else if (fmt->pad == FIMC_SD_PAD_SINK_CAM)
                vc->ci_fmt = *mf;

        ff->fmt = ffmt;

        /* Reset the crop rectangle if required. */
        if (!(fmt->pad == FIMC_SD_PAD_SOURCE && (ctx->state & FIMC_COMPOSE)))
                set_frame_crop(ff, 0, 0, mf->width, mf->height);

        if (fmt->pad != FIMC_SD_PAD_SOURCE)
                ctx->state &= ~FIMC_COMPOSE;

        mutex_unlock(&fimc->lock);
        return 0;
}

static int fimc_subdev_get_selection(struct v4l2_subdev *sd,
                                     struct v4l2_subdev_fh *fh,
                                     struct v4l2_subdev_selection *sel)
{
        struct fimc_dev *fimc = v4l2_get_subdevdata(sd);
        struct fimc_ctx *ctx = fimc->vid_cap.ctx;
        struct fimc_frame *f = &ctx->s_frame;
        struct v4l2_rect *r = &sel->r;
        struct v4l2_rect *try_sel;

        if (sel->pad == FIMC_SD_PAD_SOURCE)
                return -EINVAL;

        mutex_lock(&fimc->lock);

        switch (sel->target) {
        case V4L2_SEL_TGT_COMPOSE_BOUNDS:
                f = &ctx->d_frame;
        case V4L2_SEL_TGT_CROP_BOUNDS:
                r->width = f->o_width;
                r->height = f->o_height;
                r->left = 0;
                r->top = 0;
                mutex_unlock(&fimc->lock);
                return 0;

        case V4L2_SEL_TGT_CROP:
                try_sel = v4l2_subdev_get_try_crop(fh, sel->pad);
                break;
        case V4L2_SEL_TGT_COMPOSE:
                try_sel = v4l2_subdev_get_try_compose(fh, sel->pad);
                f = &ctx->d_frame;
                break;
        default:
                mutex_unlock(&fimc->lock);
                return -EINVAL;
        }

        if (sel->which == V4L2_SUBDEV_FORMAT_TRY) {
                sel->r = *try_sel;
        } else {
                r->left = f->offs_h;
                r->top = f->offs_v;
                r->width = f->width;
                r->height = f->height;
        }

        dbg("target %#x: l:%d, t:%d, %dx%d, f_w: %d, f_h: %d",
            sel->pad, r->left, r->top, r->width, r->height,
            f->f_width, f->f_height);

        mutex_unlock(&fimc->lock);
        return 0;
}

static int fimc_subdev_set_selection(struct v4l2_subdev *sd,
                                     struct v4l2_subdev_fh *fh,
                                     struct v4l2_subdev_selection *sel)
{
        struct fimc_dev *fimc = v4l2_get_subdevdata(sd);
        struct fimc_ctx *ctx = fimc->vid_cap.ctx;
        struct fimc_frame *f = &ctx->s_frame;
        struct v4l2_rect *r = &sel->r;
        struct v4l2_rect *try_sel;
        unsigned long flags;

        if (sel->pad == FIMC_SD_PAD_SOURCE)
                return -EINVAL;

        mutex_lock(&fimc->lock);
        fimc_capture_try_selection(ctx, r, V4L2_SEL_TGT_CROP);

        switch (sel->target) {
        case V4L2_SEL_TGT_CROP:
                try_sel = v4l2_subdev_get_try_crop(fh, sel->pad);
                break;
        case V4L2_SEL_TGT_COMPOSE:
                try_sel = v4l2_subdev_get_try_compose(fh, sel->pad);
                f = &ctx->d_frame;
                break;
        default:
                mutex_unlock(&fimc->lock);
                return -EINVAL;
        }

        if (sel->which == V4L2_SUBDEV_FORMAT_TRY) {
                *try_sel = sel->r;
        } else {
                spin_lock_irqsave(&fimc->slock, flags);
                set_frame_crop(f, r->left, r->top, r->width, r->height);
                set_bit(ST_CAPT_APPLY_CFG, &fimc->state);
                if (sel->target == V4L2_SEL_TGT_COMPOSE)
                        ctx->state |= FIMC_COMPOSE;
                spin_unlock_irqrestore(&fimc->slock, flags);
        }

        dbg("target %#x: (%d,%d)/%dx%d", sel->target, r->left, r->top,
            r->width, r->height);

        mutex_unlock(&fimc->lock);
        return 0;
}

static struct v4l2_subdev_pad_ops fimc_subdev_pad_ops = {
        .enum_mbus_code = fimc_subdev_enum_mbus_code,
        .get_selection = fimc_subdev_get_selection,
        .set_selection = fimc_subdev_set_selection,
        .get_fmt = fimc_subdev_get_fmt,
        .set_fmt = fimc_subdev_set_fmt,
};

static struct v4l2_subdev_ops fimc_subdev_ops = {
        .pad = &fimc_subdev_pad_ops,
};

/* Set default format at the sensor and host interface */
static int fimc_capture_set_default_format(struct fimc_dev *fimc)
{
        struct v4l2_format fmt = {
                .type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE,
                .fmt.pix_mp = {
                        .width          = 640,
                        .height         = 480,
                        .pixelformat    = V4L2_PIX_FMT_YUYV,
                        .field          = V4L2_FIELD_NONE,
                        .colorspace     = V4L2_COLORSPACE_JPEG,
                },
        };

        return __fimc_capture_set_format(fimc, &fmt);
}

/* fimc->lock must be already initialized */
static int fimc_register_capture_device(struct fimc_dev *fimc,
                                 struct v4l2_device *v4l2_dev)
{
        struct video_device *vfd = &fimc->vid_cap.vfd;
        struct vb2_queue *q = &fimc->vid_cap.vbq;
        struct fimc_ctx *ctx;
        struct fimc_vid_cap *vid_cap;
        int ret = -ENOMEM;

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

        ctx->fimc_dev    = fimc;
        ctx->in_path     = FIMC_IO_CAMERA;
        ctx->out_path    = FIMC_IO_DMA;
        ctx->state       = FIMC_CTX_CAP;
        ctx->s_frame.fmt = fimc_find_format(NULL, NULL, FMT_FLAGS_CAM, 0);
        ctx->d_frame.fmt = ctx->s_frame.fmt;

        memset(vfd, 0, sizeof(*vfd));
        snprintf(vfd->name, sizeof(vfd->name), "fimc.%d.capture", fimc->id);

        vfd->fops       = &fimc_capture_fops;
        vfd->ioctl_ops  = &fimc_capture_ioctl_ops;
        vfd->v4l2_dev   = v4l2_dev;
        vfd->minor      = -1;
        vfd->release    = video_device_release_empty;
        vfd->queue      = q;
        vfd->lock       = &fimc->lock;

        video_set_drvdata(vfd, fimc);
        vid_cap = &fimc->vid_cap;
        vid_cap->active_buf_cnt = 0;
        vid_cap->reqbufs_count = 0;
        vid_cap->ctx = ctx;

        INIT_LIST_HEAD(&vid_cap->pending_buf_q);
        INIT_LIST_HEAD(&vid_cap->active_buf_q);

        memset(q, 0, sizeof(*q));
        q->type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE;
        q->io_modes = VB2_MMAP | VB2_USERPTR | VB2_DMABUF;
        q->drv_priv = ctx;
        q->ops = &fimc_capture_qops;
        q->mem_ops = &vb2_dma_contig_memops;
        q->buf_struct_size = sizeof(struct fimc_vid_buffer);
        q->timestamp_type = V4L2_BUF_FLAG_TIMESTAMP_MONOTONIC;
        q->lock = &fimc->lock;

        ret = vb2_queue_init(q);
        if (ret)
                goto err_ent;

        vid_cap->vd_pad.flags = MEDIA_PAD_FL_SINK;
        ret = media_entity_init(&vfd->entity, 1, &vid_cap->vd_pad, 0);
        if (ret)
                goto err_ent;
        /*
         * For proper order of acquiring/releasing the video
         * and the graph mutex.
         */
        v4l2_disable_ioctl_locking(vfd, VIDIOC_TRY_FMT);
        v4l2_disable_ioctl_locking(vfd, VIDIOC_S_FMT);

        ret = video_register_device(vfd, VFL_TYPE_GRABBER, -1);
        if (ret)
                goto err_vd;

        v4l2_info(v4l2_dev, "Registered %s as /dev/%s\n",
                  vfd->name, video_device_node_name(vfd));

        vfd->ctrl_handler = &ctx->ctrls.handler;
        return 0;

err_vd:
        media_entity_cleanup(&vfd->entity);
err_ent:
        kfree(ctx);
        return ret;
}

static int fimc_capture_subdev_registered(struct v4l2_subdev *sd)
{
        struct fimc_dev *fimc = v4l2_get_subdevdata(sd);
        int ret;

        if (fimc == NULL)
                return -ENXIO;

        ret = fimc_register_m2m_device(fimc, sd->v4l2_dev);
        if (ret)
                return ret;

        fimc->pipeline_ops = v4l2_get_subdev_hostdata(sd);

        ret = fimc_register_capture_device(fimc, sd->v4l2_dev);
        if (ret) {
                fimc_unregister_m2m_device(fimc);
                fimc->pipeline_ops = NULL;
        }

        return ret;
}

static void fimc_capture_subdev_unregistered(struct v4l2_subdev *sd)
{
        struct fimc_dev *fimc = v4l2_get_subdevdata(sd);

        if (fimc == NULL)
                return;

        fimc_unregister_m2m_device(fimc);

        if (video_is_registered(&fimc->vid_cap.vfd)) {
                video_unregister_device(&fimc->vid_cap.vfd);
                media_entity_cleanup(&fimc->vid_cap.vfd.entity);
                fimc->pipeline_ops = NULL;
        }
        kfree(fimc->vid_cap.ctx);
        fimc->vid_cap.ctx = NULL;
}

static const struct v4l2_subdev_internal_ops fimc_capture_sd_internal_ops = {
        .registered = fimc_capture_subdev_registered,
        .unregistered = fimc_capture_subdev_unregistered,
};

int fimc_initialize_capture_subdev(struct fimc_dev *fimc)
{
        struct v4l2_subdev *sd = &fimc->vid_cap.subdev;
        int ret;

        v4l2_subdev_init(sd, &fimc_subdev_ops);
        sd->flags |= V4L2_SUBDEV_FL_HAS_DEVNODE;
        snprintf(sd->name, sizeof(sd->name), "FIMC.%d", fimc->id);

        fimc->vid_cap.sd_pads[FIMC_SD_PAD_SINK_CAM].flags = MEDIA_PAD_FL_SINK;
        fimc->vid_cap.sd_pads[FIMC_SD_PAD_SINK_FIFO].flags = MEDIA_PAD_FL_SINK;
        fimc->vid_cap.sd_pads[FIMC_SD_PAD_SOURCE].flags = MEDIA_PAD_FL_SOURCE;
        ret = media_entity_init(&sd->entity, FIMC_SD_PADS_NUM,
                                fimc->vid_cap.sd_pads, 0);
        if (ret)
                return ret;

        sd->entity.ops = &fimc_sd_media_ops;
        sd->internal_ops = &fimc_capture_sd_internal_ops;
        v4l2_set_subdevdata(sd, fimc);
        return 0;
}

void fimc_unregister_capture_subdev(struct fimc_dev *fimc)
{
        struct v4l2_subdev *sd = &fimc->vid_cap.subdev;

        v4l2_device_unregister_subdev(sd);
        media_entity_cleanup(&sd->entity);
        v4l2_set_subdevdata(sd, NULL);
}