Skip to content

Instantly share code, notes, and snippets.

@prashanthrajagopal
Last active December 31, 2023 17:26
Show Gist options
  • Star 10 You must be signed in to star a gist
  • Fork 7 You must be signed in to fork a gist
  • Save prashanthrajagopal/05f8ad157ece964d8c4d to your computer and use it in GitHub Desktop.
Save prashanthrajagopal/05f8ad157ece964d8c4d to your computer and use it in GitHub Desktop.
Take a screenshot and save as jpeg in c++
#include <stdio.h>
#include <windows.h>
#include <gdiplus.h>
#include <time.h>
int GetEncoderClsid(const WCHAR* format, CLSID* pClsid) {
using namespace Gdiplus;
UINT num = 0;
UINT size = 0;
ImageCodecInfo* pImageCodecInfo = NULL;
GetImageEncodersSize(&num, &size);
if(size == 0)
return -1;
pImageCodecInfo = (ImageCodecInfo*)(malloc(size));
if(pImageCodecInfo == NULL)
return -1;
GetImageEncoders(num, size, pImageCodecInfo);
for(UINT j = 0; j < num; ++j)
{
if( wcscmp(pImageCodecInfo[j].MimeType, format) == 0 )
{
*pClsid = pImageCodecInfo[j].Clsid;
free(pImageCodecInfo);
return j;
}
}
free(pImageCodecInfo);
return 0;
}
void gdiscreen() {
using namespace Gdiplus;
IStream* istream;
HRESULT res = CreateStreamOnHGlobal(NULL, true, &istream);
GdiplusStartupInput gdiplusStartupInput;
ULONG_PTR gdiplusToken;
GdiplusStartup(&gdiplusToken, &gdiplusStartupInput, NULL);
{
HDC scrdc, memdc;
HBITMAP membit;
scrdc = ::GetDC(0);
int Height = GetSystemMetrics(SM_CYSCREEN);
int Width = GetSystemMetrics(SM_CXSCREEN);
memdc = CreateCompatibleDC(scrdc);
membit = CreateCompatibleBitmap(scrdc, Width, Height);
HBITMAP hOldBitmap =(HBITMAP) SelectObject(memdc, membit);
BitBlt(memdc, 0, 0, Width, Height, scrdc, 0, 0, SRCCOPY);
Gdiplus::Bitmap bitmap(membit, NULL);
CLSID clsid;
GetEncoderClsid(L"image/jpeg", &clsid);
// bitmap.Save(L"screen.jpeg", &clsid, NULL); // To save the jpeg to a file
bitmap.Save(istream, &clsid, NULL);
// Create a bitmap from the stream and save it to make sure the stream has the image
// Gdiplus::Bitmap bmp(istream, NULL);
// bmp.Save(L"t1est.jpeg", &clsid, NULL);
// END
delete &clsid;
DeleteObject(memdc);
DeleteObject(membit);
::ReleaseDC(0,scrdc);
}
GdiplusShutdown(gdiplusToken);
}
int main()
{
clock_t t1 = clock();
int i;
int iterations = 10;
for(i=0;i<iterations;i++){
gdiscreen();
}
clock_t t2 = clock();
printf("%d iterations: %0.0f fps\n", iterations, iterations/((double)(t2 - t1) / CLOCKS_PER_SEC));
return 0;
}
@dbdeadka
Copy link

dbdeadka commented Nov 8, 2018

Thank you so much for a good script.
But I've to notice that it's necessary to add
istream->Release();
string
after
Gdiplus::GdiplusShutdown(gdiplusToken);
(to avoid mem leaks)

@haephrati
Copy link

Remove
delete &clsid;

@ShailMurtaza
Copy link

ShailMurtaza commented Apr 13, 2021

Can you please help me ?
I'm getting ERROR
undefined reference to __imp_CreateStreamOnHGlobal
This is the only ERROR I'm getting right
@prashanthrajagopal @haephrati @dbdeadka

@Mr-0-neti
Copy link

Can you please help me ?
I'm getting ERROR
undefined reference to __imp_CreateStreamOnHGlobal
This is the only ERROR I'm getting right
@prashanthrajagopal @haephrati @dbdeadka

yes i am gettin the same error C:\Users\USER\AppData\Local\Temp\ccTLI53G.o:screenshot.cpp:(.text+0x158): undefined reference to CreateStreamOnHGlobal@12'` ;(

@dbdeadka
Copy link

Mr-0-neti,
Add -lgdiplus to your linker options.

@ShailMurtaza
Copy link

ShailMurtaza commented Apr 20, 2021

Mr-0-neti,
Add -lgdiplus to your linker options.

@dbdeadka
I'm using -lgdiplus but not working.
Without linking GDI library you will get plenty of errors not one.

@UnidentifiedCitizen
Copy link

So, it's supports only 1 display screenshot?

