o :ήc@sddlZddlmZmZmZmZddlmZddl Z ddl m Z ddl m Z ddlmZer:d e vr:ed gd Zdad d Zd#ddZd#ddZGdddeZGdddeZGdddeZGdddeZddZd$ddZd%dd Zd!d"ZdS)&N) QtWidgetshas_qtFigureManagerQTFigureCanvasQTAgg)warn)Figure)_pylab_helpers)LinearSegmentedColormapaggzNRecommended matplotlib backend is `Agg` for full skimage.viewer functionality.) init_qtapp start_qtapp RequiredAttrfigimageLinearColormap ClearColormap FigureCanvasnew_plotupdate_axes_imagecCs tjatdurtgatS)zgInitialize QAppliction. The QApplication needs to be initialized before creating any QWidgets N)r QApplicationinstanceQApprr@/tmp/pip-target-vg8gfxp4/lib/python/skimage/viewer/utils/core.pyr s  r cCs"|durt}t|dr|jSdS)z%Return True if event loop is running.N_in_event_loopF)r hasattrrapprrris_event_loop_running"s  rcCs8|durt}t|sd|_|d|_dSd|_dS)zStart Qt mainloopNTF)r rrexec_rrrrr,s  rc@s0eZdZdZeZd ddZddZddZdS) rz.A class attribute that must be set before use.NcCs||j|df<dSN instances)selfinit_valrrr__init__=zRequiredAttr.__init__cCs"|j||f}|durtd|S)NzRequired attribute not set)r#AttributeError)r$objobjtypevaluerrr__get__@szRequiredAttr.__get__cCs||j||f<dSr!r")r$r)r+rrr__set__Fr'zRequiredAttr.__set__r!) __name__ __module__ __qualname____doc__dictr#r&r,r-rrrrr8s   rc@seZdZdZddZdS)raSLinearSegmentedColormap in which color varies smoothly. This class is a simplification of LinearSegmentedColormap, which doesn't support jumps in color intensities. Parameters ---------- name : str Name of colormap. segmented_data : dict Dictionary of 'red', 'green', 'blue', and (optionally) 'alpha' values. Each color key contains a list of `x`, `y` tuples. `x` must increase monotonically from 0 to 1 and corresponds to input values for a mappable object (e.g. an image). `y` corresponds to the color intensity. cKs,dd|D}tj|||fi|dS)NcSs i|] \}}|dd|DqS)cSsg|] \}}|||fqSrr).0xyrrr ^sz6LinearColormap.__init__...r)r3keyr+rrr ^sz+LinearColormap.__init__..)itemsr r&)r$namesegmented_datakwargsrrrr&]szLinearColormap.__init__Nr.r/r0r1r&rrrrrJs rc@seZdZdZdddZdS)rz7Color map that varies linearly from alpha = 0 to 1  clear_colorcCsV|\}}}d|fd|fgd|fd|fgd|fd|fgdd|fgd}t|||dS)Ng?)r@r@)bluegreenredalpha)rr&)r$rgb max_alphar:rgbcg_speqrrrr&fs  zClearColormap.__init__N)r>r?r=rrrrrcsrc@s eZdZdZddZddZdS)rzCanvas for displaying images.cKs8||_t||jt|tjjtjjt|dSr!)figrr& setSizePolicyr QSizePolicy ExpandingupdateGeometry)r$figurer<rrrr&qszFigureCanvas.__init__cCst|||dSr!)r resizeEvent resize_event)r$eventrrrrQys  zFigureCanvas.resizeEventN)r.r/r0r1r&rQrrrrros rcOsRtjj}|rt|dnd}|dt}||i|}t|}t||}|j S)zReturn a new figure canvas.r> FigureClass) r Gcffigskeysmaxpopr rrcanvas)argsr<allnumsnumrTrPrZ fig_managerrrr new_canvass   r_cKsB|duri}tdi|}|||j}|jdi|}||fS)aReturn new figure and axes. Parameters ---------- parent : QtWidget Qt widget that displays the plot objects. If None, you must manually call ``canvas.setParent`` and pass the parent widget. subplot_kw : dict Keyword arguments passed ``matplotlib.figure.Figure.add_subplot``. fig_kw : dict Keyword arguments passed ``matplotlib.figure.Figure``. N)r>r>r>r)r_ setParentrP add_subplot)parent subplot_kwfig_kwrZrKaxrrrrs  rr>c Ks|dur|ntjd}|dd|ddt|j\}}}tj||ftd||}t||d\}} |j d d d d d | | j |fi|| j j || fS) a/Return figure and axes with figure tightly surrounding image. Unlike pyplot.figimage, this actually plots onto an axes object, which fills the figure. Plotting the image onto an axes allows for subsequent overlays of axes artists. Parameters ---------- image : array image to plot scale : float If scale is 1, the figure and axes have the same dimension as the image. Smaller values of `scale` will shrink the figure. dpi : int Dots per inch for figure. If None, use the default rcParam. Nz figure.dpi interpolationnearestcmapgray)dtype)figsizedpirr>)leftbottomrighttop)mplrcParams setdefaultnp atleast_3dshapearrayfloatrsubplots_adjust set_axis_offimshowrPrZdraw) imagescalerlr<hwdrkrKrerrrrs   rcCs2|||jdd\}}|d||dfdS)a Update the image displayed by an image plot. This sets the image plot's array and updates its shape appropriately Parameters ---------- image_axes : `matplotlib.image.AxesImage` Image axes to update. image : array Image array. Nrr) set_arrayrv set_extent) image_axesr}rrrrrrs rr!)NN)r>N) numpyrtqtrrrr _shared.utilsr matplotlibrqmatplotlib.figurer r matplotlib.colorsr get_backendlower__all__rr rrobjectrrrrr_rrrrrrrs,         !