]>
Commit | Line | Data |
---|---|---|
1 | /* | |
2 | IMPORTANT: This Apple software is supplied to you by Apple Computer, Inc. ("Apple") in | |
3 | consideration of your agreement to the following terms, and your use, installation or modification | |
4 | of this Apple software constitutes acceptance of these terms. If you do not agree with these terms, | |
5 | please do not use, install or modify this Apple software. | |
6 | ||
7 | In consideration of your agreement to abide by the following terms, and subject to these | |
8 | terms, Apple grants you a personal, non-exclusive license, under Apple\xd5s copyrights in | |
9 | this original Apple software (the "Apple Software"), to use and modify the Apple Software, | |
10 | with or without modifications, in source and/or binary forms; provided that if you redistribute | |
11 | the Apple Software in its entirety and without modifications, you must retain this notice and | |
12 | the following text and disclaimers in all such redistributions of the Apple Software. | |
13 | Neither the name, trademarks, service marks or logos of Apple Computer, Inc. may be used | |
14 | to endorse or promote products derived from the Apple Software without specific prior | |
15 | written permission from Apple. Except as expressly tated in this notice, no other rights or | |
16 | licenses, express or implied, are granted by Apple herein, including but not limited | |
17 | to any patent rights that may be infringed by your derivative works or by other works | |
18 | in which the Apple Software may be incorporated. | |
19 | ||
20 | The Apple Software is provided by Apple on an "AS IS" basis. APPLE MAKES NO WARRANTIES, | |
21 | EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION THE IMPLIED WARRANTIES OF NON-INFRINGEMENT, | |
22 | MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE, REGARDING THE APPLE SOFTWARE OR ITS | |
23 | USE AND OPERATION ALONE OR IN COMBINATION WITH YOUR PRODUCTS. | |
24 | ||
25 | IN NO EVENT SHALL APPLE BE LIABLE FOR ANY SPECIAL, INDIRECT, INCIDENTAL OR CONSEQUENTIAL | |
26 | DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS | |
27 | OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) ARISING IN ANY WAY OUT OF THE USE, | |
28 | REPRODUCTION, MODIFICATION AND/OR DISTRIBUTION OF THE APPLE SOFTWARE, HOWEVER CAUSED AND | |
29 | WHETHER UNDER THEORY OF CONTRACT, TORT (INCLUDING NEGLIGENCE), STRICT LIABILITY OR | |
30 | OTHERWISE, EVEN IF APPLE HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
31 | */ | |
32 | ||
33 | #include "wx/wxprec.h" | |
34 | ||
35 | #if wxOSX_USE_COCOA_OR_CARBON | |
36 | ||
37 | #import <CoreFoundation/CoreFoundation.h> | |
38 | #import <ApplicationServices/ApplicationServices.h> | |
39 | #import <OpenGL/OpenGL.h> | |
40 | #import <OpenGL/gl.h> | |
41 | ||
42 | #include "wx/osx/core/private.h" | |
43 | #include "wx/osx/private/glgrab.h" | |
44 | ||
45 | extern CGColorSpaceRef wxMacGetGenericRGBColorSpace(); | |
46 | ||
47 | /* | |
48 | * perform an in-place swap from Quadrant 1 to Quadrant III format | |
49 | * (upside-down PostScript/GL to right side up QD/CG raster format) | |
50 | * We do this in-place, which requires more copying, but will touch | |
51 | * only half the pages. (Display grabs are BIG!) | |
52 | * | |
53 | * Pixel reformatting may optionally be done here if needed. | |
54 | */ | |
55 | static void swizzleBitmap(void * data, int rowBytes, int height) | |
56 | { | |
57 | int top, bottom; | |
58 | void * buffer; | |
59 | void * topP; | |
60 | void * bottomP; | |
61 | void * base; | |
62 | ||
63 | ||
64 | top = 0; | |
65 | bottom = height - 1; | |
66 | base = data; | |
67 | buffer = malloc(rowBytes); | |
68 | ||
69 | ||
70 | while ( top < bottom ) | |
71 | { | |
72 | topP = (void *)((top * rowBytes) + (intptr_t)base); | |
73 | bottomP = (void *)((bottom * rowBytes) + (intptr_t)base); | |
74 | ||
75 | ||
76 | /* | |
77 | * Save and swap scanlines. | |
78 | * | |
79 | * This code does a simple in-place exchange with a temp buffer. | |
80 | * If you need to reformat the pixels, replace the first two bcopy() | |
81 | * calls with your own custom pixel reformatter. | |
82 | */ | |
83 | bcopy( topP, buffer, rowBytes ); | |
84 | bcopy( bottomP, topP, rowBytes ); | |
85 | bcopy( buffer, bottomP, rowBytes ); | |
86 | ||
87 | ++top; | |
88 | --bottom; | |
89 | } | |
90 | free( buffer ); | |
91 | } | |
92 | ||
93 | ||
94 | /* | |
95 | * Given a display ID and a rectangle on that display, generate a CGImageRef | |
96 | * containing the display contents. | |
97 | * | |
98 | * srcRect is display-origin relative. | |
99 | * | |
100 | * This function uses a full screen OpenGL read-only context. | |
101 | * By using OpenGL, we can read the screen using a DMA transfer | |
102 | * when it's in millions of colors mode, and we can correctly read | |
103 | * a microtiled full screen OpenGL context, such as a game or full | |
104 | * screen video display. | |
105 | * | |
106 | * Returns a CGImageRef. When you are done with the CGImageRef, release it | |
107 | * using CFRelease(). | |
108 | * Returns NULL on an error. | |
109 | */ | |
110 | ||
111 | CGImageRef grabViaOpenGL(CGDirectDisplayID display, CGRect srcRect) | |
112 | { | |
113 | CGContextRef bitmap; | |
114 | CGImageRef image; | |
115 | void * data; | |
116 | long bytewidth; | |
117 | GLint width, height; | |
118 | ||
119 | CGLContextObj glContextObj; | |
120 | CGLPixelFormatObj pixelFormatObj ; | |
121 | GLint numPixelFormats ; | |
122 | CGLPixelFormatAttribute attribs[] = | |
123 | { | |
124 | kCGLPFAFullScreen, | |
125 | kCGLPFADisplayMask, | |
126 | (CGLPixelFormatAttribute) 0, /* Display mask bit goes here */ | |
127 | (CGLPixelFormatAttribute) 0 | |
128 | } ; | |
129 | ||
130 | ||
131 | if ( display == kCGNullDirectDisplay ) | |
132 | display = CGMainDisplayID(); | |
133 | attribs[2] = (CGLPixelFormatAttribute) CGDisplayIDToOpenGLDisplayMask(display); | |
134 | ||
135 | ||
136 | /* Build a full-screen GL context */ | |
137 | CGLChoosePixelFormat( attribs, &pixelFormatObj, &numPixelFormats ); | |
138 | if ( pixelFormatObj == NULL ) // No full screen context support | |
139 | return NULL; | |
140 | CGLCreateContext( pixelFormatObj, NULL, &glContextObj ) ; | |
141 | CGLDestroyPixelFormat( pixelFormatObj ) ; | |
142 | if ( glContextObj == NULL ) | |
143 | return NULL; | |
144 | ||
145 | ||
146 | CGLSetCurrentContext( glContextObj ) ; | |
147 | CGLSetFullScreen( glContextObj ) ; | |
148 | ||
149 | ||
150 | glReadBuffer(GL_FRONT); | |
151 | ||
152 | ||
153 | width = (GLint)srcRect.size.width; | |
154 | height = (GLint)srcRect.size.height; | |
155 | ||
156 | ||
157 | bytewidth = width * 4; // Assume 4 bytes/pixel for now | |
158 | bytewidth = (bytewidth + 3) & ~3; // Align to 4 bytes | |
159 | ||
160 | /* Build bitmap context */ | |
161 | data = malloc(height * bytewidth); | |
162 | if ( data == NULL ) | |
163 | { | |
164 | CGLSetCurrentContext( NULL ); | |
165 | CGLClearDrawable( glContextObj ); // disassociate from full screen | |
166 | CGLDestroyContext( glContextObj ); // and destroy the context | |
167 | return NULL; | |
168 | } | |
169 | bitmap = CGBitmapContextCreate(data, width, height, 8, bytewidth, | |
170 | wxMacGetGenericRGBColorSpace(), kCGImageAlphaNoneSkipFirst /* XRGB */); | |
171 | ||
172 | ||
173 | /* Read framebuffer into our bitmap */ | |
174 | glFinish(); /* Finish all OpenGL commands */ | |
175 | glPixelStorei(GL_PACK_ALIGNMENT, 4); /* Force 4-byte alignment */ | |
176 | glPixelStorei(GL_PACK_ROW_LENGTH, 0); | |
177 | glPixelStorei(GL_PACK_SKIP_ROWS, 0); | |
178 | glPixelStorei(GL_PACK_SKIP_PIXELS, 0); | |
179 | ||
180 | /* | |
181 | * Fetch the data in XRGB format, matching the bitmap context. | |
182 | */ | |
183 | glReadPixels((GLint)srcRect.origin.x, (GLint)srcRect.origin.y, width, height, | |
184 | GL_BGRA, | |
185 | #ifdef __BIG_ENDIAN__ | |
186 | GL_UNSIGNED_INT_8_8_8_8_REV, // for PPC | |
187 | #else | |
188 | GL_UNSIGNED_INT_8_8_8_8, // for Intel! http://lists.apple.com/archives/quartz-dev/2006/May/msg00100.html | |
189 | #endif | |
190 | data); | |
191 | /* | |
192 | * glReadPixels generates a quadrant I raster, with origin in the lower left | |
193 | * This isn't a problem for signal processing routines such as compressors, | |
194 | * as they can simply use a negative 'advance' to move between scanlines. | |
195 | * CGImageRef and CGBitmapContext assume a quadrant III raster, though, so we need to | |
196 | * invert it. Pixel reformatting can also be done here. | |
197 | */ | |
198 | swizzleBitmap(data, bytewidth, height); | |
199 | ||
200 | ||
201 | /* Make an image out of our bitmap; does a cheap vm_copy of the bitmap */ | |
202 | image = CGBitmapContextCreateImage(bitmap); | |
203 | ||
204 | /* Get rid of bitmap */ | |
205 | CFRelease(bitmap); | |
206 | free(data); | |
207 | ||
208 | ||
209 | /* Get rid of GL context */ | |
210 | CGLSetCurrentContext( NULL ); | |
211 | CGLClearDrawable( glContextObj ); // disassociate from full screen | |
212 | CGLDestroyContext( glContextObj ); // and destroy the context | |
213 | ||
214 | /* Returned image has a reference count of 1 */ | |
215 | return image; | |
216 | } | |
217 | ||
218 | #endif |