media: venus: helpers: Add a helper to map v4l2 ids to HFI ids
[linux-2.6-microblaze.git] / drivers / media / platform / qcom / venus / helpers.c
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/list.h>
7 #include <linux/mutex.h>
8 #include <linux/slab.h>
9 #include <linux/kernel.h>
10 #include <media/videobuf2-dma-sg.h>
11 #include <media/v4l2-mem2mem.h>
12 #include <asm/div64.h>
13
14 #include "core.h"
15 #include "helpers.h"
16 #include "hfi_helper.h"
17 #include "pm_helpers.h"
18
19 struct intbuf {
20         struct list_head list;
21         u32 type;
22         size_t size;
23         void *va;
24         dma_addr_t da;
25         unsigned long attrs;
26 };
27
28 bool venus_helper_check_codec(struct venus_inst *inst, u32 v4l2_pixfmt)
29 {
30         struct venus_core *core = inst->core;
31         u32 session_type = inst->session_type;
32         u32 codec;
33
34         switch (v4l2_pixfmt) {
35         case V4L2_PIX_FMT_H264:
36                 codec = HFI_VIDEO_CODEC_H264;
37                 break;
38         case V4L2_PIX_FMT_H263:
39                 codec = HFI_VIDEO_CODEC_H263;
40                 break;
41         case V4L2_PIX_FMT_MPEG1:
42                 codec = HFI_VIDEO_CODEC_MPEG1;
43                 break;
44         case V4L2_PIX_FMT_MPEG2:
45                 codec = HFI_VIDEO_CODEC_MPEG2;
46                 break;
47         case V4L2_PIX_FMT_MPEG4:
48                 codec = HFI_VIDEO_CODEC_MPEG4;
49                 break;
50         case V4L2_PIX_FMT_VC1_ANNEX_G:
51         case V4L2_PIX_FMT_VC1_ANNEX_L:
52                 codec = HFI_VIDEO_CODEC_VC1;
53                 break;
54         case V4L2_PIX_FMT_VP8:
55                 codec = HFI_VIDEO_CODEC_VP8;
56                 break;
57         case V4L2_PIX_FMT_VP9:
58                 codec = HFI_VIDEO_CODEC_VP9;
59                 break;
60         case V4L2_PIX_FMT_XVID:
61                 codec = HFI_VIDEO_CODEC_DIVX;
62                 break;
63         case V4L2_PIX_FMT_HEVC:
64                 codec = HFI_VIDEO_CODEC_HEVC;
65                 break;
66         default:
67                 return false;
68         }
69
70         if (session_type == VIDC_SESSION_TYPE_ENC && core->enc_codecs & codec)
71                 return true;
72
73         if (session_type == VIDC_SESSION_TYPE_DEC && core->dec_codecs & codec)
74                 return true;
75
76         return false;
77 }
78 EXPORT_SYMBOL_GPL(venus_helper_check_codec);
79
80 int venus_helper_queue_dpb_bufs(struct venus_inst *inst)
81 {
82         struct intbuf *buf;
83         int ret = 0;
84
85         list_for_each_entry(buf, &inst->dpbbufs, list) {
86                 struct hfi_frame_data fdata;
87
88                 memset(&fdata, 0, sizeof(fdata));
89                 fdata.alloc_len = buf->size;
90                 fdata.device_addr = buf->da;
91                 fdata.buffer_type = buf->type;
92
93                 ret = hfi_session_process_buf(inst, &fdata);
94                 if (ret)
95                         goto fail;
96         }
97
98 fail:
99         return ret;
100 }
101 EXPORT_SYMBOL_GPL(venus_helper_queue_dpb_bufs);
102
103 int venus_helper_free_dpb_bufs(struct venus_inst *inst)
104 {
105         struct intbuf *buf, *n;
106
107         list_for_each_entry_safe(buf, n, &inst->dpbbufs, list) {
108                 list_del_init(&buf->list);
109                 dma_free_attrs(inst->core->dev, buf->size, buf->va, buf->da,
110                                buf->attrs);
111                 kfree(buf);
112         }
113
114         INIT_LIST_HEAD(&inst->dpbbufs);
115
116         return 0;
117 }
118 EXPORT_SYMBOL_GPL(venus_helper_free_dpb_bufs);
119
120 int venus_helper_alloc_dpb_bufs(struct venus_inst *inst)
121 {
122         struct venus_core *core = inst->core;
123         struct device *dev = core->dev;
124         enum hfi_version ver = core->res->hfi_version;
125         struct hfi_buffer_requirements bufreq;
126         u32 buftype = inst->dpb_buftype;
127         unsigned int dpb_size = 0;
128         struct intbuf *buf;
129         unsigned int i;
130         u32 count;
131         int ret;
132
133         /* no need to allocate dpb buffers */
134         if (!inst->dpb_fmt)
135                 return 0;
136
137         if (inst->dpb_buftype == HFI_BUFFER_OUTPUT)
138                 dpb_size = inst->output_buf_size;
139         else if (inst->dpb_buftype == HFI_BUFFER_OUTPUT2)
140                 dpb_size = inst->output2_buf_size;
141
142         if (!dpb_size)
143                 return 0;
144
145         ret = venus_helper_get_bufreq(inst, buftype, &bufreq);
146         if (ret)
147                 return ret;
148
149         count = HFI_BUFREQ_COUNT_MIN(&bufreq, ver);
150
151         for (i = 0; i < count; i++) {
152                 buf = kzalloc(sizeof(*buf), GFP_KERNEL);
153                 if (!buf) {
154                         ret = -ENOMEM;
155                         goto fail;
156                 }
157
158                 buf->type = buftype;
159                 buf->size = dpb_size;
160                 buf->attrs = DMA_ATTR_WRITE_COMBINE |
161                              DMA_ATTR_NO_KERNEL_MAPPING;
162                 buf->va = dma_alloc_attrs(dev, buf->size, &buf->da, GFP_KERNEL,
163                                           buf->attrs);
164                 if (!buf->va) {
165                         kfree(buf);
166                         ret = -ENOMEM;
167                         goto fail;
168                 }
169
170                 list_add_tail(&buf->list, &inst->dpbbufs);
171         }
172
173         return 0;
174
175 fail:
176         venus_helper_free_dpb_bufs(inst);
177         return ret;
178 }
179 EXPORT_SYMBOL_GPL(venus_helper_alloc_dpb_bufs);
180
181 static int intbufs_set_buffer(struct venus_inst *inst, u32 type)
182 {
183         struct venus_core *core = inst->core;
184         struct device *dev = core->dev;
185         struct hfi_buffer_requirements bufreq;
186         struct hfi_buffer_desc bd;
187         struct intbuf *buf;
188         unsigned int i;
189         int ret;
190
191         ret = venus_helper_get_bufreq(inst, type, &bufreq);
192         if (ret)
193                 return 0;
194
195         if (!bufreq.size)
196                 return 0;
197
198         for (i = 0; i < bufreq.count_actual; i++) {
199                 buf = kzalloc(sizeof(*buf), GFP_KERNEL);
200                 if (!buf) {
201                         ret = -ENOMEM;
202                         goto fail;
203                 }
204
205                 buf->type = bufreq.type;
206                 buf->size = bufreq.size;
207                 buf->attrs = DMA_ATTR_WRITE_COMBINE |
208                              DMA_ATTR_NO_KERNEL_MAPPING;
209                 buf->va = dma_alloc_attrs(dev, buf->size, &buf->da, GFP_KERNEL,
210                                           buf->attrs);
211                 if (!buf->va) {
212                         ret = -ENOMEM;
213                         goto fail;
214                 }
215
216                 memset(&bd, 0, sizeof(bd));
217                 bd.buffer_size = buf->size;
218                 bd.buffer_type = buf->type;
219                 bd.num_buffers = 1;
220                 bd.device_addr = buf->da;
221
222                 ret = hfi_session_set_buffers(inst, &bd);
223                 if (ret) {
224                         dev_err(dev, "set session buffers failed\n");
225                         goto dma_free;
226                 }
227
228                 list_add_tail(&buf->list, &inst->internalbufs);
229         }
230
231         return 0;
232
233 dma_free:
234         dma_free_attrs(dev, buf->size, buf->va, buf->da, buf->attrs);
235 fail:
236         kfree(buf);
237         return ret;
238 }
239
240 static int intbufs_unset_buffers(struct venus_inst *inst)
241 {
242         struct hfi_buffer_desc bd = {0};
243         struct intbuf *buf, *n;
244         int ret = 0;
245
246         list_for_each_entry_safe(buf, n, &inst->internalbufs, list) {
247                 bd.buffer_size = buf->size;
248                 bd.buffer_type = buf->type;
249                 bd.num_buffers = 1;
250                 bd.device_addr = buf->da;
251                 bd.response_required = true;
252
253                 ret = hfi_session_unset_buffers(inst, &bd);
254
255                 list_del_init(&buf->list);
256                 dma_free_attrs(inst->core->dev, buf->size, buf->va, buf->da,
257                                buf->attrs);
258                 kfree(buf);
259         }
260
261         return ret;
262 }
263
264 static const unsigned int intbuf_types_1xx[] = {
265         HFI_BUFFER_INTERNAL_SCRATCH(HFI_VERSION_1XX),
266         HFI_BUFFER_INTERNAL_SCRATCH_1(HFI_VERSION_1XX),
267         HFI_BUFFER_INTERNAL_SCRATCH_2(HFI_VERSION_1XX),
268         HFI_BUFFER_INTERNAL_PERSIST,
269         HFI_BUFFER_INTERNAL_PERSIST_1,
270 };
271
272 static const unsigned int intbuf_types_4xx[] = {
273         HFI_BUFFER_INTERNAL_SCRATCH(HFI_VERSION_4XX),
274         HFI_BUFFER_INTERNAL_SCRATCH_1(HFI_VERSION_4XX),
275         HFI_BUFFER_INTERNAL_SCRATCH_2(HFI_VERSION_4XX),
276         HFI_BUFFER_INTERNAL_PERSIST,
277         HFI_BUFFER_INTERNAL_PERSIST_1,
278 };
279
280 int venus_helper_intbufs_alloc(struct venus_inst *inst)
281 {
282         const unsigned int *intbuf;
283         size_t arr_sz, i;
284         int ret;
285
286         if (IS_V4(inst->core)) {
287                 arr_sz = ARRAY_SIZE(intbuf_types_4xx);
288                 intbuf = intbuf_types_4xx;
289         } else {
290                 arr_sz = ARRAY_SIZE(intbuf_types_1xx);
291                 intbuf = intbuf_types_1xx;
292         }
293
294         for (i = 0; i < arr_sz; i++) {
295                 ret = intbufs_set_buffer(inst, intbuf[i]);
296                 if (ret)
297                         goto error;
298         }
299
300         return 0;
301
302 error:
303         intbufs_unset_buffers(inst);
304         return ret;
305 }
306 EXPORT_SYMBOL_GPL(venus_helper_intbufs_alloc);
307
308 int venus_helper_intbufs_free(struct venus_inst *inst)
309 {
310         return intbufs_unset_buffers(inst);
311 }
312 EXPORT_SYMBOL_GPL(venus_helper_intbufs_free);
313
314 int venus_helper_intbufs_realloc(struct venus_inst *inst)
315 {
316         enum hfi_version ver = inst->core->res->hfi_version;
317         struct hfi_buffer_desc bd;
318         struct intbuf *buf, *n;
319         int ret;
320
321         list_for_each_entry_safe(buf, n, &inst->internalbufs, list) {
322                 if (buf->type == HFI_BUFFER_INTERNAL_PERSIST ||
323                     buf->type == HFI_BUFFER_INTERNAL_PERSIST_1)
324                         continue;
325
326                 memset(&bd, 0, sizeof(bd));
327                 bd.buffer_size = buf->size;
328                 bd.buffer_type = buf->type;
329                 bd.num_buffers = 1;
330                 bd.device_addr = buf->da;
331                 bd.response_required = true;
332
333                 ret = hfi_session_unset_buffers(inst, &bd);
334
335                 dma_free_attrs(inst->core->dev, buf->size, buf->va, buf->da,
336                                buf->attrs);
337
338                 list_del_init(&buf->list);
339                 kfree(buf);
340         }
341
342         ret = intbufs_set_buffer(inst, HFI_BUFFER_INTERNAL_SCRATCH(ver));
343         if (ret)
344                 goto err;
345
346         ret = intbufs_set_buffer(inst, HFI_BUFFER_INTERNAL_SCRATCH_1(ver));
347         if (ret)
348                 goto err;
349
350         ret = intbufs_set_buffer(inst, HFI_BUFFER_INTERNAL_SCRATCH_2(ver));
351         if (ret)
352                 goto err;
353
354         return 0;
355 err:
356         return ret;
357 }
358 EXPORT_SYMBOL_GPL(venus_helper_intbufs_realloc);
359
360 static void fill_buffer_desc(const struct venus_buffer *buf,
361                              struct hfi_buffer_desc *bd, bool response)
362 {
363         memset(bd, 0, sizeof(*bd));
364         bd->buffer_type = HFI_BUFFER_OUTPUT;
365         bd->buffer_size = buf->size;
366         bd->num_buffers = 1;
367         bd->device_addr = buf->dma_addr;
368         bd->response_required = response;
369 }
370
371 static void return_buf_error(struct venus_inst *inst,
372                              struct vb2_v4l2_buffer *vbuf)
373 {
374         struct v4l2_m2m_ctx *m2m_ctx = inst->m2m_ctx;
375
376         if (vbuf->vb2_buf.type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE)
377                 v4l2_m2m_src_buf_remove_by_buf(m2m_ctx, vbuf);
378         else
379                 v4l2_m2m_dst_buf_remove_by_buf(m2m_ctx, vbuf);
380
381         v4l2_m2m_buf_done(vbuf, VB2_BUF_STATE_ERROR);
382 }
383
384 static void
385 put_ts_metadata(struct venus_inst *inst, struct vb2_v4l2_buffer *vbuf)
386 {
387         struct vb2_buffer *vb = &vbuf->vb2_buf;
388         unsigned int i;
389         int slot = -1;
390         u64 ts_us = vb->timestamp;
391
392         for (i = 0; i < ARRAY_SIZE(inst->tss); i++) {
393                 if (!inst->tss[i].used) {
394                         slot = i;
395                         break;
396                 }
397         }
398
399         if (slot == -1) {
400                 dev_dbg(inst->core->dev, VDBGL "no free slot\n");
401                 return;
402         }
403
404         do_div(ts_us, NSEC_PER_USEC);
405
406         inst->tss[slot].used = true;
407         inst->tss[slot].flags = vbuf->flags;
408         inst->tss[slot].tc = vbuf->timecode;
409         inst->tss[slot].ts_us = ts_us;
410         inst->tss[slot].ts_ns = vb->timestamp;
411 }
412
413 void venus_helper_get_ts_metadata(struct venus_inst *inst, u64 timestamp_us,
414                                   struct vb2_v4l2_buffer *vbuf)
415 {
416         struct vb2_buffer *vb = &vbuf->vb2_buf;
417         unsigned int i;
418
419         for (i = 0; i < ARRAY_SIZE(inst->tss); ++i) {
420                 if (!inst->tss[i].used)
421                         continue;
422
423                 if (inst->tss[i].ts_us != timestamp_us)
424                         continue;
425
426                 inst->tss[i].used = false;
427                 vbuf->flags |= inst->tss[i].flags;
428                 vbuf->timecode = inst->tss[i].tc;
429                 vb->timestamp = inst->tss[i].ts_ns;
430                 break;
431         }
432 }
433 EXPORT_SYMBOL_GPL(venus_helper_get_ts_metadata);
434
435 static int
436 session_process_buf(struct venus_inst *inst, struct vb2_v4l2_buffer *vbuf)
437 {
438         struct venus_buffer *buf = to_venus_buffer(vbuf);
439         struct vb2_buffer *vb = &vbuf->vb2_buf;
440         unsigned int type = vb->type;
441         struct hfi_frame_data fdata;
442         int ret;
443
444         memset(&fdata, 0, sizeof(fdata));
445         fdata.alloc_len = buf->size;
446         fdata.device_addr = buf->dma_addr;
447         fdata.timestamp = vb->timestamp;
448         do_div(fdata.timestamp, NSEC_PER_USEC);
449         fdata.flags = 0;
450         fdata.clnt_data = vbuf->vb2_buf.index;
451
452         if (type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) {
453                 fdata.buffer_type = HFI_BUFFER_INPUT;
454                 fdata.filled_len = vb2_get_plane_payload(vb, 0);
455                 fdata.offset = vb->planes[0].data_offset;
456
457                 if (vbuf->flags & V4L2_BUF_FLAG_LAST || !fdata.filled_len)
458                         fdata.flags |= HFI_BUFFERFLAG_EOS;
459
460                 if (inst->session_type == VIDC_SESSION_TYPE_DEC)
461                         put_ts_metadata(inst, vbuf);
462
463                 venus_pm_load_scale(inst);
464         } else if (type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE) {
465                 if (inst->session_type == VIDC_SESSION_TYPE_ENC)
466                         fdata.buffer_type = HFI_BUFFER_OUTPUT;
467                 else
468                         fdata.buffer_type = inst->opb_buftype;
469                 fdata.filled_len = 0;
470                 fdata.offset = 0;
471         }
472
473         ret = hfi_session_process_buf(inst, &fdata);
474         if (ret)
475                 return ret;
476
477         return 0;
478 }
479
480 static bool is_dynamic_bufmode(struct venus_inst *inst)
481 {
482         struct venus_core *core = inst->core;
483         struct venus_caps *caps;
484
485         /*
486          * v4 doesn't send BUFFER_ALLOC_MODE_SUPPORTED property and supports
487          * dynamic buffer mode by default for HFI_BUFFER_OUTPUT/OUTPUT2.
488          */
489         if (IS_V4(core))
490                 return true;
491
492         caps = venus_caps_by_codec(core, inst->hfi_codec, inst->session_type);
493         if (!caps)
494                 return false;
495
496         return caps->cap_bufs_mode_dynamic;
497 }
498
499 int venus_helper_unregister_bufs(struct venus_inst *inst)
500 {
501         struct venus_buffer *buf, *n;
502         struct hfi_buffer_desc bd;
503         int ret = 0;
504
505         if (is_dynamic_bufmode(inst))
506                 return 0;
507
508         list_for_each_entry_safe(buf, n, &inst->registeredbufs, reg_list) {
509                 fill_buffer_desc(buf, &bd, true);
510                 ret = hfi_session_unset_buffers(inst, &bd);
511                 list_del_init(&buf->reg_list);
512         }
513
514         return ret;
515 }
516 EXPORT_SYMBOL_GPL(venus_helper_unregister_bufs);
517
518 static int session_register_bufs(struct venus_inst *inst)
519 {
520         struct venus_core *core = inst->core;
521         struct device *dev = core->dev;
522         struct hfi_buffer_desc bd;
523         struct venus_buffer *buf;
524         int ret = 0;
525
526         if (is_dynamic_bufmode(inst))
527                 return 0;
528
529         list_for_each_entry(buf, &inst->registeredbufs, reg_list) {
530                 fill_buffer_desc(buf, &bd, false);
531                 ret = hfi_session_set_buffers(inst, &bd);
532                 if (ret) {
533                         dev_err(dev, "%s: set buffer failed\n", __func__);
534                         break;
535                 }
536         }
537
538         return ret;
539 }
540
541 static u32 to_hfi_raw_fmt(u32 v4l2_fmt)
542 {
543         switch (v4l2_fmt) {
544         case V4L2_PIX_FMT_NV12:
545                 return HFI_COLOR_FORMAT_NV12;
546         case V4L2_PIX_FMT_NV21:
547                 return HFI_COLOR_FORMAT_NV21;
548         default:
549                 break;
550         }
551
552         return 0;
553 }
554
555 int venus_helper_get_bufreq(struct venus_inst *inst, u32 type,
556                             struct hfi_buffer_requirements *req)
557 {
558         u32 ptype = HFI_PROPERTY_CONFIG_BUFFER_REQUIREMENTS;
559         union hfi_get_property hprop;
560         unsigned int i;
561         int ret;
562
563         if (req)
564                 memset(req, 0, sizeof(*req));
565
566         ret = hfi_session_get_property(inst, ptype, &hprop);
567         if (ret)
568                 return ret;
569
570         ret = -EINVAL;
571
572         for (i = 0; i < HFI_BUFFER_TYPE_MAX; i++) {
573                 if (hprop.bufreq[i].type != type)
574                         continue;
575
576                 if (req)
577                         memcpy(req, &hprop.bufreq[i], sizeof(*req));
578                 ret = 0;
579                 break;
580         }
581
582         return ret;
583 }
584 EXPORT_SYMBOL_GPL(venus_helper_get_bufreq);
585
586 struct id_mapping {
587         u32 hfi_id;
588         u32 v4l2_id;
589 };
590
591 static const struct id_mapping mpeg4_profiles[] = {
592         { HFI_MPEG4_PROFILE_SIMPLE, V4L2_MPEG_VIDEO_MPEG4_PROFILE_SIMPLE },
593         { HFI_MPEG4_PROFILE_ADVANCEDSIMPLE, V4L2_MPEG_VIDEO_MPEG4_PROFILE_ADVANCED_SIMPLE },
594 };
595
596 static const struct id_mapping mpeg4_levels[] = {
597         { HFI_MPEG4_LEVEL_0, V4L2_MPEG_VIDEO_MPEG4_LEVEL_0 },
598         { HFI_MPEG4_LEVEL_0b, V4L2_MPEG_VIDEO_MPEG4_LEVEL_0B },
599         { HFI_MPEG4_LEVEL_1, V4L2_MPEG_VIDEO_MPEG4_LEVEL_1 },
600         { HFI_MPEG4_LEVEL_2, V4L2_MPEG_VIDEO_MPEG4_LEVEL_2 },
601         { HFI_MPEG4_LEVEL_3, V4L2_MPEG_VIDEO_MPEG4_LEVEL_3 },
602         { HFI_MPEG4_LEVEL_4, V4L2_MPEG_VIDEO_MPEG4_LEVEL_4 },
603         { HFI_MPEG4_LEVEL_5, V4L2_MPEG_VIDEO_MPEG4_LEVEL_5 },
604 };
605
606 static const struct id_mapping mpeg2_profiles[] = {
607         { HFI_MPEG2_PROFILE_SIMPLE, V4L2_MPEG_VIDEO_MPEG2_PROFILE_SIMPLE },
608         { HFI_MPEG2_PROFILE_MAIN, V4L2_MPEG_VIDEO_MPEG2_PROFILE_MAIN },
609         { HFI_MPEG2_PROFILE_SNR, V4L2_MPEG_VIDEO_MPEG2_PROFILE_SNR_SCALABLE },
610         { HFI_MPEG2_PROFILE_SPATIAL, V4L2_MPEG_VIDEO_MPEG2_PROFILE_SPATIALLY_SCALABLE },
611         { HFI_MPEG2_PROFILE_HIGH, V4L2_MPEG_VIDEO_MPEG2_PROFILE_HIGH },
612 };
613
614 static const struct id_mapping mpeg2_levels[] = {
615         { HFI_MPEG2_LEVEL_LL, V4L2_MPEG_VIDEO_MPEG2_LEVEL_LOW },
616         { HFI_MPEG2_LEVEL_ML, V4L2_MPEG_VIDEO_MPEG2_LEVEL_MAIN },
617         { HFI_MPEG2_LEVEL_H14, V4L2_MPEG_VIDEO_MPEG2_LEVEL_HIGH_1440 },
618         { HFI_MPEG2_LEVEL_HL, V4L2_MPEG_VIDEO_MPEG2_LEVEL_HIGH },
619 };
620
621 static const struct id_mapping h264_profiles[] = {
622         { HFI_H264_PROFILE_BASELINE, V4L2_MPEG_VIDEO_H264_PROFILE_BASELINE },
623         { HFI_H264_PROFILE_MAIN, V4L2_MPEG_VIDEO_H264_PROFILE_MAIN },
624         { HFI_H264_PROFILE_HIGH, V4L2_MPEG_VIDEO_H264_PROFILE_HIGH },
625         { HFI_H264_PROFILE_STEREO_HIGH, V4L2_MPEG_VIDEO_H264_PROFILE_STEREO_HIGH },
626         { HFI_H264_PROFILE_MULTIVIEW_HIGH, V4L2_MPEG_VIDEO_H264_PROFILE_MULTIVIEW_HIGH },
627         { HFI_H264_PROFILE_CONSTRAINED_BASE, V4L2_MPEG_VIDEO_H264_PROFILE_CONSTRAINED_BASELINE },
628         { HFI_H264_PROFILE_CONSTRAINED_HIGH, V4L2_MPEG_VIDEO_H264_PROFILE_CONSTRAINED_HIGH },
629 };
630
631 static const struct id_mapping h264_levels[] = {
632         { HFI_H264_LEVEL_1, V4L2_MPEG_VIDEO_H264_LEVEL_1_0 },
633         { HFI_H264_LEVEL_1b, V4L2_MPEG_VIDEO_H264_LEVEL_1B },
634         { HFI_H264_LEVEL_11, V4L2_MPEG_VIDEO_H264_LEVEL_1_1 },
635         { HFI_H264_LEVEL_12, V4L2_MPEG_VIDEO_H264_LEVEL_1_2 },
636         { HFI_H264_LEVEL_13, V4L2_MPEG_VIDEO_H264_LEVEL_1_3 },
637         { HFI_H264_LEVEL_2, V4L2_MPEG_VIDEO_H264_LEVEL_2_0 },
638         { HFI_H264_LEVEL_21, V4L2_MPEG_VIDEO_H264_LEVEL_2_1 },
639         { HFI_H264_LEVEL_22, V4L2_MPEG_VIDEO_H264_LEVEL_2_2 },
640         { HFI_H264_LEVEL_3, V4L2_MPEG_VIDEO_H264_LEVEL_3_0 },
641         { HFI_H264_LEVEL_31, V4L2_MPEG_VIDEO_H264_LEVEL_3_1 },
642         { HFI_H264_LEVEL_32, V4L2_MPEG_VIDEO_H264_LEVEL_3_2 },
643         { HFI_H264_LEVEL_4, V4L2_MPEG_VIDEO_H264_LEVEL_4_0 },
644         { HFI_H264_LEVEL_41, V4L2_MPEG_VIDEO_H264_LEVEL_4_1 },
645         { HFI_H264_LEVEL_42, V4L2_MPEG_VIDEO_H264_LEVEL_4_2 },
646         { HFI_H264_LEVEL_5, V4L2_MPEG_VIDEO_H264_LEVEL_5_0 },
647         { HFI_H264_LEVEL_51, V4L2_MPEG_VIDEO_H264_LEVEL_5_1 },
648         { HFI_H264_LEVEL_52, V4L2_MPEG_VIDEO_H264_LEVEL_5_1 },
649 };
650
651 static const struct id_mapping hevc_profiles[] = {
652         { HFI_HEVC_PROFILE_MAIN, V4L2_MPEG_VIDEO_HEVC_PROFILE_MAIN },
653         { HFI_HEVC_PROFILE_MAIN_STILL_PIC, V4L2_MPEG_VIDEO_HEVC_PROFILE_MAIN_STILL_PICTURE },
654         { HFI_HEVC_PROFILE_MAIN10, V4L2_MPEG_VIDEO_HEVC_PROFILE_MAIN_10 },
655 };
656
657 static const struct id_mapping hevc_levels[] = {
658         { HFI_HEVC_LEVEL_1, V4L2_MPEG_VIDEO_HEVC_LEVEL_1 },
659         { HFI_HEVC_LEVEL_2, V4L2_MPEG_VIDEO_HEVC_LEVEL_2 },
660         { HFI_HEVC_LEVEL_21, V4L2_MPEG_VIDEO_HEVC_LEVEL_2_1 },
661         { HFI_HEVC_LEVEL_3, V4L2_MPEG_VIDEO_HEVC_LEVEL_3 },
662         { HFI_HEVC_LEVEL_31, V4L2_MPEG_VIDEO_HEVC_LEVEL_3_1 },
663         { HFI_HEVC_LEVEL_4, V4L2_MPEG_VIDEO_HEVC_LEVEL_4 },
664         { HFI_HEVC_LEVEL_41, V4L2_MPEG_VIDEO_HEVC_LEVEL_4_1 },
665         { HFI_HEVC_LEVEL_5, V4L2_MPEG_VIDEO_HEVC_LEVEL_5 },
666         { HFI_HEVC_LEVEL_51, V4L2_MPEG_VIDEO_HEVC_LEVEL_5_1 },
667         { HFI_HEVC_LEVEL_52, V4L2_MPEG_VIDEO_HEVC_LEVEL_5_2 },
668         { HFI_HEVC_LEVEL_6, V4L2_MPEG_VIDEO_HEVC_LEVEL_6 },
669         { HFI_HEVC_LEVEL_61, V4L2_MPEG_VIDEO_HEVC_LEVEL_6_1 },
670         { HFI_HEVC_LEVEL_62, V4L2_MPEG_VIDEO_HEVC_LEVEL_6_2 },
671 };
672
673 static const struct id_mapping vp8_profiles[] = {
674         { HFI_VPX_PROFILE_VERSION_0, V4L2_MPEG_VIDEO_VP8_PROFILE_0 },
675         { HFI_VPX_PROFILE_VERSION_1, V4L2_MPEG_VIDEO_VP8_PROFILE_1 },
676         { HFI_VPX_PROFILE_VERSION_2, V4L2_MPEG_VIDEO_VP8_PROFILE_2 },
677         { HFI_VPX_PROFILE_VERSION_3, V4L2_MPEG_VIDEO_VP8_PROFILE_3 },
678 };
679
680 static const struct id_mapping vp9_profiles[] = {
681         { HFI_VP9_PROFILE_P0, V4L2_MPEG_VIDEO_VP9_PROFILE_0 },
682         { HFI_VP9_PROFILE_P2_10B, V4L2_MPEG_VIDEO_VP9_PROFILE_2 },
683 };
684
685 static const struct id_mapping vp9_levels[] = {
686         { HFI_VP9_LEVEL_1, V4L2_MPEG_VIDEO_VP9_LEVEL_1_0 },
687         { HFI_VP9_LEVEL_11, V4L2_MPEG_VIDEO_VP9_LEVEL_1_1 },
688         { HFI_VP9_LEVEL_2, V4L2_MPEG_VIDEO_VP9_LEVEL_2_0},
689         { HFI_VP9_LEVEL_21, V4L2_MPEG_VIDEO_VP9_LEVEL_2_1 },
690         { HFI_VP9_LEVEL_3, V4L2_MPEG_VIDEO_VP9_LEVEL_3_0},
691         { HFI_VP9_LEVEL_31, V4L2_MPEG_VIDEO_VP9_LEVEL_3_1 },
692         { HFI_VP9_LEVEL_4, V4L2_MPEG_VIDEO_VP9_LEVEL_4_0 },
693         { HFI_VP9_LEVEL_41, V4L2_MPEG_VIDEO_VP9_LEVEL_4_1 },
694         { HFI_VP9_LEVEL_5, V4L2_MPEG_VIDEO_VP9_LEVEL_5_0 },
695         { HFI_VP9_LEVEL_51, V4L2_MPEG_VIDEO_VP9_LEVEL_5_1 },
696         { HFI_VP9_LEVEL_6, V4L2_MPEG_VIDEO_VP9_LEVEL_6_0 },
697         { HFI_VP9_LEVEL_61, V4L2_MPEG_VIDEO_VP9_LEVEL_6_1 },
698 };
699
700 static u32 find_v4l2_id(u32 hfi_id, const struct id_mapping *array, unsigned int array_sz)
701 {
702         unsigned int i;
703
704         if (!array || !array_sz)
705                 return 0;
706
707         for (i = 0; i < array_sz; i++)
708                 if (hfi_id == array[i].hfi_id)
709                         return array[i].v4l2_id;
710
711         return 0;
712 }
713
714 static u32 find_hfi_id(u32 v4l2_id, const struct id_mapping *array, unsigned int array_sz)
715 {
716         unsigned int i;
717
718         if (!array || !array_sz)
719                 return 0;
720
721         for (i = 0; i < array_sz; i++)
722                 if (v4l2_id == array[i].v4l2_id)
723                         return array[i].hfi_id;
724
725         return 0;
726 }
727
728 static void
729 v4l2_id_profile_level(u32 hfi_codec, struct hfi_profile_level *pl, u32 *profile, u32 *level)
730 {
731         u32 hfi_pf = pl->profile;
732         u32 hfi_lvl = pl->level;
733
734         switch (hfi_codec) {
735         case HFI_VIDEO_CODEC_H264:
736                 *profile = find_v4l2_id(hfi_pf, h264_profiles, ARRAY_SIZE(h264_profiles));
737                 *level = find_v4l2_id(hfi_lvl, h264_levels, ARRAY_SIZE(h264_levels));
738                 break;
739         case HFI_VIDEO_CODEC_MPEG2:
740                 *profile = find_v4l2_id(hfi_pf, mpeg2_profiles, ARRAY_SIZE(mpeg2_profiles));
741                 *level = find_v4l2_id(hfi_lvl, mpeg2_levels, ARRAY_SIZE(mpeg2_levels));
742                 break;
743         case HFI_VIDEO_CODEC_MPEG4:
744                 *profile = find_v4l2_id(hfi_pf, mpeg4_profiles, ARRAY_SIZE(mpeg4_profiles));
745                 *level = find_v4l2_id(hfi_lvl, mpeg4_levels, ARRAY_SIZE(mpeg4_levels));
746                 break;
747         case HFI_VIDEO_CODEC_VP8:
748                 *profile = find_v4l2_id(hfi_pf, vp8_profiles, ARRAY_SIZE(vp8_profiles));
749                 *level = 0;
750                 break;
751         case HFI_VIDEO_CODEC_VP9:
752                 *profile = find_v4l2_id(hfi_pf, vp9_profiles, ARRAY_SIZE(vp9_profiles));
753                 *level = find_v4l2_id(hfi_lvl, vp9_levels, ARRAY_SIZE(vp9_levels));
754                 break;
755         case HFI_VIDEO_CODEC_HEVC:
756                 *profile = find_v4l2_id(hfi_pf, hevc_profiles, ARRAY_SIZE(hevc_profiles));
757                 *level = find_v4l2_id(hfi_lvl, hevc_levels, ARRAY_SIZE(hevc_levels));
758                 break;
759         default:
760                 break;
761         }
762 }
763
764 static void
765 hfi_id_profile_level(u32 hfi_codec, u32 v4l2_pf, u32 v4l2_lvl, struct hfi_profile_level *pl)
766 {
767         switch (hfi_codec) {
768         case HFI_VIDEO_CODEC_H264:
769                 pl->profile = find_hfi_id(v4l2_pf, h264_profiles, ARRAY_SIZE(h264_profiles));
770                 pl->level = find_hfi_id(v4l2_lvl, h264_levels, ARRAY_SIZE(h264_levels));
771                 break;
772         case HFI_VIDEO_CODEC_MPEG2:
773                 pl->profile = find_hfi_id(v4l2_pf, mpeg2_profiles, ARRAY_SIZE(mpeg2_profiles));
774                 pl->level = find_hfi_id(v4l2_lvl, mpeg2_levels, ARRAY_SIZE(mpeg2_levels));
775                 break;
776         case HFI_VIDEO_CODEC_MPEG4:
777                 pl->profile = find_hfi_id(v4l2_pf, mpeg4_profiles, ARRAY_SIZE(mpeg4_profiles));
778                 pl->level = find_hfi_id(v4l2_lvl, mpeg4_levels, ARRAY_SIZE(mpeg4_levels));
779                 break;
780         case HFI_VIDEO_CODEC_VP8:
781                 pl->profile = find_hfi_id(v4l2_pf, vp8_profiles, ARRAY_SIZE(vp8_profiles));
782                 pl->level = 0;
783                 break;
784         case HFI_VIDEO_CODEC_VP9:
785                 pl->profile = find_hfi_id(v4l2_pf, vp9_profiles, ARRAY_SIZE(vp9_profiles));
786                 pl->level = find_hfi_id(v4l2_lvl, vp9_levels, ARRAY_SIZE(vp9_levels));
787                 break;
788         case HFI_VIDEO_CODEC_HEVC:
789                 pl->profile = find_hfi_id(v4l2_pf, hevc_profiles, ARRAY_SIZE(hevc_profiles));
790                 pl->level = find_hfi_id(v4l2_lvl, hevc_levels, ARRAY_SIZE(hevc_levels));
791                 break;
792         default:
793                 break;
794         }
795 }
796
797 int venus_helper_get_profile_level(struct venus_inst *inst, u32 *profile, u32 *level)
798 {
799         const u32 ptype = HFI_PROPERTY_PARAM_PROFILE_LEVEL_CURRENT;
800         union hfi_get_property hprop;
801         int ret;
802
803         ret = hfi_session_get_property(inst, ptype, &hprop);
804         if (ret)
805                 return ret;
806
807         v4l2_id_profile_level(inst->hfi_codec, &hprop.profile_level, profile, level);
808
809         return 0;
810 }
811 EXPORT_SYMBOL_GPL(venus_helper_get_profile_level);
812
813 int venus_helper_set_profile_level(struct venus_inst *inst, u32 profile, u32 level)
814 {
815         const u32 ptype = HFI_PROPERTY_PARAM_PROFILE_LEVEL_CURRENT;
816         struct hfi_profile_level pl;
817
818         hfi_id_profile_level(inst->hfi_codec, profile, level, &pl);
819
820         return hfi_session_set_property(inst, ptype, &pl);
821 }
822 EXPORT_SYMBOL_GPL(venus_helper_set_profile_level);
823
824 static u32 get_framesize_raw_nv12(u32 width, u32 height)
825 {
826         u32 y_stride, uv_stride, y_plane;
827         u32 y_sclines, uv_sclines, uv_plane;
828         u32 size;
829
830         y_stride = ALIGN(width, 128);
831         uv_stride = ALIGN(width, 128);
832         y_sclines = ALIGN(height, 32);
833         uv_sclines = ALIGN(((height + 1) >> 1), 16);
834
835         y_plane = y_stride * y_sclines;
836         uv_plane = uv_stride * uv_sclines + SZ_4K;
837         size = y_plane + uv_plane + SZ_8K;
838
839         return ALIGN(size, SZ_4K);
840 }
841
842 static u32 get_framesize_raw_nv12_ubwc(u32 width, u32 height)
843 {
844         u32 y_meta_stride, y_meta_plane;
845         u32 y_stride, y_plane;
846         u32 uv_meta_stride, uv_meta_plane;
847         u32 uv_stride, uv_plane;
848         u32 extradata = SZ_16K;
849
850         y_meta_stride = ALIGN(DIV_ROUND_UP(width, 32), 64);
851         y_meta_plane = y_meta_stride * ALIGN(DIV_ROUND_UP(height, 8), 16);
852         y_meta_plane = ALIGN(y_meta_plane, SZ_4K);
853
854         y_stride = ALIGN(width, 128);
855         y_plane = ALIGN(y_stride * ALIGN(height, 32), SZ_4K);
856
857         uv_meta_stride = ALIGN(DIV_ROUND_UP(width / 2, 16), 64);
858         uv_meta_plane = uv_meta_stride * ALIGN(DIV_ROUND_UP(height / 2, 8), 16);
859         uv_meta_plane = ALIGN(uv_meta_plane, SZ_4K);
860
861         uv_stride = ALIGN(width, 128);
862         uv_plane = ALIGN(uv_stride * ALIGN(height / 2, 32), SZ_4K);
863
864         return ALIGN(y_meta_plane + y_plane + uv_meta_plane + uv_plane +
865                      max(extradata, y_stride * 48), SZ_4K);
866 }
867
868 static u32 get_framesize_raw_p010(u32 width, u32 height)
869 {
870         u32 y_plane, uv_plane, y_stride, uv_stride, y_sclines, uv_sclines;
871
872         y_stride = ALIGN(width * 2, 256);
873         uv_stride = ALIGN(width * 2, 256);
874         y_sclines = ALIGN(height, 32);
875         uv_sclines = ALIGN((height + 1) >> 1, 16);
876         y_plane = y_stride * y_sclines;
877         uv_plane = uv_stride * uv_sclines;
878
879         return ALIGN((y_plane + uv_plane), SZ_4K);
880 }
881
882 static u32 get_framesize_raw_p010_ubwc(u32 width, u32 height)
883 {
884         u32 y_stride, uv_stride, y_sclines, uv_sclines;
885         u32 y_ubwc_plane, uv_ubwc_plane;
886         u32 y_meta_stride, y_meta_scanlines;
887         u32 uv_meta_stride, uv_meta_scanlines;
888         u32 y_meta_plane, uv_meta_plane;
889         u32 size;
890
891         y_stride = ALIGN(width * 2, 256);
892         uv_stride = ALIGN(width * 2, 256);
893         y_sclines = ALIGN(height, 16);
894         uv_sclines = ALIGN((height + 1) >> 1, 16);
895
896         y_ubwc_plane = ALIGN(y_stride * y_sclines, SZ_4K);
897         uv_ubwc_plane = ALIGN(uv_stride * uv_sclines, SZ_4K);
898         y_meta_stride = ALIGN(DIV_ROUND_UP(width, 32), 64);
899         y_meta_scanlines = ALIGN(DIV_ROUND_UP(height, 4), 16);
900         y_meta_plane = ALIGN(y_meta_stride * y_meta_scanlines, SZ_4K);
901         uv_meta_stride = ALIGN(DIV_ROUND_UP((width + 1) >> 1, 16), 64);
902         uv_meta_scanlines = ALIGN(DIV_ROUND_UP((height + 1) >> 1, 4), 16);
903         uv_meta_plane = ALIGN(uv_meta_stride * uv_meta_scanlines, SZ_4K);
904
905         size = y_ubwc_plane + uv_ubwc_plane + y_meta_plane + uv_meta_plane;
906
907         return ALIGN(size, SZ_4K);
908 }
909
910 static u32 get_framesize_raw_yuv420_tp10_ubwc(u32 width, u32 height)
911 {
912         u32 y_stride, uv_stride, y_sclines, uv_sclines;
913         u32 y_ubwc_plane, uv_ubwc_plane;
914         u32 y_meta_stride, y_meta_scanlines;
915         u32 uv_meta_stride, uv_meta_scanlines;
916         u32 y_meta_plane, uv_meta_plane;
917         u32 extradata = SZ_16K;
918         u32 size;
919
920         y_stride = ALIGN(ALIGN(width, 192) * 4 / 3, 256);
921         uv_stride = ALIGN(ALIGN(width, 192) * 4 / 3, 256);
922         y_sclines = ALIGN(height, 16);
923         uv_sclines = ALIGN((height + 1) >> 1, 16);
924
925         y_ubwc_plane = ALIGN(y_stride * y_sclines, SZ_4K);
926         uv_ubwc_plane = ALIGN(uv_stride * uv_sclines, SZ_4K);
927         y_meta_stride = ALIGN(DIV_ROUND_UP(width, 48), 64);
928         y_meta_scanlines = ALIGN(DIV_ROUND_UP(height, 4), 16);
929         y_meta_plane = ALIGN(y_meta_stride * y_meta_scanlines, SZ_4K);
930         uv_meta_stride = ALIGN(DIV_ROUND_UP((width + 1) >> 1, 24), 64);
931         uv_meta_scanlines = ALIGN(DIV_ROUND_UP((height + 1) >> 1, 4), 16);
932         uv_meta_plane = ALIGN(uv_meta_stride * uv_meta_scanlines, SZ_4K);
933
934         size = y_ubwc_plane + uv_ubwc_plane + y_meta_plane + uv_meta_plane;
935         size += max(extradata + SZ_8K, y_stride * 48);
936
937         return ALIGN(size, SZ_4K);
938 }
939
940 u32 venus_helper_get_framesz_raw(u32 hfi_fmt, u32 width, u32 height)
941 {
942         switch (hfi_fmt) {
943         case HFI_COLOR_FORMAT_NV12:
944         case HFI_COLOR_FORMAT_NV21:
945                 return get_framesize_raw_nv12(width, height);
946         case HFI_COLOR_FORMAT_NV12_UBWC:
947                 return get_framesize_raw_nv12_ubwc(width, height);
948         case HFI_COLOR_FORMAT_P010:
949                 return get_framesize_raw_p010(width, height);
950         case HFI_COLOR_FORMAT_P010_UBWC:
951                 return get_framesize_raw_p010_ubwc(width, height);
952         case HFI_COLOR_FORMAT_YUV420_TP10_UBWC:
953                 return get_framesize_raw_yuv420_tp10_ubwc(width, height);
954         default:
955                 return 0;
956         }
957 }
958 EXPORT_SYMBOL_GPL(venus_helper_get_framesz_raw);
959
960 u32 venus_helper_get_framesz(u32 v4l2_fmt, u32 width, u32 height)
961 {
962         u32 hfi_fmt, sz;
963         bool compressed;
964
965         switch (v4l2_fmt) {
966         case V4L2_PIX_FMT_MPEG:
967         case V4L2_PIX_FMT_H264:
968         case V4L2_PIX_FMT_H264_NO_SC:
969         case V4L2_PIX_FMT_H264_MVC:
970         case V4L2_PIX_FMT_H263:
971         case V4L2_PIX_FMT_MPEG1:
972         case V4L2_PIX_FMT_MPEG2:
973         case V4L2_PIX_FMT_MPEG4:
974         case V4L2_PIX_FMT_XVID:
975         case V4L2_PIX_FMT_VC1_ANNEX_G:
976         case V4L2_PIX_FMT_VC1_ANNEX_L:
977         case V4L2_PIX_FMT_VP8:
978         case V4L2_PIX_FMT_VP9:
979         case V4L2_PIX_FMT_HEVC:
980                 compressed = true;
981                 break;
982         default:
983                 compressed = false;
984                 break;
985         }
986
987         if (compressed) {
988                 sz = ALIGN(height, 32) * ALIGN(width, 32) * 3 / 2 / 2;
989                 return ALIGN(sz, SZ_4K);
990         }
991
992         hfi_fmt = to_hfi_raw_fmt(v4l2_fmt);
993         if (!hfi_fmt)
994                 return 0;
995
996         return venus_helper_get_framesz_raw(hfi_fmt, width, height);
997 }
998 EXPORT_SYMBOL_GPL(venus_helper_get_framesz);
999
1000 int venus_helper_set_input_resolution(struct venus_inst *inst,
1001                                       unsigned int width, unsigned int height)
1002 {
1003         u32 ptype = HFI_PROPERTY_PARAM_FRAME_SIZE;
1004         struct hfi_framesize fs;
1005
1006         fs.buffer_type = HFI_BUFFER_INPUT;
1007         fs.width = width;
1008         fs.height = height;
1009
1010         return hfi_session_set_property(inst, ptype, &fs);
1011 }
1012 EXPORT_SYMBOL_GPL(venus_helper_set_input_resolution);
1013
1014 int venus_helper_set_output_resolution(struct venus_inst *inst,
1015                                        unsigned int width, unsigned int height,
1016                                        u32 buftype)
1017 {
1018         u32 ptype = HFI_PROPERTY_PARAM_FRAME_SIZE;
1019         struct hfi_framesize fs;
1020
1021         fs.buffer_type = buftype;
1022         fs.width = width;
1023         fs.height = height;
1024
1025         return hfi_session_set_property(inst, ptype, &fs);
1026 }
1027 EXPORT_SYMBOL_GPL(venus_helper_set_output_resolution);
1028
1029 int venus_helper_set_work_mode(struct venus_inst *inst, u32 mode)
1030 {
1031         const u32 ptype = HFI_PROPERTY_PARAM_WORK_MODE;
1032         struct hfi_video_work_mode wm;
1033
1034         if (!IS_V4(inst->core))
1035                 return 0;
1036
1037         wm.video_work_mode = mode;
1038
1039         return hfi_session_set_property(inst, ptype, &wm);
1040 }
1041 EXPORT_SYMBOL_GPL(venus_helper_set_work_mode);
1042
1043 int venus_helper_init_codec_freq_data(struct venus_inst *inst)
1044 {
1045         const struct codec_freq_data *data;
1046         unsigned int i, data_size;
1047         u32 pixfmt;
1048         int ret = 0;
1049
1050         if (!IS_V4(inst->core))
1051                 return 0;
1052
1053         data = inst->core->res->codec_freq_data;
1054         data_size = inst->core->res->codec_freq_data_size;
1055         pixfmt = inst->session_type == VIDC_SESSION_TYPE_DEC ?
1056                         inst->fmt_out->pixfmt : inst->fmt_cap->pixfmt;
1057
1058         for (i = 0; i < data_size; i++) {
1059                 if (data[i].pixfmt == pixfmt &&
1060                     data[i].session_type == inst->session_type) {
1061                         inst->clk_data.codec_freq_data = &data[i];
1062                         break;
1063                 }
1064         }
1065
1066         if (!inst->clk_data.codec_freq_data)
1067                 ret = -EINVAL;
1068
1069         return ret;
1070 }
1071 EXPORT_SYMBOL_GPL(venus_helper_init_codec_freq_data);
1072
1073 int venus_helper_set_num_bufs(struct venus_inst *inst, unsigned int input_bufs,
1074                               unsigned int output_bufs,
1075                               unsigned int output2_bufs)
1076 {
1077         u32 ptype = HFI_PROPERTY_PARAM_BUFFER_COUNT_ACTUAL;
1078         struct hfi_buffer_count_actual buf_count;
1079         int ret;
1080
1081         buf_count.type = HFI_BUFFER_INPUT;
1082         buf_count.count_actual = input_bufs;
1083
1084         ret = hfi_session_set_property(inst, ptype, &buf_count);
1085         if (ret)
1086                 return ret;
1087
1088         buf_count.type = HFI_BUFFER_OUTPUT;
1089         buf_count.count_actual = output_bufs;
1090
1091         ret = hfi_session_set_property(inst, ptype, &buf_count);
1092         if (ret)
1093                 return ret;
1094
1095         if (output2_bufs) {
1096                 buf_count.type = HFI_BUFFER_OUTPUT2;
1097                 buf_count.count_actual = output2_bufs;
1098
1099                 ret = hfi_session_set_property(inst, ptype, &buf_count);
1100         }
1101
1102         return ret;
1103 }
1104 EXPORT_SYMBOL_GPL(venus_helper_set_num_bufs);
1105
1106 int venus_helper_set_raw_format(struct venus_inst *inst, u32 hfi_format,
1107                                 u32 buftype)
1108 {
1109         const u32 ptype = HFI_PROPERTY_PARAM_UNCOMPRESSED_FORMAT_SELECT;
1110         struct hfi_uncompressed_format_select fmt;
1111
1112         fmt.buffer_type = buftype;
1113         fmt.format = hfi_format;
1114
1115         return hfi_session_set_property(inst, ptype, &fmt);
1116 }
1117 EXPORT_SYMBOL_GPL(venus_helper_set_raw_format);
1118
1119 int venus_helper_set_color_format(struct venus_inst *inst, u32 pixfmt)
1120 {
1121         u32 hfi_format, buftype;
1122
1123         if (inst->session_type == VIDC_SESSION_TYPE_DEC)
1124                 buftype = HFI_BUFFER_OUTPUT;
1125         else if (inst->session_type == VIDC_SESSION_TYPE_ENC)
1126                 buftype = HFI_BUFFER_INPUT;
1127         else
1128                 return -EINVAL;
1129
1130         hfi_format = to_hfi_raw_fmt(pixfmt);
1131         if (!hfi_format)
1132                 return -EINVAL;
1133
1134         return venus_helper_set_raw_format(inst, hfi_format, buftype);
1135 }
1136 EXPORT_SYMBOL_GPL(venus_helper_set_color_format);
1137
1138 int venus_helper_set_multistream(struct venus_inst *inst, bool out_en,
1139                                  bool out2_en)
1140 {
1141         struct hfi_multi_stream multi = {0};
1142         u32 ptype = HFI_PROPERTY_PARAM_VDEC_MULTI_STREAM;
1143         int ret;
1144
1145         multi.buffer_type = HFI_BUFFER_OUTPUT;
1146         multi.enable = out_en;
1147
1148         ret = hfi_session_set_property(inst, ptype, &multi);
1149         if (ret)
1150                 return ret;
1151
1152         multi.buffer_type = HFI_BUFFER_OUTPUT2;
1153         multi.enable = out2_en;
1154
1155         return hfi_session_set_property(inst, ptype, &multi);
1156 }
1157 EXPORT_SYMBOL_GPL(venus_helper_set_multistream);
1158
1159 int venus_helper_set_dyn_bufmode(struct venus_inst *inst)
1160 {
1161         const u32 ptype = HFI_PROPERTY_PARAM_BUFFER_ALLOC_MODE;
1162         struct hfi_buffer_alloc_mode mode;
1163         int ret;
1164
1165         if (!is_dynamic_bufmode(inst))
1166                 return 0;
1167
1168         mode.type = HFI_BUFFER_OUTPUT;
1169         mode.mode = HFI_BUFFER_MODE_DYNAMIC;
1170
1171         ret = hfi_session_set_property(inst, ptype, &mode);
1172         if (ret)
1173                 return ret;
1174
1175         mode.type = HFI_BUFFER_OUTPUT2;
1176
1177         return hfi_session_set_property(inst, ptype, &mode);
1178 }
1179 EXPORT_SYMBOL_GPL(venus_helper_set_dyn_bufmode);
1180
1181 int venus_helper_set_bufsize(struct venus_inst *inst, u32 bufsize, u32 buftype)
1182 {
1183         const u32 ptype = HFI_PROPERTY_PARAM_BUFFER_SIZE_ACTUAL;
1184         struct hfi_buffer_size_actual bufsz;
1185
1186         bufsz.type = buftype;
1187         bufsz.size = bufsize;
1188
1189         return hfi_session_set_property(inst, ptype, &bufsz);
1190 }
1191 EXPORT_SYMBOL_GPL(venus_helper_set_bufsize);
1192
1193 unsigned int venus_helper_get_opb_size(struct venus_inst *inst)
1194 {
1195         /* the encoder has only one output */
1196         if (inst->session_type == VIDC_SESSION_TYPE_ENC)
1197                 return inst->output_buf_size;
1198
1199         if (inst->opb_buftype == HFI_BUFFER_OUTPUT)
1200                 return inst->output_buf_size;
1201         else if (inst->opb_buftype == HFI_BUFFER_OUTPUT2)
1202                 return inst->output2_buf_size;
1203
1204         return 0;
1205 }
1206 EXPORT_SYMBOL_GPL(venus_helper_get_opb_size);
1207
1208 static void delayed_process_buf_func(struct work_struct *work)
1209 {
1210         struct venus_buffer *buf, *n;
1211         struct venus_inst *inst;
1212         int ret;
1213
1214         inst = container_of(work, struct venus_inst, delayed_process_work);
1215
1216         mutex_lock(&inst->lock);
1217
1218         if (!(inst->streamon_out & inst->streamon_cap))
1219                 goto unlock;
1220
1221         list_for_each_entry_safe(buf, n, &inst->delayed_process, ref_list) {
1222                 if (buf->flags & HFI_BUFFERFLAG_READONLY)
1223                         continue;
1224
1225                 ret = session_process_buf(inst, &buf->vb);
1226                 if (ret)
1227                         return_buf_error(inst, &buf->vb);
1228
1229                 list_del_init(&buf->ref_list);
1230         }
1231 unlock:
1232         mutex_unlock(&inst->lock);
1233 }
1234
1235 void venus_helper_release_buf_ref(struct venus_inst *inst, unsigned int idx)
1236 {
1237         struct venus_buffer *buf;
1238
1239         list_for_each_entry(buf, &inst->registeredbufs, reg_list) {
1240                 if (buf->vb.vb2_buf.index == idx) {
1241                         buf->flags &= ~HFI_BUFFERFLAG_READONLY;
1242                         schedule_work(&inst->delayed_process_work);
1243                         break;
1244                 }
1245         }
1246 }
1247 EXPORT_SYMBOL_GPL(venus_helper_release_buf_ref);
1248
1249 void venus_helper_acquire_buf_ref(struct vb2_v4l2_buffer *vbuf)
1250 {
1251         struct venus_buffer *buf = to_venus_buffer(vbuf);
1252
1253         buf->flags |= HFI_BUFFERFLAG_READONLY;
1254 }
1255 EXPORT_SYMBOL_GPL(venus_helper_acquire_buf_ref);
1256
1257 static int is_buf_refed(struct venus_inst *inst, struct vb2_v4l2_buffer *vbuf)
1258 {
1259         struct venus_buffer *buf = to_venus_buffer(vbuf);
1260
1261         if (buf->flags & HFI_BUFFERFLAG_READONLY) {
1262                 list_add_tail(&buf->ref_list, &inst->delayed_process);
1263                 schedule_work(&inst->delayed_process_work);
1264                 return 1;
1265         }
1266
1267         return 0;
1268 }
1269
1270 struct vb2_v4l2_buffer *
1271 venus_helper_find_buf(struct venus_inst *inst, unsigned int type, u32 idx)
1272 {
1273         struct v4l2_m2m_ctx *m2m_ctx = inst->m2m_ctx;
1274
1275         if (type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE)
1276                 return v4l2_m2m_src_buf_remove_by_idx(m2m_ctx, idx);
1277         else
1278                 return v4l2_m2m_dst_buf_remove_by_idx(m2m_ctx, idx);
1279 }
1280 EXPORT_SYMBOL_GPL(venus_helper_find_buf);
1281
1282 int venus_helper_vb2_buf_init(struct vb2_buffer *vb)
1283 {
1284         struct venus_inst *inst = vb2_get_drv_priv(vb->vb2_queue);
1285         struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
1286         struct venus_buffer *buf = to_venus_buffer(vbuf);
1287         struct sg_table *sgt;
1288
1289         sgt = vb2_dma_sg_plane_desc(vb, 0);
1290         if (!sgt)
1291                 return -EFAULT;
1292
1293         buf->size = vb2_plane_size(vb, 0);
1294         buf->dma_addr = sg_dma_address(sgt->sgl);
1295
1296         if (vb->type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE)
1297                 list_add_tail(&buf->reg_list, &inst->registeredbufs);
1298
1299         return 0;
1300 }
1301 EXPORT_SYMBOL_GPL(venus_helper_vb2_buf_init);
1302
1303 int venus_helper_vb2_buf_prepare(struct vb2_buffer *vb)
1304 {
1305         struct venus_inst *inst = vb2_get_drv_priv(vb->vb2_queue);
1306         unsigned int out_buf_size = venus_helper_get_opb_size(inst);
1307         struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
1308
1309         if (V4L2_TYPE_IS_OUTPUT(vb->vb2_queue->type)) {
1310                 if (vbuf->field == V4L2_FIELD_ANY)
1311                         vbuf->field = V4L2_FIELD_NONE;
1312                 if (vbuf->field != V4L2_FIELD_NONE) {
1313                         dev_err(inst->core->dev, "%s field isn't supported\n",
1314                                 __func__);
1315                         return -EINVAL;
1316                 }
1317         }
1318
1319         if (vb->type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE &&
1320             vb2_plane_size(vb, 0) < out_buf_size)
1321                 return -EINVAL;
1322         if (vb->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE &&
1323             vb2_plane_size(vb, 0) < inst->input_buf_size)
1324                 return -EINVAL;
1325
1326         return 0;
1327 }
1328 EXPORT_SYMBOL_GPL(venus_helper_vb2_buf_prepare);
1329
1330 static void cache_payload(struct venus_inst *inst, struct vb2_buffer *vb)
1331 {
1332         struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
1333         unsigned int idx = vbuf->vb2_buf.index;
1334
1335         if (vbuf->vb2_buf.type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE)
1336                 inst->payloads[idx] = vb2_get_plane_payload(vb, 0);
1337 }
1338
1339 void venus_helper_vb2_buf_queue(struct vb2_buffer *vb)
1340 {
1341         struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
1342         struct venus_inst *inst = vb2_get_drv_priv(vb->vb2_queue);
1343         struct v4l2_m2m_ctx *m2m_ctx = inst->m2m_ctx;
1344         int ret;
1345
1346         mutex_lock(&inst->lock);
1347
1348         v4l2_m2m_buf_queue(m2m_ctx, vbuf);
1349
1350         cache_payload(inst, vb);
1351
1352         if (inst->session_type == VIDC_SESSION_TYPE_ENC &&
1353             !(inst->streamon_out && inst->streamon_cap))
1354                 goto unlock;
1355
1356         if (vb2_start_streaming_called(vb->vb2_queue)) {
1357                 ret = is_buf_refed(inst, vbuf);
1358                 if (ret)
1359                         goto unlock;
1360
1361                 ret = session_process_buf(inst, vbuf);
1362                 if (ret)
1363                         return_buf_error(inst, vbuf);
1364         }
1365
1366 unlock:
1367         mutex_unlock(&inst->lock);
1368 }
1369 EXPORT_SYMBOL_GPL(venus_helper_vb2_buf_queue);
1370
1371 void venus_helper_buffers_done(struct venus_inst *inst, unsigned int type,
1372                                enum vb2_buffer_state state)
1373 {
1374         struct vb2_v4l2_buffer *buf;
1375
1376         if (type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE) {
1377                 while ((buf = v4l2_m2m_src_buf_remove(inst->m2m_ctx)))
1378                         v4l2_m2m_buf_done(buf, state);
1379         } else if (type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE) {
1380                 while ((buf = v4l2_m2m_dst_buf_remove(inst->m2m_ctx)))
1381                         v4l2_m2m_buf_done(buf, state);
1382         }
1383 }
1384 EXPORT_SYMBOL_GPL(venus_helper_buffers_done);
1385
1386 void venus_helper_vb2_stop_streaming(struct vb2_queue *q)
1387 {
1388         struct venus_inst *inst = vb2_get_drv_priv(q);
1389         struct venus_core *core = inst->core;
1390         int ret;
1391
1392         mutex_lock(&inst->lock);
1393
1394         if (inst->streamon_out & inst->streamon_cap) {
1395                 ret = hfi_session_stop(inst);
1396                 ret |= hfi_session_unload_res(inst);
1397                 ret |= venus_helper_unregister_bufs(inst);
1398                 ret |= venus_helper_intbufs_free(inst);
1399                 ret |= hfi_session_deinit(inst);
1400
1401                 if (inst->session_error || core->sys_error)
1402                         ret = -EIO;
1403
1404                 if (ret)
1405                         hfi_session_abort(inst);
1406
1407                 venus_helper_free_dpb_bufs(inst);
1408
1409                 venus_pm_load_scale(inst);
1410                 INIT_LIST_HEAD(&inst->registeredbufs);
1411         }
1412
1413         venus_helper_buffers_done(inst, V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE,
1414                                   VB2_BUF_STATE_ERROR);
1415         venus_helper_buffers_done(inst, V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE,
1416                                   VB2_BUF_STATE_ERROR);
1417
1418         if (q->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE)
1419                 inst->streamon_out = 0;
1420         else
1421                 inst->streamon_cap = 0;
1422
1423         venus_pm_release_core(inst);
1424
1425         mutex_unlock(&inst->lock);
1426 }
1427 EXPORT_SYMBOL_GPL(venus_helper_vb2_stop_streaming);
1428
1429 int venus_helper_process_initial_cap_bufs(struct venus_inst *inst)
1430 {
1431         struct v4l2_m2m_ctx *m2m_ctx = inst->m2m_ctx;
1432         struct v4l2_m2m_buffer *buf, *n;
1433         int ret;
1434
1435         v4l2_m2m_for_each_dst_buf_safe(m2m_ctx, buf, n) {
1436                 ret = session_process_buf(inst, &buf->vb);
1437                 if (ret) {
1438                         return_buf_error(inst, &buf->vb);
1439                         return ret;
1440                 }
1441         }
1442
1443         return 0;
1444 }
1445 EXPORT_SYMBOL_GPL(venus_helper_process_initial_cap_bufs);
1446
1447 int venus_helper_process_initial_out_bufs(struct venus_inst *inst)
1448 {
1449         struct v4l2_m2m_ctx *m2m_ctx = inst->m2m_ctx;
1450         struct v4l2_m2m_buffer *buf, *n;
1451         int ret;
1452
1453         v4l2_m2m_for_each_src_buf_safe(m2m_ctx, buf, n) {
1454                 ret = session_process_buf(inst, &buf->vb);
1455                 if (ret) {
1456                         return_buf_error(inst, &buf->vb);
1457                         return ret;
1458                 }
1459         }
1460
1461         return 0;
1462 }
1463 EXPORT_SYMBOL_GPL(venus_helper_process_initial_out_bufs);
1464
1465 int venus_helper_vb2_start_streaming(struct venus_inst *inst)
1466 {
1467         int ret;
1468
1469         ret = venus_helper_intbufs_alloc(inst);
1470         if (ret)
1471                 return ret;
1472
1473         ret = session_register_bufs(inst);
1474         if (ret)
1475                 goto err_bufs_free;
1476
1477         venus_pm_load_scale(inst);
1478
1479         ret = hfi_session_load_res(inst);
1480         if (ret)
1481                 goto err_unreg_bufs;
1482
1483         ret = hfi_session_start(inst);
1484         if (ret)
1485                 goto err_unload_res;
1486
1487         return 0;
1488
1489 err_unload_res:
1490         hfi_session_unload_res(inst);
1491 err_unreg_bufs:
1492         venus_helper_unregister_bufs(inst);
1493 err_bufs_free:
1494         venus_helper_intbufs_free(inst);
1495         return ret;
1496 }
1497 EXPORT_SYMBOL_GPL(venus_helper_vb2_start_streaming);
1498
1499 void venus_helper_m2m_device_run(void *priv)
1500 {
1501         struct venus_inst *inst = priv;
1502         struct v4l2_m2m_ctx *m2m_ctx = inst->m2m_ctx;
1503         struct v4l2_m2m_buffer *buf, *n;
1504         int ret;
1505
1506         mutex_lock(&inst->lock);
1507
1508         v4l2_m2m_for_each_dst_buf_safe(m2m_ctx, buf, n) {
1509                 ret = session_process_buf(inst, &buf->vb);
1510                 if (ret)
1511                         return_buf_error(inst, &buf->vb);
1512         }
1513
1514         v4l2_m2m_for_each_src_buf_safe(m2m_ctx, buf, n) {
1515                 ret = session_process_buf(inst, &buf->vb);
1516                 if (ret)
1517                         return_buf_error(inst, &buf->vb);
1518         }
1519
1520         mutex_unlock(&inst->lock);
1521 }
1522 EXPORT_SYMBOL_GPL(venus_helper_m2m_device_run);
1523
1524 void venus_helper_m2m_job_abort(void *priv)
1525 {
1526         struct venus_inst *inst = priv;
1527
1528         v4l2_m2m_job_finish(inst->m2m_dev, inst->m2m_ctx);
1529 }
1530 EXPORT_SYMBOL_GPL(venus_helper_m2m_job_abort);
1531
1532 void venus_helper_init_instance(struct venus_inst *inst)
1533 {
1534         if (inst->session_type == VIDC_SESSION_TYPE_DEC) {
1535                 INIT_LIST_HEAD(&inst->delayed_process);
1536                 INIT_WORK(&inst->delayed_process_work,
1537                           delayed_process_buf_func);
1538         }
1539 }
1540 EXPORT_SYMBOL_GPL(venus_helper_init_instance);
1541
1542 static bool find_fmt_from_caps(struct venus_caps *caps, u32 buftype, u32 fmt)
1543 {
1544         unsigned int i;
1545
1546         for (i = 0; i < caps->num_fmts; i++) {
1547                 if (caps->fmts[i].buftype == buftype &&
1548                     caps->fmts[i].fmt == fmt)
1549                         return true;
1550         }
1551
1552         return false;
1553 }
1554
1555 int venus_helper_get_out_fmts(struct venus_inst *inst, u32 v4l2_fmt,
1556                               u32 *out_fmt, u32 *out2_fmt, bool ubwc)
1557 {
1558         struct venus_core *core = inst->core;
1559         struct venus_caps *caps;
1560         u32 ubwc_fmt, fmt = to_hfi_raw_fmt(v4l2_fmt);
1561         bool found, found_ubwc;
1562
1563         *out_fmt = *out2_fmt = 0;
1564
1565         if (!fmt)
1566                 return -EINVAL;
1567
1568         caps = venus_caps_by_codec(core, inst->hfi_codec, inst->session_type);
1569         if (!caps)
1570                 return -EINVAL;
1571
1572         if (inst->bit_depth == VIDC_BITDEPTH_10 &&
1573             inst->session_type == VIDC_SESSION_TYPE_DEC) {
1574                 found_ubwc =
1575                         find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT,
1576                                            HFI_COLOR_FORMAT_YUV420_TP10_UBWC);
1577                 found = find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT2,
1578                                            HFI_COLOR_FORMAT_NV12);
1579                 if (found_ubwc && found) {
1580                         /*
1581                          * Hard-code DPB buffers to be 10bit UBWC and decoder
1582                          * output buffers in 8bit NV12 until V4L2 is able to
1583                          * expose compressed/tiled formats to applications.
1584                          */
1585                         *out_fmt = HFI_COLOR_FORMAT_YUV420_TP10_UBWC;
1586                         *out2_fmt = HFI_COLOR_FORMAT_NV12;
1587                         return 0;
1588                 }
1589
1590                 return -EINVAL;
1591         }
1592
1593         if (ubwc) {
1594                 ubwc_fmt = fmt | HFI_COLOR_FORMAT_UBWC_BASE;
1595                 found_ubwc = find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT,
1596                                                 ubwc_fmt);
1597                 found = find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT2, fmt);
1598
1599                 if (found_ubwc && found) {
1600                         *out_fmt = ubwc_fmt;
1601                         *out2_fmt = fmt;
1602                         return 0;
1603                 }
1604         }
1605
1606         found = find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT, fmt);
1607         if (found) {
1608                 *out_fmt = fmt;
1609                 *out2_fmt = 0;
1610                 return 0;
1611         }
1612
1613         found = find_fmt_from_caps(caps, HFI_BUFFER_OUTPUT2, fmt);
1614         if (found) {
1615                 *out_fmt = 0;
1616                 *out2_fmt = fmt;
1617                 return 0;
1618         }
1619
1620         return -EINVAL;
1621 }
1622 EXPORT_SYMBOL_GPL(venus_helper_get_out_fmts);