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