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