Print Screen button not taking Screenshots

[Laz]

If you can convince @Alex. to implement this, here is some code wrote that makes print screen not use the stupid FL system, and instead takes nice .png files that are properly categorised by date.

Code:

#define min std::min
#define max std::max
#include <gdiplus.h>
#undef min
#undef max


    using namespace Gdiplus;

    int GetEncoderClsid(const WCHAR* format, CLSID* pClsid)
    {
        UINT  num = 0;          // number of image encoders
        UINT  size = 0;         // size of the image encoder array in bytes

        ImageCodecInfo* pImageCodecInfo = nullptr;

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

        pImageCodecInfo = (ImageCodecInfo*)(malloc(size));
        if (pImageCodecInfo == nullptr)
            return -1;  // Failure

        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;  // Success
            }
        }
        free(pImageCodecInfo);
        return -1;  // Failure
    }


    DWORD PrintScreen()
    {
        char directory[MAX_PATH];
        if (!GetScreenShotPath(directory))
        {
            return DWORD(-1);
        }

        // get the device context of the screen
        HDC hScreenDC = GetDC(nullptr);

        // and a device context to put it in
        HDC hMemoryDC = CreateCompatibleDC(hScreenDC);

        int width = GetDeviceCaps(hScreenDC, HORZRES);
        int height = GetDeviceCaps(hScreenDC, VERTRES);

        // maybe worth checking these are positive values
        HBITMAP hBitmap = CreateCompatibleBitmap(hScreenDC, width, height);

        // get a new bitmap
        HBITMAP hOldBitmap = HBITMAP(SelectObject(hMemoryDC, hBitmap));

        BitBlt(hMemoryDC, 0, 0, width, height, hScreenDC, 0, 0, SRCCOPY);
        hBitmap = (HBITMAP)SelectObject(hMemoryDC, hOldBitmap);

        std::time_t rawtime;
        std::tm* timeinfo;
        char buffer[80];

        std::time(&rawtime);
        timeinfo = std::localtime(&rawtime);

        std::strftime(buffer, 80, "%Y-%m-%d", timeinfo);
        std::puts(buffer);

        std::string dir = std::string(directory) + "/" + std::string(buffer) + "/";
        std::filesystem::create_directories(dir);

        std::strftime(buffer, 80, "%H-%M-%S", timeinfo);
        std::puts(buffer);
        std::string file = std::string(buffer) + ".png";

        GdiplusStartupInput gdiplusStartupInput;
        ULONG_PTR gdiplusToken;
        GdiplusStartup(&gdiplusToken, &gdiplusStartupInput, nullptr);
        Bitmap* image = new Bitmap(hBitmap, nullptr);

        CLSID myClsId;
        GetEncoderClsid(L"image/png", &myClsId);

        Status status = image->Save((WCHAR*)Utils::String::stows(dir + file).c_str(), &myClsId, nullptr);
        delete image;

        GdiplusShutdown(gdiplusToken);

        // clean up
        DeleteDC(hMemoryDC);
        DeleteDC(hScreenDC);

        return DWORD(-1);
    }
    
    void HackPrintScreen()
    {
        #define PrintScreenEntryPoint ((PBYTE)0x425170)
        ProtectExecuteReadWrite(PrintScreenEntryPoint, 5);
        PrintScreenEntryPoint[0] = 0xe9;
        *(DWORD*)(PrintScreenEntryPoint + 1) = PBYTE(PrintScreen) - PrintScreenEntryPoint - 5;
    }

---