1 /*
2 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
3 % %
4 % %
5 % %
6 % SSSSS CCCC RRRR EEEEE EEEEE N N SSSSS H H OOO TTTTT %
7 % SS C R R E E NN N SS H H O O T %
8 % SSS C RRRR EEE EEE N N N SSS HHHHH O O T %
9 % SS C R R E E N NN SS H H O O T %
10 % SSSSS CCCC R R EEEEE EEEEE N N SSSSS H H OOO T %
11 % %
12 % %
13 % Takes a screenshot from the monitor(s). %
14 % %
15 % Software Design %
16 % Dirk Lemstra %
17 % April 2014 %
18 % %
19 % %
20 % Copyright 1999-2016 ImageMagick Studio LLC, a non-profit organization %
21 % dedicated to making software imaging solutions freely available. %
22 % %
23 % You may not use this file except in compliance with the License. You may %
24 % obtain a copy of the License at %
25 % %
26 % http://www.imagemagick.org/script/license.php %
27 % %
28 % Unless required by applicable law or agreed to in writing, software %
29 % distributed under the License is distributed on an "AS IS" BASIS, %
30 % WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. %
31 % See the License for the specific language governing permissions and %
32 % limitations under the License. %
33 % %
34 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
35 %
36 %
37 */
38
39 /*
40 Include declarations.
41 */
42 #include "MagickCore/studio.h"
43 #if defined(MAGICKCORE_WINGDI32_DELEGATE)
44 # if defined(__CYGWIN__)
45 # include <windows.h>
46 # else
47 /* All MinGW needs ... */
48 # include "MagickCore/nt-base-private.h"
49 # include <wingdi.h>
50 # ifndef DISPLAY_DEVICE_ACTIVE
51 # define DISPLAY_DEVICE_ACTIVE 0x00000001
52 # endif
53 # endif
54 #endif
55 #include "MagickCore/blob.h"
56 #include "MagickCore/blob-private.h"
57 #include "MagickCore/cache.h"
58 #include "MagickCore/exception.h"
59 #include "MagickCore/exception-private.h"
60 #include "MagickCore/image.h"
61 #include "MagickCore/image-private.h"
62 #include "MagickCore/list.h"
63 #include "MagickCore/magick.h"
64 #include "MagickCore/memory_.h"
65 #include "MagickCore/module.h"
66 #include "MagickCore/nt-feature.h"
67 #include "MagickCore/option.h"
68 #include "MagickCore/pixel-accessor.h"
69 #include "MagickCore/quantum-private.h"
70 #include "MagickCore/static.h"
71 #include "MagickCore/string_.h"
72 #include "MagickCore/token.h"
73 #include "MagickCore/utility.h"
74 #include "MagickCore/xwindow.h"
75 #include "MagickCore/xwindow-private.h"
76
77 /*
78 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
79 % %
80 % %
81 % %
82 % R e a d S C R E E N S H O T I m a g e %
83 % %
84 % %
85 % %
86 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
87 %
88 % ReadSCREENSHOTImage() Takes a screenshot from the monitor(s).
89 %
90 % The format of the ReadSCREENSHOTImage method is:
91 %
92 % Image *ReadXImage(const ImageInfo *image_info,ExceptionInfo *exception)
93 %
94 % A description of each parameter follows:
95 %
96 % o image_info: the image info.
97 %
98 % o exception: return any errors or warnings in this structure.
99 %
100 */
ReadSCREENSHOTImage(const ImageInfo * image_info,ExceptionInfo * exception)101 static Image *ReadSCREENSHOTImage(const ImageInfo *image_info,
102 ExceptionInfo *exception)
103 {
104 Image
105 *image;
106
107 assert(image_info->signature == MagickCoreSignature);
108 if (image_info->debug != MagickFalse)
109 (void) LogMagickEvent(TraceEvent,GetMagickModule(),"%s",
110 image_info->filename);
111 assert(exception != (ExceptionInfo *) NULL);
112 assert(exception->signature == MagickCoreSignature);
113 image=(Image *) NULL;
114 #if defined(MAGICKCORE_WINGDI32_DELEGATE)
115 {
116 BITMAPINFO
117 bmi;
118
119 DISPLAY_DEVICE
120 device;
121
122 HBITMAP
123 bitmap,
124 bitmapOld;
125
126 HDC
127 bitmapDC,
128 hDC;
129
130 Image
131 *screen;
132
133 int
134 i;
135
136 MagickBooleanType
137 status;
138
139 register Quantum
140 *q;
141
142 register ssize_t
143 x;
144
145 RGBTRIPLE
146 *p;
147
148 ssize_t
149 y;
150
151 assert(image_info != (const ImageInfo *) NULL);
152 i=0;
153 device.cb = sizeof(device);
154 image=(Image *) NULL;
155 while(EnumDisplayDevices(NULL,i,&device,0) && ++i)
156 {
157 if ((device.StateFlags & DISPLAY_DEVICE_ACTIVE) != DISPLAY_DEVICE_ACTIVE)
158 continue;
159
160 hDC=CreateDC(device.DeviceName,device.DeviceName,NULL,NULL);
161 if (hDC == (HDC) NULL)
162 ThrowReaderException(CoderError,"UnableToCreateDC");
163
164 screen=AcquireImage(image_info,exception);
165 screen->columns=(size_t) GetDeviceCaps(hDC,HORZRES);
166 screen->rows=(size_t) GetDeviceCaps(hDC,VERTRES);
167 screen->storage_class=DirectClass;
168 status=SetImageExtent(screen,screen->columns,screen->rows,exception);
169 if (status == MagickFalse)
170 return(DestroyImageList(image));
171 if (image == (Image *) NULL)
172 image=screen;
173 else
174 AppendImageToList(&image,screen);
175
176 bitmapDC=CreateCompatibleDC(hDC);
177 if (bitmapDC == (HDC) NULL)
178 {
179 DeleteDC(hDC);
180 ThrowReaderException(CoderError,"UnableToCreateDC");
181 }
182 (void) ResetMagickMemory(&bmi,0,sizeof(BITMAPINFO));
183 bmi.bmiHeader.biSize=sizeof(BITMAPINFOHEADER);
184 bmi.bmiHeader.biWidth=(LONG) screen->columns;
185 bmi.bmiHeader.biHeight=(-1)*(LONG) screen->rows;
186 bmi.bmiHeader.biPlanes=1;
187 bmi.bmiHeader.biBitCount=24;
188 bmi.bmiHeader.biCompression=BI_RGB;
189 bitmap=CreateDIBSection(hDC,&bmi,DIB_RGB_COLORS,(void **) &p,NULL,0);
190 if (bitmap == (HBITMAP) NULL)
191 {
192 DeleteDC(hDC);
193 DeleteDC(bitmapDC);
194 ThrowReaderException(CoderError,"UnableToCreateBitmap");
195 }
196 bitmapOld=(HBITMAP) SelectObject(bitmapDC,bitmap);
197 if (bitmapOld == (HBITMAP) NULL)
198 {
199 DeleteDC(hDC);
200 DeleteDC(bitmapDC);
201 DeleteObject(bitmap);
202 ThrowReaderException(CoderError,"UnableToCreateBitmap");
203 }
204 BitBlt(bitmapDC,0,0,(int) screen->columns,(int) screen->rows,hDC,0,0,
205 SRCCOPY);
206 (void) SelectObject(bitmapDC,bitmapOld);
207
208 for (y=0; y < (ssize_t) screen->rows; y++)
209 {
210 q=QueueAuthenticPixels(screen,0,y,screen->columns,1,exception);
211 if (q == (Quantum *) NULL)
212 break;
213 for (x=0; x < (ssize_t) screen->columns; x++)
214 {
215 SetPixelRed(image,ScaleCharToQuantum(p->rgbtRed),q);
216 SetPixelGreen(image,ScaleCharToQuantum(p->rgbtGreen),q);
217 SetPixelBlue(image,ScaleCharToQuantum(p->rgbtBlue),q);
218 SetPixelAlpha(image,OpaqueAlpha,q);
219 p++;
220 q+=GetPixelChannels(image);
221 }
222 if (SyncAuthenticPixels(screen,exception) == MagickFalse)
223 break;
224 }
225
226 DeleteDC(hDC);
227 DeleteDC(bitmapDC);
228 DeleteObject(bitmap);
229 }
230 }
231 #elif defined(MAGICKCORE_X11_DELEGATE)
232 {
233 const char
234 *option;
235
236 XImportInfo
237 ximage_info;
238
239 XGetImportInfo(&ximage_info);
240 option=GetImageOption(image_info,"x:screen");
241 if (option != (const char *) NULL)
242 ximage_info.screen=IsStringTrue(option);
243 option=GetImageOption(image_info,"x:silent");
244 if (option != (const char *) NULL)
245 ximage_info.silent=IsStringTrue(option);
246 image=XImportImage(image_info,&ximage_info,exception);
247 }
248 #endif
249 return(image);
250 }
251
252 /*
253 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
254 % %
255 % %
256 % %
257 % R e g i s t e r S C R E E N S H O T I m a g e %
258 % %
259 % %
260 % %
261 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
262 %
263 % RegisterSCREENSHOTImage() adds attributes for the screen shot format to
264 % the list of supported formats. The attributes include the image format
265 % tag, a method to read and/or write the format, whether the format
266 % supports the saving of more than one frame to the same file or blob,
267 % whether the format supports native in-memory I/O, and a brief
268 % description of the format.
269 %
270 % The format of the RegisterScreenShotImage method is:
271 %
272 % size_t RegisterScreenShotImage(void)
273 %
274 */
RegisterSCREENSHOTImage(void)275 ModuleExport size_t RegisterSCREENSHOTImage(void)
276 {
277 MagickInfo
278 *entry;
279
280 entry=AcquireMagickInfo("SCREENSHOT","SCREENSHOT","Screen shot");
281 entry->decoder=(DecodeImageHandler *) ReadSCREENSHOTImage;
282 entry->format_type=ImplicitFormatType;
283 (void) RegisterMagickInfo(entry);
284 return(MagickImageCoderSignature);
285 }
286
287 /*
288 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
289 % %
290 % %
291 % %
292 % U n r e g i s t e r S C R E E N S H O T I m a g e %
293 % %
294 % %
295 % %
296 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
297 %
298 % UnregisterScreenShotImage() removes format registrations made by the
299 % screen shot module from the list of supported formats.
300 %
301 % The format of the UnregisterSCREENSHOTImage method is:
302 %
303 % UnregisterSCREENSHOTImage(void)
304 %
305 */
UnregisterSCREENSHOTImage(void)306 ModuleExport void UnregisterSCREENSHOTImage(void)
307 {
308 (void) UnregisterMagickInfo("SCREENSHOT");
309 }
310