src/screenshooter.c

   1 /*
   2  * Copyright © 2008-2011 Kristian Høgsberg
   3  *
   4  * Permission to use, copy, modify, distribute, and sell this software and
   5  * its documentation for any purpose is hereby granted without fee, provided
   6  * that the above copyright notice appear in all copies and that both that
   7  * copyright notice and this permission notice appear in supporting
   8  * documentation, and that the name of the copyright holders not be used in
   9  * advertising or publicity pertaining to distribution of the software
  10  * without specific, written prior permission.  The copyright holders make
  11  * no representations about the suitability of this software for any
  12  * purpose.  It is provided "as is" without express or implied warranty.
  13  *
  14  * THE COPYRIGHT HOLDERS DISCLAIM ALL WARRANTIES WITH REGARD TO THIS
  15  * SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
  16  * FITNESS, IN NO EVENT SHALL THE COPYRIGHT HOLDERS BE LIABLE FOR ANY
  17  * SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER
  18  * RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF
  19  * CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
  20  * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  21  */
  22
  23 #include <stdlib.h>
  24 #include <stdio.h>
  25 #include <string.h>
  26 #include <linux/input.h>
  27 #include <fcntl.h>
  28 #include <unistd.h>
  29 #include <sys/uio.h>
  30
  31 #include "compositor.h"
  32 #include "screenshooter-server-protocol.h"
  33
  34 #include "../wcap/wcap-decode.h"
  35
  36 struct screenshooter {
  37         struct wl_object base;
  38         struct weston_compositor *ec;
  39         struct wl_global *global;
  40         struct wl_client *client;
  41         struct weston_process process;
  42         struct wl_listener destroy_listener;
  43 };
  44
  45 struct screenshooter_read_pixels {
  46         struct weston_read_pixels base;
  47         struct wl_buffer *buffer;
  48         struct wl_resource *resource;
  49 };
  50
  51 static void
  52 copy_bgra_yflip(uint8_t *dst, uint8_t *src, int height, int stride)
  53 {
  54         uint8_t *end;
  55
  56         end = dst + height * stride;
  57         while (dst < end) {
  58                 memcpy(dst, src, stride);
  59                 dst += stride;
  60                 src -= stride;
  61         }
  62 }
  63
  64 static void
  65 copy_row_swap_RB(void *vdst, void *vsrc, int bytes)
  66 {
  67         uint32_t *dst = vdst;
  68         uint32_t *src = vsrc;
  69         uint32_t *end = dst + bytes / 4;
  70
  71         while (dst < end) {
  72                 uint32_t v = *src++;
  73                 /*                    A R G B */
  74                 uint32_t tmp = v & 0xff00ff00;
  75                 tmp |= (v >> 16) & 0x000000ff;
  76                 tmp |= (v << 16) & 0x00ff0000;
  77                 *dst++ = tmp;
  78         }
  79 }
  80
  81 static void
  82 copy_rgba_yflip(uint8_t *dst, uint8_t *src, int height, int stride)
  83 {
  84         uint8_t *end;
  85
  86         end = dst + height * stride;
  87         while (dst < end) {
  88                 copy_row_swap_RB(dst, src, stride);
  89                 dst += stride;
  90                 src -= stride;
  91         }
  92 }
  93
  94 static void
  95 screenshooter_read_pixels_done(struct weston_read_pixels *base,
  96                                struct weston_output *output)
  97 {
  98         struct screenshooter_read_pixels *r =
  99                 (struct screenshooter_read_pixels *) base;
 100         int32_t stride;
 101         uint8_t *d, *s;
 102
 103         stride = wl_shm_buffer_get_stride(r->buffer);
 104
 105         d = wl_shm_buffer_get_data(r->buffer);
 106         s = r->base.data + stride * (r->buffer->height - 1);
 107
 108         switch (output->compositor->read_format) {
 109         case GL_BGRA_EXT:
 110                 copy_bgra_yflip(d, s, output->current->height, stride);
 111                 break;
 112         case GL_RGBA:
 113                 copy_rgba_yflip(d, s, output->current->height, stride);
 114                 break;
 115         default:
 116                 break;
 117         }
 118
 119         wl_list_remove(&r->base.link);
 120
 121         screenshooter_send_done(r->resource);
 122         free(r->base.data);
 123         free(r);
 124
 125 }
 126
 127 static void
 128 screenshooter_shoot(struct wl_client *client,
 129                     struct wl_resource *resource,
 130                     struct wl_resource *output_resource,
 131                     struct wl_resource *buffer_resource)
 132 {
 133         struct weston_output *output = output_resource->data;
 134         struct screenshooter_read_pixels *r;
 135         struct wl_buffer *buffer = buffer_resource->data;
 136         int32_t stride;
 137
 138         if (!wl_buffer_is_shm(buffer))
 139                 return;
 140
 141         if (buffer->width < output->current->width ||
 142             buffer->height < output->current->height)
 143                 return;
 144
 145         r = malloc(sizeof *r);
 146         if (r == NULL) {
 147                 wl_resource_post_no_memory(resource);
 148                 return;
 149         }
 150
 151         r->base.x = 0;
 152         r->base.y = 0;
 153         r->base.width = output->current->width;
 154         r->base.height = output->current->height;
 155         r->base.done = screenshooter_read_pixels_done;
 156         r->buffer = buffer;
 157         r->resource = resource;
 158         stride = buffer->width * 4;
 159         r->base.data = malloc(stride * buffer->height);
 160
 161         if (r->base.data == NULL) {
 162                 free(r);
 163                 wl_resource_post_no_memory(resource);
 164                 return;
 165         }
 166
 167         wl_list_insert(output->read_pixels_list.prev, &r->base.link);
 168         weston_compositor_schedule_repaint(output->compositor);
 169 }
 170
 171 struct screenshooter_interface screenshooter_implementation = {
 172         screenshooter_shoot
 173 };
 174
 175 static void
 176 bind_shooter(struct wl_client *client,
 177              void *data, uint32_t version, uint32_t id)
 178 {
 179         struct screenshooter *shooter = data;
 180         struct wl_resource *resource;
 181
 182         resource = wl_client_add_object(client, &screenshooter_interface,
 183                              &screenshooter_implementation, id, data);
 184
 185         if (client != shooter->client) {
 186                 wl_resource_post_error(resource, WL_DISPLAY_ERROR_INVALID_OBJECT,
 187                                        "screenshooter failed: permission denied");
 188                 wl_resource_destroy(resource);
 189         }
 190 }
 191
 192 static void
 193 screenshooter_sigchld(struct weston_process *process, int status)
 194 {
 195         struct screenshooter *shooter =
 196                 container_of(process, struct screenshooter, process);
 197
 198         shooter->client = NULL;
 199 }
 200
 201 static void
 202 screenshooter_binding(struct wl_seat *seat, uint32_t time,
 203                  uint32_t key, uint32_t button, uint32_t axis,
 204                  int32_t state, void *data)
 205 {
 206         struct screenshooter *shooter = data;
 207         const char *screenshooter_exe = LIBEXECDIR "/weston-screenshooter";
 208
 209         if (!shooter->client)
 210                 shooter->client = weston_client_launch(shooter->ec,
 211                                         &shooter->process,
 212                                         screenshooter_exe, screenshooter_sigchld);
 213 }
 214
 215 struct weston_recorder {
 216         struct weston_output *output;
 217         uint32_t *frame, *rect;
 218         uint32_t total;
 219         int fd;
 220         struct wl_listener frame_listener;
 221         int count;
 222 };
 223
 224 static uint32_t *
 225 output_run(uint32_t *p, uint32_t delta, int run)
 226 {
 227         int i;
 228
 229         while (run > 0) {
 230                 if (run <= 0xe0) {
 231                         *p++ = delta | ((run - 1) << 24);
 232                         break;
 233                 }
 234
 235                 i = 24 - __builtin_clz(run);
 236                 *p++ = delta | ((i + 0xe0) << 24);
 237                 run -= 1 << (7 + i);
 238         }
 239
 240         return p;
 241 }
 242
 243 static void
 244 weston_recorder_frame_notify(struct wl_listener *listener, void *data)
 245 {
 246         struct weston_recorder *recorder =
 247                 container_of(listener, struct weston_recorder, frame_listener);
 248         struct weston_output *output = recorder->output;
 249         uint32_t msecs = * (uint32_t *) data;
 250         pixman_box32_t *r;
 251         pixman_region32_t damage;
 252         int i, j, k, n, width, height, run, stride;
 253         uint32_t delta, prev, *d, *s, *p, next;
 254         struct {
 255                 uint32_t msecs;
 256                 uint32_t nrects;
 257         } header;
 258         struct iovec v[2];
 259
 260         pixman_region32_init(&damage);
 261         pixman_region32_intersect(&damage, &recorder->output->region,
 262                                   &recorder->output->previous_damage);
 263
 264         r = pixman_region32_rectangles(&damage, &n);
 265         if (n == 0)
 266                 return;
 267         if (recorder->count++ == 0)
 268                 /* The first callback gives us the frame immediately
 269                  * before the weston_output_damage() call, and
 270                  * typically doesn't give us a full frame of damage.*/
 271                 return;
 272
 273         header.msecs = msecs;
 274         header.nrects = n;
 275         v[0].iov_base = &header;
 276         v[0].iov_len = sizeof header;
 277         v[1].iov_base = r;
 278         v[1].iov_len = n * sizeof *r;
 279         recorder->total += writev(recorder->fd, v, 2);
 280         stride = output->current->width;
 281
 282         for (i = 0; i < n; i++) {
 283                 width = r[i].x2 - r[i].x1;
 284                 height = r[i].y2 - r[i].y1;
 285                 glReadPixels(r[i].x1, output->current->height - r[i].y2,
 286                              width, height,
 287                              output->compositor->read_format,
 288                              GL_UNSIGNED_BYTE, recorder->rect);
 289
 290                 s = recorder->rect;
 291                 p = recorder->rect;
 292                 run = prev = 0; /* quiet gcc */
 293                 for (j = 0; j < height; j++) {
 294                         d = recorder->frame +
 295                                 stride * (r[i].y2 - j - 1) + r[i].x1;
 296                         for (k = 0; k < width; k++) {
 297                                 next = *s++;
 298                                 delta = (next - *d) & 0x00ffffff;
 299                                 *d++ = next;
 300                                 if (run == 0 || delta == prev) {
 301                                         run++;
 302                                 } else {
 303                                         p = output_run(p, prev, run);
 304                                         run = 1;
 305                                         prev = delta;
 306                                 }
 307                         }
 308                 }
 309
 310                 p = output_run(p, prev, run);
 311
 312                 recorder->total += write(recorder->fd,
 313                                          recorder->rect,
 314                                          (p - recorder->rect) * 4);
 315
 316 #if 0
 317                 fprintf(stderr,
 318                         "%dx%d at %d,%d rle from %d to %d bytes (%f) total %dM\n",
 319                         width, height, r[i].x1, r[i].y1,
 320                         width * height * 4, (int) (p - rect) * 4,
 321                         (float) (p - rect) / (width * height),
 322                         total / 1024 / 1024);
 323 #endif
 324         }
 325
 326         pixman_region32_fini(&damage);
 327 }
 328
 329 static void
 330 weston_recorder_create(struct weston_output *output, const char *filename)
 331 {
 332         struct weston_recorder *recorder;
 333         int stride, size;
 334         struct { uint32_t magic, format, width, height; } header;
 335
 336         recorder = malloc(sizeof *recorder);
 337         recorder->output = output;
 338
 339         stride = output->current->width;
 340         size = stride * 4 * output->current->height;
 341         recorder->frame = malloc(size);
 342         recorder->rect = malloc(size);
 343         recorder->total = 0;
 344         recorder->count = 0;
 345         memset(recorder->frame, 0, size);
 346
 347         recorder->fd = open(filename,
 348                             O_WRONLY | O_CREAT | O_TRUNC | O_CLOEXEC, 0644);
 349
 350         header.magic = WCAP_HEADER_MAGIC;
 351
 352         switch (output->compositor->read_format) {
 353         case GL_BGRA_EXT:
 354                 header.format = WCAP_FORMAT_XRGB8888;
 355                 break;
 356         case GL_RGBA:
 357                 header.format = WCAP_FORMAT_XBGR8888;
 358                 break;
 359         }
 360
 361         header.width = output->current->width;
 362         header.height = output->current->height;
 363         recorder->total += write(recorder->fd, &header, sizeof header);
 364
 365         recorder->frame_listener.notify = weston_recorder_frame_notify;
 366         wl_signal_add(&output->frame_signal, &recorder->frame_listener);
 367         weston_output_damage(output);
 368 }
 369
 370 static void
 371 weston_recorder_destroy(struct weston_recorder *recorder)
 372 {
 373         wl_list_remove(&recorder->frame_listener.link);
 374         close(recorder->fd);
 375         free(recorder->frame);
 376         free(recorder->rect);
 377         free(recorder);
 378 }
 379
 380 static void
 381 recorder_binding(struct wl_seat *seat, uint32_t time,
 382                  uint32_t key, uint32_t button, uint32_t axis,
 383                  int32_t state, void *data)
 384 {
 385         struct weston_seat *ws = (struct weston_seat *) seat;
 386         struct weston_compositor *ec = ws->compositor;
 387         struct weston_output *output =
 388                 container_of(ec->output_list.next,
 389                              struct weston_output, link);
 390         struct wl_listener *listener;
 391         struct weston_recorder *recorder;
 392         static const char filename[] = "capture.wcap";
 393
 394         listener = wl_signal_get(&output->frame_signal,
 395                                  weston_recorder_frame_notify);
 396         if (listener) {
 397                 recorder = container_of(listener, struct weston_recorder,
 398                                         frame_listener);
 399
 400                 fprintf(stderr,
 401                         "stopping recorder, total file size %dM, %d frames\n",
 402                         recorder->total / (1024 * 1024), recorder->count);
 403
 404                 weston_recorder_destroy(recorder);
 405         } else {
 406                 fprintf(stderr, "starting recorder, file %s\n", filename);
 407                 weston_recorder_create(output, filename);
 408         }
 409 }
 410
 411 static void
 412 screenshooter_destroy(struct wl_listener *listener, void *data)
 413 {
 414         struct screenshooter *shooter =
 415                 container_of(listener, struct screenshooter, destroy_listener);
 416
 417         wl_display_remove_global(shooter->ec->wl_display, shooter->global);
 418         free(shooter);
 419 }
 420
 421 void
 422 screenshooter_create(struct weston_compositor *ec)
 423 {
 424         struct screenshooter *shooter;
 425
 426         shooter = malloc(sizeof *shooter);
 427         if (shooter == NULL)
 428                 return;
 429
 430         shooter->base.interface = &screenshooter_interface;
 431         shooter->base.implementation =
 432                 (void(**)(void)) &screenshooter_implementation;
 433         shooter->ec = ec;
 434         shooter->client = NULL;
 435
 436         shooter->global = wl_display_add_global(ec->wl_display,
 437                                                 &screenshooter_interface,
 438                                                 shooter, bind_shooter);
 439         weston_compositor_add_binding(ec, KEY_S, 0, 0, MODIFIER_SUPER,
 440                                         screenshooter_binding, shooter);
 441         weston_compositor_add_binding(ec, KEY_R, 0, 0, MODIFIER_SUPER,
 442                                         recorder_binding, shooter);
 443
 444         shooter->destroy_listener.notify = screenshooter_destroy;
 445         wl_signal_add(&ec->destroy_signal, &shooter->destroy_listener);
 446 }