diff --git a/include/wlr/types/wlr_scene.h b/include/wlr/types/wlr_scene.h index 622dbb28..bca0b269 100644 --- a/include/wlr/types/wlr_scene.h +++ b/include/wlr/types/wlr_scene.h @@ -282,6 +282,13 @@ bool wlr_scene_output_commit(struct wlr_scene_output *scene_output); */ void wlr_scene_output_for_each_surface(struct wlr_scene_output *scene_output, wlr_surface_iterator_func_t iterator, void *user_data); +/** + * Get a scene-graph output from a wlr_output. + * + * If the output hasn't been added to the scene-graph, returns NULL. + */ +struct wlr_scene_output *wlr_scene_get_scene_output(struct wlr_scene *scene, + struct wlr_output *output); /** * Attach an output layout to a scene. diff --git a/types/scene/wlr_scene.c b/types/scene/wlr_scene.c index 03439223..d1236fe0 100644 --- a/types/scene/wlr_scene.c +++ b/types/scene/wlr_scene.c @@ -828,6 +828,18 @@ void wlr_scene_output_destroy(struct wlr_scene_output *scene_output) { free(scene_output); } +struct wlr_scene_output *wlr_scene_get_scene_output(struct wlr_scene *scene, + struct wlr_output *output) { + struct wlr_addon *addon = + wlr_addon_find(&output->addons, scene, &output_addon_impl); + if (addon == NULL) { + return NULL; + } + struct wlr_scene_output *scene_output = + wl_container_of(addon, scene_output, addon); + return scene_output; +} + void wlr_scene_output_set_position(struct wlr_scene_output *scene_output, int lx, int ly) { if (scene_output->x == lx && scene_output->y == ly) {