root/modules/access/screen/mac.c

Revision 3561b9b28f58eb7a4183e158a8fd973800d31ceb, 10.3 kB (checked in by Rémi Denis-Courmont <rdenis@simphalempin.com>, 6 months ago)

Plugins: include vlc_common.h directly instead of vlc/vlc.h

  • Property mode set to 100644
Line 
1 /*****************************************************************************
2  * mac.c: Screen capture module for the Mac.
3  *****************************************************************************
4  * Copyright (C) 2004, 2008 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Derk-Jan Hartman <hartman at videolan dot org>
8  *          arai <arai_a@mac.com>
9  *
10  * This program is free software; you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License as published by
12  * the Free Software Foundation; either version 2 of the License, or
13  * (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software
22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
23  *****************************************************************************/
24
25 /*****************************************************************************
26  * Preamble
27  *****************************************************************************/
28 #import <stdlib.h>
29
30 #ifdef HAVE_CONFIG_H
31 # import "config.h"
32 #endif
33
34 #import <vlc_common.h>
35
36 #import <ApplicationServices/ApplicationServices.h>
37 #import <OpenGL/OpenGL.h>
38 #import <OpenGL/gl.h>
39
40 typedef int CGSConnectionRef;
41 extern CGError CGSNewConnection( void *, CGSConnectionRef * );
42 extern CGError CGSReleaseConnection( CGSConnectionRef );
43 extern CGError CGSGetGlobalCursorDataSize( CGSConnectionRef, int * );
44 extern CGError CGSGetGlobalCursorData( CGSConnectionRef, unsigned char *,
45                                        int *, int *, CGRect *, CGPoint *,
46                                        int *, int *, int * );
47 extern CGError CGSGetCurrentCursorLocation( CGSConnectionRef, CGPoint * );
48
49 #import "screen.h"
50
51 struct screen_data_t
52 {
53   CGLContextObj screen;
54  
55   CGLContextObj scaled;
56   char *scaled_image;
57  
58   GLuint texture;
59   char *texture_image;
60  
61   GLuint cursor_texture;
62  
63   int left;
64   int top;
65   int src_width;
66   int src_height;
67  
68   int dest_width;
69   int dest_height;
70  
71   int screen_width;
72   int screen_height;
73  
74   CGSConnectionRef connection;
75 };
76
77 int screen_InitCapture( demux_t *p_demux )
78 {
79     demux_sys_t   *p_sys = p_demux->p_sys;
80     screen_data_t *p_data;
81     CGLPixelFormatAttribute attribs[4];
82     CGLPixelFormatObj pix;
83     GLint npix;
84     GLint viewport[4];
85    
86     p_sys->p_data = p_data =
87         ( screen_data_t * )malloc( sizeof( screen_data_t ) );
88    
89     attribs[0] = kCGLPFAFullScreen;
90     attribs[1] = kCGLPFADisplayMask;
91     attribs[2] = CGDisplayIDToOpenGLDisplayMask( CGMainDisplayID() );
92     attribs[3] = 0;
93    
94     CGLChoosePixelFormat( attribs, &pix, &npix );
95     CGLCreateContext( pix, NULL, &( p_data->screen ) );
96     CGLDestroyPixelFormat( pix );
97
98     CGLSetCurrentContext( p_data->screen );
99     CGLSetFullScreen( p_data->screen );
100    
101     glGetIntegerv( GL_VIEWPORT, viewport );
102    
103     p_data->screen_width = viewport[2];
104     p_data->screen_height = viewport[3];
105    
106     p_data->left = 0;
107     p_data->top = 0;
108     p_data->src_width = p_data->screen_width;
109     p_data->src_height = p_data->screen_height;
110     p_data->dest_width = p_data->src_width;
111     p_data->dest_height = p_data->src_height;
112    
113     attribs [0] = kCGLPFAOffScreen;
114     attribs [1] = kCGLPFAColorSize;
115     attribs [2] = 32;
116     attribs [3] = 0;
117    
118     CGLChoosePixelFormat( attribs, &pix, &npix );
119     CGLCreateContext( pix, NULL, &( p_data->scaled ) );
120     CGLDestroyPixelFormat( pix );
121
122     CGLSetCurrentContext( p_data->scaled );
123     p_data->scaled_image = ( char * )malloc( p_data->dest_width
124                                           * p_data->dest_height * 4 );
125     CGLSetOffScreen( p_data->scaled, p_data->dest_width, p_data->dest_height,
126                      p_data->dest_width * 4, p_data->scaled_image );
127    
128     es_format_Init( &p_sys->fmt, VIDEO_ES, VLC_FOURCC( 'R','V','3','2' ) );
129    
130     p_sys->fmt.video.i_width = p_data->dest_width;
131     p_sys->fmt.video.i_visible_width = p_data->dest_width;
132     p_sys->fmt.video.i_height = p_data->dest_height;
133     p_sys->fmt.video.i_bits_per_pixel = 32;
134    
135     glGenTextures( 1, &( p_data->texture ) );
136     glBindTexture( GL_TEXTURE_2D, p_data->texture );
137    
138     p_data->texture_image
139       = ( char * )malloc( p_data->src_width * p_data->src_height * 4 );
140    
141     glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST );
142     glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST );
143     glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP );
144     glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP );
145
146     glGenTextures( 1, &( p_data->cursor_texture ) );
147     glBindTexture( GL_TEXTURE_2D, p_data->cursor_texture );
148    
149     glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST );
150     glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST );
151     glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP );
152     glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP );
153    
154     CGSNewConnection( NULL, &( p_data->connection ) );
155    
156     return VLC_SUCCESS;
157 }
158
159 int screen_CloseCapture( demux_t *p_demux )
160 {
161     screen_data_t *p_data = ( screen_data_t * )p_demux->p_sys->p_data;
162    
163     CGSReleaseConnection( p_data->connection );
164    
165     CGLSetCurrentContext( NULL );
166     CGLClearDrawable( p_data->screen );
167     CGLDestroyContext( p_data->screen );
168    
169     return VLC_SUCCESS;
170 }
171
172 block_t *screen_Capture( demux_t *p_demux )
173 {
174     demux_sys_t *p_sys = p_demux->p_sys;
175     screen_data_t *p_data = ( screen_data_t * )p_sys->p_data;
176     block_t *p_block;
177     int i_size;
178    
179     i_size = p_sys->fmt.video.i_height * p_sys->fmt.video.i_width * 4;
180    
181     if( !( p_block = block_New( p_demux, i_size ) ) )
182     {
183         msg_Warn( p_demux, "cannot get block" );
184         return 0;
185     }
186    
187     CGLSetCurrentContext( p_data->screen );
188     glReadPixels( p_data->left,
189                   p_data->screen_height - p_data->top - p_data->src_height,
190                   p_data->src_width,
191                   p_data->src_height,
192                   GL_RGBA, GL_UNSIGNED_BYTE,
193                   p_data->texture_image );
194    
195     CGLSetCurrentContext( p_data->scaled );
196     glEnable( GL_TEXTURE_2D );
197     glBindTexture( GL_TEXTURE_2D, p_data->texture );
198     glTexImage2D( GL_TEXTURE_2D, 0,
199                   GL_RGBA8, p_data->src_width, p_data->src_height, 0,
200                   GL_RGBA, GL_UNSIGNED_BYTE, p_data->texture_image );
201    
202     glClearColor( 0.0f, 0.0f, 0.0f, 1.0f );
203     glClear( GL_COLOR_BUFFER_BIT );
204     glColor3f( 1.0f, 1.0f, 1.0f );
205     glEnable( GL_TEXTURE_2D );
206     glBindTexture( GL_TEXTURE_2D, p_data->texture );
207     glBegin( GL_POLYGON );
208     glTexCoord2f( 0.0, 1.0 ); glVertex2f( -1.0, -1.0 );
209     glTexCoord2f( 1.0, 1.0 ); glVertex2f( 1.0, -1.0 );
210     glTexCoord2f( 1.0, 0.0 ); glVertex2f( 1.0, 1.0 );
211     glTexCoord2f( 0.0, 0.0 ); glVertex2f( -1.0, 1.0 );
212     glEnd();
213     glDisable( GL_TEXTURE_2D );
214    
215     CGPoint cursor_pos;
216     int size;
217     int tmp1, tmp2, tmp3, tmp4;
218     unsigned char *cursor_image;
219     CGRect cursor_rect;
220     CGPoint cursor_hot;
221    
222     cursor_pos.x = 0;
223     cursor_pos.y = 0;
224    
225     if( CGSGetCurrentCursorLocation( p_data->connection, &cursor_pos )
226         == kCGErrorSuccess
227         && CGSGetGlobalCursorDataSize( p_data->connection, &size )
228         == kCGErrorSuccess )
229     {
230         cursor_image = ( unsigned char* )malloc( size );
231         if( CGSGetGlobalCursorData( p_data->connection,
232                                     cursor_image, &size,
233                                     &tmp1,
234                                     &cursor_rect, &cursor_hot,
235                                     &tmp2, &tmp3, &tmp4 )
236             == kCGErrorSuccess )
237         {
238             glEnable( GL_TEXTURE_2D );
239             glBindTexture( GL_TEXTURE_2D, p_data->cursor_texture );
240             glTexImage2D( GL_TEXTURE_2D, 0,
241                           GL_RGBA8,
242                           ( int )( cursor_rect.size.width ),
243                           ( int )( cursor_rect.size.height ), 0,
244                           GL_RGBA, GL_UNSIGNED_BYTE,
245                           ( char * )cursor_image );
246            
247             cursor_rect.origin.x = cursor_pos.x - p_data->left - cursor_hot.x;
248             cursor_rect.origin.y = cursor_pos.y - p_data->top - cursor_hot.y;
249            
250             cursor_rect.origin.x
251               = 2.0 * cursor_rect.origin.x / p_data->src_width - 1.0;
252             cursor_rect.origin.y
253               = 2.0 * cursor_rect.origin.y / p_data->src_height - 1.0;
254             cursor_rect.size.width
255               = 2.0 * cursor_rect.size.width / p_data->src_width;
256             cursor_rect.size.height
257               = 2.0 * cursor_rect.size.height / p_data->src_height;
258            
259             glColor3f( 1.0f, 1.0f, 1.0f );
260             glEnable( GL_TEXTURE_2D );
261             glEnable( GL_BLEND );
262             glBlendFunc( GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA );
263             glBindTexture( GL_TEXTURE_2D, p_data->cursor_texture );
264             glBegin( GL_POLYGON );
265             glTexCoord2f( 0.0, 0.0 ); glVertex2f( cursor_rect.origin.x,
266                                                   cursor_rect.origin.y );
267             glTexCoord2f( 1.0, 0.0 ); glVertex2f( cursor_rect.origin.x
268                                                   + cursor_rect.size.width,
269                                                   cursor_rect.origin.y );
270             glTexCoord2f( 1.0, 1.0 ); glVertex2f( cursor_rect.origin.x
271                                                   + cursor_rect.size.width,
272                                                   cursor_rect.origin.y
273                                                   + cursor_rect.size.height );
274             glTexCoord2f( 0.0, 1.0 ); glVertex2f( cursor_rect.origin.x,
275                                                   cursor_rect.origin.y
276                                                   + cursor_rect.size.height );
277             glEnd();
278             glDisable( GL_BLEND );
279             glDisable( GL_TEXTURE_2D );
280         }
281         free( cursor_image );
282     }
283    
284     glReadPixels( 0, 0,
285                   p_data->dest_width,
286                   p_data->dest_height,
287                   GL_BGRA, GL_UNSIGNED_INT_8_8_8_8_REV,
288                   p_block->p_buffer );
289    
290     return p_block;
291 }
Note: See TracBrowser for help on using the browser.