@ShailMurtaza
Copy link

ShailMurtaza commented Aug 28, 2021

So, it's supports only 1 display screenshot?

@conspiracynomad
Who said that ??

@jahid1o
Copy link

jahid1o commented Apr 6, 2022

Can you please help me ?
I'm getting ERROR
undefined reference to __imp_CreateStreamOnHGlobal
This is the only ERROR I'm getting right
@prashanthrajagopal @haephrati @dbdeadka

yes i am gettin the same error C:\Users\USER\AppData\Local\Temp\ccTLI53G.o:screenshot.cpp:(.text+0x158): undefined reference to CreateStreamOnHGlobal@12'` ;(

Add -mwindows -lgdiplus -lole32 to linker options.

@tajiknomi
Copy link

tajiknomi commented May 9, 2022

One should import gdiplus.lib in their project before compiling. If you are on windows/Visual-studio, you should include this in your source file
#pragma comment (lib, "gdiplus.lib")
OR
You could add the above lib in the VS Project like below
Project Properties > Linker > input > Additional Dependencies

Plus as stated by haephrati; remove delete &clsid;

In order to write the jpeg file; you had to uncomment the below line to save the jpeg image in the appropriate location i.e.

bitmap.Save(path/to/jpeg, &clsid, NULL);

@zoldaten
Copy link

i spent time to find out how to start it working. so i did. on windows 7. to right code is:

#include <stdio.h>
#include <windows.h>
#include <gdiplus.h>
#include <time.h>
#include <cstdio>
#pragma comment (lib, "gdiplus.lib")

int GetEncoderClsid(const WCHAR* format, CLSID* pClsid) {
	using namespace Gdiplus;
	UINT  num = 0;
	UINT  size = 0;

	ImageCodecInfo* pImageCodecInfo = NULL;

	GetImageEncodersSize(&num, &size);
	if(size == 0)
		return -1;

	pImageCodecInfo = (ImageCodecInfo*)(malloc(size));
	if(pImageCodecInfo == NULL)
		return -1;

	GetImageEncoders(num, size, pImageCodecInfo);
	for(UINT j = 0; j < num; ++j)
	{
		if( wcscmp(pImageCodecInfo[j].MimeType, format) == 0 )
		{
			*pClsid = pImageCodecInfo[j].Clsid;
			free(pImageCodecInfo);
			return j;
		}    
	}
	free(pImageCodecInfo);
	return 0;
}

void gdiscreen() {
	using namespace Gdiplus;
	IStream* istream;
	HRESULT res = CreateStreamOnHGlobal(NULL, true, &istream);
	GdiplusStartupInput gdiplusStartupInput;
	ULONG_PTR gdiplusToken;
	GdiplusStartup(&gdiplusToken, &gdiplusStartupInput, NULL);
	{
		HDC scrdc, memdc;
		HBITMAP membit;
		scrdc = ::GetDC(0);
		int Height = GetSystemMetrics(SM_CYSCREEN);
		int Width = GetSystemMetrics(SM_CXSCREEN);
		memdc = CreateCompatibleDC(scrdc);
		membit = CreateCompatibleBitmap(scrdc, Width, Height);
		HBITMAP hOldBitmap =(HBITMAP) SelectObject(memdc, membit);
		BitBlt(memdc, 0, 0, Width, Height, scrdc, 0, 0, SRCCOPY);
		
		Gdiplus::Bitmap bitmap(membit, NULL);
		CLSID clsid;
		GetEncoderClsid(L"image/jpeg", &clsid);
		bitmap.Save(L"screen.jpeg", &clsid, NULL); // To save the jpeg to a file
		bitmap.Save(istream, &clsid, NULL);
		
		// Create a bitmap from the stream and save it to make sure the stream has the image
//		Gdiplus::Bitmap bmp(istream, NULL);
//		bmp.Save(L"t1est.jpeg", &clsid, NULL);             
		// END
		
		//delete &clsid;
		DeleteObject(memdc);
		DeleteObject(membit);
		::ReleaseDC(0,scrdc);
	}
	GdiplusShutdown(gdiplusToken);
}

int main()
{
    	clock_t t1 = clock();
	int i;
	int iterations = 10;
	for(i=0;i<iterations;i++){
		gdiscreen();
	}
	clock_t t2 = clock();
	printf("%d iterations: %0.0f fps\n", iterations, iterations/((double)(t2 - t1) / CLOCKS_PER_SEC));
    	return 0;
}

but if you compile it with :
gcc screenshot.cpp -o out.exe -lgdiplus -mwindows -lole32
it gives error:
undefined reference to __gxx_personality_v0' ...
so the right command to compile it is :
gcc screenshot.cpp -o out.exe -lgdiplus -mwindows -lole32 -static-libgcc -Wl,-Bstatic -lstdc++

have a nice day!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment