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