[PATCH 04/10] rockchip/vpu: Cleanup macroblock alignment

[Ezequiel Garcia]

We need to make the macrobock alignment generic, in order
to support multiple codecs.

Signed-off-by: Ezequiel Garcia <ezequ...@collabora.com>
---
 .../media/rockchip/vpu/rockchip_vpu_enc.c     | 20 ++++++++-----------
 1 file changed, 8 insertions(+), 12 deletions(-)

diff --git a/drivers/staging/media/rockchip/vpu/rockchip_vpu_enc.c 
b/drivers/staging/media/rockchip/vpu/rockchip_vpu_enc.c
index 4451bb2dc3d7..ae1ff3d9b9d2 100644
--- a/drivers/staging/media/rockchip/vpu/rockchip_vpu_enc.c
+++ b/drivers/staging/media/rockchip/vpu/rockchip_vpu_enc.c
@@ -203,8 +203,8 @@ vidioc_try_fmt_cap_mplane(struct file *file, void *priv, 
struct v4l2_format *f)
                               fmt->frmsize.min_height,
                               fmt->frmsize.max_height);
        /* Round up to macroblocks. */
-       pix_mp->width = round_up(pix_mp->width, JPEG_MB_DIM);
-       pix_mp->height = round_up(pix_mp->height, JPEG_MB_DIM);
+       pix_mp->width = round_up(pix_mp->width, fmt->frmsize.step_width);
+       pix_mp->height = round_up(pix_mp->height, fmt->frmsize.step_height);
 
        /*
         * For compressed formats the application can specify
@@ -248,8 +248,8 @@ vidioc_try_fmt_out_mplane(struct file *file, void *priv, 
struct v4l2_format *f)
                       ctx->vpu_dst_fmt->frmsize.min_height,
                       ctx->vpu_dst_fmt->frmsize.max_height);
        /* Round up to macroblocks. */
-       width = round_up(width, JPEG_MB_DIM);
-       height = round_up(height, JPEG_MB_DIM);
+       width = round_up(width, ctx->vpu_dst_fmt->frmsize.step_width);
+       height = round_up(height, ctx->vpu_dst_fmt->frmsize.step_height);
 
        /* Fill remaining fields */
        v4l2_fill_pixfmt_mp(pix_mp, fmt->fourcc, width, height);
@@ -338,10 +338,8 @@ vidioc_s_fmt_out_mplane(struct file *file, void *priv, 
struct v4l2_format *f)
        ctx->dst_fmt.height = pix_mp->height;
 
        vpu_debug(0, "OUTPUT codec mode: %d\n", ctx->vpu_src_fmt->codec_mode);
-       vpu_debug(0, "fmt - w: %d, h: %d, mb - w: %d, h: %d\n",
-                 pix_mp->width, pix_mp->height,
-                 JPEG_MB_WIDTH(pix_mp->width),
-                 JPEG_MB_HEIGHT(pix_mp->height));
+       vpu_debug(0, "fmt - w: %d, h: %d\n",
+                 pix_mp->width, pix_mp->height);
        return 0;
 }
 
@@ -380,10 +378,8 @@ vidioc_s_fmt_cap_mplane(struct file *file, void *priv, 
struct v4l2_format *f)
        ctx->dst_fmt = *pix_mp;
 
        vpu_debug(0, "CAPTURE codec mode: %d\n", ctx->vpu_dst_fmt->codec_mode);
-       vpu_debug(0, "fmt - w: %d, h: %d, mb - w: %d, h: %d\n",
-                 pix_mp->width, pix_mp->height,
-                 JPEG_MB_WIDTH(pix_mp->width),
-                 JPEG_MB_HEIGHT(pix_mp->height));
+       vpu_debug(0, "fmt - w: %d, h: %d\n",
+                 pix_mp->width, pix_mp->height);
 
        /*
         * Current raw format might have become invalid with newly
-- 
2.20.1