Removed need for OpenCV and numpy.

This commit is contained in:
PySimpleGUI 2022-03-05 06:17:51 -05:00
parent 5b3c3e5375
commit 223cd18964
2 changed files with 16 additions and 12 deletions

View File

@ -1,8 +1,6 @@
import PySimpleGUI as sg
import win32gui
import cv2
from PIL import ImageGrab
import numpy as np
"""
Demo - Save Window screenshot
@ -10,7 +8,7 @@ import numpy as np
Saves a window as an image file. Tested saving as PNG and JPG. Input the title of the Window and it will be
saved in the format indicated by the filename.
Copyright 2020 PySimpleGUI.org
Copyright 2020, 2022 PySimpleGUI.org
"""
# --------------------------------- Function to Save Window as JPG ---------------------------------
@ -41,9 +39,8 @@ def save_win(filename=None, title=None):
fceuxHWND = win32gui.FindWindow(None, title)
rect = win32gui.GetWindowRect(fceuxHWND)
rect_cropped = (rect[0]+C, rect[1], rect[2]-C, rect[3]-C)
frame = np.array(ImageGrab.grab(bbox=rect_cropped), dtype=np.uint8)
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
cv2.imwrite(filename, frame)
grab = ImageGrab.grab(bbox=rect_cropped)
grab.save(filename)
sg.popup('Wrote image to file:',filename)
except Exception as e:
sg.popup('Error trying to save screenshot file', e)

View File

@ -6,10 +6,19 @@ import win32api
import win32con
import win32gui
import win32process
import cv2
from PIL import ImageGrab
import numpy as np
"""
Demo - Save Windows as Images
Works on WINDOWS only.
Saves a window as an image file. Tested saving as PNG and JPG.
saved in the format indicated by the filename.
The window needs to be on the primary display.
2022 update was to remove OpenCV requirement.
Copyright 2020, 2022 PySimpleGUI.org
"""
def convert_string_to_tuple(string):
"""
@ -67,7 +76,6 @@ def get_window_list():
titles = sorted(titles, key=lambda x: x[0].lower())
return titles
def save_win(filename=None, title=None, crop=True):
"""
Saves a window with the title provided as a file using the provided filename.
@ -82,9 +90,8 @@ def save_win(filename=None, title=None, crop=True):
fceuxHWND = win32gui.FindWindow(None, title)
rect = win32gui.GetWindowRect(fceuxHWND)
rect_cropped = (rect[0] + C, rect[1], rect[2] - C, rect[3] - C)
frame = np.array(ImageGrab.grab(bbox=rect_cropped), dtype=np.uint8)
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
cv2.imwrite(filename, frame)
grab = ImageGrab.grab(bbox=rect_cropped)
grab.save(filename)
sg.cprint('Wrote image to file:')
sg.cprint(filename, c='white on purple')
except Exception as e: