Flip screenshots

This commit is contained in:
emersion 2017-10-07 12:22:48 +02:00
parent d4cc82f11a
commit ecd5263d4d
No known key found for this signature in database
GPG Key ID: 0FDE7BE0E88F5E48
1 changed files with 40 additions and 7 deletions

View File

@ -1,12 +1,25 @@
#include <assert.h> #include <assert.h>
#include <stdlib.h> #include <stdlib.h>
#include <string.h>
#include <wayland-server.h> #include <wayland-server.h>
#include <wlr/types/wlr_screenshooter.h> #include <wlr/types/wlr_screenshooter.h>
#include <wlr/types/wlr_output.h> #include <wlr/types/wlr_output.h>
#include <wlr/util/log.h> #include <wlr/util/log.h>
#include "screenshooter-protocol.h" #include "screenshooter-protocol.h"
static void copy_bgra_yflip(uint8_t *dst, uint8_t *src, int32_t height,
int32_t stride) {
uint8_t *end = dst + height * stride;
while (dst < end) {
memcpy(dst, src, stride);
dst += stride;
src -= stride;
}
}
struct screenshot_state { struct screenshot_state {
int32_t width, height, stride;
uint8_t *pixels;
struct wl_shm_buffer *shm_buffer; struct wl_shm_buffer *shm_buffer;
struct wlr_output *output; struct wlr_output *output;
struct wlr_screenshot *screenshot; struct wlr_screenshot *screenshot;
@ -14,14 +27,20 @@ struct screenshot_state {
}; };
static void output_frame_notify(struct wl_listener *listener, void *_data) { static void output_frame_notify(struct wl_listener *listener, void *_data) {
struct screenshot_state *state = wl_container_of(listener, state, frame_listener); struct screenshot_state *state = wl_container_of(listener, state,
frame_listener);
wlr_output_read_pixels(state->output, state->pixels);
void *data = wl_shm_buffer_get_data(state->shm_buffer); void *data = wl_shm_buffer_get_data(state->shm_buffer);
wl_shm_buffer_begin_access(state->shm_buffer); wl_shm_buffer_begin_access(state->shm_buffer);
wlr_output_read_pixels(state->output, data); copy_bgra_yflip(data, state->pixels + state->stride * (state->height - 1),
state->height, state->stride);
wl_shm_buffer_end_access(state->shm_buffer); wl_shm_buffer_end_access(state->shm_buffer);
free(state->pixels);
wl_list_remove(&listener->link); wl_list_remove(&listener->link);
orbital_screenshot_send_done(state->screenshot->resource); orbital_screenshot_send_done(state->screenshot->resource);
// TODO: free(state) // TODO: free(state)
@ -38,23 +57,36 @@ static void screenshooter_shoot(struct wl_client *client,
struct wl_shm_buffer *shm_buffer = wl_shm_buffer_get(_buffer); struct wl_shm_buffer *shm_buffer = wl_shm_buffer_get(_buffer);
int32_t width = wl_shm_buffer_get_width(shm_buffer); int32_t width = wl_shm_buffer_get_width(shm_buffer);
int32_t height = wl_shm_buffer_get_height(shm_buffer); int32_t height = wl_shm_buffer_get_height(shm_buffer);
// TODO: int32_t stride = wl_shm_buffer_get_stride(shm_buffer); int32_t stride = wl_shm_buffer_get_stride(shm_buffer);
if (width < output->width || height < output->height) { if (width < output->width || height < output->height) {
wlr_log(L_ERROR, "Invalid buffer: too small"); wlr_log(L_ERROR, "Invalid buffer: too small");
return; return;
} }
uint8_t *pixels = malloc(stride * height);
if (pixels == NULL) {
wl_client_post_no_memory(client);
return;
}
struct wlr_screenshot *screenshot; struct wlr_screenshot *screenshot;
if (!(screenshot = calloc(1, sizeof(struct wlr_screenshot)))) { if (!(screenshot = calloc(1, sizeof(struct wlr_screenshot)))) {
return; return;
} }
screenshot->output = _output; screenshot->output = _output;
screenshot->resource = wl_resource_create(client, screenshot->resource = wl_resource_create(client,
&orbital_screenshot_interface, wl_resource_get_version(_screenshooter), id); &orbital_screenshot_interface, wl_resource_get_version(_screenshooter),
wlr_log(L_DEBUG, "new screenshot %p (res %p)", screenshot, screenshot->resource); id);
wl_resource_set_implementation(screenshot->resource, NULL, screenshot, NULL); wlr_log(L_DEBUG, "new screenshot %p (res %p)", screenshot,
screenshot->resource);
wl_resource_set_implementation(screenshot->resource, NULL, screenshot,
NULL);
struct screenshot_state *state = calloc(1, sizeof(struct screenshot_state)); struct screenshot_state *state = calloc(1, sizeof(struct screenshot_state));
state->width = width;
state->height = height;
state->stride = stride;
state->pixels = pixels;
state->shm_buffer = shm_buffer; state->shm_buffer = shm_buffer;
state->output = output; state->output = output;
state->screenshot = screenshot; state->screenshot = screenshot;
@ -71,7 +103,8 @@ static void screenshooter_bind(struct wl_client *wl_client,
struct wlr_screenshooter *screenshooter = _screenshooter; struct wlr_screenshooter *screenshooter = _screenshooter;
assert(wl_client && screenshooter); assert(wl_client && screenshooter);
if (version > 1) { if (version > 1) {
wlr_log(L_ERROR, "Client requested unsupported screenshooter version, disconnecting"); wlr_log(L_ERROR, "Client requested unsupported screenshooter version,"
"disconnecting");
wl_client_destroy(wl_client); wl_client_destroy(wl_client);
return; return;
} }