drivers/media/platform/qcom/venus/helpers.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * Copyright (c) 2012-2016, The Linux Foundation. All rights reserved.
   4  * Copyright (C) 2017 Linaro Ltd.
   5  */
   6 #include <linux/idr.h>
   7 #include <linux/list.h>
   8 #include <linux/mutex.h>
   9 #include <linux/slab.h>
  10 #include <linux/kernel.h>
  11 #include <media/videobuf2-dma-contig.h>
  12 #include <media/v4l2-mem2mem.h>
  13 #include <asm/div64.h>
  14
  15 #include "core.h"
  16 #include "helpers.h"
  17 #include "hfi_helper.h"
  18 #include "pm_helpers.h"
  19 #include "hfi_platform.h"
  20 #include "hfi_parser.h"
  21
  22 #define NUM_MBS_720P    (((ALIGN(1280, 16)) >> 4) * ((ALIGN(736, 16)) >> 4))
  23 #define NUM_MBS_4K      (((ALIGN(4096, 16)) >> 4) * ((ALIGN(2304, 16)) >> 4))
  24
  25 enum dpb_buf_owner {
  26         DRIVER,
  27         FIRMWARE,
  28 };
  29
  30 struct intbuf {
  31         struct list_head list;
  32         u32 type;
  33         size_t size;
  34         void *va;
  35         dma_addr_t da;
  36         unsigned long attrs;
  37         enum dpb_buf_owner owned_by;
  38         u32 dpb_out_tag;
  39 };
  40
  41 bool venus_helper_check_codec(struct venus_inst *inst, u32 v4l2_pixfmt)
  42 {
  43         struct venus_core *core = inst->core;
  44         u32 session_type = inst->session_type;
  45         u32 codec;
  46
  47         switch (v4l2_pixfmt) {
  48         case V4L2_PIX_FMT_H264:
  49                 codec = HFI_VIDEO_CODEC_H264;
  50                 break;
  51         case V4L2_PIX_FMT_H263:
  52                 codec = HFI_VIDEO_CODEC_H263;
  53                 break;
  54         case V4L2_PIX_FMT_MPEG1:
  55                 codec = HFI_VIDEO_CODEC_MPEG1;
  56                 break;
  57         case V4L2_PIX_FMT_MPEG2:
  58                 codec = HFI_VIDEO_CODEC_MPEG2;
  59                 break;
  60         case V4L2_PIX_FMT_MPEG4:
  61                 codec = HFI_VIDEO_CODEC_MPEG4;
  62                 break;
  63         case V4L2_PIX_FMT_VC1_ANNEX_G:
  64         case V4L2_PIX_FMT_VC1_ANNEX_L:
  65                 codec = HFI_VIDEO_CODEC_VC1;
  66                 break;
  67         case V4L2_PIX_FMT_VP8:
  68                 codec = HFI_VIDEO_CODEC_VP8;
  69                 break;
  70         case V4L2_PIX_FMT_VP9:
  71                 codec = HFI_VIDEO_CODEC_VP9;
  72                 break;
  73         case V4L2_PIX_FMT_XVID:
  74                 codec = HFI_VIDEO_CODEC_DIVX;
  75                 break;
  76         case V4L2_PIX_FMT_HEVC:
  77                 codec = HFI_VIDEO_CODEC_HEVC;
  78                 break;
  79         default:
  80                 return false;
  81         }
  82
  83         if (session_type == VIDC_SESSION_TYPE_ENC && core->enc_codecs & codec)
  84                 return true;
  85
  86         if (session_type == VIDC_SESSION_TYPE_DEC && core->dec_codecs & codec)
  87                 return true;
  88
  89         return false;
  90 }
  91 EXPORT_SYMBOL_GPL(venus_helper_check_codec);
  92
  93 static void free_dpb_buf(struct venus_inst *inst, struct intbuf *buf)
  94 {
  95         ida_free(&inst->dpb_ids, buf->dpb_out_tag);
  96
  97         list_del_init(&buf->list);
  98         dma_free_attrs(inst->core->dev, buf->size, buf->va, buf->da,
  99                        buf->attrs);
 100         kfree(buf);
 101 }
 102
 103 int venus_helper_queue_dpb_bufs(struct venus_inst *inst)
 104 {
 105         struct intbuf *buf, *next;
 106         unsigned int dpb_size = 0;
 107         int ret = 0;
 108
 109         if (inst->dpb_buftype == HFI_BUFFER_OUTPUT)
 110                 dpb_size = inst->output_buf_size;
 111         else if (inst->dpb_buftype == HFI_BUFFER_OUTPUT2)
 112                 dpb_size = inst->output2_buf_size;
 113
 114         list_for_each_entry_safe(buf, next, &inst->dpbbufs, list) {
 115                 struct hfi_frame_data fdata;
 116
 117                 memset(&fdata, 0, sizeof(fdata));
 118                 fdata.alloc_len = buf->size;
 119                 fdata.device_addr = buf->da;
 120                 fdata.buffer_type = buf->type;
 121
 122                 if (buf->owned_by == FIRMWARE)
 123                         continue;
 124
 125                 /* free buffer from previous sequence which was released later */
 126                 if (dpb_size > buf->size) {
 127                         free_dpb_buf(inst, buf);
 128                         continue;
 129                 }
 130
 131                 fdata.clnt_data = buf->dpb_out_tag;
 132
 133                 ret = hfi_session_process_buf(inst, &fdata);
 134                 if (ret)
 135                         goto fail;
 136
 137                 buf->owned_by = FIRMWARE;
 138         }
 139
 140 fail:
 141         return ret;
 142 }
 143 EXPORT_SYMBOL_GPL(venus_helper_queue_dpb_bufs);
 144
 145 int venus_helper_free_dpb_bufs(struct venus_inst *inst)
 146 {
 147         struct intbuf *buf, *n;
 148
 149         list_for_each_entry_safe(buf, n, &inst->dpbbufs, list) {
 150                 if (buf->owned_by == FIRMWARE)
 151                         continue;
 152                 free_dpb_buf(inst, buf);
 153         }
 154
 155         if (list_empty(&inst->dpbbufs))
 156                 INIT_LIST_HEAD(&inst->dpbbufs);
 157
 158         return 0;
 159 }
 160 EXPORT_SYMBOL_GPL(venus_helper_free_dpb_bufs);
 161
 162 int venus_helper_alloc_dpb_bufs(struct venus_inst *inst)
 163 {
 164         struct venus_core *core = inst->core;
 165         struct device *dev = core->dev;
 166         enum hfi_version ver = core->res->hfi_version;
 167         struct hfi_buffer_requirements bufreq;
 168         u32 buftype = inst->dpb_buftype;
 169         unsigned int dpb_size = 0;
 170         struct intbuf *buf;
 171         unsigned int i;
 172         u32 count;
 173         int ret;
 174         int id;
 175
 176         /* no need to allocate dpb buffers */
 177         if (!inst->dpb_fmt)
 178                 return 0;
 179
 180         if (inst->dpb_buftype == HFI_BUFFER_OUTPUT)
 181                 dpb_size = inst->output_buf_size;
 182         else if (inst->dpb_buftype == HFI_BUFFER_OUTPUT2)
 183                 dpb_size = inst->output2_buf_size;
 184
 185         if (!dpb_size)
 186                 return 0;
 187
 188         ret = venus_helper_get_bufreq(inst, buftype, &bufreq);
 189         if (ret)
 190                 return ret;
 191
 192         count = hfi_bufreq_get_count_min(&bufreq, ver);
 193
 194         for (i = 0; i < count; i++) {
 195                 buf = kzalloc(sizeof(*buf), GFP_KERNEL);
 196                 if (!buf) {
 197                         ret = -ENOMEM;
 198                         goto fail;
 199                 }
 200
 201                 buf->type = buftype;
 202                 buf->size = dpb_size;
 203                 buf->attrs = DMA_ATTR_WRITE_COMBINE |
 204                              DMA_ATTR_NO_KERNEL_MAPPING;
 205                 buf->va = dma_alloc_attrs(dev, buf->size, &buf->da, GFP_KERNEL,
 206                                           buf->attrs);
 207                 if (!buf->va) {
 208                         ret = -ENOMEM;
 209                         goto fail;
 210                 }
 211                 buf->owned_by = DRIVER;
 212
 213                 id = ida_alloc_min(&inst->dpb_ids, VB2_MAX_FRAME, GFP_KERNEL);
 214                 if (id < 0) {
 215                         ret = id;
 216                         goto fail;
 217                 }
 218
 219                 buf->dpb_out_tag = id;
 220
 221                 list_add_tail(&buf->list, &inst->dpbbufs);
 222         }
 223
 224         return 0;
 225
 226 fail:
 227         kfree(buf);
 228         venus_helper_free_dpb_bufs(inst);
 229         return ret;
 230 }
 231 EXPORT_SYMBOL_GPL(venus_helper_alloc_dpb_bufs);
 232
 233 static int intbufs_set_buffer(struct venus_inst *inst, u32 type)
 234 {
 235         struct venus_core *core = inst->core;
 236         struct device *dev = core->dev;
 237         struct hfi_buffer_requirements bufreq;
 238         struct hfi_buffer_desc bd;
 239         struct intbuf *buf;
 240         unsigned int i;
 241         int ret;
 242
 243         ret = venus_helper_get_bufreq(inst, type, &bufreq);
 244         if (ret)
 245                 return 0;
 246
 247         if (!bufreq.size)
 248                 return 0;
 249
 250         for (i = 0; i < bufreq.count_actual; i++) {
 251                 buf = kzalloc(sizeof(*buf), GFP_KERNEL);
 252                 if (!buf) {
 253                         ret = -ENOMEM;
 254                         goto fail;
 255                 }
 256
 257                 buf->type = bufreq.type;
 258                 buf->size = bufreq.size;
 259                 buf->attrs = DMA_ATTR_WRITE_COMBINE |
 260                              DMA_ATTR_NO_KERNEL_MAPPING;
 261                 buf->va = dma_alloc_attrs(dev, buf->size, &buf->da, GFP_KERNEL,
 262                                           buf->attrs);
 263                 if (!buf->va) {
 264                         ret = -ENOMEM;
 265                         goto fail;
 266                 }
 267
 268                 memset(&bd, 0, sizeof(bd));
 269                 bd.buffer_size = buf->size;
 270                 bd.buffer_type = buf->type;
 271                 bd.num_buffers = 1;
 272                 bd.device_addr = buf->da;
 273
 274                 ret = hfi_session_set_buffers(inst, &bd);
 275                 if (ret) {
 276                         dev_err(dev, "set session buffers failed\n");
 277                         goto dma_free;
 278                 }
 279
 280                 list_add_tail(&buf->list, &inst->internalbufs);
 281         }
 282
 283         return 0;
 284
 285 dma_free:
 286         dma_free_attrs(dev, buf->size, buf->va, buf->da, buf->attrs);
 287 fail:
 288         kfree(buf);
 289         return ret;
 290 }
 291
 292 static int intbufs_unset_buffers(struct venus_inst *inst)
 293 {
 294         struct hfi_buffer_desc bd = {0};
 295         struct intbuf *buf, *n;
 296         int ret = 0;
 297
 298         list_for_each_entry_safe(buf, n, &inst->internalbufs, list) {
 299                 bd.buffer_size = buf->size;
 300                 bd.buffer_type = buf->type;
 301                 bd.num_buffers = 1;
 302                 bd.device_addr = buf->da;
 303                 bd.response_required = true;
 304
 305                 ret = hfi_session_unset_buffers(inst, &bd);
 306
 307                 list_del_init(&buf->list);
 308                 dma_free_attrs(inst->core->dev, buf->size, buf->va, buf->da,
 309                                buf->attrs);
 310                 kfree(buf);
 311         }
 312
 313         return ret;
 314 }
 315
 316 static const unsigned int intbuf_types_1xx[] = {
 317         HFI_BUFFER_INTERNAL_SCRATCH(HFI_VERSION_1XX),
 318         HFI_BUFFER_INTERNAL_SCRATCH_1(HFI_VERSION_1XX),
 319         HFI_BUFFER_INTERNAL_SCRATCH_2(HFI_VERSION_1XX),
 320         HFI_BUFFER_INTERNAL_PERSIST,
 321         HFI_BUFFER_INTERNAL_PERSIST_1,
 322 };
 323
 324 static const unsigned int intbuf_types_4xx[] = {
 325         HFI_BUFFER_INTERNAL_SCRATCH(HFI_VERSION_4XX),
 326         HFI_BUFFER_INTERNAL_SCRATCH_1(HFI_VERSION_4XX),
 327         HFI_BUFFER_INTERNAL_SCRATCH_2(HFI_VERSION_4XX),
 328         HFI_BUFFER_INTERNAL_PERSIST,
 329         HFI_BUFFER_INTERNAL_PERSIST_1,
 330 };
 331
 332 static const unsigned int intbuf_types_6xx[] = {
 333         HFI_BUFFER_INTERNAL_SCRATCH(HFI_VERSION_6XX),
 334         HFI_BUFFER_INTERNAL_SCRATCH_1(HFI_VERSION_6XX),
 335         HFI_BUFFER_INTERNAL_SCRATCH_2(HFI_VERSION_6XX),
 336         HFI_BUFFER_INTERNAL_PERSIST,
 337         HFI_BUFFER_INTERNAL_PERSIST_1,
 338 };
 339
 340 int venus_helper_intbufs_alloc(struct venus_inst *inst)
 341 {
 342         const unsigned int *intbuf;
 343         size_t arr_sz, i;
 344         int ret;
 345
 346         if (IS_V6(inst->core)) {
 347                 arr_sz = ARRAY_SIZE(intbuf_types_6xx);
 348                 intbuf = intbuf_types_6xx;
 349         } else if (IS_V4(inst->core)) {
 350                 arr_sz = ARRAY_SIZE(intbuf_types_4xx);
 351                 intbuf = intbuf_types_4xx;
 352         } else {
 353                 arr_sz = ARRAY_SIZE(intbuf_types_1xx);
 354                 intbuf = intbuf_types_1xx;
 355         }
 356
 357         for (i = 0; i < arr_sz; i++) {
 358                 ret = intbufs_set_buffer(inst, intbuf[i]);
 359                 if (ret)
 360                         goto error;
 361         }
 362
 363         return 0;
 364
 365 error:
 366         intbufs_unset_buffers(inst);
 367         return ret;
 368 }
 369 EXPORT_SYMBOL_GPL(venus_helper_intbufs_alloc);
 370
 371 int venus_helper_intbufs_free(struct venus_inst *inst)
 372 {
 373         return intbufs_unset_buffers(inst);
 374 }
 375 EXPORT_SYMBOL_GPL(venus_helper_intbufs_free);
 376
 377 int venus_helper_intbufs_realloc(struct venus_inst *inst)
 378 {
 379         enum hfi_version ver = inst->core->res->hfi_version;
 380         struct hfi_buffer_desc bd;
 381         struct intbuf *buf, *n;
 382         int ret;
 383
 384         list_for_each_entry_safe(buf, n, &inst->internalbufs, list) {
 385                 if (buf->type == HFI_BUFFER_INTERNAL_PERSIST ||
 386                     buf->type == HFI_BUFFER_INTERNAL_PERSIST_1)
 387                         continue;
 388
 389                 memset(&bd, 0, sizeof(bd));
 390                 bd.buffer_size = buf->size;
 391                 bd.buffer_type = buf->type;
 392                 bd.num_buffers = 1;
 393                 bd.device_addr = buf->da;
 394                 bd.response_required = true;
 395
 396                 ret = hfi_session_unset_buffers(inst, &bd);
 397
 398                 dma_free_attrs(inst->core->dev, buf->size, buf->va, buf->da,
 399                                buf->attrs);
 400
 401                 list_del_init(&buf->list);
 402                 kfree(buf);
 403         }
 404
 405         ret = intbufs_set_buffer(inst, HFI_BUFFER_INTERNAL_SCRATCH(ver));
 406         if (ret)
 407                 goto err;
 408
 409         ret = intbufs_set_buffer(inst, HFI_BUFFER_INTERNAL_SCRATCH_1(ver));
 410         if (ret)
 411                 goto err;
 412
 413         ret = intbufs_set_buffer(inst, HFI_BUFFER_INTERNAL_SCRATCH_2(ver));
 414         if (ret)
 415                 goto err;
 416
 417         return 0;
 418 err:
 419         return ret;
 420 }
 421 EXPORT_SYMBOL_GPL(venus_helper_intbufs_realloc);
 422
 423 static void fill_buffer_desc(const struct venus_buffer *buf,
 424                              struct hfi_buffer_desc *bd, bool response)
 425 {
 426         memset(bd, 0, sizeof(*bd));
 427         bd->buffer_type = HFI_BUFFER_OUTPUT;
 428         bd->buffer_size = buf->size;
 429         bd->num_buffers = 1;
 430         bd->device_addr = buf->dma_addr;
 431         bd->response_required = response;
 432 }
 433
 434 static void return_buf_error(struct venus_inst *inst,
 435                              struct vb2_v4l2_buffer *vbuf)
 436 {
 437         struct v4l2_m2m_ctx *m2m_ctx = inst->m2m_ctx;
 438
 439         if (vbuf->vb2_buf.type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE)
 440                 v4l2_m2m_src_buf_remove_by_buf(m2m_ctx, vbuf);
 441         else
 442                 v4l2_m2m_dst_buf_remove_by_buf(m2m_ctx, vbuf);
 443
 444         v4l2_m2m_buf_done(vbuf, VB2_BUF_STATE_ERROR);
 445 }
 446
 447 static void
 448 put_ts_metadata(struct venus_inst *inst, struct vb2_v4l2_buffer *vbuf)
 449 {
 450         struct vb2_buffer *vb = &vbuf->vb2_buf;
 451         unsigned int i;
 452         int slot = -1;
 453         u64 ts_us = vb->timestamp;
 454
 455         for (i = 0; i < ARRAY_SIZE(inst->tss); i++) {
 456                 if (!inst->tss[i].used) {
 457                         slot = i;
 458                         break;
 459                 }
 460         }
 461
 462         if (slot == -1) {
 463                 dev_dbg(inst->core->dev, VDBGL "no free slot\n");
 464                 return;
 465         }
 466
 467         do_div(ts_us, NSEC_PER_USEC);
 468
 469         inst->tss[slot].used = true;
 470         inst->tss[slot].flags = vbuf->flags;
 471         inst->tss[slot].tc = vbuf->timecode;
 472         inst->tss[slot].ts_us = ts_us;
 473         inst->tss[slot].ts_ns = vb->timestamp;
 474 }
 475
 476 void venus_helper_get_ts_metadata(struct venus_inst *inst, u64 timestamp_us,
 477                                   struct vb2_v4l2_buffer *vbuf)
 478 {
 479         struct vb2_buffer *vb = &vbuf->vb2_buf;
 480         unsigned int i;
 481
 482         for (i = 0; i < ARRAY_SIZE(inst->tss); ++i) {
 483                 if (!inst->tss[i].used)
 484                         continue;
 485
 486                 if (inst->tss[i].ts_us != timestamp_us)
 487                         continue;
 488
 489                 inst->tss[i].used = false;
 490                 vbuf->flags |= inst->tss[i].flags;
 491                 vbuf->timecode = inst->tss[i].tc;
 492                 vb->timestamp = inst->tss[i].ts_ns;
 493                 break;
 494         }
 495 }
 496 EXPORT_SYMBOL_GPL(venus_helper_get_ts_metadata);
 497
 498 static int
 499 session_process_buf(struct venus_inst *inst, struct vb2_v4l2_buffer *vbuf)
 500 {
 501         struct venus_buffer *buf = to_venus_buffer(vbuf);
 502         struct vb2_buffer *vb = &vbuf->vb2_buf;
 503         unsigned int type = vb->type;
 504         struct hfi_frame_data fdata;
 505
 506         memset(&fdata, 0, sizeof(fdata));
 507         fdata.alloc_len = buf->size;
 508         fdata.device_addr = buf->dma_addr;
 509         fdata.timestamp = vb->timestamp;
 510         do_div(fdata.timestamp, NSEC_PER_USEC);
 511         fdata.flags = 0;
 512         fdata.clnt_data = vbuf->vb2_buf.index;
 513
 514         if (type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) {
 515                 fdata.buffer_type = HFI_BUFFER_INPUT;
 516                 fdata.filled_len = vb2_get_plane_payload(vb, 0);
 517                 fdata.offset = vb->planes[0].data_offset;
 518
 519                 if (vbuf->flags & V4L2_BUF_FLAG_LAST || !fdata.filled_len)
 520                         fdata.flags |= HFI_BUFFERFLAG_EOS;
 521
 522                 if (inst->session_type == VIDC_SESSION_TYPE_DEC)
 523                         put_ts_metadata(inst, vbuf);
 524
 525                 venus_pm_load_scale(inst);
 526         } else if (type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE) {
 527                 if (inst->session_type == VIDC_SESSION_TYPE_ENC)
 528                         fdata.buffer_type = HFI_BUFFER_OUTPUT;
 529                 else
 530                         fdata.buffer_type = inst->opb_buftype;
 531                 fdata.filled_len = 0;
 532                 fdata.offset = 0;
 533         }
 534
 535         return hfi_session_process_buf(inst, &fdata);
 536 }
 537
 538 static bool is_dynamic_bufmode(struct venus_inst *inst)
 539 {
 540         struct venus_core *core = inst->core;
 541         struct hfi_plat_caps *caps;
 542
 543         /*
 544          * v4 doesn't send BUFFER_ALLOC_MODE_SUPPORTED property and supports
 545          * dynamic buffer mode by default for HFI_BUFFER_OUTPUT/OUTPUT2.
 546          */
 547         if (IS_V4(core) || IS_V6(core))
 548                 return true;
 549
 550         caps = venus_caps_by_codec(core, inst->hfi_codec, inst->session_type);
 551         if (!caps)
 552                 return false;
 553
 554         return caps->cap_bufs_mode_dynamic;
 555 }
 556
 557 int venus_helper_unregister_bufs(struct venus_inst *inst)
 558 {
 559         struct venus_buffer *buf, *n;
 560         struct hfi_buffer_desc bd;
 561         int ret = 0;
 562
 563         if (is_dynamic_bufmode(inst))
 564                 return 0;
 565
 566         list_for_each_entry_safe(buf, n, &inst->registeredbufs, reg_list) {
 567                 fill_buffer_desc(buf, &bd, true);
 568                 ret = hfi_session_unset_buffers(inst, &bd);
 569                 list_del_init(&buf->reg_list);
 570         }
 571
 572         return ret;
 573 }
 574 EXPORT_SYMBOL_GPL(venus_helper_unregister_bufs);
 575
 576 static int session_register_bufs(struct venus_inst *inst)
 577 {
 578         struct venus_core *core = inst->core;
 579         struct device *dev = core->dev;
 580         struct hfi_buffer_desc bd;
 581         struct venus_buffer *buf;
 582         int ret = 0;
 583
 584         if (is_dynamic_bufmode(inst))
 585                 return 0;
 586
 587         list_for_each_entry(buf, &inst->registeredbufs, reg_list) {
 588                 fill_buffer_desc(buf, &bd, false);
 589                 ret = hfi_session_set_buffers(inst, &bd);
 590                 if (ret) {
 591                         dev_err(dev, "%s: set buffer failed\n", __func__);
 592                         break;
 593                 }
 594         }
 595
 596         return ret;
 597 }
 598
 599 static u32 to_hfi_raw_fmt(u32 v4l2_fmt)
 600 {
 601         switch (v4l2_fmt) {
 602         case V4L2_PIX_FMT_NV12:
 603                 return HFI_COLOR_FORMAT_NV12;
 604         case V4L2_PIX_FMT_NV21:
 605                 return HFI_COLOR_FORMAT_NV21;
 606         case V4L2_PIX_FMT_QC08C:
 607                 return HFI_COLOR_FORMAT_NV12_UBWC;
 608         case V4L2_PIX_FMT_QC10C:
 609                 return HFI_COLOR_FORMAT_YUV420_TP10_UBWC;
 610         case V4L2_PIX_FMT_P010:
 611                 return HFI_COLOR_FORMAT_P010;
 612         default:
 613                 break;
 614         }
 615
 616         return 0;
 617 }
 618
 619 static int platform_get_bufreq(struct venus_inst *inst, u32 buftype,
 620                                struct hfi_buffer_requirements *req)
 621 {
 622         enum hfi_version version = inst->core->res->hfi_version;
 623         const struct hfi_platform *hfi_plat;
 624         struct hfi_plat_buffers_params params;
 625         bool is_dec = inst->session_type == VIDC_SESSION_TYPE_DEC;
 626         struct venc_controls *enc_ctr = &inst->controls.enc;
 627
 628         hfi_plat = hfi_platform_get(version);
 629
 630         if (!hfi_plat || !hfi_plat->bufreq)
 631                 return -EINVAL;
 632
 633         params.version = version;
 634         params.num_vpp_pipes = inst->core->res->num_vpp_pipes;
 635
 636         if (is_dec) {
 637                 params.width = inst->width;
 638                 params.height = inst->height;
 639                 params.out_width = inst->out_width;
 640                 params.out_height = inst->out_height;
 641                 params.codec = inst->fmt_out->pixfmt;
 642                 params.hfi_color_fmt = to_hfi_raw_fmt(inst->fmt_cap->pixfmt);
 643                 params.dec.max_mbs_per_frame = mbs_per_frame_max(inst);
 644                 params.dec.buffer_size_limit = 0;
 645                 params.dec.is_secondary_output =
 646                         inst->opb_buftype == HFI_BUFFER_OUTPUT2;
 647                 if (params.dec.is_secondary_output)
 648                         params.hfi_dpb_color_fmt = inst->dpb_fmt;
 649                 params.dec.is_interlaced =
 650                         inst->pic_struct != HFI_INTERLACE_FRAME_PROGRESSIVE;
 651         } else {
 652                 params.width = inst->out_width;
 653                 params.height = inst->out_height;
 654                 params.codec = inst->fmt_cap->pixfmt;
 655                 params.hfi_color_fmt = to_hfi_raw_fmt(inst->fmt_out->pixfmt);
 656                 params.enc.work_mode = VIDC_WORK_MODE_2;
 657                 params.enc.rc_type = HFI_RATE_CONTROL_OFF;
 658                 if (enc_ctr->bitrate_mode == V4L2_MPEG_VIDEO_BITRATE_MODE_CQ)
 659                         params.enc.rc_type = HFI_RATE_CONTROL_CQ;
 660                 params.enc.num_b_frames = enc_ctr->num_b_frames;
 661                 params.enc.is_tenbit = inst->bit_depth == VIDC_BITDEPTH_10;
 662         }
 663
 664         return hfi_plat->bufreq(&params, inst->session_type, buftype, req);
 665 }
 666
 667 int venus_helper_get_bufreq(struct venus_inst *inst, u32 type,
 668                             struct hfi_buffer_requirements *req)
 669 {
 670         u32 ptype = HFI_PROPERTY_CONFIG_BUFFER_REQUIREMENTS;
 671         enum hfi_version ver = inst->core->res->hfi_version;
 672         union hfi_get_property hprop;
 673         unsigned int i;
 674         int ret;
 675
 676         memset(req, 0, sizeof(*req));
 677
 678         if (type == HFI_BUFFER_OUTPUT || type == HFI_BUFFER_OUTPUT2)
 679                 hfi_bufreq_set_count_min(req, ver, inst->fw_min_cnt);
 680
 681         ret = platform_get_bufreq(inst, type, req);
 682         if (!ret) {
 683                 if (type == HFI_BUFFER_OUTPUT || type == HFI_BUFFER_OUTPUT2)
 684                         inst->fw_min_cnt = hfi_bufreq_get_count_min(req, ver);
 685                 return 0;
 686         }
 687
 688         ret = hfi_session_get_property(inst, ptype, &hprop);
 689         if (ret)
 690                 return ret;
 691
 692         ret = -EINVAL;
 693
 694         for (i = 0; i < HFI_BUFFER_TYPE_MAX; i++) {
 695                 if (hprop.bufreq[i].type != type)
 696                         continue;
 697
 698                 memcpy(req, &hprop.bufreq[i], sizeof(*req));
 699                 ret = 0;
 700                 break;
 701         }
 702
 703         return ret;
 704 }
 705 EXPORT_SYMBOL_GPL(venus_helper_get_bufreq);
 706
 707 struct id_mapping {
 708         u32 hfi_id;
 709         u32 v4l2_id;
 710 };
 711
 712 static const struct id_mapping mpeg4_profiles[] = {
 713         { HFI_MPEG4_PROFILE_SIMPLE, V4L2_MPEG_VIDEO_MPEG4_PROFILE_SIMPLE },
 714         { HFI_MPEG4_PROFILE_ADVANCEDSIMPLE, V4L2_MPEG_VIDEO_MPEG4_PROFILE_ADVANCED_SIMPLE },
 715 };
 716
 717 static const struct id_mapping mpeg4_levels[] = {
 718         { HFI_MPEG4_LEVEL_0, V4L2_MPEG_VIDEO_MPEG4_LEVEL_0 },
 719         { HFI_MPEG4_LEVEL_0b, V4L2_MPEG_VIDEO_MPEG4_LEVEL_0B },
 720         { HFI_MPEG4_LEVEL_1, V4L2_MPEG_VIDEO_MPEG4_LEVEL_1 },
 721         { HFI_MPEG4_LEVEL_2, V4L2_MPEG_VIDEO_MPEG4_LEVEL_2 },
 722         { HFI_MPEG4_LEVEL_3, V4L2_MPEG_VIDEO_MPEG4_LEVEL_3 },
 723         { HFI_MPEG4_LEVEL_4, V4L2_MPEG_VIDEO_MPEG4_LEVEL_4 },
 724         { HFI_MPEG4_LEVEL_5, V4L2_MPEG_VIDEO_MPEG4_LEVEL_5 },
 725 };
 726
 727 static const struct id_mapping mpeg2_profiles[] = {
 728         { HFI_MPEG2_PROFILE_SIMPLE, V4L2_MPEG_VIDEO_MPEG2_PROFILE_SIMPLE },
 729         { HFI_MPEG2_PROFILE_MAIN, V4L2_MPEG_VIDEO_MPEG2_PROFILE_MAIN },
 730         { HFI_MPEG2_PROFILE_SNR, V4L2_MPEG_VIDEO_MPEG2_PROFILE_SNR_SCALABLE },
 731         { HFI_MPEG2_PROFILE_SPATIAL, V4L2_MPEG_VIDEO_MPEG2_PROFILE_SPATIALLY_SCALABLE },
 732         { HFI_MPEG2_PROFILE_HIGH, V4L2_MPEG_VIDEO_MPEG2_PROFILE_HIGH },
 733 };
 734
 735 static const struct id_mapping mpeg2_levels[] = {
 736         { HFI_MPEG2_LEVEL_LL, V4L2_MPEG_VIDEO_MPEG2_LEVEL_LOW },
 737         { HFI_MPEG2_LEVEL_ML, V4L2_MPEG_VIDEO_MPEG2_LEVEL_MAIN },
 738         { HFI_MPEG2_LEVEL_H14, V4L2_MPEG_VIDEO_MPEG2_LEVEL_HIGH_1440 },
 739         { HFI_MPEG2_LEVEL_HL, V4L2_MPEG_VIDEO_MPEG2_LEVEL_HIGH },
 740 };
 741
 742 static const struct id_mapping h264_profiles[] = {
 743         { HFI_H264_PROFILE_BASELINE, V4L2_MPEG_VIDEO_H264_PROFILE_BASELINE },
 744         { HFI_H264_PROFILE_MAIN, V4L2_MPEG_VIDEO_H264_PROFILE_MAIN },
 745         { HFI_H264_PROFILE_HIGH, V4L2_MPEG_VIDEO_H264_PROFILE_HIGH },
 746         { HFI_H264_PROFILE_STEREO_HIGH, V4L2_MPEG_VIDEO_H264_PROFILE_STEREO_HIGH },
 747         { HFI_H264_PROFILE_MULTIVIEW_HIGH, V4L2_MPEG_VIDEO_H264_PROFILE_MULTIVIEW_HIGH },
 748         { HFI_H264_PROFILE_CONSTRAINED_BASE, V4L2_MPEG_VIDEO_H264_PROFILE_CONSTRAINED_BASELINE },
 749         { HFI_H264_PROFILE_CONSTRAINED_HIGH, V4L2_MPEG_VIDEO_H264_PROFILE_CONSTRAINED_HIGH },
 750 };
 751
 752 static const struct id_mapping h264_levels[] = {
 753         { HFI_H264_LEVEL_1, V4L2_MPEG_VIDEO_H264_LEVEL_1_0 },
 754         { HFI_H264_LEVEL_1b, V4L2_MPEG_VIDEO_H264_LEVEL_1B },
 755         { HFI_H264_LEVEL_11, V4L2_MPEG_VIDEO_H264_LEVEL_1_1 },
 756         { HFI_H264_LEVEL_12, V4L2_MPEG_VIDEO_H264_LEVEL_1_2 },
 757         { HFI_H264_LEVEL_13, V4L2_MPEG_VIDEO_H264_LEVEL_1_3 },
 758         { HFI_H264_LEVEL_2, V4L2_MPEG_VIDEO_H264_LEVEL_2_0 },
 759         { HFI_H264_LEVEL_21, V4L2_MPEG_VIDEO_H264_LEVEL_2_1 },
 760         { HFI_H264_LEVEL_22, V4L2_MPEG_VIDEO_H264_LEVEL_2_2 },
 761         { HFI_H264_LEVEL_3, V4L2_MPEG_VIDEO_H264_LEVEL_3_0 },
 762         { HFI_H264_LEVEL_31, V4L2_MPEG_VIDEO_H264_LEVEL_3_1 },
 763         { HFI_H264_LEVEL_32, V4L2_MPEG_VIDEO_H264_LEVEL_3_2 },
 764         { HFI_H264_LEVEL_4, V4L2_MPEG_VIDEO_H264_LEVEL_4_0 },
 765         { HFI_H264_LEVEL_41, V4L2_MPEG_VIDEO_H264_LEVEL_4_1 },
 766         { HFI_H264_LEVEL_42, V4L2_MPEG_VIDEO_H264_LEVEL_4_2 },
 767         { HFI_H264_LEVEL_5, V4L2_MPEG_VIDEO_H264_LEVEL_5_0 },
 768         { HFI_H264_LEVEL_51, V4L2_MPEG_VIDEO_H264_LEVEL_5_1 },
 769         { HFI_H264_LEVEL_52, V4L2_MPEG_VIDEO_H264_LEVEL_5_1 },
 770 };
 771
 772 static const struct id_mapping hevc_profiles[] = {
 773         { HFI_HEVC_PROFILE_MAIN, V4L2_MPEG_VIDEO_HEVC_PROFILE_MAIN },
 774         { HFI_HEVC_PROFILE_MAIN_STILL_PIC, V4L2_MPEG_VIDEO_HEVC_PROFILE_MAIN_STILL_PICTURE },
 775         { HFI_HEVC_PROFILE_MAIN10, V4L2_MPEG_VIDEO_HEVC_PROFILE_MAIN_10 },
 776 };
 777
 778 static const struct id_mapping hevc_levels[] = {
 779         { HFI_HEVC_LEVEL_1, V4L2_MPEG_VIDEO_HEVC_LEVEL_1 },
 780         { HFI_HEVC_LEVEL_2, V4L2_MPEG_VIDEO_HEVC_LEVEL_2 },
 781         { HFI_HEVC_LEVEL_21, V4L2_MPEG_VIDEO_HEVC_LEVEL_2_1 },
 782         { HFI_HEVC_LEVEL_3, V4L2_MPEG_VIDEO_HEVC_LEVEL_3 },
 783         { HFI_HEVC_LEVEL_31, V4L2_MPEG_VIDEO_HEVC_LEVEL_3_1 },
 784         { HFI_HEVC_LEVEL_4, V4L2_MPEG_VIDEO_HEVC_LEVEL_4 },
 785         { HFI_HEVC_LEVEL_41, V4L2_MPEG_VIDEO_HEVC_LEVEL_4_1 },
 786         { HFI_HEVC_LEVEL_5, V4L2_MPEG_VIDEO_HEVC_LEVEL_5 },
 787         { HFI_HEVC_LEVEL_51, V4L2_MPEG_VIDEO_HEVC_LEVEL_5_1 },
 788         { HFI_HEVC_LEVEL_52, V4L2_MPEG_VIDEO_HEVC_LEVEL_5_2 },
 789         { HFI_HEVC_LEVEL_6, V4L2_MPEG_VIDEO_HEVC_LEVEL_6 },
 790         { HFI_HEVC_LEVEL_61, V4L2_MPEG_VIDEO_HEVC_LEVEL_6_1 },
 791         { HFI_HEVC_LEVEL_62, V4L2_MPEG_VIDEO_HEVC_LEVEL_6_2 },
 792 };
 793
 794 static const struct id_mapping vp8_profiles[] = {
 795         { HFI_VPX_PROFILE_VERSION_0, V4L2_MPEG_VIDEO_VP8_PROFILE_0 },
 796         { HFI_VPX_PROFILE_VERSION_1, V4L2_MPEG_VIDEO_VP8_PROFILE_1 },
 797         { HFI_VPX_PROFILE_VERSION_2, V4L2_MPEG_VIDEO_VP8_PROFILE_2 },
 798         { HFI_VPX_PROFILE_VERSION_3, V4L2_MPEG_VIDEO_VP8_PROFILE_3 },
 799 };
 800
 801 static const struct id_mapping vp9_profiles[] = {
 802         { HFI_VP9_PROFILE_P0, V4L2_MPEG_VIDEO_VP9_PROFILE_0 },
 803         { HFI_VP9_PROFILE_P2_10B, V4L2_MPEG_VIDEO_VP9_PROFILE_2 },
 804 };
 805
 806 static const struct id_mapping vp9_levels[] = {
 807         { HFI_VP9_LEVEL_1, V4L2_MPEG_VIDEO_VP9_LEVEL_1_0 },
 808         { HFI_VP9_LEVEL_11, V4L2_MPEG_VIDEO_VP9_LEVEL_1_1 },
 809         { HFI_VP9_LEVEL_2, V4L2_MPEG_VIDEO_VP9_LEVEL_2_0},
 810         { HFI_VP9_LEVEL_21, V4L2_MPEG_VIDEO_VP9_LEVEL_2_1 },
 811         { HFI_VP9_LEVEL_3, V4L2_MPEG_VIDEO_VP9_LEVEL_3_0},
 812         { HFI_VP9_LEVEL_31, V4L2_MPEG_VIDEO_VP9_LEVEL_3_1 },
 813         { HFI_VP9_LEVEL_4, V4L2_MPEG_VIDEO_VP9_LEVEL_4_0 },
 814         { HFI_VP9_LEVEL_41, V4L2_MPEG_VIDEO_VP9_LEVEL_4_1 },
 815         { HFI_VP9_LEVEL_5, V4L2_MPEG_VIDEO_VP9_LEVEL_5_0 },
 816         { HFI_VP9_LEVEL_51, V4L2_MPEG_VIDEO_VP9_LEVEL_5_1 },
 817         { HFI_VP9_LEVEL_6, V4L2_MPEG_VIDEO_VP9_LEVEL_6_0 },
 818         { HFI_VP9_LEVEL_61, V4L2_MPEG_VIDEO_VP9_LEVEL_6_1 },
 819 };
 820
 821 static u32 find_v4l2_id(u32 hfi_id, const struct id_mapping *array, unsigned int array_sz)
 822 {
 823         unsigned int i;
 824
 825         if (!array || !array_sz)
 826                 return 0;
 827
 828         for (i = 0; i < array_sz; i++)
 829                 if (hfi_id == array[i].hfi_id)
 830                         return array[i].v4l2_id;
 831
 832         return 0;
 833 }
 834
 835 static u32 find_hfi_id(u32 v4l2_id, const struct id_mapping *array, unsigned int array_sz)
 836 {
 837         unsigned int i;
 838
 839         if (!array || !array_sz)
 840                 return 0;
 841
 842         for (i = 0; i < array_sz; i++)
 843                 if (v4l2_id == array[i].v4l2_id)
 844                         return array[i].hfi_id;
 845
 846         return 0;
 847 }
 848
 849 static void
 850 v4l2_id_profile_level(u32 hfi_codec, struct hfi_profile_level *pl, u32 *profile, u32 *level)
 851 {
 852         u32 hfi_pf = pl->profile;
 853         u32 hfi_lvl = pl->level;
 854
 855         switch (hfi_codec) {
 856         case HFI_VIDEO_CODEC_H264:
 857                 *profile = find_v4l2_id(hfi_pf, h264_profiles, ARRAY_SIZE(h264_profiles));
 858                 *level = find_v4l2_id(hfi_lvl, h264_levels, ARRAY_SIZE(h264_levels));
 859                 break;
 860         case HFI_VIDEO_CODEC_MPEG2:
 861                 *profile = find_v4l2_id(hfi_pf, mpeg2_profiles, ARRAY_SIZE(mpeg2_profiles));
 862                 *level = find_v4l2_id(hfi_lvl, mpeg2_levels, ARRAY_SIZE(mpeg2_levels));
 863                 break;
 864         case HFI_VIDEO_CODEC_MPEG4:
 865                 *profile = find_v4l2_id(hfi_pf, mpeg4_profiles, ARRAY_SIZE(mpeg4_profiles));
 866                 *level = find_v4l2_id(hfi_lvl, mpeg4_levels, ARRAY_SIZE(mpeg4_levels));
 867                 break;
 868         case HFI_VIDEO_CODEC_VP8:
 869                 *profile = find_v4l2_id(hfi_pf, vp8_profiles, ARRAY_SIZE(vp8_profiles));
 870                 *level = 0;
 871                 break;
 872         case HFI_VIDEO_CODEC_VP9:
 873                 *profile = find_v4l2_id(hfi_pf, vp9_profiles, ARRAY_SIZE(vp9_profiles));
 874                 *level = find_v4l2_id(hfi_lvl, vp9_levels, ARRAY_SIZE(vp9_levels));
 875                 break;
 876         case HFI_VIDEO_CODEC_HEVC:
 877                 *profile = find_v4l2_id(hfi_pf, hevc_profiles, ARRAY_SIZE(hevc_profiles));
 878                 *level = find_v4l2_id(hfi_lvl, hevc_levels, ARRAY_SIZE(hevc_levels));
 879                 break;
 880         default:
 881                 break;
 882         }
 883 }
 884
 885 static void
 886 hfi_id_profile_level(u32 hfi_codec, u32 v4l2_pf, u32 v4l2_lvl, struct hfi_profile_level *pl)
 887 {
 888         switch (hfi_codec) {
 889         case HFI_VIDEO_CODEC_H264:
 890                 pl->profile = find_hfi_id(v4l2_pf, h264_profiles, ARRAY_SIZE(h264_profiles));
 891                 pl->level = find_hfi_id(v4l2_lvl, h264_levels, ARRAY_SIZE(h264_levels));
 892                 break;
 893         case HFI_VIDEO_CODEC_MPEG2:
 894                 pl->profile = find_hfi_id(v4l2_pf, mpeg2_profiles, ARRAY_SIZE(mpeg2_profiles));
 895                 pl->level = find_hfi_id(v4l2_lvl, mpeg2_levels, ARRAY_SIZE(mpeg2_levels));
 896                 break;
 897         case HFI_VIDEO_CODEC_MPEG4:
 898                 pl->profile = find_hfi_id(v4l2_pf, mpeg4_profiles, ARRAY_SIZE(mpeg4_profiles));
 899                 pl->level = find_hfi_id(v4l2_lvl, mpeg4_levels, ARRAY_SIZE(mpeg4_levels));
 900                 break;
 901         case HFI_VIDEO_CODEC_VP8:
 902                 pl->profile = find_hfi_id(v4l2_pf, vp8_profiles, ARRAY_SIZE(vp8_profiles));
 903                 pl->level = 0;
 904                 break;
 905         case HFI_VIDEO_CODEC_VP9:
 906                 pl->profile = find_hfi_id(v4l2_pf, vp9_profiles, ARRAY_SIZE(vp9_profiles));
 907                 pl->level = find_hfi_id(v4l2_lvl, vp9_levels, ARRAY_SIZE(vp9_levels));
 908                 break;
 909         case HFI_VIDEO_CODEC_HEVC:
 910                 pl->profile = find_hfi_id(v4l2_pf, hevc_profiles, ARRAY_SIZE(hevc_profiles));
 911                 pl->level = find_hfi_id(v4l2_lvl, hevc_levels, ARRAY_SIZE(hevc_levels));
 912                 break;
 913         default:
 914                 break;
 915         }
 916 }
 917
 918 int venus_helper_get_profile_level(struct venus_inst *inst, u32 *profile, u32 *level)
 919 {
 920         const u32 ptype = HFI_PROPERTY_PARAM_PROFILE_LEVEL_CURRENT;
 921         union hfi_get_property hprop;
 922         int ret;
 923
 924         ret = hfi_session_get_property(inst, ptype, &hprop);
 925         if (ret)
 926                 return ret;
 927
 928         v4l2_id_profile_level(inst->hfi_codec, &hprop.profile_level, profile, level);
 929
 930         return 0;
 931 }
 932 EXPORT_SYMBOL_GPL(venus_helper_get_profile_level);
 933
 934 int venus_helper_set_profile_level(struct venus_inst *inst, u32 profile, u32 level)
 935 {
 936         const u32 ptype = HFI_PROPERTY_PARAM_PROFILE_LEVEL_CURRENT;
 937         struct hfi_profile_level pl;
 938
 939         hfi_id_profile_level(inst->hfi_codec, profile, level, &pl);
 940
 941         return hfi_session_set_property(inst, ptype, &pl);
 942 }
 943 EXPORT_SYMBOL_GPL(venus_helper_set_profile_level);
 944
 945 static u32 get_framesize_raw_nv12(u32 width, u32 height)
 946 {
 947         u32 y_stride, uv_stride, y_plane;
 948         u32 y_sclines, uv_sclines, uv_plane;
 949         u32 size;
 950
 951         y_stride = ALIGN(width, 128);
 952         uv_stride = ALIGN(width, 128);
 953         y_sclines = ALIGN(height, 32);
 954         uv_sclines = ALIGN(((height + 1) >> 1), 16);
 955
 956         y_plane = y_stride * y_sclines;
 957         uv_plane = uv_stride * uv_sclines + SZ_4K;
 958         size = y_plane + uv_plane + SZ_8K;
 959
 960         return ALIGN(size, SZ_4K);
 961 }
 962
 963 static u32 get_framesize_raw_nv12_ubwc(u32 width, u32 height)
 964 {
 965         u32 y_meta_stride, y_meta_plane;
 966         u32 y_stride, y_plane;
 967         u32 uv_meta_stride, uv_meta_plane;
 968         u32 uv_stride, uv_plane;
 969         u32 extradata = SZ_16K;
 970
 971         y_meta_stride = ALIGN(DIV_ROUND_UP(width, 32), 64);
 972         y_meta_plane = y_meta_stride * ALIGN(DIV_ROUND_UP(height, 8), 16);
 973         y_meta_plane = ALIGN(y_meta_plane, SZ_4K);
 974
 975         y_stride = ALIGN(width, 128);
 976         y_plane = ALIGN(y_stride * ALIGN(height, 32), SZ_4K);
 977
 978         uv_meta_stride = ALIGN(DIV_ROUND_UP(width / 2, 16), 64);
 979         uv_meta_plane = uv_meta_stride * ALIGN(DIV_ROUND_UP(height / 2, 8), 16);
 980         uv_meta_plane = ALIGN(uv_meta_plane, SZ_4K);
 981
 982         uv_stride = ALIGN(width, 128);
 983         uv_plane = ALIGN(uv_stride * ALIGN(height / 2, 32), SZ_4K);
 984
 985         return ALIGN(y_meta_plane + y_plane + uv_meta_plane + uv_plane +
 986                      max(extradata, y_stride * 48), SZ_4K);
 987 }
 988
 989 static u32 get_framesize_raw_p010(u32 width, u32 height)
 990 {
 991         u32 y_plane, uv_plane, y_stride, uv_stride, y_sclines, uv_sclines;
 992
 993         y_stride = ALIGN(width * 2, 128);
 994         uv_stride = ALIGN(width * 2, 128);
 995         y_sclines = ALIGN(height, 32);
 996         uv_sclines = ALIGN((height + 1) >> 1, 16);
 997         y_plane = y_stride * y_sclines;
 998         uv_plane = uv_stride * uv_sclines;
 999
1000         return ALIGN((y_plane + uv_plane), SZ_4K);
1001 }
1002
1003 static u32 get_framesize_raw_p010_ubwc(u32 width, u32 height)
1004 {
1005         u32 y_stride, uv_stride, y_sclines, uv_sclines;
1006         u32 y_ubwc_plane, uv_ubwc_plane;
1007         u32 y_meta_stride, y_meta_scanlines;
1008         u32 uv_meta_stride, uv_meta_scanlines;
1009         u32 y_meta_plane, uv_meta_plane;
1010         u32 size;
1011
1012         y_stride = ALIGN(width * 2, 256);
1013         uv_stride = ALIGN(width * 2, 256);
1014         y_sclines = ALIGN(height, 16);
1015         uv_sclines = ALIGN((height + 1) >> 1, 16);
1016
1017         y_ubwc_plane = ALIGN(y_stride * y_sclines, SZ_4K);
1018         uv_ubwc_plane = ALIGN(uv_stride * uv_sclines, SZ_4K);
1019         y_meta_stride = ALIGN(DIV_ROUND_UP(width, 32), 64);
1020         y_meta_scanlines = ALIGN(DIV_ROUND_UP(height, 4), 16);
1021         y_meta_plane = ALIGN(y_meta_stride * y_meta_scanlines, SZ_4K);
1022         uv_meta_stride = ALIGN(DIV_ROUND_UP((width + 1) >> 1, 16), 64);
1023         uv_meta_scanlines = ALIGN(DIV_ROUND_UP((height + 1) >> 1, 4), 16);
1024         uv_meta_plane = ALIGN(uv_meta_stride * uv_meta_scanlines, SZ_4K);
1025
1026         size = y_ubwc_plane + uv_ubwc_plane + y_meta_plane + uv_meta_plane;
1027
1028         return ALIGN(size, SZ_4K);
1029 }
1030
1031 static u32 get_framesize_raw_yuv420_tp10_ubwc(u32 width, u32 height)
1032 {
1033         u32 y_stride, uv_stride, y_sclines, uv_sclines;
1034         u32 y_ubwc_plane, uv_ubwc_plane;
1035         u32 y_meta_stride, y_meta_scanlines;
1036         u32 uv_meta_stride, uv_meta_scanlines;
1037         u32 y_meta_plane, uv_meta_plane;
1038         u32 extradata = SZ_16K;
1039         u32 size;
1040
1041         y_stride = ALIGN(width * 4 / 3, 256);
1042         uv_stride = ALIGN(width * 4 / 3, 256);
1043         y_sclines = ALIGN(height, 16);
1044         uv_sclines = ALIGN((height + 1) >> 1, 16);
1045
1046         y_ubwc_plane = ALIGN(y_stride * y_sclines, SZ_4K);
1047         uv_ubwc_plane = ALIGN(uv_stride * uv_sclines, SZ_4K);
1048         y_meta_stride = ALIGN(DIV_ROUND_UP(width, 48), 64);
1049         y_meta_scanlines = ALIGN(DIV_ROUND_UP(height, 4), 16);
1050         y_meta_plane = ALIGN(y_meta_stride * y_meta_scanlines, SZ_4K);
1051         uv_meta_stride = ALIGN(DIV_ROUND_UP((width + 1) >> 1, 24), 64);
1052         uv_meta_scanlines = ALIGN(DIV_ROUND_UP((height + 1) >> 1, 4), 16);
1053         uv_meta_plane = ALIGN(uv_meta_stride * uv_meta_scanlines, SZ_4K);
1054
1055         size = y_ubwc_plane + uv_ubwc_plane + y_meta_plane + uv_meta_plane;
1056         size += max(extradata + SZ_8K, y_stride * 48);
1057
1058         return ALIGN(size, SZ_4K);
1059 }
1060
1061 u32 venus_helper_get_framesz_raw(u32 hfi_fmt, u32 width, u32 height)
1062 {
1063         switch (hfi_fmt) {
1064         case HFI_COLOR_FORMAT_NV12:
1065         case HFI_COLOR_FORMAT_NV21:
1066                 return get_framesize_raw_nv12(width, height);
1067         case HFI_COLOR_FORMAT_NV12_UBWC:
1068                 return get_framesize_raw_nv12_ubwc(width, height);
1069         case HFI_COLOR_FORMAT_P010:
1070                 return get_framesize_raw_p010(width, height);
1071         case HFI_COLOR_FORMAT_P010_UBWC:
1072                 return get_framesize_raw_p010_ubwc(width, height);
1073         case HFI_COLOR_FORMAT_YUV420_TP10_UBWC:
1074                 return get_framesize_raw_yuv420_tp10_ubwc(width, height);
1075         default:
1076                 return 0;
1077         }
1078 }
1079 EXPORT_SYMBOL_GPL(venus_helper_get_framesz_raw);
1080
1081 u32 venus_helper_get_framesz(u32 v4l2_fmt, u32 width, u32 height)
1082 {
1083         u32 hfi_fmt, sz;
1084         bool compressed;
1085
1086         switch (v4l2_fmt) {
1087         case V4L2_PIX_FMT_MPEG:
1088         case V4L2_PIX_FMT_H264:
1089         case V4L2_PIX_FMT_H264_NO_SC:
1090         case V4L2_PIX_FMT_H264_MVC:
1091         case V4L2_PIX_FMT_H263:
1092         case V4L2_PIX_FMT_MPEG1:
1093         case V4L2_PIX_FMT_MPEG2:
1094         case V4L2_PIX_FMT_MPEG4:
1095         case V4L2_PIX_FMT_XVID:
1096         case V4L2_PIX_FMT_VC1_ANNEX_G:
1097         case V4L2_PIX_FMT_VC1_ANNEX_L:
1098         case V4L2_PIX_FMT_VP8:
1099         case V4L2_PIX_FMT_VP9:
1100         case V4L2_PIX_FMT_HEVC:
1101                 compressed = true;
1102                 break;
1103         default:
1104                 compressed = false;
1105                 break;
1106         }
1107
1108         if (compressed) {
1109                 sz = ALIGN(height, 32) * ALIGN(width, 32) * 3 / 2 / 2;
1110                 if (width < 1280 || height < 720)
1111                         sz *= 8;
1112                 return ALIGN(sz, SZ_4K);
1113         }
1114
1115         hfi_fmt = to_hfi_raw_fmt(v4l2_fmt);
1116         if (!hfi_fmt)
1117                 return 0;
1118
1119         return venus_helper_get_framesz_raw(hfi_fmt, width, height);
1120 }
1121 EXPORT_SYMBOL_GPL(venus_helper_get_framesz);
1122
1123 int venus_helper_set_input_resolution(struct venus_inst *inst,
1124                                       unsigned int width, unsigned int height)
1125 {
1126         u32 ptype = HFI_PROPERTY_PARAM_FRAME_SIZE;
1127         struct hfi_framesize fs;
1128
1129         fs.buffer_type = HFI_BUFFER_INPUT;
1130         fs.width = width;
1131         fs.height = height;
1132
1133         return hfi_session_set_property(inst, ptype, &fs);
1134 }
1135 EXPORT_SYMBOL_GPL(venus_helper_set_input_resolution);
1136
1137 int venus_helper_set_output_resolution(struct venus_inst *inst,
1138                                        unsigned int width, unsigned int height,
1139                                        u32 buftype)
1140 {
1141         u32 ptype = HFI_PROPERTY_PARAM_FRAME_SIZE;
1142         struct hfi_framesize fs;
1143
1144         fs.buffer_type = buftype;
1145         fs.width = width;
1146         fs.height = height;
1147
1148         return hfi_session_set_property(inst, ptype, &fs);
1149 }
1150 EXPORT_SYMBOL_GPL(venus_helper_set_output_resolution);
1151
1152 static u32 venus_helper_get_work_mode(struct venus_inst *inst)
1153 {
1154         u32 mode;
1155         u32 num_mbs;
1156
1157         mode = VIDC_WORK_MODE_2;
1158         if (inst->session_type == VIDC_SESSION_TYPE_DEC) {
1159                 num_mbs = (ALIGN(inst->height, 16) * ALIGN(inst->width, 16)) / 256;
1160                 if (inst->hfi_codec == HFI_VIDEO_CODEC_MPEG2 ||
1161                     inst->pic_struct != HFI_INTERLACE_FRAME_PROGRESSIVE ||
1162                     num_mbs <= NUM_MBS_720P)
1163                         mode = VIDC_WORK_MODE_1;
1164         } else {
1165                 num_mbs = (ALIGN(inst->out_height, 16) * ALIGN(inst->out_width, 16)) / 256;
1166                 if (inst->hfi_codec == HFI_VIDEO_CODEC_VP8 &&
1167                     num_mbs <= NUM_MBS_4K)
1168                         mode = VIDC_WORK_MODE_1;
1169         }
1170
1171         return mode;
1172 }
1173
1174 int venus_helper_set_work_mode(struct venus_inst *inst)
1175 {
1176         const u32 ptype = HFI_PROPERTY_PARAM_WORK_MODE;
1177         struct hfi_video_work_mode wm;
1178         u32 mode;
1179
1180         if (!IS_V4(inst->core) && !IS_V6(inst->core))
1181                 return 0;
1182
1183         mode = venus_helper_get_work_mode(inst);
1184         wm.video_work_mode = mode;
1185         return hfi_session_set_property(inst, ptype, &wm);
1186 }
1187 EXPORT_SYMBOL_GPL(venus_helper_set_work_mode);
1188
1189 int venus_helper_set_format_constraints(struct venus_inst *inst)
1190 {
1191         const u32 ptype = HFI_PROPERTY_PARAM_UNCOMPRESSED_PLANE_ACTUAL_CONSTRAINTS_INFO;
1192         struct hfi_uncompressed_plane_actual_constraints_info pconstraint;
1193
1194         if (!IS_V6(inst->core))
1195                 return 0;
1196
1197         if (inst->opb_fmt == HFI_COLOR_FORMAT_NV12_UBWC ||
1198             inst->opb_fmt == HFI_COLOR_FORMAT_YUV420_TP10_UBWC)
1199                 return 0;
1200
1201         pconstraint.buffer_type = HFI_BUFFER_OUTPUT2;
1202         pconstraint.num_planes = 2;
1203         pconstraint.plane_format[0].stride_multiples = 128;
1204         pconstraint.plane_format[0].max_stride = 8192;
1205         pconstraint.plane_format[0].min_plane_buffer_height_multiple = 32;
1206         pconstraint.plane_format[0].buffer_alignment = 256;
1207
1208         pconstraint.plane_format[1].stride_multiples = 128;
1209         pconstraint.plane_format[1].max_stride = 8192;
1210         pconstraint.plane_format[1].min_plane_buffer_height_multiple = 16;
1211         pconstraint.plane_format[1].buffer_alignment = 256;
1212
1213         return hfi_session_set_property(inst, ptype, &pconstraint);
1214 }
1215 EXPORT_SYMBOL_GPL(venus_helper_set_format_constraints);
1216
1217 int venus_helper_set_num_bufs(struct venus_inst *inst, unsigned int input_bufs,
1218                               unsigned int output_bufs,
1219                               unsigned int output2_bufs)
1220 {
1221         u32 ptype = HFI_PROPERTY_PARAM_BUFFER_COUNT_ACTUAL;
1222         struct hfi_buffer_count_actual buf_count;
1223         int ret;
1224
1225         buf_count.type = HFI_BUFFER_INPUT;
1226         buf_count.count_actual = input_bufs;
1227
1228         ret = hfi_session_set_property(inst, ptype, &buf_count);
1229         if (ret)
1230                 return ret;
1231
1232         buf_count.type = HFI_BUFFER_OUTPUT;
1233         buf_count.count_actual = output_bufs;
1234
1235         ret = hfi_session_set_property(inst, ptype, &buf_count);
1236         if (ret)
1237                 return ret;
1238
1239         if (output2_bufs) {
1240                 buf_count.type = HFI_BUFFER_OUTPUT2;
1241                 buf_count.count_actual = output2_bufs;
1242
1243                 ret = hfi_session_set_property(inst, ptype, &buf_count);
1244         }
1245
1246         return ret;
1247 }
1248 EXPORT_SYMBOL_GPL(venus_helper_set_num_bufs);
1249
1250 int venus_helper_set_raw_format(struct venus_inst *inst, u32 hfi_format,
1251                                 u32 buftype)
1252 {
1253         const u32 ptype = HFI_PROPERTY_PARAM_UNCOMPRESSED_FORMAT_SELECT;
1254         struct hfi_uncompressed_format_select fmt;
1255
1256         fmt.buffer_type = buftype;
1257         fmt.format = hfi_format;
1258
1259         return hfi_session_set_property(inst, ptype, &fmt);
1260 }
1261 EXPORT_SYMBOL_GPL(venus_helper_set_raw_format);
1262
1263 int venus_helper_set_color_format(struct venus_inst *inst, u32 pixfmt)
1264 {
1265         u32 hfi_format, buftype;
1266
1267         if (inst->session_type == VIDC_SESSION_TYPE_DEC)
1268                 buftype = HFI_BUFFER_OUTPUT;
1269         else if (inst->session_type == VIDC_SESSION_TYPE_ENC)
1270                 buftype = HFI_BUFFER_INPUT;
1271         else
1272                 return -EINVAL;
1273
1274         hfi_format = to_hfi_raw_fmt(pixfmt);
1275         if (!hfi_format)
1276                 return -EINVAL;
1277
1278         return venus_helper_set_raw_format(inst, hfi_format, buftype);
1279 }
1280 EXPORT_SYMBOL_GPL(venus_helper_set_color_format);
1281
1282 int venus_helper_set_multistream(struct venus_inst *inst, bool out_en,
1283                                  bool out2_en)
1284 {
1285         struct hfi_multi_stream multi = {0};
1286         u32 ptype = HFI_PROPERTY_PARAM_VDEC_MULTI_STREAM;
1287         int ret;
1288
1289         multi.buffer_type = HFI_BUFFER_OUTPUT;
1290         multi.enable = out_en;
1291
1292         ret = hfi_session_set_property(inst, ptype, &multi);
1293         if (ret)
1294                 return ret;
1295
1296         multi.buffer_type = HFI_BUFFER_OUTPUT2;
1297         multi.enable = out2_en;
1298
1299         return hfi_session_set_property(inst, ptype, &multi);
1300 }
1301 EXPORT_SYMBOL_GPL(venus_helper_set_multistream);
1302
1303 int venus_helper_set_dyn_bufmode(struct venus_inst *inst)
1304 {
1305         const u32 ptype = HFI_PROPERTY_PARAM_BUFFER_ALLOC_MODE;
1306         struct hfi_buffer_alloc_mode mode;
1307         int ret;
1308
1309         if (!is_dynamic_bufmode(inst))
1310                 return 0;
1311
1312         mode.type = HFI_BUFFER_OUTPUT;
1313         mode.mode = HFI_BUFFER_MODE_DYNAMIC;
1314
1315         ret = hfi_session_set_property(inst, ptype, &mode);
1316         if (ret)
1317                 return ret;
1318
1319         mode.type = HFI_BUFFER_OUTPUT2;
1320
1321         return hfi_session_set_property(inst, ptype, &mode);
1322 }
1323 EXPORT_SYMBOL_GPL(venus_helper_set_dyn_bufmode);
1324
1325 int venus_helper_set_bufsize(struct venus_inst *inst, u32 bufsize, u32 buftype)
1326 {
1327         const u32 ptype = HFI_PROPERTY_PARAM_BUFFER_SIZE_ACTUAL;
1328         struct hfi_buffer_size_actual bufsz;
1329
1330         bufsz.type = buftype;
1331         bufsz.size = bufsize;
1332
1333         return hfi_session_set_property(inst, ptype, &bufsz);
1334 }
1335 EXPORT_SYMBOL_GPL(venus_helper_set_bufsize);
1336
1337 unsigned int venus_helper_get_opb_size(struct venus_inst *inst)
1338 {
1339         /* the encoder has only one output */
1340         if (inst->session_type == VIDC_SESSION_TYPE_ENC)
1341                 return inst->output_buf_size;
1342
1343         if (inst->opb_buftype == HFI_BUFFER_OUTPUT)
1344                 return inst->output_buf_size;
1345         else if (inst->opb_buftype == HFI_BUFFER_OUTPUT2)
1346                 return inst->output2_buf_size;
1347
1348         return 0;
1349 }
1350 EXPORT_SYMBOL_GPL(venus_helper_get_opb_size);
1351
1352 static void delayed_process_buf_func(struct work_struct *work)
1353 {
1354         struct venus_buffer *buf, *n;
1355         struct venus_inst *inst;
1356         int ret;
1357
1358         inst = container_of(work, struct venus_inst, delayed_process_work);
1359
1360         mutex_lock(&inst->lock);
1361
1362         if (!(inst->streamon_out & inst->streamon_cap))
1363                 goto unlock;
1364
1365         list_for_each_entry_safe(buf, n, &inst->delayed_process, ref_list) {
1366                 if (buf->flags & HFI_BUFFERFLAG_READONLY)
1367                         continue;
1368
1369                 ret = session_process_buf(inst, &buf->vb);
1370                 if (ret)
1371                         return_buf_error(inst, &buf->vb);
1372
1373                 list_del_init(&buf->ref_list);
1374         }
1375 unlock:
1376         mutex_unlock(&inst->lock);
1377 }
1378
1379 void venus_helper_release_buf_ref(struct venus_inst *inst, unsigned int idx)
1380 {
1381         struct venus_buffer *buf;
1382
1383         list_for_each_entry(buf, &inst->registeredbufs, reg_list) {
1384                 if (buf->vb.vb2_buf.index == idx) {
1385                         buf->flags &= ~HFI_BUFFERFLAG_READONLY;
1386                         schedule_work(&inst->delayed_process_work);
1387                         break;
1388                 }
1389         }
1390 }
1391 EXPORT_SYMBOL_GPL(venus_helper_release_buf_ref);
1392
1393 void venus_helper_acquire_buf_ref(struct vb2_v4l2_buffer *vbuf)
1394 {
1395         struct venus_buffer *buf = to_venus_buffer(vbuf);
1396
1397         buf->flags |= HFI_BUFFERFLAG_READONLY;
1398 }
1399 EXPORT_SYMBOL_GPL(venus_helper_acquire_buf_ref);
1400
1401 static int is_buf_refed(struct venus_inst *inst, struct vb2_v4l2_buffer *vbuf)
1402 {
1403         struct venus_buffer *buf = to_venus_buffer(vbuf);
1404
1405         if (buf->flags & HFI_BUFFERFLAG_READONLY) {
1406                 list_add_tail(&buf->ref_list, &inst->delayed_process);
1407                 schedule_work(&inst->delayed_process_work);
1408                 return 1;
1409         }
1410
1411         return 0;
1412 }
1413
1414 struct vb2_v4l2_buffer *
1415 venus_helper_find_buf(struct venus_inst *inst, unsigned int type, u32 idx)
1416 {
1417         struct v4l2_m2m_ctx *m2m_ctx = inst->m2m_ctx;
1418
1419         if (type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE)
1420                 return v4l2_m2m_src_buf_remove_by_idx(m2m_ctx, idx);
1421         else
1422                 return v4l2_m2m_dst_buf_remove_by_idx(m2m_ctx, idx);
1423 }
1424 EXPORT_SYMBOL_GPL(venus_helper_find_buf);
1425
1426 void venus_helper_change_dpb_owner(struct venus_inst *inst,
1427                                    struct vb2_v4l2_buffer *vbuf, unsigned int type,
1428                                    unsigned int buf_type, u32 tag)
1429 {
1430         struct intbuf *dpb_buf;
1431
1432         if (!V4L2_TYPE_IS_CAPTURE(type) ||
1433             buf_type != inst->dpb_buftype)
1434                 return;
1435
1436         list_for_each_entry(dpb_buf, &inst->dpbbufs, list)
1437                 if (dpb_buf->dpb_out_tag == tag) {
1438                         dpb_buf->owned_by = DRIVER;
1439                         break;
1440                 }
1441 }
1442 EXPORT_SYMBOL_GPL(venus_helper_change_dpb_owner);
1443
1444 int venus_helper_vb2_buf_init(struct vb2_buffer *vb)
1445 {
1446         struct venus_inst *inst = vb2_get_drv_priv(vb->vb2_queue);
1447         struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
1448         struct venus_buffer *buf = to_venus_buffer(vbuf);
1449
1450         buf->size = vb2_plane_size(vb, 0);
1451         buf->dma_addr = vb2_dma_contig_plane_dma_addr(vb, 0);
1452
1453         if (vb->type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE)
1454                 list_add_tail(&buf->reg_list, &inst->registeredbufs);
1455
1456         return 0;
1457 }
1458 EXPORT_SYMBOL_GPL(venus_helper_vb2_buf_init);
1459
1460 int venus_helper_vb2_buf_prepare(struct vb2_buffer *vb)
1461 {
1462         struct venus_inst *inst = vb2_get_drv_priv(vb->vb2_queue);
1463         unsigned int out_buf_size = venus_helper_get_opb_size(inst);
1464         struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
1465
1466         if (V4L2_TYPE_IS_OUTPUT(vb->vb2_queue->type)) {
1467                 if (vbuf->field == V4L2_FIELD_ANY)
1468                         vbuf->field = V4L2_FIELD_NONE;
1469                 if (vbuf->field != V4L2_FIELD_NONE) {
1470                         dev_err(inst->core->dev, "%s field isn't supported\n",
1471                                 __func__);
1472                         return -EINVAL;
1473                 }
1474         }
1475
1476         if (vb->type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE &&
1477             vb2_plane_size(vb, 0) < out_buf_size)
1478                 return -EINVAL;
1479         if (vb->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE &&
1480             vb2_plane_size(vb, 0) < inst->input_buf_size)
1481                 return -EINVAL;
1482
1483         return 0;
1484 }
1485 EXPORT_SYMBOL_GPL(venus_helper_vb2_buf_prepare);
1486
1487 static void cache_payload(struct venus_inst *inst, struct vb2_buffer *vb)
1488 {
1489         struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
1490         unsigned int idx = vbuf->vb2_buf.index;
1491
1492         if (vbuf->vb2_buf.type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE)
1493                 inst->payloads[idx] = vb2_get_plane_payload(vb, 0);
1494 }
1495
1496 void venus_helper_vb2_buf_queue(struct vb2_buffer *vb)
1497 {
1498         struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
1499         struct venus_inst *inst = vb2_get_drv_priv(vb->vb2_queue);
1500         struct v4l2_m2m_ctx *m2m_ctx = inst->m2m_ctx;
1501         int ret;
1502
1503         v4l2_m2m_buf_queue(m2m_ctx, vbuf);
1504
1505         /* Skip processing queued capture buffers after LAST flag */
1506         if (inst->session_type == VIDC_SESSION_TYPE_DEC &&
1507             V4L2_TYPE_IS_CAPTURE(vb->vb2_queue->type) &&
1508             inst->codec_state == VENUS_DEC_STATE_DRC)
1509                 return;
1510
1511         cache_payload(inst, vb);
1512
1513         if (inst->session_type == VIDC_SESSION_TYPE_ENC &&
1514             !(inst->streamon_out && inst->streamon_cap))
1515                 return;
1516
1517         if (vb2_start_streaming_called(vb->vb2_queue)) {
1518                 ret = is_buf_refed(inst, vbuf);
1519                 if (ret)
1520                         return;
1521
1522                 ret = session_process_buf(inst, vbuf);
1523                 if (ret)
1524                         return_buf_error(inst, vbuf);
1525         }
1526 }
1527 EXPORT_SYMBOL_GPL(venus_helper_vb2_buf_queue);
1528
1529 void venus_helper_buffers_done(struct venus_inst *inst, unsigned int type,
1530                                enum vb2_buffer_state state)
1531 {
1532         struct vb2_v4l2_buffer *buf;
1533
1534         if (type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) {
1535                 while ((buf = v4l2_m2m_src_buf_remove(inst->m2m_ctx)))
1536                         v4l2_m2m_buf_done(buf, state);
1537         } else if (type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE) {
1538                 while ((buf = v4l2_m2m_dst_buf_remove(inst->m2m_ctx)))
1539                         v4l2_m2m_buf_done(buf, state);
1540         }
1541 }
1542 EXPORT_SYMBOL_GPL(venus_helper_buffers_done);
1543
1544 void venus_helper_vb2_stop_streaming(struct vb2_queue *q)
1545 {
1546         struct venus_inst *inst = vb2_get_drv_priv(q);
1547         struct venus_core *core = inst->core;
1548         int ret;
1549
1550         mutex_lock(&inst->lock);
1551
1552         if (inst->streamon_out & inst->streamon_cap) {
1553                 ret = hfi_session_stop(inst);
1554                 ret |= hfi_session_unload_res(inst);
1555                 ret |= venus_helper_unregister_bufs(inst);
1556                 ret |= venus_helper_intbufs_free(inst);
1557                 ret |= hfi_session_deinit(inst);
1558
1559                 if (inst->session_error || test_bit(0, &core->sys_error))
1560                         ret = -EIO;
1561
1562                 if (ret)
1563                         hfi_session_abort(inst);
1564
1565                 venus_helper_free_dpb_bufs(inst);
1566
1567                 venus_pm_load_scale(inst);
1568                 INIT_LIST_HEAD(&inst->registeredbufs);
1569         }
1570
1571         venus_helper_buffers_done(inst, V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE,
1572                                   VB2_BUF_STATE_ERROR);
1573         venus_helper_buffers_done(inst, V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE,
1574                                   VB2_BUF_STATE_ERROR);
1575
1576         if (q->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE)
1577                 inst->streamon_out = 0;
1578         else
1579                 inst->streamon_cap = 0;
1580
1581         venus_pm_release_core(inst);
1582
1583         inst->session_error = 0;
1584
1585         mutex_unlock(&inst->lock);
1586 }
1587 EXPORT_SYMBOL_GPL(venus_helper_vb2_stop_streaming);
1588
1589 void venus_helper_vb2_queue_error(struct venus_inst *inst)
1590 {
1591         struct v4l2_m2m_ctx *m2m_ctx = inst->m2m_ctx;
1592         struct vb2_queue *q;
1593
1594         q = v4l2_m2m_get_src_vq(m2m_ctx);
1595         vb2_queue_error(q);
1596         q = v4l2_m2m_get_dst_vq(m2m_ctx);
1597         vb2_queue_error(q);
1598 }
1599 EXPORT_SYMBOL_GPL(venus_helper_vb2_queue_error);
1600
1601 int venus_helper_process_initial_cap_bufs(struct venus_inst *inst)
1602 {
1603         struct v4l2_m2m_ctx *m2m_ctx = inst->m2m_ctx;
1604         struct v4l2_m2m_buffer *buf, *n;
1605         int ret;
1606
1607         v4l2_m2m_for_each_dst_buf_safe(m2m_ctx, buf, n) {
1608                 ret = session_process_buf(inst, &buf->vb);
1609                 if (ret) {
1610                         return_buf_error(inst, &buf->vb);
1611                         return ret;
1612                 }
1613         }
1614
1615         return 0;
1616 }
1617 EXPORT_SYMBOL_GPL(venus_helper_process_initial_cap_bufs);
1618
1619 int venus_helper_process_initial_out_bufs(struct venus_inst *inst)
1620 {
1621         struct v4l2_m2m_ctx *m2m_ctx = inst->m2m_ctx;
1622         struct v4l2_m2m_buffer *buf, *n;
1623         int ret;
1624
1625         v4l2_m2m_for_each_src_buf_safe(m2m_ctx, buf, n) {
1626                 ret = session_process_buf(inst, &buf->vb);
1627                 if (ret) {
1628                         return_buf_error(inst, &buf->vb);
1629                         return ret;
1630                 }
1631         }
1632
1633         return 0;
1634 }
1635 EXPORT_SYMBOL_GPL(venus_helper_process_initial_out_bufs);
1636
1637 int venus_helper_vb2_start_streaming(struct venus_inst *inst)
1638 {
1639         int ret;
1640
1641         ret = venus_helper_intbufs_alloc(inst);
1642         if (ret)
1643                 return ret;
1644
1645         ret = session_register_bufs(inst);
1646         if (ret)
1647                 goto err_bufs_free;
1648
1649         venus_pm_load_scale(inst);
1650
1651         ret = hfi_session_load_res(inst);
1652         if (ret)
1653                 goto err_unreg_bufs;
1654
1655         ret = hfi_session_start(inst);
1656         if (ret)
1657                 goto err_unload_res;
1658
1659         return 0;
1660
1661 err_unload_res:
1662         hfi_session_unload_res(inst);
1663 err_unreg_bufs:
1664         venus_helper_unregister_bufs(inst);
1665 err_bufs_free:
1666         venus_helper_intbufs_free(inst);
1667         return ret;
1668 }
1669 EXPORT_SYMBOL_GPL(venus_helper_vb2_start_streaming);
1670
1671 void venus_helper_m2m_device_run(void *priv)
1672 {
1673         struct venus_inst *inst = priv;
1674         struct v4l2_m2m_ctx *m2m_ctx = inst->m2m_ctx;
1675         struct v4l2_m2m_buffer *buf, *n;
1676         int ret;
1677
1678         mutex_lock(&inst->lock);
1679
1680         v4l2_m2m_for_each_dst_buf_safe(m2m_ctx, buf, n) {
1681                 ret = session_process_buf(inst, &buf->vb);
1682                 if (ret)
1683                         return_buf_error(inst, &buf->vb);
1684         }
1685
1686         v4l2_m2m_for_each_src_buf_safe(m2m_ctx, buf, n) {
1687                 ret = session_process_buf(inst, &buf->vb);
1688                 if (ret)
1689                         return_buf_error(inst, &buf->vb);
1690         }
1691
1692         mutex_unlock(&inst->lock);
1693 }
1694 EXPORT_SYMBOL_GPL(venus_helper_m2m_device_run);
1695
1696 void venus_helper_m2m_job_abort(void *priv)
1697 {
1698         struct venus_inst *inst = priv;
1699
1700         v4l2_m2m_job_finish(inst->m2m_dev, inst->m2m_ctx);
1701 }
1702 EXPORT_SYMBOL_GPL(venus_helper_m2m_job_abort);
1703
1704 int venus_helper_session_init(struct venus_inst *inst)
1705 {
1706         enum hfi_version version = inst->core->res->hfi_version;
1707         u32 session_type = inst->session_type;
1708         u32 codec;
1709         int ret;
1710
1711         codec = inst->session_type == VIDC_SESSION_TYPE_DEC ?
1712                         inst->fmt_out->pixfmt : inst->fmt_cap->pixfmt;
1713
1714         ret = hfi_session_init(inst, codec);
1715         if (ret)
1716                 return ret;
1717
1718         inst->clk_data.vpp_freq = hfi_platform_get_codec_vpp_freq(version, codec,
1719                                                                   session_type);
1720         inst->clk_data.vsp_freq = hfi_platform_get_codec_vsp_freq(version, codec,
1721                                                                   session_type);
1722         inst->clk_data.low_power_freq = hfi_platform_get_codec_lp_freq(version, codec,
1723                                                                        session_type);
1724
1725         return 0;
1726 }
1727 EXPORT_SYMBOL_GPL(venus_helper_session_init);
1728
1729 void venus_helper_init_instance(struct venus_inst *inst)
1730 {
1731         if (inst->session_type == VIDC_SESSION_TYPE_DEC) {
1732                 INIT_LIST_HEAD(&inst->delayed_process);
1733                 INIT_WORK(&inst->delayed_process_work,
1734                           delayed_process_buf_func);
1735         }
1736 }
1737 EXPORT_SYMBOL_GPL(venus_helper_init_instance);
1738
1739 static bool find_fmt_from_caps(struct hfi_plat_caps *caps, u32 buftype, u32 fmt)
1740 {
1741         unsigned int i;
1742
1743         for (i = 0; i < caps->num_fmts; i++) {
1744                 if (caps->fmts[i].buftype == buftype &&
1745                     caps->fmts[i].fmt == fmt)
1746                         return true;
1747         }
1748
1749         return false;
1750 }
1751
1752 int venus_helper_get_out_fmts(struct venus_inst *inst, u32 v4l2_fmt,
1753                               u32 *out_fmt, u32 *out2_fmt, bool ubwc)
1754 {
1755         struct venus_core *core = inst->core;
1756         struct hfi_plat_caps *caps;
1757         u32 ubwc_fmt, fmt = to_hfi_raw_fmt(v4l2_fmt);
1758         bool found, found_ubwc;
1759
1760         *out_fmt = *out2_fmt = 0;
1761
1762         if (!fmt)
1763                 return -EINVAL;
1764
1765         caps = venus_caps_by_codec(core, inst->hfi_codec, inst->session_type);
1766         if (!caps)
1767                 return -EINVAL;
1768
1769         if (inst->bit_depth == VIDC_BITDEPTH_10 && inst->session_type == VIDC_SESSION_TYPE_DEC) {
1770                 found_ubwc = find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT,
1771                                                 HFI_COLOR_FORMAT_YUV420_TP10_UBWC);
1772                 found = find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT2, fmt);
1773                 if (found_ubwc && found) {
1774                         /*
1775                          * Hard-code DPB buffers to be 10bit UBWC
1776                          * until V4L2 is able to expose compressed/tiled
1777                          * formats to applications.
1778                          */
1779                         *out_fmt = HFI_COLOR_FORMAT_YUV420_TP10_UBWC;
1780                         *out2_fmt = fmt;
1781                         return 0;
1782                 }
1783         }
1784
1785         if (ubwc) {
1786                 ubwc_fmt = fmt | HFI_COLOR_FORMAT_UBWC_BASE;
1787                 found_ubwc = find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT,
1788                                                 ubwc_fmt);
1789                 found = find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT2, fmt);
1790
1791                 if (found_ubwc && found) {
1792                         *out_fmt = ubwc_fmt;
1793                         *out2_fmt = fmt;
1794                         return 0;
1795                 }
1796         }
1797
1798         found = find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT, fmt);
1799         if (found) {
1800                 *out_fmt = fmt;
1801                 *out2_fmt = 0;
1802                 return 0;
1803         }
1804
1805         found = find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT2, fmt);
1806         if (found) {
1807                 *out_fmt = 0;
1808                 *out2_fmt = fmt;
1809                 return 0;
1810         }
1811
1812         return -EINVAL;
1813 }
1814 EXPORT_SYMBOL_GPL(venus_helper_get_out_fmts);
1815
1816 bool venus_helper_check_format(struct venus_inst *inst, u32 v4l2_pixfmt)
1817 {
1818         struct venus_core *core = inst->core;
1819         u32 fmt = to_hfi_raw_fmt(v4l2_pixfmt);
1820         struct hfi_plat_caps *caps;
1821         bool found;
1822
1823         if (!fmt)
1824                 return false;
1825
1826         caps = venus_caps_by_codec(core, inst->hfi_codec, inst->session_type);
1827         if (!caps)
1828                 return false;
1829
1830         found = find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT, fmt);
1831         if (found)
1832                 goto done;
1833
1834         found = find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT2, fmt);
1835 done:
1836         return found;
1837 }
1838 EXPORT_SYMBOL_GPL(venus_helper_check_format);
1839
1840 int venus_helper_set_stride(struct venus_inst *inst,
1841                             unsigned int width, unsigned int height)
1842 {
1843         const u32 ptype = HFI_PROPERTY_PARAM_UNCOMPRESSED_PLANE_ACTUAL_INFO;
1844
1845         struct hfi_uncompressed_plane_actual_info plane_actual_info;
1846
1847         plane_actual_info.buffer_type = HFI_BUFFER_INPUT;
1848         plane_actual_info.num_planes = 2;
1849         plane_actual_info.plane_format[0].actual_stride = width;
1850         plane_actual_info.plane_format[0].actual_plane_buffer_height = height;
1851         plane_actual_info.plane_format[1].actual_stride = width;
1852         plane_actual_info.plane_format[1].actual_plane_buffer_height = height / 2;
1853
1854         return hfi_session_set_property(inst, ptype, &plane_actual_info);
1855 }
1856 EXPORT_SYMBOL_GPL(venus_helper_set_stride);