def make_ICONINFO(w, h, rgb_data, rgb_format="BGRA"):
log("make_ICONINFO(%i, %i, %i bytes, %s)", w, h, len(rgb_data), rgb_format)
bitmap = 0
mask = 0
try:
bytes_per_pixel = len(rgb_format)
bitmap = rgb_to_bitmap(rgb_data, bytes_per_pixel, w, h)
log("rgb_to_bitmap(%i bytes, %i, %i, %i)=%s", len(rgb_data),
bytes_per_pixel, w, h, bitmap)
mask = CreateBitmap(w, h, 1, 1, None)
log("CreateBitmap(%i, %i, 1, 1, None)=%#x", w, h, mask or 0)
if not mask:
raise ctypes.WinError(ctypes.get_last_error())
iconinfo = ICONINFO()
iconinfo.fIcon = True
iconinfo.hbmMask = mask
iconinfo.hbmColor = bitmap
hicon = CreateIconIndirect(byref(iconinfo))
log("CreateIconIndirect()=%#x", hicon or 0)
if not hicon:
raise ctypes.WinError(ctypes.get_last_error())
return hicon
except Exception:
log.error("Error: failed to set tray icon", exc_info=True)
return FALLBACK_ICON
finally:
if mask:
DeleteObject(mask)
if bitmap:
DeleteObject(bitmap)
def make_ICONINFO(w, h, bgra):
bitmap = 0
mask = 0
try:
bitmap = rgba_to_bitmap(bgra, w, h)
mask = CreateBitmap(w, h, 1, 1, None)
iconinfo = ICONINFO()
iconinfo.fIcon = True
iconinfo.hbmMask = mask
iconinfo.hbmColor = bitmap
hicon = CreateIconIndirect(byref(iconinfo))
log("CreateIconIndirect()=%#x", hicon)
if not hicon:
raise ctypes.WinError(ctypes.get_last_error())
return hicon
except Exception:
log.error("Error: failed to set tray icon", exc_info=True)
return FALLBACK_ICON
finally:
if mask:
DeleteObject(mask)
if bitmap:
DeleteObject(bitmap)
def get_cursor_data(hCursor):
#w, h = get_fixed_cursor_size()
if not hCursor:
return None
x, y = 0, 0
dc = None
memdc = None
bitmap = None
old_handle = None
pixels = None
try:
ii = ICONINFO()
ii.cbSize = sizeof(ICONINFO)
if not GetIconInfo(hCursor, byref(ii)):
raise WindowsError() #@UndefinedVariable
x = ii.xHotspot
y = ii.yHotspot
cursorlog(
"get_cursor_data(%#x) hotspot at %ix%i, hbmColor=%#x, hbmMask=%#x",
hCursor, x, y, ii.hbmColor or 0, ii.hbmMask or 0)
if not ii.hbmColor:
#FIXME: we don't handle black and white cursors
return None
iie = ICONINFOEXW()
iie.cbSize = sizeof(ICONINFOEXW)
if not GetIconInfoExW(hCursor, byref(iie)):
raise WindowsError() #@UndefinedVariable
name = iie.szResName[:MAX_PATH]
cursorlog("wResID=%#x, sxModName=%s, szResName=%s", iie.wResID,
iie.sxModName[:MAX_PATH], name)
bm = Bitmap()
if not GetObjectA(ii.hbmColor, sizeof(Bitmap), byref(bm)):
raise WindowsError() #@UndefinedVariable
cursorlog(
"cursor bitmap: type=%i, width=%i, height=%i, width bytes=%i, planes=%i, bits pixel=%i, bits=%#x",
bm.bmType, bm.bmWidth, bm.bmHeight, bm.bmWidthBytes, bm.bmPlanes,
bm.bmBitsPixel, bm.bmBits or 0)
w = bm.bmWidth
h = bm.bmHeight
dc = GetDC(None)
assert dc, "failed to get a drawing context"
memdc = CreateCompatibleDC(dc)
assert memdc, "failed to get a compatible drawing context from %s" % dc
bitmap = CreateCompatibleBitmap(dc, w, h)
assert bitmap, "failed to get a compatible bitmap from %s" % dc
old_handle = SelectObject(memdc, bitmap)
#check if icon is animated:
UINT_MAX = 2**32 - 1
if not DrawIconEx(memdc, 0, 0, hCursor, w, h, UINT_MAX, 0, 0):
cursorlog("cursor is animated!")
#if not DrawIcon(memdc, 0, 0, hCursor):
if not DrawIconEx(memdc, 0, 0, hCursor, w, h, 0, 0,
win32con.DI_NORMAL):
raise WindowsError() #@UndefinedVariable
buf_size = bm.bmWidthBytes * h
buf = create_string_buffer(b"", buf_size)
r = GetBitmapBits(bitmap, buf_size, byref(buf))
cursorlog("get_cursor_data(%#x) GetBitmapBits(%#x, %#x, %#x)=%i",
hCursor, bitmap, buf_size, addressof(buf), r)
if not r:
cursorlog.error("Error: failed to copy screen bitmap data")
return None
elif r != buf_size:
cursorlog.warn(
"Warning: invalid cursor buffer size, got %i bytes but expected %i",
r, buf_size)
return None
else:
#32-bit data:
pixels = bytearray(strtobytes(buf.raw))
has_alpha = False
has_pixels = False
for i in range(len(pixels) // 4):
has_pixels = has_pixels or pixels[i * 4] != 0 or pixels[
i * 4 + 1] != 0 or pixels[i * 4 + 2] != 0
has_alpha = has_alpha or pixels[i * 4 + 3] != 0
if has_pixels and has_alpha:
break
if has_pixels and not has_alpha:
#generate missing alpha - don't ask me why
for i in range(len(pixels) // 4):
if pixels[i * 4] != 0 or pixels[i * 4 +
1] != 0 or pixels[i * 4 +
2] != 0:
pixels[i * 4 + 3] = 0xff
return [0, 0, w, h, x, y, hCursor, bytes(pixels), strtobytes(name)]
except Exception as e:
cursorlog("get_cursor_data(%#x)", hCursor, exc_info=True)
cursorlog.error("Error: failed to grab cursor:")
cursorlog.error(" %s", str(e) or type(e))
return None
finally:
if old_handle:
SelectObject(memdc, old_handle)
if bitmap:
DeleteObject(bitmap)
if memdc:
DeleteDC(memdc)
if dc:
ReleaseDC(None, dc)