1 /* 2 * Copyright (c) 2012-2016, The Linux Foundation. All rights reserved. 3 * Copyright (C) 2017 Linaro Ltd. 4 * 5 * This program is free software; you can redistribute it and/or modify 6 * it under the terms of the GNU General Public License version 2 and 7 * only version 2 as published by the Free Software Foundation. 8 * 9 * This program is distributed in the hope that it will be useful, 10 * but WITHOUT ANY WARRANTY; without even the implied warranty of 11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 12 * GNU General Public License for more details. 13 * 14 */ 15 #include <linux/clk.h> 16 #include <linux/iopoll.h> 17 #include <linux/list.h> 18 #include <linux/mutex.h> 19 #include <linux/pm_runtime.h> 20 #include <linux/slab.h> 21 #include <media/videobuf2-dma-sg.h> 22 #include <media/v4l2-mem2mem.h> 23 #include <asm/div64.h> 24 25 #include "core.h" 26 #include "helpers.h" 27 #include "hfi_helper.h" 28 #include "hfi_venus_io.h" 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 }; 38 39 bool venus_helper_check_codec(struct venus_inst *inst, u32 v4l2_pixfmt) 40 { 41 struct venus_core *core = inst->core; 42 u32 session_type = inst->session_type; 43 u32 codec; 44 45 switch (v4l2_pixfmt) { 46 case V4L2_PIX_FMT_H264: 47 codec = HFI_VIDEO_CODEC_H264; 48 break; 49 case V4L2_PIX_FMT_H263: 50 codec = HFI_VIDEO_CODEC_H263; 51 break; 52 case V4L2_PIX_FMT_MPEG1: 53 codec = HFI_VIDEO_CODEC_MPEG1; 54 break; 55 case V4L2_PIX_FMT_MPEG2: 56 codec = HFI_VIDEO_CODEC_MPEG2; 57 break; 58 case V4L2_PIX_FMT_MPEG4: 59 codec = HFI_VIDEO_CODEC_MPEG4; 60 break; 61 case V4L2_PIX_FMT_VC1_ANNEX_G: 62 case V4L2_PIX_FMT_VC1_ANNEX_L: 63 codec = HFI_VIDEO_CODEC_VC1; 64 break; 65 case V4L2_PIX_FMT_VP8: 66 codec = HFI_VIDEO_CODEC_VP8; 67 break; 68 case V4L2_PIX_FMT_VP9: 69 codec = HFI_VIDEO_CODEC_VP9; 70 break; 71 case V4L2_PIX_FMT_XVID: 72 codec = HFI_VIDEO_CODEC_DIVX; 73 break; 74 case V4L2_PIX_FMT_HEVC: 75 codec = HFI_VIDEO_CODEC_HEVC; 76 break; 77 default: 78 return false; 79 } 80 81 if (session_type == VIDC_SESSION_TYPE_ENC && core->enc_codecs & codec) 82 return true; 83 84 if (session_type == VIDC_SESSION_TYPE_DEC && core->dec_codecs & codec) 85 return true; 86 87 return false; 88 } 89 EXPORT_SYMBOL_GPL(venus_helper_check_codec); 90 91 static int venus_helper_queue_dpb_bufs(struct venus_inst *inst) 92 { 93 struct intbuf *buf; 94 int ret = 0; 95 96 list_for_each_entry(buf, &inst->dpbbufs, list) { 97 struct hfi_frame_data fdata; 98 99 memset(&fdata, 0, sizeof(fdata)); 100 fdata.alloc_len = buf->size; 101 fdata.device_addr = buf->da; 102 fdata.buffer_type = buf->type; 103 104 ret = hfi_session_process_buf(inst, &fdata); 105 if (ret) 106 goto fail; 107 } 108 109 fail: 110 return ret; 111 } 112 113 int venus_helper_free_dpb_bufs(struct venus_inst *inst) 114 { 115 struct intbuf *buf, *n; 116 117 list_for_each_entry_safe(buf, n, &inst->dpbbufs, list) { 118 list_del_init(&buf->list); 119 dma_free_attrs(inst->core->dev, buf->size, buf->va, buf->da, 120 buf->attrs); 121 kfree(buf); 122 } 123 124 INIT_LIST_HEAD(&inst->dpbbufs); 125 126 return 0; 127 } 128 EXPORT_SYMBOL_GPL(venus_helper_free_dpb_bufs); 129 130 int venus_helper_alloc_dpb_bufs(struct venus_inst *inst) 131 { 132 struct venus_core *core = inst->core; 133 struct device *dev = core->dev; 134 enum hfi_version ver = core->res->hfi_version; 135 struct hfi_buffer_requirements bufreq; 136 u32 buftype = inst->dpb_buftype; 137 unsigned int dpb_size = 0; 138 struct intbuf *buf; 139 unsigned int i; 140 u32 count; 141 int ret; 142 143 /* no need to allocate dpb buffers */ 144 if (!inst->dpb_fmt) 145 return 0; 146 147 if (inst->dpb_buftype == HFI_BUFFER_OUTPUT) 148 dpb_size = inst->output_buf_size; 149 else if (inst->dpb_buftype == HFI_BUFFER_OUTPUT2) 150 dpb_size = inst->output2_buf_size; 151 152 if (!dpb_size) 153 return 0; 154 155 ret = venus_helper_get_bufreq(inst, buftype, &bufreq); 156 if (ret) 157 return ret; 158 159 count = HFI_BUFREQ_COUNT_MIN(&bufreq, ver); 160 161 for (i = 0; i < count; i++) { 162 buf = kzalloc(sizeof(*buf), GFP_KERNEL); 163 if (!buf) { 164 ret = -ENOMEM; 165 goto fail; 166 } 167 168 buf->type = buftype; 169 buf->size = dpb_size; 170 buf->attrs = DMA_ATTR_WRITE_COMBINE | 171 DMA_ATTR_NO_KERNEL_MAPPING; 172 buf->va = dma_alloc_attrs(dev, buf->size, &buf->da, GFP_KERNEL, 173 buf->attrs); 174 if (!buf->va) { 175 kfree(buf); 176 ret = -ENOMEM; 177 goto fail; 178 } 179 180 list_add_tail(&buf->list, &inst->dpbbufs); 181 } 182 183 return 0; 184 185 fail: 186 venus_helper_free_dpb_bufs(inst); 187 return ret; 188 } 189 EXPORT_SYMBOL_GPL(venus_helper_alloc_dpb_bufs); 190 191 static int intbufs_set_buffer(struct venus_inst *inst, u32 type) 192 { 193 struct venus_core *core = inst->core; 194 struct device *dev = core->dev; 195 struct hfi_buffer_requirements bufreq; 196 struct hfi_buffer_desc bd; 197 struct intbuf *buf; 198 unsigned int i; 199 int ret; 200 201 ret = venus_helper_get_bufreq(inst, type, &bufreq); 202 if (ret) 203 return 0; 204 205 if (!bufreq.size) 206 return 0; 207 208 for (i = 0; i < bufreq.count_actual; i++) { 209 buf = kzalloc(sizeof(*buf), GFP_KERNEL); 210 if (!buf) { 211 ret = -ENOMEM; 212 goto fail; 213 } 214 215 buf->type = bufreq.type; 216 buf->size = bufreq.size; 217 buf->attrs = DMA_ATTR_WRITE_COMBINE | 218 DMA_ATTR_NO_KERNEL_MAPPING; 219 buf->va = dma_alloc_attrs(dev, buf->size, &buf->da, GFP_KERNEL, 220 buf->attrs); 221 if (!buf->va) { 222 ret = -ENOMEM; 223 goto fail; 224 } 225 226 memset(&bd, 0, sizeof(bd)); 227 bd.buffer_size = buf->size; 228 bd.buffer_type = buf->type; 229 bd.num_buffers = 1; 230 bd.device_addr = buf->da; 231 232 ret = hfi_session_set_buffers(inst, &bd); 233 if (ret) { 234 dev_err(dev, "set session buffers failed\n"); 235 goto dma_free; 236 } 237 238 list_add_tail(&buf->list, &inst->internalbufs); 239 } 240 241 return 0; 242 243 dma_free: 244 dma_free_attrs(dev, buf->size, buf->va, buf->da, buf->attrs); 245 fail: 246 kfree(buf); 247 return ret; 248 } 249 250 static int intbufs_unset_buffers(struct venus_inst *inst) 251 { 252 struct hfi_buffer_desc bd = {0}; 253 struct intbuf *buf, *n; 254 int ret = 0; 255 256 list_for_each_entry_safe(buf, n, &inst->internalbufs, list) { 257 bd.buffer_size = buf->size; 258 bd.buffer_type = buf->type; 259 bd.num_buffers = 1; 260 bd.device_addr = buf->da; 261 bd.response_required = true; 262 263 ret = hfi_session_unset_buffers(inst, &bd); 264 265 list_del_init(&buf->list); 266 dma_free_attrs(inst->core->dev, buf->size, buf->va, buf->da, 267 buf->attrs); 268 kfree(buf); 269 } 270 271 return ret; 272 } 273 274 static const unsigned int intbuf_types_1xx[] = { 275 HFI_BUFFER_INTERNAL_SCRATCH(HFI_VERSION_1XX), 276 HFI_BUFFER_INTERNAL_SCRATCH_1(HFI_VERSION_1XX), 277 HFI_BUFFER_INTERNAL_SCRATCH_2(HFI_VERSION_1XX), 278 HFI_BUFFER_INTERNAL_PERSIST, 279 HFI_BUFFER_INTERNAL_PERSIST_1, 280 }; 281 282 static const unsigned int intbuf_types_4xx[] = { 283 HFI_BUFFER_INTERNAL_SCRATCH(HFI_VERSION_4XX), 284 HFI_BUFFER_INTERNAL_SCRATCH_1(HFI_VERSION_4XX), 285 HFI_BUFFER_INTERNAL_SCRATCH_2(HFI_VERSION_4XX), 286 HFI_BUFFER_INTERNAL_PERSIST, 287 HFI_BUFFER_INTERNAL_PERSIST_1, 288 }; 289 290 static int intbufs_alloc(struct venus_inst *inst) 291 { 292 const unsigned int *intbuf; 293 size_t arr_sz, i; 294 int ret; 295 296 if (IS_V4(inst->core)) { 297 arr_sz = ARRAY_SIZE(intbuf_types_4xx); 298 intbuf = intbuf_types_4xx; 299 } else { 300 arr_sz = ARRAY_SIZE(intbuf_types_1xx); 301 intbuf = intbuf_types_1xx; 302 } 303 304 for (i = 0; i < arr_sz; i++) { 305 ret = intbufs_set_buffer(inst, intbuf[i]); 306 if (ret) 307 goto error; 308 } 309 310 return 0; 311 312 error: 313 intbufs_unset_buffers(inst); 314 return ret; 315 } 316 317 static int intbufs_free(struct venus_inst *inst) 318 { 319 return intbufs_unset_buffers(inst); 320 } 321 322 static u32 load_per_instance(struct venus_inst *inst) 323 { 324 u32 mbs; 325 326 if (!inst || !(inst->state >= INST_INIT && inst->state < INST_STOP)) 327 return 0; 328 329 mbs = (ALIGN(inst->width, 16) / 16) * (ALIGN(inst->height, 16) / 16); 330 331 return mbs * inst->fps; 332 } 333 334 static u32 load_per_type(struct venus_core *core, u32 session_type) 335 { 336 struct venus_inst *inst = NULL; 337 u32 mbs_per_sec = 0; 338 339 mutex_lock(&core->lock); 340 list_for_each_entry(inst, &core->instances, list) { 341 if (inst->session_type != session_type) 342 continue; 343 344 mbs_per_sec += load_per_instance(inst); 345 } 346 mutex_unlock(&core->lock); 347 348 return mbs_per_sec; 349 } 350 351 static int load_scale_clocks(struct venus_core *core) 352 { 353 const struct freq_tbl *table = core->res->freq_tbl; 354 unsigned int num_rows = core->res->freq_tbl_size; 355 unsigned long freq = table[0].freq; 356 struct clk *clk = core->clks[0]; 357 struct device *dev = core->dev; 358 u32 mbs_per_sec; 359 unsigned int i; 360 int ret; 361 362 mbs_per_sec = load_per_type(core, VIDC_SESSION_TYPE_ENC) + 363 load_per_type(core, VIDC_SESSION_TYPE_DEC); 364 365 if (mbs_per_sec > core->res->max_load) 366 dev_warn(dev, "HW is overloaded, needed: %d max: %d\n", 367 mbs_per_sec, core->res->max_load); 368 369 if (!mbs_per_sec && num_rows > 1) { 370 freq = table[num_rows - 1].freq; 371 goto set_freq; 372 } 373 374 for (i = 0; i < num_rows; i++) { 375 if (mbs_per_sec > table[i].load) 376 break; 377 freq = table[i].freq; 378 } 379 380 set_freq: 381 382 ret = clk_set_rate(clk, freq); 383 if (ret) 384 goto err; 385 386 ret = clk_set_rate(core->core0_clk, freq); 387 if (ret) 388 goto err; 389 390 ret = clk_set_rate(core->core1_clk, freq); 391 if (ret) 392 goto err; 393 394 return 0; 395 396 err: 397 dev_err(dev, "failed to set clock rate %lu (%d)\n", freq, ret); 398 return ret; 399 } 400 401 static void fill_buffer_desc(const struct venus_buffer *buf, 402 struct hfi_buffer_desc *bd, bool response) 403 { 404 memset(bd, 0, sizeof(*bd)); 405 bd->buffer_type = HFI_BUFFER_OUTPUT; 406 bd->buffer_size = buf->size; 407 bd->num_buffers = 1; 408 bd->device_addr = buf->dma_addr; 409 bd->response_required = response; 410 } 411 412 static void return_buf_error(struct venus_inst *inst, 413 struct vb2_v4l2_buffer *vbuf) 414 { 415 struct v4l2_m2m_ctx *m2m_ctx = inst->m2m_ctx; 416 417 if (vbuf->vb2_buf.type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) 418 v4l2_m2m_src_buf_remove_by_buf(m2m_ctx, vbuf); 419 else 420 v4l2_m2m_dst_buf_remove_by_buf(m2m_ctx, vbuf); 421 422 v4l2_m2m_buf_done(vbuf, VB2_BUF_STATE_ERROR); 423 } 424 425 static int 426 session_process_buf(struct venus_inst *inst, struct vb2_v4l2_buffer *vbuf) 427 { 428 struct venus_buffer *buf = to_venus_buffer(vbuf); 429 struct vb2_buffer *vb = &vbuf->vb2_buf; 430 unsigned int type = vb->type; 431 struct hfi_frame_data fdata; 432 int ret; 433 434 memset(&fdata, 0, sizeof(fdata)); 435 fdata.alloc_len = buf->size; 436 fdata.device_addr = buf->dma_addr; 437 fdata.timestamp = vb->timestamp; 438 do_div(fdata.timestamp, NSEC_PER_USEC); 439 fdata.flags = 0; 440 fdata.clnt_data = vbuf->vb2_buf.index; 441 442 if (!fdata.timestamp) 443 fdata.flags |= HFI_BUFFERFLAG_TIMESTAMPINVALID; 444 445 if (type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) { 446 fdata.buffer_type = HFI_BUFFER_INPUT; 447 fdata.filled_len = vb2_get_plane_payload(vb, 0); 448 fdata.offset = vb->planes[0].data_offset; 449 450 if (vbuf->flags & V4L2_BUF_FLAG_LAST || !fdata.filled_len) 451 fdata.flags |= HFI_BUFFERFLAG_EOS; 452 } else if (type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE) { 453 if (inst->session_type == VIDC_SESSION_TYPE_ENC) 454 fdata.buffer_type = HFI_BUFFER_OUTPUT; 455 else 456 fdata.buffer_type = inst->opb_buftype; 457 fdata.filled_len = 0; 458 fdata.offset = 0; 459 } 460 461 ret = hfi_session_process_buf(inst, &fdata); 462 if (ret) 463 return ret; 464 465 return 0; 466 } 467 468 static bool is_dynamic_bufmode(struct venus_inst *inst) 469 { 470 struct venus_core *core = inst->core; 471 struct venus_caps *caps; 472 473 caps = venus_caps_by_codec(core, inst->hfi_codec, inst->session_type); 474 if (!caps) 475 return false; 476 477 return caps->cap_bufs_mode_dynamic; 478 } 479 480 static int session_unregister_bufs(struct venus_inst *inst) 481 { 482 struct venus_buffer *buf, *n; 483 struct hfi_buffer_desc bd; 484 int ret = 0; 485 486 if (is_dynamic_bufmode(inst)) 487 return 0; 488 489 list_for_each_entry_safe(buf, n, &inst->registeredbufs, reg_list) { 490 fill_buffer_desc(buf, &bd, true); 491 ret = hfi_session_unset_buffers(inst, &bd); 492 list_del_init(&buf->reg_list); 493 } 494 495 return ret; 496 } 497 498 static int session_register_bufs(struct venus_inst *inst) 499 { 500 struct venus_core *core = inst->core; 501 struct device *dev = core->dev; 502 struct hfi_buffer_desc bd; 503 struct venus_buffer *buf; 504 int ret = 0; 505 506 if (is_dynamic_bufmode(inst)) 507 return 0; 508 509 list_for_each_entry(buf, &inst->registeredbufs, reg_list) { 510 fill_buffer_desc(buf, &bd, false); 511 ret = hfi_session_set_buffers(inst, &bd); 512 if (ret) { 513 dev_err(dev, "%s: set buffer failed\n", __func__); 514 break; 515 } 516 } 517 518 return ret; 519 } 520 521 static u32 to_hfi_raw_fmt(u32 v4l2_fmt) 522 { 523 switch (v4l2_fmt) { 524 case V4L2_PIX_FMT_NV12: 525 return HFI_COLOR_FORMAT_NV12; 526 case V4L2_PIX_FMT_NV21: 527 return HFI_COLOR_FORMAT_NV21; 528 default: 529 break; 530 } 531 532 return 0; 533 } 534 535 int venus_helper_get_bufreq(struct venus_inst *inst, u32 type, 536 struct hfi_buffer_requirements *req) 537 { 538 u32 ptype = HFI_PROPERTY_CONFIG_BUFFER_REQUIREMENTS; 539 union hfi_get_property hprop; 540 unsigned int i; 541 int ret; 542 543 if (req) 544 memset(req, 0, sizeof(*req)); 545 546 ret = hfi_session_get_property(inst, ptype, &hprop); 547 if (ret) 548 return ret; 549 550 ret = -EINVAL; 551 552 for (i = 0; i < HFI_BUFFER_TYPE_MAX; i++) { 553 if (hprop.bufreq[i].type != type) 554 continue; 555 556 if (req) 557 memcpy(req, &hprop.bufreq[i], sizeof(*req)); 558 ret = 0; 559 break; 560 } 561 562 return ret; 563 } 564 EXPORT_SYMBOL_GPL(venus_helper_get_bufreq); 565 566 static u32 get_framesize_raw_nv12(u32 width, u32 height) 567 { 568 u32 y_stride, uv_stride, y_plane; 569 u32 y_sclines, uv_sclines, uv_plane; 570 u32 size; 571 572 y_stride = ALIGN(width, 128); 573 uv_stride = ALIGN(width, 128); 574 y_sclines = ALIGN(height, 32); 575 uv_sclines = ALIGN(((height + 1) >> 1), 16); 576 577 y_plane = y_stride * y_sclines; 578 uv_plane = uv_stride * uv_sclines + SZ_4K; 579 size = y_plane + uv_plane + SZ_8K; 580 581 return ALIGN(size, SZ_4K); 582 } 583 584 static u32 get_framesize_raw_nv12_ubwc(u32 width, u32 height) 585 { 586 u32 y_meta_stride, y_meta_plane; 587 u32 y_stride, y_plane; 588 u32 uv_meta_stride, uv_meta_plane; 589 u32 uv_stride, uv_plane; 590 u32 extradata = SZ_16K; 591 592 y_meta_stride = ALIGN(DIV_ROUND_UP(width, 32), 64); 593 y_meta_plane = y_meta_stride * ALIGN(DIV_ROUND_UP(height, 8), 16); 594 y_meta_plane = ALIGN(y_meta_plane, SZ_4K); 595 596 y_stride = ALIGN(width, 128); 597 y_plane = ALIGN(y_stride * ALIGN(height, 32), SZ_4K); 598 599 uv_meta_stride = ALIGN(DIV_ROUND_UP(width / 2, 16), 64); 600 uv_meta_plane = uv_meta_stride * ALIGN(DIV_ROUND_UP(height / 2, 8), 16); 601 uv_meta_plane = ALIGN(uv_meta_plane, SZ_4K); 602 603 uv_stride = ALIGN(width, 128); 604 uv_plane = ALIGN(uv_stride * ALIGN(height / 2, 32), SZ_4K); 605 606 return ALIGN(y_meta_plane + y_plane + uv_meta_plane + uv_plane + 607 max(extradata, y_stride * 48), SZ_4K); 608 } 609 610 u32 venus_helper_get_framesz_raw(u32 hfi_fmt, u32 width, u32 height) 611 { 612 switch (hfi_fmt) { 613 case HFI_COLOR_FORMAT_NV12: 614 case HFI_COLOR_FORMAT_NV21: 615 return get_framesize_raw_nv12(width, height); 616 case HFI_COLOR_FORMAT_NV12_UBWC: 617 return get_framesize_raw_nv12_ubwc(width, height); 618 default: 619 return 0; 620 } 621 } 622 EXPORT_SYMBOL_GPL(venus_helper_get_framesz_raw); 623 624 u32 venus_helper_get_framesz(u32 v4l2_fmt, u32 width, u32 height) 625 { 626 u32 hfi_fmt, sz; 627 bool compressed; 628 629 switch (v4l2_fmt) { 630 case V4L2_PIX_FMT_MPEG: 631 case V4L2_PIX_FMT_H264: 632 case V4L2_PIX_FMT_H264_NO_SC: 633 case V4L2_PIX_FMT_H264_MVC: 634 case V4L2_PIX_FMT_H263: 635 case V4L2_PIX_FMT_MPEG1: 636 case V4L2_PIX_FMT_MPEG2: 637 case V4L2_PIX_FMT_MPEG4: 638 case V4L2_PIX_FMT_XVID: 639 case V4L2_PIX_FMT_VC1_ANNEX_G: 640 case V4L2_PIX_FMT_VC1_ANNEX_L: 641 case V4L2_PIX_FMT_VP8: 642 case V4L2_PIX_FMT_VP9: 643 case V4L2_PIX_FMT_HEVC: 644 compressed = true; 645 break; 646 default: 647 compressed = false; 648 break; 649 } 650 651 if (compressed) { 652 sz = ALIGN(height, 32) * ALIGN(width, 32) * 3 / 2 / 2; 653 return ALIGN(sz, SZ_4K); 654 } 655 656 hfi_fmt = to_hfi_raw_fmt(v4l2_fmt); 657 if (!hfi_fmt) 658 return 0; 659 660 return venus_helper_get_framesz_raw(hfi_fmt, width, height); 661 } 662 EXPORT_SYMBOL_GPL(venus_helper_get_framesz); 663 664 int venus_helper_set_input_resolution(struct venus_inst *inst, 665 unsigned int width, unsigned int height) 666 { 667 u32 ptype = HFI_PROPERTY_PARAM_FRAME_SIZE; 668 struct hfi_framesize fs; 669 670 fs.buffer_type = HFI_BUFFER_INPUT; 671 fs.width = width; 672 fs.height = height; 673 674 return hfi_session_set_property(inst, ptype, &fs); 675 } 676 EXPORT_SYMBOL_GPL(venus_helper_set_input_resolution); 677 678 int venus_helper_set_output_resolution(struct venus_inst *inst, 679 unsigned int width, unsigned int height, 680 u32 buftype) 681 { 682 u32 ptype = HFI_PROPERTY_PARAM_FRAME_SIZE; 683 struct hfi_framesize fs; 684 685 fs.buffer_type = buftype; 686 fs.width = width; 687 fs.height = height; 688 689 return hfi_session_set_property(inst, ptype, &fs); 690 } 691 EXPORT_SYMBOL_GPL(venus_helper_set_output_resolution); 692 693 int venus_helper_set_work_mode(struct venus_inst *inst, u32 mode) 694 { 695 const u32 ptype = HFI_PROPERTY_PARAM_WORK_MODE; 696 struct hfi_video_work_mode wm; 697 698 if (!IS_V4(inst->core)) 699 return 0; 700 701 wm.video_work_mode = mode; 702 703 return hfi_session_set_property(inst, ptype, &wm); 704 } 705 EXPORT_SYMBOL_GPL(venus_helper_set_work_mode); 706 707 int venus_helper_set_core_usage(struct venus_inst *inst, u32 usage) 708 { 709 const u32 ptype = HFI_PROPERTY_CONFIG_VIDEOCORES_USAGE; 710 struct hfi_videocores_usage_type cu; 711 712 if (!IS_V4(inst->core)) 713 return 0; 714 715 cu.video_core_enable_mask = usage; 716 717 return hfi_session_set_property(inst, ptype, &cu); 718 } 719 EXPORT_SYMBOL_GPL(venus_helper_set_core_usage); 720 721 int venus_helper_set_num_bufs(struct venus_inst *inst, unsigned int input_bufs, 722 unsigned int output_bufs, 723 unsigned int output2_bufs) 724 { 725 u32 ptype = HFI_PROPERTY_PARAM_BUFFER_COUNT_ACTUAL; 726 struct hfi_buffer_count_actual buf_count; 727 int ret; 728 729 buf_count.type = HFI_BUFFER_INPUT; 730 buf_count.count_actual = input_bufs; 731 732 ret = hfi_session_set_property(inst, ptype, &buf_count); 733 if (ret) 734 return ret; 735 736 buf_count.type = HFI_BUFFER_OUTPUT; 737 buf_count.count_actual = output_bufs; 738 739 ret = hfi_session_set_property(inst, ptype, &buf_count); 740 if (ret) 741 return ret; 742 743 if (output2_bufs) { 744 buf_count.type = HFI_BUFFER_OUTPUT2; 745 buf_count.count_actual = output2_bufs; 746 747 ret = hfi_session_set_property(inst, ptype, &buf_count); 748 } 749 750 return ret; 751 } 752 EXPORT_SYMBOL_GPL(venus_helper_set_num_bufs); 753 754 int venus_helper_set_raw_format(struct venus_inst *inst, u32 hfi_format, 755 u32 buftype) 756 { 757 const u32 ptype = HFI_PROPERTY_PARAM_UNCOMPRESSED_FORMAT_SELECT; 758 struct hfi_uncompressed_format_select fmt; 759 760 fmt.buffer_type = buftype; 761 fmt.format = hfi_format; 762 763 return hfi_session_set_property(inst, ptype, &fmt); 764 } 765 EXPORT_SYMBOL_GPL(venus_helper_set_raw_format); 766 767 int venus_helper_set_color_format(struct venus_inst *inst, u32 pixfmt) 768 { 769 u32 hfi_format, buftype; 770 771 if (inst->session_type == VIDC_SESSION_TYPE_DEC) 772 buftype = HFI_BUFFER_OUTPUT; 773 else if (inst->session_type == VIDC_SESSION_TYPE_ENC) 774 buftype = HFI_BUFFER_INPUT; 775 else 776 return -EINVAL; 777 778 hfi_format = to_hfi_raw_fmt(pixfmt); 779 if (!hfi_format) 780 return -EINVAL; 781 782 return venus_helper_set_raw_format(inst, hfi_format, buftype); 783 } 784 EXPORT_SYMBOL_GPL(venus_helper_set_color_format); 785 786 int venus_helper_set_multistream(struct venus_inst *inst, bool out_en, 787 bool out2_en) 788 { 789 struct hfi_multi_stream multi = {0}; 790 u32 ptype = HFI_PROPERTY_PARAM_VDEC_MULTI_STREAM; 791 int ret; 792 793 multi.buffer_type = HFI_BUFFER_OUTPUT; 794 multi.enable = out_en; 795 796 ret = hfi_session_set_property(inst, ptype, &multi); 797 if (ret) 798 return ret; 799 800 multi.buffer_type = HFI_BUFFER_OUTPUT2; 801 multi.enable = out2_en; 802 803 return hfi_session_set_property(inst, ptype, &multi); 804 } 805 EXPORT_SYMBOL_GPL(venus_helper_set_multistream); 806 807 int venus_helper_set_dyn_bufmode(struct venus_inst *inst) 808 { 809 const u32 ptype = HFI_PROPERTY_PARAM_BUFFER_ALLOC_MODE; 810 struct hfi_buffer_alloc_mode mode; 811 int ret; 812 813 if (!is_dynamic_bufmode(inst)) 814 return 0; 815 816 mode.type = HFI_BUFFER_OUTPUT; 817 mode.mode = HFI_BUFFER_MODE_DYNAMIC; 818 819 ret = hfi_session_set_property(inst, ptype, &mode); 820 if (ret) 821 return ret; 822 823 mode.type = HFI_BUFFER_OUTPUT2; 824 825 return hfi_session_set_property(inst, ptype, &mode); 826 } 827 EXPORT_SYMBOL_GPL(venus_helper_set_dyn_bufmode); 828 829 int venus_helper_set_bufsize(struct venus_inst *inst, u32 bufsize, u32 buftype) 830 { 831 const u32 ptype = HFI_PROPERTY_PARAM_BUFFER_SIZE_ACTUAL; 832 struct hfi_buffer_size_actual bufsz; 833 834 bufsz.type = buftype; 835 bufsz.size = bufsize; 836 837 return hfi_session_set_property(inst, ptype, &bufsz); 838 } 839 EXPORT_SYMBOL_GPL(venus_helper_set_bufsize); 840 841 unsigned int venus_helper_get_opb_size(struct venus_inst *inst) 842 { 843 /* the encoder has only one output */ 844 if (inst->session_type == VIDC_SESSION_TYPE_ENC) 845 return inst->output_buf_size; 846 847 if (inst->opb_buftype == HFI_BUFFER_OUTPUT) 848 return inst->output_buf_size; 849 else if (inst->opb_buftype == HFI_BUFFER_OUTPUT2) 850 return inst->output2_buf_size; 851 852 return 0; 853 } 854 EXPORT_SYMBOL_GPL(venus_helper_get_opb_size); 855 856 static void delayed_process_buf_func(struct work_struct *work) 857 { 858 struct venus_buffer *buf, *n; 859 struct venus_inst *inst; 860 int ret; 861 862 inst = container_of(work, struct venus_inst, delayed_process_work); 863 864 mutex_lock(&inst->lock); 865 866 if (!(inst->streamon_out & inst->streamon_cap)) 867 goto unlock; 868 869 list_for_each_entry_safe(buf, n, &inst->delayed_process, ref_list) { 870 if (buf->flags & HFI_BUFFERFLAG_READONLY) 871 continue; 872 873 ret = session_process_buf(inst, &buf->vb); 874 if (ret) 875 return_buf_error(inst, &buf->vb); 876 877 list_del_init(&buf->ref_list); 878 } 879 unlock: 880 mutex_unlock(&inst->lock); 881 } 882 883 void venus_helper_release_buf_ref(struct venus_inst *inst, unsigned int idx) 884 { 885 struct venus_buffer *buf; 886 887 list_for_each_entry(buf, &inst->registeredbufs, reg_list) { 888 if (buf->vb.vb2_buf.index == idx) { 889 buf->flags &= ~HFI_BUFFERFLAG_READONLY; 890 schedule_work(&inst->delayed_process_work); 891 break; 892 } 893 } 894 } 895 EXPORT_SYMBOL_GPL(venus_helper_release_buf_ref); 896 897 void venus_helper_acquire_buf_ref(struct vb2_v4l2_buffer *vbuf) 898 { 899 struct venus_buffer *buf = to_venus_buffer(vbuf); 900 901 buf->flags |= HFI_BUFFERFLAG_READONLY; 902 } 903 EXPORT_SYMBOL_GPL(venus_helper_acquire_buf_ref); 904 905 static int is_buf_refed(struct venus_inst *inst, struct vb2_v4l2_buffer *vbuf) 906 { 907 struct venus_buffer *buf = to_venus_buffer(vbuf); 908 909 if (buf->flags & HFI_BUFFERFLAG_READONLY) { 910 list_add_tail(&buf->ref_list, &inst->delayed_process); 911 schedule_work(&inst->delayed_process_work); 912 return 1; 913 } 914 915 return 0; 916 } 917 918 struct vb2_v4l2_buffer * 919 venus_helper_find_buf(struct venus_inst *inst, unsigned int type, u32 idx) 920 { 921 struct v4l2_m2m_ctx *m2m_ctx = inst->m2m_ctx; 922 923 if (type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) 924 return v4l2_m2m_src_buf_remove_by_idx(m2m_ctx, idx); 925 else 926 return v4l2_m2m_dst_buf_remove_by_idx(m2m_ctx, idx); 927 } 928 EXPORT_SYMBOL_GPL(venus_helper_find_buf); 929 930 int venus_helper_vb2_buf_init(struct vb2_buffer *vb) 931 { 932 struct venus_inst *inst = vb2_get_drv_priv(vb->vb2_queue); 933 struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb); 934 struct venus_buffer *buf = to_venus_buffer(vbuf); 935 struct sg_table *sgt; 936 937 sgt = vb2_dma_sg_plane_desc(vb, 0); 938 if (!sgt) 939 return -EFAULT; 940 941 buf->size = vb2_plane_size(vb, 0); 942 buf->dma_addr = sg_dma_address(sgt->sgl); 943 944 if (vb->type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE) 945 list_add_tail(&buf->reg_list, &inst->registeredbufs); 946 947 return 0; 948 } 949 EXPORT_SYMBOL_GPL(venus_helper_vb2_buf_init); 950 951 int venus_helper_vb2_buf_prepare(struct vb2_buffer *vb) 952 { 953 struct venus_inst *inst = vb2_get_drv_priv(vb->vb2_queue); 954 unsigned int out_buf_size = venus_helper_get_opb_size(inst); 955 956 if (vb->type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE && 957 vb2_plane_size(vb, 0) < out_buf_size) 958 return -EINVAL; 959 if (vb->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE && 960 vb2_plane_size(vb, 0) < inst->input_buf_size) 961 return -EINVAL; 962 963 return 0; 964 } 965 EXPORT_SYMBOL_GPL(venus_helper_vb2_buf_prepare); 966 967 void venus_helper_vb2_buf_queue(struct vb2_buffer *vb) 968 { 969 struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb); 970 struct venus_inst *inst = vb2_get_drv_priv(vb->vb2_queue); 971 struct v4l2_m2m_ctx *m2m_ctx = inst->m2m_ctx; 972 int ret; 973 974 mutex_lock(&inst->lock); 975 976 v4l2_m2m_buf_queue(m2m_ctx, vbuf); 977 978 if (!(inst->streamon_out & inst->streamon_cap)) 979 goto unlock; 980 981 ret = is_buf_refed(inst, vbuf); 982 if (ret) 983 goto unlock; 984 985 ret = session_process_buf(inst, vbuf); 986 if (ret) 987 return_buf_error(inst, vbuf); 988 989 unlock: 990 mutex_unlock(&inst->lock); 991 } 992 EXPORT_SYMBOL_GPL(venus_helper_vb2_buf_queue); 993 994 void venus_helper_buffers_done(struct venus_inst *inst, 995 enum vb2_buffer_state state) 996 { 997 struct vb2_v4l2_buffer *buf; 998 999 while ((buf = v4l2_m2m_src_buf_remove(inst->m2m_ctx))) 1000 v4l2_m2m_buf_done(buf, state); 1001 while ((buf = v4l2_m2m_dst_buf_remove(inst->m2m_ctx))) 1002 v4l2_m2m_buf_done(buf, state); 1003 } 1004 EXPORT_SYMBOL_GPL(venus_helper_buffers_done); 1005 1006 void venus_helper_vb2_stop_streaming(struct vb2_queue *q) 1007 { 1008 struct venus_inst *inst = vb2_get_drv_priv(q); 1009 struct venus_core *core = inst->core; 1010 int ret; 1011 1012 mutex_lock(&inst->lock); 1013 1014 if (inst->streamon_out & inst->streamon_cap) { 1015 ret = hfi_session_stop(inst); 1016 ret |= hfi_session_unload_res(inst); 1017 ret |= session_unregister_bufs(inst); 1018 ret |= intbufs_free(inst); 1019 ret |= hfi_session_deinit(inst); 1020 1021 if (inst->session_error || core->sys_error) 1022 ret = -EIO; 1023 1024 if (ret) 1025 hfi_session_abort(inst); 1026 1027 venus_helper_free_dpb_bufs(inst); 1028 1029 load_scale_clocks(core); 1030 INIT_LIST_HEAD(&inst->registeredbufs); 1031 } 1032 1033 venus_helper_buffers_done(inst, VB2_BUF_STATE_ERROR); 1034 1035 if (q->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) 1036 inst->streamon_out = 0; 1037 else 1038 inst->streamon_cap = 0; 1039 1040 mutex_unlock(&inst->lock); 1041 } 1042 EXPORT_SYMBOL_GPL(venus_helper_vb2_stop_streaming); 1043 1044 int venus_helper_vb2_start_streaming(struct venus_inst *inst) 1045 { 1046 struct venus_core *core = inst->core; 1047 int ret; 1048 1049 ret = intbufs_alloc(inst); 1050 if (ret) 1051 return ret; 1052 1053 ret = session_register_bufs(inst); 1054 if (ret) 1055 goto err_bufs_free; 1056 1057 load_scale_clocks(core); 1058 1059 ret = hfi_session_load_res(inst); 1060 if (ret) 1061 goto err_unreg_bufs; 1062 1063 ret = hfi_session_start(inst); 1064 if (ret) 1065 goto err_unload_res; 1066 1067 ret = venus_helper_queue_dpb_bufs(inst); 1068 if (ret) 1069 goto err_session_stop; 1070 1071 return 0; 1072 1073 err_session_stop: 1074 hfi_session_stop(inst); 1075 err_unload_res: 1076 hfi_session_unload_res(inst); 1077 err_unreg_bufs: 1078 session_unregister_bufs(inst); 1079 err_bufs_free: 1080 intbufs_free(inst); 1081 return ret; 1082 } 1083 EXPORT_SYMBOL_GPL(venus_helper_vb2_start_streaming); 1084 1085 void venus_helper_m2m_device_run(void *priv) 1086 { 1087 struct venus_inst *inst = priv; 1088 struct v4l2_m2m_ctx *m2m_ctx = inst->m2m_ctx; 1089 struct v4l2_m2m_buffer *buf, *n; 1090 int ret; 1091 1092 mutex_lock(&inst->lock); 1093 1094 v4l2_m2m_for_each_dst_buf_safe(m2m_ctx, buf, n) { 1095 ret = session_process_buf(inst, &buf->vb); 1096 if (ret) 1097 return_buf_error(inst, &buf->vb); 1098 } 1099 1100 v4l2_m2m_for_each_src_buf_safe(m2m_ctx, buf, n) { 1101 ret = session_process_buf(inst, &buf->vb); 1102 if (ret) 1103 return_buf_error(inst, &buf->vb); 1104 } 1105 1106 mutex_unlock(&inst->lock); 1107 } 1108 EXPORT_SYMBOL_GPL(venus_helper_m2m_device_run); 1109 1110 void venus_helper_m2m_job_abort(void *priv) 1111 { 1112 struct venus_inst *inst = priv; 1113 1114 v4l2_m2m_job_finish(inst->m2m_dev, inst->m2m_ctx); 1115 } 1116 EXPORT_SYMBOL_GPL(venus_helper_m2m_job_abort); 1117 1118 void venus_helper_init_instance(struct venus_inst *inst) 1119 { 1120 if (inst->session_type == VIDC_SESSION_TYPE_DEC) { 1121 INIT_LIST_HEAD(&inst->delayed_process); 1122 INIT_WORK(&inst->delayed_process_work, 1123 delayed_process_buf_func); 1124 } 1125 } 1126 EXPORT_SYMBOL_GPL(venus_helper_init_instance); 1127 1128 static bool find_fmt_from_caps(struct venus_caps *caps, u32 buftype, u32 fmt) 1129 { 1130 unsigned int i; 1131 1132 for (i = 0; i < caps->num_fmts; i++) { 1133 if (caps->fmts[i].buftype == buftype && 1134 caps->fmts[i].fmt == fmt) 1135 return true; 1136 } 1137 1138 return false; 1139 } 1140 1141 int venus_helper_get_out_fmts(struct venus_inst *inst, u32 v4l2_fmt, 1142 u32 *out_fmt, u32 *out2_fmt, bool ubwc) 1143 { 1144 struct venus_core *core = inst->core; 1145 struct venus_caps *caps; 1146 u32 ubwc_fmt, fmt = to_hfi_raw_fmt(v4l2_fmt); 1147 bool found, found_ubwc; 1148 1149 *out_fmt = *out2_fmt = 0; 1150 1151 if (!fmt) 1152 return -EINVAL; 1153 1154 caps = venus_caps_by_codec(core, inst->hfi_codec, inst->session_type); 1155 if (!caps) 1156 return -EINVAL; 1157 1158 if (ubwc) { 1159 ubwc_fmt = fmt | HFI_COLOR_FORMAT_UBWC_BASE; 1160 found_ubwc = find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT, 1161 ubwc_fmt); 1162 found = find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT2, fmt); 1163 1164 if (found_ubwc && found) { 1165 *out_fmt = ubwc_fmt; 1166 *out2_fmt = fmt; 1167 return 0; 1168 } 1169 } 1170 1171 found = find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT, fmt); 1172 if (found) { 1173 *out_fmt = fmt; 1174 *out2_fmt = 0; 1175 return 0; 1176 } 1177 1178 found = find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT2, fmt); 1179 if (found) { 1180 *out_fmt = 0; 1181 *out2_fmt = fmt; 1182 return 0; 1183 } 1184 1185 return -EINVAL; 1186 } 1187 EXPORT_SYMBOL_GPL(venus_helper_get_out_fmts); 1188 1189 int venus_helper_power_enable(struct venus_core *core, u32 session_type, 1190 bool enable) 1191 { 1192 void __iomem *ctrl, *stat; 1193 u32 val; 1194 int ret; 1195 1196 if (!IS_V3(core) && !IS_V4(core)) 1197 return 0; 1198 1199 if (IS_V3(core)) { 1200 if (session_type == VIDC_SESSION_TYPE_DEC) 1201 ctrl = core->base + WRAPPER_VDEC_VCODEC_POWER_CONTROL; 1202 else 1203 ctrl = core->base + WRAPPER_VENC_VCODEC_POWER_CONTROL; 1204 if (enable) 1205 writel(0, ctrl); 1206 else 1207 writel(1, ctrl); 1208 1209 return 0; 1210 } 1211 1212 if (session_type == VIDC_SESSION_TYPE_DEC) { 1213 ctrl = core->base + WRAPPER_VCODEC0_MMCC_POWER_CONTROL; 1214 stat = core->base + WRAPPER_VCODEC0_MMCC_POWER_STATUS; 1215 } else { 1216 ctrl = core->base + WRAPPER_VCODEC1_MMCC_POWER_CONTROL; 1217 stat = core->base + WRAPPER_VCODEC1_MMCC_POWER_STATUS; 1218 } 1219 1220 if (enable) { 1221 writel(0, ctrl); 1222 1223 ret = readl_poll_timeout(stat, val, val & BIT(1), 1, 100); 1224 if (ret) 1225 return ret; 1226 } else { 1227 writel(1, ctrl); 1228 1229 ret = readl_poll_timeout(stat, val, !(val & BIT(1)), 1, 100); 1230 if (ret) 1231 return ret; 1232 } 1233 1234 return 0; 1235 } 1236 EXPORT_SYMBOL_GPL(venus_helper_power_enable); 1237