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