Commit | Line | Data |
---|---|---|
0ddb7051 TC |
1 | #include "imext.h" |
2 | #include <windows.h> | |
3 | #include <string.h> | |
4 | ||
5 | i_img * | |
6 | imss_win32(unsigned hwnd_u, int include_decor) { | |
7 | HWND hwnd = (HWND)hwnd_u; | |
8 | HDC wdc, bmdc; | |
9 | RECT rect; | |
10 | HBITMAP work_bmp, old_dc_bmp; | |
11 | int width, height; | |
12 | BITMAPINFO bmi; | |
13 | unsigned char *di_bits; | |
14 | i_img *result = NULL; | |
15 | ||
16 | i_clear_error(); | |
17 | ||
18 | if (!hwnd) | |
19 | hwnd = GetDesktopWindow(); | |
20 | ||
21 | if (include_decor) { | |
22 | wdc = GetWindowDC(hwnd); | |
23 | GetWindowRect(hwnd, &rect); | |
24 | } | |
25 | else { | |
26 | wdc = GetDC(hwnd); | |
27 | GetClientRect(hwnd, &rect); | |
28 | } | |
29 | if (!wdc) { | |
30 | i_push_error(0, "Cannot get window DC - invalid hwnd?"); | |
31 | return NULL; | |
32 | } | |
33 | ||
34 | width = rect.right - rect.left; | |
35 | height = rect.bottom - rect.top; | |
36 | work_bmp = CreateCompatibleBitmap(wdc, width, height); | |
37 | bmdc = CreateCompatibleDC(wdc); | |
38 | old_dc_bmp = SelectObject(bmdc, work_bmp); | |
39 | BitBlt(bmdc, 0, 0, width, height, wdc, 0, 0, SRCCOPY); | |
40 | ||
41 | /* make a dib */ | |
42 | memset(&bmi, 0, sizeof(bmi)); | |
43 | bmi.bmiHeader.biSize = sizeof(bmi); | |
44 | bmi.bmiHeader.biWidth = width; | |
45 | bmi.bmiHeader.biHeight = -height; | |
46 | bmi.bmiHeader.biPlanes = 1; | |
47 | bmi.bmiHeader.biBitCount = 32; | |
48 | bmi.bmiHeader.biCompression = BI_RGB; | |
49 | ||
50 | di_bits = mymalloc(4 * width * height); | |
51 | if (GetDIBits(bmdc, work_bmp, 0, height, di_bits, &bmi, DIB_RGB_COLORS)) { | |
52 | i_color *line = mymalloc(sizeof(i_color) * width); | |
53 | i_color *cp; | |
54 | int x, y; | |
55 | unsigned char *ch_pp = di_bits; | |
56 | result = i_img_8_new(width, height, 3); | |
57 | ||
58 | for (y = 0; y < height; ++y) { | |
59 | cp = line; | |
60 | for (x = 0; x < width; ++x) { | |
61 | cp->rgb.b = *ch_pp++; | |
62 | cp->rgb.g = *ch_pp++; | |
63 | cp->rgb.r = *ch_pp++; | |
64 | ch_pp++; | |
65 | cp++; | |
66 | } | |
67 | i_plin(result, 0, width, y, line); | |
68 | } | |
69 | myfree(line); | |
70 | } | |
71 | ||
72 | /* clean up */ | |
73 | myfree(di_bits); | |
74 | SelectObject(bmdc, old_dc_bmp); | |
75 | DeleteDC(bmdc); | |
76 | DeleteObject(work_bmp); | |
77 | ReleaseDC(hwnd, wdc); | |
78 | ||
79 | return result; | |
80 | } |