vkms_writeback.c

  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_edid.h>
  7#include <drm/drm_fourcc.h>
  8#include <drm/drm_writeback.h>
  9#include <drm/drm_print.h>
 10#include <drm/drm_probe_helper.h>
 11#include <drm/drm_atomic_helper.h>
 12#include <drm/drm_gem_framebuffer_helper.h>
 13#include <drm/drm_gem_shmem_helper.h>
 14
 15#include "vkms_drv.h"
 16#include "vkms_formats.h"
 17
 18static const u32 vkms_wb_formats[] = {
 19	DRM_FORMAT_ARGB8888,
 20	DRM_FORMAT_XRGB8888,
 21	DRM_FORMAT_ABGR8888,
 22	DRM_FORMAT_XRGB16161616,
 23	DRM_FORMAT_ARGB16161616,
 24	DRM_FORMAT_RGB565
 25};
 26
 27static const struct drm_connector_funcs vkms_wb_connector_funcs = {
 28	.fill_modes = drm_helper_probe_single_connector_modes,
 29	.reset = drm_atomic_helper_connector_reset,
 30	.atomic_duplicate_state = drm_atomic_helper_connector_duplicate_state,
 31	.atomic_destroy_state = drm_atomic_helper_connector_destroy_state,
 32};
 33
 34static int vkms_wb_atomic_check(struct drm_connector *connector,
 35				struct drm_atomic_state *state)
 36{
 37	struct drm_connector_state *conn_state =
 38		drm_atomic_get_new_connector_state(state, connector);
 39	struct drm_crtc_state *crtc_state;
 40	struct drm_framebuffer *fb;
 41	const struct drm_display_mode *mode;
 42	int ret;
 43
 44	if (!conn_state->writeback_job || !conn_state->writeback_job->fb)
 45		return 0;
 46
 47	if (!conn_state->crtc)
 48		return 0;
 49
 50	crtc_state = drm_atomic_get_new_crtc_state(state, conn_state->crtc);
 51	mode = &crtc_state->mode;
 52
 53	fb = conn_state->writeback_job->fb;
 54	if (fb->width != mode->hdisplay || fb->height != mode->vdisplay) {
 55		DRM_DEBUG_KMS("Invalid framebuffer size %ux%u\n",
 56			      fb->width, fb->height);
 57		return -EINVAL;
 58	}
 59
 60	ret = drm_atomic_helper_check_wb_connector_state(connector, state);
 61	if (ret < 0)
 62		return ret;
 63
 64	return 0;
 65}
 66
 67static int vkms_wb_connector_get_modes(struct drm_connector *connector)
 68{
 69	struct drm_device *dev = connector->dev;
 70
 71	return drm_add_modes_noedid(connector, dev->mode_config.max_width,
 72				    dev->mode_config.max_height);
 73}
 74
 75static int vkms_wb_prepare_job(struct drm_writeback_connector *wb_connector,
 76			       struct drm_writeback_job *job)
 77{
 78	struct vkms_writeback_job *vkmsjob;
 79	int ret;
 80
 81	if (!job->fb)
 82		return 0;
 83
 84	vkmsjob = kzalloc(sizeof(*vkmsjob), GFP_KERNEL);
 85	if (!vkmsjob)
 86		return -ENOMEM;
 87
 88	ret = drm_gem_fb_vmap(job->fb, vkmsjob->wb_frame_info.map, vkmsjob->data);
 89	if (ret) {
 90		DRM_ERROR("vmap failed: %d\n", ret);
 91		goto err_kfree;
 92	}
 93
 94	vkmsjob->wb_frame_info.fb = job->fb;
 95	drm_framebuffer_get(vkmsjob->wb_frame_info.fb);
 96
 97	job->priv = vkmsjob;
 98
 99	return 0;
100
101err_kfree:
102	kfree(vkmsjob);
103	return ret;
104}
105
106static void vkms_wb_cleanup_job(struct drm_writeback_connector *connector,
107				struct drm_writeback_job *job)
108{
109	struct vkms_writeback_job *vkmsjob = job->priv;
110	struct vkms_output *vkms_output = container_of(connector,
111						       struct vkms_output,
112						       wb_connector);
113
114	if (!job->fb)
115		return;
116
117	drm_gem_fb_vunmap(job->fb, vkmsjob->wb_frame_info.map);
118
119	drm_framebuffer_put(vkmsjob->wb_frame_info.fb);
120
121	vkms_set_composer(vkms_output, false);
122	kfree(vkmsjob);
123}
124
125static void vkms_wb_atomic_commit(struct drm_connector *conn,
126				  struct drm_atomic_state *state)
127{
128	struct drm_connector_state *connector_state = drm_atomic_get_new_connector_state(state,
129											 conn);
130	struct vkms_output *output = drm_crtc_to_vkms_output(connector_state->crtc);
131	struct drm_writeback_connector *wb_conn = &output->wb_connector;
132	struct drm_connector_state *conn_state = wb_conn->base.state;
133	struct vkms_crtc_state *crtc_state = output->composer_state;
134	struct drm_framebuffer *fb = connector_state->writeback_job->fb;
135	u16 crtc_height = crtc_state->base.mode.vdisplay;
136	u16 crtc_width = crtc_state->base.mode.hdisplay;
137	struct vkms_writeback_job *active_wb;
138	struct vkms_frame_info *wb_frame_info;
139	u32 wb_format = fb->format->format;
140
141	if (!conn_state)
142		return;
143
144	vkms_set_composer(output, true);
145
146	active_wb = conn_state->writeback_job->priv;
147	wb_frame_info = &active_wb->wb_frame_info;
148
149	spin_lock_irq(&output->composer_lock);
150	crtc_state->active_writeback = active_wb;
151	crtc_state->wb_pending = true;
152	spin_unlock_irq(&output->composer_lock);
153	drm_writeback_queue_job(wb_conn, connector_state);
154	active_wb->pixel_write = get_pixel_write_function(wb_format);
155	drm_rect_init(&wb_frame_info->src, 0, 0, crtc_width, crtc_height);
156	drm_rect_init(&wb_frame_info->dst, 0, 0, crtc_width, crtc_height);
157}
158
159static const struct drm_connector_helper_funcs vkms_wb_conn_helper_funcs = {
160	.get_modes = vkms_wb_connector_get_modes,
161	.prepare_writeback_job = vkms_wb_prepare_job,
162	.cleanup_writeback_job = vkms_wb_cleanup_job,
163	.atomic_commit = vkms_wb_atomic_commit,
164	.atomic_check = vkms_wb_atomic_check,
165};
166
167int vkms_enable_writeback_connector(struct vkms_device *vkmsdev,
168				    struct vkms_output *vkms_output)
169{
170	struct drm_writeback_connector *wb = &vkms_output->wb_connector;
171	int ret;
172
173	ret = drmm_encoder_init(&vkmsdev->drm, &vkms_output->wb_encoder,
174				NULL, DRM_MODE_ENCODER_VIRTUAL, NULL);
175	if (ret)
176		return ret;
177	vkms_output->wb_encoder.possible_crtcs |= drm_crtc_mask(&vkms_output->crtc);
178	vkms_output->wb_encoder.possible_clones |=
179		drm_encoder_mask(&vkms_output->wb_encoder);
180
181	drm_connector_helper_add(&wb->base, &vkms_wb_conn_helper_funcs);
182
183	return drmm_writeback_connector_init(&vkmsdev->drm, wb,
184					     &vkms_wb_connector_funcs,
185					     &vkms_output->wb_encoder,
186					     vkms_wb_formats,
187					     ARRAY_SIZE(vkms_wb_formats));
188}