Простой класс для сохранения снимков окна (C++)

#ifndef __WINDOW_SHOT_H__
#define __WINDOW_SHOT_H__
 
class window_shot {
 
    // Идентификатор окна для Subclassing
    // Так же служит признаком успешной инициализации класса
    HWND m_wnd;
 
    // делать снимок всего окна или только клиентской области
    BOOL m_shot_full_window;
 
    void shot( HWND hWnd ) const {
        HDC hDC;
        RECT rc;
 
        // снимок всего окна, или только клиентской области?
        if ( m_shot_full_window ) {
            GetWindowRect( hWnd, &rc );
            hDC = GetWindowDC( hWnd );
        } else {
            GetClientRect( hWnd, &rc );
            hDC = GetDC( hWnd );
        }
 
        // получаем имя окна
        char caption[64];
        GetWindowText( hWnd, caption, 63 );
 
        // если окно без имени, задаём префикс файла по умолчанию
        if ( caption[0] == '\0' ) {
            strcpy( caption, "shot" );
        }
 
        // ищем первое свободное имя файла
        char filename[80];
        for ( int file_index = 0; file_index < 10000; ++file_index ) {
 
            //sprintf_s( filename, 80, "%s - %04d.bmp", caption, file_index );
            sprintf( filename, "%s - %04d.bmp", caption, file_index );
 
            // имя файла свободно?
            if ( GetFileAttributes( filename ) == INVALID_FILE_ATTRIBUTES ) break;
        }
 
        // раземры изображения
        LONG lWidth = rc.right - rc.left;
        LONG lHeight = rc.bottom - rc.top;
        DWORD dwSize = lHeight * (lWidth << 2); // размер в байтах
 
        BITMAPINFO bi = { { sizeof( BITMAPINFOHEADER ), lWidth, lHeight, 1, 32, BI_RGB, dwSize, 0, 0, 0, 0 }, { 0, 0, 0, 0 } };
 
        HDC hDCmem = CreateCompatibleDC( hDC );
        HBITMAP hBMP = CreateCompatibleBitmap( hDC, lWidth, lHeight );
        SelectObject( hDCmem, hBMP );
        BitBlt( hDCmem, 0, 0, lWidth, lHeight, hDC, 0, 0, SRCCOPY );
 
        char * p_image = new char [lWidth * lHeight * 4];
 
        GetDIBits( hDC, hBMP, 0, lHeight, p_image, &bi, DIB_RGB_COLORS );
 
        DWORD dwHdrSize = sizeof( BITMAPFILEHEADER ) + sizeof( BITMAPINFOHEADER );
        BITMAPFILEHEADER bfh = { ('M' << 8) | 'B', dwHdrSize + dwSize, 0, 0, dwHdrSize };
 
        // запись изображения
        HANDLE hFile = CreateFile( filename, GENERIC_WRITE, 0, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL );
        if ( hFile != INVALID_HANDLE_VALUE ) {
            DWORD dwWritten;
            WriteFile( hFile, &bfh, sizeof( bfh ), &dwWritten, NULL );
            WriteFile( hFile, &bi.bmiHeader, sizeof( bi.bmiHeader ), &dwWritten, NULL );
            WriteFile( hFile, p_image, dwSize, &dwWritten, NULL );
            CloseHandle( hFile );
        }
        delete [] p_image;
 
        DeleteObject( hBMP );
        DeleteDC( hDCmem );
        ReleaseDC( hWnd, hDC );
    }
 
    static LRESULT CALLBACK subclassproc( HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam, UINT_PTR, DWORD_PTR dwRefData ) {
        // VK_SNAPSHOT не посылается через WM_KEYDOWN
        if ( uMsg == WM_KEYUP && wParam == VK_SNAPSHOT ) {
            ((window_shot *)dwRefData)->shot( hWnd );
        }
        return DefSubclassProc( hWnd, uMsg, wParam, lParam );
    }
 
public:
    explicit window_shot( HWND hWnd, BOOL bShotFullWindow = TRUE ) : m_wnd( hWnd ), m_shot_full_window( bShotFullWindow ) {
        // Обрабатывать нажатие клавиши 'Print Screen' будем через Subclassing
        if ( !SetWindowSubclass( m_wnd, subclassproc, 0, (DWORD_PTR)this ) ) {
            m_wnd = NULL;
        }
    }
 
    virtual ~window_shot() {
        if ( m_wnd ) {
            RemoveWindowSubclass( m_wnd, subclassproc, 0 );
        }
    }
};
 
#endif // __WINDOW_SHOT_H__

#include <windows.h>
#include <commctrl.h>
#include <cstdio>
 
#include "window_shot.h"
 
static LRESULT CALLBACK wnd_proc( HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam ) {
    switch ( uMsg ) {
        case WM_DESTROY:
            PostQuitMessage( 0 );
            break;
 
        default:
            return DefWindowProc( hWnd, uMsg, wParam, lParam );
    }
    return 0;
}
 
int main()
{
    WNDCLASS wc;
    ZeroMemory( &wc, sizeof( WNDCLASS ) );
    wc.lpfnWndProc      = wnd_proc;
    wc.hInstance        = GetModuleHandle( NULL );
    wc.lpszClassName    = "test32cls";
    wc.hCursor          = LoadCursor( NULL, IDC_ARROW );
    wc.hbrBackground    = (HBRUSH)(COLOR_BTNFACE + 1);
    RegisterClass( &wc );
 
    HWND hWnd = CreateWindow( "test32cls", "Test Window", WS_OVERLAPPEDWINDOW | WS_VISIBLE, 100, 100, 300, 200, NULL, NULL, wc.hInstance, NULL );
 
    window_shot shooter( hWnd );
 
    MSG msg;
    while ( GetMessage( &msg, NULL, 0, 0 ) ) {
        TranslateMessage( &msg );
        DispatchMessage( &msg );
    }
    return 0;
}