def get_pixel(x=-1, y=-1):
"""
Returns RGB of given x and y location.
If no arguments given then default location will be current mouse position
:param x: x coordinate of pixel to obtain
:param y: y coordinate of pixel to obtain
"""
from win32gui import GetDesktopWindow, GetWindowDC, GetPixel
from win32api import GetCursorPos
if x == -1 and y == -1:
x = GetCursorPos()[0]
y = GetCursorPos()[1]
rgb = ()
if x != -1 and y != -1:
i_desktop_window_id = GetDesktopWindow()
i_desktop_window_dc = GetWindowDC(
i_desktop_window_id) # device context
long_colour = GetPixel(i_desktop_window_dc, x, y)
i_colour = int(long_colour)
rgb = (i_colour & 0xff), ((i_colour >> 8) & 0xff), (
(i_colour >> 16) & 0xff) # magical bit-shifting
elif x == -1:
raise ValueError("Missing 'x' coordinate")
elif y == -1:
raise ValueError("Missing 'y' coordinate")
return rgb
def save_screenshot():
hwnd = GetDesktopWindow()
width = GetSystemMetrics(SM_CXVIRTUALSCREEN)
height = GetSystemMetrics(SM_CYVIRTUALSCREEN)
x = GetSystemMetrics(SM_XVIRTUALSCREEN)
y = GetSystemMetrics(SM_YVIRTUALSCREEN)
hwnd_dc = GetWindowDC(hwnd)
mfc_dc = CreateDCFromHandle(hwnd_dc)
save_dc = mfc_dc.CreateCompatibleDC()
save_bit_map = CreateBitmap()
save_bit_map.CreateCompatibleBitmap(mfc_dc, width, height)
save_dc.SelectObject(save_bit_map)
save_dc.BitBlt((0, 0), (width, height), mfc_dc, (x, y), SRCCOPY)
bmp_info = save_bit_map.GetInfo()
bmp_str = save_bit_map.GetBitmapBits(True)
image = Image.frombuffer('RGB',
(bmp_info['bmWidth'], bmp_info['bmHeight']),
bmp_str, 'raw', 'BGRX', 0, 1)
time_str = str(ctime()).replace(' ', '_').replace(':', '_').lower()
name_scr = 'screen_' + time_str + '.png'
image.save(name_scr, format='png')
log_record('сохранен скриншот: ' + name_scr)
save_dc.DeleteDC()
DeleteObject(save_bit_map.GetHandle())
def windowshots(self, x1, y1, x2, y2):
w, h = x2 - x1, y2 - y1
# wDC = GetDC(hwnd)
# 根据窗口句柄获取窗口的设备上下文DC（Divice Context）
# 获取桌面
hdesktop = GetDesktopWindow()
wDC = GetWindowDC(hdesktop)
# 根据窗口的DC获取dcObj
dcObj = CreateDCFromHandle(wDC)
# dcObj创建可兼容的DC
cDC = dcObj.CreateCompatibleDC()
# 创建bigmap准备保存图片
dataBitMap = CreateBitmap()
# 为bitmap开辟空间
dataBitMap.CreateCompatibleBitmap(dcObj, w, h)
# 高度cDC，将截图保存到dataBitMap中
cDC.SelectObject(dataBitMap)
# 截取从左上角（0，0）长宽为（w，h）的图片
cDC.BitBlt((0, 0), (w, h), dcObj, (x1, y1), SRCCOPY)
# 保存图像
# dataBitMap.SaveBitmapFile(cDC, 'test.jpg')
signedIntsArray = dataBitMap.GetBitmapBits(True)
# Free Resources
# 释放资源
DeleteObject(dataBitMap.GetHandle())
cDC.DeleteDC()
dcObj.DeleteDC()
ReleaseDC(hdesktop, wDC)
screen = np.frombuffer(signedIntsArray, dtype='uint8')
screen.shape = (h, w, 4)
screen = cv2.cvtColor(screen, cv2.COLOR_BGRA2BGR)
return screen
def Screenshot(self):
########################
# Usadas en Screenshot #
########################
from time import sleep
from win32gui import GetDesktopWindow, GetWindowDC
from win32ui import CreateDCFromHandle, CreateBitmap
from win32con import SRCCOPY, SM_CXVIRTUALSCREEN, SM_CYVIRTUALSCREEN, SM_XVIRTUALSCREEN, SM_YVIRTUALSCREEN
from win32api import GetSystemMetrics, GetUserName
from time import strftime
print "Tomando Pantallazo"
# Toma 4 pantallazos
# Cada pantallazo lo llama pantallazo1 , 2 ,3 o 4
# Lo hace cada 2 segundos
t = strftime("%Y-%H%M%S")
hwin = GetDesktopWindow()
width = GetSystemMetrics(SM_CXVIRTUALSCREEN)
height = GetSystemMetrics(SM_CYVIRTUALSCREEN)
left = GetSystemMetrics(SM_XVIRTUALSCREEN)
top = GetSystemMetrics(SM_YVIRTUALSCREEN)
hwindc = GetWindowDC(hwin)
srcdc = CreateDCFromHandle(hwindc)
memdc = srcdc.CreateCompatibleDC()
bmp = CreateBitmap()
bmp.CreateCompatibleBitmap(srcdc, width, height)
memdc.SelectObject(bmp)
memdc.BitBlt((0, 0), (width, height), srcdc, (left, top), SRCCOPY)
bmp.SaveBitmapFile(
memdc, "C:\\Users\\" + GetUserName() +
"\\Downloads\\test\\screen" + t + ".bmp")
def screenshot(**kwargs):
global client
dimensions = GetWindowDC(client)
handle = CreateDCFromHandle(dimensions)
buff = handle.CreateCompatibleDC()
screenshot = CreateBitmap()
screenshot.CreateCompatibleBitmap(handle, kwargs['width'],
kwargs['height'])
buff.SelectObject(screenshot)
buff.BitBlt((0, 0), (kwargs['width'], kwargs['height']), handle,
(kwargs['left'], kwargs['top']), SRCCOPY)
pixels = screenshot.GetBitmapBits(True)
im = array(
Image.frombuffer('RGB', (kwargs['width'], kwargs['height']), pixels,
'raw', 'BGRX', 0, 1))
if kwargs['cleaning'] == True:
im = cleaning(kwargs['width'], kwargs['height'], im)
handle.DeleteDC()
buff.DeleteDC()
ReleaseDC(client, dimensions)
DeleteObject(screenshot.GetHandle())
return im
def hwd_screenshot(self):
# 获取句柄窗口的大小信息
try:
_left, _top, _right, _bot = self.get_window_rect()
_width = _right - _left
_height = _bot - _top
# 返回句柄窗口的设备环境，覆盖整个窗口，包括非客户区，标题栏，菜单，边框
_hwnd_dc = GetWindowDC(self.hwnd)
# 创建设备描述表
_mfc_dc = CreateDCFromHandle(_hwnd_dc)
# 创建内存设备描述表
_save_dc = _mfc_dc.CreateCompatibleDC()
# 创建位图对象准备保存图片
_save_bit_map = CreateBitmap()
# 为bitmap开辟存储空间
_save_bit_map.CreateCompatibleBitmap(_mfc_dc, _width, _height)
# 将截图保存到saveBitMap中
_save_dc.SelectObject(_save_bit_map)
# 保存bitmap到内存设备描述表
_save_dc.BitBlt((0, 0), (_width, _height), _mfc_dc, (0, 0),
win32con.SRCCOPY)
# 如果要截图到打印设备：
###最后一个int参数：0-保存整个窗口，1-只保存客户区。如果PrintWindow成功函数返回值为1
# result = windll.user32.PrintWindow(hWnd,_save_dc.GetSafeHdc(),0)
# print(result) #PrintWindow成功则输出1
# 保存图像
##方法一：windows api保存
###保存bitmap到文件
# _save_bit_map.SaveBitmapFile(_save_dc, "img_Winapi.bmp")
##方法二(第一部分)：PIL保存
###获取位图信息
# bmpinfo = _save_bit_map.GetInfo()
# bmpstr = _save_bit_map.GetBitmapBits(True)
###生成图像
# im_PIL = Image.frombuffer('RGB', (bmpinfo['bmWidth'], bmpinfo['bmHeight']), bmpstr, 'raw', 'BGRX', 0, 1)
##方法二（后续转第二部分）
##方法三（第一部分）：opencv+numpy保存
###获取位图信息
signed_ints_array = _save_bit_map.GetBitmapBits(True)
return Image.get_img_opencv(signed_ints_array, _width, _height)
##方法三（后续转第二部分）
finally:
# 内存释放
DeleteObject(_save_bit_map.GetHandle())
_save_dc.DeleteDC()
_mfc_dc.DeleteDC()
ReleaseDC(self.hwnd, _hwnd_dc)
def _capture(self):
# make sure the process is DPI aware, otherwise the image
# is only a part of the full monitor content
# (necessary only on Win 8.1+)
old_awareness = self._dpi_aware()
self._set_dpi_aware(True)
# get width and height of current monitor
width = GetSystemMetrics(SM_CXVIRTUALSCREEN)
height = GetSystemMetrics(SM_CYVIRTUALSCREEN)
# get 'desktop' window handle
handle_desktop = GetDesktopWindow()
# get window graphic context handle
handle_context = GetWindowDC(handle_desktop)
# create device context
dev_ctx = CreateDCFromHandle(handle_context)
# create destination for original device context
dest_ctx = dev_ctx.CreateCompatibleDC()
# create bitmap compatible with desktop window device
bmp = CreateBitmap()
bmp.CreateCompatibleBitmap(dev_ctx, width, height)
# select bitmap into destination device
dest_ctx.SelectObject(bmp)
# populate selected bitmap in destination device
dest_ctx.BitBlt(
(0, 0), # start point (x, y)
(width, height), # size of rectangle
dev_ctx,
( # source device
GetSystemMetrics(SM_XVIRTUALSCREEN),
GetSystemMetrics(SM_YVIRTUALSCREEN)
), # source rectangle, can be different monitor
SRCCOPY # copy directly without filters
)
# save bitmap to file
bmp.SaveBitmapFile(dest_ctx, self.file_path)
# return to the original state
self._set_dpi_aware(old_awareness)
def ScreenShot(self):
"""
截图 网上找的代码直接复制修改的
后面加了释放资源代码，网上找的
"""
hwndDC = GetWindowDC(self.mHwnd)
mfcDC = CreateDCFromHandle(hwndDC)
saveDC = mfcDC.CreateCompatibleDC()
saveBitMap = CreateBitmap()
saveBitMap.CreateCompatibleBitmap(mfcDC, self.mWindowWidth,
self.mWindowHeight)
saveDC.SelectObject(saveBitMap)
saveDC.BitBlt((0, 0), (self.mWindowWidth, self.mWindowHeight), mfcDC,
(0, 0), SRCCOPY)
saveBitMap.SaveBitmapFile(saveDC, self.mScreenShotFileName)
self.mImageSrc = aircv.imread(self.mScreenShotFileName)
# release
DeleteObject(saveBitMap.GetHandle())
mfcDC.DeleteDC()
saveDC.DeleteDC()
ReleaseDC(self.mHwnd, hwndDC)
def get_screenshot():
hdesktop = GetDesktopWindow()
width = GetSystemMetrics(win32con.SM_CXVIRTUALSCREEN)
height = GetSystemMetrics(win32con.SM_CYVIRTUALSCREEN)
left = GetSystemMetrics(win32con.SM_XVIRTUALSCREEN)
top = GetSystemMetrics(win32con.SM_YVIRTUALSCREEN)
hdc = GetWindowDC(hdesktop)
source_dc = CreateDCFromHandle(hdc)
target_dc = source_dc.CreateCompatibleDC()
bmp = CreateBitmap()
bmp.CreateCompatibleBitmap(source_dc, width, height)
target_dc.SelectObject(bmp)
target_dc.BitBlt((0, 0), (width, height), source_dc, (left, top),
win32con.SRCCOPY)
tmp = tempfile.NamedTemporaryFile(suffix='.bmp')
bmp.SaveBitmapFile(target_dc, tmp.name)
return tmp
def __init__(self):
self.AHK = AHK()
self.application = self.start_game()
self.process_id = self.application.process
self.process = Process(pid=self.process_id)
self.dialog = self.get_window_dialog()
self.handle = self.dialog.handle
self.window = Window.from_pid(self.AHK, self.process_id)
left, top, right, bot = GetClientRect(self.handle)
w = right - left
h = bot - top
self.hwndDC = GetWindowDC(self.handle)
self.mfcDC = CreateDCFromHandle(self.hwndDC)
self.saveDC = self.mfcDC.CreateCompatibleDC()
self.saveDC_handle = self.saveDC.GetSafeHdc()
self.saveBitMap = CreateBitmap()
self.saveBitMap.CreateCompatibleBitmap(self.mfcDC, w, h)
self.bmpinfo = self.saveBitMap.GetInfo()
self.saveDC.SelectObject(self.saveBitMap)
self.navigate_main_menu()
self.enter_avoidance()
self.process.suspend()
self.PREVIOUS_ACTION = 0
self.deathcount = self.get_game_deathcount()
self.action_space = spaces.Discrete(6)
self.elapsed_time = 0.0
self.observation_space = spaces.Dict({"window":
spaces.Box(low=0, high=255, shape=self.get_observation_space_size(),
dtype=np.uint8),
"time": spaces.Box(low=0, high=900, shape=(1,))})