Remove stdlib.h
[vlc.git] / modules / access / screen / x11.c
1 /*****************************************************************************
2  * x11.c: Screen capture module.
3  *****************************************************************************
4  * Copyright (C) 2004 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Gildas Bazin <gbazin@videolan.org>
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22  *****************************************************************************/
23
24 /*****************************************************************************
25  * Preamble
26  *****************************************************************************/
27
28 #include <vlc/vlc.h>
29
30 #include <X11/Xlib.h>
31 #include <X11/Xutil.h>
32
33 #include "screen.h"
34
35 int screen_InitCapture( demux_t *p_demux )
36 {
37     demux_sys_t *p_sys = p_demux->p_sys;
38     Display *p_display;
39     XWindowAttributes win_info;
40     int i_chroma;
41
42     /* Open the display */
43     p_display = XOpenDisplay( NULL );
44     if( !p_display )
45     {
46         msg_Err( p_demux, "cannot open display" );
47         return VLC_EGENERIC;
48     }
49     p_sys->p_data = (void *)p_display;
50
51     /* Get the parameters of the root window */
52     if( !XGetWindowAttributes( p_display,
53                                DefaultRootWindow( p_display ),
54                                &win_info ) )
55     {
56         msg_Err( p_demux, "can't get root window attributes" );
57         XCloseDisplay( p_display );
58         return VLC_EGENERIC;
59     }
60
61     switch( win_info.depth )
62     {
63     case 8: /* FIXME: set the palette */
64         i_chroma = VLC_FOURCC('R','G','B','2'); break;
65     case 15:
66         i_chroma = VLC_FOURCC('R','V','1','5'); break;
67     case 16:
68         i_chroma = VLC_FOURCC('R','V','1','6'); break;
69     case 24:
70     case 32:
71         i_chroma = VLC_FOURCC('R','V','3','2');
72         win_info.depth = 32;
73         break;
74     default:
75         msg_Err( p_demux, "unknown screen depth %i", win_info.depth );
76         XCloseDisplay( p_display );
77         return VLC_EGENERIC;
78     }
79
80     es_format_Init( &p_sys->fmt, VIDEO_ES, i_chroma );
81     p_sys->fmt.video.i_width  = win_info.width;
82     p_sys->fmt.video.i_height = win_info.height;
83     p_sys->fmt.video.i_bits_per_pixel = win_info.depth;
84
85 #if 0
86     win_info.visual->red_mask;
87     win_info.visual->green_mask;
88     win_info.visual->blue_mask;
89     win_info.visual->bits_per_rgb;
90 #endif
91
92     return VLC_SUCCESS;
93 }
94
95 int screen_CloseCapture( demux_t *p_demux )
96 {
97     demux_sys_t *p_sys = p_demux->p_sys;
98     Display *p_display = (Display *)p_sys->p_data;
99
100     XCloseDisplay( p_display );
101     return VLC_SUCCESS;
102 }
103
104 block_t *screen_Capture( demux_t *p_demux )
105 {
106     demux_sys_t *p_sys = p_demux->p_sys;
107     Display *p_display = (Display *)p_sys->p_data;
108     block_t *p_block;
109     XImage *image;
110     int i_size;
111
112     image = XGetImage( p_display, DefaultRootWindow( p_display ),
113                        0, 0, p_sys->fmt.video.i_width,
114                        p_sys->fmt.video.i_height, AllPlanes, ZPixmap );
115
116     if( !image )
117     {
118         msg_Warn( p_demux, "cannot get image" );
119         return 0;
120     }
121
122     i_size = image->bytes_per_line * image->height;
123
124     if( !( p_block = block_New( p_demux, i_size ) ) )
125     {
126         msg_Warn( p_demux, "cannot get block" );
127         XDestroyImage( image );
128         return 0;
129     }
130
131     memcpy( p_block->p_buffer, image->data, i_size );
132
133     XDestroyImage( image );
134
135     return p_block;
136 }
137