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 *
vkms_plane_duplicate_state(struct drm_plane * plane)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
vkms_plane_destroy_state(struct drm_plane * plane,struct drm_plane_state * old_state)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
vkms_plane_reset(struct drm_plane * plane)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
vkms_plane_atomic_update(struct drm_plane * plane,struct drm_atomic_state * state)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
vkms_plane_atomic_check(struct drm_plane * plane,struct drm_atomic_state * state)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
vkms_prepare_fb(struct drm_plane * plane,struct drm_plane_state * state)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
vkms_cleanup_fb(struct drm_plane * plane,struct drm_plane_state * state)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
vkms_plane_init(struct vkms_device * vkmsdev,enum drm_plane_type type)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