U % d @sTdZddlZddlZddlmZGdddeZGdddeZGd d d eZdS) zD PIL-based formats to take screenshots and grab from the clipboard. N)FormatcsJeZdZdZdZdZfddZddZdd ZGd d d e j Z Z S) BaseGrabFormatzBase format for grab formats.FNcs tt|j||t|_dSN)superr__init__ threadingRLock_lock)selfargskwargs __class__8/tmp/pip-unpacked-wheel-x84omsqu/imageio/plugins/grab.pyrszBaseGrabFormat.__init__cCsdS)NFrr requestrrr _can_writeszBaseGrabFormat._can_writec Csv|jd|jsfd|_ddl}t|ds.tdzddlm}Wn tk r^YW5QRdSX||_W5QRX|jS)NTr __version__z(Imageio Pillow requires Pillow, not PIL!) ImageGrab)r _pillow_importedPILhasattr ImportErrorr _ImageGrab)r rrrrr _init_pillows zBaseGrabFormat._init_pillowc@s$eZdZddZddZddZdS)zBaseGrabFormat.ReadercCsdSrrr rrr_open)szBaseGrabFormat.Reader._opencCsdSrrrrrr_close,szBaseGrabFormat.Reader._closecCs |j|Sr)format _get_data)r indexrrrr!/szBaseGrabFormat.Reader._get_dataN)__name__ __module__ __qualname__rrr!rrrrReader(sr&) r#r$r%__doc__rrrrrrr& __classcell__rrrrr s rc@s eZdZdZddZddZdS)ScreenGrabFormataThe 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|jdkrdSt|S)NzFfilenameboolrrrrr _can_read?s zScreenGrabFormat._can_readcCs6|}|st|}|dk s$tt|}|ifSr)rAssertionErrorZgrabnpasarrayr r"rZpil_imZimrrrr!Ds   zScreenGrabFormat._get_dataNr#r$r%r'r-r!rrrrr)3s r)c@s eZdZdZddZddZdS)ClipboardGrabFormata(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|jdkrdSt|S)Nz Fr*rrrrr-Zs zClipboardGrabFormat._can_readcCs:|}|st|}|dkr(tdt|}|ifS)Nz5There seems to be no image data on the clipboard now.)rr.Z grabclipboard RuntimeErrorr/r0r1rrrr!_s zClipboardGrabFormat._get_dataNr2rrrrr3Ns r3) r'rZnumpyr/corerrr)r3rrrrs  '