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