# HG changeset patch # User gpoirier # Date 1165962728 0 # Node ID 003fede0eb671b9d8687b29c0c21805d20ce14f4 # Parent 6b8f97401a345ddd5f49e9d2bcddfa8f2e597e07 RFC 3&4 stage: FFmpeg style aint't my style - Tabs are 4 width - No tabs allowed diff -r 6b8f97401a34 -r 003fede0eb67 x11grab.c --- a/x11grab.c Tue Dec 12 22:32:06 2006 +0000 +++ b/x11grab.c Tue Dec 12 22:32:08 2006 +0000 @@ -8,7 +8,7 @@ * Edouard Gomez * * This file contains code from grab.c: - * Copyright (c) 2000-2001 Fabrice Bellard + * Copyright (c) 2000-2001 Fabrice Bellard * * This file contains code from the xvidcap project: * Copyright (C) 1997-1998 Rasca, Berlin @@ -48,273 +48,273 @@ typedef struct { - Display *dpy; - int frame_format; - int frame_size; - int frame_rate; - int frame_rate_base; - int64_t time_frame; + Display *dpy; + int frame_format; + int frame_size; + int frame_rate; + int frame_rate_base; + int64_t time_frame; - int height; - int width; - int x_off; - int y_off; - XImage *image; - int use_shm; - XShmSegmentInfo shminfo; - int mouse_wanted; + int height; + int width; + int x_off; + int y_off; + XImage *image; + int use_shm; + XShmSegmentInfo shminfo; + int mouse_wanted; } X11Grab; static int x11grab_read_header(AVFormatContext *s1, AVFormatParameters *ap) { - X11Grab *x11grab = s1->priv_data; - Display *dpy; - AVStream *st = NULL; - int width, height; - int frame_rate, frame_rate_base, frame_size; - int input_pixfmt; - XImage *image; - int x_off=0; int y_off = 0; - int use_shm; + X11Grab *x11grab = s1->priv_data; + Display *dpy; + AVStream *st = NULL; + int width, height; + int frame_rate, frame_rate_base, frame_size; + int input_pixfmt; + XImage *image; + int x_off=0; int y_off = 0; + int use_shm; + + dpy = XOpenDisplay(NULL); + if(!dpy) { + goto fail; + } - dpy = XOpenDisplay(NULL); - if(!dpy) { - goto fail; - } + sscanf(ap->device, "x11:%d,%d", &x_off, &y_off); + av_log(s1, AV_LOG_INFO, "device: %s -> x: %d y: %d width: %d height: %d\n", ap->device, x_off, y_off, ap->width, ap->height); + + if (!ap || ap->width <= 0 || ap->height <= 0 || ap->time_base.den <= 0) { + av_log(s1, AV_LOG_ERROR, "AVParameters don't have any video size. Use -s.\n"); + return AVERROR_IO; + } + + width = ap->width; + height = ap->height; + frame_rate = ap->time_base.den; + frame_rate_base = ap->time_base.num; + + st = av_new_stream(s1, 0); + if (!st) { + return -ENOMEM; + } + av_set_pts_info(st, 64, 1, 1000000); /* 64 bits pts in us */ - sscanf(ap->device, "x11:%d,%d", &x_off, &y_off); - av_log(s1, AV_LOG_INFO, "device: %s -> x: %d y: %d width: %d height: %d\n", ap->device, x_off, y_off, ap->width, ap->height); - - if (!ap || ap->width <= 0 || ap->height <= 0 || ap->time_base.den <= 0) { - av_log(s1, AV_LOG_ERROR, "AVParameters don't have any video size. Use -s.\n"); - return AVERROR_IO; - } + use_shm = XShmQueryExtension(dpy); + av_log(s1, AV_LOG_INFO, "shared memory extension %s\n", use_shm ? "found" : "not found"); - width = ap->width; - height = ap->height; - frame_rate = ap->time_base.den; - frame_rate_base = ap->time_base.num; + if(use_shm) { + int scr = XDefaultScreen(dpy); + image = XShmCreateImage(dpy, + DefaultVisual(dpy, scr), + DefaultDepth(dpy, scr), + ZPixmap, + NULL, + &x11grab->shminfo, + ap->width, ap->height); + x11grab->shminfo.shmid = shmget(IPC_PRIVATE, + image->bytes_per_line * image->height, + IPC_CREAT|0777); + if (x11grab->shminfo.shmid == -1) { + av_log(s1, AV_LOG_ERROR, "Fatal: Can't get shared memory!\n"); + return -ENOMEM; + } + x11grab->shminfo.shmaddr = image->data = shmat(x11grab->shminfo.shmid, 0, 0); + x11grab->shminfo.readOnly = False; - st = av_new_stream(s1, 0); - if (!st) { - return -ENOMEM; - } - av_set_pts_info(st, 64, 1, 1000000); /* 64 bits pts in us */ - - use_shm = XShmQueryExtension(dpy); - av_log(s1, AV_LOG_INFO, "shared memory extension %s\n", use_shm ? "found" : "not found"); + if (!XShmAttach(dpy, &x11grab->shminfo)) { + av_log(s1, AV_LOG_ERROR, "Fatal: Failed to attach shared memory!\n"); + /* needs some better error subroutine :) */ + return AVERROR_IO; + } + } else { + image = XGetImage(dpy, RootWindow(dpy, DefaultScreen(dpy)), + x_off,y_off, + ap->width,ap->height, + AllPlanes, ZPixmap); + } - if(use_shm) { - int scr = XDefaultScreen(dpy); - image = XShmCreateImage(dpy, - DefaultVisual(dpy, scr), - DefaultDepth(dpy, scr), - ZPixmap, - NULL, - &x11grab->shminfo, - ap->width, ap->height); - x11grab->shminfo.shmid = shmget(IPC_PRIVATE, - image->bytes_per_line * image->height, - IPC_CREAT|0777); - if (x11grab->shminfo.shmid == -1) { - av_log(s1, AV_LOG_ERROR, "Fatal: Can't get shared memory!\n"); - return -ENOMEM; - } - x11grab->shminfo.shmaddr = image->data = shmat(x11grab->shminfo.shmid, 0, 0); - x11grab->shminfo.readOnly = False; - - if (!XShmAttach(dpy, &x11grab->shminfo)) { - av_log(s1, AV_LOG_ERROR, "Fatal: Failed to attach shared memory!\n"); - /* needs some better error subroutine :) */ - return AVERROR_IO; - } - } else { - image = XGetImage(dpy, RootWindow(dpy, DefaultScreen(dpy)), - x_off,y_off, - ap->width,ap->height, - AllPlanes, ZPixmap); - } - - switch (image->bits_per_pixel) { - case 8: - av_log (s1, AV_LOG_DEBUG, "8 bit pallete\n"); - input_pixfmt = PIX_FMT_PAL8; - break; - case 16: - if ( image->red_mask == 0xF800 && image->green_mask == 0x07E0 - && image->blue_mask == 0x1F ) { - av_log (s1, AV_LOG_DEBUG, "16 bit RGB565\n"); - input_pixfmt = PIX_FMT_RGB565; - } else if ( image->red_mask == 0x7C00 && - image->green_mask == 0x03E0 && - image->blue_mask == 0x1F ) { - av_log(s1, AV_LOG_DEBUG, "16 bit RGB555\n"); - input_pixfmt = PIX_FMT_RGB555; - } else { - av_log(s1, AV_LOG_ERROR, "RGB ordering at image depth %i not supported ... aborting\n", image->bits_per_pixel); - av_log(s1, AV_LOG_ERROR, "color masks: r 0x%.6lx g 0x%.6lx b 0x%.6lx\n", image->red_mask, image->green_mask, image->blue_mask); - return AVERROR_IO; - } - break; - case 24: - if ( image->red_mask == 0xFF0000 && - image->green_mask == 0xFF00 - && image->blue_mask == 0xFF ) { - input_pixfmt = PIX_FMT_BGR24; - } else if ( image->red_mask == 0xFF && image->green_mask == 0xFF00 - && image->blue_mask == 0xFF0000 ) { - input_pixfmt = PIX_FMT_RGB24; - } else { - av_log(s1, AV_LOG_ERROR,"rgb ordering at image depth %i not supported ... aborting\n", image->bits_per_pixel); - av_log(s1, AV_LOG_ERROR, "color masks: r 0x%.6lx g 0x%.6lx b 0x%.6lx\n", image->red_mask, image->green_mask, image->blue_mask); - return AVERROR_IO; - } - break; - case 32: + switch (image->bits_per_pixel) { + case 8: + av_log (s1, AV_LOG_DEBUG, "8 bit pallete\n"); + input_pixfmt = PIX_FMT_PAL8; + break; + case 16: + if ( image->red_mask == 0xF800 && image->green_mask == 0x07E0 + && image->blue_mask == 0x1F ) { + av_log (s1, AV_LOG_DEBUG, "16 bit RGB565\n"); + input_pixfmt = PIX_FMT_RGB565; + } else if ( image->red_mask == 0x7C00 && + image->green_mask == 0x03E0 && + image->blue_mask == 0x1F ) { + av_log(s1, AV_LOG_DEBUG, "16 bit RGB555\n"); + input_pixfmt = PIX_FMT_RGB555; + } else { + av_log(s1, AV_LOG_ERROR, "RGB ordering at image depth %i not supported ... aborting\n", image->bits_per_pixel); + av_log(s1, AV_LOG_ERROR, "color masks: r 0x%.6lx g 0x%.6lx b 0x%.6lx\n", image->red_mask, image->green_mask, image->blue_mask); + return AVERROR_IO; + } + break; + case 24: + if ( image->red_mask == 0xFF0000 && + image->green_mask == 0xFF00 + && image->blue_mask == 0xFF ) { + input_pixfmt = PIX_FMT_BGR24; + } else if ( image->red_mask == 0xFF && image->green_mask == 0xFF00 + && image->blue_mask == 0xFF0000 ) { + input_pixfmt = PIX_FMT_RGB24; + } else { + av_log(s1, AV_LOG_ERROR,"rgb ordering at image depth %i not supported ... aborting\n", image->bits_per_pixel); + av_log(s1, AV_LOG_ERROR, "color masks: r 0x%.6lx g 0x%.6lx b 0x%.6lx\n", image->red_mask, image->green_mask, image->blue_mask); + return AVERROR_IO; + } + break; + case 32: #if 0 - GetColorInfo (image, &c_info); - if ( c_info.alpha_mask == 0xFF000000 && image->green_mask == 0xFF00 ) { - /* byte order is relevant here, not endianness - * endianness is handled by avcodec, but atm no such thing - * as having ABGR, instead of ARGB in a word. Since we - * need this for Solaris/SPARC, but need to do the conversion - * for every frame we do it outside of this loop, cf. below - * this matches both ARGB32 and ABGR32 */ - input_pixfmt = PIX_FMT_ARGB32; - } else { - av_log(s1, AV_LOG_ERROR,"image depth %i not supported ... aborting\n", image->bits_per_pixel); - return AVERROR_IO; - } + GetColorInfo (image, &c_info); + if ( c_info.alpha_mask == 0xFF000000 && image->green_mask == 0xFF00 ) { + /* byte order is relevant here, not endianness + * endianness is handled by avcodec, but atm no such thing + * as having ABGR, instead of ARGB in a word. Since we + * need this for Solaris/SPARC, but need to do the conversion + * for every frame we do it outside of this loop, cf. below + * this matches both ARGB32 and ABGR32 */ + input_pixfmt = PIX_FMT_ARGB32; + } else { + av_log(s1, AV_LOG_ERROR,"image depth %i not supported ... aborting\n", image->bits_per_pixel); + return AVERROR_IO; + } #endif - input_pixfmt = PIX_FMT_RGBA32; - break; - default: - av_log(s1, AV_LOG_ERROR, "image depth %i not supported ... aborting\n", image->bits_per_pixel); - return -1; - } + input_pixfmt = PIX_FMT_RGBA32; + break; + default: + av_log(s1, AV_LOG_ERROR, "image depth %i not supported ... aborting\n", image->bits_per_pixel); + return -1; + } - frame_size = width * height * image->bits_per_pixel/8; - x11grab->frame_size = frame_size; - x11grab->dpy = dpy; - x11grab->width = ap->width; - x11grab->height = ap->height; - x11grab->frame_rate = frame_rate; - x11grab->frame_rate_base = frame_rate_base; - x11grab->time_frame = av_gettime() * frame_rate / frame_rate_base; - x11grab->x_off = x_off; - x11grab->y_off = y_off; - x11grab->image = image; - x11grab->use_shm = use_shm; - x11grab->mouse_wanted = 1; + frame_size = width * height * image->bits_per_pixel/8; + x11grab->frame_size = frame_size; + x11grab->dpy = dpy; + x11grab->width = ap->width; + x11grab->height = ap->height; + x11grab->frame_rate = frame_rate; + x11grab->frame_rate_base = frame_rate_base; + x11grab->time_frame = av_gettime() * frame_rate / frame_rate_base; + x11grab->x_off = x_off; + x11grab->y_off = y_off; + x11grab->image = image; + x11grab->use_shm = use_shm; + x11grab->mouse_wanted = 1; - st->codec->codec_type = CODEC_TYPE_VIDEO; - st->codec->codec_id = CODEC_ID_RAWVIDEO; - st->codec->width = width; - st->codec->height = height; - st->codec->pix_fmt = input_pixfmt; - st->codec->time_base.den = frame_rate; - st->codec->time_base.num = frame_rate_base; - st->codec->bit_rate = frame_size * 1/av_q2d(st->codec->time_base) * 8; - - return 0; + st->codec->codec_type = CODEC_TYPE_VIDEO; + st->codec->codec_id = CODEC_ID_RAWVIDEO; + st->codec->width = width; + st->codec->height = height; + st->codec->pix_fmt = input_pixfmt; + st->codec->time_base.den = frame_rate; + st->codec->time_base.num = frame_rate_base; + st->codec->bit_rate = frame_size * 1/av_q2d(st->codec->time_base) * 8; + + return 0; fail: - av_free(st); - return AVERROR_IO; + av_free(st); + return AVERROR_IO; } static void getCurrentPointer(AVFormatContext *s1, X11Grab *s, int *x, int *y) { - Window mrootwindow, childwindow; - int dummy; - Display *dpy = s->dpy; - - mrootwindow = DefaultRootWindow(dpy); - - if (XQueryPointer(dpy, mrootwindow, &mrootwindow, &childwindow, - x, y, &dummy, &dummy, (unsigned int*)&dummy)) { - } else { - av_log(s1, AV_LOG_INFO, "couldn't find mouse pointer\n"); - *x = -1; - *y = -1; - } + Window mrootwindow, childwindow; + int dummy; + Display *dpy = s->dpy; + + mrootwindow = DefaultRootWindow(dpy); + + if (XQueryPointer(dpy, mrootwindow, &mrootwindow, &childwindow, + x, y, &dummy, &dummy, (unsigned int*)&dummy)) { + } else { + av_log(s1, AV_LOG_INFO, "couldn't find mouse pointer\n"); + *x = -1; + *y = -1; + } } -#define DRAW_CURSOR_TEMPLATE(type_t) \ -do { \ -type_t *cursor; \ -int width_cursor; \ -uint16_t bm_b, bm_w, mask; \ -\ -for (line = 0; line < min(20, (y_off + height) - *y); line++ ) { \ - if (s->mouse_wanted == 1) { \ - bm_b = mousePointerBlack[line]; \ - bm_w = mousePointerWhite[line]; \ - } else { \ - bm_b = mousePointerWhite[line]; \ - bm_w = mousePointerBlack[line]; \ - } \ - mask = ( 0x0001 << 15 ); \ -\ - for (cursor = (type_t*) im_data, width_cursor = 0; \ - ((width_cursor + *x) < (width + x_off) && width_cursor < 16); \ - cursor++, width_cursor++) { \ - if ( ( bm_b & mask ) > 0 ) { \ - *cursor &= ((~ image->red_mask) & (~ image->green_mask) & (~image->blue_mask )); \ - } else if ( ( bm_w & mask ) > 0 ) { \ - *cursor |= (image->red_mask | image->green_mask | image->blue_mask ); \ - } \ - mask >>= 1; \ - } \ - im_data += image->bytes_per_line; \ -} \ -} while (0) +#define DRAW_CURSOR_TEMPLATE(type_t) \ + do { \ + type_t *cursor; \ + int width_cursor; \ + uint16_t bm_b, bm_w, mask; \ + \ + for (line = 0; line < min(20, (y_off + height) - *y); line++ ) { \ + if (s->mouse_wanted == 1) { \ + bm_b = mousePointerBlack[line]; \ + bm_w = mousePointerWhite[line]; \ + } else { \ + bm_b = mousePointerWhite[line]; \ + bm_w = mousePointerBlack[line]; \ + } \ + mask = ( 0x0001 << 15 ); \ + \ + for (cursor = (type_t*) im_data, width_cursor = 0; \ + ((width_cursor + *x) < (width + x_off) && width_cursor < 16); \ + cursor++, width_cursor++) { \ + if ( ( bm_b & mask ) > 0 ) { \ + *cursor &= ((~ image->red_mask) & (~ image->green_mask) & (~image->blue_mask )); \ + } else if ( ( bm_w & mask ) > 0 ) { \ + *cursor |= (image->red_mask | image->green_mask | image->blue_mask ); \ + } \ + mask >>= 1; \ + } \ + im_data += image->bytes_per_line; \ + } \ + } while (0) static void paintMousePointer(AVFormatContext *s1, X11Grab *s, int *x, int *y, XImage *image) { - static const uint16_t const mousePointerBlack[] = - { - 0, 49152, 40960, 36864, 34816, - 33792, 33280, 33024, 32896, 32832, - 33728, 37376, 43264, 51456, 1152, - 1152, 576, 576, 448, 0 - }; + static const uint16_t const mousePointerBlack[] = + { + 0, 49152, 40960, 36864, 34816, + 33792, 33280, 33024, 32896, 32832, + 33728, 37376, 43264, 51456, 1152, + 1152, 576, 576, 448, 0 + }; - static const uint16_t const mousePointerWhite[] = - { - 0, 0, 16384, 24576, 28672, - 30720, 31744, 32256, 32512, 32640, - 31744, 27648, 17920, 1536, 768, - 768, 384, 384, 0, 0 - }; + static const uint16_t const mousePointerWhite[] = + { + 0, 0, 16384, 24576, 28672, + 30720, 31744, 32256, 32512, 32640, + 31744, 27648, 17920, 1536, 768, + 768, 384, 384, 0, 0 + }; - int x_off = s->x_off; - int y_off = s->y_off; - int width = s->width; - int height = s->height; + int x_off = s->x_off; + int y_off = s->y_off; + int width = s->width; + int height = s->height; - if ( (*x - x_off) >= 0 && *x < (width + x_off) - && (*y - y_off) >= 0 && *y < (height + y_off) ) { - int line; - uint8_t *im_data = (uint8_t*)image->data; - - im_data += (image->bytes_per_line * (*y - y_off)); // shift to right line - im_data += (image->bits_per_pixel / 8 * (*x - x_off)); // shift to right pixel - - switch(image->bits_per_pixel) { - case 32: - DRAW_CURSOR_TEMPLATE(uint32_t); - break; - case 16: - DRAW_CURSOR_TEMPLATE(uint16_t); - break; - default: - /* XXX: How do we deal with 24bit and 8bit modes ? */ - break; - } - } + if ( (*x - x_off) >= 0 && *x < (width + x_off) + && (*y - y_off) >= 0 && *y < (height + y_off) ) { + int line; + uint8_t *im_data = (uint8_t*)image->data; + + im_data += (image->bytes_per_line * (*y - y_off)); // shift to right line + im_data += (image->bits_per_pixel / 8 * (*x - x_off)); // shift to right pixel + + switch(image->bits_per_pixel) { + case 32: + DRAW_CURSOR_TEMPLATE(uint32_t); + break; + case 16: + DRAW_CURSOR_TEMPLATE(uint16_t); + break; + default: + /* XXX: How do we deal with 24bit and 8bit modes ? */ + break; + } + } }