B N[ @sdZddlmZmZmZddlZddlZddlm Z ddl m Z Gddde Z Gd d d e Z Gd d d e Ze d dgdZe eeddgdZe edS)zD PIL-based formats to take screenshots and grab from the clipboard. )absolute_importprint_functiondivisionN)formats)FormatcsJeZdZdZdZdZfddZddZdd ZGd d d e j Z Z S) BaseGrabFormatz# Base format for grab formats. FNcs tt|j||t|_dS)N)superr__init__ threadingRLock_lock)selfargskwargs) __class__3lib/python3.7/site-packages/imageio/plugins/grab.pyr szBaseGrabFormat.__init__cCsdS)NFr)rrequestrrr _can_writeszBaseGrabFormat._can_writec Csj|jX|jsZd|_ddl}t|ds.tdyddlm}Wntk rRdSX||_WdQRX|jS)NTr __version__z(Imageio Pillow requires Pillow, not PIL!) ImageGrab)r _pillow_importedPILhasattr ImportErrorr _ImageGrab)rrrrrr _init_pillows zBaseGrabFormat._init_pillowc@s$eZdZddZddZddZdS)zBaseGrabFormat.ReadercCsdS)Nr)rrrr_open-szBaseGrabFormat.Reader._opencCsdS)Nr)rrrr_close0szBaseGrabFormat.Reader._closecCs |j|S)N)format _get_data)rindexrrrr!3szBaseGrabFormat.Reader._get_dataN)__name__ __module__ __qualname__rrr!rrrrReader,sr&) r#r$r%__doc__rrr rrrr& __classcell__rr)rrrs rc@s eZdZdZddZddZdS)ScreenGrabFormata" The ScreenGrabFormat provided a means to grab screenshots using the uri of "". This functionality is provided via Pillow. Note that "" is only supported on Windows and OS X. Parameters for reading ---------------------- No parameters. cCs,|jddkrdS|jdkr dSt|S)Nzi?Fz)modefilenameboolr)rrrrr _can_readCs  zScreenGrabFormat._can_readcCs6|}|st|}|dk s$tt|}|ifS)N)rAssertionErrorZgrabnpasarray)rr"rpil_imimrrrr!Js   zScreenGrabFormat._get_dataN)r#r$r%r'r.r!rrrrr)7s r)c@s eZdZdZddZddZdS)ClipboardGrabFormata1 The ClipboardGrabFormat provided a means to grab image data from the clipboard, using the uri "" This functionality is provided via Pillow. Note that "" is only supported on Windows. Parameters for reading ---------------------- No parameters. cCs,|jddkrdS|jdkr dSt|S)Nr*zi?Fz )r+r,r-r)rrrrrr.`s  zClipboardGrabFormat._can_readcCs:|}|st|}|dkr(tdt|}|ifS)Nz5There seems to be no image data on the clipboard now.)rr/Z grabclipboard RuntimeErrorr0r1)rr"rr2r3rrrr!gs zClipboardGrabFormat._get_dataN)r#r$r%r'r.r!rrrrr4Ts r4Z screengrabz(Grab screenshots (Windows and OS X only)iZ clipboardgrabz"Grab from clipboard (Windows only))r'Z __future__rrrr Znumpyr0rZcorerrr)r4r Z add_formatrrrrs  (!