截屏代码

  1 #define ALIGNEDONDWORD(theBytes) ( ( ( (DWORD)(theBytes) + 31 ) & ~31 ) >> 3 )
  2 
  3 BOOL BmpFromHandle( HDC hDC, HBITMAP hBitmap, WORD nBits, LPCTSTR lpszFile )
  4 {
  5     BITMAP bmp;
  6     WORD cClrBits;
  7     LPBITMAPINFO bi;
  8     BITMAPFILEHEADER bf;
  9     LPVOID lpData;
 10     HANDLE hFile;
 11     DWORD dwBytes;
 12     int nRet;
 13     BOOL bRet;
 14 
 15     nRet = ::GetObject( ( HANDLE )hBitmap, sizeof( BITMAP ), &bmp );
 16     if ( nRet != sizeof( BITMAP ) )
 17     {
 18         return FALSE;
 19     }
 20 
 21     bmp.bmBitsPixel = nBits;
 22 
 23     cClrBits = ( WORD )( bmp.bmPlanes * bmp.bmBitsPixel );
 24 
 25     if ( cClrBits == 1 )
 26     {
 27         cClrBits = 1;
 28     }
 29     else if ( cClrBits <= 4 )
 30     {
 31         cClrBits = 4;
 32     }
 33     else if ( cClrBits <= 8 )
 34     {
 35         cClrBits = 8;
 36     }
 37     else if ( cClrBits <= 16 )
 38     {
 39         cClrBits = 16;
 40     }
 41     else if ( cClrBits <= 24 )
 42     {
 43         cClrBits = 24;
 44     }
 45     else
 46     {
 47         cClrBits = 32;
 48     }
 49 
 50     if ( cClrBits < 24 )
 51     {
 52         bi = ( LPBITMAPINFO )calloc( 1, sizeof( BITMAPINFOHEADER ) + sizeof( RGBQUAD ) * ( 1 << cClrBits ) );
 53     }
 54     else
 55     {
 56         bi = ( LPBITMAPINFO )calloc( 1, sizeof( BITMAPINFOHEADER ) );
 57     }
 58 
 59     if ( bi == NULL )
 60     {
 61         return FALSE;
 62     }
 63 
 64     bi->bmiHeader.biSize = sizeof( BITMAPINFOHEADER );
 65     bi->bmiHeader.biWidth = bmp.bmWidth;
 66     bi->bmiHeader.biHeight = bmp.bmHeight;
 67     bi->bmiHeader.biPlanes = bmp.bmPlanes;
 68     bi->bmiHeader.biBitCount = bmp.bmBitsPixel;
 69     bi->bmiHeader.biCompression = BI_RGB;
 70     bi->bmiHeader.biSizeImage = ( ( ( bi->bmiHeader.biWidth * cClrBits + 31 ) & ~31 ) >> 3 ) * bi->bmiHeader.biHeight;
 71     /*bi->bmiHeader.biXPelsPerMeter = 0;
 72     bi->bmiHeader.biYPelsPerMeter = 0;*/
 73 
 74     if ( cClrBits < 24 )
 75     {
 76         bi->bmiHeader.biClrUsed = ( 1 << cClrBits );
 77     }
 78     /*else
 79     {
 80         bi->bmiHeader.biClrUsed = 0;
 81     }*/
 82 
 83     /*bi->bmiHeader.biClrImportant = 0;*/
 84 
 85 
 86     lpData = calloc( ( bi->bmiHeader.biSizeImage + 4095 ) >> 12, 4096 );
 87     if ( lpData == NULL )
 88     {
 89         free( bi );
 90         return FALSE;
 91     }
 92 
 93     nRet = ::GetDIBits( hDC, hBitmap, 0, bi->bmiHeader.biHeight, lpData, bi, DIB_RGB_COLORS );
 94     if ( nRet != bi->bmiHeader.biHeight )
 95     {
 96         goto err_exit;
 97     }
 98 
 99     if ( cClrBits < 24 )
100     {
101         bi->bmiHeader.biClrUsed = ( 1 << cClrBits );
102     }
103 
104     memset( &bf, 0, sizeof( BITMAPFILEHEADER ) );
105     bf.bfType = 0x4d42;
106     bf.bfSize = sizeof( BITMAPFILEHEADER ) + sizeof( BITMAPINFOHEADER ) + bi->bmiHeader.biClrUsed * sizeof( RGBQUAD ) + bi->bmiHeader.biSizeImage;
107     /*bf.bfReserved1 = 0;
108     bf.bfReserved2 = 0;*/
109     bf.bfOffBits = bf.bfSize - bi->bmiHeader.biSizeImage;
110 
111     hFile = ::CreateFile( lpszFile, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL );
112     if ( hFile == INVALID_HANDLE_VALUE )
113     {
114         goto err_exit;
115     }
116 
117     bRet = ::WriteFile( hFile, &bf, sizeof( BITMAPFILEHEADER ), &dwBytes, NULL );
118     if ( !bRet || dwBytes != sizeof( BITMAPFILEHEADER ) )
119     {
120         ::CloseHandle( hFile );
121         goto err_exit;
122     }
123 
124     bRet = ::WriteFile( hFile, bi, bf.bfOffBits - sizeof( BITMAPFILEHEADER ), &dwBytes, NULL );
125     if ( !bRet || dwBytes != bf.bfOffBits - sizeof( BITMAPFILEHEADER ) )
126     {
127         ::CloseHandle( hFile );
128         goto err_exit;
129     }
130 
131     bRet = ::WriteFile( hFile, lpData, bi->bmiHeader.biSizeImage, &dwBytes, NULL );
132     if ( !bRet || dwBytes != bi->bmiHeader.biSizeImage )
133     {
134         ::CloseHandle( hFile );
135         goto err_exit;
136     }
137 
138     ::CloseHandle( hFile );
139 
140     free( lpData );
141     free( bi );
142     return TRUE;
143 
144 err_exit:
145     free( lpData );
146     free( bi );
147     return FALSE;
148 }
149 
150 BOOL CopyScreen( LPCTSTR lpszFile, WORD nBits = 24 )
151 {
152     HDC hDC, hMemDC;
153     HBITMAP hBitmap, hBitmapOld;
154     int nWidth, nHeight;
155     BOOL bRet;
156 
157     bRet = TRUE;
158 
159     hDC = ::GetDC( NULL );
160     nWidth = ::GetSystemMetrics( SM_CXSCREEN );
161     nHeight = ::GetSystemMetrics( SM_CYSCREEN );
162     hMemDC = ::CreateCompatibleDC( hDC );
163     hBitmap = ::CreateCompatibleBitmap( hDC, nWidth, nHeight );
164     hBitmapOld = ( HBITMAP )::SelectObject( hMemDC, ( HGDIOBJ )hBitmap );
165 
166     bRet = ::BitBlt( hMemDC, 0, 0, nWidth, nHeight, hDC, 0, 0, SRCCOPY );
167     if ( bRet )
168     {
169         bRet = BmpFromHandle( hMemDC, hBitmap, nBits, lpszFile );
170     }
171 
172     ::SelectObject( hMemDC, ( HGDIOBJ )hBitmapOld );
173     ::DeleteObject( ( HGDIOBJ )hBitmap );
174     ::DeleteDC( hMemDC );
175     ::ReleaseDC( NULL, hDC );
176     return bRet;
177 }
178 
179 BOOL CopyRC( UINT nBitmapID, LPCTSTR lpszFile, WORD nBits = 24 )
180 {
181     HDC hDC;
182     HBITMAP hBitmap;
183     BOOL bRet;
184 
185     bRet = TRUE;
186 
187     hDC = ::GetDC( NULL );
188 
189     hBitmap = ::LoadBitmap( ::GetModuleHandle( NULL ), MAKEINTRESOURCE( nBitmapID ) );
190     if ( hBitmap )
191     {
192         bRet = BmpFromHandle( hDC, hBitmap, nBits, lpszFile );
193     }
194     else
195     {
196         bRet = FALSE;
197     }
198 
199     ::DeleteObject( ( HGDIOBJ )hBitmap );
200     ::ReleaseDC( NULL, hDC );
201 
202     return bRet;
203 }