Снимок экрана активного окна с Python PIL и Windows API: как бороться с закругленными углами?

Для этого проекта я делаю снимки экрана с помощью Windows API (для работы с несколькими экранами) и конвертирую их в изображение PIL; затем при желании добавляю тень вокруг окна.

Моя проблема в том, что снимок экрана на самом деле представляет собой прямоугольник окна; Это означает, что я снимаю фон вокруг закругленных углов, а я этого не хочу. Я немного погуглил и нашел документы и советы по прозрачности, и я предполагаю, что должен найти способ получить форму окна, чтобы сделать его маской, которую я применил к (прямоугольному) изображению, которое у меня есть. Но я так и не нашел эту маску. Может ли кто-нибудь помочь?

Ниже мой код:

hwnd = win32gui.GetForegroundWindow()

l, t, r, b = win32gui.GetWindowRect(hwnd)
w = r - l
h = b - t

hwndDC = win32gui.GetWindowDC(hwnd)
mfcDC  = win32ui.CreateDCFromHandle(hwndDC)
saveDC = mfcDC.CreateCompatibleDC()

saveBitMap = win32ui.CreateBitmap()
saveBitMap.CreateCompatibleBitmap(mfcDC, w, h)
saveDC.SelectObject(saveBitMap)

saveDC.BitBlt((0, 0), (w, h),  mfcDC,  (0, 0),  win32con.SRCCOPY)

#add cursor
if showcursor:
    curFlags, curH, (curX, curY) = win32gui.GetCursorInfo()
    saveDC.DrawIcon((curX, curY), curH)

#load into PIL image
"""http://stackoverflow.com/questions/4199497/image-frombuffer-with-16-bit-image-data"""
bmpinfo = saveBitMap.GetInfo()
bmpstr = saveBitMap.GetBitmapBits(True)
im = Image.frombuffer(
    'RGB',
    (bmpinfo['bmWidth'], bmpinfo['bmHeight']),
    bmpstr, 'raw', 'BGRX', 0, 1)

win32gui.DeleteObject(saveBitMap.GetHandle())
saveDC.DeleteDC()
mfcDC.DeleteDC()
win32gui.ReleaseDC(hwnd, hwndDC)

return im

Ниже приведен немного увеличенный снимок экрана с окном над синим фоном:

Как вы можете видеть, есть синие углы, которых не должно быть.

16
задан martineau 5 November 2012 в 12:44
поделиться