Edit File: smartdisplay.cpython-311.pyc
� m�dg � � � d dl Z d dlZd dlmZmZ d dlmZ d dlmZ e j e � � Z G d� de � � Z d� Z G d� d e� � ZdS ) � N)�Image� ImageChops)�grab)�Displayc � � e Zd ZdS )�DisplayTimeoutErrorN)�__name__� __module__� __qualname__� � �n/builddir/build/BUILD/cloudlinux-venv-1.0.2/venv/lib/python3.11/site-packages/pyvirtualdisplay/smartdisplay.pyr r s � � � � � ��Dr r c � � | j dk r| � d� � } t j d| j |� � }t j | |� � }|� � � }|r| � |� � S dS )z�Crop borders off an image. :param im: Source image. :param bgcolor: Background color, using either a color tuple. :return: An image without borders, or None if there's no actual content in the image. �RGBN) �mode�convertr �new�sizer � difference�getbbox�crop)�im�bgcolor�bg�diff�bboxs r �autocropr ss � � �w�%��� �Z�Z�� � �� ��5�"�'�7� +� +�B� � ��R� (� (�D� �<�<�>�>�D� � ��w�w�t�}�}� � �4r c �$ � e Zd Zd� Zdd�Zdd�ZdS ) � SmartDisplayc �, � t || j � � S )z�Crop borders off an image. :param im: Source image. :return: An image without borders, or None if there's no actual content in the image. )r �_bgcolor)�selfr s r r zSmartDisplay.autocrop" s � � ��D�M�*�*�*r Tc �^ � t | j �� � }|r| � |� � }|S )N)�xdisplay)r �new_display_varr )r" r �imgs r r zSmartDisplay.grab* s3 � ��D� 0�1�1�1�� � %��-�-��$�$�C�� r �< Nc � � d}d}d} t � dt |� � z � � t j |� � ||z }| � d�� � }| � |� � }|r|}|r|snL ||� � rn@|}|dz }||k rd|�d |�d �} t | � � �t � d � � ��|sJ �|S ) a� start process and create screenshot. Repeat screenshot until it is not empty and cb_imgcheck callback function returns True for current screenshot. :param autocrop: True -> crop screenshot :param timeout: int :param cb_imgcheck: None or callback for testing img, True = accept img, False = reject img r g333333�?g �?� zsleeping %s secsF)r zTimeout! elapsed time:z timeout:� z screenshot is empty, next try..)�log�debug�str�time�sleepr r r ) r" �timeoutr � cb_imgcheck�t� sleep_time� repeat_timer& �img_crop�msgs r �waitgrabzSmartDisplay.waitgrab2 s � � ��� �� � 9� �I�I�(�3�z�?�?�:� ;� ;� ;� �J�z� "� "� "� ��O�A��)�)�U�)�+�+�C��}�}�S�)�)�H�� � ��� �"� �� �;�s�#�#� ��$�J� �3� �K��7�{�{�{�@A���7�7�7�K��)�#�.�.�.� �I�I�7� 8� 8� 8�) 9�* � � � � � r )T)r' TN)r r r r r r7 r r r r r ! sK � � � � � �+� +� +�� � � �'� '� '� '� '� 'r r )�loggingr. �PILr r � PIL.ImageGrabr �pyvirtualdisplayr � getLoggerr r+ � Exceptionr r r r r r �<module>r> s� �� ���� � � � � !� !� !� !� !� !� !� !� � � � � � � $� $� $� $� $� $��g���!�!�� � � � � �)� � � �� � �"8� 8� 8� 8� 8�7� 8� 8� 8� 8� 8r