2021-10-04 14:45:36 +00:00
|
|
|
|
#include "screen_capture.h"
|
|
|
|
|
|
|
|
|
|
#include <conio.h>
|
|
|
|
|
#include <stdio.h>
|
|
|
|
|
#include <QDebug>
|
|
|
|
|
#include <QString>
|
|
|
|
|
|
2021-10-06 15:31:47 +00:00
|
|
|
|
#if _MSC_VER >= 1600
|
|
|
|
|
#pragma execution_character_set("utf-8")
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
2021-10-05 02:27:53 +00:00
|
|
|
|
#define WIDEN2(x) L ## x
|
2021-10-06 15:31:47 +00:00
|
|
|
|
#define WIDEN(x) WIDEN2(x)
|
|
|
|
|
#define __WFILE__ WIDEN(__FILE__)
|
|
|
|
|
#define HRCHECK(__expr) {hr=(__expr);if(FAILED(hr)){wprintf(L"FAILURE 0x%08X (%i)\n\tline: %u file: '%s'\n\texpr: '" WIDEN(#__expr) L"'\n",hr, hr, __LINE__,__WFILE__);goto cleanup;}}
|
|
|
|
|
#define RELEASE(__p) {if(__p!=nullptr){__p->Release();__p=nullptr;}}
|
2021-10-05 02:27:53 +00:00
|
|
|
|
|
|
|
|
|
// ȱenum<75><6D><EFBFBD><EFBFBD>
|
|
|
|
|
// https://www.gamedev.net/forums/topic/132636-enum-displaymode--dx9--false/
|
|
|
|
|
|
2021-10-06 15:31:47 +00:00
|
|
|
|
HRESULT SavePixelsToFile32bppPBGRA(UINT width, UINT height, UINT stride,
|
|
|
|
|
LPBYTE pixels, LPWSTR filePath, const GUID &format)
|
|
|
|
|
{
|
2021-10-05 02:27:53 +00:00
|
|
|
|
if (!filePath || !pixels)
|
|
|
|
|
return E_INVALIDARG;
|
|
|
|
|
|
|
|
|
|
HRESULT hr = S_OK;
|
|
|
|
|
IWICImagingFactory *factory = nullptr;
|
|
|
|
|
IWICBitmapEncoder *encoder = nullptr;
|
|
|
|
|
IWICBitmapFrameEncode *frame = nullptr;
|
|
|
|
|
IWICStream *stream = nullptr;
|
|
|
|
|
GUID pf = GUID_WICPixelFormat32bppPBGRA;
|
|
|
|
|
BOOL coInit = CoInitialize(nullptr);
|
|
|
|
|
|
|
|
|
|
HRCHECK(CoCreateInstance(CLSID_WICImagingFactory, nullptr, CLSCTX_INPROC_SERVER, IID_PPV_ARGS(&factory)));
|
|
|
|
|
HRCHECK(factory->CreateStream(&stream));
|
|
|
|
|
HRCHECK(stream->InitializeFromFilename(filePath, GENERIC_WRITE));
|
|
|
|
|
HRCHECK(factory->CreateEncoder(format, nullptr, &encoder));
|
|
|
|
|
HRCHECK(encoder->Initialize(stream, WICBitmapEncoderNoCache));
|
|
|
|
|
HRCHECK(encoder->CreateNewFrame(&frame, nullptr)); // we don't use options here
|
|
|
|
|
HRCHECK(frame->Initialize(nullptr)); // we dont' use any options here
|
|
|
|
|
HRCHECK(frame->SetSize(width, height));
|
|
|
|
|
HRCHECK(frame->SetPixelFormat(&pf));
|
|
|
|
|
HRCHECK(frame->WritePixels(height, stride, stride * height, pixels));
|
|
|
|
|
HRCHECK(frame->Commit());
|
|
|
|
|
HRCHECK(encoder->Commit());
|
|
|
|
|
|
2021-10-06 15:31:47 +00:00
|
|
|
|
cleanup:
|
2021-10-05 02:27:53 +00:00
|
|
|
|
RELEASE(stream);
|
|
|
|
|
RELEASE(frame);
|
|
|
|
|
RELEASE(encoder);
|
|
|
|
|
RELEASE(factory);
|
|
|
|
|
if (coInit) CoUninitialize();
|
|
|
|
|
return hr;
|
2021-10-06 15:31:47 +00:00
|
|
|
|
}
|
2021-10-05 02:27:53 +00:00
|
|
|
|
|
|
|
|
|
HRESULT Direct3D9TakeScreenshots(UINT adapter, UINT count)
|
|
|
|
|
{
|
|
|
|
|
HRESULT hr = S_OK;
|
|
|
|
|
IDirect3D9 *d3d = nullptr;
|
|
|
|
|
IDirect3DDevice9 *device = nullptr;
|
|
|
|
|
IDirect3DSurface9 *surface = nullptr;
|
|
|
|
|
D3DPRESENT_PARAMETERS parameters = { 0 };
|
|
|
|
|
D3DDISPLAYMODE mode;
|
|
|
|
|
D3DLOCKED_RECT rc;
|
|
|
|
|
UINT pitch;
|
|
|
|
|
SYSTEMTIME st;
|
|
|
|
|
LPBYTE *shots = nullptr;
|
|
|
|
|
|
|
|
|
|
// init D3D and get screen size
|
|
|
|
|
d3d = Direct3DCreate9(D3D_SDK_VERSION);
|
|
|
|
|
HRCHECK(d3d->GetAdapterDisplayMode(adapter, &mode));
|
|
|
|
|
|
|
|
|
|
parameters.Windowed = TRUE;
|
|
|
|
|
parameters.BackBufferCount = 1;
|
|
|
|
|
parameters.BackBufferHeight = mode.Height;
|
|
|
|
|
parameters.BackBufferWidth = mode.Width;
|
|
|
|
|
parameters.SwapEffect = D3DSWAPEFFECT_DISCARD;
|
|
|
|
|
parameters.hDeviceWindow = NULL;
|
|
|
|
|
|
|
|
|
|
// create device & capture surface
|
|
|
|
|
HRCHECK(d3d->CreateDevice(adapter, D3DDEVTYPE_HAL, NULL, D3DCREATE_SOFTWARE_VERTEXPROCESSING, ¶meters, &device));
|
|
|
|
|
HRCHECK(device->CreateOffscreenPlainSurface(mode.Width, mode.Height, D3DFMT_A8R8G8B8, D3DPOOL_SYSTEMMEM, &surface, nullptr));
|
|
|
|
|
|
|
|
|
|
// compute the required buffer size
|
|
|
|
|
HRCHECK(surface->LockRect(&rc, NULL, 0));
|
|
|
|
|
pitch = rc.Pitch;
|
|
|
|
|
HRCHECK(surface->UnlockRect());
|
|
|
|
|
|
|
|
|
|
// allocate screenshots buffers
|
|
|
|
|
shots = new LPBYTE[count];
|
|
|
|
|
for (UINT i = 0; i < count; i++)
|
|
|
|
|
{
|
|
|
|
|
shots[i] = new BYTE[pitch * mode.Height];
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
GetSystemTime(&st); // measure the time we spend doing <count> captures
|
|
|
|
|
wprintf(L"%i:%i:%i.%i\n", st.wHour, st.wMinute, st.wSecond, st.wMilliseconds);
|
|
|
|
|
for (UINT i = 0; i < count; i++)
|
|
|
|
|
{
|
|
|
|
|
// get the data
|
|
|
|
|
HRCHECK(device->GetFrontBufferData(0, surface));
|
|
|
|
|
|
|
|
|
|
// copy it into our buffers
|
|
|
|
|
HRCHECK(surface->LockRect(&rc, NULL, 0));
|
|
|
|
|
CopyMemory(shots[i], rc.pBits, rc.Pitch * mode.Height);
|
|
|
|
|
HRCHECK(surface->UnlockRect());
|
|
|
|
|
}
|
|
|
|
|
GetSystemTime(&st);
|
|
|
|
|
wprintf(L"%i:%i:%i.%i\n", st.wHour, st.wMinute, st.wSecond, st.wMilliseconds);
|
|
|
|
|
|
|
|
|
|
// save all screenshots
|
|
|
|
|
for (UINT i = 0; i < count; i++)
|
|
|
|
|
{
|
|
|
|
|
WCHAR file[100];
|
|
|
|
|
wsprintf(file, L"cap%i.png", i);
|
|
|
|
|
HRCHECK(SavePixelsToFile32bppPBGRA(mode.Width, mode.Height, pitch, shots[i], file, GUID_ContainerFormatPng));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
cleanup:
|
|
|
|
|
if (shots != nullptr)
|
|
|
|
|
{
|
|
|
|
|
for (UINT i = 0; i < count; i++)
|
|
|
|
|
{
|
|
|
|
|
delete shots[i];
|
|
|
|
|
}
|
|
|
|
|
delete[] shots;
|
|
|
|
|
}
|
|
|
|
|
RELEASE(surface);
|
|
|
|
|
RELEASE(device);
|
|
|
|
|
RELEASE(d3d);
|
|
|
|
|
return hr;
|
|
|
|
|
}
|
|
|
|
|
|
2021-10-04 14:45:36 +00:00
|
|
|
|
ScreenCapture::ScreenCapture()
|
|
|
|
|
{
|
2021-10-06 15:31:47 +00:00
|
|
|
|
m_d3d9_dev = ::Direct3DCreate9(D3D_SDK_VERSION);
|
2021-10-04 14:45:36 +00:00
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
2021-10-06 15:31:47 +00:00
|
|
|
|
BOOL CALLBACK MonitorEnumProc(HMONITOR hMonitor,HDC hdcMonitor,
|
|
|
|
|
LPRECT lprcMonitor,LPARAM dwData)
|
2021-10-04 14:45:36 +00:00
|
|
|
|
{
|
|
|
|
|
MONITORINFOEX mi;
|
|
|
|
|
mi.cbSize=sizeof(MONITORINFOEX);
|
|
|
|
|
GetMonitorInfo(hMonitor,&mi);
|
|
|
|
|
qDebug()<<QString::asprintf("Device name:%s\t",mi.szDevice);
|
|
|
|
|
if(mi.dwFlags==MONITORINFOF_PRIMARY) printf("Primary monitor!\n");
|
|
|
|
|
else printf("\n");
|
|
|
|
|
qDebug()<<QString::asprintf("Monitor rectangle:(%d,%d,%d,%d)\n",mi.rcMonitor.left,mi.rcMonitor.top,
|
|
|
|
|
mi.rcMonitor.right,mi.rcMonitor.bottom);
|
|
|
|
|
qDebug()<<QString::asprintf("Work rectangle:(%d,%d,%d,%d)\n",mi.rcWork.left,mi.rcWork.top,
|
|
|
|
|
mi.rcWork.right,mi.rcWork.bottom);
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void ScreenCapture::EnumScreen()
|
|
|
|
|
{
|
2021-10-06 15:31:47 +00:00
|
|
|
|
// EnumDisplayMonitors(NULL,NULL,MonitorEnumProc,NULL);
|
|
|
|
|
if(m_d3d9_dev == NULL)
|
|
|
|
|
{
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
D3DADAPTER_IDENTIFIER9 adapterID; // Used to store device info
|
|
|
|
|
char strBuffer[20480];
|
|
|
|
|
DWORD dwDisplayCount = m_d3d9_dev->GetAdapterCount();
|
|
|
|
|
for(DWORD i = 0; i < dwDisplayCount; i++)
|
|
|
|
|
{
|
|
|
|
|
if( m_d3d9_dev->GetAdapterIdentifier( i/*D3DADAPTER_DEFAULT*/, 0,&adapterID ) != D3D_OK )
|
|
|
|
|
{
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
qDebug()<<adapterID.DeviceName;
|
|
|
|
|
}
|
2021-10-04 14:45:36 +00:00
|
|
|
|
}
|
2021-10-06 15:31:47 +00:00
|
|
|
|
|
|
|
|
|
|