You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
o3de/Tools/LyTestTools/ly_test_tools/image/image_capture.py

39 lines
1.2 KiB
Python

"""
Copyright (c) Contributors to the Open 3D Engine Project.
For complete copyright and license terms please see the LICENSE at the root of this distribution.
SPDX-License-Identifier: Apache-2.0 OR MIT
Utility package for LyTestTools.
Provides image capturing functionality.
"""
import pyscreenshot
def screencap(x1=None, y1=None, x2=None, y2=None, filename='screenshot.png'):
"""
Capture an arbitrary portion (in absolute coordinates) of the screen to a file. Note currently accepts coords
for the leftmost screen only. Any span beyond/below will be blank.
If any of the coordinate parameters is None, the entire screen will be captured.
Note this is a brittle way to verify a portion of an application.
:param x1: Top left X
:param y1: Top left Y
:param x2: Bottom right X
:param y2: Bottom right Y
:param filename: Filename to save to.
:return: Image that was saved in PIL "Image" class.
"""
if x1 is not None and x2 is not None and y1 is not None and y2 is not None:
im = pyscreenshot.grab(bbox=(x1, y1, x2, y2), childprocess=False)
else:
im = pyscreenshot.grab(childprocess=False)
im.save(filename)
return im