2239 lines · c
1// SPDX-License-Identifier: GPL-2.0-only2/*3 * A virtual codec example device.4 *5 * Copyright 2018 Cisco Systems, Inc. and/or its affiliates. All rights reserved.6 *7 * This is a virtual codec device driver for testing the codec framework.8 * It simulates a device that uses memory buffers for both source and9 * destination and encodes or decodes the data.10 */11 12#include <linux/module.h>13#include <linux/delay.h>14#include <linux/fs.h>15#include <linux/sched.h>16#include <linux/slab.h>17 18#include <linux/platform_device.h>19#include <media/v4l2-mem2mem.h>20#include <media/v4l2-device.h>21#include <media/v4l2-ioctl.h>22#include <media/v4l2-ctrls.h>23#include <media/v4l2-event.h>24#include <media/videobuf2-vmalloc.h>25 26#include "codec-v4l2-fwht.h"27 28MODULE_DESCRIPTION("Virtual codec device");29MODULE_AUTHOR("Hans Verkuil <hans.verkuil@cisco.com>");30MODULE_LICENSE("GPL v2");31 32static bool multiplanar;33module_param(multiplanar, bool, 0444);34MODULE_PARM_DESC(multiplanar,35 " use multi-planar API instead of single-planar API");36 37static unsigned int debug;38module_param(debug, uint, 0644);39MODULE_PARM_DESC(debug, " activates debug info");40 41#define VICODEC_NAME "vicodec"42#define MAX_WIDTH 4096U43#define MIN_WIDTH 640U44#define MAX_HEIGHT 2160U45#define MIN_HEIGHT 360U46 47#define dprintk(dev, fmt, arg...) \48 v4l2_dbg(1, debug, &dev->v4l2_dev, "%s: " fmt, __func__, ## arg)49 50 51struct pixfmt_info {52 u32 id;53 unsigned int bytesperline_mult;54 unsigned int sizeimage_mult;55 unsigned int sizeimage_div;56 unsigned int luma_step;57 unsigned int chroma_step;58 /* Chroma plane subsampling */59 unsigned int width_div;60 unsigned int height_div;61};62 63static const struct v4l2_fwht_pixfmt_info pixfmt_fwht = {64 V4L2_PIX_FMT_FWHT, 0, 3, 1, 1, 1, 1, 1, 0, 165};66 67static const struct v4l2_fwht_pixfmt_info pixfmt_stateless_fwht = {68 V4L2_PIX_FMT_FWHT_STATELESS, 0, 3, 1, 1, 1, 1, 1, 0, 169};70 71static void vicodec_dev_release(struct device *dev)72{73}74 75static struct platform_device vicodec_pdev = {76 .name = VICODEC_NAME,77 .dev.release = vicodec_dev_release,78};79 80/* Per-queue, driver-specific private data */81struct vicodec_q_data {82 unsigned int coded_width;83 unsigned int coded_height;84 unsigned int visible_width;85 unsigned int visible_height;86 unsigned int sizeimage;87 unsigned int vb2_sizeimage;88 unsigned int sequence;89 const struct v4l2_fwht_pixfmt_info *info;90};91 92enum {93 V4L2_M2M_SRC = 0,94 V4L2_M2M_DST = 1,95};96 97struct vicodec_dev_instance {98 struct video_device vfd;99 struct mutex mutex;100 spinlock_t lock;101 struct v4l2_m2m_dev *m2m_dev;102};103 104struct vicodec_dev {105 struct v4l2_device v4l2_dev;106 struct vicodec_dev_instance stateful_enc;107 struct vicodec_dev_instance stateful_dec;108 struct vicodec_dev_instance stateless_dec;109#ifdef CONFIG_MEDIA_CONTROLLER110 struct media_device mdev;111#endif112 113};114 115struct vicodec_ctx {116 struct v4l2_fh fh;117 struct vicodec_dev *dev;118 bool is_enc;119 bool is_stateless;120 spinlock_t *lock;121 122 struct v4l2_ctrl_handler hdl;123 124 /* Source and destination queue data */125 struct vicodec_q_data q_data[2];126 struct v4l2_fwht_state state;127 128 u32 cur_buf_offset;129 u32 comp_max_size;130 u32 comp_size;131 u32 header_size;132 u32 comp_magic_cnt;133 bool comp_has_frame;134 bool comp_has_next_frame;135 bool first_source_change_sent;136 bool source_changed;137};138 139static const struct v4l2_event vicodec_eos_event = {140 .type = V4L2_EVENT_EOS141};142 143static inline struct vicodec_ctx *file2ctx(struct file *file)144{145 return container_of(file->private_data, struct vicodec_ctx, fh);146}147 148static struct vicodec_q_data *get_q_data(struct vicodec_ctx *ctx,149 enum v4l2_buf_type type)150{151 switch (type) {152 case V4L2_BUF_TYPE_VIDEO_OUTPUT:153 case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE:154 return &ctx->q_data[V4L2_M2M_SRC];155 case V4L2_BUF_TYPE_VIDEO_CAPTURE:156 case V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE:157 return &ctx->q_data[V4L2_M2M_DST];158 default:159 break;160 }161 return NULL;162}163 164static void copy_cap_to_ref(const u8 *cap, const struct v4l2_fwht_pixfmt_info *info,165 struct v4l2_fwht_state *state)166{167 int plane_idx;168 u8 *p_ref = state->ref_frame.buf;169 unsigned int cap_stride = state->stride;170 unsigned int ref_stride = state->ref_stride;171 172 for (plane_idx = 0; plane_idx < info->planes_num; plane_idx++) {173 int i;174 unsigned int h_div = (plane_idx == 1 || plane_idx == 2) ?175 info->height_div : 1;176 const u8 *row_cap = cap;177 u8 *row_ref = p_ref;178 179 if (info->planes_num == 3 && plane_idx == 1) {180 cap_stride /= 2;181 ref_stride /= 2;182 }183 184 if (plane_idx == 1 &&185 (info->id == V4L2_PIX_FMT_NV24 ||186 info->id == V4L2_PIX_FMT_NV42)) {187 cap_stride *= 2;188 ref_stride *= 2;189 }190 191 for (i = 0; i < state->visible_height / h_div; i++) {192 memcpy(row_ref, row_cap, ref_stride);193 row_ref += ref_stride;194 row_cap += cap_stride;195 }196 cap += cap_stride * (state->coded_height / h_div);197 p_ref += ref_stride * (state->coded_height / h_div);198 }199}200 201static bool validate_by_version(unsigned int flags, unsigned int version)202{203 if (!version || version > V4L2_FWHT_VERSION)204 return false;205 206 if (version >= 2) {207 unsigned int components_num = 1 +208 ((flags & V4L2_FWHT_FL_COMPONENTS_NUM_MSK) >>209 V4L2_FWHT_FL_COMPONENTS_NUM_OFFSET);210 unsigned int pixenc = flags & V4L2_FWHT_FL_PIXENC_MSK;211 212 if (components_num == 0 || components_num > 4 || !pixenc)213 return false;214 }215 return true;216}217 218static bool validate_stateless_params_flags(const struct v4l2_ctrl_fwht_params *params,219 const struct v4l2_fwht_pixfmt_info *cur_info)220{221 unsigned int width_div =222 (params->flags & V4L2_FWHT_FL_CHROMA_FULL_WIDTH) ? 1 : 2;223 unsigned int height_div =224 (params->flags & V4L2_FWHT_FL_CHROMA_FULL_HEIGHT) ? 1 : 2;225 unsigned int components_num = 3;226 unsigned int pixenc = 0;227 228 if (params->version < 3)229 return false;230 231 components_num = 1 + ((params->flags & V4L2_FWHT_FL_COMPONENTS_NUM_MSK) >>232 V4L2_FWHT_FL_COMPONENTS_NUM_OFFSET);233 pixenc = (params->flags & V4L2_FWHT_FL_PIXENC_MSK);234 if (v4l2_fwht_validate_fmt(cur_info, width_div, height_div,235 components_num, pixenc))236 return true;237 return false;238}239 240 241static void update_state_from_header(struct vicodec_ctx *ctx)242{243 const struct fwht_cframe_hdr *p_hdr = &ctx->state.header;244 245 ctx->state.visible_width = ntohl(p_hdr->width);246 ctx->state.visible_height = ntohl(p_hdr->height);247 ctx->state.colorspace = ntohl(p_hdr->colorspace);248 ctx->state.xfer_func = ntohl(p_hdr->xfer_func);249 ctx->state.ycbcr_enc = ntohl(p_hdr->ycbcr_enc);250 ctx->state.quantization = ntohl(p_hdr->quantization);251}252 253static int device_process(struct vicodec_ctx *ctx,254 struct vb2_v4l2_buffer *src_vb,255 struct vb2_v4l2_buffer *dst_vb)256{257 struct vicodec_dev *dev = ctx->dev;258 struct v4l2_fwht_state *state = &ctx->state;259 u8 *p_src, *p_dst;260 int ret = 0;261 262 if (ctx->is_enc || ctx->is_stateless)263 p_src = vb2_plane_vaddr(&src_vb->vb2_buf, 0);264 else265 p_src = state->compressed_frame;266 267 if (ctx->is_stateless) {268 struct media_request *src_req = src_vb->vb2_buf.req_obj.req;269 270 ret = v4l2_ctrl_request_setup(src_req, &ctx->hdl);271 if (ret)272 return ret;273 update_state_from_header(ctx);274 275 ctx->state.header.size =276 htonl(vb2_get_plane_payload(&src_vb->vb2_buf, 0));277 /*278 * set the reference buffer from the reference timestamp279 * only if this is a P-frame280 */281 if (!(ntohl(ctx->state.header.flags) & V4L2_FWHT_FL_I_FRAME)) {282 struct vb2_buffer *ref_vb2_buf;283 struct vb2_queue *vq_cap =284 v4l2_m2m_get_vq(ctx->fh.m2m_ctx,285 V4L2_BUF_TYPE_VIDEO_CAPTURE);286 287 ref_vb2_buf = vb2_find_buffer(vq_cap, ctx->state.ref_frame_ts);288 if (!ref_vb2_buf)289 return -EINVAL;290 if (ref_vb2_buf->state == VB2_BUF_STATE_ERROR)291 ret = -EINVAL;292 ctx->state.ref_frame.buf =293 vb2_plane_vaddr(ref_vb2_buf, 0);294 } else {295 ctx->state.ref_frame.buf = NULL;296 }297 }298 p_dst = vb2_plane_vaddr(&dst_vb->vb2_buf, 0);299 if (!p_src || !p_dst) {300 v4l2_err(&dev->v4l2_dev,301 "Acquiring kernel pointers to buffers failed\n");302 return -EFAULT;303 }304 305 if (ctx->is_enc) {306 struct vicodec_q_data *q_src;307 int comp_sz_or_errcode;308 309 q_src = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT);310 state->info = q_src->info;311 comp_sz_or_errcode = v4l2_fwht_encode(state, p_src, p_dst);312 if (comp_sz_or_errcode < 0)313 return comp_sz_or_errcode;314 vb2_set_plane_payload(&dst_vb->vb2_buf, 0, comp_sz_or_errcode);315 } else {316 struct vicodec_q_data *q_dst;317 unsigned int comp_frame_size = ntohl(ctx->state.header.size);318 319 q_dst = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE);320 if (comp_frame_size > ctx->comp_max_size)321 return -EINVAL;322 state->info = q_dst->info;323 ret = v4l2_fwht_decode(state, p_src, p_dst);324 if (ret < 0)325 return ret;326 if (!ctx->is_stateless)327 copy_cap_to_ref(p_dst, ctx->state.info, &ctx->state);328 329 vb2_set_plane_payload(&dst_vb->vb2_buf, 0, q_dst->sizeimage);330 if (ntohl(ctx->state.header.flags) & V4L2_FWHT_FL_I_FRAME)331 dst_vb->flags |= V4L2_BUF_FLAG_KEYFRAME;332 else333 dst_vb->flags |= V4L2_BUF_FLAG_PFRAME;334 }335 return ret;336}337 338/*339 * mem2mem callbacks340 */341static enum vb2_buffer_state get_next_header(struct vicodec_ctx *ctx,342 u8 **pp, u32 sz)343{344 static const u8 magic[] = {345 0x4f, 0x4f, 0x4f, 0x4f, 0xff, 0xff, 0xff, 0xff346 };347 u8 *p = *pp;348 u32 state;349 u8 *header = (u8 *)&ctx->state.header;350 351 state = VB2_BUF_STATE_DONE;352 353 if (!ctx->header_size) {354 state = VB2_BUF_STATE_ERROR;355 for (; p < *pp + sz; p++) {356 u32 copy;357 358 p = memchr(p, magic[ctx->comp_magic_cnt],359 *pp + sz - p);360 if (!p) {361 ctx->comp_magic_cnt = 0;362 p = *pp + sz;363 break;364 }365 copy = sizeof(magic) - ctx->comp_magic_cnt;366 if (*pp + sz - p < copy)367 copy = *pp + sz - p;368 369 memcpy(header + ctx->comp_magic_cnt, p, copy);370 ctx->comp_magic_cnt += copy;371 if (!memcmp(header, magic, ctx->comp_magic_cnt)) {372 p += copy;373 state = VB2_BUF_STATE_DONE;374 break;375 }376 ctx->comp_magic_cnt = 0;377 }378 if (ctx->comp_magic_cnt < sizeof(magic)) {379 *pp = p;380 return state;381 }382 ctx->header_size = sizeof(magic);383 }384 385 if (ctx->header_size < sizeof(struct fwht_cframe_hdr)) {386 u32 copy = sizeof(struct fwht_cframe_hdr) - ctx->header_size;387 388 if (*pp + sz - p < copy)389 copy = *pp + sz - p;390 391 memcpy(header + ctx->header_size, p, copy);392 p += copy;393 ctx->header_size += copy;394 }395 *pp = p;396 return state;397}398 399/* device_run() - prepares and starts the device */400static void device_run(void *priv)401{402 struct vicodec_ctx *ctx = priv;403 struct vicodec_dev *dev = ctx->dev;404 struct vb2_v4l2_buffer *src_buf, *dst_buf;405 struct vicodec_q_data *q_src, *q_dst;406 u32 state;407 struct media_request *src_req;408 409 src_buf = v4l2_m2m_next_src_buf(ctx->fh.m2m_ctx);410 dst_buf = v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx);411 src_req = src_buf->vb2_buf.req_obj.req;412 413 q_src = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT);414 q_dst = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE);415 416 state = VB2_BUF_STATE_DONE;417 if (device_process(ctx, src_buf, dst_buf))418 state = VB2_BUF_STATE_ERROR;419 else420 dst_buf->sequence = q_dst->sequence++;421 dst_buf->flags &= ~V4L2_BUF_FLAG_LAST;422 v4l2_m2m_buf_copy_metadata(src_buf, dst_buf, false);423 424 spin_lock(ctx->lock);425 if (!ctx->comp_has_next_frame &&426 v4l2_m2m_is_last_draining_src_buf(ctx->fh.m2m_ctx, src_buf)) {427 dst_buf->flags |= V4L2_BUF_FLAG_LAST;428 v4l2_event_queue_fh(&ctx->fh, &vicodec_eos_event);429 v4l2_m2m_mark_stopped(ctx->fh.m2m_ctx);430 }431 if (ctx->is_enc || ctx->is_stateless) {432 src_buf->sequence = q_src->sequence++;433 src_buf = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx);434 v4l2_m2m_buf_done(src_buf, state);435 } else if (vb2_get_plane_payload(&src_buf->vb2_buf, 0) == ctx->cur_buf_offset) {436 src_buf->sequence = q_src->sequence++;437 src_buf = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx);438 v4l2_m2m_buf_done(src_buf, state);439 ctx->cur_buf_offset = 0;440 ctx->comp_has_next_frame = false;441 }442 v4l2_m2m_buf_done(dst_buf, state);443 444 ctx->comp_size = 0;445 ctx->header_size = 0;446 ctx->comp_magic_cnt = 0;447 ctx->comp_has_frame = false;448 spin_unlock(ctx->lock);449 if (ctx->is_stateless && src_req)450 v4l2_ctrl_request_complete(src_req, &ctx->hdl);451 452 if (ctx->is_enc)453 v4l2_m2m_job_finish(dev->stateful_enc.m2m_dev, ctx->fh.m2m_ctx);454 else if (ctx->is_stateless)455 v4l2_m2m_job_finish(dev->stateless_dec.m2m_dev,456 ctx->fh.m2m_ctx);457 else458 v4l2_m2m_job_finish(dev->stateful_dec.m2m_dev, ctx->fh.m2m_ctx);459}460 461static void job_remove_src_buf(struct vicodec_ctx *ctx, u32 state)462{463 struct vb2_v4l2_buffer *src_buf;464 struct vicodec_q_data *q_src;465 466 q_src = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT);467 spin_lock(ctx->lock);468 src_buf = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx);469 src_buf->sequence = q_src->sequence++;470 v4l2_m2m_buf_done(src_buf, state);471 ctx->cur_buf_offset = 0;472 spin_unlock(ctx->lock);473}474 475static const struct v4l2_fwht_pixfmt_info *476info_from_header(const struct fwht_cframe_hdr *p_hdr)477{478 unsigned int flags = ntohl(p_hdr->flags);479 unsigned int width_div = (flags & V4L2_FWHT_FL_CHROMA_FULL_WIDTH) ? 1 : 2;480 unsigned int height_div = (flags & V4L2_FWHT_FL_CHROMA_FULL_HEIGHT) ? 1 : 2;481 unsigned int components_num = 3;482 unsigned int pixenc = 0;483 unsigned int version = ntohl(p_hdr->version);484 485 if (version >= 2) {486 components_num = 1 + ((flags & V4L2_FWHT_FL_COMPONENTS_NUM_MSK) >>487 V4L2_FWHT_FL_COMPONENTS_NUM_OFFSET);488 pixenc = (flags & V4L2_FWHT_FL_PIXENC_MSK);489 }490 return v4l2_fwht_find_nth_fmt(width_div, height_div,491 components_num, pixenc, 0);492}493 494static bool is_header_valid(const struct fwht_cframe_hdr *p_hdr)495{496 const struct v4l2_fwht_pixfmt_info *info;497 unsigned int w = ntohl(p_hdr->width);498 unsigned int h = ntohl(p_hdr->height);499 unsigned int version = ntohl(p_hdr->version);500 unsigned int flags = ntohl(p_hdr->flags);501 502 if (w < MIN_WIDTH || w > MAX_WIDTH || h < MIN_HEIGHT || h > MAX_HEIGHT)503 return false;504 505 if (!validate_by_version(flags, version))506 return false;507 508 info = info_from_header(p_hdr);509 if (!info)510 return false;511 return true;512}513 514static void update_capture_data_from_header(struct vicodec_ctx *ctx)515{516 struct vicodec_q_data *q_dst = get_q_data(ctx,517 V4L2_BUF_TYPE_VIDEO_CAPTURE);518 const struct fwht_cframe_hdr *p_hdr = &ctx->state.header;519 const struct v4l2_fwht_pixfmt_info *info = info_from_header(p_hdr);520 unsigned int flags = ntohl(p_hdr->flags);521 unsigned int hdr_width_div = (flags & V4L2_FWHT_FL_CHROMA_FULL_WIDTH) ? 1 : 2;522 unsigned int hdr_height_div = (flags & V4L2_FWHT_FL_CHROMA_FULL_HEIGHT) ? 1 : 2;523 524 /*525 * This function should not be used by a stateless codec since526 * it changes values in q_data that are not request specific527 */528 WARN_ON(ctx->is_stateless);529 530 q_dst->info = info;531 q_dst->visible_width = ntohl(p_hdr->width);532 q_dst->visible_height = ntohl(p_hdr->height);533 q_dst->coded_width = vic_round_dim(q_dst->visible_width, hdr_width_div);534 q_dst->coded_height = vic_round_dim(q_dst->visible_height,535 hdr_height_div);536 537 q_dst->sizeimage = q_dst->coded_width * q_dst->coded_height *538 q_dst->info->sizeimage_mult / q_dst->info->sizeimage_div;539 ctx->state.colorspace = ntohl(p_hdr->colorspace);540 541 ctx->state.xfer_func = ntohl(p_hdr->xfer_func);542 ctx->state.ycbcr_enc = ntohl(p_hdr->ycbcr_enc);543 ctx->state.quantization = ntohl(p_hdr->quantization);544}545 546static void set_last_buffer(struct vb2_v4l2_buffer *dst_buf,547 const struct vb2_v4l2_buffer *src_buf,548 struct vicodec_ctx *ctx)549{550 struct vicodec_q_data *q_dst = get_q_data(ctx,551 V4L2_BUF_TYPE_VIDEO_CAPTURE);552 553 vb2_set_plane_payload(&dst_buf->vb2_buf, 0, 0);554 dst_buf->sequence = q_dst->sequence++;555 556 v4l2_m2m_buf_copy_metadata(src_buf, dst_buf, !ctx->is_enc);557 dst_buf->flags |= V4L2_BUF_FLAG_LAST;558 v4l2_m2m_buf_done(dst_buf, VB2_BUF_STATE_DONE);559}560 561static int job_ready(void *priv)562{563 static const u8 magic[] = {564 0x4f, 0x4f, 0x4f, 0x4f, 0xff, 0xff, 0xff, 0xff565 };566 struct vicodec_ctx *ctx = priv;567 struct vb2_v4l2_buffer *src_buf;568 u8 *p_src;569 u8 *p;570 u32 sz;571 u32 state;572 struct vicodec_q_data *q_dst = get_q_data(ctx,573 V4L2_BUF_TYPE_VIDEO_CAPTURE);574 unsigned int flags;575 unsigned int hdr_width_div;576 unsigned int hdr_height_div;577 unsigned int max_to_copy;578 unsigned int comp_frame_size;579 580 if (ctx->source_changed)581 return 0;582 if (ctx->is_stateless || ctx->is_enc || ctx->comp_has_frame)583 return 1;584 585restart:586 ctx->comp_has_next_frame = false;587 src_buf = v4l2_m2m_next_src_buf(ctx->fh.m2m_ctx);588 if (!src_buf)589 return 0;590 p_src = vb2_plane_vaddr(&src_buf->vb2_buf, 0);591 sz = vb2_get_plane_payload(&src_buf->vb2_buf, 0);592 p = p_src + ctx->cur_buf_offset;593 594 state = VB2_BUF_STATE_DONE;595 596 if (ctx->header_size < sizeof(struct fwht_cframe_hdr)) {597 state = get_next_header(ctx, &p, p_src + sz - p);598 if (ctx->header_size < sizeof(struct fwht_cframe_hdr)) {599 if (v4l2_m2m_is_last_draining_src_buf(ctx->fh.m2m_ctx,600 src_buf))601 return 1;602 job_remove_src_buf(ctx, state);603 goto restart;604 }605 }606 607 comp_frame_size = ntohl(ctx->state.header.size);608 609 /*610 * The current scanned frame might be the first frame of a new611 * resolution so its size might be larger than ctx->comp_max_size.612 * In that case it is copied up to the current buffer capacity and613 * the copy will continue after allocating new large enough buffer614 * when restreaming615 */616 max_to_copy = min(comp_frame_size, ctx->comp_max_size);617 618 if (ctx->comp_size < max_to_copy) {619 u32 copy = max_to_copy - ctx->comp_size;620 621 if (copy > p_src + sz - p)622 copy = p_src + sz - p;623 624 memcpy(ctx->state.compressed_frame + ctx->comp_size,625 p, copy);626 p += copy;627 ctx->comp_size += copy;628 if (ctx->comp_size < max_to_copy) {629 if (v4l2_m2m_is_last_draining_src_buf(ctx->fh.m2m_ctx,630 src_buf))631 return 1;632 job_remove_src_buf(ctx, state);633 goto restart;634 }635 }636 ctx->cur_buf_offset = p - p_src;637 if (ctx->comp_size == comp_frame_size)638 ctx->comp_has_frame = true;639 ctx->comp_has_next_frame = false;640 if (ctx->comp_has_frame && sz - ctx->cur_buf_offset >=641 sizeof(struct fwht_cframe_hdr)) {642 struct fwht_cframe_hdr *p_hdr = (struct fwht_cframe_hdr *)p;643 u32 frame_size = ntohl(p_hdr->size);644 u32 remaining = sz - ctx->cur_buf_offset - sizeof(*p_hdr);645 646 if (!memcmp(p, magic, sizeof(magic)))647 ctx->comp_has_next_frame = remaining >= frame_size;648 }649 /*650 * if the header is invalid the device_run will just drop the frame651 * with an error652 */653 if (!is_header_valid(&ctx->state.header) && ctx->comp_has_frame)654 return 1;655 flags = ntohl(ctx->state.header.flags);656 hdr_width_div = (flags & V4L2_FWHT_FL_CHROMA_FULL_WIDTH) ? 1 : 2;657 hdr_height_div = (flags & V4L2_FWHT_FL_CHROMA_FULL_HEIGHT) ? 1 : 2;658 659 if (ntohl(ctx->state.header.width) != q_dst->visible_width ||660 ntohl(ctx->state.header.height) != q_dst->visible_height ||661 !q_dst->info ||662 hdr_width_div != q_dst->info->width_div ||663 hdr_height_div != q_dst->info->height_div) {664 static const struct v4l2_event rs_event = {665 .type = V4L2_EVENT_SOURCE_CHANGE,666 .u.src_change.changes = V4L2_EVENT_SRC_CH_RESOLUTION,667 };668 669 struct vb2_v4l2_buffer *dst_buf =670 v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx);671 672 update_capture_data_from_header(ctx);673 v4l2_event_queue_fh(&ctx->fh, &rs_event);674 set_last_buffer(dst_buf, src_buf, ctx);675 ctx->source_changed = true;676 return 0;677 }678 return 1;679}680 681/*682 * video ioctls683 */684 685static const struct v4l2_fwht_pixfmt_info *find_fmt(u32 fmt)686{687 const struct v4l2_fwht_pixfmt_info *info =688 v4l2_fwht_find_pixfmt(fmt);689 690 if (!info)691 info = v4l2_fwht_get_pixfmt(0);692 return info;693}694 695static int vidioc_querycap(struct file *file, void *priv,696 struct v4l2_capability *cap)697{698 strscpy(cap->driver, VICODEC_NAME, sizeof(cap->driver));699 strscpy(cap->card, VICODEC_NAME, sizeof(cap->card));700 snprintf(cap->bus_info, sizeof(cap->bus_info),701 "platform:%s", VICODEC_NAME);702 return 0;703}704 705static int enum_fmt(struct v4l2_fmtdesc *f, struct vicodec_ctx *ctx,706 bool is_out)707{708 bool is_uncomp = (ctx->is_enc && is_out) || (!ctx->is_enc && !is_out);709 710 if (V4L2_TYPE_IS_MULTIPLANAR(f->type) && !multiplanar)711 return -EINVAL;712 if (!V4L2_TYPE_IS_MULTIPLANAR(f->type) && multiplanar)713 return -EINVAL;714 715 if (is_uncomp) {716 const struct v4l2_fwht_pixfmt_info *info =717 get_q_data(ctx, f->type)->info;718 719 if (ctx->is_enc ||720 !vb2_is_streaming(&ctx->fh.m2m_ctx->cap_q_ctx.q))721 info = v4l2_fwht_get_pixfmt(f->index);722 else723 info = v4l2_fwht_find_nth_fmt(info->width_div,724 info->height_div,725 info->components_num,726 info->pixenc,727 f->index);728 if (!info)729 return -EINVAL;730 f->pixelformat = info->id;731 } else {732 if (f->index)733 return -EINVAL;734 f->pixelformat = ctx->is_stateless ?735 V4L2_PIX_FMT_FWHT_STATELESS : V4L2_PIX_FMT_FWHT;736 if (!ctx->is_enc && !ctx->is_stateless)737 f->flags = V4L2_FMT_FLAG_DYN_RESOLUTION |738 V4L2_FMT_FLAG_CONTINUOUS_BYTESTREAM;739 }740 return 0;741}742 743static int vidioc_enum_fmt_vid_cap(struct file *file, void *priv,744 struct v4l2_fmtdesc *f)745{746 struct vicodec_ctx *ctx = file2ctx(file);747 748 return enum_fmt(f, ctx, false);749}750 751static int vidioc_enum_fmt_vid_out(struct file *file, void *priv,752 struct v4l2_fmtdesc *f)753{754 struct vicodec_ctx *ctx = file2ctx(file);755 756 return enum_fmt(f, ctx, true);757}758 759static int vidioc_g_fmt(struct vicodec_ctx *ctx, struct v4l2_format *f)760{761 struct vb2_queue *vq;762 struct vicodec_q_data *q_data;763 struct v4l2_pix_format_mplane *pix_mp;764 struct v4l2_pix_format *pix;765 const struct v4l2_fwht_pixfmt_info *info;766 767 vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, f->type);768 if (!vq)769 return -EINVAL;770 771 q_data = get_q_data(ctx, f->type);772 info = q_data->info;773 774 switch (f->type) {775 case V4L2_BUF_TYPE_VIDEO_CAPTURE:776 case V4L2_BUF_TYPE_VIDEO_OUTPUT:777 if (multiplanar)778 return -EINVAL;779 pix = &f->fmt.pix;780 pix->width = q_data->coded_width;781 pix->height = q_data->coded_height;782 pix->field = V4L2_FIELD_NONE;783 pix->pixelformat = info->id;784 pix->bytesperline = q_data->coded_width *785 info->bytesperline_mult;786 pix->sizeimage = q_data->sizeimage;787 pix->colorspace = ctx->state.colorspace;788 pix->xfer_func = ctx->state.xfer_func;789 pix->ycbcr_enc = ctx->state.ycbcr_enc;790 pix->quantization = ctx->state.quantization;791 break;792 793 case V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE:794 case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE:795 if (!multiplanar)796 return -EINVAL;797 pix_mp = &f->fmt.pix_mp;798 pix_mp->width = q_data->coded_width;799 pix_mp->height = q_data->coded_height;800 pix_mp->field = V4L2_FIELD_NONE;801 pix_mp->pixelformat = info->id;802 pix_mp->num_planes = 1;803 pix_mp->plane_fmt[0].bytesperline =804 q_data->coded_width * info->bytesperline_mult;805 pix_mp->plane_fmt[0].sizeimage = q_data->sizeimage;806 pix_mp->colorspace = ctx->state.colorspace;807 pix_mp->xfer_func = ctx->state.xfer_func;808 pix_mp->ycbcr_enc = ctx->state.ycbcr_enc;809 pix_mp->quantization = ctx->state.quantization;810 break;811 default:812 return -EINVAL;813 }814 return 0;815}816 817static int vidioc_g_fmt_vid_out(struct file *file, void *priv,818 struct v4l2_format *f)819{820 return vidioc_g_fmt(file2ctx(file), f);821}822 823static int vidioc_g_fmt_vid_cap(struct file *file, void *priv,824 struct v4l2_format *f)825{826 return vidioc_g_fmt(file2ctx(file), f);827}828 829static int vidioc_try_fmt(struct vicodec_ctx *ctx, struct v4l2_format *f)830{831 struct v4l2_pix_format_mplane *pix_mp;832 struct v4l2_pix_format *pix;833 struct v4l2_plane_pix_format *plane;834 const struct v4l2_fwht_pixfmt_info *info = ctx->is_stateless ?835 &pixfmt_stateless_fwht : &pixfmt_fwht;836 837 switch (f->type) {838 case V4L2_BUF_TYPE_VIDEO_CAPTURE:839 case V4L2_BUF_TYPE_VIDEO_OUTPUT:840 pix = &f->fmt.pix;841 if (pix->pixelformat != V4L2_PIX_FMT_FWHT &&842 pix->pixelformat != V4L2_PIX_FMT_FWHT_STATELESS)843 info = find_fmt(pix->pixelformat);844 845 pix->width = clamp(pix->width, MIN_WIDTH, MAX_WIDTH);846 pix->width = vic_round_dim(pix->width, info->width_div);847 848 pix->height = clamp(pix->height, MIN_HEIGHT, MAX_HEIGHT);849 pix->height = vic_round_dim(pix->height, info->height_div);850 851 pix->field = V4L2_FIELD_NONE;852 pix->bytesperline =853 pix->width * info->bytesperline_mult;854 pix->sizeimage = pix->width * pix->height *855 info->sizeimage_mult / info->sizeimage_div;856 if (pix->pixelformat == V4L2_PIX_FMT_FWHT)857 pix->sizeimage += sizeof(struct fwht_cframe_hdr);858 break;859 case V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE:860 case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE:861 pix_mp = &f->fmt.pix_mp;862 plane = pix_mp->plane_fmt;863 if (pix_mp->pixelformat != V4L2_PIX_FMT_FWHT &&864 pix_mp->pixelformat != V4L2_PIX_FMT_FWHT_STATELESS)865 info = find_fmt(pix_mp->pixelformat);866 pix_mp->num_planes = 1;867 868 pix_mp->width = clamp(pix_mp->width, MIN_WIDTH, MAX_WIDTH);869 pix_mp->width = vic_round_dim(pix_mp->width, info->width_div);870 871 pix_mp->height = clamp(pix_mp->height, MIN_HEIGHT, MAX_HEIGHT);872 pix_mp->height = vic_round_dim(pix_mp->height,873 info->height_div);874 875 pix_mp->field = V4L2_FIELD_NONE;876 plane->bytesperline =877 pix_mp->width * info->bytesperline_mult;878 plane->sizeimage = pix_mp->width * pix_mp->height *879 info->sizeimage_mult / info->sizeimage_div;880 if (pix_mp->pixelformat == V4L2_PIX_FMT_FWHT)881 plane->sizeimage += sizeof(struct fwht_cframe_hdr);882 break;883 default:884 return -EINVAL;885 }886 887 return 0;888}889 890static int vidioc_try_fmt_vid_cap(struct file *file, void *priv,891 struct v4l2_format *f)892{893 struct vicodec_ctx *ctx = file2ctx(file);894 struct v4l2_pix_format_mplane *pix_mp;895 struct v4l2_pix_format *pix;896 897 switch (f->type) {898 case V4L2_BUF_TYPE_VIDEO_CAPTURE:899 if (multiplanar)900 return -EINVAL;901 pix = &f->fmt.pix;902 pix->pixelformat = ctx->is_enc ? V4L2_PIX_FMT_FWHT :903 find_fmt(f->fmt.pix.pixelformat)->id;904 pix->colorspace = ctx->state.colorspace;905 pix->xfer_func = ctx->state.xfer_func;906 pix->ycbcr_enc = ctx->state.ycbcr_enc;907 pix->quantization = ctx->state.quantization;908 break;909 case V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE:910 if (!multiplanar)911 return -EINVAL;912 pix_mp = &f->fmt.pix_mp;913 pix_mp->pixelformat = ctx->is_enc ? V4L2_PIX_FMT_FWHT :914 find_fmt(pix_mp->pixelformat)->id;915 pix_mp->colorspace = ctx->state.colorspace;916 pix_mp->xfer_func = ctx->state.xfer_func;917 pix_mp->ycbcr_enc = ctx->state.ycbcr_enc;918 pix_mp->quantization = ctx->state.quantization;919 break;920 default:921 return -EINVAL;922 }923 924 return vidioc_try_fmt(ctx, f);925}926 927static int vidioc_try_fmt_vid_out(struct file *file, void *priv,928 struct v4l2_format *f)929{930 struct vicodec_ctx *ctx = file2ctx(file);931 struct v4l2_pix_format_mplane *pix_mp;932 struct v4l2_pix_format *pix;933 934 switch (f->type) {935 case V4L2_BUF_TYPE_VIDEO_OUTPUT:936 if (multiplanar)937 return -EINVAL;938 pix = &f->fmt.pix;939 if (ctx->is_enc)940 pix->pixelformat = find_fmt(pix->pixelformat)->id;941 else if (ctx->is_stateless)942 pix->pixelformat = V4L2_PIX_FMT_FWHT_STATELESS;943 else944 pix->pixelformat = V4L2_PIX_FMT_FWHT;945 if (!pix->colorspace)946 pix->colorspace = V4L2_COLORSPACE_REC709;947 break;948 case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE:949 if (!multiplanar)950 return -EINVAL;951 pix_mp = &f->fmt.pix_mp;952 if (ctx->is_enc)953 pix_mp->pixelformat = find_fmt(pix_mp->pixelformat)->id;954 else if (ctx->is_stateless)955 pix_mp->pixelformat = V4L2_PIX_FMT_FWHT_STATELESS;956 else957 pix_mp->pixelformat = V4L2_PIX_FMT_FWHT;958 if (!pix_mp->colorspace)959 pix_mp->colorspace = V4L2_COLORSPACE_REC709;960 break;961 default:962 return -EINVAL;963 }964 965 return vidioc_try_fmt(ctx, f);966}967 968static int vidioc_s_fmt(struct vicodec_ctx *ctx, struct v4l2_format *f)969{970 struct vicodec_q_data *q_data;971 struct vb2_queue *vq;972 bool fmt_changed = true;973 struct v4l2_pix_format_mplane *pix_mp;974 struct v4l2_pix_format *pix;975 976 vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, f->type);977 if (!vq)978 return -EINVAL;979 980 q_data = get_q_data(ctx, f->type);981 if (!q_data)982 return -EINVAL;983 984 switch (f->type) {985 case V4L2_BUF_TYPE_VIDEO_CAPTURE:986 case V4L2_BUF_TYPE_VIDEO_OUTPUT:987 pix = &f->fmt.pix;988 if (ctx->is_enc && V4L2_TYPE_IS_OUTPUT(f->type))989 fmt_changed =990 !q_data->info ||991 q_data->info->id != pix->pixelformat ||992 q_data->coded_width != pix->width ||993 q_data->coded_height != pix->height;994 995 if (vb2_is_busy(vq) && fmt_changed)996 return -EBUSY;997 998 if (pix->pixelformat == V4L2_PIX_FMT_FWHT)999 q_data->info = &pixfmt_fwht;1000 else if (pix->pixelformat == V4L2_PIX_FMT_FWHT_STATELESS)1001 q_data->info = &pixfmt_stateless_fwht;1002 else1003 q_data->info = find_fmt(pix->pixelformat);1004 q_data->coded_width = pix->width;1005 q_data->coded_height = pix->height;1006 q_data->sizeimage = pix->sizeimage;1007 break;1008 case V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE:1009 case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE:1010 pix_mp = &f->fmt.pix_mp;1011 if (ctx->is_enc && V4L2_TYPE_IS_OUTPUT(f->type))1012 fmt_changed =1013 !q_data->info ||1014 q_data->info->id != pix_mp->pixelformat ||1015 q_data->coded_width != pix_mp->width ||1016 q_data->coded_height != pix_mp->height;1017 1018 if (vb2_is_busy(vq) && fmt_changed)1019 return -EBUSY;1020 1021 if (pix_mp->pixelformat == V4L2_PIX_FMT_FWHT)1022 q_data->info = &pixfmt_fwht;1023 else if (pix_mp->pixelformat == V4L2_PIX_FMT_FWHT_STATELESS)1024 q_data->info = &pixfmt_stateless_fwht;1025 else1026 q_data->info = find_fmt(pix_mp->pixelformat);1027 q_data->coded_width = pix_mp->width;1028 q_data->coded_height = pix_mp->height;1029 q_data->sizeimage = pix_mp->plane_fmt[0].sizeimage;1030 break;1031 default:1032 return -EINVAL;1033 }1034 1035 dprintk(ctx->dev,1036 "Setting format for type %d, coded wxh: %dx%d, fourcc: 0x%08x\n",1037 f->type, q_data->coded_width, q_data->coded_height,1038 q_data->info->id);1039 1040 return 0;1041}1042 1043static int vidioc_s_fmt_vid_cap(struct file *file, void *priv,1044 struct v4l2_format *f)1045{1046 int ret;1047 1048 ret = vidioc_try_fmt_vid_cap(file, priv, f);1049 if (ret)1050 return ret;1051 1052 return vidioc_s_fmt(file2ctx(file), f);1053}1054 1055static int vidioc_s_fmt_vid_out(struct file *file, void *priv,1056 struct v4l2_format *f)1057{1058 struct vicodec_ctx *ctx = file2ctx(file);1059 struct vicodec_q_data *q_data;1060 struct vicodec_q_data *q_data_cap;1061 struct v4l2_pix_format *pix;1062 struct v4l2_pix_format_mplane *pix_mp;1063 u32 coded_w = 0, coded_h = 0;1064 unsigned int size = 0;1065 int ret;1066 1067 q_data = get_q_data(ctx, f->type);1068 q_data_cap = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE);1069 1070 ret = vidioc_try_fmt_vid_out(file, priv, f);1071 if (ret)1072 return ret;1073 1074 if (ctx->is_enc) {1075 struct vb2_queue *vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, f->type);1076 struct vb2_queue *vq_cap = v4l2_m2m_get_vq(ctx->fh.m2m_ctx,1077 V4L2_BUF_TYPE_VIDEO_CAPTURE);1078 const struct v4l2_fwht_pixfmt_info *info = ctx->is_stateless ?1079 &pixfmt_stateless_fwht : &pixfmt_fwht;1080 1081 if (f->type == V4L2_BUF_TYPE_VIDEO_OUTPUT) {1082 coded_w = f->fmt.pix.width;1083 coded_h = f->fmt.pix.height;1084 } else {1085 coded_w = f->fmt.pix_mp.width;1086 coded_h = f->fmt.pix_mp.height;1087 }1088 if (vb2_is_busy(vq) && (coded_w != q_data->coded_width ||1089 coded_h != q_data->coded_height))1090 return -EBUSY;1091 size = coded_w * coded_h *1092 info->sizeimage_mult / info->sizeimage_div;1093 if (!ctx->is_stateless)1094 size += sizeof(struct fwht_cframe_hdr);1095 1096 if (vb2_is_busy(vq_cap) && size > q_data_cap->sizeimage)1097 return -EBUSY;1098 }1099 1100 ret = vidioc_s_fmt(file2ctx(file), f);1101 if (!ret) {1102 if (ctx->is_enc) {1103 q_data->visible_width = coded_w;1104 q_data->visible_height = coded_h;1105 q_data_cap->coded_width = coded_w;1106 q_data_cap->coded_height = coded_h;1107 q_data_cap->sizeimage = size;1108 }1109 1110 switch (f->type) {1111 case V4L2_BUF_TYPE_VIDEO_OUTPUT:1112 pix = &f->fmt.pix;1113 ctx->state.colorspace = pix->colorspace;1114 ctx->state.xfer_func = pix->xfer_func;1115 ctx->state.ycbcr_enc = pix->ycbcr_enc;1116 ctx->state.quantization = pix->quantization;1117 break;1118 case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE:1119 pix_mp = &f->fmt.pix_mp;1120 ctx->state.colorspace = pix_mp->colorspace;1121 ctx->state.xfer_func = pix_mp->xfer_func;1122 ctx->state.ycbcr_enc = pix_mp->ycbcr_enc;1123 ctx->state.quantization = pix_mp->quantization;1124 break;1125 default:1126 break;1127 }1128 }1129 return ret;1130}1131 1132static int vidioc_g_selection(struct file *file, void *priv,1133 struct v4l2_selection *s)1134{1135 struct vicodec_ctx *ctx = file2ctx(file);1136 struct vicodec_q_data *q_data;1137 1138 q_data = get_q_data(ctx, s->type);1139 if (!q_data)1140 return -EINVAL;1141 /*1142 * encoder supports only cropping on the OUTPUT buffer1143 * decoder supports only composing on the CAPTURE buffer1144 */1145 if (ctx->is_enc && s->type == V4L2_BUF_TYPE_VIDEO_OUTPUT) {1146 switch (s->target) {1147 case V4L2_SEL_TGT_CROP:1148 s->r.left = 0;1149 s->r.top = 0;1150 s->r.width = q_data->visible_width;1151 s->r.height = q_data->visible_height;1152 return 0;1153 case V4L2_SEL_TGT_CROP_DEFAULT:1154 case V4L2_SEL_TGT_CROP_BOUNDS:1155 s->r.left = 0;1156 s->r.top = 0;1157 s->r.width = q_data->coded_width;1158 s->r.height = q_data->coded_height;1159 return 0;1160 }1161 } else if (!ctx->is_enc && s->type == V4L2_BUF_TYPE_VIDEO_CAPTURE) {1162 switch (s->target) {1163 case V4L2_SEL_TGT_COMPOSE:1164 s->r.left = 0;1165 s->r.top = 0;1166 s->r.width = q_data->visible_width;1167 s->r.height = q_data->visible_height;1168 return 0;1169 case V4L2_SEL_TGT_COMPOSE_DEFAULT:1170 case V4L2_SEL_TGT_COMPOSE_BOUNDS:1171 s->r.left = 0;1172 s->r.top = 0;1173 s->r.width = q_data->coded_width;1174 s->r.height = q_data->coded_height;1175 return 0;1176 }1177 }1178 return -EINVAL;1179}1180 1181static int vidioc_s_selection(struct file *file, void *priv,1182 struct v4l2_selection *s)1183{1184 struct vicodec_ctx *ctx = file2ctx(file);1185 struct vicodec_q_data *q_data;1186 1187 if (s->type != V4L2_BUF_TYPE_VIDEO_OUTPUT)1188 return -EINVAL;1189 1190 q_data = get_q_data(ctx, s->type);1191 if (!q_data)1192 return -EINVAL;1193 1194 if (!ctx->is_enc || s->target != V4L2_SEL_TGT_CROP)1195 return -EINVAL;1196 1197 s->r.left = 0;1198 s->r.top = 0;1199 q_data->visible_width = clamp(s->r.width, MIN_WIDTH,1200 q_data->coded_width);1201 s->r.width = q_data->visible_width;1202 q_data->visible_height = clamp(s->r.height, MIN_HEIGHT,1203 q_data->coded_height);1204 s->r.height = q_data->visible_height;1205 return 0;1206}1207 1208static int vicodec_encoder_cmd(struct file *file, void *fh,1209 struct v4l2_encoder_cmd *ec)1210{1211 struct vicodec_ctx *ctx = file2ctx(file);1212 int ret;1213 1214 ret = v4l2_m2m_ioctl_try_encoder_cmd(file, fh, ec);1215 if (ret < 0)1216 return ret;1217 1218 if (!vb2_is_streaming(&ctx->fh.m2m_ctx->out_q_ctx.q))1219 return 0;1220 1221 ret = v4l2_m2m_ioctl_encoder_cmd(file, fh, ec);1222 if (ret < 0)1223 return ret;1224 1225 if (ec->cmd == V4L2_ENC_CMD_STOP &&1226 v4l2_m2m_has_stopped(ctx->fh.m2m_ctx))1227 v4l2_event_queue_fh(&ctx->fh, &vicodec_eos_event);1228 1229 if (ec->cmd == V4L2_ENC_CMD_START &&1230 v4l2_m2m_has_stopped(ctx->fh.m2m_ctx))1231 vb2_clear_last_buffer_dequeued(&ctx->fh.m2m_ctx->cap_q_ctx.q);1232 1233 return 0;1234}1235 1236static int vicodec_decoder_cmd(struct file *file, void *fh,1237 struct v4l2_decoder_cmd *dc)1238{1239 struct vicodec_ctx *ctx = file2ctx(file);1240 int ret;1241 1242 /*1243 * This ioctl should not be used with a stateless codec that doesn't1244 * support holding buffers and the associated flush command.1245 */1246 WARN_ON(ctx->is_stateless);1247 1248 ret = v4l2_m2m_ioctl_try_decoder_cmd(file, fh, dc);1249 if (ret < 0)1250 return ret;1251 1252 if (!vb2_is_streaming(&ctx->fh.m2m_ctx->out_q_ctx.q))1253 return 0;1254 1255 ret = v4l2_m2m_ioctl_decoder_cmd(file, fh, dc);1256 if (ret < 0)1257 return ret;1258 1259 if (dc->cmd == V4L2_DEC_CMD_STOP &&1260 v4l2_m2m_has_stopped(ctx->fh.m2m_ctx))1261 v4l2_event_queue_fh(&ctx->fh, &vicodec_eos_event);1262 1263 if (dc->cmd == V4L2_DEC_CMD_START &&1264 v4l2_m2m_has_stopped(ctx->fh.m2m_ctx))1265 vb2_clear_last_buffer_dequeued(&ctx->fh.m2m_ctx->cap_q_ctx.q);1266 1267 return 0;1268}1269 1270static int vicodec_enum_framesizes(struct file *file, void *fh,1271 struct v4l2_frmsizeenum *fsize)1272{1273 switch (fsize->pixel_format) {1274 case V4L2_PIX_FMT_FWHT_STATELESS:1275 break;1276 case V4L2_PIX_FMT_FWHT:1277 break;1278 default:1279 if (find_fmt(fsize->pixel_format)->id == fsize->pixel_format)1280 break;1281 return -EINVAL;1282 }1283 1284 if (fsize->index)1285 return -EINVAL;1286 1287 fsize->type = V4L2_FRMSIZE_TYPE_STEPWISE;1288 1289 fsize->stepwise.min_width = MIN_WIDTH;1290 fsize->stepwise.max_width = MAX_WIDTH;1291 fsize->stepwise.step_width = 8;1292 fsize->stepwise.min_height = MIN_HEIGHT;1293 fsize->stepwise.max_height = MAX_HEIGHT;1294 fsize->stepwise.step_height = 8;1295 1296 return 0;1297}1298 1299static int vicodec_subscribe_event(struct v4l2_fh *fh,1300 const struct v4l2_event_subscription *sub)1301{1302 struct vicodec_ctx *ctx = container_of(fh, struct vicodec_ctx, fh);1303 1304 switch (sub->type) {1305 case V4L2_EVENT_SOURCE_CHANGE:1306 if (ctx->is_enc)1307 return -EINVAL;1308 fallthrough;1309 case V4L2_EVENT_EOS:1310 if (ctx->is_stateless)1311 return -EINVAL;1312 return v4l2_event_subscribe(fh, sub, 0, NULL);1313 default:1314 return v4l2_ctrl_subscribe_event(fh, sub);1315 }1316}1317 1318static const struct v4l2_ioctl_ops vicodec_ioctl_ops = {1319 .vidioc_querycap = vidioc_querycap,1320 1321 .vidioc_enum_fmt_vid_cap = vidioc_enum_fmt_vid_cap,1322 .vidioc_g_fmt_vid_cap = vidioc_g_fmt_vid_cap,1323 .vidioc_try_fmt_vid_cap = vidioc_try_fmt_vid_cap,1324 .vidioc_s_fmt_vid_cap = vidioc_s_fmt_vid_cap,1325 1326 .vidioc_g_fmt_vid_cap_mplane = vidioc_g_fmt_vid_cap,1327 .vidioc_try_fmt_vid_cap_mplane = vidioc_try_fmt_vid_cap,1328 .vidioc_s_fmt_vid_cap_mplane = vidioc_s_fmt_vid_cap,1329 1330 .vidioc_enum_fmt_vid_out = vidioc_enum_fmt_vid_out,1331 .vidioc_g_fmt_vid_out = vidioc_g_fmt_vid_out,1332 .vidioc_try_fmt_vid_out = vidioc_try_fmt_vid_out,1333 .vidioc_s_fmt_vid_out = vidioc_s_fmt_vid_out,1334 1335 .vidioc_g_fmt_vid_out_mplane = vidioc_g_fmt_vid_out,1336 .vidioc_try_fmt_vid_out_mplane = vidioc_try_fmt_vid_out,1337 .vidioc_s_fmt_vid_out_mplane = vidioc_s_fmt_vid_out,1338 1339 .vidioc_reqbufs = v4l2_m2m_ioctl_reqbufs,1340 .vidioc_querybuf = v4l2_m2m_ioctl_querybuf,1341 .vidioc_qbuf = v4l2_m2m_ioctl_qbuf,1342 .vidioc_dqbuf = v4l2_m2m_ioctl_dqbuf,1343 .vidioc_prepare_buf = v4l2_m2m_ioctl_prepare_buf,1344 .vidioc_create_bufs = v4l2_m2m_ioctl_create_bufs,1345 .vidioc_expbuf = v4l2_m2m_ioctl_expbuf,1346 .vidioc_remove_bufs = v4l2_m2m_ioctl_remove_bufs,1347 1348 .vidioc_streamon = v4l2_m2m_ioctl_streamon,1349 .vidioc_streamoff = v4l2_m2m_ioctl_streamoff,1350 1351 .vidioc_g_selection = vidioc_g_selection,1352 .vidioc_s_selection = vidioc_s_selection,1353 1354 .vidioc_try_encoder_cmd = v4l2_m2m_ioctl_try_encoder_cmd,1355 .vidioc_encoder_cmd = vicodec_encoder_cmd,1356 .vidioc_try_decoder_cmd = v4l2_m2m_ioctl_try_decoder_cmd,1357 .vidioc_decoder_cmd = vicodec_decoder_cmd,1358 .vidioc_enum_framesizes = vicodec_enum_framesizes,1359 1360 .vidioc_subscribe_event = vicodec_subscribe_event,1361 .vidioc_unsubscribe_event = v4l2_event_unsubscribe,1362};1363 1364 1365/*1366 * Queue operations1367 */1368 1369static int vicodec_queue_setup(struct vb2_queue *vq, unsigned int *nbuffers,1370 unsigned int *nplanes, unsigned int sizes[],1371 struct device *alloc_devs[])1372{1373 struct vicodec_ctx *ctx = vb2_get_drv_priv(vq);1374 struct vicodec_q_data *q_data = get_q_data(ctx, vq->type);1375 unsigned int size = q_data->sizeimage;1376 1377 if (*nplanes)1378 return sizes[0] < size ? -EINVAL : 0;1379 1380 *nplanes = 1;1381 sizes[0] = size;1382 q_data->vb2_sizeimage = size;1383 return 0;1384}1385 1386static int vicodec_buf_out_validate(struct vb2_buffer *vb)1387{1388 struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);1389 1390 vbuf->field = V4L2_FIELD_NONE;1391 return 0;1392}1393 1394static int vicodec_buf_prepare(struct vb2_buffer *vb)1395{1396 struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);1397 struct vicodec_ctx *ctx = vb2_get_drv_priv(vb->vb2_queue);1398 struct vicodec_q_data *q_data;1399 1400 dprintk(ctx->dev, "type: %d\n", vb->vb2_queue->type);1401 1402 q_data = get_q_data(ctx, vb->vb2_queue->type);1403 if (V4L2_TYPE_IS_OUTPUT(vb->vb2_queue->type)) {1404 if (vbuf->field == V4L2_FIELD_ANY)1405 vbuf->field = V4L2_FIELD_NONE;1406 if (vbuf->field != V4L2_FIELD_NONE) {1407 dprintk(ctx->dev, "%s field isn't supported\n",1408 __func__);1409 return -EINVAL;1410 }1411 }1412 1413 if (vb2_plane_size(vb, 0) < q_data->vb2_sizeimage) {1414 dprintk(ctx->dev,1415 "%s data will not fit into plane (%lu < %lu)\n",1416 __func__, vb2_plane_size(vb, 0),1417 (long)q_data->vb2_sizeimage);1418 return -EINVAL;1419 }1420 1421 return 0;1422}1423 1424static void vicodec_buf_queue(struct vb2_buffer *vb)1425{1426 struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);1427 struct vicodec_ctx *ctx = vb2_get_drv_priv(vb->vb2_queue);1428 unsigned int sz = vb2_get_plane_payload(&vbuf->vb2_buf, 0);1429 u8 *p_src = vb2_plane_vaddr(&vbuf->vb2_buf, 0);1430 u8 *p = p_src;1431 struct vb2_queue *vq_out = v4l2_m2m_get_vq(ctx->fh.m2m_ctx,1432 V4L2_BUF_TYPE_VIDEO_OUTPUT);1433 struct vb2_queue *vq_cap = v4l2_m2m_get_vq(ctx->fh.m2m_ctx,1434 V4L2_BUF_TYPE_VIDEO_CAPTURE);1435 bool header_valid = false;1436 static const struct v4l2_event rs_event = {1437 .type = V4L2_EVENT_SOURCE_CHANGE,1438 .u.src_change.changes = V4L2_EVENT_SRC_CH_RESOLUTION,1439 };1440 1441 if (V4L2_TYPE_IS_CAPTURE(vb->vb2_queue->type) &&1442 vb2_is_streaming(vb->vb2_queue) &&1443 v4l2_m2m_dst_buf_is_last(ctx->fh.m2m_ctx)) {1444 unsigned int i;1445 1446 for (i = 0; i < vb->num_planes; i++)1447 vb2_set_plane_payload(vb, i, 0);1448 1449 vbuf->field = V4L2_FIELD_NONE;1450 vbuf->sequence =1451 get_q_data(ctx, vb->vb2_queue->type)->sequence++;1452 1453 v4l2_m2m_last_buffer_done(ctx->fh.m2m_ctx, vbuf);1454 v4l2_event_queue_fh(&ctx->fh, &vicodec_eos_event);1455 return;1456 }1457 1458 /* buf_queue handles only the first source change event */1459 if (ctx->first_source_change_sent) {1460 v4l2_m2m_buf_queue(ctx->fh.m2m_ctx, vbuf);1461 return;1462 }1463 1464 /*1465 * if both queues are streaming, the source change event is1466 * handled in job_ready1467 */1468 if (vb2_is_streaming(vq_cap) && vb2_is_streaming(vq_out)) {1469 v4l2_m2m_buf_queue(ctx->fh.m2m_ctx, vbuf);1470 return;1471 }1472 1473 /*1474 * source change event is relevant only for the stateful decoder1475 * in the compressed stream1476 */1477 if (ctx->is_stateless || ctx->is_enc ||1478 V4L2_TYPE_IS_CAPTURE(vb->vb2_queue->type)) {1479 v4l2_m2m_buf_queue(ctx->fh.m2m_ctx, vbuf);1480 return;1481 }1482 1483 do {1484 enum vb2_buffer_state state =1485 get_next_header(ctx, &p, p_src + sz - p);1486 1487 if (ctx->header_size < sizeof(struct fwht_cframe_hdr)) {1488 v4l2_m2m_buf_done(vbuf, state);1489 return;1490 }1491 header_valid = is_header_valid(&ctx->state.header);1492 /*1493 * p points right after the end of the header in the1494 * buffer. If the header is invalid we set p to point1495 * to the next byte after the start of the header1496 */1497 if (!header_valid) {1498 p = p - sizeof(struct fwht_cframe_hdr) + 1;1499 if (p < p_src)1500 p = p_src;1501 ctx->header_size = 0;1502 ctx->comp_magic_cnt = 0;1503 }1504 1505 } while (!header_valid);1506 1507 ctx->cur_buf_offset = p - p_src;1508 update_capture_data_from_header(ctx);1509 ctx->first_source_change_sent = true;1510 v4l2_event_queue_fh(&ctx->fh, &rs_event);1511 v4l2_m2m_buf_queue(ctx->fh.m2m_ctx, vbuf);1512}1513 1514static void vicodec_return_bufs(struct vb2_queue *q, u32 state)1515{1516 struct vicodec_ctx *ctx = vb2_get_drv_priv(q);1517 struct vb2_v4l2_buffer *vbuf;1518 1519 for (;;) {1520 if (V4L2_TYPE_IS_OUTPUT(q->type))1521 vbuf = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx);1522 else1523 vbuf = v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx);1524 if (vbuf == NULL)1525 return;1526 v4l2_ctrl_request_complete(vbuf->vb2_buf.req_obj.req,1527 &ctx->hdl);1528 spin_lock(ctx->lock);1529 v4l2_m2m_buf_done(vbuf, state);1530 spin_unlock(ctx->lock);1531 }1532}1533 1534static unsigned int total_frame_size(struct vicodec_q_data *q_data)1535{1536 unsigned int size;1537 unsigned int chroma_div;1538 1539 if (!q_data->info) {1540 WARN_ON(1);1541 return 0;1542 }1543 size = q_data->coded_width * q_data->coded_height;1544 chroma_div = q_data->info->width_div * q_data->info->height_div;1545 1546 if (q_data->info->components_num == 4)1547 return 2 * size + 2 * (size / chroma_div);1548 else if (q_data->info->components_num == 3)1549 return size + 2 * (size / chroma_div);1550 return size;1551}1552 1553static int vicodec_start_streaming(struct vb2_queue *q,1554 unsigned int count)1555{1556 struct vicodec_ctx *ctx = vb2_get_drv_priv(q);1557 struct vicodec_q_data *q_data = get_q_data(ctx, q->type);1558 struct v4l2_fwht_state *state = &ctx->state;1559 const struct v4l2_fwht_pixfmt_info *info = q_data->info;1560 unsigned int size = q_data->coded_width * q_data->coded_height;1561 unsigned int chroma_div;1562 unsigned int total_planes_size;1563 u8 *new_comp_frame = NULL;1564 1565 chroma_div = info->width_div * info->height_div;1566 q_data->sequence = 0;1567 1568 v4l2_m2m_update_start_streaming_state(ctx->fh.m2m_ctx, q);1569 1570 state->gop_cnt = 0;1571 1572 if ((V4L2_TYPE_IS_OUTPUT(q->type) && !ctx->is_enc) ||1573 (V4L2_TYPE_IS_CAPTURE(q->type) && ctx->is_enc))1574 return 0;1575 1576 if (info->id == V4L2_PIX_FMT_FWHT ||1577 info->id == V4L2_PIX_FMT_FWHT_STATELESS) {1578 vicodec_return_bufs(q, VB2_BUF_STATE_QUEUED);1579 return -EINVAL;1580 }1581 total_planes_size = total_frame_size(q_data);1582 ctx->comp_max_size = total_planes_size;1583 1584 state->visible_width = q_data->visible_width;1585 state->visible_height = q_data->visible_height;1586 state->coded_width = q_data->coded_width;1587 state->coded_height = q_data->coded_height;1588 state->stride = q_data->coded_width *1589 info->bytesperline_mult;1590 1591 if (ctx->is_stateless) {1592 state->ref_stride = state->stride;1593 return 0;1594 }1595 state->ref_stride = q_data->coded_width * info->luma_alpha_step;1596 1597 state->ref_frame.buf = kvmalloc(total_planes_size, GFP_KERNEL);1598 state->ref_frame.luma = state->ref_frame.buf;1599 new_comp_frame = kvmalloc(ctx->comp_max_size, GFP_KERNEL);1600 1601 if (!state->ref_frame.luma || !new_comp_frame) {1602 kvfree(state->ref_frame.luma);1603 kvfree(new_comp_frame);1604 vicodec_return_bufs(q, VB2_BUF_STATE_QUEUED);1605 return -ENOMEM;1606 }1607 /*1608 * if state->compressed_frame was already allocated then1609 * it contain data of the first frame of the new resolution1610 */1611 if (state->compressed_frame) {1612 if (ctx->comp_size > ctx->comp_max_size)1613 ctx->comp_size = ctx->comp_max_size;1614 1615 memcpy(new_comp_frame,1616 state->compressed_frame, ctx->comp_size);1617 }1618 1619 kvfree(state->compressed_frame);1620 state->compressed_frame = new_comp_frame;1621 1622 if (info->components_num < 3) {1623 state->ref_frame.cb = NULL;1624 state->ref_frame.cr = NULL;1625 state->ref_frame.alpha = NULL;1626 return 0;1627 }1628 1629 state->ref_frame.cb = state->ref_frame.luma + size;1630 state->ref_frame.cr = state->ref_frame.cb + size / chroma_div;1631 1632 if (info->components_num == 4)1633 state->ref_frame.alpha =1634 state->ref_frame.cr + size / chroma_div;1635 else1636 state->ref_frame.alpha = NULL;1637 1638 return 0;1639}1640 1641static void vicodec_stop_streaming(struct vb2_queue *q)1642{1643 struct vicodec_ctx *ctx = vb2_get_drv_priv(q);1644 1645 vicodec_return_bufs(q, VB2_BUF_STATE_ERROR);1646 1647 v4l2_m2m_update_stop_streaming_state(ctx->fh.m2m_ctx, q);1648 1649 if (V4L2_TYPE_IS_OUTPUT(q->type) &&1650 v4l2_m2m_has_stopped(ctx->fh.m2m_ctx))1651 v4l2_event_queue_fh(&ctx->fh, &vicodec_eos_event);1652 1653 if (!ctx->is_enc && V4L2_TYPE_IS_OUTPUT(q->type))1654 ctx->first_source_change_sent = false;1655 1656 if ((!V4L2_TYPE_IS_OUTPUT(q->type) && !ctx->is_enc) ||1657 (V4L2_TYPE_IS_OUTPUT(q->type) && ctx->is_enc)) {1658 if (!ctx->is_stateless)1659 kvfree(ctx->state.ref_frame.buf);1660 ctx->state.ref_frame.buf = NULL;1661 ctx->state.ref_frame.luma = NULL;1662 ctx->comp_max_size = 0;1663 ctx->source_changed = false;1664 }1665 if (V4L2_TYPE_IS_OUTPUT(q->type) && !ctx->is_enc) {1666 ctx->cur_buf_offset = 0;1667 ctx->comp_size = 0;1668 ctx->header_size = 0;1669 ctx->comp_magic_cnt = 0;1670 ctx->comp_has_frame = false;1671 ctx->comp_has_next_frame = false;1672 }1673}1674 1675static void vicodec_buf_request_complete(struct vb2_buffer *vb)1676{1677 struct vicodec_ctx *ctx = vb2_get_drv_priv(vb->vb2_queue);1678 1679 v4l2_ctrl_request_complete(vb->req_obj.req, &ctx->hdl);1680}1681 1682 1683static const struct vb2_ops vicodec_qops = {1684 .queue_setup = vicodec_queue_setup,1685 .buf_out_validate = vicodec_buf_out_validate,1686 .buf_prepare = vicodec_buf_prepare,1687 .buf_queue = vicodec_buf_queue,1688 .buf_request_complete = vicodec_buf_request_complete,1689 .start_streaming = vicodec_start_streaming,1690 .stop_streaming = vicodec_stop_streaming,1691 .wait_prepare = vb2_ops_wait_prepare,1692 .wait_finish = vb2_ops_wait_finish,1693};1694 1695static int queue_init(void *priv, struct vb2_queue *src_vq,1696 struct vb2_queue *dst_vq)1697{1698 struct vicodec_ctx *ctx = priv;1699 int ret;1700 1701 src_vq->type = (multiplanar ?1702 V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE :1703 V4L2_BUF_TYPE_VIDEO_OUTPUT);1704 src_vq->io_modes = VB2_MMAP | VB2_USERPTR | VB2_DMABUF;1705 src_vq->drv_priv = ctx;1706 src_vq->buf_struct_size = sizeof(struct v4l2_m2m_buffer);1707 src_vq->ops = &vicodec_qops;1708 src_vq->mem_ops = &vb2_vmalloc_memops;1709 src_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY;1710 if (ctx->is_enc)1711 src_vq->lock = &ctx->dev->stateful_enc.mutex;1712 else if (ctx->is_stateless)1713 src_vq->lock = &ctx->dev->stateless_dec.mutex;1714 else1715 src_vq->lock = &ctx->dev->stateful_dec.mutex;1716 src_vq->supports_requests = ctx->is_stateless;1717 src_vq->requires_requests = ctx->is_stateless;1718 ret = vb2_queue_init(src_vq);1719 if (ret)1720 return ret;1721 1722 dst_vq->type = (multiplanar ?1723 V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE :1724 V4L2_BUF_TYPE_VIDEO_CAPTURE);1725 dst_vq->io_modes = VB2_MMAP | VB2_USERPTR | VB2_DMABUF;1726 dst_vq->max_num_buffers = 64;1727 dst_vq->drv_priv = ctx;1728 dst_vq->buf_struct_size = sizeof(struct v4l2_m2m_buffer);1729 dst_vq->ops = &vicodec_qops;1730 dst_vq->mem_ops = &vb2_vmalloc_memops;1731 dst_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY;1732 dst_vq->lock = src_vq->lock;1733 1734 return vb2_queue_init(dst_vq);1735}1736 1737static int vicodec_try_ctrl(struct v4l2_ctrl *ctrl)1738{1739 struct vicodec_ctx *ctx = container_of(ctrl->handler,1740 struct vicodec_ctx, hdl);1741 const struct v4l2_ctrl_fwht_params *params;1742 struct vicodec_q_data *q_dst = get_q_data(ctx,1743 V4L2_BUF_TYPE_VIDEO_CAPTURE);1744 1745 switch (ctrl->id) {1746 case V4L2_CID_STATELESS_FWHT_PARAMS:1747 if (!q_dst->info)1748 return -EINVAL;1749 params = ctrl->p_new.p_fwht_params;1750 if (params->width > q_dst->coded_width ||1751 params->width < MIN_WIDTH ||1752 params->height > q_dst->coded_height ||1753 params->height < MIN_HEIGHT)1754 return -EINVAL;1755 if (!validate_by_version(params->flags, params->version))1756 return -EINVAL;1757 if (!validate_stateless_params_flags(params, q_dst->info))1758 return -EINVAL;1759 return 0;1760 default:1761 return 0;1762 }1763 return 0;1764}1765 1766static void update_header_from_stateless_params(struct vicodec_ctx *ctx,1767 const struct v4l2_ctrl_fwht_params *params)1768{1769 struct fwht_cframe_hdr *p_hdr = &ctx->state.header;1770 1771 p_hdr->magic1 = FWHT_MAGIC1;1772 p_hdr->magic2 = FWHT_MAGIC2;1773 p_hdr->version = htonl(params->version);1774 p_hdr->width = htonl(params->width);1775 p_hdr->height = htonl(params->height);1776 p_hdr->flags = htonl(params->flags);1777 p_hdr->colorspace = htonl(params->colorspace);1778 p_hdr->xfer_func = htonl(params->xfer_func);1779 p_hdr->ycbcr_enc = htonl(params->ycbcr_enc);1780 p_hdr->quantization = htonl(params->quantization);1781}1782 1783static int vicodec_s_ctrl(struct v4l2_ctrl *ctrl)1784{1785 struct vicodec_ctx *ctx = container_of(ctrl->handler,1786 struct vicodec_ctx, hdl);1787 const struct v4l2_ctrl_fwht_params *params;1788 1789 switch (ctrl->id) {1790 case V4L2_CID_MPEG_VIDEO_GOP_SIZE:1791 ctx->state.gop_size = ctrl->val;1792 return 0;1793 case V4L2_CID_FWHT_I_FRAME_QP:1794 ctx->state.i_frame_qp = ctrl->val;1795 return 0;1796 case V4L2_CID_FWHT_P_FRAME_QP:1797 ctx->state.p_frame_qp = ctrl->val;1798 return 0;1799 case V4L2_CID_STATELESS_FWHT_PARAMS:1800 params = ctrl->p_new.p_fwht_params;1801 update_header_from_stateless_params(ctx, params);1802 ctx->state.ref_frame_ts = params->backward_ref_ts;1803 return 0;1804 }1805 return -EINVAL;1806}1807 1808static const struct v4l2_ctrl_ops vicodec_ctrl_ops = {1809 .s_ctrl = vicodec_s_ctrl,1810 .try_ctrl = vicodec_try_ctrl,1811};1812 1813static const struct v4l2_ctrl_config vicodec_ctrl_stateless_state = {1814 .ops = &vicodec_ctrl_ops,1815 .id = V4L2_CID_STATELESS_FWHT_PARAMS,1816 .elem_size = sizeof(struct v4l2_ctrl_fwht_params),1817};1818 1819/*1820 * File operations1821 */1822static int vicodec_open(struct file *file)1823{1824 const struct v4l2_fwht_pixfmt_info *info = v4l2_fwht_get_pixfmt(0);1825 struct video_device *vfd = video_devdata(file);1826 struct vicodec_dev *dev = video_drvdata(file);1827 struct vicodec_ctx *ctx = NULL;1828 struct v4l2_ctrl_handler *hdl;1829 unsigned int raw_size;1830 unsigned int comp_size;1831 int rc = 0;1832 1833 if (mutex_lock_interruptible(vfd->lock))1834 return -ERESTARTSYS;1835 ctx = kzalloc(sizeof(*ctx), GFP_KERNEL);1836 if (!ctx) {1837 rc = -ENOMEM;1838 goto open_unlock;1839 }1840 1841 if (vfd == &dev->stateful_enc.vfd)1842 ctx->is_enc = true;1843 else if (vfd == &dev->stateless_dec.vfd)1844 ctx->is_stateless = true;1845 1846 v4l2_fh_init(&ctx->fh, video_devdata(file));1847 file->private_data = &ctx->fh;1848 ctx->dev = dev;1849 hdl = &ctx->hdl;1850 v4l2_ctrl_handler_init(hdl, 5);1851 v4l2_ctrl_new_std(hdl, &vicodec_ctrl_ops, V4L2_CID_MPEG_VIDEO_GOP_SIZE,1852 1, 16, 1, 10);1853 v4l2_ctrl_new_std(hdl, &vicodec_ctrl_ops, V4L2_CID_FWHT_I_FRAME_QP,1854 1, 31, 1, 20);1855 v4l2_ctrl_new_std(hdl, &vicodec_ctrl_ops, V4L2_CID_FWHT_P_FRAME_QP,1856 1, 31, 1, 20);1857 if (ctx->is_enc)1858 v4l2_ctrl_new_std(hdl, &vicodec_ctrl_ops,1859 V4L2_CID_MIN_BUFFERS_FOR_OUTPUT, 1, 1, 1, 1);1860 if (ctx->is_stateless)1861 v4l2_ctrl_new_custom(hdl, &vicodec_ctrl_stateless_state, NULL);1862 if (hdl->error) {1863 rc = hdl->error;1864 v4l2_ctrl_handler_free(hdl);1865 kfree(ctx);1866 goto open_unlock;1867 }1868 ctx->fh.ctrl_handler = hdl;1869 v4l2_ctrl_handler_setup(hdl);1870 1871 if (ctx->is_enc)1872 ctx->q_data[V4L2_M2M_SRC].info = info;1873 else if (ctx->is_stateless)1874 ctx->q_data[V4L2_M2M_SRC].info = &pixfmt_stateless_fwht;1875 else1876 ctx->q_data[V4L2_M2M_SRC].info = &pixfmt_fwht;1877 ctx->q_data[V4L2_M2M_SRC].coded_width = 1280;1878 ctx->q_data[V4L2_M2M_SRC].coded_height = 720;1879 ctx->q_data[V4L2_M2M_SRC].visible_width = 1280;1880 ctx->q_data[V4L2_M2M_SRC].visible_height = 720;1881 raw_size = 1280 * 720 * info->sizeimage_mult / info->sizeimage_div;1882 comp_size = 1280 * 720 * pixfmt_fwht.sizeimage_mult /1883 pixfmt_fwht.sizeimage_div;1884 if (ctx->is_enc)1885 ctx->q_data[V4L2_M2M_SRC].sizeimage = raw_size;1886 else if (ctx->is_stateless)1887 ctx->q_data[V4L2_M2M_SRC].sizeimage = comp_size;1888 else1889 ctx->q_data[V4L2_M2M_SRC].sizeimage =1890 comp_size + sizeof(struct fwht_cframe_hdr);1891 ctx->q_data[V4L2_M2M_DST] = ctx->q_data[V4L2_M2M_SRC];1892 if (ctx->is_enc) {1893 ctx->q_data[V4L2_M2M_DST].info = &pixfmt_fwht;1894 ctx->q_data[V4L2_M2M_DST].sizeimage =1895 comp_size + sizeof(struct fwht_cframe_hdr);1896 } else {1897 ctx->q_data[V4L2_M2M_DST].info = info;1898 ctx->q_data[V4L2_M2M_DST].sizeimage = raw_size;1899 }1900 1901 ctx->state.colorspace = V4L2_COLORSPACE_REC709;1902 1903 if (ctx->is_enc) {1904 ctx->fh.m2m_ctx = v4l2_m2m_ctx_init(dev->stateful_enc.m2m_dev,1905 ctx, &queue_init);1906 ctx->lock = &dev->stateful_enc.lock;1907 } else if (ctx->is_stateless) {1908 ctx->fh.m2m_ctx = v4l2_m2m_ctx_init(dev->stateless_dec.m2m_dev,1909 ctx, &queue_init);1910 ctx->lock = &dev->stateless_dec.lock;1911 } else {1912 ctx->fh.m2m_ctx = v4l2_m2m_ctx_init(dev->stateful_dec.m2m_dev,1913 ctx, &queue_init);1914 ctx->lock = &dev->stateful_dec.lock;1915 }1916 1917 if (IS_ERR(ctx->fh.m2m_ctx)) {1918 rc = PTR_ERR(ctx->fh.m2m_ctx);1919 1920 v4l2_ctrl_handler_free(hdl);1921 v4l2_fh_exit(&ctx->fh);1922 kfree(ctx);1923 goto open_unlock;1924 }1925 1926 v4l2_fh_add(&ctx->fh);1927 1928open_unlock:1929 mutex_unlock(vfd->lock);1930 return rc;1931}1932 1933static int vicodec_release(struct file *file)1934{1935 struct video_device *vfd = video_devdata(file);1936 struct vicodec_ctx *ctx = file2ctx(file);1937 1938 mutex_lock(vfd->lock);1939 v4l2_m2m_ctx_release(ctx->fh.m2m_ctx);1940 mutex_unlock(vfd->lock);1941 v4l2_fh_del(&ctx->fh);1942 v4l2_fh_exit(&ctx->fh);1943 v4l2_ctrl_handler_free(&ctx->hdl);1944 kvfree(ctx->state.compressed_frame);1945 kfree(ctx);1946 1947 return 0;1948}1949 1950static int vicodec_request_validate(struct media_request *req)1951{1952 struct media_request_object *obj;1953 struct v4l2_ctrl_handler *parent_hdl, *hdl;1954 struct vicodec_ctx *ctx = NULL;1955 struct v4l2_ctrl *ctrl;1956 unsigned int count;1957 1958 list_for_each_entry(obj, &req->objects, list) {1959 struct vb2_buffer *vb;1960 1961 if (vb2_request_object_is_buffer(obj)) {1962 vb = container_of(obj, struct vb2_buffer, req_obj);1963 ctx = vb2_get_drv_priv(vb->vb2_queue);1964 1965 break;1966 }1967 }1968 1969 if (!ctx) {1970 pr_err("No buffer was provided with the request\n");1971 return -ENOENT;1972 }1973 1974 count = vb2_request_buffer_cnt(req);1975 if (!count) {1976 v4l2_info(&ctx->dev->v4l2_dev,1977 "No buffer was provided with the request\n");1978 return -ENOENT;1979 } else if (count > 1) {1980 v4l2_info(&ctx->dev->v4l2_dev,1981 "More than one buffer was provided with the request\n");1982 return -EINVAL;1983 }1984 1985 parent_hdl = &ctx->hdl;1986 1987 hdl = v4l2_ctrl_request_hdl_find(req, parent_hdl);1988 if (!hdl) {1989 v4l2_info(&ctx->dev->v4l2_dev, "Missing codec control\n");1990 return -ENOENT;1991 }1992 ctrl = v4l2_ctrl_request_hdl_ctrl_find(hdl,1993 vicodec_ctrl_stateless_state.id);1994 v4l2_ctrl_request_hdl_put(hdl);1995 if (!ctrl) {1996 v4l2_info(&ctx->dev->v4l2_dev,1997 "Missing required codec control\n");1998 return -ENOENT;1999 }2000 2001 return vb2_request_validate(req);2002}2003 2004static const struct v4l2_file_operations vicodec_fops = {2005 .owner = THIS_MODULE,2006 .open = vicodec_open,2007 .release = vicodec_release,2008 .poll = v4l2_m2m_fop_poll,2009 .unlocked_ioctl = video_ioctl2,2010 .mmap = v4l2_m2m_fop_mmap,2011};2012 2013static const struct video_device vicodec_videodev = {2014 .name = VICODEC_NAME,2015 .vfl_dir = VFL_DIR_M2M,2016 .fops = &vicodec_fops,2017 .ioctl_ops = &vicodec_ioctl_ops,2018 .minor = -1,2019 .release = video_device_release_empty,2020};2021 2022static const struct media_device_ops vicodec_m2m_media_ops = {2023 .req_validate = vicodec_request_validate,2024 .req_queue = v4l2_m2m_request_queue,2025};2026 2027static const struct v4l2_m2m_ops m2m_ops = {2028 .device_run = device_run,2029 .job_ready = job_ready,2030};2031 2032static int register_instance(struct vicodec_dev *dev,2033 struct vicodec_dev_instance *dev_instance,2034 const char *name, bool is_enc, bool is_stateless)2035{2036 struct video_device *vfd;2037 int ret;2038 2039 spin_lock_init(&dev_instance->lock);2040 mutex_init(&dev_instance->mutex);2041 dev_instance->m2m_dev = v4l2_m2m_init(&m2m_ops);2042 if (IS_ERR(dev_instance->m2m_dev)) {2043 v4l2_err(&dev->v4l2_dev, "Failed to init vicodec enc device\n");2044 return PTR_ERR(dev_instance->m2m_dev);2045 }2046 2047 dev_instance->vfd = vicodec_videodev;2048 vfd = &dev_instance->vfd;2049 vfd->lock = &dev_instance->mutex;2050 vfd->v4l2_dev = &dev->v4l2_dev;2051 strscpy(vfd->name, name, sizeof(vfd->name));2052 vfd->device_caps = V4L2_CAP_STREAMING |2053 (multiplanar ? V4L2_CAP_VIDEO_M2M_MPLANE : V4L2_CAP_VIDEO_M2M);2054 if (is_enc || is_stateless) {2055 v4l2_disable_ioctl(vfd, VIDIOC_DECODER_CMD);2056 v4l2_disable_ioctl(vfd, VIDIOC_TRY_DECODER_CMD);2057 }2058 if (!is_enc) {2059 v4l2_disable_ioctl(vfd, VIDIOC_ENCODER_CMD);2060 v4l2_disable_ioctl(vfd, VIDIOC_TRY_ENCODER_CMD);2061 }2062 video_set_drvdata(vfd, dev);2063 2064 ret = video_register_device(vfd, VFL_TYPE_VIDEO, 0);2065 if (ret) {2066 v4l2_err(&dev->v4l2_dev, "Failed to register video device '%s'\n", name);2067 v4l2_m2m_release(dev_instance->m2m_dev);2068 return ret;2069 }2070 v4l2_info(&dev->v4l2_dev, "Device '%s' registered as /dev/video%d\n",2071 name, vfd->num);2072 return 0;2073}2074 2075static void vicodec_v4l2_dev_release(struct v4l2_device *v4l2_dev)2076{2077 struct vicodec_dev *dev = container_of(v4l2_dev, struct vicodec_dev, v4l2_dev);2078 2079 v4l2_device_unregister(&dev->v4l2_dev);2080 v4l2_m2m_release(dev->stateful_enc.m2m_dev);2081 v4l2_m2m_release(dev->stateful_dec.m2m_dev);2082 v4l2_m2m_release(dev->stateless_dec.m2m_dev);2083#ifdef CONFIG_MEDIA_CONTROLLER2084 media_device_cleanup(&dev->mdev);2085#endif2086 kfree(dev);2087}2088 2089static int vicodec_probe(struct platform_device *pdev)2090{2091 struct vicodec_dev *dev;2092 int ret;2093 2094 dev = kzalloc(sizeof(*dev), GFP_KERNEL);2095 if (!dev)2096 return -ENOMEM;2097 2098 ret = v4l2_device_register(&pdev->dev, &dev->v4l2_dev);2099 if (ret)2100 goto free_dev;2101 2102 dev->v4l2_dev.release = vicodec_v4l2_dev_release;2103 2104#ifdef CONFIG_MEDIA_CONTROLLER2105 dev->mdev.dev = &pdev->dev;2106 strscpy(dev->mdev.model, "vicodec", sizeof(dev->mdev.model));2107 strscpy(dev->mdev.bus_info, "platform:vicodec",2108 sizeof(dev->mdev.bus_info));2109 media_device_init(&dev->mdev);2110 dev->mdev.ops = &vicodec_m2m_media_ops;2111 dev->v4l2_dev.mdev = &dev->mdev;2112#endif2113 2114 platform_set_drvdata(pdev, dev);2115 2116 ret = register_instance(dev, &dev->stateful_enc, "stateful-encoder",2117 true, false);2118 if (ret)2119 goto unreg_dev;2120 2121 ret = register_instance(dev, &dev->stateful_dec, "stateful-decoder",2122 false, false);2123 if (ret)2124 goto unreg_sf_enc;2125 2126 ret = register_instance(dev, &dev->stateless_dec, "stateless-decoder",2127 false, true);2128 if (ret)2129 goto unreg_sf_dec;2130 2131#ifdef CONFIG_MEDIA_CONTROLLER2132 ret = v4l2_m2m_register_media_controller(dev->stateful_enc.m2m_dev,2133 &dev->stateful_enc.vfd,2134 MEDIA_ENT_F_PROC_VIDEO_ENCODER);2135 if (ret) {2136 v4l2_err(&dev->v4l2_dev, "Failed to init mem2mem media controller for enc\n");2137 goto unreg_m2m;2138 }2139 2140 ret = v4l2_m2m_register_media_controller(dev->stateful_dec.m2m_dev,2141 &dev->stateful_dec.vfd,2142 MEDIA_ENT_F_PROC_VIDEO_DECODER);2143 if (ret) {2144 v4l2_err(&dev->v4l2_dev, "Failed to init mem2mem media controller for dec\n");2145 goto unreg_m2m_sf_enc_mc;2146 }2147 2148 ret = v4l2_m2m_register_media_controller(dev->stateless_dec.m2m_dev,2149 &dev->stateless_dec.vfd,2150 MEDIA_ENT_F_PROC_VIDEO_DECODER);2151 if (ret) {2152 v4l2_err(&dev->v4l2_dev, "Failed to init mem2mem media controller for stateless dec\n");2153 goto unreg_m2m_sf_dec_mc;2154 }2155 2156 ret = media_device_register(&dev->mdev);2157 if (ret) {2158 v4l2_err(&dev->v4l2_dev, "Failed to register mem2mem media device\n");2159 goto unreg_m2m_sl_dec_mc;2160 }2161#endif2162 return 0;2163 2164#ifdef CONFIG_MEDIA_CONTROLLER2165unreg_m2m_sl_dec_mc:2166 v4l2_m2m_unregister_media_controller(dev->stateless_dec.m2m_dev);2167unreg_m2m_sf_dec_mc:2168 v4l2_m2m_unregister_media_controller(dev->stateful_dec.m2m_dev);2169unreg_m2m_sf_enc_mc:2170 v4l2_m2m_unregister_media_controller(dev->stateful_enc.m2m_dev);2171unreg_m2m:2172 video_unregister_device(&dev->stateless_dec.vfd);2173 v4l2_m2m_release(dev->stateless_dec.m2m_dev);2174#endif2175unreg_sf_dec:2176 video_unregister_device(&dev->stateful_dec.vfd);2177 v4l2_m2m_release(dev->stateful_dec.m2m_dev);2178unreg_sf_enc:2179 video_unregister_device(&dev->stateful_enc.vfd);2180 v4l2_m2m_release(dev->stateful_enc.m2m_dev);2181unreg_dev:2182 v4l2_device_unregister(&dev->v4l2_dev);2183free_dev:2184 kfree(dev);2185 2186 return ret;2187}2188 2189static void vicodec_remove(struct platform_device *pdev)2190{2191 struct vicodec_dev *dev = platform_get_drvdata(pdev);2192 2193 v4l2_info(&dev->v4l2_dev, "Removing " VICODEC_NAME);2194 2195#ifdef CONFIG_MEDIA_CONTROLLER2196 media_device_unregister(&dev->mdev);2197 v4l2_m2m_unregister_media_controller(dev->stateful_enc.m2m_dev);2198 v4l2_m2m_unregister_media_controller(dev->stateful_dec.m2m_dev);2199 v4l2_m2m_unregister_media_controller(dev->stateless_dec.m2m_dev);2200#endif2201 2202 video_unregister_device(&dev->stateful_enc.vfd);2203 video_unregister_device(&dev->stateful_dec.vfd);2204 video_unregister_device(&dev->stateless_dec.vfd);2205 v4l2_device_put(&dev->v4l2_dev);2206}2207 2208static struct platform_driver vicodec_pdrv = {2209 .probe = vicodec_probe,2210 .remove_new = vicodec_remove,2211 .driver = {2212 .name = VICODEC_NAME,2213 },2214};2215 2216static void __exit vicodec_exit(void)2217{2218 platform_driver_unregister(&vicodec_pdrv);2219 platform_device_unregister(&vicodec_pdev);2220}2221 2222static int __init vicodec_init(void)2223{2224 int ret;2225 2226 ret = platform_device_register(&vicodec_pdev);2227 if (ret)2228 return ret;2229 2230 ret = platform_driver_register(&vicodec_pdrv);2231 if (ret)2232 platform_device_unregister(&vicodec_pdev);2233 2234 return ret;2235}2236 2237module_init(vicodec_init);2238module_exit(vicodec_exit);2239