forked from openkylin/imagemagick
337 lines
12 KiB
C
337 lines
12 KiB
C
|
/*
|
|||
|
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
|||
|
% %
|
|||
|
% %
|
|||
|
% %
|
|||
|
% SSSSS CCCC RRRR EEEEE EEEEE N N SSSSS H H OOO TTTTT %
|
|||
|
% SS C R R E E NN N SS H H O O T %
|
|||
|
% SSS C RRRR EEE EEE N N N SSS HHHHH O O T %
|
|||
|
% SS C R R E E N NN SS H H O O T %
|
|||
|
% SSSSS CCCC R R EEEEE EEEEE N N SSSSS H H OOO T %
|
|||
|
% %
|
|||
|
% %
|
|||
|
% Takes a screenshot from the monitor(s). %
|
|||
|
% %
|
|||
|
% Software Design %
|
|||
|
% Dirk Lemstra %
|
|||
|
% April 2014 %
|
|||
|
% %
|
|||
|
% %
|
|||
|
% Copyright 1999-2021 ImageMagick Studio LLC, a non-profit organization %
|
|||
|
% dedicated to making software imaging solutions freely available. %
|
|||
|
% %
|
|||
|
% You may not use this file except in compliance with the License. You may %
|
|||
|
% obtain a copy of the License at %
|
|||
|
% %
|
|||
|
% https://imagemagick.org/script/license.php %
|
|||
|
% %
|
|||
|
% Unless required by applicable law or agreed to in writing, software %
|
|||
|
% distributed under the License is distributed on an "AS IS" BASIS, %
|
|||
|
% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. %
|
|||
|
% See the License for the specific language governing permissions and %
|
|||
|
% limitations under the License. %
|
|||
|
% %
|
|||
|
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
|||
|
%
|
|||
|
%
|
|||
|
*/
|
|||
|
|
|||
|
/*
|
|||
|
Include declarations.
|
|||
|
*/
|
|||
|
#include "magick/studio.h"
|
|||
|
#if defined(MAGICKCORE_WINGDI32_DELEGATE)
|
|||
|
# if defined(__CYGWIN__)
|
|||
|
# include <windows.h>
|
|||
|
# else
|
|||
|
/* All MinGW needs ... */
|
|||
|
# include "magick/nt-base-private.h"
|
|||
|
# include <wingdi.h>
|
|||
|
# ifndef DISPLAY_DEVICE_ACTIVE
|
|||
|
# define DISPLAY_DEVICE_ACTIVE 0x00000001
|
|||
|
# endif
|
|||
|
# endif
|
|||
|
#endif
|
|||
|
#include "magick/blob.h"
|
|||
|
#include "magick/blob-private.h"
|
|||
|
#include "magick/cache.h"
|
|||
|
#include "magick/exception.h"
|
|||
|
#include "magick/exception-private.h"
|
|||
|
#include "magick/image.h"
|
|||
|
#include "magick/image-private.h"
|
|||
|
#include "magick/list.h"
|
|||
|
#include "magick/magick.h"
|
|||
|
#include "magick/memory_.h"
|
|||
|
#include "magick/module.h"
|
|||
|
#include "magick/nt-feature.h"
|
|||
|
#include "magick/option.h"
|
|||
|
#include "magick/pixel-accessor.h"
|
|||
|
#include "magick/quantum-private.h"
|
|||
|
#include "magick/static.h"
|
|||
|
#include "magick/string_.h"
|
|||
|
#include "magick/token.h"
|
|||
|
#include "magick/transform.h"
|
|||
|
#include "magick/utility.h"
|
|||
|
#include "magick/xwindow.h"
|
|||
|
#include "magick/xwindow-private.h"
|
|||
|
|
|||
|
/*
|
|||
|
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
|||
|
% %
|
|||
|
% %
|
|||
|
% %
|
|||
|
% R e a d S C R E E N S H O T I m a g e %
|
|||
|
% %
|
|||
|
% %
|
|||
|
% %
|
|||
|
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
|||
|
%
|
|||
|
% ReadSCREENSHOTImage() Takes a screenshot from the monitor(s).
|
|||
|
%
|
|||
|
% The format of the ReadSCREENSHOTImage method is:
|
|||
|
%
|
|||
|
% Image *ReadXImage(const ImageInfo *image_info,ExceptionInfo *exception)
|
|||
|
%
|
|||
|
% A description of each parameter follows:
|
|||
|
%
|
|||
|
% o image_info: the image info.
|
|||
|
%
|
|||
|
% o exception: return any errors or warnings in this structure.
|
|||
|
%
|
|||
|
*/
|
|||
|
static Image *ReadSCREENSHOTImage(const ImageInfo *image_info,
|
|||
|
ExceptionInfo *exception)
|
|||
|
{
|
|||
|
Image
|
|||
|
*image;
|
|||
|
|
|||
|
assert(image_info->signature == MagickCoreSignature);
|
|||
|
if (image_info->debug != MagickFalse)
|
|||
|
(void) LogMagickEvent(TraceEvent,GetMagickModule(),"%s",
|
|||
|
image_info->filename);
|
|||
|
assert(exception != (ExceptionInfo *) NULL);
|
|||
|
assert(exception->signature == MagickCoreSignature);
|
|||
|
image=(Image *) NULL;
|
|||
|
#if defined(MAGICKCORE_WINGDI32_DELEGATE)
|
|||
|
{
|
|||
|
BITMAPINFO
|
|||
|
bmi;
|
|||
|
|
|||
|
DISPLAY_DEVICE
|
|||
|
device;
|
|||
|
|
|||
|
HBITMAP
|
|||
|
bitmap,
|
|||
|
bitmapOld;
|
|||
|
|
|||
|
HDC
|
|||
|
bitmapDC,
|
|||
|
hDC;
|
|||
|
|
|||
|
Image
|
|||
|
*screen;
|
|||
|
|
|||
|
int
|
|||
|
i;
|
|||
|
|
|||
|
MagickBooleanType
|
|||
|
status;
|
|||
|
|
|||
|
PixelPacket
|
|||
|
*q;
|
|||
|
|
|||
|
ssize_t
|
|||
|
x;
|
|||
|
|
|||
|
RGBQUAD
|
|||
|
*p;
|
|||
|
|
|||
|
ssize_t
|
|||
|
y;
|
|||
|
|
|||
|
assert(image_info != (const ImageInfo *) NULL);
|
|||
|
i=0;
|
|||
|
device.cb = sizeof(device);
|
|||
|
image=(Image *) NULL;
|
|||
|
while(EnumDisplayDevices(NULL,i,&device,0) && ++i)
|
|||
|
{
|
|||
|
if ((device.StateFlags & DISPLAY_DEVICE_ACTIVE) != DISPLAY_DEVICE_ACTIVE)
|
|||
|
continue;
|
|||
|
|
|||
|
hDC=CreateDC(device.DeviceName,device.DeviceName,NULL,NULL);
|
|||
|
if (hDC == (HDC) NULL)
|
|||
|
ThrowReaderException(CoderError,"UnableToCreateDC");
|
|||
|
|
|||
|
screen=AcquireImage(image_info);
|
|||
|
screen->columns=(size_t) GetDeviceCaps(hDC,HORZRES);
|
|||
|
screen->rows=(size_t) GetDeviceCaps(hDC,VERTRES);
|
|||
|
screen->storage_class=DirectClass;
|
|||
|
if (image == (Image *) NULL)
|
|||
|
image=screen;
|
|||
|
else
|
|||
|
AppendImageToList(&image,screen);
|
|||
|
status=SetImageExtent(screen,screen->columns,screen->rows);
|
|||
|
if (status == MagickFalse)
|
|||
|
{
|
|||
|
InheritException(exception,&image->exception);
|
|||
|
return(DestroyImageList(image));
|
|||
|
}
|
|||
|
|
|||
|
bitmapDC=CreateCompatibleDC(hDC);
|
|||
|
if (bitmapDC == (HDC) NULL)
|
|||
|
{
|
|||
|
DeleteDC(hDC);
|
|||
|
ThrowReaderException(CoderError,"UnableToCreateDC");
|
|||
|
}
|
|||
|
(void) memset(&bmi,0,sizeof(BITMAPINFO));
|
|||
|
bmi.bmiHeader.biSize=sizeof(BITMAPINFOHEADER);
|
|||
|
bmi.bmiHeader.biWidth=(LONG) screen->columns;
|
|||
|
bmi.bmiHeader.biHeight=(-1)*(LONG) screen->rows;
|
|||
|
bmi.bmiHeader.biPlanes=1;
|
|||
|
bmi.bmiHeader.biBitCount=32;
|
|||
|
bmi.bmiHeader.biCompression=BI_RGB;
|
|||
|
bitmap=CreateDIBSection(hDC,&bmi,DIB_RGB_COLORS,(void **) &p,NULL,0);
|
|||
|
if (bitmap == (HBITMAP) NULL)
|
|||
|
{
|
|||
|
DeleteDC(hDC);
|
|||
|
DeleteDC(bitmapDC);
|
|||
|
ThrowReaderException(CoderError,"UnableToCreateBitmap");
|
|||
|
}
|
|||
|
bitmapOld=(HBITMAP) SelectObject(bitmapDC,bitmap);
|
|||
|
if (bitmapOld == (HBITMAP) NULL)
|
|||
|
{
|
|||
|
DeleteDC(hDC);
|
|||
|
DeleteDC(bitmapDC);
|
|||
|
DeleteObject(bitmap);
|
|||
|
ThrowReaderException(CoderError,"UnableToCreateBitmap");
|
|||
|
}
|
|||
|
BitBlt(bitmapDC,0,0,(int) screen->columns,(int) screen->rows,hDC,0,0,
|
|||
|
SRCCOPY);
|
|||
|
(void) SelectObject(bitmapDC,bitmapOld);
|
|||
|
|
|||
|
for (y=0; y < (ssize_t) screen->rows; y++)
|
|||
|
{
|
|||
|
q=QueueAuthenticPixels(screen,0,y,screen->columns,1,exception);
|
|||
|
if (q == (PixelPacket *) NULL)
|
|||
|
break;
|
|||
|
for (x=0; x < (ssize_t) screen->columns; x++)
|
|||
|
{
|
|||
|
SetPixelRed(q,ScaleCharToQuantum(p->rgbRed));
|
|||
|
SetPixelGreen(q,ScaleCharToQuantum(p->rgbGreen));
|
|||
|
SetPixelBlue(q,ScaleCharToQuantum(p->rgbBlue));
|
|||
|
SetPixelOpacity(q,OpaqueOpacity);
|
|||
|
p++;
|
|||
|
q++;
|
|||
|
}
|
|||
|
if (SyncAuthenticPixels(screen,exception) == MagickFalse)
|
|||
|
break;
|
|||
|
}
|
|||
|
|
|||
|
DeleteDC(hDC);
|
|||
|
DeleteDC(bitmapDC);
|
|||
|
DeleteObject(bitmap);
|
|||
|
}
|
|||
|
}
|
|||
|
#elif defined(MAGICKCORE_X11_DELEGATE)
|
|||
|
{
|
|||
|
const char
|
|||
|
*option;
|
|||
|
|
|||
|
XImportInfo
|
|||
|
ximage_info;
|
|||
|
|
|||
|
(void) exception;
|
|||
|
XGetImportInfo(&ximage_info);
|
|||
|
option=GetImageOption(image_info,"x:screen");
|
|||
|
if (option != (const char *) NULL)
|
|||
|
ximage_info.screen=IsMagickTrue(option);
|
|||
|
option=GetImageOption(image_info,"x:silent");
|
|||
|
if (option != (const char *) NULL)
|
|||
|
ximage_info.silent=IsMagickTrue(option);
|
|||
|
image=XImportImage(image_info,&ximage_info);
|
|||
|
if ((image != (Image *) NULL) && (image_info->extract != (char *) NULL))
|
|||
|
{
|
|||
|
Image
|
|||
|
*crop_image;
|
|||
|
|
|||
|
RectangleInfo
|
|||
|
crop_info;
|
|||
|
|
|||
|
/*
|
|||
|
Crop image as defined by the extract rectangle.
|
|||
|
*/
|
|||
|
(void) ParsePageGeometry(image,image_info->extract,&crop_info,
|
|||
|
exception);
|
|||
|
crop_image=CropImage(image,&crop_info,exception);
|
|||
|
if (crop_image != (Image *) NULL)
|
|||
|
{
|
|||
|
image=DestroyImage(image);
|
|||
|
image=crop_image;
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
#endif
|
|||
|
return(image);
|
|||
|
}
|
|||
|
|
|||
|
/*
|
|||
|
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
|||
|
% %
|
|||
|
% %
|
|||
|
% %
|
|||
|
% R e g i s t e r S C R E E N S H O T I m a g e %
|
|||
|
% %
|
|||
|
% %
|
|||
|
% %
|
|||
|
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
|||
|
%
|
|||
|
% RegisterSCREENSHOTImage() adds attributes for the screen shot format to
|
|||
|
% the list of supported formats. The attributes include the image format
|
|||
|
% tag, a method to read and/or write the format, whether the format
|
|||
|
% supports the saving of more than one frame to the same file or blob,
|
|||
|
% whether the format supports native in-memory I/O, and a brief
|
|||
|
% description of the format.
|
|||
|
%
|
|||
|
% The format of the RegisterScreenShotImage method is:
|
|||
|
%
|
|||
|
% size_t RegisterScreenShotImage(void)
|
|||
|
%
|
|||
|
*/
|
|||
|
ModuleExport size_t RegisterSCREENSHOTImage(void)
|
|||
|
{
|
|||
|
MagickInfo
|
|||
|
*entry;
|
|||
|
|
|||
|
entry=SetMagickInfo("SCREENSHOT");
|
|||
|
entry->decoder=(DecodeImageHandler *) ReadSCREENSHOTImage;
|
|||
|
entry->format_type=ImplicitFormatType;
|
|||
|
entry->description=ConstantString("Screen shot");
|
|||
|
entry->magick_module=ConstantString("SCREENSHOT");
|
|||
|
(void) RegisterMagickInfo(entry);
|
|||
|
return(MagickImageCoderSignature);
|
|||
|
}
|
|||
|
|
|||
|
/*
|
|||
|
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
|||
|
% %
|
|||
|
% %
|
|||
|
% %
|
|||
|
% 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 %
|
|||
|
% %
|
|||
|
% %
|
|||
|
% %
|
|||
|
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
|||
|
%
|
|||
|
% UnregisterScreenShotImage() removes format registrations made by the
|
|||
|
% screen shot module from the list of supported formats.
|
|||
|
%
|
|||
|
% The format of the UnregisterSCREENSHOTImage method is:
|
|||
|
%
|
|||
|
% UnregisterSCREENSHOTImage(void)
|
|||
|
%
|
|||
|
*/
|
|||
|
ModuleExport void UnregisterSCREENSHOTImage(void)
|
|||
|
{
|
|||
|
(void) UnregisterMagickInfo("SCREENSHOT");
|
|||
|
}
|