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 18 static 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 27 static 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 34 static 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 67 static 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 75 static 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 101 err_kfree: 102 kfree(vkmsjob); 103 return ret; 104 } 105 106 static 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 125 static 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 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 .atomic_check = vkms_wb_atomic_check, 165 }; 166 167 int 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 } 189