forked from Qortal/Brooklyn
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
2237 lines
61 KiB
2237 lines
61 KiB
// SPDX-License-Identifier: GPL-2.0-only |
|
/* |
|
* A virtual codec example device. |
|
* |
|
* Copyright 2018 Cisco Systems, Inc. and/or its affiliates. All rights reserved. |
|
* |
|
* This is a virtual codec device driver for testing the codec framework. |
|
* It simulates a device that uses memory buffers for both source and |
|
* destination and encodes or decodes the data. |
|
*/ |
|
|
|
#include <linux/module.h> |
|
#include <linux/delay.h> |
|
#include <linux/fs.h> |
|
#include <linux/sched.h> |
|
#include <linux/slab.h> |
|
|
|
#include <linux/platform_device.h> |
|
#include <media/v4l2-mem2mem.h> |
|
#include <media/v4l2-device.h> |
|
#include <media/v4l2-ioctl.h> |
|
#include <media/v4l2-ctrls.h> |
|
#include <media/v4l2-event.h> |
|
#include <media/videobuf2-vmalloc.h> |
|
|
|
#include "codec-v4l2-fwht.h" |
|
|
|
MODULE_DESCRIPTION("Virtual codec device"); |
|
MODULE_AUTHOR("Hans Verkuil <[email protected]>"); |
|
MODULE_LICENSE("GPL v2"); |
|
|
|
static bool multiplanar; |
|
module_param(multiplanar, bool, 0444); |
|
MODULE_PARM_DESC(multiplanar, |
|
" use multi-planar API instead of single-planar API"); |
|
|
|
static unsigned int debug; |
|
module_param(debug, uint, 0644); |
|
MODULE_PARM_DESC(debug, " activates debug info"); |
|
|
|
#define VICODEC_NAME "vicodec" |
|
#define MAX_WIDTH 4096U |
|
#define MIN_WIDTH 640U |
|
#define MAX_HEIGHT 2160U |
|
#define MIN_HEIGHT 360U |
|
|
|
#define dprintk(dev, fmt, arg...) \ |
|
v4l2_dbg(1, debug, &dev->v4l2_dev, "%s: " fmt, __func__, ## arg) |
|
|
|
|
|
struct pixfmt_info { |
|
u32 id; |
|
unsigned int bytesperline_mult; |
|
unsigned int sizeimage_mult; |
|
unsigned int sizeimage_div; |
|
unsigned int luma_step; |
|
unsigned int chroma_step; |
|
/* Chroma plane subsampling */ |
|
unsigned int width_div; |
|
unsigned int height_div; |
|
}; |
|
|
|
static const struct v4l2_fwht_pixfmt_info pixfmt_fwht = { |
|
V4L2_PIX_FMT_FWHT, 0, 3, 1, 1, 1, 1, 1, 0, 1 |
|
}; |
|
|
|
static const struct v4l2_fwht_pixfmt_info pixfmt_stateless_fwht = { |
|
V4L2_PIX_FMT_FWHT_STATELESS, 0, 3, 1, 1, 1, 1, 1, 0, 1 |
|
}; |
|
|
|
static void vicodec_dev_release(struct device *dev) |
|
{ |
|
} |
|
|
|
static struct platform_device vicodec_pdev = { |
|
.name = VICODEC_NAME, |
|
.dev.release = vicodec_dev_release, |
|
}; |
|
|
|
/* Per-queue, driver-specific private data */ |
|
struct vicodec_q_data { |
|
unsigned int coded_width; |
|
unsigned int coded_height; |
|
unsigned int visible_width; |
|
unsigned int visible_height; |
|
unsigned int sizeimage; |
|
unsigned int vb2_sizeimage; |
|
unsigned int sequence; |
|
const struct v4l2_fwht_pixfmt_info *info; |
|
}; |
|
|
|
enum { |
|
V4L2_M2M_SRC = 0, |
|
V4L2_M2M_DST = 1, |
|
}; |
|
|
|
struct vicodec_dev_instance { |
|
struct video_device vfd; |
|
struct mutex mutex; |
|
spinlock_t lock; |
|
struct v4l2_m2m_dev *m2m_dev; |
|
}; |
|
|
|
struct vicodec_dev { |
|
struct v4l2_device v4l2_dev; |
|
struct vicodec_dev_instance stateful_enc; |
|
struct vicodec_dev_instance stateful_dec; |
|
struct vicodec_dev_instance stateless_dec; |
|
#ifdef CONFIG_MEDIA_CONTROLLER |
|
struct media_device mdev; |
|
#endif |
|
|
|
}; |
|
|
|
struct vicodec_ctx { |
|
struct v4l2_fh fh; |
|
struct vicodec_dev *dev; |
|
bool is_enc; |
|
bool is_stateless; |
|
spinlock_t *lock; |
|
|
|
struct v4l2_ctrl_handler hdl; |
|
|
|
/* Source and destination queue data */ |
|
struct vicodec_q_data q_data[2]; |
|
struct v4l2_fwht_state state; |
|
|
|
u32 cur_buf_offset; |
|
u32 comp_max_size; |
|
u32 comp_size; |
|
u32 header_size; |
|
u32 comp_magic_cnt; |
|
bool comp_has_frame; |
|
bool comp_has_next_frame; |
|
bool first_source_change_sent; |
|
bool source_changed; |
|
}; |
|
|
|
static const struct v4l2_event vicodec_eos_event = { |
|
.type = V4L2_EVENT_EOS |
|
}; |
|
|
|
static inline struct vicodec_ctx *file2ctx(struct file *file) |
|
{ |
|
return container_of(file->private_data, struct vicodec_ctx, fh); |
|
} |
|
|
|
static struct vicodec_q_data *get_q_data(struct vicodec_ctx *ctx, |
|
enum v4l2_buf_type type) |
|
{ |
|
switch (type) { |
|
case V4L2_BUF_TYPE_VIDEO_OUTPUT: |
|
case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE: |
|
return &ctx->q_data[V4L2_M2M_SRC]; |
|
case V4L2_BUF_TYPE_VIDEO_CAPTURE: |
|
case V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE: |
|
return &ctx->q_data[V4L2_M2M_DST]; |
|
default: |
|
break; |
|
} |
|
return NULL; |
|
} |
|
|
|
static void copy_cap_to_ref(const u8 *cap, const struct v4l2_fwht_pixfmt_info *info, |
|
struct v4l2_fwht_state *state) |
|
{ |
|
int plane_idx; |
|
u8 *p_ref = state->ref_frame.buf; |
|
unsigned int cap_stride = state->stride; |
|
unsigned int ref_stride = state->ref_stride; |
|
|
|
for (plane_idx = 0; plane_idx < info->planes_num; plane_idx++) { |
|
int i; |
|
unsigned int h_div = (plane_idx == 1 || plane_idx == 2) ? |
|
info->height_div : 1; |
|
const u8 *row_cap = cap; |
|
u8 *row_ref = p_ref; |
|
|
|
if (info->planes_num == 3 && plane_idx == 1) { |
|
cap_stride /= 2; |
|
ref_stride /= 2; |
|
} |
|
|
|
if (plane_idx == 1 && |
|
(info->id == V4L2_PIX_FMT_NV24 || |
|
info->id == V4L2_PIX_FMT_NV42)) { |
|
cap_stride *= 2; |
|
ref_stride *= 2; |
|
} |
|
|
|
for (i = 0; i < state->visible_height / h_div; i++) { |
|
memcpy(row_ref, row_cap, ref_stride); |
|
row_ref += ref_stride; |
|
row_cap += cap_stride; |
|
} |
|
cap += cap_stride * (state->coded_height / h_div); |
|
p_ref += ref_stride * (state->coded_height / h_div); |
|
} |
|
} |
|
|
|
static bool validate_by_version(unsigned int flags, unsigned int version) |
|
{ |
|
if (!version || version > V4L2_FWHT_VERSION) |
|
return false; |
|
|
|
if (version >= 2) { |
|
unsigned int components_num = 1 + |
|
((flags & V4L2_FWHT_FL_COMPONENTS_NUM_MSK) >> |
|
V4L2_FWHT_FL_COMPONENTS_NUM_OFFSET); |
|
unsigned int pixenc = flags & V4L2_FWHT_FL_PIXENC_MSK; |
|
|
|
if (components_num == 0 || components_num > 4 || !pixenc) |
|
return false; |
|
} |
|
return true; |
|
} |
|
|
|
static bool validate_stateless_params_flags(const struct v4l2_ctrl_fwht_params *params, |
|
const struct v4l2_fwht_pixfmt_info *cur_info) |
|
{ |
|
unsigned int width_div = |
|
(params->flags & V4L2_FWHT_FL_CHROMA_FULL_WIDTH) ? 1 : 2; |
|
unsigned int height_div = |
|
(params->flags & V4L2_FWHT_FL_CHROMA_FULL_HEIGHT) ? 1 : 2; |
|
unsigned int components_num = 3; |
|
unsigned int pixenc = 0; |
|
|
|
if (params->version < 3) |
|
return false; |
|
|
|
components_num = 1 + ((params->flags & V4L2_FWHT_FL_COMPONENTS_NUM_MSK) >> |
|
V4L2_FWHT_FL_COMPONENTS_NUM_OFFSET); |
|
pixenc = (params->flags & V4L2_FWHT_FL_PIXENC_MSK); |
|
if (v4l2_fwht_validate_fmt(cur_info, width_div, height_div, |
|
components_num, pixenc)) |
|
return true; |
|
return false; |
|
} |
|
|
|
|
|
static void update_state_from_header(struct vicodec_ctx *ctx) |
|
{ |
|
const struct fwht_cframe_hdr *p_hdr = &ctx->state.header; |
|
|
|
ctx->state.visible_width = ntohl(p_hdr->width); |
|
ctx->state.visible_height = ntohl(p_hdr->height); |
|
ctx->state.colorspace = ntohl(p_hdr->colorspace); |
|
ctx->state.xfer_func = ntohl(p_hdr->xfer_func); |
|
ctx->state.ycbcr_enc = ntohl(p_hdr->ycbcr_enc); |
|
ctx->state.quantization = ntohl(p_hdr->quantization); |
|
} |
|
|
|
static int device_process(struct vicodec_ctx *ctx, |
|
struct vb2_v4l2_buffer *src_vb, |
|
struct vb2_v4l2_buffer *dst_vb) |
|
{ |
|
struct vicodec_dev *dev = ctx->dev; |
|
struct v4l2_fwht_state *state = &ctx->state; |
|
u8 *p_src, *p_dst; |
|
int ret = 0; |
|
|
|
if (ctx->is_enc || ctx->is_stateless) |
|
p_src = vb2_plane_vaddr(&src_vb->vb2_buf, 0); |
|
else |
|
p_src = state->compressed_frame; |
|
|
|
if (ctx->is_stateless) { |
|
struct media_request *src_req = src_vb->vb2_buf.req_obj.req; |
|
|
|
ret = v4l2_ctrl_request_setup(src_req, &ctx->hdl); |
|
if (ret) |
|
return ret; |
|
update_state_from_header(ctx); |
|
|
|
ctx->state.header.size = |
|
htonl(vb2_get_plane_payload(&src_vb->vb2_buf, 0)); |
|
/* |
|
* set the reference buffer from the reference timestamp |
|
* only if this is a P-frame |
|
*/ |
|
if (!(ntohl(ctx->state.header.flags) & V4L2_FWHT_FL_I_FRAME)) { |
|
struct vb2_buffer *ref_vb2_buf; |
|
int ref_buf_idx; |
|
struct vb2_queue *vq_cap = |
|
v4l2_m2m_get_vq(ctx->fh.m2m_ctx, |
|
V4L2_BUF_TYPE_VIDEO_CAPTURE); |
|
|
|
ref_buf_idx = vb2_find_timestamp(vq_cap, |
|
ctx->state.ref_frame_ts, 0); |
|
if (ref_buf_idx < 0) |
|
return -EINVAL; |
|
|
|
ref_vb2_buf = vq_cap->bufs[ref_buf_idx]; |
|
if (ref_vb2_buf->state == VB2_BUF_STATE_ERROR) |
|
ret = -EINVAL; |
|
ctx->state.ref_frame.buf = |
|
vb2_plane_vaddr(ref_vb2_buf, 0); |
|
} else { |
|
ctx->state.ref_frame.buf = NULL; |
|
} |
|
} |
|
p_dst = vb2_plane_vaddr(&dst_vb->vb2_buf, 0); |
|
if (!p_src || !p_dst) { |
|
v4l2_err(&dev->v4l2_dev, |
|
"Acquiring kernel pointers to buffers failed\n"); |
|
return -EFAULT; |
|
} |
|
|
|
if (ctx->is_enc) { |
|
struct vicodec_q_data *q_src; |
|
int comp_sz_or_errcode; |
|
|
|
q_src = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT); |
|
state->info = q_src->info; |
|
comp_sz_or_errcode = v4l2_fwht_encode(state, p_src, p_dst); |
|
if (comp_sz_or_errcode < 0) |
|
return comp_sz_or_errcode; |
|
vb2_set_plane_payload(&dst_vb->vb2_buf, 0, comp_sz_or_errcode); |
|
} else { |
|
struct vicodec_q_data *q_dst; |
|
unsigned int comp_frame_size = ntohl(ctx->state.header.size); |
|
|
|
q_dst = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE); |
|
if (comp_frame_size > ctx->comp_max_size) |
|
return -EINVAL; |
|
state->info = q_dst->info; |
|
ret = v4l2_fwht_decode(state, p_src, p_dst); |
|
if (ret < 0) |
|
return ret; |
|
if (!ctx->is_stateless) |
|
copy_cap_to_ref(p_dst, ctx->state.info, &ctx->state); |
|
|
|
vb2_set_plane_payload(&dst_vb->vb2_buf, 0, q_dst->sizeimage); |
|
if (ntohl(ctx->state.header.flags) & V4L2_FWHT_FL_I_FRAME) |
|
dst_vb->flags |= V4L2_BUF_FLAG_KEYFRAME; |
|
else |
|
dst_vb->flags |= V4L2_BUF_FLAG_PFRAME; |
|
} |
|
return ret; |
|
} |
|
|
|
/* |
|
* mem2mem callbacks |
|
*/ |
|
static enum vb2_buffer_state get_next_header(struct vicodec_ctx *ctx, |
|
u8 **pp, u32 sz) |
|
{ |
|
static const u8 magic[] = { |
|
0x4f, 0x4f, 0x4f, 0x4f, 0xff, 0xff, 0xff, 0xff |
|
}; |
|
u8 *p = *pp; |
|
u32 state; |
|
u8 *header = (u8 *)&ctx->state.header; |
|
|
|
state = VB2_BUF_STATE_DONE; |
|
|
|
if (!ctx->header_size) { |
|
state = VB2_BUF_STATE_ERROR; |
|
for (; p < *pp + sz; p++) { |
|
u32 copy; |
|
|
|
p = memchr(p, magic[ctx->comp_magic_cnt], |
|
*pp + sz - p); |
|
if (!p) { |
|
ctx->comp_magic_cnt = 0; |
|
p = *pp + sz; |
|
break; |
|
} |
|
copy = sizeof(magic) - ctx->comp_magic_cnt; |
|
if (*pp + sz - p < copy) |
|
copy = *pp + sz - p; |
|
|
|
memcpy(header + ctx->comp_magic_cnt, p, copy); |
|
ctx->comp_magic_cnt += copy; |
|
if (!memcmp(header, magic, ctx->comp_magic_cnt)) { |
|
p += copy; |
|
state = VB2_BUF_STATE_DONE; |
|
break; |
|
} |
|
ctx->comp_magic_cnt = 0; |
|
} |
|
if (ctx->comp_magic_cnt < sizeof(magic)) { |
|
*pp = p; |
|
return state; |
|
} |
|
ctx->header_size = sizeof(magic); |
|
} |
|
|
|
if (ctx->header_size < sizeof(struct fwht_cframe_hdr)) { |
|
u32 copy = sizeof(struct fwht_cframe_hdr) - ctx->header_size; |
|
|
|
if (*pp + sz - p < copy) |
|
copy = *pp + sz - p; |
|
|
|
memcpy(header + ctx->header_size, p, copy); |
|
p += copy; |
|
ctx->header_size += copy; |
|
} |
|
*pp = p; |
|
return state; |
|
} |
|
|
|
/* device_run() - prepares and starts the device */ |
|
static void device_run(void *priv) |
|
{ |
|
struct vicodec_ctx *ctx = priv; |
|
struct vicodec_dev *dev = ctx->dev; |
|
struct vb2_v4l2_buffer *src_buf, *dst_buf; |
|
struct vicodec_q_data *q_src, *q_dst; |
|
u32 state; |
|
struct media_request *src_req; |
|
|
|
src_buf = v4l2_m2m_next_src_buf(ctx->fh.m2m_ctx); |
|
dst_buf = v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx); |
|
src_req = src_buf->vb2_buf.req_obj.req; |
|
|
|
q_src = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT); |
|
q_dst = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE); |
|
|
|
state = VB2_BUF_STATE_DONE; |
|
if (device_process(ctx, src_buf, dst_buf)) |
|
state = VB2_BUF_STATE_ERROR; |
|
else |
|
dst_buf->sequence = q_dst->sequence++; |
|
dst_buf->flags &= ~V4L2_BUF_FLAG_LAST; |
|
v4l2_m2m_buf_copy_metadata(src_buf, dst_buf, false); |
|
|
|
spin_lock(ctx->lock); |
|
if (!ctx->comp_has_next_frame && |
|
v4l2_m2m_is_last_draining_src_buf(ctx->fh.m2m_ctx, src_buf)) { |
|
dst_buf->flags |= V4L2_BUF_FLAG_LAST; |
|
v4l2_event_queue_fh(&ctx->fh, &vicodec_eos_event); |
|
v4l2_m2m_mark_stopped(ctx->fh.m2m_ctx); |
|
} |
|
if (ctx->is_enc || ctx->is_stateless) { |
|
src_buf->sequence = q_src->sequence++; |
|
src_buf = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx); |
|
v4l2_m2m_buf_done(src_buf, state); |
|
} else if (vb2_get_plane_payload(&src_buf->vb2_buf, 0) == ctx->cur_buf_offset) { |
|
src_buf->sequence = q_src->sequence++; |
|
src_buf = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx); |
|
v4l2_m2m_buf_done(src_buf, state); |
|
ctx->cur_buf_offset = 0; |
|
ctx->comp_has_next_frame = false; |
|
} |
|
v4l2_m2m_buf_done(dst_buf, state); |
|
|
|
ctx->comp_size = 0; |
|
ctx->header_size = 0; |
|
ctx->comp_magic_cnt = 0; |
|
ctx->comp_has_frame = false; |
|
spin_unlock(ctx->lock); |
|
if (ctx->is_stateless && src_req) |
|
v4l2_ctrl_request_complete(src_req, &ctx->hdl); |
|
|
|
if (ctx->is_enc) |
|
v4l2_m2m_job_finish(dev->stateful_enc.m2m_dev, ctx->fh.m2m_ctx); |
|
else if (ctx->is_stateless) |
|
v4l2_m2m_job_finish(dev->stateless_dec.m2m_dev, |
|
ctx->fh.m2m_ctx); |
|
else |
|
v4l2_m2m_job_finish(dev->stateful_dec.m2m_dev, ctx->fh.m2m_ctx); |
|
} |
|
|
|
static void job_remove_src_buf(struct vicodec_ctx *ctx, u32 state) |
|
{ |
|
struct vb2_v4l2_buffer *src_buf; |
|
struct vicodec_q_data *q_src; |
|
|
|
q_src = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT); |
|
spin_lock(ctx->lock); |
|
src_buf = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx); |
|
src_buf->sequence = q_src->sequence++; |
|
v4l2_m2m_buf_done(src_buf, state); |
|
ctx->cur_buf_offset = 0; |
|
spin_unlock(ctx->lock); |
|
} |
|
|
|
static const struct v4l2_fwht_pixfmt_info * |
|
info_from_header(const struct fwht_cframe_hdr *p_hdr) |
|
{ |
|
unsigned int flags = ntohl(p_hdr->flags); |
|
unsigned int width_div = (flags & V4L2_FWHT_FL_CHROMA_FULL_WIDTH) ? 1 : 2; |
|
unsigned int height_div = (flags & V4L2_FWHT_FL_CHROMA_FULL_HEIGHT) ? 1 : 2; |
|
unsigned int components_num = 3; |
|
unsigned int pixenc = 0; |
|
unsigned int version = ntohl(p_hdr->version); |
|
|
|
if (version >= 2) { |
|
components_num = 1 + ((flags & V4L2_FWHT_FL_COMPONENTS_NUM_MSK) >> |
|
V4L2_FWHT_FL_COMPONENTS_NUM_OFFSET); |
|
pixenc = (flags & V4L2_FWHT_FL_PIXENC_MSK); |
|
} |
|
return v4l2_fwht_find_nth_fmt(width_div, height_div, |
|
components_num, pixenc, 0); |
|
} |
|
|
|
static bool is_header_valid(const struct fwht_cframe_hdr *p_hdr) |
|
{ |
|
const struct v4l2_fwht_pixfmt_info *info; |
|
unsigned int w = ntohl(p_hdr->width); |
|
unsigned int h = ntohl(p_hdr->height); |
|
unsigned int version = ntohl(p_hdr->version); |
|
unsigned int flags = ntohl(p_hdr->flags); |
|
|
|
if (w < MIN_WIDTH || w > MAX_WIDTH || h < MIN_HEIGHT || h > MAX_HEIGHT) |
|
return false; |
|
|
|
if (!validate_by_version(flags, version)) |
|
return false; |
|
|
|
info = info_from_header(p_hdr); |
|
if (!info) |
|
return false; |
|
return true; |
|
} |
|
|
|
static void update_capture_data_from_header(struct vicodec_ctx *ctx) |
|
{ |
|
struct vicodec_q_data *q_dst = get_q_data(ctx, |
|
V4L2_BUF_TYPE_VIDEO_CAPTURE); |
|
const struct fwht_cframe_hdr *p_hdr = &ctx->state.header; |
|
const struct v4l2_fwht_pixfmt_info *info = info_from_header(p_hdr); |
|
unsigned int flags = ntohl(p_hdr->flags); |
|
unsigned int hdr_width_div = (flags & V4L2_FWHT_FL_CHROMA_FULL_WIDTH) ? 1 : 2; |
|
unsigned int hdr_height_div = (flags & V4L2_FWHT_FL_CHROMA_FULL_HEIGHT) ? 1 : 2; |
|
|
|
/* |
|
* This function should not be used by a stateless codec since |
|
* it changes values in q_data that are not request specific |
|
*/ |
|
WARN_ON(ctx->is_stateless); |
|
|
|
q_dst->info = info; |
|
q_dst->visible_width = ntohl(p_hdr->width); |
|
q_dst->visible_height = ntohl(p_hdr->height); |
|
q_dst->coded_width = vic_round_dim(q_dst->visible_width, hdr_width_div); |
|
q_dst->coded_height = vic_round_dim(q_dst->visible_height, |
|
hdr_height_div); |
|
|
|
q_dst->sizeimage = q_dst->coded_width * q_dst->coded_height * |
|
q_dst->info->sizeimage_mult / q_dst->info->sizeimage_div; |
|
ctx->state.colorspace = ntohl(p_hdr->colorspace); |
|
|
|
ctx->state.xfer_func = ntohl(p_hdr->xfer_func); |
|
ctx->state.ycbcr_enc = ntohl(p_hdr->ycbcr_enc); |
|
ctx->state.quantization = ntohl(p_hdr->quantization); |
|
} |
|
|
|
static void set_last_buffer(struct vb2_v4l2_buffer *dst_buf, |
|
const struct vb2_v4l2_buffer *src_buf, |
|
struct vicodec_ctx *ctx) |
|
{ |
|
struct vicodec_q_data *q_dst = get_q_data(ctx, |
|
V4L2_BUF_TYPE_VIDEO_CAPTURE); |
|
|
|
vb2_set_plane_payload(&dst_buf->vb2_buf, 0, 0); |
|
dst_buf->sequence = q_dst->sequence++; |
|
|
|
v4l2_m2m_buf_copy_metadata(src_buf, dst_buf, !ctx->is_enc); |
|
dst_buf->flags |= V4L2_BUF_FLAG_LAST; |
|
v4l2_m2m_buf_done(dst_buf, VB2_BUF_STATE_DONE); |
|
} |
|
|
|
static int job_ready(void *priv) |
|
{ |
|
static const u8 magic[] = { |
|
0x4f, 0x4f, 0x4f, 0x4f, 0xff, 0xff, 0xff, 0xff |
|
}; |
|
struct vicodec_ctx *ctx = priv; |
|
struct vb2_v4l2_buffer *src_buf; |
|
u8 *p_src; |
|
u8 *p; |
|
u32 sz; |
|
u32 state; |
|
struct vicodec_q_data *q_dst = get_q_data(ctx, |
|
V4L2_BUF_TYPE_VIDEO_CAPTURE); |
|
unsigned int flags; |
|
unsigned int hdr_width_div; |
|
unsigned int hdr_height_div; |
|
unsigned int max_to_copy; |
|
unsigned int comp_frame_size; |
|
|
|
if (ctx->source_changed) |
|
return 0; |
|
if (ctx->is_stateless || ctx->is_enc || ctx->comp_has_frame) |
|
return 1; |
|
|
|
restart: |
|
ctx->comp_has_next_frame = false; |
|
src_buf = v4l2_m2m_next_src_buf(ctx->fh.m2m_ctx); |
|
if (!src_buf) |
|
return 0; |
|
p_src = vb2_plane_vaddr(&src_buf->vb2_buf, 0); |
|
sz = vb2_get_plane_payload(&src_buf->vb2_buf, 0); |
|
p = p_src + ctx->cur_buf_offset; |
|
|
|
state = VB2_BUF_STATE_DONE; |
|
|
|
if (ctx->header_size < sizeof(struct fwht_cframe_hdr)) { |
|
state = get_next_header(ctx, &p, p_src + sz - p); |
|
if (ctx->header_size < sizeof(struct fwht_cframe_hdr)) { |
|
if (v4l2_m2m_is_last_draining_src_buf(ctx->fh.m2m_ctx, |
|
src_buf)) |
|
return 1; |
|
job_remove_src_buf(ctx, state); |
|
goto restart; |
|
} |
|
} |
|
|
|
comp_frame_size = ntohl(ctx->state.header.size); |
|
|
|
/* |
|
* The current scanned frame might be the first frame of a new |
|
* resolution so its size might be larger than ctx->comp_max_size. |
|
* In that case it is copied up to the current buffer capacity and |
|
* the copy will continue after allocating new large enough buffer |
|
* when restreaming |
|
*/ |
|
max_to_copy = min(comp_frame_size, ctx->comp_max_size); |
|
|
|
if (ctx->comp_size < max_to_copy) { |
|
u32 copy = max_to_copy - ctx->comp_size; |
|
|
|
if (copy > p_src + sz - p) |
|
copy = p_src + sz - p; |
|
|
|
memcpy(ctx->state.compressed_frame + ctx->comp_size, |
|
p, copy); |
|
p += copy; |
|
ctx->comp_size += copy; |
|
if (ctx->comp_size < max_to_copy) { |
|
if (v4l2_m2m_is_last_draining_src_buf(ctx->fh.m2m_ctx, |
|
src_buf)) |
|
return 1; |
|
job_remove_src_buf(ctx, state); |
|
goto restart; |
|
} |
|
} |
|
ctx->cur_buf_offset = p - p_src; |
|
if (ctx->comp_size == comp_frame_size) |
|
ctx->comp_has_frame = true; |
|
ctx->comp_has_next_frame = false; |
|
if (ctx->comp_has_frame && sz - ctx->cur_buf_offset >= |
|
sizeof(struct fwht_cframe_hdr)) { |
|
struct fwht_cframe_hdr *p_hdr = (struct fwht_cframe_hdr *)p; |
|
u32 frame_size = ntohl(p_hdr->size); |
|
u32 remaining = sz - ctx->cur_buf_offset - sizeof(*p_hdr); |
|
|
|
if (!memcmp(p, magic, sizeof(magic))) |
|
ctx->comp_has_next_frame = remaining >= frame_size; |
|
} |
|
/* |
|
* if the header is invalid the device_run will just drop the frame |
|
* with an error |
|
*/ |
|
if (!is_header_valid(&ctx->state.header) && ctx->comp_has_frame) |
|
return 1; |
|
flags = ntohl(ctx->state.header.flags); |
|
hdr_width_div = (flags & V4L2_FWHT_FL_CHROMA_FULL_WIDTH) ? 1 : 2; |
|
hdr_height_div = (flags & V4L2_FWHT_FL_CHROMA_FULL_HEIGHT) ? 1 : 2; |
|
|
|
if (ntohl(ctx->state.header.width) != q_dst->visible_width || |
|
ntohl(ctx->state.header.height) != q_dst->visible_height || |
|
!q_dst->info || |
|
hdr_width_div != q_dst->info->width_div || |
|
hdr_height_div != q_dst->info->height_div) { |
|
static const struct v4l2_event rs_event = { |
|
.type = V4L2_EVENT_SOURCE_CHANGE, |
|
.u.src_change.changes = V4L2_EVENT_SRC_CH_RESOLUTION, |
|
}; |
|
|
|
struct vb2_v4l2_buffer *dst_buf = |
|
v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx); |
|
|
|
update_capture_data_from_header(ctx); |
|
v4l2_event_queue_fh(&ctx->fh, &rs_event); |
|
set_last_buffer(dst_buf, src_buf, ctx); |
|
ctx->source_changed = true; |
|
return 0; |
|
} |
|
return 1; |
|
} |
|
|
|
/* |
|
* video ioctls |
|
*/ |
|
|
|
static const struct v4l2_fwht_pixfmt_info *find_fmt(u32 fmt) |
|
{ |
|
const struct v4l2_fwht_pixfmt_info *info = |
|
v4l2_fwht_find_pixfmt(fmt); |
|
|
|
if (!info) |
|
info = v4l2_fwht_get_pixfmt(0); |
|
return info; |
|
} |
|
|
|
static int vidioc_querycap(struct file *file, void *priv, |
|
struct v4l2_capability *cap) |
|
{ |
|
strscpy(cap->driver, VICODEC_NAME, sizeof(cap->driver)); |
|
strscpy(cap->card, VICODEC_NAME, sizeof(cap->card)); |
|
snprintf(cap->bus_info, sizeof(cap->bus_info), |
|
"platform:%s", VICODEC_NAME); |
|
return 0; |
|
} |
|
|
|
static int enum_fmt(struct v4l2_fmtdesc *f, struct vicodec_ctx *ctx, |
|
bool is_out) |
|
{ |
|
bool is_uncomp = (ctx->is_enc && is_out) || (!ctx->is_enc && !is_out); |
|
|
|
if (V4L2_TYPE_IS_MULTIPLANAR(f->type) && !multiplanar) |
|
return -EINVAL; |
|
if (!V4L2_TYPE_IS_MULTIPLANAR(f->type) && multiplanar) |
|
return -EINVAL; |
|
|
|
if (is_uncomp) { |
|
const struct v4l2_fwht_pixfmt_info *info = |
|
get_q_data(ctx, f->type)->info; |
|
|
|
if (ctx->is_enc || |
|
!vb2_is_streaming(&ctx->fh.m2m_ctx->cap_q_ctx.q)) |
|
info = v4l2_fwht_get_pixfmt(f->index); |
|
else |
|
info = v4l2_fwht_find_nth_fmt(info->width_div, |
|
info->height_div, |
|
info->components_num, |
|
info->pixenc, |
|
f->index); |
|
if (!info) |
|
return -EINVAL; |
|
f->pixelformat = info->id; |
|
} else { |
|
if (f->index) |
|
return -EINVAL; |
|
f->pixelformat = ctx->is_stateless ? |
|
V4L2_PIX_FMT_FWHT_STATELESS : V4L2_PIX_FMT_FWHT; |
|
if (!ctx->is_enc && !ctx->is_stateless) |
|
f->flags = V4L2_FMT_FLAG_DYN_RESOLUTION | |
|
V4L2_FMT_FLAG_CONTINUOUS_BYTESTREAM; |
|
} |
|
return 0; |
|
} |
|
|
|
static int vidioc_enum_fmt_vid_cap(struct file *file, void *priv, |
|
struct v4l2_fmtdesc *f) |
|
{ |
|
struct vicodec_ctx *ctx = file2ctx(file); |
|
|
|
return enum_fmt(f, ctx, false); |
|
} |
|
|
|
static int vidioc_enum_fmt_vid_out(struct file *file, void *priv, |
|
struct v4l2_fmtdesc *f) |
|
{ |
|
struct vicodec_ctx *ctx = file2ctx(file); |
|
|
|
return enum_fmt(f, ctx, true); |
|
} |
|
|
|
static int vidioc_g_fmt(struct vicodec_ctx *ctx, struct v4l2_format *f) |
|
{ |
|
struct vb2_queue *vq; |
|
struct vicodec_q_data *q_data; |
|
struct v4l2_pix_format_mplane *pix_mp; |
|
struct v4l2_pix_format *pix; |
|
const struct v4l2_fwht_pixfmt_info *info; |
|
|
|
vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, f->type); |
|
if (!vq) |
|
return -EINVAL; |
|
|
|
q_data = get_q_data(ctx, f->type); |
|
info = q_data->info; |
|
|
|
switch (f->type) { |
|
case V4L2_BUF_TYPE_VIDEO_CAPTURE: |
|
case V4L2_BUF_TYPE_VIDEO_OUTPUT: |
|
if (multiplanar) |
|
return -EINVAL; |
|
pix = &f->fmt.pix; |
|
pix->width = q_data->coded_width; |
|
pix->height = q_data->coded_height; |
|
pix->field = V4L2_FIELD_NONE; |
|
pix->pixelformat = info->id; |
|
pix->bytesperline = q_data->coded_width * |
|
info->bytesperline_mult; |
|
pix->sizeimage = q_data->sizeimage; |
|
pix->colorspace = ctx->state.colorspace; |
|
pix->xfer_func = ctx->state.xfer_func; |
|
pix->ycbcr_enc = ctx->state.ycbcr_enc; |
|
pix->quantization = ctx->state.quantization; |
|
break; |
|
|
|
case V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE: |
|
case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE: |
|
if (!multiplanar) |
|
return -EINVAL; |
|
pix_mp = &f->fmt.pix_mp; |
|
pix_mp->width = q_data->coded_width; |
|
pix_mp->height = q_data->coded_height; |
|
pix_mp->field = V4L2_FIELD_NONE; |
|
pix_mp->pixelformat = info->id; |
|
pix_mp->num_planes = 1; |
|
pix_mp->plane_fmt[0].bytesperline = |
|
q_data->coded_width * info->bytesperline_mult; |
|
pix_mp->plane_fmt[0].sizeimage = q_data->sizeimage; |
|
pix_mp->colorspace = ctx->state.colorspace; |
|
pix_mp->xfer_func = ctx->state.xfer_func; |
|
pix_mp->ycbcr_enc = ctx->state.ycbcr_enc; |
|
pix_mp->quantization = ctx->state.quantization; |
|
break; |
|
default: |
|
return -EINVAL; |
|
} |
|
return 0; |
|
} |
|
|
|
static int vidioc_g_fmt_vid_out(struct file *file, void *priv, |
|
struct v4l2_format *f) |
|
{ |
|
return vidioc_g_fmt(file2ctx(file), f); |
|
} |
|
|
|
static int vidioc_g_fmt_vid_cap(struct file *file, void *priv, |
|
struct v4l2_format *f) |
|
{ |
|
return vidioc_g_fmt(file2ctx(file), f); |
|
} |
|
|
|
static int vidioc_try_fmt(struct vicodec_ctx *ctx, struct v4l2_format *f) |
|
{ |
|
struct v4l2_pix_format_mplane *pix_mp; |
|
struct v4l2_pix_format *pix; |
|
struct v4l2_plane_pix_format *plane; |
|
const struct v4l2_fwht_pixfmt_info *info = ctx->is_stateless ? |
|
&pixfmt_stateless_fwht : &pixfmt_fwht; |
|
|
|
switch (f->type) { |
|
case V4L2_BUF_TYPE_VIDEO_CAPTURE: |
|
case V4L2_BUF_TYPE_VIDEO_OUTPUT: |
|
pix = &f->fmt.pix; |
|
if (pix->pixelformat != V4L2_PIX_FMT_FWHT && |
|
pix->pixelformat != V4L2_PIX_FMT_FWHT_STATELESS) |
|
info = find_fmt(pix->pixelformat); |
|
|
|
pix->width = clamp(pix->width, MIN_WIDTH, MAX_WIDTH); |
|
pix->width = vic_round_dim(pix->width, info->width_div); |
|
|
|
pix->height = clamp(pix->height, MIN_HEIGHT, MAX_HEIGHT); |
|
pix->height = vic_round_dim(pix->height, info->height_div); |
|
|
|
pix->field = V4L2_FIELD_NONE; |
|
pix->bytesperline = |
|
pix->width * info->bytesperline_mult; |
|
pix->sizeimage = pix->width * pix->height * |
|
info->sizeimage_mult / info->sizeimage_div; |
|
if (pix->pixelformat == V4L2_PIX_FMT_FWHT) |
|
pix->sizeimage += sizeof(struct fwht_cframe_hdr); |
|
break; |
|
case V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE: |
|
case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE: |
|
pix_mp = &f->fmt.pix_mp; |
|
plane = pix_mp->plane_fmt; |
|
if (pix_mp->pixelformat != V4L2_PIX_FMT_FWHT && |
|
pix_mp->pixelformat != V4L2_PIX_FMT_FWHT_STATELESS) |
|
info = find_fmt(pix_mp->pixelformat); |
|
pix_mp->num_planes = 1; |
|
|
|
pix_mp->width = clamp(pix_mp->width, MIN_WIDTH, MAX_WIDTH); |
|
pix_mp->width = vic_round_dim(pix_mp->width, info->width_div); |
|
|
|
pix_mp->height = clamp(pix_mp->height, MIN_HEIGHT, MAX_HEIGHT); |
|
pix_mp->height = vic_round_dim(pix_mp->height, |
|
info->height_div); |
|
|
|
pix_mp->field = V4L2_FIELD_NONE; |
|
plane->bytesperline = |
|
pix_mp->width * info->bytesperline_mult; |
|
plane->sizeimage = pix_mp->width * pix_mp->height * |
|
info->sizeimage_mult / info->sizeimage_div; |
|
if (pix_mp->pixelformat == V4L2_PIX_FMT_FWHT) |
|
plane->sizeimage += sizeof(struct fwht_cframe_hdr); |
|
break; |
|
default: |
|
return -EINVAL; |
|
} |
|
|
|
return 0; |
|
} |
|
|
|
static int vidioc_try_fmt_vid_cap(struct file *file, void *priv, |
|
struct v4l2_format *f) |
|
{ |
|
struct vicodec_ctx *ctx = file2ctx(file); |
|
struct v4l2_pix_format_mplane *pix_mp; |
|
struct v4l2_pix_format *pix; |
|
|
|
switch (f->type) { |
|
case V4L2_BUF_TYPE_VIDEO_CAPTURE: |
|
if (multiplanar) |
|
return -EINVAL; |
|
pix = &f->fmt.pix; |
|
pix->pixelformat = ctx->is_enc ? V4L2_PIX_FMT_FWHT : |
|
find_fmt(f->fmt.pix.pixelformat)->id; |
|
pix->colorspace = ctx->state.colorspace; |
|
pix->xfer_func = ctx->state.xfer_func; |
|
pix->ycbcr_enc = ctx->state.ycbcr_enc; |
|
pix->quantization = ctx->state.quantization; |
|
break; |
|
case V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE: |
|
if (!multiplanar) |
|
return -EINVAL; |
|
pix_mp = &f->fmt.pix_mp; |
|
pix_mp->pixelformat = ctx->is_enc ? V4L2_PIX_FMT_FWHT : |
|
find_fmt(pix_mp->pixelformat)->id; |
|
pix_mp->colorspace = ctx->state.colorspace; |
|
pix_mp->xfer_func = ctx->state.xfer_func; |
|
pix_mp->ycbcr_enc = ctx->state.ycbcr_enc; |
|
pix_mp->quantization = ctx->state.quantization; |
|
break; |
|
default: |
|
return -EINVAL; |
|
} |
|
|
|
return vidioc_try_fmt(ctx, f); |
|
} |
|
|
|
static int vidioc_try_fmt_vid_out(struct file *file, void *priv, |
|
struct v4l2_format *f) |
|
{ |
|
struct vicodec_ctx *ctx = file2ctx(file); |
|
struct v4l2_pix_format_mplane *pix_mp; |
|
struct v4l2_pix_format *pix; |
|
|
|
switch (f->type) { |
|
case V4L2_BUF_TYPE_VIDEO_OUTPUT: |
|
if (multiplanar) |
|
return -EINVAL; |
|
pix = &f->fmt.pix; |
|
if (ctx->is_enc) |
|
pix->pixelformat = find_fmt(pix->pixelformat)->id; |
|
else if (ctx->is_stateless) |
|
pix->pixelformat = V4L2_PIX_FMT_FWHT_STATELESS; |
|
else |
|
pix->pixelformat = V4L2_PIX_FMT_FWHT; |
|
if (!pix->colorspace) |
|
pix->colorspace = V4L2_COLORSPACE_REC709; |
|
break; |
|
case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE: |
|
if (!multiplanar) |
|
return -EINVAL; |
|
pix_mp = &f->fmt.pix_mp; |
|
if (ctx->is_enc) |
|
pix_mp->pixelformat = find_fmt(pix_mp->pixelformat)->id; |
|
else if (ctx->is_stateless) |
|
pix_mp->pixelformat = V4L2_PIX_FMT_FWHT_STATELESS; |
|
else |
|
pix_mp->pixelformat = V4L2_PIX_FMT_FWHT; |
|
if (!pix_mp->colorspace) |
|
pix_mp->colorspace = V4L2_COLORSPACE_REC709; |
|
break; |
|
default: |
|
return -EINVAL; |
|
} |
|
|
|
return vidioc_try_fmt(ctx, f); |
|
} |
|
|
|
static int vidioc_s_fmt(struct vicodec_ctx *ctx, struct v4l2_format *f) |
|
{ |
|
struct vicodec_q_data *q_data; |
|
struct vb2_queue *vq; |
|
bool fmt_changed = true; |
|
struct v4l2_pix_format_mplane *pix_mp; |
|
struct v4l2_pix_format *pix; |
|
|
|
vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, f->type); |
|
if (!vq) |
|
return -EINVAL; |
|
|
|
q_data = get_q_data(ctx, f->type); |
|
if (!q_data) |
|
return -EINVAL; |
|
|
|
switch (f->type) { |
|
case V4L2_BUF_TYPE_VIDEO_CAPTURE: |
|
case V4L2_BUF_TYPE_VIDEO_OUTPUT: |
|
pix = &f->fmt.pix; |
|
if (ctx->is_enc && V4L2_TYPE_IS_OUTPUT(f->type)) |
|
fmt_changed = |
|
!q_data->info || |
|
q_data->info->id != pix->pixelformat || |
|
q_data->coded_width != pix->width || |
|
q_data->coded_height != pix->height; |
|
|
|
if (vb2_is_busy(vq) && fmt_changed) |
|
return -EBUSY; |
|
|
|
if (pix->pixelformat == V4L2_PIX_FMT_FWHT) |
|
q_data->info = &pixfmt_fwht; |
|
else if (pix->pixelformat == V4L2_PIX_FMT_FWHT_STATELESS) |
|
q_data->info = &pixfmt_stateless_fwht; |
|
else |
|
q_data->info = find_fmt(pix->pixelformat); |
|
q_data->coded_width = pix->width; |
|
q_data->coded_height = pix->height; |
|
q_data->sizeimage = pix->sizeimage; |
|
break; |
|
case V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE: |
|
case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE: |
|
pix_mp = &f->fmt.pix_mp; |
|
if (ctx->is_enc && V4L2_TYPE_IS_OUTPUT(f->type)) |
|
fmt_changed = |
|
!q_data->info || |
|
q_data->info->id != pix_mp->pixelformat || |
|
q_data->coded_width != pix_mp->width || |
|
q_data->coded_height != pix_mp->height; |
|
|
|
if (vb2_is_busy(vq) && fmt_changed) |
|
return -EBUSY; |
|
|
|
if (pix_mp->pixelformat == V4L2_PIX_FMT_FWHT) |
|
q_data->info = &pixfmt_fwht; |
|
else if (pix_mp->pixelformat == V4L2_PIX_FMT_FWHT_STATELESS) |
|
q_data->info = &pixfmt_stateless_fwht; |
|
else |
|
q_data->info = find_fmt(pix_mp->pixelformat); |
|
q_data->coded_width = pix_mp->width; |
|
q_data->coded_height = pix_mp->height; |
|
q_data->sizeimage = pix_mp->plane_fmt[0].sizeimage; |
|
break; |
|
default: |
|
return -EINVAL; |
|
} |
|
|
|
dprintk(ctx->dev, |
|
"Setting format for type %d, coded wxh: %dx%d, fourcc: 0x%08x\n", |
|
f->type, q_data->coded_width, q_data->coded_height, |
|
q_data->info->id); |
|
|
|
return 0; |
|
} |
|
|
|
static int vidioc_s_fmt_vid_cap(struct file *file, void *priv, |
|
struct v4l2_format *f) |
|
{ |
|
int ret; |
|
|
|
ret = vidioc_try_fmt_vid_cap(file, priv, f); |
|
if (ret) |
|
return ret; |
|
|
|
return vidioc_s_fmt(file2ctx(file), f); |
|
} |
|
|
|
static int vidioc_s_fmt_vid_out(struct file *file, void *priv, |
|
struct v4l2_format *f) |
|
{ |
|
struct vicodec_ctx *ctx = file2ctx(file); |
|
struct vicodec_q_data *q_data; |
|
struct vicodec_q_data *q_data_cap; |
|
struct v4l2_pix_format *pix; |
|
struct v4l2_pix_format_mplane *pix_mp; |
|
u32 coded_w = 0, coded_h = 0; |
|
unsigned int size = 0; |
|
int ret; |
|
|
|
q_data = get_q_data(ctx, f->type); |
|
q_data_cap = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE); |
|
|
|
ret = vidioc_try_fmt_vid_out(file, priv, f); |
|
if (ret) |
|
return ret; |
|
|
|
if (ctx->is_enc) { |
|
struct vb2_queue *vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, f->type); |
|
struct vb2_queue *vq_cap = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, |
|
V4L2_BUF_TYPE_VIDEO_CAPTURE); |
|
const struct v4l2_fwht_pixfmt_info *info = ctx->is_stateless ? |
|
&pixfmt_stateless_fwht : &pixfmt_fwht; |
|
|
|
if (f->type == V4L2_BUF_TYPE_VIDEO_OUTPUT) { |
|
coded_w = f->fmt.pix.width; |
|
coded_h = f->fmt.pix.height; |
|
} else { |
|
coded_w = f->fmt.pix_mp.width; |
|
coded_h = f->fmt.pix_mp.height; |
|
} |
|
if (vb2_is_busy(vq) && (coded_w != q_data->coded_width || |
|
coded_h != q_data->coded_height)) |
|
return -EBUSY; |
|
size = coded_w * coded_h * |
|
info->sizeimage_mult / info->sizeimage_div; |
|
if (!ctx->is_stateless) |
|
size += sizeof(struct fwht_cframe_hdr); |
|
|
|
if (vb2_is_busy(vq_cap) && size > q_data_cap->sizeimage) |
|
return -EBUSY; |
|
} |
|
|
|
ret = vidioc_s_fmt(file2ctx(file), f); |
|
if (!ret) { |
|
if (ctx->is_enc) { |
|
q_data->visible_width = coded_w; |
|
q_data->visible_height = coded_h; |
|
q_data_cap->coded_width = coded_w; |
|
q_data_cap->coded_height = coded_h; |
|
q_data_cap->sizeimage = size; |
|
} |
|
|
|
switch (f->type) { |
|
case V4L2_BUF_TYPE_VIDEO_OUTPUT: |
|
pix = &f->fmt.pix; |
|
ctx->state.colorspace = pix->colorspace; |
|
ctx->state.xfer_func = pix->xfer_func; |
|
ctx->state.ycbcr_enc = pix->ycbcr_enc; |
|
ctx->state.quantization = pix->quantization; |
|
break; |
|
case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE: |
|
pix_mp = &f->fmt.pix_mp; |
|
ctx->state.colorspace = pix_mp->colorspace; |
|
ctx->state.xfer_func = pix_mp->xfer_func; |
|
ctx->state.ycbcr_enc = pix_mp->ycbcr_enc; |
|
ctx->state.quantization = pix_mp->quantization; |
|
break; |
|
default: |
|
break; |
|
} |
|
} |
|
return ret; |
|
} |
|
|
|
static int vidioc_g_selection(struct file *file, void *priv, |
|
struct v4l2_selection *s) |
|
{ |
|
struct vicodec_ctx *ctx = file2ctx(file); |
|
struct vicodec_q_data *q_data; |
|
|
|
q_data = get_q_data(ctx, s->type); |
|
if (!q_data) |
|
return -EINVAL; |
|
/* |
|
* encoder supports only cropping on the OUTPUT buffer |
|
* decoder supports only composing on the CAPTURE buffer |
|
*/ |
|
if (ctx->is_enc && s->type == V4L2_BUF_TYPE_VIDEO_OUTPUT) { |
|
switch (s->target) { |
|
case V4L2_SEL_TGT_CROP: |
|
s->r.left = 0; |
|
s->r.top = 0; |
|
s->r.width = q_data->visible_width; |
|
s->r.height = q_data->visible_height; |
|
return 0; |
|
case V4L2_SEL_TGT_CROP_DEFAULT: |
|
case V4L2_SEL_TGT_CROP_BOUNDS: |
|
s->r.left = 0; |
|
s->r.top = 0; |
|
s->r.width = q_data->coded_width; |
|
s->r.height = q_data->coded_height; |
|
return 0; |
|
} |
|
} else if (!ctx->is_enc && s->type == V4L2_BUF_TYPE_VIDEO_CAPTURE) { |
|
switch (s->target) { |
|
case V4L2_SEL_TGT_COMPOSE: |
|
s->r.left = 0; |
|
s->r.top = 0; |
|
s->r.width = q_data->visible_width; |
|
s->r.height = q_data->visible_height; |
|
return 0; |
|
case V4L2_SEL_TGT_COMPOSE_DEFAULT: |
|
case V4L2_SEL_TGT_COMPOSE_BOUNDS: |
|
s->r.left = 0; |
|
s->r.top = 0; |
|
s->r.width = q_data->coded_width; |
|
s->r.height = q_data->coded_height; |
|
return 0; |
|
} |
|
} |
|
return -EINVAL; |
|
} |
|
|
|
static int vidioc_s_selection(struct file *file, void *priv, |
|
struct v4l2_selection *s) |
|
{ |
|
struct vicodec_ctx *ctx = file2ctx(file); |
|
struct vicodec_q_data *q_data; |
|
|
|
if (s->type != V4L2_BUF_TYPE_VIDEO_OUTPUT) |
|
return -EINVAL; |
|
|
|
q_data = get_q_data(ctx, s->type); |
|
if (!q_data) |
|
return -EINVAL; |
|
|
|
if (!ctx->is_enc || s->target != V4L2_SEL_TGT_CROP) |
|
return -EINVAL; |
|
|
|
s->r.left = 0; |
|
s->r.top = 0; |
|
q_data->visible_width = clamp(s->r.width, MIN_WIDTH, |
|
q_data->coded_width); |
|
s->r.width = q_data->visible_width; |
|
q_data->visible_height = clamp(s->r.height, MIN_HEIGHT, |
|
q_data->coded_height); |
|
s->r.height = q_data->visible_height; |
|
return 0; |
|
} |
|
|
|
static int vicodec_encoder_cmd(struct file *file, void *fh, |
|
struct v4l2_encoder_cmd *ec) |
|
{ |
|
struct vicodec_ctx *ctx = file2ctx(file); |
|
int ret; |
|
|
|
ret = v4l2_m2m_ioctl_try_encoder_cmd(file, fh, ec); |
|
if (ret < 0) |
|
return ret; |
|
|
|
if (!vb2_is_streaming(&ctx->fh.m2m_ctx->cap_q_ctx.q) || |
|
!vb2_is_streaming(&ctx->fh.m2m_ctx->out_q_ctx.q)) |
|
return 0; |
|
|
|
ret = v4l2_m2m_ioctl_encoder_cmd(file, fh, ec); |
|
if (ret < 0) |
|
return ret; |
|
|
|
if (ec->cmd == V4L2_ENC_CMD_STOP && |
|
v4l2_m2m_has_stopped(ctx->fh.m2m_ctx)) |
|
v4l2_event_queue_fh(&ctx->fh, &vicodec_eos_event); |
|
|
|
if (ec->cmd == V4L2_ENC_CMD_START && |
|
v4l2_m2m_has_stopped(ctx->fh.m2m_ctx)) |
|
vb2_clear_last_buffer_dequeued(&ctx->fh.m2m_ctx->cap_q_ctx.q); |
|
|
|
return 0; |
|
} |
|
|
|
static int vicodec_decoder_cmd(struct file *file, void *fh, |
|
struct v4l2_decoder_cmd *dc) |
|
{ |
|
struct vicodec_ctx *ctx = file2ctx(file); |
|
int ret; |
|
|
|
ret = v4l2_m2m_ioctl_try_decoder_cmd(file, fh, dc); |
|
if (ret < 0) |
|
return ret; |
|
|
|
if (!vb2_is_streaming(&ctx->fh.m2m_ctx->cap_q_ctx.q) || |
|
!vb2_is_streaming(&ctx->fh.m2m_ctx->out_q_ctx.q)) |
|
return 0; |
|
|
|
ret = v4l2_m2m_ioctl_decoder_cmd(file, fh, dc); |
|
if (ret < 0) |
|
return ret; |
|
|
|
if (dc->cmd == V4L2_DEC_CMD_STOP && |
|
v4l2_m2m_has_stopped(ctx->fh.m2m_ctx)) |
|
v4l2_event_queue_fh(&ctx->fh, &vicodec_eos_event); |
|
|
|
if (dc->cmd == V4L2_DEC_CMD_START && |
|
v4l2_m2m_has_stopped(ctx->fh.m2m_ctx)) |
|
vb2_clear_last_buffer_dequeued(&ctx->fh.m2m_ctx->cap_q_ctx.q); |
|
|
|
return 0; |
|
} |
|
|
|
static int vicodec_enum_framesizes(struct file *file, void *fh, |
|
struct v4l2_frmsizeenum *fsize) |
|
{ |
|
switch (fsize->pixel_format) { |
|
case V4L2_PIX_FMT_FWHT_STATELESS: |
|
break; |
|
case V4L2_PIX_FMT_FWHT: |
|
break; |
|
default: |
|
if (find_fmt(fsize->pixel_format)->id == fsize->pixel_format) |
|
break; |
|
return -EINVAL; |
|
} |
|
|
|
if (fsize->index) |
|
return -EINVAL; |
|
|
|
fsize->type = V4L2_FRMSIZE_TYPE_STEPWISE; |
|
|
|
fsize->stepwise.min_width = MIN_WIDTH; |
|
fsize->stepwise.max_width = MAX_WIDTH; |
|
fsize->stepwise.step_width = 8; |
|
fsize->stepwise.min_height = MIN_HEIGHT; |
|
fsize->stepwise.max_height = MAX_HEIGHT; |
|
fsize->stepwise.step_height = 8; |
|
|
|
return 0; |
|
} |
|
|
|
static int vicodec_subscribe_event(struct v4l2_fh *fh, |
|
const struct v4l2_event_subscription *sub) |
|
{ |
|
struct vicodec_ctx *ctx = container_of(fh, struct vicodec_ctx, fh); |
|
|
|
switch (sub->type) { |
|
case V4L2_EVENT_SOURCE_CHANGE: |
|
if (ctx->is_enc) |
|
return -EINVAL; |
|
fallthrough; |
|
case V4L2_EVENT_EOS: |
|
if (ctx->is_stateless) |
|
return -EINVAL; |
|
return v4l2_event_subscribe(fh, sub, 0, NULL); |
|
default: |
|
return v4l2_ctrl_subscribe_event(fh, sub); |
|
} |
|
} |
|
|
|
static const struct v4l2_ioctl_ops vicodec_ioctl_ops = { |
|
.vidioc_querycap = vidioc_querycap, |
|
|
|
.vidioc_enum_fmt_vid_cap = vidioc_enum_fmt_vid_cap, |
|
.vidioc_g_fmt_vid_cap = vidioc_g_fmt_vid_cap, |
|
.vidioc_try_fmt_vid_cap = vidioc_try_fmt_vid_cap, |
|
.vidioc_s_fmt_vid_cap = vidioc_s_fmt_vid_cap, |
|
|
|
.vidioc_g_fmt_vid_cap_mplane = vidioc_g_fmt_vid_cap, |
|
.vidioc_try_fmt_vid_cap_mplane = vidioc_try_fmt_vid_cap, |
|
.vidioc_s_fmt_vid_cap_mplane = vidioc_s_fmt_vid_cap, |
|
|
|
.vidioc_enum_fmt_vid_out = vidioc_enum_fmt_vid_out, |
|
.vidioc_g_fmt_vid_out = vidioc_g_fmt_vid_out, |
|
.vidioc_try_fmt_vid_out = vidioc_try_fmt_vid_out, |
|
.vidioc_s_fmt_vid_out = vidioc_s_fmt_vid_out, |
|
|
|
.vidioc_g_fmt_vid_out_mplane = vidioc_g_fmt_vid_out, |
|
.vidioc_try_fmt_vid_out_mplane = vidioc_try_fmt_vid_out, |
|
.vidioc_s_fmt_vid_out_mplane = vidioc_s_fmt_vid_out, |
|
|
|
.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_prepare_buf = v4l2_m2m_ioctl_prepare_buf, |
|
.vidioc_create_bufs = v4l2_m2m_ioctl_create_bufs, |
|
.vidioc_expbuf = v4l2_m2m_ioctl_expbuf, |
|
|
|
.vidioc_streamon = v4l2_m2m_ioctl_streamon, |
|
.vidioc_streamoff = v4l2_m2m_ioctl_streamoff, |
|
|
|
.vidioc_g_selection = vidioc_g_selection, |
|
.vidioc_s_selection = vidioc_s_selection, |
|
|
|
.vidioc_try_encoder_cmd = v4l2_m2m_ioctl_try_encoder_cmd, |
|
.vidioc_encoder_cmd = vicodec_encoder_cmd, |
|
.vidioc_try_decoder_cmd = v4l2_m2m_ioctl_try_decoder_cmd, |
|
.vidioc_decoder_cmd = vicodec_decoder_cmd, |
|
.vidioc_enum_framesizes = vicodec_enum_framesizes, |
|
|
|
.vidioc_subscribe_event = vicodec_subscribe_event, |
|
.vidioc_unsubscribe_event = v4l2_event_unsubscribe, |
|
}; |
|
|
|
|
|
/* |
|
* Queue operations |
|
*/ |
|
|
|
static int vicodec_queue_setup(struct vb2_queue *vq, unsigned int *nbuffers, |
|
unsigned int *nplanes, unsigned int sizes[], |
|
struct device *alloc_devs[]) |
|
{ |
|
struct vicodec_ctx *ctx = vb2_get_drv_priv(vq); |
|
struct vicodec_q_data *q_data = get_q_data(ctx, vq->type); |
|
unsigned int size = q_data->sizeimage; |
|
|
|
if (*nplanes) |
|
return sizes[0] < size ? -EINVAL : 0; |
|
|
|
*nplanes = 1; |
|
sizes[0] = size; |
|
q_data->vb2_sizeimage = size; |
|
return 0; |
|
} |
|
|
|
static int vicodec_buf_out_validate(struct vb2_buffer *vb) |
|
{ |
|
struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb); |
|
|
|
vbuf->field = V4L2_FIELD_NONE; |
|
return 0; |
|
} |
|
|
|
static int vicodec_buf_prepare(struct vb2_buffer *vb) |
|
{ |
|
struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb); |
|
struct vicodec_ctx *ctx = vb2_get_drv_priv(vb->vb2_queue); |
|
struct vicodec_q_data *q_data; |
|
|
|
dprintk(ctx->dev, "type: %d\n", vb->vb2_queue->type); |
|
|
|
q_data = get_q_data(ctx, vb->vb2_queue->type); |
|
if (V4L2_TYPE_IS_OUTPUT(vb->vb2_queue->type)) { |
|
if (vbuf->field == V4L2_FIELD_ANY) |
|
vbuf->field = V4L2_FIELD_NONE; |
|
if (vbuf->field != V4L2_FIELD_NONE) { |
|
dprintk(ctx->dev, "%s field isn't supported\n", |
|
__func__); |
|
return -EINVAL; |
|
} |
|
} |
|
|
|
if (vb2_plane_size(vb, 0) < q_data->vb2_sizeimage) { |
|
dprintk(ctx->dev, |
|
"%s data will not fit into plane (%lu < %lu)\n", |
|
__func__, vb2_plane_size(vb, 0), |
|
(long)q_data->vb2_sizeimage); |
|
return -EINVAL; |
|
} |
|
|
|
return 0; |
|
} |
|
|
|
static void vicodec_buf_queue(struct vb2_buffer *vb) |
|
{ |
|
struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb); |
|
struct vicodec_ctx *ctx = vb2_get_drv_priv(vb->vb2_queue); |
|
unsigned int sz = vb2_get_plane_payload(&vbuf->vb2_buf, 0); |
|
u8 *p_src = vb2_plane_vaddr(&vbuf->vb2_buf, 0); |
|
u8 *p = p_src; |
|
struct vb2_queue *vq_out = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, |
|
V4L2_BUF_TYPE_VIDEO_OUTPUT); |
|
struct vb2_queue *vq_cap = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, |
|
V4L2_BUF_TYPE_VIDEO_CAPTURE); |
|
bool header_valid = false; |
|
static const struct v4l2_event rs_event = { |
|
.type = V4L2_EVENT_SOURCE_CHANGE, |
|
.u.src_change.changes = V4L2_EVENT_SRC_CH_RESOLUTION, |
|
}; |
|
|
|
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)) { |
|
unsigned int i; |
|
|
|
for (i = 0; i < vb->num_planes; i++) |
|
vb->planes[i].bytesused = 0; |
|
|
|
vbuf->field = V4L2_FIELD_NONE; |
|
vbuf->sequence = |
|
get_q_data(ctx, vb->vb2_queue->type)->sequence++; |
|
|
|
v4l2_m2m_last_buffer_done(ctx->fh.m2m_ctx, vbuf); |
|
v4l2_event_queue_fh(&ctx->fh, &vicodec_eos_event); |
|
return; |
|
} |
|
|
|
/* buf_queue handles only the first source change event */ |
|
if (ctx->first_source_change_sent) { |
|
v4l2_m2m_buf_queue(ctx->fh.m2m_ctx, vbuf); |
|
return; |
|
} |
|
|
|
/* |
|
* if both queues are streaming, the source change event is |
|
* handled in job_ready |
|
*/ |
|
if (vb2_is_streaming(vq_cap) && vb2_is_streaming(vq_out)) { |
|
v4l2_m2m_buf_queue(ctx->fh.m2m_ctx, vbuf); |
|
return; |
|
} |
|
|
|
/* |
|
* source change event is relevant only for the stateful decoder |
|
* in the compressed stream |
|
*/ |
|
if (ctx->is_stateless || ctx->is_enc || |
|
V4L2_TYPE_IS_CAPTURE(vb->vb2_queue->type)) { |
|
v4l2_m2m_buf_queue(ctx->fh.m2m_ctx, vbuf); |
|
return; |
|
} |
|
|
|
do { |
|
enum vb2_buffer_state state = |
|
get_next_header(ctx, &p, p_src + sz - p); |
|
|
|
if (ctx->header_size < sizeof(struct fwht_cframe_hdr)) { |
|
v4l2_m2m_buf_done(vbuf, state); |
|
return; |
|
} |
|
header_valid = is_header_valid(&ctx->state.header); |
|
/* |
|
* p points right after the end of the header in the |
|
* buffer. If the header is invalid we set p to point |
|
* to the next byte after the start of the header |
|
*/ |
|
if (!header_valid) { |
|
p = p - sizeof(struct fwht_cframe_hdr) + 1; |
|
if (p < p_src) |
|
p = p_src; |
|
ctx->header_size = 0; |
|
ctx->comp_magic_cnt = 0; |
|
} |
|
|
|
} while (!header_valid); |
|
|
|
ctx->cur_buf_offset = p - p_src; |
|
update_capture_data_from_header(ctx); |
|
ctx->first_source_change_sent = true; |
|
v4l2_event_queue_fh(&ctx->fh, &rs_event); |
|
v4l2_m2m_buf_queue(ctx->fh.m2m_ctx, vbuf); |
|
} |
|
|
|
static void vicodec_return_bufs(struct vb2_queue *q, u32 state) |
|
{ |
|
struct vicodec_ctx *ctx = vb2_get_drv_priv(q); |
|
struct vb2_v4l2_buffer *vbuf; |
|
|
|
for (;;) { |
|
if (V4L2_TYPE_IS_OUTPUT(q->type)) |
|
vbuf = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx); |
|
else |
|
vbuf = v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx); |
|
if (vbuf == NULL) |
|
return; |
|
v4l2_ctrl_request_complete(vbuf->vb2_buf.req_obj.req, |
|
&ctx->hdl); |
|
spin_lock(ctx->lock); |
|
v4l2_m2m_buf_done(vbuf, state); |
|
spin_unlock(ctx->lock); |
|
} |
|
} |
|
|
|
static unsigned int total_frame_size(struct vicodec_q_data *q_data) |
|
{ |
|
unsigned int size; |
|
unsigned int chroma_div; |
|
|
|
if (!q_data->info) { |
|
WARN_ON(1); |
|
return 0; |
|
} |
|
size = q_data->coded_width * q_data->coded_height; |
|
chroma_div = q_data->info->width_div * q_data->info->height_div; |
|
|
|
if (q_data->info->components_num == 4) |
|
return 2 * size + 2 * (size / chroma_div); |
|
else if (q_data->info->components_num == 3) |
|
return size + 2 * (size / chroma_div); |
|
return size; |
|
} |
|
|
|
static int vicodec_start_streaming(struct vb2_queue *q, |
|
unsigned int count) |
|
{ |
|
struct vicodec_ctx *ctx = vb2_get_drv_priv(q); |
|
struct vicodec_q_data *q_data = get_q_data(ctx, q->type); |
|
struct v4l2_fwht_state *state = &ctx->state; |
|
const struct v4l2_fwht_pixfmt_info *info = q_data->info; |
|
unsigned int size = q_data->coded_width * q_data->coded_height; |
|
unsigned int chroma_div; |
|
unsigned int total_planes_size; |
|
u8 *new_comp_frame = NULL; |
|
|
|
chroma_div = info->width_div * info->height_div; |
|
q_data->sequence = 0; |
|
|
|
v4l2_m2m_update_start_streaming_state(ctx->fh.m2m_ctx, q); |
|
|
|
state->gop_cnt = 0; |
|
|
|
if ((V4L2_TYPE_IS_OUTPUT(q->type) && !ctx->is_enc) || |
|
(V4L2_TYPE_IS_CAPTURE(q->type) && ctx->is_enc)) |
|
return 0; |
|
|
|
if (info->id == V4L2_PIX_FMT_FWHT || |
|
info->id == V4L2_PIX_FMT_FWHT_STATELESS) { |
|
vicodec_return_bufs(q, VB2_BUF_STATE_QUEUED); |
|
return -EINVAL; |
|
} |
|
total_planes_size = total_frame_size(q_data); |
|
ctx->comp_max_size = total_planes_size; |
|
|
|
state->visible_width = q_data->visible_width; |
|
state->visible_height = q_data->visible_height; |
|
state->coded_width = q_data->coded_width; |
|
state->coded_height = q_data->coded_height; |
|
state->stride = q_data->coded_width * |
|
info->bytesperline_mult; |
|
|
|
if (ctx->is_stateless) { |
|
state->ref_stride = state->stride; |
|
return 0; |
|
} |
|
state->ref_stride = q_data->coded_width * info->luma_alpha_step; |
|
|
|
state->ref_frame.buf = kvmalloc(total_planes_size, GFP_KERNEL); |
|
state->ref_frame.luma = state->ref_frame.buf; |
|
new_comp_frame = kvmalloc(ctx->comp_max_size, GFP_KERNEL); |
|
|
|
if (!state->ref_frame.luma || !new_comp_frame) { |
|
kvfree(state->ref_frame.luma); |
|
kvfree(new_comp_frame); |
|
vicodec_return_bufs(q, VB2_BUF_STATE_QUEUED); |
|
return -ENOMEM; |
|
} |
|
/* |
|
* if state->compressed_frame was already allocated then |
|
* it contain data of the first frame of the new resolution |
|
*/ |
|
if (state->compressed_frame) { |
|
if (ctx->comp_size > ctx->comp_max_size) |
|
ctx->comp_size = ctx->comp_max_size; |
|
|
|
memcpy(new_comp_frame, |
|
state->compressed_frame, ctx->comp_size); |
|
} |
|
|
|
kvfree(state->compressed_frame); |
|
state->compressed_frame = new_comp_frame; |
|
|
|
if (info->components_num < 3) { |
|
state->ref_frame.cb = NULL; |
|
state->ref_frame.cr = NULL; |
|
state->ref_frame.alpha = NULL; |
|
return 0; |
|
} |
|
|
|
state->ref_frame.cb = state->ref_frame.luma + size; |
|
state->ref_frame.cr = state->ref_frame.cb + size / chroma_div; |
|
|
|
if (info->components_num == 4) |
|
state->ref_frame.alpha = |
|
state->ref_frame.cr + size / chroma_div; |
|
else |
|
state->ref_frame.alpha = NULL; |
|
|
|
return 0; |
|
} |
|
|
|
static void vicodec_stop_streaming(struct vb2_queue *q) |
|
{ |
|
struct vicodec_ctx *ctx = vb2_get_drv_priv(q); |
|
|
|
vicodec_return_bufs(q, VB2_BUF_STATE_ERROR); |
|
|
|
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, &vicodec_eos_event); |
|
|
|
if (!ctx->is_enc && V4L2_TYPE_IS_OUTPUT(q->type)) |
|
ctx->first_source_change_sent = false; |
|
|
|
if ((!V4L2_TYPE_IS_OUTPUT(q->type) && !ctx->is_enc) || |
|
(V4L2_TYPE_IS_OUTPUT(q->type) && ctx->is_enc)) { |
|
if (!ctx->is_stateless) |
|
kvfree(ctx->state.ref_frame.buf); |
|
ctx->state.ref_frame.buf = NULL; |
|
ctx->state.ref_frame.luma = NULL; |
|
ctx->comp_max_size = 0; |
|
ctx->source_changed = false; |
|
} |
|
if (V4L2_TYPE_IS_OUTPUT(q->type) && !ctx->is_enc) { |
|
ctx->cur_buf_offset = 0; |
|
ctx->comp_size = 0; |
|
ctx->header_size = 0; |
|
ctx->comp_magic_cnt = 0; |
|
ctx->comp_has_frame = false; |
|
ctx->comp_has_next_frame = false; |
|
} |
|
} |
|
|
|
static void vicodec_buf_request_complete(struct vb2_buffer *vb) |
|
{ |
|
struct vicodec_ctx *ctx = vb2_get_drv_priv(vb->vb2_queue); |
|
|
|
v4l2_ctrl_request_complete(vb->req_obj.req, &ctx->hdl); |
|
} |
|
|
|
|
|
static const struct vb2_ops vicodec_qops = { |
|
.queue_setup = vicodec_queue_setup, |
|
.buf_out_validate = vicodec_buf_out_validate, |
|
.buf_prepare = vicodec_buf_prepare, |
|
.buf_queue = vicodec_buf_queue, |
|
.buf_request_complete = vicodec_buf_request_complete, |
|
.start_streaming = vicodec_start_streaming, |
|
.stop_streaming = vicodec_stop_streaming, |
|
.wait_prepare = vb2_ops_wait_prepare, |
|
.wait_finish = vb2_ops_wait_finish, |
|
}; |
|
|
|
static int queue_init(void *priv, struct vb2_queue *src_vq, |
|
struct vb2_queue *dst_vq) |
|
{ |
|
struct vicodec_ctx *ctx = priv; |
|
int ret; |
|
|
|
src_vq->type = (multiplanar ? |
|
V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE : |
|
V4L2_BUF_TYPE_VIDEO_OUTPUT); |
|
src_vq->io_modes = VB2_MMAP | VB2_USERPTR | VB2_DMABUF; |
|
src_vq->drv_priv = ctx; |
|
src_vq->buf_struct_size = sizeof(struct v4l2_m2m_buffer); |
|
src_vq->ops = &vicodec_qops; |
|
src_vq->mem_ops = &vb2_vmalloc_memops; |
|
src_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY; |
|
if (ctx->is_enc) |
|
src_vq->lock = &ctx->dev->stateful_enc.mutex; |
|
else if (ctx->is_stateless) |
|
src_vq->lock = &ctx->dev->stateless_dec.mutex; |
|
else |
|
src_vq->lock = &ctx->dev->stateful_dec.mutex; |
|
src_vq->supports_requests = ctx->is_stateless; |
|
src_vq->requires_requests = ctx->is_stateless; |
|
ret = vb2_queue_init(src_vq); |
|
if (ret) |
|
return ret; |
|
|
|
dst_vq->type = (multiplanar ? |
|
V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE : |
|
V4L2_BUF_TYPE_VIDEO_CAPTURE); |
|
dst_vq->io_modes = VB2_MMAP | VB2_USERPTR | VB2_DMABUF; |
|
dst_vq->drv_priv = ctx; |
|
dst_vq->buf_struct_size = sizeof(struct v4l2_m2m_buffer); |
|
dst_vq->ops = &vicodec_qops; |
|
dst_vq->mem_ops = &vb2_vmalloc_memops; |
|
dst_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY; |
|
dst_vq->lock = src_vq->lock; |
|
|
|
return vb2_queue_init(dst_vq); |
|
} |
|
|
|
static int vicodec_try_ctrl(struct v4l2_ctrl *ctrl) |
|
{ |
|
struct vicodec_ctx *ctx = container_of(ctrl->handler, |
|
struct vicodec_ctx, hdl); |
|
const struct v4l2_ctrl_fwht_params *params; |
|
struct vicodec_q_data *q_dst = get_q_data(ctx, |
|
V4L2_BUF_TYPE_VIDEO_CAPTURE); |
|
|
|
switch (ctrl->id) { |
|
case V4L2_CID_STATELESS_FWHT_PARAMS: |
|
if (!q_dst->info) |
|
return -EINVAL; |
|
params = ctrl->p_new.p_fwht_params; |
|
if (params->width > q_dst->coded_width || |
|
params->width < MIN_WIDTH || |
|
params->height > q_dst->coded_height || |
|
params->height < MIN_HEIGHT) |
|
return -EINVAL; |
|
if (!validate_by_version(params->flags, params->version)) |
|
return -EINVAL; |
|
if (!validate_stateless_params_flags(params, q_dst->info)) |
|
return -EINVAL; |
|
return 0; |
|
default: |
|
return 0; |
|
} |
|
return 0; |
|
} |
|
|
|
static void update_header_from_stateless_params(struct vicodec_ctx *ctx, |
|
const struct v4l2_ctrl_fwht_params *params) |
|
{ |
|
struct fwht_cframe_hdr *p_hdr = &ctx->state.header; |
|
|
|
p_hdr->magic1 = FWHT_MAGIC1; |
|
p_hdr->magic2 = FWHT_MAGIC2; |
|
p_hdr->version = htonl(params->version); |
|
p_hdr->width = htonl(params->width); |
|
p_hdr->height = htonl(params->height); |
|
p_hdr->flags = htonl(params->flags); |
|
p_hdr->colorspace = htonl(params->colorspace); |
|
p_hdr->xfer_func = htonl(params->xfer_func); |
|
p_hdr->ycbcr_enc = htonl(params->ycbcr_enc); |
|
p_hdr->quantization = htonl(params->quantization); |
|
} |
|
|
|
static int vicodec_s_ctrl(struct v4l2_ctrl *ctrl) |
|
{ |
|
struct vicodec_ctx *ctx = container_of(ctrl->handler, |
|
struct vicodec_ctx, hdl); |
|
const struct v4l2_ctrl_fwht_params *params; |
|
|
|
switch (ctrl->id) { |
|
case V4L2_CID_MPEG_VIDEO_GOP_SIZE: |
|
ctx->state.gop_size = ctrl->val; |
|
return 0; |
|
case V4L2_CID_FWHT_I_FRAME_QP: |
|
ctx->state.i_frame_qp = ctrl->val; |
|
return 0; |
|
case V4L2_CID_FWHT_P_FRAME_QP: |
|
ctx->state.p_frame_qp = ctrl->val; |
|
return 0; |
|
case V4L2_CID_STATELESS_FWHT_PARAMS: |
|
params = ctrl->p_new.p_fwht_params; |
|
update_header_from_stateless_params(ctx, params); |
|
ctx->state.ref_frame_ts = params->backward_ref_ts; |
|
return 0; |
|
} |
|
return -EINVAL; |
|
} |
|
|
|
static const struct v4l2_ctrl_ops vicodec_ctrl_ops = { |
|
.s_ctrl = vicodec_s_ctrl, |
|
.try_ctrl = vicodec_try_ctrl, |
|
}; |
|
|
|
static const struct v4l2_ctrl_config vicodec_ctrl_stateless_state = { |
|
.ops = &vicodec_ctrl_ops, |
|
.id = V4L2_CID_STATELESS_FWHT_PARAMS, |
|
.elem_size = sizeof(struct v4l2_ctrl_fwht_params), |
|
}; |
|
|
|
/* |
|
* File operations |
|
*/ |
|
static int vicodec_open(struct file *file) |
|
{ |
|
const struct v4l2_fwht_pixfmt_info *info = v4l2_fwht_get_pixfmt(0); |
|
struct video_device *vfd = video_devdata(file); |
|
struct vicodec_dev *dev = video_drvdata(file); |
|
struct vicodec_ctx *ctx = NULL; |
|
struct v4l2_ctrl_handler *hdl; |
|
unsigned int raw_size; |
|
unsigned int comp_size; |
|
int rc = 0; |
|
|
|
if (mutex_lock_interruptible(vfd->lock)) |
|
return -ERESTARTSYS; |
|
ctx = kzalloc(sizeof(*ctx), GFP_KERNEL); |
|
if (!ctx) { |
|
rc = -ENOMEM; |
|
goto open_unlock; |
|
} |
|
|
|
if (vfd == &dev->stateful_enc.vfd) |
|
ctx->is_enc = true; |
|
else if (vfd == &dev->stateless_dec.vfd) |
|
ctx->is_stateless = true; |
|
|
|
v4l2_fh_init(&ctx->fh, video_devdata(file)); |
|
file->private_data = &ctx->fh; |
|
ctx->dev = dev; |
|
hdl = &ctx->hdl; |
|
v4l2_ctrl_handler_init(hdl, 5); |
|
v4l2_ctrl_new_std(hdl, &vicodec_ctrl_ops, V4L2_CID_MPEG_VIDEO_GOP_SIZE, |
|
1, 16, 1, 10); |
|
v4l2_ctrl_new_std(hdl, &vicodec_ctrl_ops, V4L2_CID_FWHT_I_FRAME_QP, |
|
1, 31, 1, 20); |
|
v4l2_ctrl_new_std(hdl, &vicodec_ctrl_ops, V4L2_CID_FWHT_P_FRAME_QP, |
|
1, 31, 1, 20); |
|
if (ctx->is_enc) |
|
v4l2_ctrl_new_std(hdl, &vicodec_ctrl_ops, |
|
V4L2_CID_MIN_BUFFERS_FOR_OUTPUT, 1, 1, 1, 1); |
|
if (ctx->is_stateless) |
|
v4l2_ctrl_new_custom(hdl, &vicodec_ctrl_stateless_state, NULL); |
|
if (hdl->error) { |
|
rc = hdl->error; |
|
v4l2_ctrl_handler_free(hdl); |
|
kfree(ctx); |
|
goto open_unlock; |
|
} |
|
ctx->fh.ctrl_handler = hdl; |
|
v4l2_ctrl_handler_setup(hdl); |
|
|
|
if (ctx->is_enc) |
|
ctx->q_data[V4L2_M2M_SRC].info = info; |
|
else if (ctx->is_stateless) |
|
ctx->q_data[V4L2_M2M_SRC].info = &pixfmt_stateless_fwht; |
|
else |
|
ctx->q_data[V4L2_M2M_SRC].info = &pixfmt_fwht; |
|
ctx->q_data[V4L2_M2M_SRC].coded_width = 1280; |
|
ctx->q_data[V4L2_M2M_SRC].coded_height = 720; |
|
ctx->q_data[V4L2_M2M_SRC].visible_width = 1280; |
|
ctx->q_data[V4L2_M2M_SRC].visible_height = 720; |
|
raw_size = 1280 * 720 * info->sizeimage_mult / info->sizeimage_div; |
|
comp_size = 1280 * 720 * pixfmt_fwht.sizeimage_mult / |
|
pixfmt_fwht.sizeimage_div; |
|
if (ctx->is_enc) |
|
ctx->q_data[V4L2_M2M_SRC].sizeimage = raw_size; |
|
else if (ctx->is_stateless) |
|
ctx->q_data[V4L2_M2M_SRC].sizeimage = comp_size; |
|
else |
|
ctx->q_data[V4L2_M2M_SRC].sizeimage = |
|
comp_size + sizeof(struct fwht_cframe_hdr); |
|
ctx->q_data[V4L2_M2M_DST] = ctx->q_data[V4L2_M2M_SRC]; |
|
if (ctx->is_enc) { |
|
ctx->q_data[V4L2_M2M_DST].info = &pixfmt_fwht; |
|
ctx->q_data[V4L2_M2M_DST].sizeimage = |
|
comp_size + sizeof(struct fwht_cframe_hdr); |
|
} else { |
|
ctx->q_data[V4L2_M2M_DST].info = info; |
|
ctx->q_data[V4L2_M2M_DST].sizeimage = raw_size; |
|
} |
|
|
|
ctx->state.colorspace = V4L2_COLORSPACE_REC709; |
|
|
|
if (ctx->is_enc) { |
|
ctx->fh.m2m_ctx = v4l2_m2m_ctx_init(dev->stateful_enc.m2m_dev, |
|
ctx, &queue_init); |
|
ctx->lock = &dev->stateful_enc.lock; |
|
} else if (ctx->is_stateless) { |
|
ctx->fh.m2m_ctx = v4l2_m2m_ctx_init(dev->stateless_dec.m2m_dev, |
|
ctx, &queue_init); |
|
ctx->lock = &dev->stateless_dec.lock; |
|
} else { |
|
ctx->fh.m2m_ctx = v4l2_m2m_ctx_init(dev->stateful_dec.m2m_dev, |
|
ctx, &queue_init); |
|
ctx->lock = &dev->stateful_dec.lock; |
|
} |
|
|
|
if (IS_ERR(ctx->fh.m2m_ctx)) { |
|
rc = PTR_ERR(ctx->fh.m2m_ctx); |
|
|
|
v4l2_ctrl_handler_free(hdl); |
|
v4l2_fh_exit(&ctx->fh); |
|
kfree(ctx); |
|
goto open_unlock; |
|
} |
|
|
|
v4l2_fh_add(&ctx->fh); |
|
|
|
open_unlock: |
|
mutex_unlock(vfd->lock); |
|
return rc; |
|
} |
|
|
|
static int vicodec_release(struct file *file) |
|
{ |
|
struct video_device *vfd = video_devdata(file); |
|
struct vicodec_ctx *ctx = file2ctx(file); |
|
|
|
mutex_lock(vfd->lock); |
|
v4l2_m2m_ctx_release(ctx->fh.m2m_ctx); |
|
mutex_unlock(vfd->lock); |
|
v4l2_fh_del(&ctx->fh); |
|
v4l2_fh_exit(&ctx->fh); |
|
v4l2_ctrl_handler_free(&ctx->hdl); |
|
kvfree(ctx->state.compressed_frame); |
|
kfree(ctx); |
|
|
|
return 0; |
|
} |
|
|
|
static int vicodec_request_validate(struct media_request *req) |
|
{ |
|
struct media_request_object *obj; |
|
struct v4l2_ctrl_handler *parent_hdl, *hdl; |
|
struct vicodec_ctx *ctx = NULL; |
|
struct v4l2_ctrl *ctrl; |
|
unsigned int count; |
|
|
|
list_for_each_entry(obj, &req->objects, list) { |
|
struct vb2_buffer *vb; |
|
|
|
if (vb2_request_object_is_buffer(obj)) { |
|
vb = container_of(obj, struct vb2_buffer, req_obj); |
|
ctx = vb2_get_drv_priv(vb->vb2_queue); |
|
|
|
break; |
|
} |
|
} |
|
|
|
if (!ctx) { |
|
pr_err("No buffer was provided with the request\n"); |
|
return -ENOENT; |
|
} |
|
|
|
count = vb2_request_buffer_cnt(req); |
|
if (!count) { |
|
v4l2_info(&ctx->dev->v4l2_dev, |
|
"No buffer was provided with the request\n"); |
|
return -ENOENT; |
|
} else if (count > 1) { |
|
v4l2_info(&ctx->dev->v4l2_dev, |
|
"More than one buffer was provided with the request\n"); |
|
return -EINVAL; |
|
} |
|
|
|
parent_hdl = &ctx->hdl; |
|
|
|
hdl = v4l2_ctrl_request_hdl_find(req, parent_hdl); |
|
if (!hdl) { |
|
v4l2_info(&ctx->dev->v4l2_dev, "Missing codec control\n"); |
|
return -ENOENT; |
|
} |
|
ctrl = v4l2_ctrl_request_hdl_ctrl_find(hdl, |
|
vicodec_ctrl_stateless_state.id); |
|
v4l2_ctrl_request_hdl_put(hdl); |
|
if (!ctrl) { |
|
v4l2_info(&ctx->dev->v4l2_dev, |
|
"Missing required codec control\n"); |
|
return -ENOENT; |
|
} |
|
|
|
return vb2_request_validate(req); |
|
} |
|
|
|
static const struct v4l2_file_operations vicodec_fops = { |
|
.owner = THIS_MODULE, |
|
.open = vicodec_open, |
|
.release = vicodec_release, |
|
.poll = v4l2_m2m_fop_poll, |
|
.unlocked_ioctl = video_ioctl2, |
|
.mmap = v4l2_m2m_fop_mmap, |
|
}; |
|
|
|
static const struct video_device vicodec_videodev = { |
|
.name = VICODEC_NAME, |
|
.vfl_dir = VFL_DIR_M2M, |
|
.fops = &vicodec_fops, |
|
.ioctl_ops = &vicodec_ioctl_ops, |
|
.minor = -1, |
|
.release = video_device_release_empty, |
|
}; |
|
|
|
static const struct media_device_ops vicodec_m2m_media_ops = { |
|
.req_validate = vicodec_request_validate, |
|
.req_queue = v4l2_m2m_request_queue, |
|
}; |
|
|
|
static const struct v4l2_m2m_ops m2m_ops = { |
|
.device_run = device_run, |
|
.job_ready = job_ready, |
|
}; |
|
|
|
static int register_instance(struct vicodec_dev *dev, |
|
struct vicodec_dev_instance *dev_instance, |
|
const char *name, bool is_enc) |
|
{ |
|
struct video_device *vfd; |
|
int ret; |
|
|
|
spin_lock_init(&dev_instance->lock); |
|
mutex_init(&dev_instance->mutex); |
|
dev_instance->m2m_dev = v4l2_m2m_init(&m2m_ops); |
|
if (IS_ERR(dev_instance->m2m_dev)) { |
|
v4l2_err(&dev->v4l2_dev, "Failed to init vicodec enc device\n"); |
|
return PTR_ERR(dev_instance->m2m_dev); |
|
} |
|
|
|
dev_instance->vfd = vicodec_videodev; |
|
vfd = &dev_instance->vfd; |
|
vfd->lock = &dev_instance->mutex; |
|
vfd->v4l2_dev = &dev->v4l2_dev; |
|
strscpy(vfd->name, name, sizeof(vfd->name)); |
|
vfd->device_caps = V4L2_CAP_STREAMING | |
|
(multiplanar ? V4L2_CAP_VIDEO_M2M_MPLANE : V4L2_CAP_VIDEO_M2M); |
|
if (is_enc) { |
|
v4l2_disable_ioctl(vfd, VIDIOC_DECODER_CMD); |
|
v4l2_disable_ioctl(vfd, VIDIOC_TRY_DECODER_CMD); |
|
} else { |
|
v4l2_disable_ioctl(vfd, VIDIOC_ENCODER_CMD); |
|
v4l2_disable_ioctl(vfd, VIDIOC_TRY_ENCODER_CMD); |
|
} |
|
video_set_drvdata(vfd, dev); |
|
|
|
ret = video_register_device(vfd, VFL_TYPE_VIDEO, 0); |
|
if (ret) { |
|
v4l2_err(&dev->v4l2_dev, "Failed to register video device '%s'\n", name); |
|
v4l2_m2m_release(dev_instance->m2m_dev); |
|
return ret; |
|
} |
|
v4l2_info(&dev->v4l2_dev, "Device '%s' registered as /dev/video%d\n", |
|
name, vfd->num); |
|
return 0; |
|
} |
|
|
|
static void vicodec_v4l2_dev_release(struct v4l2_device *v4l2_dev) |
|
{ |
|
struct vicodec_dev *dev = container_of(v4l2_dev, struct vicodec_dev, v4l2_dev); |
|
|
|
v4l2_device_unregister(&dev->v4l2_dev); |
|
v4l2_m2m_release(dev->stateful_enc.m2m_dev); |
|
v4l2_m2m_release(dev->stateful_dec.m2m_dev); |
|
v4l2_m2m_release(dev->stateless_dec.m2m_dev); |
|
#ifdef CONFIG_MEDIA_CONTROLLER |
|
media_device_cleanup(&dev->mdev); |
|
#endif |
|
kfree(dev); |
|
} |
|
|
|
static int vicodec_probe(struct platform_device *pdev) |
|
{ |
|
struct vicodec_dev *dev; |
|
int ret; |
|
|
|
dev = kzalloc(sizeof(*dev), GFP_KERNEL); |
|
if (!dev) |
|
return -ENOMEM; |
|
|
|
ret = v4l2_device_register(&pdev->dev, &dev->v4l2_dev); |
|
if (ret) |
|
goto free_dev; |
|
|
|
dev->v4l2_dev.release = vicodec_v4l2_dev_release; |
|
|
|
#ifdef CONFIG_MEDIA_CONTROLLER |
|
dev->mdev.dev = &pdev->dev; |
|
strscpy(dev->mdev.model, "vicodec", sizeof(dev->mdev.model)); |
|
strscpy(dev->mdev.bus_info, "platform:vicodec", |
|
sizeof(dev->mdev.bus_info)); |
|
media_device_init(&dev->mdev); |
|
dev->mdev.ops = &vicodec_m2m_media_ops; |
|
dev->v4l2_dev.mdev = &dev->mdev; |
|
#endif |
|
|
|
platform_set_drvdata(pdev, dev); |
|
|
|
ret = register_instance(dev, &dev->stateful_enc, "stateful-encoder", |
|
true); |
|
if (ret) |
|
goto unreg_dev; |
|
|
|
ret = register_instance(dev, &dev->stateful_dec, "stateful-decoder", |
|
false); |
|
if (ret) |
|
goto unreg_sf_enc; |
|
|
|
ret = register_instance(dev, &dev->stateless_dec, "stateless-decoder", |
|
false); |
|
if (ret) |
|
goto unreg_sf_dec; |
|
|
|
#ifdef CONFIG_MEDIA_CONTROLLER |
|
ret = v4l2_m2m_register_media_controller(dev->stateful_enc.m2m_dev, |
|
&dev->stateful_enc.vfd, |
|
MEDIA_ENT_F_PROC_VIDEO_ENCODER); |
|
if (ret) { |
|
v4l2_err(&dev->v4l2_dev, "Failed to init mem2mem media controller for enc\n"); |
|
goto unreg_m2m; |
|
} |
|
|
|
ret = v4l2_m2m_register_media_controller(dev->stateful_dec.m2m_dev, |
|
&dev->stateful_dec.vfd, |
|
MEDIA_ENT_F_PROC_VIDEO_DECODER); |
|
if (ret) { |
|
v4l2_err(&dev->v4l2_dev, "Failed to init mem2mem media controller for dec\n"); |
|
goto unreg_m2m_sf_enc_mc; |
|
} |
|
|
|
ret = v4l2_m2m_register_media_controller(dev->stateless_dec.m2m_dev, |
|
&dev->stateless_dec.vfd, |
|
MEDIA_ENT_F_PROC_VIDEO_DECODER); |
|
if (ret) { |
|
v4l2_err(&dev->v4l2_dev, "Failed to init mem2mem media controller for stateless dec\n"); |
|
goto unreg_m2m_sf_dec_mc; |
|
} |
|
|
|
ret = media_device_register(&dev->mdev); |
|
if (ret) { |
|
v4l2_err(&dev->v4l2_dev, "Failed to register mem2mem media device\n"); |
|
goto unreg_m2m_sl_dec_mc; |
|
} |
|
#endif |
|
return 0; |
|
|
|
#ifdef CONFIG_MEDIA_CONTROLLER |
|
unreg_m2m_sl_dec_mc: |
|
v4l2_m2m_unregister_media_controller(dev->stateless_dec.m2m_dev); |
|
unreg_m2m_sf_dec_mc: |
|
v4l2_m2m_unregister_media_controller(dev->stateful_dec.m2m_dev); |
|
unreg_m2m_sf_enc_mc: |
|
v4l2_m2m_unregister_media_controller(dev->stateful_enc.m2m_dev); |
|
unreg_m2m: |
|
video_unregister_device(&dev->stateless_dec.vfd); |
|
v4l2_m2m_release(dev->stateless_dec.m2m_dev); |
|
#endif |
|
unreg_sf_dec: |
|
video_unregister_device(&dev->stateful_dec.vfd); |
|
v4l2_m2m_release(dev->stateful_dec.m2m_dev); |
|
unreg_sf_enc: |
|
video_unregister_device(&dev->stateful_enc.vfd); |
|
v4l2_m2m_release(dev->stateful_enc.m2m_dev); |
|
unreg_dev: |
|
v4l2_device_unregister(&dev->v4l2_dev); |
|
free_dev: |
|
kfree(dev); |
|
|
|
return ret; |
|
} |
|
|
|
static int vicodec_remove(struct platform_device *pdev) |
|
{ |
|
struct vicodec_dev *dev = platform_get_drvdata(pdev); |
|
|
|
v4l2_info(&dev->v4l2_dev, "Removing " VICODEC_NAME); |
|
|
|
#ifdef CONFIG_MEDIA_CONTROLLER |
|
media_device_unregister(&dev->mdev); |
|
v4l2_m2m_unregister_media_controller(dev->stateful_enc.m2m_dev); |
|
v4l2_m2m_unregister_media_controller(dev->stateful_dec.m2m_dev); |
|
v4l2_m2m_unregister_media_controller(dev->stateless_dec.m2m_dev); |
|
#endif |
|
|
|
video_unregister_device(&dev->stateful_enc.vfd); |
|
video_unregister_device(&dev->stateful_dec.vfd); |
|
video_unregister_device(&dev->stateless_dec.vfd); |
|
v4l2_device_put(&dev->v4l2_dev); |
|
|
|
return 0; |
|
} |
|
|
|
static struct platform_driver vicodec_pdrv = { |
|
.probe = vicodec_probe, |
|
.remove = vicodec_remove, |
|
.driver = { |
|
.name = VICODEC_NAME, |
|
}, |
|
}; |
|
|
|
static void __exit vicodec_exit(void) |
|
{ |
|
platform_driver_unregister(&vicodec_pdrv); |
|
platform_device_unregister(&vicodec_pdev); |
|
} |
|
|
|
static int __init vicodec_init(void) |
|
{ |
|
int ret; |
|
|
|
ret = platform_device_register(&vicodec_pdev); |
|
if (ret) |
|
return ret; |
|
|
|
ret = platform_driver_register(&vicodec_pdrv); |
|
if (ret) |
|
platform_device_unregister(&vicodec_pdev); |
|
|
|
return ret; |
|
} |
|
|
|
module_init(vicodec_init); |
|
module_exit(vicodec_exit);
|
|
|