drm/rockchip: vop: Add yuv plane support

vop support yuv with NV12, NV16 and NV24, only 2 plane yuv.

Signed-off-by: Mark Yao <mark.yao@rock-chips.com>

1 files changed, 54 insertions, 2 deletions

diff --git a/drivers/gpu/drm/rockchip/rockchip_drm_vop.c b/drivers/gpu/drm/rockchip/rockchip_drm_vop.c
index da72de9eb3d5..c1264d58b03b 100644
--- a/drivers/gpu/drm/rockchip/rockchip_drm_vop.c
+++ b/drivers/gpu/drm/rockchip/rockchip_drm_vop.c
@@ -393,6 +393,18 @@ static enum vop_data_format vop_convert_format(uint32_t format)
         }
 }
 
+static bool is_yuv_support(uint32_t format)
+{
+        switch (format) {
+        case DRM_FORMAT_NV12:
+        case DRM_FORMAT_NV16:
+        case DRM_FORMAT_NV24:
+                return true;
+        default:
+                return false;
+        }
+}
+
 static bool is_alpha_support(uint32_t format)
 {
         switch (format) {
@@ -598,17 +610,22 @@ static int vop_update_plane_event(struct drm_plane *plane,
         struct vop *vop = to_vop(crtc);
         struct drm_gem_object *obj;
         struct rockchip_gem_object *rk_obj;
+        struct drm_gem_object *uv_obj;
+        struct rockchip_gem_object *rk_uv_obj;
         unsigned long offset;
         unsigned int actual_w;
         unsigned int actual_h;
         unsigned int dsp_stx;
         unsigned int dsp_sty;
         unsigned int y_vir_stride;
+        unsigned int uv_vir_stride = 0;
         dma_addr_t yrgb_mst;
+        dma_addr_t uv_mst = 0;
         enum vop_data_format format;
         uint32_t val;
         bool is_alpha;
         bool rb_swap;
+        bool is_yuv;
         bool visible;
         int ret;
         struct drm_rect dest = {
@@ -643,6 +660,8 @@ static int vop_update_plane_event(struct drm_plane *plane,
 
         is_alpha = is_alpha_support(fb->pixel_format);
         rb_swap = has_rb_swapped(fb->pixel_format);
+        is_yuv = is_yuv_support(fb->pixel_format);
+
         format = vop_convert_format(fb->pixel_format);
         if (format < 0)
                 return format;
@@ -655,18 +674,47 @@ static int vop_update_plane_event(struct drm_plane *plane,
 
         rk_obj = to_rockchip_obj(obj);
 
+        if (is_yuv) {
+                /*
+                 * Src.x1 can be odd when do clip, but yuv plane start point
+                 * need align with 2 pixel.
+                 */
+                val = (src.x1 >> 16) % 2;
+                src.x1 += val << 16;
+                src.x2 += val << 16;
+        }
+
         actual_w = (src.x2 - src.x1) >> 16;
         actual_h = (src.y2 - src.y1) >> 16;
 
         dsp_stx = dest.x1 + crtc->mode.htotal - crtc->mode.hsync_start;
         dsp_sty = dest.y1 + crtc->mode.vtotal - crtc->mode.vsync_start;
 
-        offset = (src.x1 >> 16) * (fb->bits_per_pixel >> 3);
+        offset = (src.x1 >> 16) * drm_format_plane_cpp(fb->pixel_format, 0);
         offset += (src.y1 >> 16) * fb->pitches[0];
-        yrgb_mst = rk_obj->dma_addr + offset;
 
+        yrgb_mst = rk_obj->dma_addr + offset + fb->offsets[0];
         y_vir_stride = fb->pitches[0] >> 2;
 
+        if (is_yuv) {
+                int hsub = drm_format_horz_chroma_subsampling(fb->pixel_format);
+                int vsub = drm_format_vert_chroma_subsampling(fb->pixel_format);
+                int bpp = drm_format_plane_cpp(fb->pixel_format, 1);
+
+                uv_obj = rockchip_fb_get_gem_obj(fb, 1);
+                if (!uv_obj) {
+                        DRM_ERROR("fail to get uv object from framebuffer\n");
+                        return -EINVAL;
+                }
+                rk_uv_obj = to_rockchip_obj(uv_obj);
+                uv_vir_stride = fb->pitches[1] >> 2;
+
+                offset = (src.x1 >> 16) * bpp / hsub;
+                offset += (src.y1 >> 16) * fb->pitches[1] / vsub;
+
+                uv_mst = rk_uv_obj->dma_addr + offset + fb->offsets[1];
+        }
+
         /*
          * If this plane update changes the plane's framebuffer, (or more
          * precisely, if this update has a different framebuffer than the last
@@ -702,6 +750,10 @@ static int vop_update_plane_event(struct drm_plane *plane,
         VOP_WIN_SET(vop, win, format, format);
         VOP_WIN_SET(vop, win, yrgb_vir, y_vir_stride);
         VOP_WIN_SET(vop, win, yrgb_mst, yrgb_mst);
+        if (is_yuv) {
+                VOP_WIN_SET(vop, win, uv_vir, uv_vir_stride);
+                VOP_WIN_SET(vop, win, uv_mst, uv_mst);
+        }
         val = (actual_h - 1) << 16;
         val |= (actual_w - 1) & 0xffff;
         VOP_WIN_SET(vop, win, act_info, val);