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