ó  ‰\c@sddlZddljZddljZejddddddgƒZ dd l m Z d gZ d e fd „ƒYZ d efd „ƒYZedkrejjƒddlmZddlmZejƒZeeƒZe eejƒZejƒndS(iÿÿÿÿNtwhitetredt dodgerbluetgoldt greenyellowt bluevioleti(tCanvasToolBaset PaintToolcBsòeZdZddddddd„Zed„ƒZejd„ƒZed„ƒZejd„ƒZed„ƒZ e jd „ƒZ ed „ƒZ e jd „ƒZ d „Z d „Z d„Z d„Zd„Zd„Zed„ƒZRS(sWidget for painting on top of a plot. Parameters ---------- manager : Viewer or PlotPlugin. Skimage viewer or plot plugin object. overlay_shape : shape tuple 2D shape tuple used to initialize overlay image. alpha : float (between [0, 1]) Opacity of overlay on_move : function Function called whenever a control handle is moved. This function must accept the end points of line as the only argument. on_release : function Function called whenever the control handle is released. on_enter : function Function called whenever the "enter" key is pressed. rect_props : dict Properties for :class:`matplotlib.patches.Rectangle`. This class redefines defaults in :class:`matplotlib.widgets.RectangleSelector`. Attributes ---------- overlay : array Overlay of painted labels displayed on top of image. label : int Current paint color. Examples ---------- >>> from skimage.data import camera >>> import matplotlib.pyplot as plt >>> from skimage.viewer.canvastools import PaintTool >>> import numpy as np >>> img = camera() #doctest: +SKIP >>> ax = plt.subplot(111) #doctest: +SKIP >>> plt.imshow(img, cmap=plt.cm.gray) #doctest: +SKIP >>> p = PaintTool(ax,np.shape(img[:-1]),10,0.2) #doctest: +SKIP >>> plt.show() #doctest: +SKIP >>> mask = p.overlay #doctest: +SKIP >>> plt.imshow(mask,cmap=plt.cm.gray) #doctest: +SKIP >>> plt.show() #doctest: +SKIP ig333333Ó?c Cstt|ƒj|d|d|d|ƒtdddddd d tƒ} | j|dk ra|niƒ||_t|_ d|_ ||_ t j d d d | |_|jjtƒ|jj|jƒd |_||_|j |jg|_|jj|ƒdS(Nton_moveton_entert on_releaset edgecolortrt facecolors0.7talphagà?tanimatedii(ii(tsuperRt__init__tdicttTruetupdatetNoneRt LABELS_CMAPtcmapt _overlay_plottshapetpltt Rectanglet_cursort set_visibletFalsetaxt add_patchtlabeltradiustartiststmanagertadd_tool( tselfR$t overlay_shapeR"RRR R t rect_propstprops((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyR;s !      cCs|jS(N(t_label(R&((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyR!VscCsX||jjkr2tdt|jdƒƒ‚n||_|jj|j|ƒƒdS(NsMaximum label value = %si(RtNt ValueErrortlenR*Rt set_edgecolor(R&tvalue((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyR!Zs  cCs|jS(N(t_radius(R&((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyR"ascCsY||_d|d|_|jj|jƒ|jj|jƒt||jƒ|_dS(Nii(R0t_widthRt set_widtht set_heighttCenteredWindowt_shapetwindow(R&R ((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyR"es  cCs|jS(N(t_overlay(R&((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pytoverlaymsc Cs¬||_|dkr7|jjj|jƒd|_ng|jdkrŽtd|jd|jdt j ƒdt ƒ}|jj |||_n|jj |ƒ|jƒdS(NRRtnormR(R7RRtimagestremoveRRRRtmcolorstNoNormRtimshowtset_datatredraw(R&timageR)((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyR8qs   cCs|jS(N(R5(R&((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyRscCso||_|jdk rS|jjd|dd|dddfƒ|j|_ntj|ddƒ|_dS(Ngà¿igà?itdtypetuint8( R5RRt set_extentR0R"tnptzerosR8(R&R((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyRƒs  cCs0|jdkr,|j|jƒ|jƒndS(Ntenter(tkeytcallback_on_entertgeometryR@(R&tevent((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyt on_key_pressŒscCsV|jdks"|jj|ƒ r&dS|j|j|jƒ|j|j|jƒdS(Ni(tbuttonRtin_axest update_cursortxdatatydatatupdate_overlay(R&RK((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyton_mouse_press‘s"cCs'|jdkrdS|j|jƒdS(Ni(RMtcallback_on_releaseRJ(R&RK((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyton_mouse_release—scCs|jj|ƒs0|jjtƒ|jƒdS|jjtƒ|j|j|j ƒ|j dkrs|jƒdS|j |j|j ƒ|j |j ƒdS(Ni(RRNRRRR@RRORPRQRMRRtcallback_on_moveRJ(R&RK((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyRœs  cCs2|j}|j||jj||ƒ<||_dS(N(R8R!R6tat(R&txtyR8((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyRRªs cCs<||jd}||jd}|jj||fƒdS(Ni(R"Rtset_xy(R&RXRY((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyRO°scCs|jS(N(R8(R&((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyRJµsN(t__name__t __module__t__doc__RRtpropertyR!tsetterR"R8RRLRSRURRRRORJ(((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyR s&.        R4cBs eZdZd„Zd„ZRS(sqWindow that create slices numpy arrays over 2D windows. Examples -------- >>> a = np.arange(16).reshape(4, 4) >>> w = CenteredWindow(1, a.shape) >>> a[w.at(1, 1)] array([[ 0, 1, 2], [ 4, 5, 6], [ 8, 9, 10]]) >>> a[w.at(0, 0)] array([[0, 1], [4, 5]]) >>> a[w.at(4, 3)] array([[14, 15]]) cCs||_||_dS(N(R"t array_shape(R&R"R`((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyRËs c Csˆ|j\}}|j}td||ƒ}t|||dƒ}td||ƒ}t|||dƒ} t|| ƒt||ƒgS(Nii(R`R"tmaxtmintslice( R&trowtcolthtwR txmintxmaxtymintymax((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyRWÏs (R[R\R]RRW(((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyR4ºs t__main__(tdata(t ImageViewer(tnumpyREtmatplotlib.pyplottpyplotRtmatplotlib.colorstcolorsR<tListedColormapRtviewer.canvastools.baseRt__all__RtobjectR4R[ttestingtrundocstRmtviewerRntcameraRARt paint_tooltshow(((sClib/python2.7/site-packages/skimage/viewer/canvastools/painttool.pyts  ®