Remove cairo dependency, write raw pixels

This commit is contained in:
emersion 2017-10-07 19:01:11 +02:00
parent b27b6cd69c
commit a87f016017
No known key found for this signature in database
GPG Key ID: 0FDE7BE0E88F5E48
4 changed files with 37 additions and 14 deletions

View File

@ -26,5 +26,5 @@ executable(
executable(
'screenshot',
'screenshot.c',
dependencies: [wayland_client, wlr_protos, dependency('cairo')],
dependencies: [wayland_client, wlr_protos],
)

View File

@ -33,7 +33,6 @@
#include <wayland-client.h>
#include <limits.h>
#include <sys/param.h>
#include <cairo.h>
#include <screenshooter-client-protocol.h>
#include "../backend/wayland/os-compatibility.c"
@ -174,11 +173,20 @@ create_shm_buffer(int width, int height, void **data_out)
return buffer;
}
static void argb_to_rgba(uint32_t *data, size_t height, size_t stride) {
size_t n = height*stride/4;
for (size_t i = 0; i < n; ++i) {
uint32_t v = data[i];
uint32_t rgb = v & 0x00ffffff;
uint32_t a = (v & 0xff000000) >> 24;
data[i] = (rgb << 8) | a;
}
}
static void
write_png(int width, int height)
{
int output_stride, buffer_stride, i;
cairo_surface_t *surface;
void *data, *d, *s;
struct screenshooter_output *output, *next;
@ -203,11 +211,13 @@ write_png(int width, int height)
free(output);
}
surface = cairo_image_surface_create_for_data(data,
CAIRO_FORMAT_RGB24,
width, height, buffer_stride);
cairo_surface_write_to_png(surface, "wayland-screenshot.png");
cairo_surface_destroy(surface);
argb_to_rgba(data, height, buffer_stride);
// TODO: call convert
FILE *f = fopen("wayland-screenshot", "w");
fwrite(data, buffer_stride * height, 1, f);
fclose(f);
free(data);
}

View File

@ -71,6 +71,10 @@ void wlr_output_swap_buffers(struct wlr_output *output);
void wlr_output_set_gamma(struct wlr_output *output,
uint16_t size, uint16_t *r, uint16_t *g, uint16_t *b);
uint16_t wlr_output_get_gamma_size(struct wlr_output *output);
/**
* Reads all pixels from the output and stores them as ARGB.
*/
void wlr_output_read_pixels(struct wlr_output *output, void *out_data);
#endif

View File

@ -8,7 +8,6 @@
#include <wlr/util/list.h>
#include <wlr/util/log.h>
#include <GLES2/gl2.h>
#include <GLES2/gl2ext.h>
#include <wlr/render/matrix.h>
#include <wlr/render/gles2.h>
#include <wlr/render.h>
@ -252,9 +251,19 @@ uint16_t wlr_output_get_gamma_size(struct wlr_output *output) {
return output->impl->get_gamma_size(output);
}
void wlr_output_read_pixels(struct wlr_output *output, void *out_data) {
// TODO: is wlr_output_make_current required?
wlr_output_make_current(output);
glReadPixels(0, 0, output->width, output->height, GL_BGRA_EXT,
GL_UNSIGNED_BYTE, out_data);
static void rgba_to_argb(uint32_t *data, size_t height, size_t stride) {
size_t n = height*stride/4;
for (size_t i = 0; i < n; ++i) {
uint32_t v = data[i];
uint32_t rgb = (v & 0xffffff00) >> 8;
uint32_t a = v & 0x000000ff;
data[i] = rgb | (a << 24);
}
}
void wlr_output_read_pixels(struct wlr_output *output, void *out_data) {
wlr_output_make_current(output);
glReadPixels(0, 0, output->width, output->height, GL_RGBA, GL_UNSIGNED_BYTE,
out_data);
rgba_to_argb(out_data, output->height, output->width*4);
}