FIXED GRABSCREEN

New template

SVFBFuncs/grabscreen.py

@@ -1,9 +1,7 @@
-# Done by Frannecklp
-
 import cv2
 import numpy as np
 import win32gui, win32ui, win32con, win32api
-
+from ctypes import windll
 
 def grab_screen(region):
 
@@ -16,10 +14,10 @@ def grab_screen(region):
 
     else:
 
-        hwin = win32gui.FindWindow(None, "Stardew Valley")
+        hwnd = win32gui.FindWindow(None, "Stardew Valley")
 
-        if hwin:
-            rect = win32gui.GetWindowRect(hwin)
+        if hwnd:
+            rect = win32gui.GetWindowRect(hwnd)
 
             # If any dimension is negative the game is minimized, so we need to get the resolution from save file
             if any(i < 0 for i in rect):
@@ -34,9 +32,30 @@ def grab_screen(region):
             else:
                 print("game running")
 
-                left, top = rect[0:2]
-                width = rect[2] - rect[0]
-                height = rect[3] - rect[1]
+                left, top, right, bot = win32gui.GetWindowRect(hwnd)
+                w = right - left
+                h = bot - top
+                hwndDC = win32gui.GetWindowDC(hwnd)
+                mfcDC = win32ui.CreateDCFromHandle(hwndDC)
+                saveDC = mfcDC.CreateCompatibleDC()
+
+                saveBitMap = win32ui.CreateBitmap()
+                saveBitMap.CreateCompatibleBitmap(mfcDC, w, h)
+
+                saveDC.SelectObject(saveBitMap)
+
+                result = windll.user32.PrintWindow(hwnd, saveDC.GetSafeHdc(), 1)
+
+                bmpinfo = saveBitMap.GetInfo()
+                bmpstr = saveBitMap.GetBitmapBits(True)
+
+                img = np.fromstring(bmpstr, dtype='uint8')
+                img.shape = (bmpinfo['bmHeight'], bmpinfo['bmWidth'], 4)
+                win32gui.DeleteObject(saveBitMap.GetHandle())
+                saveDC.DeleteDC()
+                mfcDC.DeleteDC()
+                win32gui.ReleaseDC(hwnd, hwndDC)
+                return cv2.cvtColor(img, cv2.COLOR_BGRA2GRAY)
 
         else:
             print("game not running")
@@ -46,21 +65,5 @@ def grab_screen(region):
             left = win32api.GetSystemMetrics(win32con.SM_XVIRTUALSCREEN)
             top = win32api.GetSystemMetrics(win32con.SM_YVIRTUALSCREEN)
 
-    hwindc = win32gui.GetWindowDC(hwin)
-    srcdc = win32ui.CreateDCFromHandle(hwindc)
-    memdc = srcdc.CreateCompatibleDC()
-    bmp = win32ui.CreateBitmap()
-    bmp.CreateCompatibleBitmap(srcdc, width, height)
-    memdc.SelectObject(bmp)
-    memdc.BitBlt((0, 0), (width, height), srcdc, (left, top), win32con.SRCCOPY)
-
-    signedIntsArray = bmp.GetBitmapBits(True)
-    img = np.fromstring(signedIntsArray, dtype='uint8')
-    img.shape = (height, width, 4)
 
-    srcdc.DeleteDC()
-    memdc.DeleteDC()
-    win32gui.ReleaseDC(hwin, hwindc)
-    win32gui.DeleteObject(bmp.GetHandle())
 
-    return cv2.cvtColor(img, cv2.COLOR_BGRA2GRAY)