#include <linux/clk.h>
#include <linux/dma-mapping.h>
#include <linux/err.h>
#include <linux/interrupt.h>
#include <linux/ioctl.h>
#include <linux/module.h>
#include <linux/of_device.h>
#include <linux/pm_runtime.h>
#include <media/jpeg.h>
#include <media/v4l2-common.h>
#include <media/v4l2-ctrls.h>
#include <media/v4l2-device.h>
#include <media/v4l2-event.h>
#include <media/v4l2-ioctl.h>
#include <media/v4l2-jpeg.h>
#include <media/v4l2-rect.h>
#include <media/v4l2-mem2mem.h>
#include <media/videobuf2-dma-contig.h>
#include <media/videobuf2-v4l2.h>
#include "e5010-jpeg-enc.h"
#include "e5010-jpeg-enc-hw.h"
static const struct of_device_id e5010_of_match[];
static const struct v4l2_file_operations e5010_fops;
static const struct v4l2_ioctl_ops e5010_ioctl_ops;
static const struct vb2_ops e5010_video_ops;
static const struct v4l2_m2m_ops e5010_m2m_ops;
static struct e5010_fmt e5010_formats[] = {
{
.fourcc = V4L2_PIX_FMT_NV12,
.num_planes = 1,
.type = V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE,
.subsampling = V4L2_JPEG_CHROMA_SUBSAMPLING_420,
.chroma_order = CHROMA_ORDER_CB_CR,
.frmsize = { MIN_DIMENSION, MAX_DIMENSION, 64,
MIN_DIMENSION, MAX_DIMENSION, 8 },
},
{
.fourcc = V4L2_PIX_FMT_NV12M,
.num_planes = 2,
.type = V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE,
.subsampling = V4L2_JPEG_CHROMA_SUBSAMPLING_420,
.chroma_order = CHROMA_ORDER_CB_CR,
.frmsize = { MIN_DIMENSION, MAX_DIMENSION, 64,
MIN_DIMENSION, MAX_DIMENSION, 8 },
},
{
.fourcc = V4L2_PIX_FMT_NV21,
.num_planes = 1,
.type = V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE,
.subsampling = V4L2_JPEG_CHROMA_SUBSAMPLING_420,
.chroma_order = CHROMA_ORDER_CR_CB,
.frmsize = { MIN_DIMENSION, MAX_DIMENSION, 64,
MIN_DIMENSION, MAX_DIMENSION, 8 },
},
{
.fourcc = V4L2_PIX_FMT_NV21M,
.num_planes = 2,
.type = V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE,
.subsampling = V4L2_JPEG_CHROMA_SUBSAMPLING_420,
.chroma_order = CHROMA_ORDER_CR_CB,
.frmsize = { MIN_DIMENSION, MAX_DIMENSION, 64,
MIN_DIMENSION, MAX_DIMENSION, 8 },
},
{
.fourcc = V4L2_PIX_FMT_NV16,
.num_planes = 1,
.type = V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE,
.subsampling = V4L2_JPEG_CHROMA_SUBSAMPLING_422,
.chroma_order = CHROMA_ORDER_CB_CR,
.frmsize = { MIN_DIMENSION, MAX_DIMENSION, 64,
MIN_DIMENSION, MAX_DIMENSION, 8 },
},
{
.fourcc = V4L2_PIX_FMT_NV16M,
.num_planes = 2,
.type = V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE,
.subsampling = V4L2_JPEG_CHROMA_SUBSAMPLING_422,
.chroma_order = CHROMA_ORDER_CB_CR,
.frmsize = { MIN_DIMENSION, MAX_DIMENSION, 64,
MIN_DIMENSION, MAX_DIMENSION, 8 },
},
{
.fourcc = V4L2_PIX_FMT_NV61,
.num_planes = 1,
.type = V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE,
.subsampling = V4L2_JPEG_CHROMA_SUBSAMPLING_422,
.chroma_order = CHROMA_ORDER_CR_CB,
.frmsize = { MIN_DIMENSION, MAX_DIMENSION, 64,
MIN_DIMENSION, MAX_DIMENSION, 8 },
},
{
.fourcc = V4L2_PIX_FMT_NV61M,
.num_planes = 2,
.type = V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE,
.subsampling = V4L2_JPEG_CHROMA_SUBSAMPLING_422,
.chroma_order = CHROMA_ORDER_CR_CB,
.frmsize = { MIN_DIMENSION, MAX_DIMENSION, 64,
MIN_DIMENSION, MAX_DIMENSION, 8 },
},
{
.fourcc = V4L2_PIX_FMT_JPEG,
.num_planes = 1,
.type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE,
.subsampling = 0,
.chroma_order = 0,
.frmsize = { MIN_DIMENSION, MAX_DIMENSION, 16,
MIN_DIMENSION, MAX_DIMENSION, 8 },
},
};
static unsigned int debug;
module_param(debug, uint, 0644);
MODULE_PARM_DESC(debug, "debug level");
#define dprintk(dev, lvl, fmt, arg...) \
v4l2_dbg(lvl, debug, &(dev)->v4l2_dev, "%s: " fmt, __func__, ## arg)
static const struct v4l2_event e5010_eos_event = {
.type = V4L2_EVENT_EOS
};
static const char *type_name(enum v4l2_buf_type type)
{
switch (type) {
case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE:
return "Output";
case V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE:
return "Capture";
default:
return "Invalid";
}
}
static struct e5010_q_data *get_queue(struct e5010_context *ctx, enum v4l2_buf_type type)
{
return (type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) ? &ctx->out_queue : &ctx->cap_queue;
}
static void calculate_qp_tables(struct e5010_context *ctx)
{
long long luminosity, contrast;
int quality, i;
quality = 50 - ctx->quality;
luminosity = LUMINOSITY * quality / 50;
contrast = CONTRAST * quality / 50;
if (quality > 0) {
luminosity *= INCREASE;
contrast *= INCREASE;
}
for (i = 0; i < V4L2_JPEG_PIXELS_IN_BLOCK; i++) {
long long delta = v4l2_jpeg_ref_table_chroma_qt[i] * contrast + luminosity;
int val = (int)(v4l2_jpeg_ref_table_chroma_qt[i] + delta);
clamp(val, 1, 255);
ctx->chroma_qp[i] = quality == -50 ? 1 : val;
delta = v4l2_jpeg_ref_table_luma_qt[i] * contrast + luminosity;
val = (int)(v4l2_jpeg_ref_table_luma_qt[i] + delta);
clamp(val, 1, 255);
ctx->luma_qp[i] = quality == -50 ? 1 : val;
}
ctx->update_qp = true;
}
static int update_qp_tables(struct e5010_context *ctx)
{
struct e5010_dev *e5010 = ctx->e5010;
int i, ret = 0;
u32 lvalue, cvalue;
lvalue = 0;
cvalue = 0;
for (i = 0; i < QP_TABLE_SIZE; i++) {
lvalue |= ctx->luma_qp[i] << (8 * (i % 4));
cvalue |= ctx->chroma_qp[i] << (8 * (i % 4));
if (i % 4 == 3) {
ret |= e5010_hw_set_qpvalue(e5010->core_base,
JASPER_LUMA_QUANTIZATION_TABLE0_OFFSET
+ QP_TABLE_FIELD_OFFSET * ((i - 3) / 4),
lvalue);
ret |= e5010_hw_set_qpvalue(e5010->core_base,
JASPER_CHROMA_QUANTIZATION_TABLE0_OFFSET
+ QP_TABLE_FIELD_OFFSET * ((i - 3) / 4),
cvalue);
lvalue = 0;
cvalue = 0;
}
}
return ret;
}
static int e5010_set_input_subsampling(void __iomem *core_base, int subsampling)
{
switch (subsampling) {
case V4L2_JPEG_CHROMA_SUBSAMPLING_420:
return e5010_hw_set_input_subsampling(core_base, SUBSAMPLING_420);
case V4L2_JPEG_CHROMA_SUBSAMPLING_422:
return e5010_hw_set_input_subsampling(core_base, SUBSAMPLING_422);
default:
return -EINVAL;
};
}
static int e5010_querycap(struct file *file, void *priv, struct v4l2_capability *cap)
{
strscpy(cap->driver, E5010_MODULE_NAME, sizeof(cap->driver));
strscpy(cap->card, E5010_MODULE_NAME, sizeof(cap->card));
return 0;
}
static struct e5010_fmt *find_format(struct v4l2_format *f)
{
int i;
for (i = 0; i < ARRAY_SIZE(e5010_formats); ++i) {
if (e5010_formats[i].fourcc == f->fmt.pix_mp.pixelformat &&
e5010_formats[i].type == f->type)
return &e5010_formats[i];
}
return NULL;
}
static int e5010_enum_fmt(struct file *file, void *priv, struct v4l2_fmtdesc *f)
{
int i, index = 0;
struct e5010_fmt *fmt = NULL;
struct e5010_context *ctx = to_e5010_context(file);
if (!V4L2_TYPE_IS_MULTIPLANAR(f->type)) {
v4l2_err(&ctx->e5010->v4l2_dev, "ENUMFMT with Invalid type: %d\n", f->type);
return -EINVAL;
}
for (i = 0; i < ARRAY_SIZE(e5010_formats); ++i) {
if (e5010_formats[i].type == f->type) {
if (index == f->index) {
fmt = &e5010_formats[i];
break;
}
index++;
}
}
if (!fmt)
return -EINVAL;
f->pixelformat = fmt->fourcc;
return 0;
}
static int e5010_g_fmt(struct file *file, void *priv, struct v4l2_format *f)
{
struct e5010_context *ctx = to_e5010_context(file);
struct e5010_q_data *queue;
int i;
struct v4l2_pix_format_mplane *pix_mp = &f->fmt.pix_mp;
struct v4l2_plane_pix_format *plane_fmt = pix_mp->plane_fmt;
queue = get_queue(ctx, f->type);
pix_mp->flags = 0;
pix_mp->field = V4L2_FIELD_NONE;
pix_mp->pixelformat = queue->fmt->fourcc;
pix_mp->width = queue->width_adjusted;
pix_mp->height = queue->height_adjusted;
pix_mp->num_planes = queue->fmt->num_planes;
if (V4L2_TYPE_IS_OUTPUT(f->type)) {
if (!pix_mp->colorspace)
pix_mp->colorspace = V4L2_COLORSPACE_SRGB;
for (i = 0; i < queue->fmt->num_planes; i++) {
plane_fmt[i].sizeimage = queue->sizeimage[i];
plane_fmt[i].bytesperline = queue->bytesperline[i];
}
} else {
pix_mp->colorspace = V4L2_COLORSPACE_JPEG;
plane_fmt[0].bytesperline = 0;
plane_fmt[0].sizeimage = queue->sizeimage[0];
}
pix_mp->ycbcr_enc = V4L2_YCBCR_ENC_DEFAULT;
pix_mp->xfer_func = V4L2_XFER_FUNC_DEFAULT;
pix_mp->quantization = V4L2_QUANTIZATION_DEFAULT;
return 0;
}
static int e5010_jpeg_try_fmt(struct v4l2_format *f, struct e5010_context *ctx)
{
struct e5010_fmt *fmt;
struct v4l2_pix_format_mplane *pix_mp = &f->fmt.pix_mp;
struct v4l2_plane_pix_format *plane_fmt = pix_mp->plane_fmt;
fmt = find_format(f);
if (!fmt) {
if (V4L2_TYPE_IS_OUTPUT(f->type))
pix_mp->pixelformat = V4L2_PIX_FMT_NV12;
else
pix_mp->pixelformat = V4L2_PIX_FMT_JPEG;
fmt = find_format(f);
if (!fmt)
return -EINVAL;
}
if (V4L2_TYPE_IS_OUTPUT(f->type)) {
if (!pix_mp->colorspace)
pix_mp->colorspace = V4L2_COLORSPACE_JPEG;
if (!pix_mp->ycbcr_enc)
pix_mp->ycbcr_enc = V4L2_YCBCR_ENC_DEFAULT;
if (!pix_mp->quantization)
pix_mp->quantization = V4L2_QUANTIZATION_DEFAULT;
if (!pix_mp->xfer_func)
pix_mp->xfer_func = V4L2_XFER_FUNC_DEFAULT;
v4l2_apply_frmsize_constraints(&pix_mp->width,
&pix_mp->height,
&fmt->frmsize);
v4l2_fill_pixfmt_mp(pix_mp, pix_mp->pixelformat,
pix_mp->width, pix_mp->height);
} else {
pix_mp->colorspace = V4L2_COLORSPACE_JPEG;
pix_mp->ycbcr_enc = V4L2_YCBCR_ENC_DEFAULT;
pix_mp->quantization = V4L2_QUANTIZATION_DEFAULT;
pix_mp->xfer_func = V4L2_XFER_FUNC_DEFAULT;
v4l2_apply_frmsize_constraints(&pix_mp->width,
&pix_mp->height,
&fmt->frmsize);
plane_fmt[0].sizeimage = pix_mp->width * pix_mp->height * JPEG_MAX_BYTES_PER_PIXEL;
plane_fmt[0].sizeimage += HEADER_SIZE;
plane_fmt[0].bytesperline = 0;
pix_mp->pixelformat = fmt->fourcc;
pix_mp->num_planes = fmt->num_planes;
}
pix_mp->flags = 0;
pix_mp->field = V4L2_FIELD_NONE;
dprintk(ctx->e5010, 2,
"ctx: 0x%p: format type %s:, wxh: %dx%d (plane0 : %d bytes, plane1 : %d bytes),fmt: %c%c%c%c\n",
ctx, type_name(f->type), pix_mp->width, pix_mp->height,
plane_fmt[0].sizeimage, plane_fmt[1].sizeimage,
(fmt->fourcc & 0xff),
(fmt->fourcc >> 8) & 0xff,
(fmt->fourcc >> 16) & 0xff,
(fmt->fourcc >> 24) & 0xff);
return 0;
}
static int e5010_try_fmt(struct file *file, void *priv, struct v4l2_format *f)
{
struct e5010_context *ctx = to_e5010_context(file);
return e5010_jpeg_try_fmt(f, ctx);
}
static int e5010_s_fmt(struct file *file, void *priv, struct v4l2_format *f)
{
struct e5010_context *ctx = to_e5010_context(file);
struct vb2_queue *vq;
int ret = 0, i = 0;
struct v4l2_pix_format_mplane *pix_mp = &f->fmt.pix_mp;
struct v4l2_plane_pix_format *plane_fmt = pix_mp->plane_fmt;
struct e5010_q_data *queue;
struct e5010_fmt *fmt;
vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, f->type);
if (vb2_is_busy(vq)) {
v4l2_err(&ctx->e5010->v4l2_dev, "queue busy\n");
return -EBUSY;
}
ret = e5010_jpeg_try_fmt(f, ctx);
if (ret)
return ret;
fmt = find_format(f);
queue = get_queue(ctx, f->type);
queue->fmt = fmt;
queue->width = pix_mp->width;
queue->height = pix_mp->height;
if (V4L2_TYPE_IS_OUTPUT(f->type)) {
for (i = 0; i < fmt->num_planes; i++) {
queue->bytesperline[i] = plane_fmt[i].bytesperline;
queue->sizeimage[i] = plane_fmt[i].sizeimage;
}
queue->crop.left = 0;
queue->crop.top = 0;
queue->crop.width = queue->width;
queue->crop.height = queue->height;
} else {
queue->sizeimage[0] = plane_fmt[0].sizeimage;
queue->sizeimage[1] = 0;
queue->bytesperline[0] = 0;
queue->bytesperline[1] = 0;
}
return 0;
}
static int e5010_enum_framesizes(struct file *file, void *priv, struct v4l2_frmsizeenum *fsize)
{
struct v4l2_format f;
struct e5010_fmt *fmt;
if (fsize->index != 0)
return -EINVAL;
f.fmt.pix_mp.pixelformat = fsize->pixel_format;
if (f.fmt.pix_mp.pixelformat == V4L2_PIX_FMT_JPEG)
f.type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE;
else
f.type = V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE;
fmt = find_format(&f);
if (!fmt)
return -EINVAL;
fsize->type = V4L2_FRMSIZE_TYPE_STEPWISE;
fsize->stepwise = fmt->frmsize;
fsize->reserved[0] = 0;
fsize->reserved[1] = 0;
return 0;
}
static int e5010_g_selection(struct file *file, void *fh, struct v4l2_selection *s)
{
struct e5010_context *ctx = to_e5010_context(file);
struct e5010_q_data *queue;
if (s->type != V4L2_BUF_TYPE_VIDEO_OUTPUT)
return -EINVAL;
queue = get_queue(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE);
switch (s->target) {
case V4L2_SEL_TGT_CROP_DEFAULT:
case V4L2_SEL_TGT_CROP_BOUNDS:
s->r.left = 0;
s->r.top = 0;
s->r.width = queue->width;
s->r.height = queue->height;
break;
case V4L2_SEL_TGT_CROP:
memcpy(&s->r, &queue->crop, sizeof(s->r));
break;
default:
return -EINVAL;
}
return 0;
}
static int e5010_s_selection(struct file *file, void *fh, struct v4l2_selection *s)
{
struct e5010_context *ctx = to_e5010_context(file);
struct e5010_q_data *queue;
struct vb2_queue *vq;
struct v4l2_rect base_rect;
vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, s->type);
if (vb2_is_streaming(vq))
return -EBUSY;
if (s->target != V4L2_SEL_TGT_CROP ||
s->type != V4L2_BUF_TYPE_VIDEO_OUTPUT)
return -EINVAL;
queue = get_queue(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE);
base_rect.top = 0;
base_rect.left = 0;
base_rect.width = queue->width;
base_rect.height = queue->height;
switch (s->flags) {
case 0:
s->r.width = round_down(s->r.width, queue->fmt->frmsize.step_width);
s->r.height = round_down(s->r.height, queue->fmt->frmsize.step_height);
s->r.left = round_down(s->r.left, queue->fmt->frmsize.step_width);
s->r.top = round_down(s->r.top, 2);
if (s->r.left + s->r.width > queue->width)
s->r.width = round_down(s->r.width + s->r.left - queue->width,
queue->fmt->frmsize.step_width);
if (s->r.top + s->r.height > queue->height)
s->r.top = round_down(s->r.top + s->r.height - queue->height, 2);
break;
case V4L2_SEL_FLAG_GE:
s->r.width = round_up(s->r.width, queue->fmt->frmsize.step_width);
s->r.height = round_up(s->r.height, queue->fmt->frmsize.step_height);
s->r.left = round_up(s->r.left, queue->fmt->frmsize.step_width);
s->r.top = round_up(s->r.top, 2);
break;
case V4L2_SEL_FLAG_LE:
s->r.width = round_down(s->r.width, queue->fmt->frmsize.step_width);
s->r.height = round_down(s->r.height, queue->fmt->frmsize.step_height);
s->r.left = round_down(s->r.left, queue->fmt->frmsize.step_width);
s->r.top = round_down(s->r.top, 2);
break;
case V4L2_SEL_FLAG_LE | V4L2_SEL_FLAG_GE:
if (!IS_ALIGNED(s->r.width, queue->fmt->frmsize.step_width) ||
!IS_ALIGNED(s->r.height, queue->fmt->frmsize.step_height) ||
!IS_ALIGNED(s->r.left, queue->fmt->frmsize.step_width) ||
!IS_ALIGNED(s->r.top, 2))
return -ERANGE;
break;
default:
return -EINVAL;
}
if (!v4l2_rect_enclosed(&s->r, &base_rect))
return -ERANGE;
memcpy(&queue->crop, &s->r, sizeof(s->r));
if (!v4l2_rect_equal(&s->r, &base_rect))
queue->crop_set = true;
dprintk(ctx->e5010, 2, "ctx: 0x%p: crop rectangle: w: %d, h : %d, l : %d, t : %d\n",
ctx, queue->crop.width, queue->crop.height, queue->crop.left, queue->crop.top);
return 0;
}
static int e5010_subscribe_event(struct v4l2_fh *fh, const struct v4l2_event_subscription *sub)
{
switch (sub->type) {
case V4L2_EVENT_EOS:
return v4l2_event_subscribe(fh, sub, 0, NULL);
case V4L2_EVENT_CTRL:
return v4l2_ctrl_subscribe_event(fh, sub);
default:
return -EINVAL;
}
return 0;
}
static int queue_init(void *priv, struct vb2_queue *src_vq, struct vb2_queue *dst_vq)
{
struct e5010_context *ctx = priv;
struct e5010_dev *e5010 = ctx->e5010;
int ret = 0;
memset(src_vq, 0, sizeof(*src_vq));
src_vq->type = V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE;
src_vq->io_modes = VB2_MMAP | VB2_DMABUF;
src_vq->drv_priv = ctx;
src_vq->buf_struct_size = sizeof(struct e5010_buffer);
src_vq->ops = &e5010_video_ops;
src_vq->mem_ops = &vb2_dma_contig_memops;
src_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY;
src_vq->lock = &e5010->mutex;
src_vq->dev = e5010->v4l2_dev.dev;
ret = vb2_queue_init(src_vq);
if (ret)
return ret;
memset(dst_vq, 0, sizeof(*dst_vq));
dst_vq->type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE;
dst_vq->io_modes = VB2_MMAP | VB2_DMABUF;
dst_vq->drv_priv = ctx;
dst_vq->buf_struct_size = sizeof(struct e5010_buffer);
dst_vq->ops = &e5010_video_ops;
dst_vq->mem_ops = &vb2_dma_contig_memops;
dst_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY;
dst_vq->lock = &e5010->mutex;
dst_vq->dev = e5010->v4l2_dev.dev;
ret = vb2_queue_init(dst_vq);
if (ret) {
vb2_queue_release(src_vq);
return ret;
}
return 0;
}
static int e5010_s_ctrl(struct v4l2_ctrl *ctrl)
{
struct e5010_context *ctx =
container_of(ctrl->handler, struct e5010_context, ctrl_handler);
switch (ctrl->id) {
case V4L2_CID_JPEG_COMPRESSION_QUALITY:
ctx->quality = ctrl->val;
calculate_qp_tables(ctx);
dprintk(ctx->e5010, 2, "ctx: 0x%p compression quality set to : %d\n", ctx,
ctx->quality);
break;
default:
return -EINVAL;
}
return 0;
}
static const struct v4l2_ctrl_ops e5010_ctrl_ops = {
.s_ctrl = e5010_s_ctrl,
};
static void e5010_encode_ctrls(struct e5010_context *ctx)
{
v4l2_ctrl_new_std(&ctx->ctrl_handler, &e5010_ctrl_ops,
V4L2_CID_JPEG_COMPRESSION_QUALITY, 1, 100, 1, 75);
}
static int e5010_ctrls_setup(struct e5010_context *ctx)
{
int err;
v4l2_ctrl_handler_init(&ctx->ctrl_handler, 1);
e5010_encode_ctrls(ctx);
if (ctx->ctrl_handler.error) {
err = ctx->ctrl_handler.error;
v4l2_ctrl_handler_free(&ctx->ctrl_handler);
return err;
}
err = v4l2_ctrl_handler_setup(&ctx->ctrl_handler);
if (err)
v4l2_ctrl_handler_free(&ctx->ctrl_handler);
return err;
}
static void e5010_jpeg_set_default_params(struct e5010_context *ctx)
{
struct e5010_q_data *queue;
struct v4l2_format f;
struct e5010_fmt *fmt;
struct v4l2_pix_format_mplane *pix_mp = &f.fmt.pix_mp;
struct v4l2_plane_pix_format *plane_fmt = pix_mp->plane_fmt;
int i = 0;
f.type = V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE;
f.fmt.pix_mp.pixelformat = V4L2_PIX_FMT_NV12;
fmt = find_format(&f);
queue = &ctx->out_queue;
queue->fmt = fmt;
queue->width = DEFAULT_WIDTH;
queue->height = DEFAULT_HEIGHT;
pix_mp->width = queue->width;
pix_mp->height = queue->height;
queue->crop.left = 0;
queue->crop.top = 0;
queue->crop.width = queue->width;
queue->crop.height = queue->height;
v4l2_apply_frmsize_constraints(&pix_mp->width,
&pix_mp->height,
&fmt->frmsize);
v4l2_fill_pixfmt_mp(pix_mp, pix_mp->pixelformat,
pix_mp->width, pix_mp->height);
for (i = 0; i < fmt->num_planes; i++) {
queue->bytesperline[i] = plane_fmt[i].bytesperline;
queue->sizeimage[i] = plane_fmt[i].sizeimage;
}
queue->width_adjusted = pix_mp->width;
queue->height_adjusted = pix_mp->height;
f.type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE;
f.fmt.pix_mp.pixelformat = V4L2_PIX_FMT_JPEG;
fmt = find_format(&f);
queue = &ctx->cap_queue;
queue->fmt = fmt;
queue->width = DEFAULT_WIDTH;
queue->height = DEFAULT_HEIGHT;
pix_mp->width = queue->width;
pix_mp->height = queue->height;
v4l2_apply_frmsize_constraints(&pix_mp->width,
&pix_mp->height,
&fmt->frmsize);
queue->sizeimage[0] = pix_mp->width * pix_mp->height * JPEG_MAX_BYTES_PER_PIXEL;
queue->sizeimage[0] += HEADER_SIZE;
queue->sizeimage[1] = 0;
queue->bytesperline[0] = 0;
queue->bytesperline[1] = 0;
queue->width_adjusted = pix_mp->width;
queue->height_adjusted = pix_mp->height;
}
static int e5010_open(struct file *file)
{
struct e5010_dev *e5010 = video_drvdata(file);
struct video_device *vdev = video_devdata(file);
struct e5010_context *ctx;
int ret = 0;
ctx = kzalloc_obj(*ctx);
if (!ctx)
return -ENOMEM;
if (mutex_lock_interruptible(&e5010->mutex)) {
ret = -ERESTARTSYS;
goto free;
}
v4l2_fh_init(&ctx->fh, vdev);
v4l2_fh_add(&ctx->fh, file);
ctx->e5010 = e5010;
ctx->fh.m2m_ctx = v4l2_m2m_ctx_init(e5010->m2m_dev, ctx, queue_init);
if (IS_ERR(ctx->fh.m2m_ctx)) {
v4l2_err(&e5010->v4l2_dev, "failed to init m2m ctx\n");
ret = PTR_ERR(ctx->fh.m2m_ctx);
goto exit;
}
ret = e5010_ctrls_setup(ctx);
if (ret) {
v4l2_err(&e5010->v4l2_dev, "failed to setup e5010 jpeg controls\n");
goto err_ctrls_setup;
}
ctx->fh.ctrl_handler = &ctx->ctrl_handler;
e5010_jpeg_set_default_params(ctx);
dprintk(e5010, 1, "Created instance: 0x%p, m2m_ctx: 0x%p\n", ctx, ctx->fh.m2m_ctx);
mutex_unlock(&e5010->mutex);
return 0;
err_ctrls_setup:
v4l2_m2m_ctx_release(ctx->fh.m2m_ctx);
exit:
v4l2_fh_del(&ctx->fh, file);
v4l2_fh_exit(&ctx->fh);
mutex_unlock(&e5010->mutex);
free:
kfree(ctx);
return ret;
}
static int e5010_release(struct file *file)
{
struct e5010_dev *e5010 = video_drvdata(file);
struct e5010_context *ctx = to_e5010_context(file);
dprintk(e5010, 1, "Releasing instance: 0x%p, m2m_ctx: 0x%p\n", ctx, ctx->fh.m2m_ctx);
mutex_lock(&e5010->mutex);
v4l2_ctrl_handler_free(&ctx->ctrl_handler);
v4l2_m2m_ctx_release(ctx->fh.m2m_ctx);
v4l2_fh_del(&ctx->fh, file);
v4l2_fh_exit(&ctx->fh);
kfree(ctx);
mutex_unlock(&e5010->mutex);
return 0;
}
static void header_write(struct e5010_context *ctx, u8 *addr, unsigned int *offset,
unsigned int no_bytes, unsigned long bits)
{
u8 *w_addr = addr + *offset;
int i;
if ((*offset + no_bytes) > HEADER_SIZE) {
v4l2_warn(&ctx->e5010->v4l2_dev, "%s: %s: %d: Problem writing header. %d > HEADER_SIZE %d\n",
__FILE__, __func__, __LINE__, *offset + no_bytes, HEADER_SIZE);
return;
}
for (i = no_bytes - 1; i >= 0; i--)
*(w_addr++) = ((u8 *)&bits)[i];
*offset += no_bytes;
}
static void encode_marker_segment(struct e5010_context *ctx, void *addr, unsigned int *offset)
{
u8 *buffer = (u8 *)addr;
int i;
header_write(ctx, buffer, offset, 2, START_OF_IMAGE);
header_write(ctx, buffer, offset, 2, DQT_MARKER);
header_write(ctx, buffer, offset, 3, LQPQ << 4);
for (i = 0; i < V4L2_JPEG_PIXELS_IN_BLOCK; i++)
header_write(ctx, buffer, offset, 1, ctx->luma_qp[v4l2_jpeg_zigzag_scan_index[i]]);
header_write(ctx, buffer, offset, 2, DQT_MARKER);
header_write(ctx, buffer, offset, 3, (LQPQ << 4) | 1);
for (i = 0; i < V4L2_JPEG_PIXELS_IN_BLOCK; i++)
header_write(ctx, buffer, offset, 1,
ctx->chroma_qp[v4l2_jpeg_zigzag_scan_index[i]]);
header_write(ctx, buffer, offset, 2, DHT_MARKER);
header_write(ctx, buffer, offset, 2, LH_DC);
header_write(ctx, buffer, offset, 1, V4L2_JPEG_LUM_HT | V4L2_JPEG_DC_HT);
for (i = 0 ; i < V4L2_JPEG_REF_HT_DC_LEN; i++)
header_write(ctx, buffer, offset, 1, v4l2_jpeg_ref_table_luma_dc_ht[i]);
header_write(ctx, buffer, offset, 2, DHT_MARKER);
header_write(ctx, buffer, offset, 2, LH_AC);
header_write(ctx, buffer, offset, 1, V4L2_JPEG_LUM_HT | V4L2_JPEG_AC_HT);
for (i = 0 ; i < V4L2_JPEG_REF_HT_AC_LEN; i++)
header_write(ctx, buffer, offset, 1, v4l2_jpeg_ref_table_luma_ac_ht[i]);
header_write(ctx, buffer, offset, 2, DHT_MARKER);
header_write(ctx, buffer, offset, 2, LH_DC);
header_write(ctx, buffer, offset, 1, V4L2_JPEG_CHR_HT | V4L2_JPEG_DC_HT);
for (i = 0 ; i < V4L2_JPEG_REF_HT_DC_LEN; i++)
header_write(ctx, buffer, offset, 1, v4l2_jpeg_ref_table_chroma_dc_ht[i]);
header_write(ctx, buffer, offset, 2, DHT_MARKER);
header_write(ctx, buffer, offset, 2, LH_AC);
header_write(ctx, buffer, offset, 1, V4L2_JPEG_CHR_HT | V4L2_JPEG_AC_HT);
for (i = 0 ; i < V4L2_JPEG_REF_HT_AC_LEN; i++)
header_write(ctx, buffer, offset, 1, v4l2_jpeg_ref_table_chroma_ac_ht[i]);
}
static void encode_frame_header(struct e5010_context *ctx, void *addr, unsigned int *offset)
{
u8 *buffer = (u8 *)addr;
header_write(ctx, buffer, offset, 2, SOF_BASELINE_DCT);
header_write(ctx, buffer, offset, 2, 8 + (3 * UC_NUM_COMP));
header_write(ctx, buffer, offset, 1, PRECISION);
header_write(ctx, buffer, offset, 2, ctx->out_queue.crop.height);
header_write(ctx, buffer, offset, 2, ctx->out_queue.crop.width);
header_write(ctx, buffer, offset, 1, UC_NUM_COMP);
header_write(ctx, buffer, offset, 1, 1);
if (ctx->out_queue.fmt->subsampling == V4L2_JPEG_CHROMA_SUBSAMPLING_422)
header_write(ctx, buffer, offset, 1,
HORZ_SAMPLING_FACTOR | (VERT_SAMPLING_FACTOR_422));
else
header_write(ctx, buffer, offset, 1,
HORZ_SAMPLING_FACTOR | (VERT_SAMPLING_FACTOR_420));
header_write(ctx, buffer, offset, 1, 0);
header_write(ctx, buffer, offset, 1, 2);
header_write(ctx, buffer, offset, 1, (HORZ_SAMPLING_FACTOR >> 1) | 1);
header_write(ctx, buffer, offset, 1, 1);
header_write(ctx, buffer, offset, 1, 3);
header_write(ctx, buffer, offset, 1, (HORZ_SAMPLING_FACTOR >> 1) | 1);
header_write(ctx, buffer, offset, 1, 1);
}
static void jpg_encode_sos_header(struct e5010_context *ctx, void *addr, unsigned int *offset)
{
u8 *buffer = (u8 *)addr;
int i;
header_write(ctx, buffer, offset, 2, START_OF_SCAN);
header_write(ctx, buffer, offset, 2, 6 + (COMPONENTS_IN_SCAN << 1));
header_write(ctx, buffer, offset, 1, COMPONENTS_IN_SCAN);
for (i = 0; i < COMPONENTS_IN_SCAN; i++) {
header_write(ctx, buffer, offset, 1, i + 1);
if (i == 0)
header_write(ctx, buffer, offset, 1, 0);
else
header_write(ctx, buffer, offset, 1, 17);
}
header_write(ctx, buffer, offset, 1, 0);
header_write(ctx, buffer, offset, 1, 63);
header_write(ctx, buffer, offset, 1, 0);
}
static void write_header(struct e5010_context *ctx, void *addr)
{
unsigned int offset = 0;
encode_marker_segment(ctx, addr, &offset);
encode_frame_header(ctx, addr, &offset);
jpg_encode_sos_header(ctx, addr, &offset);
}
static irqreturn_t e5010_irq(int irq, void *data)
{
struct e5010_dev *e5010 = data;
struct e5010_context *ctx;
int output_size;
struct vb2_v4l2_buffer *src_buf, *dst_buf;
bool pic_done, out_addr_err;
spin_lock(&e5010->hw_lock);
pic_done = e5010_hw_pic_done_irq(e5010->core_base);
out_addr_err = e5010_hw_output_address_irq(e5010->core_base);
if (!pic_done && !out_addr_err) {
spin_unlock(&e5010->hw_lock);
return IRQ_NONE;
}
ctx = v4l2_m2m_get_curr_priv(e5010->m2m_dev);
if (WARN_ON(!ctx))
goto job_unlock;
dst_buf = v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx);
src_buf = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx);
if (!dst_buf || !src_buf) {
v4l2_err(&e5010->v4l2_dev, "ctx: 0x%p No source or destination buffer\n", ctx);
goto job_unlock;
}
if (out_addr_err) {
e5010_hw_clear_output_error(e5010->core_base, 1);
v4l2_warn(&e5010->v4l2_dev,
"ctx: 0x%p Output bitstream size exceeded max size\n", ctx);
v4l2_m2m_buf_done(src_buf, VB2_BUF_STATE_ERROR);
vb2_set_plane_payload(&dst_buf->vb2_buf, 0, dst_buf->planes[0].length);
v4l2_m2m_buf_done(dst_buf, VB2_BUF_STATE_ERROR);
if (v4l2_m2m_is_last_draining_src_buf(ctx->fh.m2m_ctx, src_buf)) {
dst_buf->flags |= V4L2_BUF_FLAG_LAST;
v4l2_m2m_mark_stopped(ctx->fh.m2m_ctx);
v4l2_event_queue_fh(&ctx->fh, &e5010_eos_event);
dprintk(e5010, 2, "ctx: 0x%p Sending EOS\n", ctx);
}
}
if (pic_done) {
e5010_hw_clear_picture_done(e5010->core_base, 1);
dprintk(e5010, 3, "ctx: 0x%p Got output bitstream of size %d bytes\n",
ctx, readl(e5010->core_base + JASPER_OUTPUT_SIZE_OFFSET));
if (v4l2_m2m_is_last_draining_src_buf(ctx->fh.m2m_ctx, src_buf)) {
dst_buf->flags |= V4L2_BUF_FLAG_LAST;
v4l2_m2m_mark_stopped(ctx->fh.m2m_ctx);
v4l2_event_queue_fh(&ctx->fh, &e5010_eos_event);
dprintk(e5010, 2, "ctx: 0x%p Sending EOS\n", ctx);
}
v4l2_m2m_buf_done(src_buf, VB2_BUF_STATE_DONE);
output_size = e5010_hw_get_output_size(e5010->core_base);
vb2_set_plane_payload(&dst_buf->vb2_buf, 0, output_size + HEADER_SIZE);
v4l2_m2m_buf_done(dst_buf, VB2_BUF_STATE_DONE);
dprintk(e5010, 3,
"ctx: 0x%p frame done for dst_buf->sequence: %d src_buf->sequence: %d\n",
ctx, dst_buf->sequence, src_buf->sequence);
}
v4l2_m2m_job_finish(e5010->m2m_dev, ctx->fh.m2m_ctx);
dprintk(e5010, 3, "ctx: 0x%p Finish job\n", ctx);
job_unlock:
spin_unlock(&e5010->hw_lock);
return IRQ_HANDLED;
}
static int e5010_init_device(struct e5010_dev *e5010)
{
int ret = 0;
e5010_hw_bypass_mmu(e5010->mmu_base, 1);
if (e5010_hw_enable_auto_clock_gating(e5010->core_base, 1))
v4l2_warn(&e5010->v4l2_dev, "failed to enable auto clock gating\n");
if (e5010_hw_enable_manual_clock_gating(e5010->core_base, 0))
v4l2_warn(&e5010->v4l2_dev, "failed to disable manual clock gating\n");
if (e5010_hw_enable_crc_check(e5010->core_base, 0))
v4l2_warn(&e5010->v4l2_dev, "failed to disable CRC check\n");
if (e5010_hw_enable_output_address_error_irq(e5010->core_base, 1))
v4l2_err(&e5010->v4l2_dev, "failed to enable Output Address Error interrupts\n");
ret = e5010_hw_set_input_source_to_memory(e5010->core_base, 1);
if (ret) {
v4l2_err(&e5010->v4l2_dev, "failed to set input source to memory\n");
return ret;
}
ret = e5010_hw_enable_picture_done_irq(e5010->core_base, 1);
if (ret)
v4l2_err(&e5010->v4l2_dev, "failed to enable Picture Done interrupts\n");
return ret;
}
static int e5010_probe(struct platform_device *pdev)
{
struct e5010_dev *e5010;
int irq, ret = 0;
struct device *dev = &pdev->dev;
ret = dma_set_mask(dev, DMA_BIT_MASK(32));
if (ret)
return dev_err_probe(dev, ret, "32-bit consistent DMA enable failed\n");
e5010 = devm_kzalloc(dev, sizeof(*e5010), GFP_KERNEL);
if (!e5010)
return -ENOMEM;
platform_set_drvdata(pdev, e5010);
e5010->dev = dev;
mutex_init(&e5010->mutex);
spin_lock_init(&e5010->hw_lock);
e5010->vdev = video_device_alloc();
if (!e5010->vdev) {
dev_err(dev, "failed to allocate video device\n");
return -ENOMEM;
}
snprintf(e5010->vdev->name, sizeof(e5010->vdev->name), "%s", E5010_MODULE_NAME);
e5010->vdev->fops = &e5010_fops;
e5010->vdev->ioctl_ops = &e5010_ioctl_ops;
e5010->vdev->minor = -1;
e5010->vdev->release = video_device_release;
e5010->vdev->vfl_dir = VFL_DIR_M2M;
e5010->vdev->device_caps = V4L2_CAP_VIDEO_M2M_MPLANE | V4L2_CAP_STREAMING;
e5010->vdev->v4l2_dev = &e5010->v4l2_dev;
e5010->vdev->lock = &e5010->mutex;
ret = v4l2_device_register(dev, &e5010->v4l2_dev);
if (ret) {
dev_err_probe(dev, ret, "failed to register v4l2 device\n");
goto fail_after_video_device_alloc;
}
e5010->m2m_dev = v4l2_m2m_init(&e5010_m2m_ops);
if (IS_ERR(e5010->m2m_dev)) {
ret = PTR_ERR(e5010->m2m_dev);
e5010->m2m_dev = NULL;
dev_err_probe(dev, ret, "failed to init mem2mem device\n");
goto fail_after_v4l2_register;
}
video_set_drvdata(e5010->vdev, e5010);
e5010->core_base = devm_platform_ioremap_resource_byname(pdev, "core");
if (IS_ERR(e5010->core_base)) {
ret = PTR_ERR(e5010->core_base);
dev_err_probe(dev, ret, "Missing 'core' resources area\n");
goto fail_after_v4l2_register;
}
e5010->mmu_base = devm_platform_ioremap_resource_byname(pdev, "mmu");
if (IS_ERR(e5010->mmu_base)) {
ret = PTR_ERR(e5010->mmu_base);
dev_err_probe(dev, ret, "Missing 'mmu' resources area\n");
goto fail_after_v4l2_register;
}
e5010->last_context_run = NULL;
irq = platform_get_irq(pdev, 0);
ret = devm_request_irq(dev, irq, e5010_irq, 0,
E5010_MODULE_NAME, e5010);
if (ret) {
dev_err_probe(dev, ret, "failed to register IRQ %d\n", irq);
goto fail_after_v4l2_register;
}
e5010->clk = devm_clk_get(dev, NULL);
if (IS_ERR(e5010->clk)) {
ret = PTR_ERR(e5010->clk);
dev_err_probe(dev, ret, "failed to get clock\n");
goto fail_after_v4l2_register;
}
pm_runtime_enable(dev);
ret = video_register_device(e5010->vdev, VFL_TYPE_VIDEO, 0);
if (ret) {
dev_err_probe(dev, ret, "failed to register video device\n");
goto fail_after_video_register_device;
}
v4l2_info(&e5010->v4l2_dev, "Device registered as /dev/video%d\n",
e5010->vdev->num);
return 0;
fail_after_video_register_device:
v4l2_m2m_release(e5010->m2m_dev);
fail_after_v4l2_register:
v4l2_device_unregister(&e5010->v4l2_dev);
fail_after_video_device_alloc:
video_device_release(e5010->vdev);
return ret;
}
static void e5010_remove(struct platform_device *pdev)
{
struct e5010_dev *e5010 = platform_get_drvdata(pdev);
pm_runtime_disable(e5010->dev);
video_unregister_device(e5010->vdev);
v4l2_m2m_release(e5010->m2m_dev);
v4l2_device_unregister(&e5010->v4l2_dev);
}
static void e5010_vb2_buffers_return(struct vb2_queue *q, enum vb2_buffer_state state)
{
struct vb2_v4l2_buffer *vbuf;
struct e5010_context *ctx = vb2_get_drv_priv(q);
if (V4L2_TYPE_IS_OUTPUT(q->type)) {
while ((vbuf = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx))) {
dprintk(ctx->e5010, 2, "ctx: 0x%p, buf type %s | index %d\n",
ctx, type_name(vbuf->vb2_buf.type), vbuf->vb2_buf.index);
v4l2_m2m_buf_done(vbuf, state);
}
} else {
while ((vbuf = v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx))) {
dprintk(ctx->e5010, 2, "ctx: 0x%p, buf type %s | index %d\n",
ctx, type_name(vbuf->vb2_buf.type), vbuf->vb2_buf.index);
vb2_set_plane_payload(&vbuf->vb2_buf, 0, 0);
v4l2_m2m_buf_done(vbuf, state);
}
}
}
static int e5010_queue_setup(struct vb2_queue *vq, unsigned int *nbuffers, unsigned int *nplanes,
unsigned int sizes[], struct device *alloc_devs[])
{
struct e5010_context *ctx = vb2_get_drv_priv(vq);
struct e5010_q_data *queue;
int i;
queue = get_queue(ctx, vq->type);
if (*nplanes) {
if (*nplanes != queue->fmt->num_planes)
return -EINVAL;
for (i = 0; i < *nplanes; i++) {
if (sizes[i] < queue->sizeimage[i])
return -EINVAL;
}
return 0;
}
*nplanes = queue->fmt->num_planes;
for (i = 0; i < *nplanes; i++)
sizes[i] = queue->sizeimage[i];
dprintk(ctx->e5010, 2,
"ctx: 0x%p, type %s, buffer(s): %d, planes %d, plane1: bytes %d plane2: %d bytes\n",
ctx, type_name(vq->type), *nbuffers, *nplanes, sizes[0], sizes[1]);
return 0;
}
static void e5010_buf_finish(struct vb2_buffer *vb)
{
struct e5010_context *ctx = vb2_get_drv_priv(vb->vb2_queue);
void *d_addr;
if (vb->state != VB2_BUF_STATE_DONE || V4L2_TYPE_IS_OUTPUT(vb->vb2_queue->type))
return;
d_addr = vb2_plane_vaddr(vb, 0);
write_header(ctx, d_addr);
}
static int e5010_buf_out_validate(struct vb2_buffer *vb)
{
struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
struct e5010_context *ctx = vb2_get_drv_priv(vb->vb2_queue);
if (vbuf->field != V4L2_FIELD_NONE)
dprintk(ctx->e5010, 1, "ctx: 0x%p, field isn't supported\n", ctx);
vbuf->field = V4L2_FIELD_NONE;
return 0;
}
static int e5010_buf_prepare(struct vb2_buffer *vb)
{
struct e5010_context *ctx = vb2_get_drv_priv(vb->vb2_queue);
struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
struct e5010_q_data *queue;
int i;
vbuf->field = V4L2_FIELD_NONE;
queue = get_queue(ctx, vb->vb2_queue->type);
for (i = 0; i < queue->fmt->num_planes; i++) {
if (vb2_plane_size(vb, i) < (unsigned long)queue->sizeimage[i]) {
v4l2_err(&ctx->e5010->v4l2_dev, "plane %d too small (%lu < %lu)", i,
vb2_plane_size(vb, i), (unsigned long)queue->sizeimage[i]);
return -EINVAL;
}
}
if (V4L2_TYPE_IS_CAPTURE(vb->vb2_queue->type)) {
vb2_set_plane_payload(vb, 0, 0);
vb2_set_plane_payload(vb, 1, 0);
}
return 0;
}
static void e5010_buf_queue(struct vb2_buffer *vb)
{
struct e5010_context *ctx = vb2_get_drv_priv(vb->vb2_queue);
struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
if (V4L2_TYPE_IS_CAPTURE(vb->vb2_queue->type) &&
vb2_is_streaming(vb->vb2_queue) &&
v4l2_m2m_dst_buf_is_last(ctx->fh.m2m_ctx)) {
struct e5010_q_data *queue = get_queue(ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE);
vbuf->sequence = queue->sequence++;
v4l2_m2m_last_buffer_done(ctx->fh.m2m_ctx, vbuf);
v4l2_event_queue_fh(&ctx->fh, &e5010_eos_event);
return;
}
v4l2_m2m_buf_queue(ctx->fh.m2m_ctx, vbuf);
}
static int e5010_encoder_cmd(struct file *file, void *priv,
struct v4l2_encoder_cmd *cmd)
{
struct e5010_context *ctx = to_e5010_context(file);
int ret;
struct vb2_queue *cap_vq;
cap_vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE);
ret = v4l2_m2m_ioctl_try_encoder_cmd(file, &ctx->fh, cmd);
if (ret < 0)
return ret;
if (!vb2_is_streaming(v4l2_m2m_get_src_vq(ctx->fh.m2m_ctx)) ||
!vb2_is_streaming(v4l2_m2m_get_dst_vq(ctx->fh.m2m_ctx)))
return 0;
ret = v4l2_m2m_ioctl_encoder_cmd(file, &ctx->fh, cmd);
if (ret < 0)
return ret;
if (cmd->cmd == V4L2_ENC_CMD_STOP &&
v4l2_m2m_has_stopped(ctx->fh.m2m_ctx))
v4l2_event_queue_fh(&ctx->fh, &e5010_eos_event);
if (cmd->cmd == V4L2_ENC_CMD_START &&
v4l2_m2m_has_stopped(ctx->fh.m2m_ctx))
vb2_clear_last_buffer_dequeued(cap_vq);
return 0;
}
static int e5010_start_streaming(struct vb2_queue *q, unsigned int count)
{
struct e5010_context *ctx = vb2_get_drv_priv(q);
int ret;
struct e5010_q_data *queue = get_queue(ctx, q->type);
v4l2_m2m_update_start_streaming_state(ctx->fh.m2m_ctx, q);
queue->sequence = 0;
ret = pm_runtime_resume_and_get(ctx->e5010->dev);
if (ret < 0) {
v4l2_err(&ctx->e5010->v4l2_dev, "failed to power up jpeg\n");
goto fail;
}
ret = e5010_init_device(ctx->e5010);
if (ret) {
v4l2_err(&ctx->e5010->v4l2_dev, "failed to Enable e5010 device\n");
goto fail;
}
return 0;
fail:
e5010_vb2_buffers_return(q, VB2_BUF_STATE_QUEUED);
return ret;
}
static void e5010_stop_streaming(struct vb2_queue *q)
{
struct e5010_context *ctx = vb2_get_drv_priv(q);
e5010_vb2_buffers_return(q, VB2_BUF_STATE_ERROR);
if (V4L2_TYPE_IS_OUTPUT(q->type))
v4l2_m2m_update_stop_streaming_state(ctx->fh.m2m_ctx, q);
if (V4L2_TYPE_IS_OUTPUT(q->type) &&
v4l2_m2m_has_stopped(ctx->fh.m2m_ctx)) {
v4l2_event_queue_fh(&ctx->fh, &e5010_eos_event);
}
pm_runtime_put_sync(ctx->e5010->dev);
}
static void e5010_device_run(void *priv)
{
struct e5010_context *ctx = priv;
struct e5010_dev *e5010 = ctx->e5010;
struct vb2_v4l2_buffer *s_vb, *d_vb;
u32 reg = 0;
int ret = 0, luma_crop_offset = 0, chroma_crop_offset = 0;
unsigned long flags;
int num_planes = ctx->out_queue.fmt->num_planes;
spin_lock_irqsave(&e5010->hw_lock, flags);
s_vb = v4l2_m2m_next_src_buf(ctx->fh.m2m_ctx);
WARN_ON(!s_vb);
d_vb = v4l2_m2m_next_dst_buf(ctx->fh.m2m_ctx);
WARN_ON(!d_vb);
if (!s_vb || !d_vb)
goto no_ready_buf_err;
s_vb->sequence = ctx->out_queue.sequence++;
d_vb->sequence = ctx->cap_queue.sequence++;
v4l2_m2m_buf_copy_metadata(s_vb, d_vb);
if (ctx != e5010->last_context_run || ctx->update_qp) {
dprintk(e5010, 1, "ctx updated: 0x%p -> 0x%p, updating qp tables\n",
e5010->last_context_run, ctx);
ret = update_qp_tables(ctx);
}
if (ret) {
ctx->update_qp = true;
v4l2_err(&e5010->v4l2_dev, "failed to update QP tables\n");
goto device_busy_err;
} else {
e5010->last_context_run = ctx;
ctx->update_qp = false;
}
reg = (u32)vb2_dma_contig_plane_dma_addr(&s_vb->vb2_buf, 0);
if (ctx->out_queue.crop_set) {
luma_crop_offset = ctx->out_queue.bytesperline[0] * ctx->out_queue.crop.top +
ctx->out_queue.crop.left;
if (ctx->out_queue.fmt->subsampling == V4L2_JPEG_CHROMA_SUBSAMPLING_422) {
chroma_crop_offset =
ctx->out_queue.bytesperline[0] * ctx->out_queue.crop.top
+ ctx->out_queue.crop.left;
} else {
chroma_crop_offset =
ctx->out_queue.bytesperline[0] * ctx->out_queue.crop.top / 2
+ ctx->out_queue.crop.left;
}
dprintk(e5010, 1, "Luma crop offset : %x, chroma crop offset : %x\n",
luma_crop_offset, chroma_crop_offset);
}
ret = e5010_hw_set_input_luma_addr(e5010->core_base, reg + luma_crop_offset);
if (ret || !reg) {
v4l2_err(&e5010->v4l2_dev, "failed to set input luma address\n");
goto device_busy_err;
}
if (num_planes == 1)
reg += (ctx->out_queue.bytesperline[0]) * (ctx->out_queue.height);
else
reg = (u32)vb2_dma_contig_plane_dma_addr(&s_vb->vb2_buf, 1);
dprintk(e5010, 3,
"ctx: 0x%p, luma_addr: 0x%x, chroma_addr: 0x%x, out_addr: 0x%x\n",
ctx, (u32)vb2_dma_contig_plane_dma_addr(&s_vb->vb2_buf, 0) + luma_crop_offset,
reg + chroma_crop_offset, (u32)vb2_dma_contig_plane_dma_addr(&d_vb->vb2_buf, 0));
dprintk(e5010, 3,
"ctx: 0x%p, buf indices: src_index: %d, dst_index: %d\n",
ctx, s_vb->vb2_buf.index, d_vb->vb2_buf.index);
ret = e5010_hw_set_input_chroma_addr(e5010->core_base, reg + chroma_crop_offset);
if (ret || !reg) {
v4l2_err(&e5010->v4l2_dev, "failed to set input chroma address\n");
goto device_busy_err;
}
reg = (u32)vb2_dma_contig_plane_dma_addr(&d_vb->vb2_buf, 0);
reg += HEADER_SIZE;
ret = e5010_hw_set_output_base_addr(e5010->core_base, reg);
if (ret || !reg) {
v4l2_err(&e5010->v4l2_dev, "failed to set output base address\n");
goto device_busy_err;
}
ret = e5010_hw_set_horizontal_size(e5010->core_base, ctx->out_queue.crop.width - 1);
if (ret) {
v4l2_err(&e5010->v4l2_dev, "failed to set input width\n");
goto device_busy_err;
}
ret = e5010_hw_set_vertical_size(e5010->core_base, ctx->out_queue.crop.height - 1);
if (ret) {
v4l2_err(&e5010->v4l2_dev, "failed to set input width\n");
goto device_busy_err;
}
ret = e5010_hw_set_luma_stride(e5010->core_base, ctx->out_queue.bytesperline[0]);
if (ret) {
v4l2_err(&e5010->v4l2_dev, "failed to set luma stride\n");
goto device_busy_err;
}
ret = e5010_hw_set_chroma_stride(e5010->core_base, ctx->out_queue.bytesperline[0]);
if (ret) {
v4l2_err(&e5010->v4l2_dev, "failed to set chroma stride\n");
goto device_busy_err;
}
ret = e5010_set_input_subsampling(e5010->core_base, ctx->out_queue.fmt->subsampling);
if (ret) {
v4l2_err(&e5010->v4l2_dev, "failed to set input subsampling\n");
goto device_busy_err;
}
ret = e5010_hw_set_chroma_order(e5010->core_base, ctx->out_queue.fmt->chroma_order);
if (ret) {
v4l2_err(&e5010->v4l2_dev, "failed to set chroma order\n");
goto device_busy_err;
}
e5010_hw_set_output_max_size(e5010->core_base, d_vb->planes[0].length);
e5010_hw_encode_start(e5010->core_base, 1);
spin_unlock_irqrestore(&e5010->hw_lock, flags);
return;
device_busy_err:
e5010_reset(e5010->dev, e5010->core_base, e5010->mmu_base);
no_ready_buf_err:
if (s_vb) {
v4l2_m2m_src_buf_remove_by_buf(ctx->fh.m2m_ctx, s_vb);
v4l2_m2m_buf_done(s_vb, VB2_BUF_STATE_ERROR);
}
if (d_vb) {
v4l2_m2m_dst_buf_remove_by_buf(ctx->fh.m2m_ctx, d_vb);
vb2_set_plane_payload(&d_vb->vb2_buf, 0, 1);
v4l2_m2m_buf_done(d_vb, VB2_BUF_STATE_ERROR);
}
v4l2_m2m_job_finish(e5010->m2m_dev, ctx->fh.m2m_ctx);
spin_unlock_irqrestore(&e5010->hw_lock, flags);
}
#ifdef CONFIG_PM
static int e5010_runtime_resume(struct device *dev)
{
struct e5010_dev *e5010 = dev_get_drvdata(dev);
int ret;
ret = clk_prepare_enable(e5010->clk);
if (ret < 0) {
v4l2_err(&e5010->v4l2_dev, "failed to enable clock\n");
return ret;
}
return 0;
}
static int e5010_runtime_suspend(struct device *dev)
{
struct e5010_dev *e5010 = dev_get_drvdata(dev);
clk_disable_unprepare(e5010->clk);
return 0;
}
#endif
#ifdef CONFIG_PM_SLEEP
static int e5010_suspend(struct device *dev)
{
struct e5010_dev *e5010 = dev_get_drvdata(dev);
v4l2_m2m_suspend(e5010->m2m_dev);
return pm_runtime_force_suspend(dev);
}
static int e5010_resume(struct device *dev)
{
struct e5010_dev *e5010 = dev_get_drvdata(dev);
int ret;
ret = pm_runtime_force_resume(dev);
if (ret < 0)
return ret;
ret = e5010_init_device(e5010);
if (ret) {
dev_err(dev, "Failed to re-enable e5010 device\n");
return ret;
}
v4l2_m2m_resume(e5010->m2m_dev);
return ret;
}
#endif
static const struct dev_pm_ops e5010_pm_ops = {
SET_RUNTIME_PM_OPS(e5010_runtime_suspend,
e5010_runtime_resume, NULL)
SET_SYSTEM_SLEEP_PM_OPS(e5010_suspend, e5010_resume)
};
static const struct v4l2_ioctl_ops e5010_ioctl_ops = {
.vidioc_querycap = e5010_querycap,
.vidioc_enum_fmt_vid_cap = e5010_enum_fmt,
.vidioc_g_fmt_vid_cap_mplane = e5010_g_fmt,
.vidioc_try_fmt_vid_cap_mplane = e5010_try_fmt,
.vidioc_s_fmt_vid_cap_mplane = e5010_s_fmt,
.vidioc_enum_fmt_vid_out = e5010_enum_fmt,
.vidioc_g_fmt_vid_out_mplane = e5010_g_fmt,
.vidioc_try_fmt_vid_out_mplane = e5010_try_fmt,
.vidioc_s_fmt_vid_out_mplane = e5010_s_fmt,
.vidioc_g_selection = e5010_g_selection,
.vidioc_s_selection = e5010_s_selection,
.vidioc_reqbufs = v4l2_m2m_ioctl_reqbufs,
.vidioc_querybuf = v4l2_m2m_ioctl_querybuf,
.vidioc_qbuf = v4l2_m2m_ioctl_qbuf,
.vidioc_dqbuf = v4l2_m2m_ioctl_dqbuf,
.vidioc_expbuf = v4l2_m2m_ioctl_expbuf,
.vidioc_create_bufs = v4l2_m2m_ioctl_create_bufs,
.vidioc_prepare_buf = v4l2_m2m_ioctl_prepare_buf,
.vidioc_streamon = v4l2_m2m_ioctl_streamon,
.vidioc_streamoff = v4l2_m2m_ioctl_streamoff,
.vidioc_log_status = v4l2_ctrl_log_status,
.vidioc_subscribe_event = e5010_subscribe_event,
.vidioc_unsubscribe_event = v4l2_event_unsubscribe,
.vidioc_try_encoder_cmd = v4l2_m2m_ioctl_try_encoder_cmd,
.vidioc_encoder_cmd = e5010_encoder_cmd,
.vidioc_enum_framesizes = e5010_enum_framesizes,
};
static const struct vb2_ops e5010_video_ops = {
.queue_setup = e5010_queue_setup,
.buf_queue = e5010_buf_queue,
.buf_finish = e5010_buf_finish,
.buf_prepare = e5010_buf_prepare,
.buf_out_validate = e5010_buf_out_validate,
.start_streaming = e5010_start_streaming,
.stop_streaming = e5010_stop_streaming,
};
static const struct v4l2_file_operations e5010_fops = {
.owner = THIS_MODULE,
.open = e5010_open,
.release = e5010_release,
.poll = v4l2_m2m_fop_poll,
.unlocked_ioctl = video_ioctl2,
.mmap = v4l2_m2m_fop_mmap,
};
static const struct v4l2_m2m_ops e5010_m2m_ops = {
.device_run = e5010_device_run,
};
static const struct of_device_id e5010_of_match[] = {
{.compatible = "img,e5010-jpeg-enc"}, { },
};
MODULE_DEVICE_TABLE(of, e5010_of_match);
static struct platform_driver e5010_driver = {
.probe = e5010_probe,
.remove = e5010_remove,
.driver = {
.name = E5010_MODULE_NAME,
.of_match_table = e5010_of_match,
.pm = &e5010_pm_ops,
},
};
module_platform_driver(e5010_driver);
MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("Imagination E5010 JPEG encoder driver");
