Скриншот окна - PullRequest
Новая
       22

Скриншот окна

1 голос
/ 23 декабря 2009

Я писал некоторый код, чтобы сделать скриншот окна (в Windows). Код работает нормально, однако перед захватом экрана я должен вывести окно вперед, которое я хочу захватить, и вызвать перерисовку.

Я выполняю перерисовку с помощью InvalidateRect, затем мне нужно выкачать некоторые сообщения из цикла сообщений, чтобы обработать WM_PAINT. Это явно немного неубедительно, так как я не знаю, сколько сообщений прокачать.

Я пытался использовать RedrawWindow с RDW_ALLCHILDREN, однако приложение, с которого я беру экран, является приложением MDI и, похоже, не перерисовывает всех своих детей.

Итак, мой вопрос: есть ли лучший способ перерисовать окно до захвата экрана?

Приветствие Рич

1 Ответ

4 голосов
/ 27 декабря 2009

Поскольку вы не упомянули язык, который вы используете, я надеюсь, что следующий код на C ++ поможет вам! 

void getScreenShot( int texWidth, int texHeight, unsigned char* pBuffer, HWND handle )
{
    /* Local variables */
    HDC screenDC;
    RECT screenRect;
    int extraBytesPerRow;
    BITMAPINFO bitmapInfo;
    HDC bitmapDC;
    void* bitmapDataPtr;
    HBITMAP hBitmap;
    HBITMAP hPrevBitmap;
    unsigned char* pIn;
    unsigned char* pOut;
    int rowIndex;
    int colIndex;
    /* Get a DC from the desktop window */
    screenDC = GetDC(handle);
    GetClientRect(handle, &screenRect );
    /* Determine the extra bytes we need per row (each row of bitmap data must end on a 32bit boundary) */
    extraBytesPerRow = ( texWidth * 3 ) % 4;
    extraBytesPerRow = extraBytesPerRow ? 4 - extraBytesPerRow : 0;
    /* Setup the bitmap info structure */
    memset( &bitmapInfo, 0, sizeof( bitmapInfo ) );
    bitmapInfo.bmiHeader.biSize = sizeof( BITMAPINFOHEADER );
    bitmapInfo.bmiHeader.biWidth = texWidth;
    bitmapInfo.bmiHeader.biHeight = texHeight;
    bitmapInfo.bmiHeader.biPlanes = 1;
    bitmapInfo.bmiHeader.biBitCount = 24;
    bitmapInfo.bmiHeader.biCompression = BI_RGB;
    /* Create a bitmap device context (bitmapDataPtr will be a pointer to the bits in the bitmap) */
    bitmapDC = CreateCompatibleDC( NULL );
    hBitmap = CreateDIBSection( bitmapDC, ( BITMAPINFO* )&bitmapInfo.bmiHeader, DIB_RGB_COLORS, &bitmapDataPtr, NULL, 0 );
    hPrevBitmap = ( HBITMAP )SelectObject( bitmapDC, hBitmap );
    /* BitBlt or StretchBlt the image from the input DC into our bitmap DC */
    if ( ( texWidth != screenRect.right ) || ( texHeight != screenRect.bottom ) )
    {
        SetStretchBltMode( bitmapDC, HALFTONE );
        StretchBlt( bitmapDC, 0, 0, texWidth, texHeight, screenDC, 0, 0, screenRect.right, screenRect.bottom, SRCCOPY );
    }
    else
    {
        BitBlt( bitmapDC, 0, 0, texWidth, texHeight, screenDC, 0, 0, SRCCOPY);
    }
    /* Copy the data from the bitmap to the user's buffer (bitmap data is BGR and 4 byte aligned on each row, we want tightly-packed RGB) */
    pIn = ( unsigned char* )bitmapDataPtr;
    pOut = pBuffer;
    for ( rowIndex = 0; rowIndex < texHeight; rowIndex++ )
    {
        for ( colIndex = 0; colIndex < texWidth; colIndex++ )
        {
            pOut[ 0 ] = pIn[2];
            pOut[ 1 ] = pIn[1];
            pOut[ 2 ] = pIn[0];
            pOut += 3;
            pIn += 3;
        }
        pIn += extraBytesPerRow;
    }
    /* Free memory used by the bitmap */
    SelectObject( bitmapDC, hPrevBitmap );
    DeleteObject( hBitmap );
    DeleteDC( bitmapDC );
    /* Release the screen DC */
    ReleaseDC(handle, screenDC );
}

На самом деле вам не нужно форсировать перерисовку .. Но в случае, если окно свернуто, вам может потребоваться его вызвать, прежде чем вызывать функцию с дескриптором окна ...! texWidth и texHeight - это размер окна, которое вы собираетесь захватить; чтобы получить это, вы можете использовать, GetWindowRect (..) или проверить ссылку здесь: ссылка

...