1 // SPDX-License-Identifier: GPL-2.0+ 2 3 #include <linux/iosys-map.h> 4 5 #include <drm/drm_atomic.h> 6 #include <drm/drm_atomic_helper.h> 7 #include <drm/drm_blend.h> 8 #include <drm/drm_fourcc.h> 9 #include <drm/drm_gem_atomic_helper.h> 10 #include <drm/drm_gem_framebuffer_helper.h> 11 12 #include "vkms_drv.h" 13 #include "vkms_formats.h" 14 15 static const u32 vkms_formats[] = { 16 DRM_FORMAT_ARGB8888, 17 DRM_FORMAT_ABGR8888, 18 DRM_FORMAT_BGRA8888, 19 DRM_FORMAT_RGBA8888, 20 DRM_FORMAT_XRGB8888, 21 DRM_FORMAT_XBGR8888, 22 DRM_FORMAT_RGB888, 23 DRM_FORMAT_BGR888, 24 DRM_FORMAT_XRGB16161616, 25 DRM_FORMAT_XBGR16161616, 26 DRM_FORMAT_ARGB16161616, 27 DRM_FORMAT_ABGR16161616, 28 DRM_FORMAT_RGB565, 29 DRM_FORMAT_BGR565, 30 DRM_FORMAT_NV12, 31 DRM_FORMAT_NV16, 32 DRM_FORMAT_NV24, 33 DRM_FORMAT_NV21, 34 DRM_FORMAT_NV61, 35 DRM_FORMAT_NV42, 36 DRM_FORMAT_YUV420, 37 DRM_FORMAT_YUV422, 38 DRM_FORMAT_YUV444, 39 DRM_FORMAT_YVU420, 40 DRM_FORMAT_YVU422, 41 DRM_FORMAT_YVU444, 42 DRM_FORMAT_P010, 43 DRM_FORMAT_P012, 44 DRM_FORMAT_P016, 45 DRM_FORMAT_R1, 46 DRM_FORMAT_R2, 47 DRM_FORMAT_R4, 48 DRM_FORMAT_R8, 49 }; 50 51 static struct drm_plane_state * 52 vkms_plane_duplicate_state(struct drm_plane *plane) 53 { 54 struct vkms_plane_state *vkms_state; 55 struct vkms_frame_info *frame_info; 56 57 vkms_state = kzalloc(sizeof(*vkms_state), GFP_KERNEL); 58 if (!vkms_state) 59 return NULL; 60 61 frame_info = kzalloc(sizeof(*frame_info), GFP_KERNEL); 62 if (!frame_info) { 63 DRM_DEBUG_KMS("Couldn't allocate frame_info\n"); 64 kfree(vkms_state); 65 return NULL; 66 } 67 68 vkms_state->frame_info = frame_info; 69 70 __drm_gem_duplicate_shadow_plane_state(plane, &vkms_state->base); 71 72 return &vkms_state->base.base; 73 } 74 75 static void vkms_plane_destroy_state(struct drm_plane *plane, 76 struct drm_plane_state *old_state) 77 { 78 struct vkms_plane_state *vkms_state = to_vkms_plane_state(old_state); 79 struct drm_crtc *crtc = vkms_state->base.base.crtc; 80 81 if (crtc && vkms_state->frame_info->fb) { 82 /* dropping the reference we acquired in 83 * vkms_primary_plane_update() 84 */ 85 if (drm_framebuffer_read_refcount(vkms_state->frame_info->fb)) 86 drm_framebuffer_put(vkms_state->frame_info->fb); 87 } 88 89 kfree(vkms_state->frame_info); 90 vkms_state->frame_info = NULL; 91 92 __drm_gem_destroy_shadow_plane_state(&vkms_state->base); 93 kfree(vkms_state); 94 } 95 96 static void vkms_plane_reset(struct drm_plane *plane) 97 { 98 struct vkms_plane_state *vkms_state; 99 100 if (plane->state) { 101 vkms_plane_destroy_state(plane, plane->state); 102 plane->state = NULL; /* must be set to NULL here */ 103 } 104 105 vkms_state = kzalloc(sizeof(*vkms_state), GFP_KERNEL); 106 if (!vkms_state) { 107 DRM_ERROR("Cannot allocate vkms_plane_state\n"); 108 return; 109 } 110 111 __drm_gem_reset_shadow_plane(plane, &vkms_state->base); 112 } 113 114 static const struct drm_plane_funcs vkms_plane_funcs = { 115 .update_plane = drm_atomic_helper_update_plane, 116 .disable_plane = drm_atomic_helper_disable_plane, 117 .reset = vkms_plane_reset, 118 .atomic_duplicate_state = vkms_plane_duplicate_state, 119 .atomic_destroy_state = vkms_plane_destroy_state, 120 }; 121 122 static void vkms_plane_atomic_update(struct drm_plane *plane, 123 struct drm_atomic_state *state) 124 { 125 struct drm_plane_state *new_state = drm_atomic_get_new_plane_state(state, 126 plane); 127 struct vkms_plane_state *vkms_plane_state; 128 struct drm_shadow_plane_state *shadow_plane_state; 129 struct drm_framebuffer *fb = new_state->fb; 130 struct vkms_frame_info *frame_info; 131 u32 fmt; 132 133 if (!new_state->crtc || !fb) 134 return; 135 136 fmt = fb->format->format; 137 vkms_plane_state = to_vkms_plane_state(new_state); 138 shadow_plane_state = &vkms_plane_state->base; 139 140 frame_info = vkms_plane_state->frame_info; 141 memcpy(&frame_info->src, &new_state->src, sizeof(struct drm_rect)); 142 memcpy(&frame_info->dst, &new_state->dst, sizeof(struct drm_rect)); 143 frame_info->fb = fb; 144 memcpy(&frame_info->map, &shadow_plane_state->data, sizeof(frame_info->map)); 145 drm_framebuffer_get(frame_info->fb); 146 frame_info->rotation = new_state->rotation; 147 148 vkms_plane_state->pixel_read_line = get_pixel_read_line_function(fmt); 149 get_conversion_matrix_to_argb_u16(fmt, new_state->color_encoding, new_state->color_range, 150 &vkms_plane_state->conversion_matrix); 151 } 152 153 static int vkms_plane_atomic_check(struct drm_plane *plane, 154 struct drm_atomic_state *state) 155 { 156 struct drm_plane_state *new_plane_state = drm_atomic_get_new_plane_state(state, 157 plane); 158 struct drm_crtc_state *crtc_state; 159 int ret; 160 161 if (!new_plane_state->fb || WARN_ON(!new_plane_state->crtc)) 162 return 0; 163 164 crtc_state = drm_atomic_get_crtc_state(state, 165 new_plane_state->crtc); 166 if (IS_ERR(crtc_state)) 167 return PTR_ERR(crtc_state); 168 169 ret = drm_atomic_helper_check_plane_state(new_plane_state, crtc_state, 170 DRM_PLANE_NO_SCALING, 171 DRM_PLANE_NO_SCALING, 172 true, true); 173 if (ret != 0) 174 return ret; 175 176 return 0; 177 } 178 179 static int vkms_prepare_fb(struct drm_plane *plane, 180 struct drm_plane_state *state) 181 { 182 struct drm_shadow_plane_state *shadow_plane_state; 183 struct drm_framebuffer *fb = state->fb; 184 int ret; 185 186 if (!fb) 187 return 0; 188 189 shadow_plane_state = to_drm_shadow_plane_state(state); 190 191 ret = drm_gem_plane_helper_prepare_fb(plane, state); 192 if (ret) 193 return ret; 194 195 return drm_gem_fb_vmap(fb, shadow_plane_state->map, shadow_plane_state->data); 196 } 197 198 static void vkms_cleanup_fb(struct drm_plane *plane, 199 struct drm_plane_state *state) 200 { 201 struct drm_shadow_plane_state *shadow_plane_state; 202 struct drm_framebuffer *fb = state->fb; 203 204 if (!fb) 205 return; 206 207 shadow_plane_state = to_drm_shadow_plane_state(state); 208 209 drm_gem_fb_vunmap(fb, shadow_plane_state->map); 210 } 211 212 static const struct drm_plane_helper_funcs vkms_plane_helper_funcs = { 213 .atomic_update = vkms_plane_atomic_update, 214 .atomic_check = vkms_plane_atomic_check, 215 .prepare_fb = vkms_prepare_fb, 216 .cleanup_fb = vkms_cleanup_fb, 217 }; 218 219 struct vkms_plane *vkms_plane_init(struct vkms_device *vkmsdev, 220 enum drm_plane_type type) 221 { 222 struct drm_device *dev = &vkmsdev->drm; 223 struct vkms_plane *plane; 224 225 plane = drmm_universal_plane_alloc(dev, struct vkms_plane, base, 0, 226 &vkms_plane_funcs, 227 vkms_formats, ARRAY_SIZE(vkms_formats), 228 NULL, type, NULL); 229 if (IS_ERR(plane)) 230 return plane; 231 232 drm_plane_helper_add(&plane->base, &vkms_plane_helper_funcs); 233 234 drm_plane_create_rotation_property(&plane->base, DRM_MODE_ROTATE_0, 235 DRM_MODE_ROTATE_MASK | DRM_MODE_REFLECT_MASK); 236 237 drm_plane_create_color_properties(&plane->base, 238 BIT(DRM_COLOR_YCBCR_BT601) | 239 BIT(DRM_COLOR_YCBCR_BT709) | 240 BIT(DRM_COLOR_YCBCR_BT2020), 241 BIT(DRM_COLOR_YCBCR_LIMITED_RANGE) | 242 BIT(DRM_COLOR_YCBCR_FULL_RANGE), 243 DRM_COLOR_YCBCR_BT601, 244 DRM_COLOR_YCBCR_FULL_RANGE); 245 246 return plane; 247 } 248