1// SPDX-License-Identifier: GPL-2.0+
  2
  3#include <linux/dma-buf-map.h>
  4
  5#include <drm/drm_atomic.h>
  6#include <drm/drm_atomic_helper.h>
  7#include <drm/drm_fourcc.h>
  8#include <drm/drm_gem_atomic_helper.h>
  9#include <drm/drm_gem_framebuffer_helper.h>
 10#include <drm/drm_plane_helper.h>
 11
 12#include "vkms_drv.h"
 13
 14static const u32 vkms_formats[] = {
 15	DRM_FORMAT_XRGB8888,
 16};
 17
 18static const u32 vkms_plane_formats[] = {
 19	DRM_FORMAT_ARGB8888,
 20	DRM_FORMAT_XRGB8888
 21};
 22
 23static struct drm_plane_state *
 24vkms_plane_duplicate_state(struct drm_plane *plane)
 25{
 26	struct vkms_plane_state *vkms_state;
 27	struct vkms_composer *composer;
 28
 29	vkms_state = kzalloc(sizeof(*vkms_state), GFP_KERNEL);
 30	if (!vkms_state)
 31		return NULL;
 32
 33	composer = kzalloc(sizeof(*composer), GFP_KERNEL);
 34	if (!composer) {
 35		DRM_DEBUG_KMS("Couldn't allocate composer\n");
 36		kfree(vkms_state);
 37		return NULL;
 38	}
 39
 40	vkms_state->composer = composer;
 41
 42	__drm_atomic_helper_plane_duplicate_state(plane,
 43						  &vkms_state->base);
 44
 45	return &vkms_state->base;
 46}
 47
 48static 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.crtc;
 53
 54	if (crtc) {
 55		/* dropping the reference we acquired in
 56		 * vkms_primary_plane_update()
 57		 */
 58		if (drm_framebuffer_read_refcount(&vkms_state->composer->fb))
 59			drm_framebuffer_put(&vkms_state->composer->fb);
 60	}
 61
 62	kfree(vkms_state->composer);
 63	vkms_state->composer = NULL;
 64
 65	__drm_atomic_helper_plane_destroy_state(old_state);
 66	kfree(vkms_state);
 67}
 68
 69static 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
 76	vkms_state = kzalloc(sizeof(*vkms_state), GFP_KERNEL);
 77	if (!vkms_state) {
 78		DRM_ERROR("Cannot allocate vkms_plane_state\n");
 79		return;
 80	}
 81
 82	plane->state = &vkms_state->base;
 83	plane->state->plane = plane;
 84}
 85
 86static 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
 94static 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_framebuffer *fb = new_state->fb;
101	struct vkms_composer *composer;
102
103	if (!new_state->crtc || !fb)
104		return;
105
106	vkms_plane_state = to_vkms_plane_state(new_state);
107
108	composer = vkms_plane_state->composer;
109	memcpy(&composer->src, &new_state->src, sizeof(struct drm_rect));
110	memcpy(&composer->dst, &new_state->dst, sizeof(struct drm_rect));
111	memcpy(&composer->fb, fb, sizeof(struct drm_framebuffer));
112	drm_framebuffer_get(&composer->fb);
113	composer->offset = fb->offsets[0];
114	composer->pitch = fb->pitches[0];
115	composer->cpp = fb->format->cpp[0];
116}
117
118static int vkms_plane_atomic_check(struct drm_plane *plane,
119				   struct drm_atomic_state *state)
120{
121	struct drm_plane_state *new_plane_state = drm_atomic_get_new_plane_state(state,
122										 plane);
123	struct drm_crtc_state *crtc_state;
124	bool can_position = false;
125	int ret;
126
127	if (!new_plane_state->fb || WARN_ON(!new_plane_state->crtc))
128		return 0;
129
130	crtc_state = drm_atomic_get_crtc_state(state,
131					       new_plane_state->crtc);
132	if (IS_ERR(crtc_state))
133		return PTR_ERR(crtc_state);
134
135	if (plane->type != DRM_PLANE_TYPE_PRIMARY)
136		can_position = true;
137
138	ret = drm_atomic_helper_check_plane_state(new_plane_state, crtc_state,
139						  DRM_PLANE_HELPER_NO_SCALING,
140						  DRM_PLANE_HELPER_NO_SCALING,
141						  can_position, true);
142	if (ret != 0)
143		return ret;
144
145	/* for now primary plane must be visible and full screen */
146	if (!new_plane_state->visible && !can_position)
147		return -EINVAL;
148
149	return 0;
150}
151
152static const struct drm_plane_helper_funcs vkms_primary_helper_funcs = {
153	.atomic_update		= vkms_plane_atomic_update,
154	.atomic_check		= vkms_plane_atomic_check,
155	DRM_GEM_SHADOW_PLANE_HELPER_FUNCS,
156};
157
158struct vkms_plane *vkms_plane_init(struct vkms_device *vkmsdev,
159				   enum drm_plane_type type, int index)
160{
161	struct drm_device *dev = &vkmsdev->drm;
162	const struct drm_plane_helper_funcs *funcs;
163	struct vkms_plane *plane;
164	const u32 *formats;
165	int nformats;
166
167	switch (type) {
168	case DRM_PLANE_TYPE_PRIMARY:
169		formats = vkms_formats;
170		nformats = ARRAY_SIZE(vkms_formats);
171		funcs = &vkms_primary_helper_funcs;
172		break;
173	case DRM_PLANE_TYPE_CURSOR:
174	case DRM_PLANE_TYPE_OVERLAY:
175		formats = vkms_plane_formats;
176		nformats = ARRAY_SIZE(vkms_plane_formats);
177		funcs = &vkms_primary_helper_funcs;
178		break;
179	default:
180		formats = vkms_formats;
181		nformats = ARRAY_SIZE(vkms_formats);
182		funcs = &vkms_primary_helper_funcs;
183		break;
184	}
185
186	plane = drmm_universal_plane_alloc(dev, struct vkms_plane, base, 1 << index,
187					   &vkms_plane_funcs,
188					   formats, nformats,
189					   NULL, type, NULL);
190	if (IS_ERR(plane))
191		return plane;
192
193	drm_plane_helper_add(&plane->base, funcs);
194
195	return plane;
196}