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