def prep_menu_icon(self,
icon): #Couldn't get this to work with pngs, only ico
# First load the icon.
ico_x = win32api.GetSystemMetrics(win32con.SM_CXSMICON)
ico_y = win32api.GetSystemMetrics(win32con.SM_CYSMICON)
hicon = win32gui.LoadImage(0, icon, win32con.IMAGE_ICON, ico_x, ico_y,
win32con.LR_LOADFROMFILE)
hdcBitmap = win32gui.CreateCompatibleDC(0)
hdcScreen = win32gui.GetDC(0)
hbm = win32gui.CreateCompatibleBitmap(hdcScreen, ico_x, ico_y)
hbmOld = win32gui.SelectObject(hdcBitmap, hbm)
# Fill the background.
brush = win32gui.GetSysColorBrush(win32con.COLOR_MENU)
win32gui.FillRect(hdcBitmap, (0, 0, 16, 16), brush)
# unclear if brush needs to be feed. Best clue I can find is:
# "GetSysColorBrush returns a cached brush instead of allocating a new
# one." - implies no DeleteObject
# draw the icon
win32gui.DrawIconEx(hdcBitmap, 0, 0, hicon, ico_x, ico_y, 0, 0,
win32con.DI_NORMAL)
win32gui.SelectObject(hdcBitmap, hbmOld)
win32gui.DeleteDC(hdcBitmap)
return hbm
def _set_icon_menu(self, icon):
"""Load icons into the tray items.
Got from https://stackoverflow.com/a/45890829.
"""
ico_x = win32api.GetSystemMetrics(win32con.SM_CXSMICON)
ico_y = win32api.GetSystemMetrics(win32con.SM_CYSMICON)
hIcon = win32gui.LoadImage(0, icon, win32con.IMAGE_ICON, ico_x, ico_y,
win32con.LR_LOADFROMFILE)
hwndDC = win32gui.GetWindowDC(self.hwnd)
dc = win32ui.CreateDCFromHandle(hwndDC)
memDC = dc.CreateCompatibleDC()
iconBitmap = win32ui.CreateBitmap()
iconBitmap.CreateCompatibleBitmap(dc, ico_x, ico_y)
oldBmp = memDC.SelectObject(iconBitmap)
brush = win32gui.GetSysColorBrush(win32con.COLOR_MENU)
win32gui.FillRect(memDC.GetSafeHdc(), (0, 0, ico_x, ico_y), brush)
win32gui.DrawIconEx(memDC.GetSafeHdc(), 0, 0, hIcon, ico_x, ico_y, 0,
0, win32con.DI_NORMAL)
memDC.SelectObject(oldBmp)
memDC.DeleteDC()
win32gui.ReleaseDC(self.hwnd, hwndDC)
self.logger.debug('Set menu icon.')
return iconBitmap.GetHandle()
def IconToFile(hlib, tmpDirName, group_name):
libc = ctypes.CDLL(ctypes.util.find_library('c'))
libc.memcpy.argtypes = [ctypes.c_void_p, ctypes.c_void_p, ctypes.c_size_t]
libc.memcpy.restype = ctypes.c_char_p
# patch FindResourceW, ctypes.windll.kernel32.SizeofResource
FindResourceW = ctypes.windll.kernel32.FindResourceW
FindResourceW.argtypes = [
ctypes.c_void_p, ctypes.c_void_p, ctypes.c_void_p
]
FindResourceW.restype = ctypes.c_void_p
SizeofResource = ctypes.windll.kernel32.SizeofResource
SizeofResource.argtypes = [ctypes.c_void_p, ctypes.c_void_p]
SizeofResource.restype = ctypes.c_size_t
hRes = win32api.LoadResource(hlib, win32con.RT_GROUP_ICON, group_name)
mem_icon_dir = ctypes.windll.kernel32.LockResource(hRes)
# 32 bits color; 16 and 256 colors are too old
icon_size = 256
icon_name = ctypes.windll.user32.LookupIconIdFromDirectoryEx(
mem_icon_dir, True, icon_size, icon_size, 0x00000000)
sys.stderr.write("icon_name=%s\n" % str(icon_name))
hResInfo = FindResourceW(hlib, icon_name, win32con.RT_ICON)
size = ctypes.windll.kernel32.SizeofResource(hlib, hResInfo)
rec = win32api.LoadResource(hlib, win32con.RT_ICON, icon_name)
mem_icon = ctypes.windll.kernel32.LockResource(rec)
# And this is some differ (copy data to Python buffer)
binary_data = (ctypes.c_ubyte * size)()
libc.memcpy(binary_data, mem_icon, size)
hIconRet = ctypes.windll.user32.CreateIconFromResourceEx(
binary_data, size, True, 0x00030000, 0, 0, 0x00000000)
info = win32gui.GetIconInfo(hIconRet)
bminfo = win32gui.GetObject(info[4])
# generate bitmap by drawing the icon
hdc = win32ui.CreateDCFromHandle(win32gui.GetDC(0))
hbmp = win32ui.CreateBitmap()
hbmp.CreateCompatibleBitmap(hdc, bminfo.bmWidth, bminfo.bmHeight)
hcdc = hdc.CreateCompatibleDC()
hcdc.SelectObject(hbmp)
win32gui.DrawIconEx(hcdc.GetHandleOutput(), 0, 0, hIconRet, bminfo.bmWidth,
bminfo.bmHeight, 0, 0, 0x0003)
# Need temp directory or generate image on the fly, for inclusion into an HTML page.
# Temp file visible from HTTP server.
# An alternative is to send the file content to a socket stream.
# MIME type is "image/bmp"
imgFilNam = "icon-%03dx%03d-%05d-%03d.bmp" % (
bminfo.bmWidth, bminfo.bmHeight, group_name, icon_name)
if tmpDirName:
imgFilNam = tmpDirName + imgFilNam
sys.stderr.write("Generating %s\n" % imgFilNam)
hbmp.SaveBitmapFile(hcdc, imgFilNam)
win32gui.DestroyIcon(hIconRet)
return imgFilNam
def IconToFile(hlib, group_name):
"""The group might be a string or an integer and its type must be kept."""
# patch FindResourceW, ctypes.windll.kernel32.SizeofResource
FindResourceW = ctypes.windll.kernel32.FindResourceW
FindResourceW.argtypes = [
ctypes.c_void_p, ctypes.c_void_p, ctypes.c_void_p
]
FindResourceW.restype = ctypes.c_void_p
SizeofResource = ctypes.windll.kernel32.SizeofResource
SizeofResource.argtypes = [ctypes.c_void_p, ctypes.c_void_p]
SizeofResource.restype = ctypes.c_size_t
hRes = win32api.LoadResource(hlib, win32con.RT_GROUP_ICON, group_name)
mem_icon_dir = ctypes.windll.kernel32.LockResource(hRes)
# 32 bits color; 16 and 256 colors are too old
icon_size = 256
icon_name = ctypes.windll.user32.LookupIconIdFromDirectoryEx(
mem_icon_dir, True, icon_size, icon_size, 0x00000000)
h_res_info = FindResourceW(hlib, icon_name, win32con.RT_ICON)
size = ctypes.windll.kernel32.SizeofResource(hlib, h_res_info)
rec = win32api.LoadResource(hlib, win32con.RT_ICON, icon_name)
mem_icon = ctypes.windll.kernel32.LockResource(rec)
# And this is some differ (copy data to Python buffer)
binary_data = (ctypes.c_ubyte * size)()
ctypes.memmove(binary_data, mem_icon, size)
h_icon_ret = ctypes.windll.user32.CreateIconFromResourceEx(
binary_data, size, True, 0x00030000, 0, 0, 0x00000000)
info = win32gui.GetIconInfo(h_icon_ret)
bminfo = win32gui.GetObject(info[4])
# generate bitmap by drawing the icon
hdc = win32ui.CreateDCFromHandle(win32gui.GetDC(0))
hbmp = win32ui.CreateBitmap()
hbmp.CreateCompatibleBitmap(hdc, bminfo.bmWidth, bminfo.bmHeight)
hcdc = hdc.CreateCompatibleDC()
hcdc.SelectObject(hbmp)
win32gui.DrawIconEx(hcdc.GetHandleOutput(), 0, 0, h_icon_ret,
bminfo.bmWidth, bminfo.bmHeight, 0, 0, 0x0003)
# MIME type is "image/bmp"
# The group name might be a number: 110 etc... or a string such as 'ICO_MYCOMPUTER'.
# This is the prefix of the temporary BMP file name containing the extracted icon.
img_fil_nam_prefix = "icon-%03dx%03d-%s-%03d" % (
bminfo.bmWidth, bminfo.bmHeight, str(group_name), icon_name)
# The destructor will remove the file.
obj_temp_file = lib_util.TmpFile(img_fil_nam_prefix, "bmp")
img_fil_nam = obj_temp_file.Name
hbmp.SaveBitmapFile(hcdc, img_fil_nam)
win32gui.DestroyIcon(h_icon_ret)
return obj_temp_file
def iconToQImage(self, hIcon):
hdc = win32ui.CreateDCFromHandle(win32gui.GetDC(0))
hbmp = win32ui.CreateBitmap()
hbmp.CreateCompatibleBitmap(hdc, hIcon.width, hIcon.height)
hdc = hdc.CreateCompatibleDC()
hdc.SelectObject(hbmp)
win32gui.DrawIconEx(hdc.GetHandleOutput(), 0, 0, hIcon.hIcon,
hIcon.width, hIcon.height, 0, None, 0x0003)
bitmapbits = hbmp.GetBitmapBits(True)
image = QImage(bitmapbits, hIcon.width, hIcon.height,
QImage.Format_ARGB32_Premultiplied)
return image
def prep(self): ico_x = win32api.GetSystemMetrics(win32con.SM_CXSMICON) ico_y = win32api.GetSystemMetrics(win32con.SM_CYSMICON) lr, sm = win32gui.ExtractIconEx(self.executable,0) win32gui.DestroyIcon(lr[0]) hicon = sm[0] hdcBitmap = win32gui.CreateCompatibleDC(0) hdcScreen = win32gui.GetDC(0) hbm = win32gui.CreateCompatibleBitmap(hdcScreen, ico_x, ico_y) hbmOld = win32gui.SelectObject(hdcBitmap, hbm) brush = win32gui.GetSysColorBrush(win32con.COLOR_MENU) win32gui.FillRect(hdcBitmap, (0, 0, 16, 16), brush) win32gui.DrawIconEx(hdcBitmap, 0, 0, hicon, ico_x, ico_y, 0, 0, win32con.DI_NORMAL) win32gui.SelectObject(hdcBitmap, hbmOld) win32gui.DeleteDC(hdcBitmap) return hbm
def prep(self):
ico_x = win32api.GetSystemMetrics(win32con.SM_CXSMICON)
ico_y = win32api.GetSystemMetrics(win32con.SM_CYSMICON)
hicon = win32gui.LoadImage(0, self.icon, win32con.IMAGE_ICON, \
ico_x, ico_y, win32con.LR_LOADFROMFILE)
hdcBitmap = win32gui.CreateCompatibleDC(0)
hdcScreen = win32gui.GetDC(0)
hbm = win32gui.CreateCompatibleBitmap(hdcScreen, ico_x, ico_y)
hbmOld = win32gui.SelectObject(hdcBitmap, hbm)
brush = win32gui.GetSysColorBrush(win32con.COLOR_MENU)
win32gui.FillRect(hdcBitmap, (0, 0, 16, 16), brush)
win32gui.DrawIconEx(hdcBitmap, 0, 0, hicon, ico_x, ico_y, 0, 0, \
win32con.DI_NORMAL)
win32gui.SelectObject(hdcBitmap, hbmOld)
win32gui.DeleteDC(hdcBitmap)
return hbm
def prep_menu_icon(self, icon):
# first load the icon
ico_x = win32api.GetSystemMetrics(win32con.SM_CXSMICON)
ico_y = win32api.GetSystemMetrics(win32con.SM_CYSMICON)
hicon = win32gui.LoadImage(0, icon, win32con.IMAGE_ICON, ico_x, ico_y, win32con.LR_LOADFROMFILE)
hdcBitmap = win32gui.CreateCompatibleDC(None)
hdcScreen = win32gui.GetDC(None)
hbm = win32gui.CreateCompatibleBitmap(hdcScreen, ico_x, ico_y)
hbmOld = win32gui.SelectObject(hdcBitmap, hbm)
# fill the background
brush = win32gui.GetSysColorBrush(win32con.COLOR_MENU)
win32gui.FillRect(hdcBitmap, (0, 0, 16, 16), brush)
# draw the icon
win32gui.DrawIconEx(hdcBitmap, 0, 0, hicon, ico_x, ico_y, 0, 0, win32con.DI_NORMAL)
win32gui.SelectObject(hdcBitmap, hbmOld)
# no need to free the brush
win32gui.DeleteDC(hdcBitmap)
return hbm
def prep_menu_icon(self, icon):
# First load the icon.
ico_x = win32api.GetSystemMetrics(win32con.SM_CXSMICON)
ico_y = win32api.GetSystemMetrics(win32con.SM_CYSMICON)
hIcon = win32gui.LoadImage(0, icon, win32con.IMAGE_ICON, ico_x, ico_y,
win32con.LR_LOADFROMFILE)
hwndDC = win32gui.GetWindowDC(self.hwnd)
dc = win32ui.CreateDCFromHandle(hwndDC)
memDC = dc.CreateCompatibleDC()
iconBitmap = win32ui.CreateBitmap()
iconBitmap.CreateCompatibleBitmap(dc, ico_x, ico_y)
oldBmp = memDC.SelectObject(iconBitmap)
brush = win32gui.GetSysColorBrush(win32con.COLOR_MENU)
win32gui.FillRect(memDC.GetSafeHdc(), (0, 0, ico_x, ico_y), brush)
win32gui.DrawIconEx(memDC.GetSafeHdc(), 0, 0, hIcon, ico_x, ico_y, 0,
0, win32con.DI_NORMAL)
memDC.SelectObject(oldBmp)
memDC.DeleteDC()
win32gui.ReleaseDC(self.hwnd, hwndDC)
return iconBitmap.GetHandle()
def prep_menu_icon(icon):
# 首先加载图标
ico_x = win32api.GetSystemMetrics(win32con.SM_CXSMICON)
ico_y = win32api.GetSystemMetrics(win32con.SM_CYSMICON)
hicon = win32gui.LoadImage(0, icon, win32con.IMAGE_ICON, ico_x, ico_y, win32con.LR_LOADFROMFILE)
hdc_bitmap = win32gui.CreateCompatibleDC(0)
hdc_screen = win32gui.GetDC(0)
hbm = win32gui.CreateCompatibleBitmap(hdc_screen, ico_x, ico_y)
hbm_old = win32gui.SelectObject(hdc_bitmap, hbm)
# 填满背景
brush = win32gui.GetSysColorBrush(win32con.COLOR_MENU)
win32gui.FillRect(hdc_bitmap, (0, 0, 16, 16), brush)
# "GetSysColorBrush返回缓存的画笔而不是分配新的画笔"
# - 暗示没有DeleteObject
# 画出图标
win32gui.DrawIconEx(hdc_bitmap, 0, 0, hicon, ico_x, ico_y, 0, 0, win32con.DI_NORMAL)
win32gui.SelectObject(hdc_bitmap, hbm_old)
win32gui.DeleteDC(hdc_bitmap)
return hbm
def prep_menu_icon(self, icon):
if DEBUG: print(sys._getframe(0).f_code.co_name)
# 首先加载图标。
ico_x = win32api.GetSystemMetrics(win32con.SM_CXSMICON)
ico_y = win32api.GetSystemMetrics(win32con.SM_CYSMICON)
hicon = win32gui.LoadImage(0, icon, win32con.IMAGE_ICON, ico_x, ico_y, win32con.LR_LOADFROMFILE)
hdcBitmap = win32gui.CreateCompatibleDC(0)
hdcScreen = win32gui.GetDC(0)
hbm = win32gui.CreateCompatibleBitmap(hdcScreen, ico_x, ico_y)
hbmOld = win32gui.SelectObject(hdcBitmap, hbm)
# 填满背景。
brush = win32gui.GetSysColorBrush(win32con.COLOR_MENU)
win32gui.FillRect(hdcBitmap, (0, 0, 16, 16), brush)
# "GetSysColorBrush返回缓存的画笔而不是分配新的画笔。"
# - 暗示没有DeleteObject
# 画出图标
win32gui.DrawIconEx(hdcBitmap, 0, 0, hicon, ico_x, ico_y, 0, 0, win32con.DI_NORMAL)
win32gui.SelectObject(hdcBitmap, hbmOld)
win32gui.DeleteDC(hdcBitmap)
return hbm
def wnd_proc(h_wnd, message, w_param, l_param):
"""Displays a transparent window with some graphic elements
Displays a transparent window with some graphic elements
:param h_wnd: an input argument
:returns: nothing
"""
if message == win32con.WM_QUIT:
# print("Closing the window overlay")
win32gui.PostQuitMessage(0)
return 0
if message == win32con.WM_PAINT:
hdc, paint_struct = win32gui.BeginPaint(h_wnd)
win32gui.SetGraphicsMode(hdc, win32con.GM_ADVANCED)
win32gui.BringWindowToTop(h_wnd)
for r in self.graphical_elements:
if 'geometry' in r:
geometry = r['geometry']
else:
geometry = None
if 'x' in r:
x = r['x'] - self.x_min
else:
x = 0
if 'y' in r:
y = r['y'] - self.y_min
else:
y = 0
if 'width' in r:
width = r['width']
else:
width = 100
if 'height' in r:
height = r['height']
else:
height = 100
if 'xyrgb_array' in r:
xyrgb_array = r['xyrgb_array']
else:
xyrgb_array = ((15, 15, 255, 0, 0), (15, 45, 0, 255,
0), (45, 30, 0, 0,
255))
if 'color' in r:
color_r = r['color'][0]
color_g = r['color'][1]
color_b = r['color'][2]
else:
color_r = 255
color_g = 0
color_b = 0
if 'thickness' in r:
thickness = r['thickness']
else:
thickness = 0
if 'angle' in r:
angle = r['angle']
else:
angle = 0
if 'geometry' in r and r['geometry'] is Shape.triangle:
vertices = ()
for xyrgb in xyrgb_array:
vertices = vertices + (
{
'x': int(round(xyrgb[0])) - self.x_min,
'y': int(round(xyrgb[1])) - self.y_min,
'Red': xyrgb[2] * 256,
'Green': xyrgb[3] * 256,
'Blue': xyrgb[4] * 256,
'Alpha': 0
}, )
mesh = ((0, 1, 2), )
win32gui.GradientFill(hdc, vertices, mesh,
win32con.GRADIENT_FILL_TRIANGLE)
if 'brush_color' in r:
brush_color_r = r['brush_color'][0]
brush_color_g = r['brush_color'][1]
brush_color_b = r['brush_color'][2]
else:
brush_color_r = 255
brush_color_g = 255
brush_color_b = 255
if thickness == 0:
color_r = brush_color_r
color_g = brush_color_g
color_b = brush_color_b
if 'font_size' in r:
font_size = r['font_size']
else:
font_size = 18
if 'font_name' in r:
font_name = r['font_name']
else:
font_name = "Arial"
my_brush = None
if 'brush' in r and width > 1 and height > 1:
brush = r['brush']
brush_color = win32api.RGB(brush_color_r,
brush_color_g,
brush_color_b)
if brush is Brush.solid:
my_brush = win32gui.CreateSolidBrush(brush_color)
elif brush is Brush.b_diagonal:
my_brush = win32gui.CreateHatchBrush(
win32con.HS_BDIAGONAL, brush_color)
elif brush is Brush.cross:
my_brush = win32gui.CreateHatchBrush(
win32con.HS_CROSS, brush_color)
elif brush is Brush.diag_cross:
my_brush = win32gui.CreateHatchBrush(
win32con.HS_DIAGCROSS, brush_color)
elif brush is Brush.f_diagonal:
my_brush = win32gui.CreateHatchBrush(
win32con.HS_FDIAGONAL, brush_color)
elif brush is Brush.horizontal:
my_brush = win32gui.CreateHatchBrush(
win32con.HS_HORIZONTAL, brush_color)
elif brush is Brush.vertical:
my_brush = win32gui.CreateHatchBrush(
win32con.HS_VERTICAL, brush_color)
old_brush = win32gui.SelectObject(hdc, my_brush)
pen = win32gui.CreatePen(
win32con.PS_GEOMETRIC, thickness,
win32api.RGB(color_r, color_g, color_b))
old_pen = win32gui.SelectObject(hdc, pen)
if 'center_of_rotation' in r:
center_of_rotation_x = r['center_of_rotation'][0]
center_of_rotation_y = r['center_of_rotation'][1]
else:
center_of_rotation_x = 0
center_of_rotation_y = 0
if angle != 0:
r_angle = angle * (math.pi / 180)
Py_XFORM = win32gui.GetWorldTransform(hdc)
win32gui.SetWorldTransform(
hdc, {
'M11': math.cos(r_angle),
'M12': math.sin(r_angle),
'M21': math.sin(r_angle) * -1,
'M22': math.cos(r_angle),
'Dx': x,
'Dy': y
})
x, y = -center_of_rotation_x, -center_of_rotation_y
if 'text_format' in r:
text_format = eval(r['text_format'])
else:
text_format = win32con.DT_CENTER | win32con.DT_SINGLELINE | win32con.DT_VCENTER
if 'geometry' in r:
if r['geometry'] is Shape.rectangle:
win32gui.Rectangle(hdc, int(round(x)),
int(round(y)),
int(round(x + width)),
int(round(y + height)))
elif r['geometry'] is Shape.ellipse:
win32gui.Ellipse(hdc, int(round(x)), int(round(y)),
int(round(x + width)),
int(round(y + height)))
elif r['geometry'] is Shape.arrow:
a = thickness
t = ((x - int(a * 1.4), y), (x - a * 4, y + a * 3),
(x, y), (x - a * 4, y - a * 3),
(x - int(a * 1.4), y), (x - a * 9, y))
win32gui.Polyline(hdc, t)
elif r['geometry'] is Shape.image:
hicon = r['hicon']
win32gui.DrawIconEx(hdc, x, y, hicon, 0, 0, 0,
None, 0x0003)
elif r['geometry'] is Shape.triangle and thickness > 0:
t = ()
for xyrgb in xyrgb_array:
t = t + ((int(round(
xyrgb[0])), +int(round(xyrgb[1]))), )
t = t + ((int(round(xyrgb_array[0][0])),
int(round(xyrgb_array[0][1]))), )
win32gui.Polyline(hdc, t)
if angle != 0:
win32gui.SetWorldTransform(hdc, Py_XFORM)
win32gui.SelectObject(hdc, old_pen)
if 'brush' in r and width > 1 and height > 1:
win32gui.SelectObject(hdc, old_brush)
if 'text' in r:
text = r['text']
lf = win32gui.LOGFONT()
lf.lfFaceName = font_name
lf.lfHeight = font_size
lf.lfWeight = win32con.FW_NORMAL
lf.lfQuality = win32con.ANTIALIASED_QUALITY
hf = win32gui.CreateFontIndirect(lf)
old_font = win32gui.SelectObject(hdc, hf)
if 'text_color' in r:
text_color_r = r['text_color'][0]
text_color_g = r['text_color'][1]
text_color_b = r['text_color'][2]
else:
text_color_r = 0
text_color_g = 0
text_color_b = 0
win32gui.SetTextColor(
hdc,
win32api.RGB(text_color_r, text_color_g,
text_color_b))
if 'text_bg_color' in r:
text_bg_color_r = r['text_bg_color'][0]
text_bg_color_g = r['text_bg_color'][1]
text_bg_color_b = r['text_bg_color'][2]
else:
text_bg_color_r = brush_color_r
text_bg_color_g = brush_color_g
text_bg_color_b = brush_color_b
win32gui.SetBkMode(hdc, win32con.TRANSPARENT)
win32gui.SetBkColor(
hdc,
win32api.RGB(text_bg_color_r, text_bg_color_g,
text_bg_color_b))
tuple_r = tuple([
int(round(x)),
int(round(y)),
int(round(x + width)),
int(round(y + height))
])
win32gui.DrawTextW(hdc, text, -1, tuple_r, text_format)
win32gui.SelectObject(hdc, old_font)
win32gui.EndPaint(h_wnd, paint_struct)
return 0
else:
return win32gui.DefWindowProc(h_wnd, message, w_param, l_param)
