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 *
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 *
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 
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
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 
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 *
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 
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 
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 
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
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
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 
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 
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 
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 
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 
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 
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 
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 
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 
606 static void venc_pm_touch(struct venus_inst *inst)
607 {
608 	pm_runtime_mark_last_busy(inst->core->dev_enc);
609 }
610 
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, &ltr_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 
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 
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 
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 
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 
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 
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 
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 
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 
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 
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 
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 
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 
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 
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 
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 
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 
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 
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 
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