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/clk.h> 7 #include <linux/module.h> 8 #include <linux/mod_devicetable.h> 9 #include <linux/platform_device.h> 10 #include <linux/pm_runtime.h> 11 #include <linux/slab.h> 12 #include <media/v4l2-mem2mem.h> 13 #include <media/videobuf2-dma-contig.h> 14 #include <media/v4l2-ioctl.h> 15 #include <media/v4l2-event.h> 16 #include <media/v4l2-ctrls.h> 17 18 #include "hfi_venus_io.h" 19 #include "hfi_parser.h" 20 #include "core.h" 21 #include "helpers.h" 22 #include "venc.h" 23 #include "pm_helpers.h" 24 25 #define NUM_B_FRAMES_MAX 4 26 27 /* 28 * Three resons to keep MPLANE formats (despite that the number of planes 29 * currently is one): 30 * - the MPLANE formats allow only one plane to be used 31 * - the downstream driver use MPLANE formats too 32 * - future firmware versions could add support for >1 planes 33 */ 34 static const struct venus_format venc_formats[] = { 35 [VENUS_FMT_NV12] = { 36 .pixfmt = V4L2_PIX_FMT_NV12, 37 .num_planes = 1, 38 .type = V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE, 39 }, 40 [VENUS_FMT_H264] = { 41 .pixfmt = V4L2_PIX_FMT_H264, 42 .num_planes = 1, 43 .type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE, 44 }, 45 [VENUS_FMT_VP8] = { 46 .pixfmt = V4L2_PIX_FMT_VP8, 47 .num_planes = 1, 48 .type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE, 49 }, 50 [VENUS_FMT_HEVC] = { 51 .pixfmt = V4L2_PIX_FMT_HEVC, 52 .num_planes = 1, 53 .type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE, 54 }, 55 [VENUS_FMT_MPEG4] = { 56 .pixfmt = V4L2_PIX_FMT_MPEG4, 57 .num_planes = 1, 58 .type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE, 59 }, 60 [VENUS_FMT_H263] = { 61 .pixfmt = V4L2_PIX_FMT_H263, 62 .num_planes = 1, 63 .type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE, 64 }, 65 }; 66 67 static const struct venus_format * 68 find_format(struct venus_inst *inst, u32 pixfmt, u32 type) 69 { 70 const struct venus_format *fmt = venc_formats; 71 unsigned int size = ARRAY_SIZE(venc_formats); 72 unsigned int i; 73 74 for (i = 0; i < size; i++) { 75 if (fmt[i].pixfmt == pixfmt) 76 break; 77 } 78 79 if (i == size || fmt[i].type != type) 80 return NULL; 81 82 if (type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE && 83 !venus_helper_check_codec(inst, fmt[i].pixfmt)) 84 return NULL; 85 86 return &fmt[i]; 87 } 88 89 static const struct venus_format * 90 find_format_by_index(struct venus_inst *inst, unsigned int index, u32 type) 91 { 92 const struct venus_format *fmt = venc_formats; 93 unsigned int size = ARRAY_SIZE(venc_formats); 94 unsigned int i, k = 0; 95 96 if (index > size) 97 return NULL; 98 99 for (i = 0; i < size; i++) { 100 bool valid; 101 102 if (fmt[i].type != type) 103 continue; 104 valid = type != V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE || 105 venus_helper_check_codec(inst, fmt[i].pixfmt); 106 if (k == index && valid) 107 break; 108 if (valid) 109 k++; 110 } 111 112 if (i == size) 113 return NULL; 114 115 return &fmt[i]; 116 } 117 118 static int venc_v4l2_to_hfi(int id, int value) 119 { 120 switch (id) { 121 case V4L2_CID_MPEG_VIDEO_H264_ENTROPY_MODE: 122 switch (value) { 123 case V4L2_MPEG_VIDEO_H264_ENTROPY_MODE_CAVLC: 124 default: 125 return HFI_H264_ENTROPY_CAVLC; 126 case V4L2_MPEG_VIDEO_H264_ENTROPY_MODE_CABAC: 127 return HFI_H264_ENTROPY_CABAC; 128 } 129 case V4L2_CID_MPEG_VIDEO_H264_LOOP_FILTER_MODE: 130 switch (value) { 131 case V4L2_MPEG_VIDEO_H264_LOOP_FILTER_MODE_ENABLED: 132 default: 133 return HFI_H264_DB_MODE_ALL_BOUNDARY; 134 case V4L2_MPEG_VIDEO_H264_LOOP_FILTER_MODE_DISABLED: 135 return HFI_H264_DB_MODE_DISABLE; 136 case V4L2_MPEG_VIDEO_H264_LOOP_FILTER_MODE_DISABLED_AT_SLICE_BOUNDARY: 137 return HFI_H264_DB_MODE_SKIP_SLICE_BOUNDARY; 138 } 139 } 140 141 return 0; 142 } 143 144 static int 145 venc_querycap(struct file *file, void *fh, struct v4l2_capability *cap) 146 { 147 strscpy(cap->driver, "qcom-venus", sizeof(cap->driver)); 148 strscpy(cap->card, "Qualcomm Venus video encoder", sizeof(cap->card)); 149 strscpy(cap->bus_info, "platform:qcom-venus", sizeof(cap->bus_info)); 150 151 return 0; 152 } 153 154 static int venc_enum_fmt(struct file *file, void *fh, struct v4l2_fmtdesc *f) 155 { 156 struct venus_inst *inst = to_inst(file); 157 const struct venus_format *fmt; 158 159 fmt = find_format_by_index(inst, f->index, f->type); 160 161 memset(f->reserved, 0, sizeof(f->reserved)); 162 163 if (!fmt) 164 return -EINVAL; 165 166 f->pixelformat = fmt->pixfmt; 167 168 return 0; 169 } 170 171 static const struct venus_format * 172 venc_try_fmt_common(struct venus_inst *inst, struct v4l2_format *f) 173 { 174 struct v4l2_pix_format_mplane *pixmp = &f->fmt.pix_mp; 175 struct v4l2_plane_pix_format *pfmt = pixmp->plane_fmt; 176 const struct venus_format *fmt; 177 u32 sizeimage; 178 179 memset(pfmt[0].reserved, 0, sizeof(pfmt[0].reserved)); 180 memset(pixmp->reserved, 0, sizeof(pixmp->reserved)); 181 182 fmt = find_format(inst, pixmp->pixelformat, f->type); 183 if (!fmt) { 184 if (f->type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE) 185 pixmp->pixelformat = V4L2_PIX_FMT_H264; 186 else if (f->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) 187 pixmp->pixelformat = V4L2_PIX_FMT_NV12; 188 else 189 return NULL; 190 fmt = find_format(inst, pixmp->pixelformat, f->type); 191 if (!fmt) 192 return NULL; 193 } 194 195 pixmp->width = clamp(pixmp->width, frame_width_min(inst), 196 frame_width_max(inst)); 197 pixmp->height = clamp(pixmp->height, frame_height_min(inst), 198 frame_height_max(inst)); 199 200 pixmp->width = ALIGN(pixmp->width, 128); 201 pixmp->height = ALIGN(pixmp->height, 32); 202 203 pixmp->width = ALIGN(pixmp->width, 2); 204 pixmp->height = ALIGN(pixmp->height, 2); 205 206 if (pixmp->field == V4L2_FIELD_ANY) 207 pixmp->field = V4L2_FIELD_NONE; 208 pixmp->num_planes = fmt->num_planes; 209 pixmp->flags = 0; 210 211 sizeimage = venus_helper_get_framesz(pixmp->pixelformat, 212 pixmp->width, 213 pixmp->height); 214 pfmt[0].sizeimage = max(ALIGN(pfmt[0].sizeimage, SZ_4K), sizeimage); 215 216 if (f->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) 217 pfmt[0].bytesperline = ALIGN(pixmp->width, 128); 218 else 219 pfmt[0].bytesperline = 0; 220 221 return fmt; 222 } 223 224 static int venc_try_fmt(struct file *file, void *fh, struct v4l2_format *f) 225 { 226 struct venus_inst *inst = to_inst(file); 227 228 venc_try_fmt_common(inst, f); 229 230 return 0; 231 } 232 233 static int venc_s_fmt(struct file *file, void *fh, struct v4l2_format *f) 234 { 235 struct venus_inst *inst = to_inst(file); 236 struct v4l2_pix_format_mplane *pixmp = &f->fmt.pix_mp; 237 struct v4l2_pix_format_mplane orig_pixmp; 238 const struct venus_format *fmt; 239 struct v4l2_format format; 240 u32 pixfmt_out = 0, pixfmt_cap = 0; 241 struct vb2_queue *q; 242 243 q = v4l2_m2m_get_vq(inst->m2m_ctx, f->type); 244 if (!q) 245 return -EINVAL; 246 247 if (vb2_is_busy(q)) 248 return -EBUSY; 249 250 orig_pixmp = *pixmp; 251 252 fmt = venc_try_fmt_common(inst, f); 253 if (!fmt) 254 return -EINVAL; 255 256 if (f->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) { 257 pixfmt_out = pixmp->pixelformat; 258 pixfmt_cap = inst->fmt_cap->pixfmt; 259 } else if (f->type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE) { 260 pixfmt_cap = pixmp->pixelformat; 261 pixfmt_out = inst->fmt_out->pixfmt; 262 } 263 264 memset(&format, 0, sizeof(format)); 265 266 format.type = V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE; 267 format.fmt.pix_mp.pixelformat = pixfmt_out; 268 format.fmt.pix_mp.width = orig_pixmp.width; 269 format.fmt.pix_mp.height = orig_pixmp.height; 270 venc_try_fmt_common(inst, &format); 271 272 if (f->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) { 273 inst->out_width = format.fmt.pix_mp.width; 274 inst->out_height = format.fmt.pix_mp.height; 275 inst->colorspace = pixmp->colorspace; 276 inst->ycbcr_enc = pixmp->ycbcr_enc; 277 inst->quantization = pixmp->quantization; 278 inst->xfer_func = pixmp->xfer_func; 279 } 280 281 memset(&format, 0, sizeof(format)); 282 283 format.type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE; 284 format.fmt.pix_mp.pixelformat = pixfmt_cap; 285 format.fmt.pix_mp.width = orig_pixmp.width; 286 format.fmt.pix_mp.height = orig_pixmp.height; 287 venc_try_fmt_common(inst, &format); 288 289 inst->width = format.fmt.pix_mp.width; 290 inst->height = format.fmt.pix_mp.height; 291 292 if (f->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) 293 inst->fmt_out = fmt; 294 else if (f->type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE) { 295 inst->fmt_cap = fmt; 296 inst->output_buf_size = pixmp->plane_fmt[0].sizeimage; 297 } 298 299 return 0; 300 } 301 302 static int venc_g_fmt(struct file *file, void *fh, struct v4l2_format *f) 303 { 304 struct v4l2_pix_format_mplane *pixmp = &f->fmt.pix_mp; 305 struct venus_inst *inst = to_inst(file); 306 const struct venus_format *fmt; 307 308 if (f->type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE) 309 fmt = inst->fmt_cap; 310 else if (f->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) 311 fmt = inst->fmt_out; 312 else 313 return -EINVAL; 314 315 pixmp->pixelformat = fmt->pixfmt; 316 317 if (f->type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE) { 318 pixmp->width = inst->width; 319 pixmp->height = inst->height; 320 pixmp->colorspace = inst->colorspace; 321 pixmp->ycbcr_enc = inst->ycbcr_enc; 322 pixmp->quantization = inst->quantization; 323 pixmp->xfer_func = inst->xfer_func; 324 } else if (f->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) { 325 pixmp->width = inst->out_width; 326 pixmp->height = inst->out_height; 327 } 328 329 venc_try_fmt_common(inst, f); 330 331 return 0; 332 } 333 334 static int 335 venc_g_selection(struct file *file, void *fh, struct v4l2_selection *s) 336 { 337 struct venus_inst *inst = to_inst(file); 338 339 if (s->type != V4L2_BUF_TYPE_VIDEO_OUTPUT) 340 return -EINVAL; 341 342 switch (s->target) { 343 case V4L2_SEL_TGT_CROP_DEFAULT: 344 case V4L2_SEL_TGT_CROP_BOUNDS: 345 s->r.width = inst->out_width; 346 s->r.height = inst->out_height; 347 break; 348 case V4L2_SEL_TGT_CROP: 349 s->r.width = inst->width; 350 s->r.height = inst->height; 351 break; 352 default: 353 return -EINVAL; 354 } 355 356 s->r.top = 0; 357 s->r.left = 0; 358 359 return 0; 360 } 361 362 static int 363 venc_s_selection(struct file *file, void *fh, struct v4l2_selection *s) 364 { 365 struct venus_inst *inst = to_inst(file); 366 367 if (s->type != V4L2_BUF_TYPE_VIDEO_OUTPUT) 368 return -EINVAL; 369 370 if (s->r.width > inst->out_width || 371 s->r.height > inst->out_height) 372 return -EINVAL; 373 374 s->r.width = ALIGN(s->r.width, 2); 375 s->r.height = ALIGN(s->r.height, 2); 376 377 switch (s->target) { 378 case V4L2_SEL_TGT_CROP: 379 s->r.top = 0; 380 s->r.left = 0; 381 inst->width = s->r.width; 382 inst->height = s->r.height; 383 break; 384 default: 385 return -EINVAL; 386 } 387 388 return 0; 389 } 390 391 static int venc_s_parm(struct file *file, void *fh, struct v4l2_streamparm *a) 392 { 393 struct venus_inst *inst = to_inst(file); 394 struct v4l2_outputparm *out = &a->parm.output; 395 struct v4l2_fract *timeperframe = &out->timeperframe; 396 u64 us_per_frame, fps; 397 398 if (a->type != V4L2_BUF_TYPE_VIDEO_OUTPUT && 399 a->type != V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) 400 return -EINVAL; 401 402 memset(out->reserved, 0, sizeof(out->reserved)); 403 404 if (!timeperframe->denominator) 405 timeperframe->denominator = inst->timeperframe.denominator; 406 if (!timeperframe->numerator) 407 timeperframe->numerator = inst->timeperframe.numerator; 408 409 out->capability = V4L2_CAP_TIMEPERFRAME; 410 411 us_per_frame = timeperframe->numerator * (u64)USEC_PER_SEC; 412 do_div(us_per_frame, timeperframe->denominator); 413 414 if (!us_per_frame) 415 return -EINVAL; 416 417 fps = (u64)USEC_PER_SEC; 418 do_div(fps, us_per_frame); 419 420 inst->timeperframe = *timeperframe; 421 inst->fps = fps; 422 423 return 0; 424 } 425 426 static int venc_g_parm(struct file *file, void *fh, struct v4l2_streamparm *a) 427 { 428 struct venus_inst *inst = to_inst(file); 429 430 if (a->type != V4L2_BUF_TYPE_VIDEO_OUTPUT && 431 a->type != V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) 432 return -EINVAL; 433 434 a->parm.output.capability |= V4L2_CAP_TIMEPERFRAME; 435 a->parm.output.timeperframe = inst->timeperframe; 436 437 return 0; 438 } 439 440 static int venc_enum_framesizes(struct file *file, void *fh, 441 struct v4l2_frmsizeenum *fsize) 442 { 443 struct venus_inst *inst = to_inst(file); 444 const struct venus_format *fmt; 445 446 fsize->type = V4L2_FRMSIZE_TYPE_STEPWISE; 447 448 fmt = find_format(inst, fsize->pixel_format, 449 V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE); 450 if (!fmt) { 451 fmt = find_format(inst, fsize->pixel_format, 452 V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE); 453 if (!fmt) 454 return -EINVAL; 455 } 456 457 if (fsize->index) 458 return -EINVAL; 459 460 fsize->stepwise.min_width = frame_width_min(inst); 461 fsize->stepwise.max_width = frame_width_max(inst); 462 fsize->stepwise.step_width = frame_width_step(inst); 463 fsize->stepwise.min_height = frame_height_min(inst); 464 fsize->stepwise.max_height = frame_height_max(inst); 465 fsize->stepwise.step_height = frame_height_step(inst); 466 467 return 0; 468 } 469 470 static int venc_enum_frameintervals(struct file *file, void *fh, 471 struct v4l2_frmivalenum *fival) 472 { 473 struct venus_inst *inst = to_inst(file); 474 const struct venus_format *fmt; 475 unsigned int framerate_factor = 1; 476 477 fival->type = V4L2_FRMIVAL_TYPE_STEPWISE; 478 479 fmt = find_format(inst, fival->pixel_format, 480 V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE); 481 if (!fmt) { 482 fmt = find_format(inst, fival->pixel_format, 483 V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE); 484 if (!fmt) 485 return -EINVAL; 486 } 487 488 if (fival->index) 489 return -EINVAL; 490 491 if (!fival->width || !fival->height) 492 return -EINVAL; 493 494 if (fival->width > frame_width_max(inst) || 495 fival->width < frame_width_min(inst) || 496 fival->height > frame_height_max(inst) || 497 fival->height < frame_height_min(inst)) 498 return -EINVAL; 499 500 if (IS_V1(inst->core)) { 501 /* framerate is reported in 1/65535 fps unit */ 502 framerate_factor = (1 << 16); 503 } 504 505 fival->stepwise.min.numerator = 1; 506 fival->stepwise.min.denominator = frate_max(inst) / framerate_factor; 507 fival->stepwise.max.numerator = 1; 508 fival->stepwise.max.denominator = frate_min(inst) / framerate_factor; 509 fival->stepwise.step.numerator = 1; 510 fival->stepwise.step.denominator = frate_max(inst) / framerate_factor; 511 512 return 0; 513 } 514 515 static int venc_subscribe_event(struct v4l2_fh *fh, 516 const struct v4l2_event_subscription *sub) 517 { 518 switch (sub->type) { 519 case V4L2_EVENT_EOS: 520 return v4l2_event_subscribe(fh, sub, 2, NULL); 521 case V4L2_EVENT_CTRL: 522 return v4l2_ctrl_subscribe_event(fh, sub); 523 default: 524 return -EINVAL; 525 } 526 } 527 528 static int 529 venc_encoder_cmd(struct file *file, void *fh, struct v4l2_encoder_cmd *cmd) 530 { 531 struct venus_inst *inst = to_inst(file); 532 struct hfi_frame_data fdata = {0}; 533 int ret = 0; 534 535 ret = v4l2_m2m_ioctl_try_encoder_cmd(file, fh, cmd); 536 if (ret) 537 return ret; 538 539 mutex_lock(&inst->lock); 540 541 if (cmd->cmd == V4L2_ENC_CMD_STOP && 542 inst->enc_state == VENUS_ENC_STATE_ENCODING) { 543 /* 544 * Implement V4L2_ENC_CMD_STOP by enqueue an empty buffer on 545 * encoder input to signal EOS. 546 */ 547 if (!(inst->streamon_out && inst->streamon_cap)) 548 goto unlock; 549 550 fdata.buffer_type = HFI_BUFFER_INPUT; 551 fdata.flags |= HFI_BUFFERFLAG_EOS; 552 fdata.device_addr = 0xdeadb000; 553 554 ret = hfi_session_process_buf(inst, &fdata); 555 556 inst->enc_state = VENUS_ENC_STATE_DRAIN; 557 } else if (cmd->cmd == V4L2_ENC_CMD_START) { 558 if (inst->enc_state == VENUS_ENC_STATE_DRAIN) { 559 ret = -EBUSY; 560 goto unlock; 561 } 562 if (inst->enc_state == VENUS_ENC_STATE_STOPPED) { 563 vb2_clear_last_buffer_dequeued(&inst->fh.m2m_ctx->cap_q_ctx.q); 564 inst->enc_state = VENUS_ENC_STATE_ENCODING; 565 } 566 } 567 568 unlock: 569 mutex_unlock(&inst->lock); 570 return ret; 571 } 572 573 static const struct v4l2_ioctl_ops venc_ioctl_ops = { 574 .vidioc_querycap = venc_querycap, 575 .vidioc_enum_fmt_vid_cap = venc_enum_fmt, 576 .vidioc_enum_fmt_vid_out = venc_enum_fmt, 577 .vidioc_s_fmt_vid_cap_mplane = venc_s_fmt, 578 .vidioc_s_fmt_vid_out_mplane = venc_s_fmt, 579 .vidioc_g_fmt_vid_cap_mplane = venc_g_fmt, 580 .vidioc_g_fmt_vid_out_mplane = venc_g_fmt, 581 .vidioc_try_fmt_vid_cap_mplane = venc_try_fmt, 582 .vidioc_try_fmt_vid_out_mplane = venc_try_fmt, 583 .vidioc_g_selection = venc_g_selection, 584 .vidioc_s_selection = venc_s_selection, 585 .vidioc_reqbufs = v4l2_m2m_ioctl_reqbufs, 586 .vidioc_querybuf = v4l2_m2m_ioctl_querybuf, 587 .vidioc_create_bufs = v4l2_m2m_ioctl_create_bufs, 588 .vidioc_prepare_buf = v4l2_m2m_ioctl_prepare_buf, 589 .vidioc_qbuf = v4l2_m2m_ioctl_qbuf, 590 .vidioc_expbuf = v4l2_m2m_ioctl_expbuf, 591 .vidioc_dqbuf = v4l2_m2m_ioctl_dqbuf, 592 .vidioc_streamon = v4l2_m2m_ioctl_streamon, 593 .vidioc_streamoff = v4l2_m2m_ioctl_streamoff, 594 .vidioc_s_parm = venc_s_parm, 595 .vidioc_g_parm = venc_g_parm, 596 .vidioc_enum_framesizes = venc_enum_framesizes, 597 .vidioc_enum_frameintervals = venc_enum_frameintervals, 598 .vidioc_subscribe_event = venc_subscribe_event, 599 .vidioc_unsubscribe_event = v4l2_event_unsubscribe, 600 .vidioc_try_encoder_cmd = v4l2_m2m_ioctl_try_encoder_cmd, 601 .vidioc_encoder_cmd = venc_encoder_cmd, 602 }; 603 604 static int venc_pm_get(struct venus_inst *inst) 605 { 606 struct venus_core *core = inst->core; 607 struct device *dev = core->dev_enc; 608 int ret; 609 610 mutex_lock(&core->pm_lock); 611 ret = pm_runtime_resume_and_get(dev); 612 mutex_unlock(&core->pm_lock); 613 614 return ret < 0 ? ret : 0; 615 } 616 617 static int venc_pm_put(struct venus_inst *inst, bool autosuspend) 618 { 619 struct venus_core *core = inst->core; 620 struct device *dev = core->dev_enc; 621 int ret; 622 623 mutex_lock(&core->pm_lock); 624 625 if (autosuspend) 626 ret = pm_runtime_put_autosuspend(dev); 627 else 628 ret = pm_runtime_put_sync(dev); 629 630 mutex_unlock(&core->pm_lock); 631 632 return ret < 0 ? ret : 0; 633 } 634 635 static int venc_pm_get_put(struct venus_inst *inst) 636 { 637 struct venus_core *core = inst->core; 638 struct device *dev = core->dev_enc; 639 int ret = 0; 640 641 mutex_lock(&core->pm_lock); 642 643 if (pm_runtime_suspended(dev)) { 644 ret = pm_runtime_resume_and_get(dev); 645 if (ret < 0) 646 goto error; 647 648 ret = pm_runtime_put_autosuspend(dev); 649 } 650 651 error: 652 mutex_unlock(&core->pm_lock); 653 654 return ret < 0 ? ret : 0; 655 } 656 657 static void venc_pm_touch(struct venus_inst *inst) 658 { 659 pm_runtime_mark_last_busy(inst->core->dev_enc); 660 } 661 662 static int venc_set_properties(struct venus_inst *inst) 663 { 664 struct venc_controls *ctr = &inst->controls.enc; 665 struct hfi_intra_period intra_period; 666 struct hfi_framerate frate; 667 struct hfi_bitrate brate; 668 struct hfi_idr_period idrp; 669 struct hfi_quantization quant; 670 struct hfi_quantization_range quant_range; 671 struct hfi_quantization_range_v2 quant_range_v2; 672 struct hfi_enable en; 673 struct hfi_ltr_mode ltr_mode; 674 struct hfi_intra_refresh intra_refresh = {}; 675 u32 ptype, rate_control, bitrate; 676 u32 profile, level; 677 int ret; 678 679 ret = venus_helper_set_work_mode(inst); 680 if (ret) 681 return ret; 682 683 ptype = HFI_PROPERTY_CONFIG_FRAME_RATE; 684 frate.buffer_type = HFI_BUFFER_OUTPUT; 685 frate.framerate = inst->fps * (1 << 16); 686 687 ret = hfi_session_set_property(inst, ptype, &frate); 688 if (ret) 689 return ret; 690 691 if (inst->fmt_cap->pixfmt == V4L2_PIX_FMT_H264) { 692 struct hfi_h264_vui_timing_info info; 693 struct hfi_h264_entropy_control entropy; 694 struct hfi_h264_db_control deblock; 695 struct hfi_h264_8x8_transform h264_transform; 696 697 ptype = HFI_PROPERTY_PARAM_VENC_H264_VUI_TIMING_INFO; 698 info.enable = 1; 699 info.fixed_framerate = 1; 700 info.time_scale = NSEC_PER_SEC; 701 702 ret = hfi_session_set_property(inst, ptype, &info); 703 if (ret) 704 return ret; 705 706 ptype = HFI_PROPERTY_PARAM_VENC_H264_ENTROPY_CONTROL; 707 entropy.entropy_mode = venc_v4l2_to_hfi( 708 V4L2_CID_MPEG_VIDEO_H264_ENTROPY_MODE, 709 ctr->h264_entropy_mode); 710 entropy.cabac_model = HFI_H264_CABAC_MODEL_0; 711 712 ret = hfi_session_set_property(inst, ptype, &entropy); 713 if (ret) 714 return ret; 715 716 ptype = HFI_PROPERTY_PARAM_VENC_H264_DEBLOCK_CONTROL; 717 deblock.mode = venc_v4l2_to_hfi( 718 V4L2_CID_MPEG_VIDEO_H264_LOOP_FILTER_MODE, 719 ctr->h264_loop_filter_mode); 720 deblock.slice_alpha_offset = ctr->h264_loop_filter_alpha; 721 deblock.slice_beta_offset = ctr->h264_loop_filter_beta; 722 723 ret = hfi_session_set_property(inst, ptype, &deblock); 724 if (ret) 725 return ret; 726 727 ptype = HFI_PROPERTY_PARAM_VENC_H264_TRANSFORM_8X8; 728 h264_transform.enable_type = 0; 729 if (ctr->profile.h264 == V4L2_MPEG_VIDEO_H264_PROFILE_HIGH || 730 ctr->profile.h264 == V4L2_MPEG_VIDEO_H264_PROFILE_CONSTRAINED_HIGH) 731 h264_transform.enable_type = ctr->h264_8x8_transform; 732 733 ret = hfi_session_set_property(inst, ptype, &h264_transform); 734 if (ret) 735 return ret; 736 737 if (ctr->layer_bitrate) { 738 unsigned int i; 739 740 ptype = HFI_PROPERTY_PARAM_VENC_HIER_P_MAX_NUM_ENH_LAYER; 741 ret = hfi_session_set_property(inst, ptype, &ctr->h264_hier_layers); 742 if (ret) 743 return ret; 744 745 ptype = HFI_PROPERTY_CONFIG_VENC_HIER_P_ENH_LAYER; 746 ret = hfi_session_set_property(inst, ptype, &ctr->layer_bitrate); 747 if (ret) 748 return ret; 749 750 for (i = 0; i < ctr->h264_hier_layers; ++i) { 751 ptype = HFI_PROPERTY_CONFIG_VENC_TARGET_BITRATE; 752 brate.bitrate = ctr->h264_hier_layer_bitrate[i]; 753 brate.layer_id = i; 754 755 ret = hfi_session_set_property(inst, ptype, &brate); 756 if (ret) 757 return ret; 758 } 759 } 760 } 761 762 if (inst->fmt_cap->pixfmt == V4L2_PIX_FMT_H264 || 763 inst->fmt_cap->pixfmt == V4L2_PIX_FMT_HEVC) { 764 /* IDR periodicity, n: 765 * n = 0 - only the first I-frame is IDR frame 766 * n = 1 - all I-frames will be IDR frames 767 * n > 1 - every n-th I-frame will be IDR frame 768 */ 769 ptype = HFI_PROPERTY_CONFIG_VENC_IDR_PERIOD; 770 idrp.idr_period = 0; 771 ret = hfi_session_set_property(inst, ptype, &idrp); 772 if (ret) 773 return ret; 774 } 775 776 if (inst->fmt_cap->pixfmt == V4L2_PIX_FMT_HEVC && 777 ctr->profile.hevc == V4L2_MPEG_VIDEO_HEVC_PROFILE_MAIN_10) { 778 struct hfi_hdr10_pq_sei hdr10; 779 unsigned int c; 780 781 ptype = HFI_PROPERTY_PARAM_VENC_HDR10_PQ_SEI; 782 783 for (c = 0; c < 3; c++) { 784 hdr10.mastering.display_primaries_x[c] = 785 ctr->mastering.display_primaries_x[c]; 786 hdr10.mastering.display_primaries_y[c] = 787 ctr->mastering.display_primaries_y[c]; 788 } 789 790 hdr10.mastering.white_point_x = ctr->mastering.white_point_x; 791 hdr10.mastering.white_point_y = ctr->mastering.white_point_y; 792 hdr10.mastering.max_display_mastering_luminance = 793 ctr->mastering.max_display_mastering_luminance; 794 hdr10.mastering.min_display_mastering_luminance = 795 ctr->mastering.min_display_mastering_luminance; 796 797 hdr10.cll.max_content_light = ctr->cll.max_content_light_level; 798 hdr10.cll.max_pic_average_light = 799 ctr->cll.max_pic_average_light_level; 800 801 ret = hfi_session_set_property(inst, ptype, &hdr10); 802 if (ret) 803 return ret; 804 } 805 806 if (ctr->num_b_frames) { 807 u32 max_num_b_frames = NUM_B_FRAMES_MAX; 808 809 ptype = HFI_PROPERTY_PARAM_VENC_MAX_NUM_B_FRAMES; 810 ret = hfi_session_set_property(inst, ptype, &max_num_b_frames); 811 if (ret) 812 return ret; 813 } 814 815 ptype = HFI_PROPERTY_CONFIG_VENC_INTRA_PERIOD; 816 intra_period.pframes = ctr->num_p_frames; 817 intra_period.bframes = ctr->num_b_frames; 818 819 ret = hfi_session_set_property(inst, ptype, &intra_period); 820 if (ret) 821 return ret; 822 823 if (!ctr->rc_enable) 824 rate_control = HFI_RATE_CONTROL_OFF; 825 else if (ctr->bitrate_mode == V4L2_MPEG_VIDEO_BITRATE_MODE_VBR) 826 rate_control = ctr->frame_skip_mode ? HFI_RATE_CONTROL_VBR_VFR : 827 HFI_RATE_CONTROL_VBR_CFR; 828 else if (ctr->bitrate_mode == V4L2_MPEG_VIDEO_BITRATE_MODE_CBR) 829 rate_control = ctr->frame_skip_mode ? HFI_RATE_CONTROL_CBR_VFR : 830 HFI_RATE_CONTROL_CBR_CFR; 831 else if (ctr->bitrate_mode == V4L2_MPEG_VIDEO_BITRATE_MODE_CQ) 832 rate_control = HFI_RATE_CONTROL_CQ; 833 834 ptype = HFI_PROPERTY_PARAM_VENC_RATE_CONTROL; 835 ret = hfi_session_set_property(inst, ptype, &rate_control); 836 if (ret) 837 return ret; 838 839 if (rate_control == HFI_RATE_CONTROL_CQ && ctr->const_quality) { 840 struct hfi_heic_frame_quality quality = {}; 841 842 ptype = HFI_PROPERTY_CONFIG_HEIC_FRAME_QUALITY; 843 quality.frame_quality = ctr->const_quality; 844 ret = hfi_session_set_property(inst, ptype, &quality); 845 if (ret) 846 return ret; 847 } 848 849 if (!ctr->layer_bitrate) { 850 if (!ctr->bitrate) 851 bitrate = 64000; 852 else 853 bitrate = ctr->bitrate; 854 855 ptype = HFI_PROPERTY_CONFIG_VENC_TARGET_BITRATE; 856 brate.bitrate = bitrate; 857 brate.layer_id = 0; 858 859 ret = hfi_session_set_property(inst, ptype, &brate); 860 if (ret) 861 return ret; 862 863 if (!ctr->bitrate_peak) 864 bitrate *= 2; 865 else 866 bitrate = ctr->bitrate_peak; 867 868 ptype = HFI_PROPERTY_CONFIG_VENC_MAX_BITRATE; 869 brate.bitrate = bitrate; 870 brate.layer_id = 0; 871 872 ret = hfi_session_set_property(inst, ptype, &brate); 873 if (ret) 874 return ret; 875 } 876 877 if (inst->fmt_cap->pixfmt == V4L2_PIX_FMT_H264 || 878 inst->fmt_cap->pixfmt == V4L2_PIX_FMT_HEVC) { 879 ptype = HFI_PROPERTY_CONFIG_VENC_SYNC_FRAME_SEQUENCE_HEADER; 880 if (ctr->header_mode == V4L2_MPEG_VIDEO_HEADER_MODE_SEPARATE) 881 en.enable = 0; 882 else 883 en.enable = 1; 884 885 ret = hfi_session_set_property(inst, ptype, &en); 886 if (ret) 887 return ret; 888 } 889 890 ptype = HFI_PROPERTY_PARAM_VENC_SESSION_QP; 891 if (inst->fmt_cap->pixfmt == V4L2_PIX_FMT_HEVC) { 892 quant.qp_i = ctr->hevc_i_qp; 893 quant.qp_p = ctr->hevc_p_qp; 894 quant.qp_b = ctr->hevc_b_qp; 895 } else { 896 quant.qp_i = ctr->h264_i_qp; 897 quant.qp_p = ctr->h264_p_qp; 898 quant.qp_b = ctr->h264_b_qp; 899 } 900 quant.layer_id = 0; 901 ret = hfi_session_set_property(inst, ptype, &quant); 902 if (ret) 903 return ret; 904 905 if (inst->core->res->hfi_version == HFI_VERSION_4XX || 906 inst->core->res->hfi_version == HFI_VERSION_6XX) { 907 ptype = HFI_PROPERTY_PARAM_VENC_SESSION_QP_RANGE_V2; 908 909 if (inst->fmt_cap->pixfmt == V4L2_PIX_FMT_HEVC) { 910 quant_range_v2.min_qp.qp_packed = ctr->hevc_min_qp; 911 quant_range_v2.max_qp.qp_packed = ctr->hevc_max_qp; 912 } else if (inst->fmt_cap->pixfmt == V4L2_PIX_FMT_VP8) { 913 quant_range_v2.min_qp.qp_packed = ctr->vp8_min_qp; 914 quant_range_v2.max_qp.qp_packed = ctr->vp8_max_qp; 915 } else { 916 quant_range_v2.min_qp.qp_packed = ctr->h264_min_qp; 917 quant_range_v2.max_qp.qp_packed = ctr->h264_max_qp; 918 } 919 920 ret = hfi_session_set_property(inst, ptype, &quant_range_v2); 921 } else { 922 ptype = HFI_PROPERTY_PARAM_VENC_SESSION_QP_RANGE; 923 924 if (inst->fmt_cap->pixfmt == V4L2_PIX_FMT_HEVC) { 925 quant_range.min_qp = ctr->hevc_min_qp; 926 quant_range.max_qp = ctr->hevc_max_qp; 927 } else if (inst->fmt_cap->pixfmt == V4L2_PIX_FMT_VP8) { 928 quant_range.min_qp = ctr->vp8_min_qp; 929 quant_range.max_qp = ctr->vp8_max_qp; 930 } else { 931 quant_range.min_qp = ctr->h264_min_qp; 932 quant_range.max_qp = ctr->h264_max_qp; 933 } 934 935 quant_range.layer_id = 0; 936 ret = hfi_session_set_property(inst, ptype, &quant_range); 937 } 938 939 if (ret) 940 return ret; 941 942 ptype = HFI_PROPERTY_PARAM_VENC_LTRMODE; 943 ltr_mode.ltr_count = ctr->ltr_count; 944 ltr_mode.ltr_mode = HFI_LTR_MODE_MANUAL; 945 ltr_mode.trust_mode = 1; 946 ret = hfi_session_set_property(inst, ptype, <r_mode); 947 if (ret) 948 return ret; 949 950 switch (inst->hfi_codec) { 951 case HFI_VIDEO_CODEC_H264: 952 profile = ctr->profile.h264; 953 level = ctr->level.h264; 954 break; 955 case HFI_VIDEO_CODEC_MPEG4: 956 profile = ctr->profile.mpeg4; 957 level = ctr->level.mpeg4; 958 break; 959 case HFI_VIDEO_CODEC_VP8: 960 profile = ctr->profile.vp8; 961 level = 0; 962 break; 963 case HFI_VIDEO_CODEC_VP9: 964 profile = ctr->profile.vp9; 965 level = ctr->level.vp9; 966 break; 967 case HFI_VIDEO_CODEC_HEVC: 968 profile = ctr->profile.hevc; 969 level = ctr->level.hevc; 970 break; 971 case HFI_VIDEO_CODEC_MPEG2: 972 default: 973 profile = 0; 974 level = 0; 975 break; 976 } 977 978 ret = venus_helper_set_profile_level(inst, profile, level); 979 if (ret) 980 return ret; 981 982 if (inst->fmt_cap->pixfmt == V4L2_PIX_FMT_H264 || 983 inst->fmt_cap->pixfmt == V4L2_PIX_FMT_HEVC) { 984 struct hfi_enable en = {}; 985 986 ptype = HFI_PROPERTY_PARAM_VENC_H264_GENERATE_AUDNAL; 987 988 if (ctr->aud_enable) 989 en.enable = 1; 990 991 ret = hfi_session_set_property(inst, ptype, &en); 992 } 993 994 if ((inst->fmt_cap->pixfmt == V4L2_PIX_FMT_H264 || 995 inst->fmt_cap->pixfmt == V4L2_PIX_FMT_HEVC) && 996 (rate_control == HFI_RATE_CONTROL_CBR_VFR || 997 rate_control == HFI_RATE_CONTROL_CBR_CFR)) { 998 intra_refresh.mode = HFI_INTRA_REFRESH_NONE; 999 intra_refresh.cir_mbs = 0; 1000 1001 if (ctr->intra_refresh_period) { 1002 u32 mbs; 1003 1004 mbs = ALIGN(inst->width, 16) * ALIGN(inst->height, 16); 1005 mbs /= 16 * 16; 1006 if (mbs % ctr->intra_refresh_period) 1007 mbs++; 1008 mbs /= ctr->intra_refresh_period; 1009 1010 intra_refresh.cir_mbs = mbs; 1011 if (ctr->intra_refresh_type == 1012 V4L2_CID_MPEG_VIDEO_INTRA_REFRESH_PERIOD_TYPE_CYCLIC) 1013 intra_refresh.mode = HFI_INTRA_REFRESH_CYCLIC; 1014 else 1015 intra_refresh.mode = HFI_INTRA_REFRESH_RANDOM; 1016 } 1017 1018 ptype = HFI_PROPERTY_PARAM_VENC_INTRA_REFRESH; 1019 1020 ret = hfi_session_set_property(inst, ptype, &intra_refresh); 1021 if (ret) 1022 return ret; 1023 } 1024 1025 return 0; 1026 } 1027 1028 static int venc_init_session(struct venus_inst *inst) 1029 { 1030 int ret; 1031 1032 ret = venus_helper_session_init(inst); 1033 if (ret == -EALREADY) 1034 return 0; 1035 else if (ret) 1036 return ret; 1037 1038 ret = venus_helper_set_stride(inst, inst->out_width, 1039 inst->out_height); 1040 if (ret) 1041 goto deinit; 1042 1043 ret = venus_helper_set_input_resolution(inst, inst->width, 1044 inst->height); 1045 if (ret) 1046 goto deinit; 1047 1048 ret = venus_helper_set_output_resolution(inst, inst->width, 1049 inst->height, 1050 HFI_BUFFER_OUTPUT); 1051 if (ret) 1052 goto deinit; 1053 1054 ret = venus_helper_set_color_format(inst, inst->fmt_out->pixfmt); 1055 if (ret) 1056 goto deinit; 1057 1058 ret = venc_set_properties(inst); 1059 if (ret) 1060 goto deinit; 1061 1062 return 0; 1063 deinit: 1064 hfi_session_deinit(inst); 1065 return ret; 1066 } 1067 1068 static int venc_out_num_buffers(struct venus_inst *inst, unsigned int *num) 1069 { 1070 struct hfi_buffer_requirements bufreq; 1071 int ret; 1072 1073 ret = venus_helper_get_bufreq(inst, HFI_BUFFER_INPUT, &bufreq); 1074 if (ret) 1075 return ret; 1076 1077 *num = bufreq.count_actual; 1078 1079 return 0; 1080 } 1081 1082 static int venc_queue_setup(struct vb2_queue *q, 1083 unsigned int *num_buffers, unsigned int *num_planes, 1084 unsigned int sizes[], struct device *alloc_devs[]) 1085 { 1086 struct venus_inst *inst = vb2_get_drv_priv(q); 1087 struct venus_core *core = inst->core; 1088 unsigned int num, min = 4; 1089 int ret; 1090 1091 if (*num_planes) { 1092 if (q->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE && 1093 *num_planes != inst->fmt_out->num_planes) 1094 return -EINVAL; 1095 1096 if (q->type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE && 1097 *num_planes != inst->fmt_cap->num_planes) 1098 return -EINVAL; 1099 1100 if (q->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE && 1101 sizes[0] < inst->input_buf_size) 1102 return -EINVAL; 1103 1104 if (q->type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE && 1105 sizes[0] < inst->output_buf_size) 1106 return -EINVAL; 1107 1108 return 0; 1109 } 1110 1111 if (test_bit(0, &core->sys_error)) { 1112 if (inst->nonblock) 1113 return -EAGAIN; 1114 1115 ret = wait_event_interruptible(core->sys_err_done, 1116 !test_bit(0, &core->sys_error)); 1117 if (ret) 1118 return ret; 1119 } 1120 1121 ret = venc_pm_get(inst); 1122 if (ret) 1123 return ret; 1124 1125 mutex_lock(&inst->lock); 1126 ret = venc_init_session(inst); 1127 mutex_unlock(&inst->lock); 1128 1129 if (ret) 1130 goto put_power; 1131 1132 ret = venc_pm_put(inst, false); 1133 if (ret) 1134 return ret; 1135 1136 switch (q->type) { 1137 case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE: 1138 *num_planes = inst->fmt_out->num_planes; 1139 1140 ret = venc_out_num_buffers(inst, &num); 1141 if (ret) 1142 break; 1143 1144 num = max(num, min); 1145 *num_buffers = max(*num_buffers, num); 1146 inst->num_input_bufs = *num_buffers; 1147 1148 sizes[0] = venus_helper_get_framesz(inst->fmt_out->pixfmt, 1149 inst->out_width, 1150 inst->out_height); 1151 inst->input_buf_size = sizes[0]; 1152 break; 1153 case V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE: 1154 *num_planes = inst->fmt_cap->num_planes; 1155 *num_buffers = max(*num_buffers, min); 1156 inst->num_output_bufs = *num_buffers; 1157 sizes[0] = venus_helper_get_framesz(inst->fmt_cap->pixfmt, 1158 inst->width, 1159 inst->height); 1160 sizes[0] = max(sizes[0], inst->output_buf_size); 1161 inst->output_buf_size = sizes[0]; 1162 break; 1163 default: 1164 ret = -EINVAL; 1165 break; 1166 } 1167 1168 return ret; 1169 put_power: 1170 venc_pm_put(inst, false); 1171 return ret; 1172 } 1173 1174 static int venc_buf_init(struct vb2_buffer *vb) 1175 { 1176 struct venus_inst *inst = vb2_get_drv_priv(vb->vb2_queue); 1177 1178 inst->buf_count++; 1179 1180 return venus_helper_vb2_buf_init(vb); 1181 } 1182 1183 static void venc_release_session(struct venus_inst *inst) 1184 { 1185 int ret; 1186 1187 venc_pm_get(inst); 1188 1189 mutex_lock(&inst->lock); 1190 1191 ret = hfi_session_deinit(inst); 1192 if (ret || inst->session_error) 1193 hfi_session_abort(inst); 1194 1195 mutex_unlock(&inst->lock); 1196 1197 venus_pm_load_scale(inst); 1198 INIT_LIST_HEAD(&inst->registeredbufs); 1199 venus_pm_release_core(inst); 1200 1201 venc_pm_put(inst, false); 1202 } 1203 1204 static void venc_buf_cleanup(struct vb2_buffer *vb) 1205 { 1206 struct venus_inst *inst = vb2_get_drv_priv(vb->vb2_queue); 1207 struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb); 1208 struct venus_buffer *buf = to_venus_buffer(vbuf); 1209 1210 mutex_lock(&inst->lock); 1211 if (vb->type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE) 1212 if (!list_empty(&inst->registeredbufs)) 1213 list_del_init(&buf->reg_list); 1214 mutex_unlock(&inst->lock); 1215 1216 inst->buf_count--; 1217 if (!inst->buf_count) 1218 venc_release_session(inst); 1219 } 1220 1221 static int venc_verify_conf(struct venus_inst *inst) 1222 { 1223 enum hfi_version ver = inst->core->res->hfi_version; 1224 struct hfi_buffer_requirements bufreq; 1225 int ret; 1226 1227 if (!inst->num_input_bufs || !inst->num_output_bufs) 1228 return -EINVAL; 1229 1230 ret = venus_helper_get_bufreq(inst, HFI_BUFFER_OUTPUT, &bufreq); 1231 if (ret) 1232 return ret; 1233 1234 if (inst->num_output_bufs < bufreq.count_actual || 1235 inst->num_output_bufs < hfi_bufreq_get_count_min(&bufreq, ver)) 1236 return -EINVAL; 1237 1238 ret = venus_helper_get_bufreq(inst, HFI_BUFFER_INPUT, &bufreq); 1239 if (ret) 1240 return ret; 1241 1242 if (inst->num_input_bufs < bufreq.count_actual || 1243 inst->num_input_bufs < hfi_bufreq_get_count_min(&bufreq, ver)) 1244 return -EINVAL; 1245 1246 return 0; 1247 } 1248 1249 static int venc_start_streaming(struct vb2_queue *q, unsigned int count) 1250 { 1251 struct venus_inst *inst = vb2_get_drv_priv(q); 1252 int ret; 1253 1254 mutex_lock(&inst->lock); 1255 1256 if (q->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) 1257 inst->streamon_out = 1; 1258 else 1259 inst->streamon_cap = 1; 1260 1261 if (!(inst->streamon_out & inst->streamon_cap)) { 1262 mutex_unlock(&inst->lock); 1263 return 0; 1264 } 1265 1266 venus_helper_init_instance(inst); 1267 1268 inst->sequence_cap = 0; 1269 inst->sequence_out = 0; 1270 1271 ret = venc_pm_get(inst); 1272 if (ret) 1273 goto error; 1274 1275 ret = venus_pm_acquire_core(inst); 1276 if (ret) 1277 goto put_power; 1278 1279 ret = venc_pm_put(inst, true); 1280 if (ret) 1281 goto error; 1282 1283 ret = venc_set_properties(inst); 1284 if (ret) 1285 goto error; 1286 1287 ret = venc_verify_conf(inst); 1288 if (ret) 1289 goto error; 1290 1291 ret = venus_helper_set_num_bufs(inst, inst->num_input_bufs, 1292 inst->num_output_bufs, 0); 1293 if (ret) 1294 goto error; 1295 1296 ret = venus_helper_vb2_start_streaming(inst); 1297 if (ret) 1298 goto error; 1299 1300 inst->enc_state = VENUS_ENC_STATE_ENCODING; 1301 1302 mutex_unlock(&inst->lock); 1303 1304 return 0; 1305 1306 put_power: 1307 venc_pm_put(inst, false); 1308 error: 1309 venus_helper_buffers_done(inst, q->type, VB2_BUF_STATE_QUEUED); 1310 if (q->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) 1311 inst->streamon_out = 0; 1312 else 1313 inst->streamon_cap = 0; 1314 mutex_unlock(&inst->lock); 1315 return ret; 1316 } 1317 1318 static void venc_vb2_buf_queue(struct vb2_buffer *vb) 1319 { 1320 struct venus_inst *inst = vb2_get_drv_priv(vb->vb2_queue); 1321 struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb); 1322 1323 venc_pm_get_put(inst); 1324 1325 mutex_lock(&inst->lock); 1326 1327 if (inst->enc_state == VENUS_ENC_STATE_STOPPED) { 1328 vbuf->sequence = inst->sequence_cap++; 1329 vbuf->field = V4L2_FIELD_NONE; 1330 vb2_set_plane_payload(vb, 0, 0); 1331 v4l2_m2m_buf_done(vbuf, VB2_BUF_STATE_DONE); 1332 mutex_unlock(&inst->lock); 1333 return; 1334 } 1335 1336 venus_helper_vb2_buf_queue(vb); 1337 mutex_unlock(&inst->lock); 1338 } 1339 1340 static const struct vb2_ops venc_vb2_ops = { 1341 .queue_setup = venc_queue_setup, 1342 .buf_init = venc_buf_init, 1343 .buf_cleanup = venc_buf_cleanup, 1344 .buf_prepare = venus_helper_vb2_buf_prepare, 1345 .start_streaming = venc_start_streaming, 1346 .stop_streaming = venus_helper_vb2_stop_streaming, 1347 .buf_queue = venc_vb2_buf_queue, 1348 }; 1349 1350 static void venc_buf_done(struct venus_inst *inst, unsigned int buf_type, 1351 u32 tag, u32 bytesused, u32 data_offset, u32 flags, 1352 u32 hfi_flags, u64 timestamp_us) 1353 { 1354 struct vb2_v4l2_buffer *vbuf; 1355 struct vb2_buffer *vb; 1356 unsigned int type; 1357 1358 venc_pm_touch(inst); 1359 1360 if (buf_type == HFI_BUFFER_INPUT) 1361 type = V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE; 1362 else 1363 type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE; 1364 1365 vbuf = venus_helper_find_buf(inst, type, tag); 1366 if (!vbuf) 1367 return; 1368 1369 vbuf->flags = flags; 1370 1371 if (type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE) { 1372 vb = &vbuf->vb2_buf; 1373 vb2_set_plane_payload(vb, 0, bytesused + data_offset); 1374 vb->planes[0].data_offset = data_offset; 1375 vb->timestamp = timestamp_us * NSEC_PER_USEC; 1376 vbuf->sequence = inst->sequence_cap++; 1377 if ((vbuf->flags & V4L2_BUF_FLAG_LAST) && 1378 inst->enc_state == VENUS_ENC_STATE_DRAIN) { 1379 inst->enc_state = VENUS_ENC_STATE_STOPPED; 1380 } 1381 } else { 1382 vbuf->sequence = inst->sequence_out++; 1383 } 1384 1385 v4l2_m2m_buf_done(vbuf, VB2_BUF_STATE_DONE); 1386 } 1387 1388 static void venc_event_notify(struct venus_inst *inst, u32 event, 1389 struct hfi_event_data *data) 1390 { 1391 struct device *dev = inst->core->dev_enc; 1392 1393 venc_pm_touch(inst); 1394 1395 if (event == EVT_SESSION_ERROR) { 1396 inst->session_error = true; 1397 venus_helper_vb2_queue_error(inst); 1398 dev_err(dev, "enc: event session error %x\n", inst->error); 1399 } 1400 } 1401 1402 static const struct hfi_inst_ops venc_hfi_ops = { 1403 .buf_done = venc_buf_done, 1404 .event_notify = venc_event_notify, 1405 }; 1406 1407 static const struct v4l2_m2m_ops venc_m2m_ops = { 1408 .device_run = venus_helper_m2m_device_run, 1409 .job_abort = venus_helper_m2m_job_abort, 1410 }; 1411 1412 static int m2m_queue_init(void *priv, struct vb2_queue *src_vq, 1413 struct vb2_queue *dst_vq) 1414 { 1415 struct venus_inst *inst = priv; 1416 int ret; 1417 1418 src_vq->type = V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE; 1419 src_vq->io_modes = VB2_MMAP | VB2_USERPTR | VB2_DMABUF; 1420 src_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY; 1421 src_vq->ops = &venc_vb2_ops; 1422 src_vq->mem_ops = &vb2_dma_contig_memops; 1423 src_vq->drv_priv = inst; 1424 src_vq->buf_struct_size = sizeof(struct venus_buffer); 1425 src_vq->allow_zero_bytesused = 1; 1426 src_vq->min_queued_buffers = 1; 1427 src_vq->dev = inst->core->dev; 1428 src_vq->lock = &inst->ctx_q_lock; 1429 if (inst->core->res->hfi_version == HFI_VERSION_1XX) 1430 src_vq->bidirectional = 1; 1431 ret = vb2_queue_init(src_vq); 1432 if (ret) 1433 return ret; 1434 1435 dst_vq->type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE; 1436 dst_vq->io_modes = VB2_MMAP | VB2_USERPTR | VB2_DMABUF; 1437 dst_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY; 1438 dst_vq->ops = &venc_vb2_ops; 1439 dst_vq->mem_ops = &vb2_dma_contig_memops; 1440 dst_vq->drv_priv = inst; 1441 dst_vq->buf_struct_size = sizeof(struct venus_buffer); 1442 dst_vq->allow_zero_bytesused = 1; 1443 dst_vq->min_queued_buffers = 1; 1444 dst_vq->dev = inst->core->dev; 1445 dst_vq->lock = &inst->ctx_q_lock; 1446 return vb2_queue_init(dst_vq); 1447 } 1448 1449 static void venc_inst_init(struct venus_inst *inst) 1450 { 1451 inst->fmt_cap = &venc_formats[VENUS_FMT_H264]; 1452 inst->fmt_out = &venc_formats[VENUS_FMT_NV12]; 1453 inst->width = 1280; 1454 inst->height = ALIGN(720, 32); 1455 inst->out_width = 1280; 1456 inst->out_height = 720; 1457 inst->fps = 15; 1458 inst->timeperframe.numerator = 1; 1459 inst->timeperframe.denominator = 15; 1460 inst->hfi_codec = HFI_VIDEO_CODEC_H264; 1461 } 1462 1463 static int venc_open(struct file *file) 1464 { 1465 struct venus_core *core = video_drvdata(file); 1466 struct venus_inst *inst; 1467 int ret; 1468 1469 inst = kzalloc(sizeof(*inst), GFP_KERNEL); 1470 if (!inst) 1471 return -ENOMEM; 1472 1473 INIT_LIST_HEAD(&inst->dpbbufs); 1474 INIT_LIST_HEAD(&inst->registeredbufs); 1475 INIT_LIST_HEAD(&inst->internalbufs); 1476 INIT_LIST_HEAD(&inst->list); 1477 mutex_init(&inst->lock); 1478 mutex_init(&inst->ctx_q_lock); 1479 1480 inst->core = core; 1481 inst->session_type = VIDC_SESSION_TYPE_ENC; 1482 inst->clk_data.core_id = VIDC_CORE_ID_DEFAULT; 1483 inst->core_acquired = false; 1484 inst->nonblock = file->f_flags & O_NONBLOCK; 1485 1486 if (inst->enc_state == VENUS_ENC_STATE_DEINIT) 1487 inst->enc_state = VENUS_ENC_STATE_INIT; 1488 1489 venus_helper_init_instance(inst); 1490 1491 ret = venc_ctrl_init(inst); 1492 if (ret) 1493 goto err_free; 1494 1495 ret = hfi_session_create(inst, &venc_hfi_ops); 1496 if (ret) 1497 goto err_ctrl_deinit; 1498 1499 venc_inst_init(inst); 1500 1501 /* 1502 * create m2m device for every instance, the m2m context scheduling 1503 * is made by firmware side so we do not need to care about. 1504 */ 1505 inst->m2m_dev = v4l2_m2m_init(&venc_m2m_ops); 1506 if (IS_ERR(inst->m2m_dev)) { 1507 ret = PTR_ERR(inst->m2m_dev); 1508 goto err_session_destroy; 1509 } 1510 1511 inst->m2m_ctx = v4l2_m2m_ctx_init(inst->m2m_dev, inst, m2m_queue_init); 1512 if (IS_ERR(inst->m2m_ctx)) { 1513 ret = PTR_ERR(inst->m2m_ctx); 1514 goto err_m2m_release; 1515 } 1516 1517 v4l2_fh_init(&inst->fh, core->vdev_enc); 1518 1519 inst->fh.ctrl_handler = &inst->ctrl_handler; 1520 v4l2_fh_add(&inst->fh); 1521 inst->fh.m2m_ctx = inst->m2m_ctx; 1522 file->private_data = &inst->fh; 1523 1524 return 0; 1525 1526 err_m2m_release: 1527 v4l2_m2m_release(inst->m2m_dev); 1528 err_session_destroy: 1529 hfi_session_destroy(inst); 1530 err_ctrl_deinit: 1531 v4l2_ctrl_handler_free(&inst->ctrl_handler); 1532 err_free: 1533 kfree(inst); 1534 return ret; 1535 } 1536 1537 static int venc_close(struct file *file) 1538 { 1539 struct venus_inst *inst = to_inst(file); 1540 1541 venc_pm_get(inst); 1542 venus_close_common(inst); 1543 inst->enc_state = VENUS_ENC_STATE_DEINIT; 1544 venc_pm_put(inst, false); 1545 1546 kfree(inst); 1547 return 0; 1548 } 1549 1550 static const struct v4l2_file_operations venc_fops = { 1551 .owner = THIS_MODULE, 1552 .open = venc_open, 1553 .release = venc_close, 1554 .unlocked_ioctl = video_ioctl2, 1555 .poll = v4l2_m2m_fop_poll, 1556 .mmap = v4l2_m2m_fop_mmap, 1557 }; 1558 1559 static int venc_probe(struct platform_device *pdev) 1560 { 1561 struct device *dev = &pdev->dev; 1562 struct video_device *vdev; 1563 struct venus_core *core; 1564 int ret; 1565 1566 if (!dev->parent) 1567 return -EPROBE_DEFER; 1568 1569 core = dev_get_drvdata(dev->parent); 1570 if (!core) 1571 return -EPROBE_DEFER; 1572 1573 platform_set_drvdata(pdev, core); 1574 1575 if (core->pm_ops->venc_get) { 1576 ret = core->pm_ops->venc_get(dev); 1577 if (ret) 1578 return ret; 1579 } 1580 1581 vdev = video_device_alloc(); 1582 if (!vdev) 1583 return -ENOMEM; 1584 1585 strscpy(vdev->name, "qcom-venus-encoder", sizeof(vdev->name)); 1586 vdev->release = video_device_release; 1587 vdev->fops = &venc_fops; 1588 vdev->ioctl_ops = &venc_ioctl_ops; 1589 vdev->vfl_dir = VFL_DIR_M2M; 1590 vdev->v4l2_dev = &core->v4l2_dev; 1591 vdev->device_caps = V4L2_CAP_VIDEO_M2M_MPLANE | V4L2_CAP_STREAMING; 1592 1593 ret = video_register_device(vdev, VFL_TYPE_VIDEO, -1); 1594 if (ret) 1595 goto err_vdev_release; 1596 1597 core->vdev_enc = vdev; 1598 core->dev_enc = dev; 1599 1600 video_set_drvdata(vdev, core); 1601 pm_runtime_set_autosuspend_delay(dev, 2000); 1602 pm_runtime_use_autosuspend(dev); 1603 pm_runtime_enable(dev); 1604 1605 return 0; 1606 1607 err_vdev_release: 1608 video_device_release(vdev); 1609 return ret; 1610 } 1611 1612 static void venc_remove(struct platform_device *pdev) 1613 { 1614 struct venus_core *core = dev_get_drvdata(pdev->dev.parent); 1615 1616 video_unregister_device(core->vdev_enc); 1617 pm_runtime_disable(core->dev_enc); 1618 1619 if (core->pm_ops->venc_put) 1620 core->pm_ops->venc_put(core->dev_enc); 1621 } 1622 1623 static __maybe_unused int venc_runtime_suspend(struct device *dev) 1624 { 1625 struct venus_core *core = dev_get_drvdata(dev); 1626 const struct venus_pm_ops *pm_ops = core->pm_ops; 1627 int ret = 0; 1628 1629 if (pm_ops->venc_power) 1630 ret = pm_ops->venc_power(dev, POWER_OFF); 1631 1632 return ret; 1633 } 1634 1635 static __maybe_unused int venc_runtime_resume(struct device *dev) 1636 { 1637 struct venus_core *core = dev_get_drvdata(dev); 1638 const struct venus_pm_ops *pm_ops = core->pm_ops; 1639 int ret = 0; 1640 1641 if (pm_ops->venc_power) 1642 ret = pm_ops->venc_power(dev, POWER_ON); 1643 1644 return ret; 1645 } 1646 1647 static const struct dev_pm_ops venc_pm_ops = { 1648 SET_SYSTEM_SLEEP_PM_OPS(pm_runtime_force_suspend, 1649 pm_runtime_force_resume) 1650 SET_RUNTIME_PM_OPS(venc_runtime_suspend, venc_runtime_resume, NULL) 1651 }; 1652 1653 static const struct of_device_id venc_dt_match[] = { 1654 { .compatible = "venus-encoder" }, 1655 { } 1656 }; 1657 MODULE_DEVICE_TABLE(of, venc_dt_match); 1658 1659 static struct platform_driver qcom_venus_enc_driver = { 1660 .probe = venc_probe, 1661 .remove = venc_remove, 1662 .driver = { 1663 .name = "qcom-venus-encoder", 1664 .of_match_table = venc_dt_match, 1665 .pm = &venc_pm_ops, 1666 }, 1667 }; 1668 module_platform_driver(qcom_venus_enc_driver); 1669 1670 MODULE_ALIAS("platform:qcom-venus-encoder"); 1671 MODULE_DESCRIPTION("Qualcomm Venus video encoder driver"); 1672 MODULE_LICENSE("GPL v2"); 1673