1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Copyright (c) 2015 MediaTek Inc.
4  * Author: CK Hu <ck.hu@mediatek.com>
5  */
6 
7 #include <drm/drm_atomic.h>
8 #include <drm/drm_atomic_helper.h>
9 #include <drm/drm_atomic_uapi.h>
10 #include <drm/drm_blend.h>
11 #include <drm/drm_fourcc.h>
12 #include <drm/drm_framebuffer.h>
13 #include <drm/drm_gem_atomic_helper.h>
14 #include <linux/align.h>
15 
16 #include "mtk_drm_crtc.h"
17 #include "mtk_drm_ddp_comp.h"
18 #include "mtk_drm_drv.h"
19 #include "mtk_drm_gem.h"
20 #include "mtk_drm_plane.h"
21 
22 static const u64 modifiers[] = {
23 	DRM_FORMAT_MOD_LINEAR,
24 	DRM_FORMAT_MOD_ARM_AFBC(AFBC_FORMAT_MOD_BLOCK_SIZE_32x8 |
25 				AFBC_FORMAT_MOD_SPLIT |
26 				AFBC_FORMAT_MOD_SPARSE),
27 	DRM_FORMAT_MOD_INVALID,
28 };
29 
30 static void mtk_plane_reset(struct drm_plane *plane)
31 {
32 	struct mtk_plane_state *state;
33 
34 	if (plane->state) {
35 		__drm_atomic_helper_plane_destroy_state(plane->state);
36 
37 		state = to_mtk_plane_state(plane->state);
38 		memset(state, 0, sizeof(*state));
39 	} else {
40 		state = kzalloc(sizeof(*state), GFP_KERNEL);
41 		if (!state)
42 			return;
43 	}
44 
45 	__drm_atomic_helper_plane_reset(plane, &state->base);
46 
47 	state->base.plane = plane;
48 	state->pending.format = DRM_FORMAT_RGB565;
49 	state->pending.modifier = DRM_FORMAT_MOD_LINEAR;
50 }
51 
52 static struct drm_plane_state *mtk_plane_duplicate_state(struct drm_plane *plane)
53 {
54 	struct mtk_plane_state *old_state = to_mtk_plane_state(plane->state);
55 	struct mtk_plane_state *state;
56 
57 	state = kmalloc(sizeof(*state), GFP_KERNEL);
58 	if (!state)
59 		return NULL;
60 
61 	__drm_atomic_helper_plane_duplicate_state(plane, &state->base);
62 
63 	WARN_ON(state->base.plane != plane);
64 
65 	state->pending = old_state->pending;
66 
67 	return &state->base;
68 }
69 
70 static bool mtk_plane_format_mod_supported(struct drm_plane *plane,
71 					   uint32_t format,
72 					   uint64_t modifier)
73 {
74 	if (modifier == DRM_FORMAT_MOD_LINEAR)
75 		return true;
76 
77 	if (modifier != DRM_FORMAT_MOD_ARM_AFBC(
78 				AFBC_FORMAT_MOD_BLOCK_SIZE_32x8 |
79 				AFBC_FORMAT_MOD_SPLIT |
80 				AFBC_FORMAT_MOD_SPARSE))
81 		return false;
82 
83 	if (format != DRM_FORMAT_XRGB8888 &&
84 	    format != DRM_FORMAT_ARGB8888 &&
85 	    format != DRM_FORMAT_BGRX8888 &&
86 	    format != DRM_FORMAT_BGRA8888 &&
87 	    format != DRM_FORMAT_ABGR8888 &&
88 	    format != DRM_FORMAT_XBGR8888 &&
89 	    format != DRM_FORMAT_RGB888 &&
90 	    format != DRM_FORMAT_BGR888)
91 		return false;
92 
93 	return true;
94 }
95 
96 static void mtk_drm_plane_destroy_state(struct drm_plane *plane,
97 					struct drm_plane_state *state)
98 {
99 	__drm_atomic_helper_plane_destroy_state(state);
100 	kfree(to_mtk_plane_state(state));
101 }
102 
103 static int mtk_plane_atomic_async_check(struct drm_plane *plane,
104 					struct drm_atomic_state *state)
105 {
106 	struct drm_plane_state *new_plane_state = drm_atomic_get_new_plane_state(state,
107 										 plane);
108 	struct drm_crtc_state *crtc_state;
109 	int ret;
110 
111 	if (plane != new_plane_state->crtc->cursor)
112 		return -EINVAL;
113 
114 	if (!plane->state)
115 		return -EINVAL;
116 
117 	if (!plane->state->fb)
118 		return -EINVAL;
119 
120 	ret = mtk_drm_crtc_plane_check(new_plane_state->crtc, plane,
121 				       to_mtk_plane_state(new_plane_state));
122 	if (ret)
123 		return ret;
124 
125 	if (state)
126 		crtc_state = drm_atomic_get_existing_crtc_state(state,
127 								new_plane_state->crtc);
128 	else /* Special case for asynchronous cursor updates. */
129 		crtc_state = new_plane_state->crtc->state;
130 
131 	return drm_atomic_helper_check_plane_state(plane->state, crtc_state,
132 						   DRM_PLANE_NO_SCALING,
133 						   DRM_PLANE_NO_SCALING,
134 						   true, true);
135 }
136 
137 static void mtk_plane_update_new_state(struct drm_plane_state *new_state,
138 				       struct mtk_plane_state *mtk_plane_state)
139 {
140 	struct drm_framebuffer *fb = new_state->fb;
141 	struct drm_gem_object *gem;
142 	struct mtk_drm_gem_obj *mtk_gem;
143 	unsigned int pitch, format;
144 	u64 modifier;
145 	dma_addr_t addr;
146 	dma_addr_t hdr_addr = 0;
147 	unsigned int hdr_pitch = 0;
148 
149 	gem = fb->obj[0];
150 	mtk_gem = to_mtk_gem_obj(gem);
151 	addr = mtk_gem->dma_addr;
152 	pitch = fb->pitches[0];
153 	format = fb->format->format;
154 	modifier = fb->modifier;
155 
156 	if (modifier == DRM_FORMAT_MOD_LINEAR) {
157 		addr += (new_state->src.x1 >> 16) * fb->format->cpp[0];
158 		addr += (new_state->src.y1 >> 16) * pitch;
159 	} else {
160 		int width_in_blocks = ALIGN(fb->width, AFBC_DATA_BLOCK_WIDTH)
161 				      / AFBC_DATA_BLOCK_WIDTH;
162 		int height_in_blocks = ALIGN(fb->height, AFBC_DATA_BLOCK_HEIGHT)
163 				       / AFBC_DATA_BLOCK_HEIGHT;
164 		int x_offset_in_blocks = (new_state->src.x1 >> 16) / AFBC_DATA_BLOCK_WIDTH;
165 		int y_offset_in_blocks = (new_state->src.y1 >> 16) / AFBC_DATA_BLOCK_HEIGHT;
166 		int hdr_size;
167 
168 		hdr_pitch = width_in_blocks * AFBC_HEADER_BLOCK_SIZE;
169 		pitch = width_in_blocks * AFBC_DATA_BLOCK_WIDTH *
170 			AFBC_DATA_BLOCK_HEIGHT * fb->format->cpp[0];
171 
172 		hdr_size = ALIGN(hdr_pitch * height_in_blocks, AFBC_HEADER_ALIGNMENT);
173 
174 		hdr_addr = addr + hdr_pitch * y_offset_in_blocks +
175 			   AFBC_HEADER_BLOCK_SIZE * x_offset_in_blocks;
176 		/* The data plane is offset by 1 additional block. */
177 		addr = addr + hdr_size +
178 		       pitch * y_offset_in_blocks +
179 		       AFBC_DATA_BLOCK_WIDTH * AFBC_DATA_BLOCK_HEIGHT *
180 		       fb->format->cpp[0] * (x_offset_in_blocks + 1);
181 	}
182 
183 	mtk_plane_state->pending.enable = true;
184 	mtk_plane_state->pending.pitch = pitch;
185 	mtk_plane_state->pending.hdr_pitch = hdr_pitch;
186 	mtk_plane_state->pending.format = format;
187 	mtk_plane_state->pending.modifier = modifier;
188 	mtk_plane_state->pending.addr = addr;
189 	mtk_plane_state->pending.hdr_addr = hdr_addr;
190 	mtk_plane_state->pending.x = new_state->dst.x1;
191 	mtk_plane_state->pending.y = new_state->dst.y1;
192 	mtk_plane_state->pending.width = drm_rect_width(&new_state->dst);
193 	mtk_plane_state->pending.height = drm_rect_height(&new_state->dst);
194 	mtk_plane_state->pending.rotation = new_state->rotation;
195 	mtk_plane_state->pending.color_encoding = new_state->color_encoding;
196 }
197 
198 static void mtk_plane_atomic_async_update(struct drm_plane *plane,
199 					  struct drm_atomic_state *state)
200 {
201 	struct drm_plane_state *new_state = drm_atomic_get_new_plane_state(state,
202 									   plane);
203 	struct mtk_plane_state *new_plane_state = to_mtk_plane_state(plane->state);
204 
205 	plane->state->crtc_x = new_state->crtc_x;
206 	plane->state->crtc_y = new_state->crtc_y;
207 	plane->state->crtc_h = new_state->crtc_h;
208 	plane->state->crtc_w = new_state->crtc_w;
209 	plane->state->src_x = new_state->src_x;
210 	plane->state->src_y = new_state->src_y;
211 	plane->state->src_h = new_state->src_h;
212 	plane->state->src_w = new_state->src_w;
213 	swap(plane->state->fb, new_state->fb);
214 
215 	mtk_plane_update_new_state(new_state, new_plane_state);
216 	wmb(); /* Make sure the above parameters are set before update */
217 	new_plane_state->pending.async_dirty = true;
218 	mtk_drm_crtc_async_update(new_state->crtc, plane, state);
219 }
220 
221 static const struct drm_plane_funcs mtk_plane_funcs = {
222 	.update_plane = drm_atomic_helper_update_plane,
223 	.disable_plane = drm_atomic_helper_disable_plane,
224 	.destroy = drm_plane_cleanup,
225 	.reset = mtk_plane_reset,
226 	.atomic_duplicate_state = mtk_plane_duplicate_state,
227 	.atomic_destroy_state = mtk_drm_plane_destroy_state,
228 	.format_mod_supported = mtk_plane_format_mod_supported,
229 };
230 
231 static int mtk_plane_atomic_check(struct drm_plane *plane,
232 				  struct drm_atomic_state *state)
233 {
234 	struct drm_plane_state *new_plane_state = drm_atomic_get_new_plane_state(state,
235 										 plane);
236 	struct drm_framebuffer *fb = new_plane_state->fb;
237 	struct drm_crtc_state *crtc_state;
238 	int ret;
239 
240 	if (!fb)
241 		return 0;
242 
243 	if (WARN_ON(!new_plane_state->crtc))
244 		return 0;
245 
246 	ret = mtk_drm_crtc_plane_check(new_plane_state->crtc, plane,
247 				       to_mtk_plane_state(new_plane_state));
248 	if (ret)
249 		return ret;
250 
251 	crtc_state = drm_atomic_get_crtc_state(state,
252 					       new_plane_state->crtc);
253 	if (IS_ERR(crtc_state))
254 		return PTR_ERR(crtc_state);
255 
256 	return drm_atomic_helper_check_plane_state(new_plane_state,
257 						   crtc_state,
258 						   DRM_PLANE_NO_SCALING,
259 						   DRM_PLANE_NO_SCALING,
260 						   true, true);
261 }
262 
263 static void mtk_plane_atomic_disable(struct drm_plane *plane,
264 				     struct drm_atomic_state *state)
265 {
266 	struct drm_plane_state *new_state = drm_atomic_get_new_plane_state(state,
267 									   plane);
268 	struct mtk_plane_state *mtk_plane_state = to_mtk_plane_state(new_state);
269 	mtk_plane_state->pending.enable = false;
270 	wmb(); /* Make sure the above parameter is set before update */
271 	mtk_plane_state->pending.dirty = true;
272 }
273 
274 static void mtk_plane_atomic_update(struct drm_plane *plane,
275 				    struct drm_atomic_state *state)
276 {
277 	struct drm_plane_state *new_state = drm_atomic_get_new_plane_state(state,
278 									   plane);
279 	struct mtk_plane_state *mtk_plane_state = to_mtk_plane_state(new_state);
280 
281 	if (!new_state->crtc || WARN_ON(!new_state->fb))
282 		return;
283 
284 	if (!new_state->visible) {
285 		mtk_plane_atomic_disable(plane, state);
286 		return;
287 	}
288 
289 	mtk_plane_update_new_state(new_state, mtk_plane_state);
290 	wmb(); /* Make sure the above parameters are set before update */
291 	mtk_plane_state->pending.dirty = true;
292 }
293 
294 static const struct drm_plane_helper_funcs mtk_plane_helper_funcs = {
295 	.atomic_check = mtk_plane_atomic_check,
296 	.atomic_update = mtk_plane_atomic_update,
297 	.atomic_disable = mtk_plane_atomic_disable,
298 	.atomic_async_update = mtk_plane_atomic_async_update,
299 	.atomic_async_check = mtk_plane_atomic_async_check,
300 };
301 
302 int mtk_plane_init(struct drm_device *dev, struct drm_plane *plane,
303 		   unsigned long possible_crtcs, enum drm_plane_type type,
304 		   unsigned int supported_rotations, const u32 *formats,
305 		   size_t num_formats)
306 {
307 	int err;
308 
309 	if (!formats || !num_formats) {
310 		DRM_ERROR("no formats for plane\n");
311 		return -EINVAL;
312 	}
313 
314 	err = drm_universal_plane_init(dev, plane, possible_crtcs,
315 				       &mtk_plane_funcs, formats,
316 				       num_formats, modifiers, type, NULL);
317 	if (err) {
318 		DRM_ERROR("failed to initialize plane\n");
319 		return err;
320 	}
321 
322 	if (supported_rotations & ~DRM_MODE_ROTATE_0) {
323 		err = drm_plane_create_rotation_property(plane,
324 							 DRM_MODE_ROTATE_0,
325 							 supported_rotations);
326 		if (err)
327 			DRM_INFO("Create rotation property failed\n");
328 	}
329 
330 	drm_plane_helper_add(plane, &mtk_plane_helper_funcs);
331 
332 	return 0;
333 }
334