mirror of
https://git.kernel.org/pub/scm/linux/kernel/git/stable/linux.git
synced 2024-11-01 17:08:10 +00:00
media: hantro: postproc: Fix buffer size calculation
When allocating aux buffers for postprocessing, it's assumed that base buffer size is the same as that of output. Coincidentally, that's true most of the time, but not always. 10-bit source also needs aux buffer size which is appropriate for 10-bit native format, even if the output format is 8-bit. Similarly, mv sizes and other extra buffer size also depends on source width/height, not destination. Reviewed-by: Ezequiel Garcia <ezequiel@vanguardiasur.com.ar> Tested-by: Benjamin Gaignard <benjamin.gaignard@collabora.com> Signed-off-by: Jernej Skrabec <jernej.skrabec@gmail.com> Signed-off-by: Hans Verkuil <hverkuil-cisco@xs4all.nl> Signed-off-by: Mauro Carvalho Chehab <mchehab@kernel.org>
This commit is contained in:
parent
8d03c2ab88
commit
b273deab2b
3 changed files with 20 additions and 8 deletions
|
@ -12,6 +12,7 @@
|
|||
#include "hantro_hw.h"
|
||||
#include "hantro_g1_regs.h"
|
||||
#include "hantro_g2_regs.h"
|
||||
#include "hantro_v4l2.h"
|
||||
|
||||
#define HANTRO_PP_REG_WRITE(vpu, reg_name, val) \
|
||||
{ \
|
||||
|
@ -174,18 +175,27 @@ int hantro_postproc_alloc(struct hantro_ctx *ctx)
|
|||
struct v4l2_m2m_ctx *m2m_ctx = ctx->fh.m2m_ctx;
|
||||
struct vb2_queue *cap_queue = &m2m_ctx->cap_q_ctx.q;
|
||||
unsigned int num_buffers = cap_queue->num_buffers;
|
||||
struct v4l2_pix_format_mplane pix_mp;
|
||||
const struct hantro_fmt *fmt;
|
||||
unsigned int i, buf_size;
|
||||
|
||||
buf_size = ctx->dst_fmt.plane_fmt[0].sizeimage;
|
||||
/* this should always pick native format */
|
||||
fmt = hantro_get_default_fmt(ctx, false);
|
||||
if (!fmt)
|
||||
return -EINVAL;
|
||||
v4l2_fill_pixfmt_mp(&pix_mp, fmt->fourcc, ctx->src_fmt.width,
|
||||
ctx->src_fmt.height);
|
||||
|
||||
buf_size = pix_mp.plane_fmt[0].sizeimage;
|
||||
if (ctx->vpu_src_fmt->fourcc == V4L2_PIX_FMT_H264_SLICE)
|
||||
buf_size += hantro_h264_mv_size(ctx->dst_fmt.width,
|
||||
ctx->dst_fmt.height);
|
||||
buf_size += hantro_h264_mv_size(pix_mp.width,
|
||||
pix_mp.height);
|
||||
else if (ctx->vpu_src_fmt->fourcc == V4L2_PIX_FMT_VP9_FRAME)
|
||||
buf_size += hantro_vp9_mv_size(ctx->dst_fmt.width,
|
||||
ctx->dst_fmt.height);
|
||||
buf_size += hantro_vp9_mv_size(pix_mp.width,
|
||||
pix_mp.height);
|
||||
else if (ctx->vpu_src_fmt->fourcc == V4L2_PIX_FMT_HEVC_SLICE)
|
||||
buf_size += hantro_hevc_mv_size(ctx->dst_fmt.width,
|
||||
ctx->dst_fmt.height);
|
||||
buf_size += hantro_hevc_mv_size(pix_mp.width,
|
||||
pix_mp.height);
|
||||
|
||||
for (i = 0; i < num_buffers; ++i) {
|
||||
struct hantro_aux_buf *priv = &ctx->postproc.dec_q[i];
|
||||
|
|
|
@ -118,7 +118,7 @@ hantro_find_format(const struct hantro_ctx *ctx, u32 fourcc)
|
|||
return NULL;
|
||||
}
|
||||
|
||||
static const struct hantro_fmt *
|
||||
const struct hantro_fmt *
|
||||
hantro_get_default_fmt(const struct hantro_ctx *ctx, bool bitstream)
|
||||
{
|
||||
const struct hantro_fmt *formats;
|
||||
|
|
|
@ -23,5 +23,7 @@ extern const struct vb2_ops hantro_queue_ops;
|
|||
|
||||
void hantro_reset_fmts(struct hantro_ctx *ctx);
|
||||
int hantro_get_format_depth(u32 fourcc);
|
||||
const struct hantro_fmt *
|
||||
hantro_get_default_fmt(const struct hantro_ctx *ctx, bool bitstream);
|
||||
|
||||
#endif /* HANTRO_V4L2_H_ */
|
||||
|
|
Loading…
Reference in a new issue