[PATCH weston v0 05/11] tests: add screenshot recording to weston-test

[Derek Foreman]

On 10/11/14 12:31 PM, Bill Spitzak wrote:
> It is possible to call libpng directly if you want to avoid the cairo
> dependency.

Yup, certainly is - and you probably know how disgusting the resulting
code looks. ;)

If adding the cairo dependency to weston-test is a bad move I'll use
libpng directly - however, we already need cairo for some of the other
bits, just not for weston-test yet.

If nobody complains too loudly though, it's way nicer to write a png in
a couple of cairo calls than to do the requisite setjmp shenanigans for
libpng.  Using cairo also makes bryce's follow on patch to mask a region
look a little more intuitive than it would otherwise.


> 
> On 11/09/2014 01:41 PM, Bryce Harrington wrote:
>> From: Derek Foreman <derekf at osg.samsung.com>
>>
>> Adds wl_test_record_screenshot() to weston test.  This commit also
>> adds a dependency on cairo to weston-test to use it for writing PNG
>> files.
>>
>> Signed-off-by: Bryce Harrington <bryce at osg.samsung.com>
>> ---
>>   Makefile.am               |  4 +--
>>   protocol/wayland-test.xml |  3 +++
>>   tests/weston-test.c       | 68
>> +++++++++++++++++++++++++++++++++++++++++++++++
>>   3 files changed, 73 insertions(+), 2 deletions(-)
>>
>> diff --git a/Makefile.am b/Makefile.am
>> index 1e7cc81..26dd473 100644
>> --- a/Makefile.am
>> +++ b/Makefile.am
>> @@ -881,7 +881,7 @@ noinst_PROGRAMS +=            \
>>       matrix-test
>>
>>   test_module_ldflags = \
>> -    -module -avoid-version -rpath $(libdir) $(COMPOSITOR_LIBS)
>> +    -module -avoid-version -rpath $(libdir) $(COMPOSITOR_LIBS)
>> $(CAIRO_LIBS)
>>
>>   surface_global_test_la_SOURCES = tests/surface-global-test.c
>>   surface_global_test_la_LDFLAGS = $(test_module_ldflags)
>> @@ -893,7 +893,7 @@ surface_test_la_CFLAGS = $(GCC_CFLAGS)
>> $(COMPOSITOR_CFLAGS)
>>
>>   weston_test_la_LIBADD = $(COMPOSITOR_LIBS) libshared.la
>>   weston_test_la_LDFLAGS = $(test_module_ldflags)
>> -weston_test_la_CFLAGS = $(GCC_CFLAGS) $(COMPOSITOR_CFLAGS)
>> +weston_test_la_CFLAGS = $(GCC_CFLAGS) $(COMPOSITOR_CFLAGS)
>> $(CAIRO_CFLAGS)
>>   weston_test_la_SOURCES = tests/weston-test.c
>>   nodist_weston_test_la_SOURCES =            \
>>       protocol/wayland-test-protocol.c    \
>> diff --git a/protocol/wayland-test.xml b/protocol/wayland-test.xml
>> index 18b6625..a22a6ac 100644
>> --- a/protocol/wayland-test.xml
>> +++ b/protocol/wayland-test.xml
>> @@ -58,5 +58,8 @@
>>       <event name="n_egl_buffers">
>>         <arg name="n" type="uint"/>
>>       </event>
>> +    <request name="record_screenshot">
>> +      <arg name="basename" type="string"/>
>> +    </request>
>>     </interface>
>>   </protocol>
>> diff --git a/tests/weston-test.c b/tests/weston-test.c
>> index f1e45c1..16f20c6 100644
>> --- a/tests/weston-test.c
>> +++ b/tests/weston-test.c
>> @@ -35,6 +35,8 @@
>>   #include <EGL/eglext.h>
>>   #endif /* ENABLE_EGL */
>>
>> +#include <cairo.h>
>> +
>>   struct weston_test {
>>       struct weston_compositor *compositor;
>>       struct weston_layer layer;
>> @@ -235,6 +237,71 @@ get_n_buffers(struct wl_client *client, struct
>> wl_resource *resource)
>>       wl_test_send_n_egl_buffers(resource, n_buffers);
>>   }
>>
>> +static void
>> +dump_image(const char *filename, int x, int y, uint32_t *image)
>> +{
>> +    cairo_surface_t *surface, *flipped;
>> +    cairo_t *cr;
>> +
>> +    surface = cairo_image_surface_create_for_data((unsigned char
>> *)image,
>> +                              CAIRO_FORMAT_ARGB32,
>> +                              x, y, x * 4);
>> +    flipped = cairo_surface_create_similar_image(surface,
>> CAIRO_FORMAT_ARGB32, x, y);
>> +
>> +    cr = cairo_create(flipped);
>> +    cairo_translate(cr, 0.0, y);
>> +    cairo_scale(cr, 1.0, -1.0);
>> +    cairo_set_source_surface(cr, surface, 0, 0);
>> +    cairo_paint(cr);
>> +    cairo_destroy(cr);
>> +    cairo_surface_destroy(surface);
>> +
>> +    cairo_surface_write_to_png(flipped, filename);
>> +    cairo_surface_destroy(flipped);
>> +}
>> +
>> +static void
>> +record_screenshot(struct wl_client *client, struct wl_resource
>> *resource,
>> +          const char *basename)
>> +{
>> +    struct weston_output *o;
>> +    struct weston_test *test = wl_resource_get_user_data(resource);
>> +    char *filename;
>> +    uint32_t *buffer;
>> +    int w, h, head = 0;
>> +
>> +    wl_list_for_each(o, &test->compositor->output_list, link) {
>> +        switch (o->transform) {
>> +        case WL_OUTPUT_TRANSFORM_90:
>> +        case WL_OUTPUT_TRANSFORM_270:
>> +        case WL_OUTPUT_TRANSFORM_FLIPPED_90:
>> +        case WL_OUTPUT_TRANSFORM_FLIPPED_270:
>> +            w = o->height;
>> +            h = o->width;
>> +            break;
>> +        default:
>> +            w = o->width;
>> +            h = o->height;
>> +            break;
>> +        }
>> +        buffer = malloc(w * h * 4);
>> +        if (!buffer)
>> +            return;
>> +
>> +        test->compositor->renderer->read_pixels(o,
>> +                          o->compositor->read_format,
>> +                          buffer, 0, 0, w, h);
>> +
>> +        if (asprintf(&filename, "%s-%d.png", basename, head) < 0)
>> +            return;
>> +
>> +        dump_image(filename, w, h, buffer);
>> +        free(filename);
>> +        free(buffer);
>> +        head++;
>> +    }
>> +}
>> +
>>   static const struct wl_test_interface test_implementation = {
>>       move_surface,
>>       move_pointer,
>> @@ -242,6 +309,7 @@ static const struct wl_test_interface
>> test_implementation = {
>>       activate_surface,
>>       send_key,
>>       get_n_buffers,
>> +    record_screenshot
>>   };
>>
>>   static void
>>