a ¤Â=bØ5ã@sÒdZddlZddlmZddlmZmZddlm Z ddl m Z m Z m Z mZdd lmZdd lmZmZmZmZmZdd lmZmZdd lmZdd lmZddgZdd„ZGdd„de j ƒZ!Gdd„de!ƒZ"dS)z< ImageViewer class for viewing and interacting with images. éNé)Úio)Ú dtype_rangeÚ img_as_float)Úrescale_intensityé)Ú QtWidgetsÚQtGuiÚQtÚSignal)ÚSlider)ÚdialogsÚ init_qtappÚfigimageÚ start_qtappÚupdate_axes_image)Ú BlitManagerÚ EventManager)ÚPlugin)ÚwarnÚ ImageViewerÚCollectionViewercCsr| ¡}|jdkr>|jj|jjf}t||d}| t|ƒ¡}n,|jdkrj|jddkrjt   |t   |¡f¡}t|ƒS)a¶Return RGB image from the given matplotlib image object. Each image in a matplotlib figure has its own colormap and normalization function. Return RGBA (RGB + alpha channel) image with float dtype. Parameters ---------- mpl_image : matplotlib.image.AxesImage object The image being converted. Returns ------- img : array of float, shape (M, N, 4) An image of float values in [0, 1]. r)Zin_ranger) Z get_arrayÚndimZnormZvminZvmaxrZcmaprÚshapeÚnpZdstackÚ ones_like)Z mpl_imageÚimageZ input_range©rú:lib/python3.9/site-packages/skimage/viewer/viewers/core.pyÚmpl_image_to_rgbas  rcsòeZdZdZejejejejdœZ e e j ƒZ d.‡fdd„ Zdd„Zd/d d „Zd0d d „Zdd„Zdd„Zd1dd„Zdd„Zd2‡fdd„ Zd3dd„Zdd„Zedd„ƒZejdd„ƒZd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Z d*d+„Z!d,d-„Z"‡Z#S)4ra,Viewer for displaying images. This viewer is a simple container object that holds a Matplotlib axes for showing images. `ImageViewer` doesn't subclass the Matplotlib axes (or figure) because of the high probability of name collisions. Subclasses and plugins will likely extend the `update_image` method to add custom overlays or filter the displayed image. Parameters ---------- image : array Image being viewed. Attributes ---------- canvas, fig, ax : Matplotlib canvas, figure, and axes Matplotlib canvas, figure, and axes used to display image. image : array Image being viewed. Setting this value will update the displayed frame. original_image : array Plugins typically operate on (but don't change) the *original* image. plugins : list List of attached plugins. Examples -------- >>> from skimage import data >>> image = data.coins() >>> viewer = ImageViewer(image) # doctest: +SKIP >>> viewer.show() # doctest: +SKIP )ÚtopZbottomÚleftÚrightTcsÒtdtddtƒtt|ƒ ¡| tj¡|  d¡t   d|¡|_ |j   d|jtjtj¡|j   d|jtjtj¡|j   d|jtjtj¡| ¡ |j ¡t  ¡|_| |j¡t|tƒrê|}|j}|j |j¡|j  |j!¡t"|ƒ\|_#|_$|j#j%|_%|j% &|¡|j$j'd d g|_(||_)|r) rCZwidgetrGZ widget_sizeZ viewer_sizeZdxZdyÚwÚhrrrrM¬s zImageViewer._add_widget_sizeNcCs4|durt ¡}|durdSt |¡}| |¡dS)z&Open image file and display in viewer.N)r Zopen_file_dialogrZimreadr/)rCÚfilenamerrrrr*ºs  zImageViewer.open_filecCs| |¡dS)zUpdate displayed image. This method can be overridden or extended in subclasses and plugins to react to image changes. N)r/©rCrrrrÚ update_imageÃszImageViewer.update_imagecCs ||_| ¡|_|j |¡dS©N)Úoriginal_imageÚcopyrrIÚemitrSrrrr/Ës z"ImageViewer._update_original_imagecCs|durt ¡}|durdSt|jjƒdkróz$ImageViewer.show..)r0rr9)rCZ main_windowrrrr`øszImageViewer.showcCs |jr|j ¡n |j ¡dSrU)r4r5Úredrawr2Z draw_idle©rCrrrrgs zImageViewer.redrawcCs|jSrU)Ú_imgrhrrrrszImageViewer.imagecCs–||_t|j|ƒ|jdd…\}}|j d|¡|j |d¡t|jj }|ddkrp|  ¡dkrpd|df}|j  |¡|j rŠd|j _| ¡dS)NrrrY)rirr8rr1Zset_xlimZset_ylimrZdtypeÚtypeÚminZset_climr4r5Z backgroundrg)rCrrQrPZclimrrrr s    cCs|j ¡|_dSrU)rVrWrrhrrrÚ reset_image!szImageViewer.reset_imagecCs|j ||¡}|S)z||jvrdS|jr"|j |j¡|j |¡|j |¡dSrU)r3r4r5Zremove_artistsrqÚremover6ÚdetachrsrrrÚ remove_tool9s   zImageViewer.remove_toolcCsLt|dƒ}t|dƒ}zd|j||f||fWStyFYdS0dS)Ngà?z%4s @ [%4s, %4s]ro)ÚintrÚ IndexError)rCraÚyrrrrpAs    zImageViewer._format_coord)T)r?)N)N)r)T)$Ú__name__Ú __module__Ú __qualname__Ú__doc__r ZTopDockWidgetAreaZBottomDockWidgetAreaZLeftDockWidgetAreaZRightDockWidgetArearKr rZndarrayrIr(rOrMr*rTr/r+r^r0r`rgÚpropertyrÚsetterrlrArnrBrtrwrpÚ __classcell__rrrErr1s:"ý =      cs2eZdZdZd ‡fdd„ Zdd„Zdd„Z‡ZS) raUViewer for displaying image collections. Select the displayed frame of the image collection using the slider or with the following keyboard shortcuts: left/right arrows Previous/next image in collection. number keys, 0--9 0% to 90% of collection. For example, "5" goes to the image in the middle (i.e. 50%) of the collection. home/end keys First/last image in collection. Parameters ---------- image_collection : list of images List of images to be displayed. update_on : {'move' | 'release'} Control whether image is updated on slide or release of the image slider. Using 'on_release' will give smoother behavior when displaying large images or when writing a plugin/subclass that requires heavy computation. r_c sŽtdtdd||_d|_t|jƒ|_|d}tt|ƒ |¡t dd|jdd}||d<|j |d<d |d <t d i|¤Ž|_ |j  |j ¡dS) Nz›`CollectionViewer` is deprecated and will be removed in 0.20. For alternatives, refer to https://scikit-image.org/docs/stable/user_guide/visualization.htmlrr#rrY)ÚvalueZlowZhighÚ update_onrmrxZ value_typeÚframe)r„)rr&Úimage_collectionÚindexrZÚ num_imagesr'rr(ÚdictÚ update_indexr Úsliderr:r;)rCr…rƒÚkwargsZ first_imageZ slider_kwsrErrr(dsý  zCollectionViewer.__init__cCsVtt|ƒƒ}||jkrdSt|dƒ}t||jdƒ}||_||j_| |j |¡dS)z:Select image on display using index into image collection.NrrY) rxÚroundr†Úmaxrkr‡rŠÚvalrTr…)rCrLr†rrrr‰}s   zCollectionViewer.update_indexcCspt|ƒtjkrd| ¡}d|kr*dkrZnn,td|d|jƒ}| d|¡| ¡ql| ¡n| ¡dS)Né0é:gš™™™™™¹?ro) rjr Z QKeyEventÚkeyrxr‡r‰ZacceptÚignore)rCr]r‘r†rrrÚ keyPressEventŒs   zCollectionViewer.keyPressEvent)r_)r{r|r}r~r(r‰r“rrrrErrKs)#r~ZnumpyrrorZ util.dtyperrZexposurerZqtrr r r Zwidgetsr Zutilsr rrrrZ utils.canvasrrZ plugins.baserÚwarningsrÚ__all__rZ QMainWindowrrrrrrÚs