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