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