ó  ‰\c@sªddlmZddlZddlZddljZddlm Z ddl m Z ddl m Z dZdZd „Zd d „Zd ed „Zd ed „ZdS(iÿÿÿÿ(treduceN(tPolygon(tPatchCollectioni(tpolygoniic Cstj|dtƒ}g|D]\}}|^q}g|D]\}}|^q;}t|||ƒ\}} |||| f<|S(Ntdtype(tnptzerostintR( tverticestshapetlabeltmasktxtytprtpctrrtcc((sAlib/python2.7/site-packages/skimage/future/manual_segmentation.pyt_mask_from_verticess gš™™™™™Ù?cCsJt|dtƒ}t|gdtd|ƒ}|j|ƒ}tjƒ|S(Ntclosedtmatch_originaltalpha(RtTrueRtadd_collectiontplttdraw(taxRRRtptpolygon_object((sAlib/python2.7/site-packages/skimage/future/manual_segmentation.pyt _draw_polygons  csmg‰g‰g‰g‰ˆjdkr6tdƒ‚ntjƒ\‰‰ˆjddƒˆjˆddƒˆjƒ‡‡‡fd†}ˆjd d d d gƒ‰tj j ˆd ƒ}|j |ƒ‡‡‡‡‡‡‡‡fd †}ˆj j d|ƒtjdtƒ‡fd†tˆddƒDƒ}|rCtj|ƒSttj|tjdˆjd ƒƒSdS(slReturn a label image based on polygon selections made with the mouse. Parameters ---------- image : (M, N[, 3]) array Grayscale or RGB image. alpha : float, optional Transparency value for polygons drawn over the image. return_all : bool, optional If True, an array containing each separate polygon drawn is returned. (The polygons may overlap.) If False (default), latter polygons "overwrite" earlier ones where they overlap. Returns ------- labels : array of int, shape ([Q, ]M, N) The segmented regions. If mode is `'separate'`, the leading dimension of the array corresponds to the number of regions that the user drew. Notes ----- Use left click to select the vertices of the polygon and right click to confirm the selection once all vertices are selected. Examples -------- >>> from skimage import data, future, io >>> camera = data.camera() >>> mask = future.manual_polygon_segmentation(camera) # doctest: +SKIP >>> io.imshow(mask) # doctest: +SKIP >>> io.show() # doctest: +SKIP iis.Only 2D grayscale or RGB images are supported.tbottomgš™™™™™É?tcmaptgraycs:ˆr6ˆjƒˆjƒ}|jƒˆjjƒndS(N(tpoptremovetcanvast draw_idle(targstkwargst last_poly(tfigtlist_of_vertex_liststpolygons_drawn(sAlib/python2.7/site-packages/skimage/future/manual_segmentation.pyt_undoPs    g333333ë?gš™™™™™©?g333333³?u⟲cs"|jdks|jˆkr"dSˆjjjjdk r>dS|jtkr®ˆj|j |j gƒˆr…ˆj ƒ}|j ƒnt ˆˆdˆdƒ}ˆj|ƒnp|jtkrˆsÇdSˆjˆƒt ˆˆdˆƒ}ˆj|ƒˆj ƒ}|j ƒˆ2tjƒndS(NRgffffffö?(tinaxestNoneR#tmanagerttoolbart_activetbuttont LEFT_CLICKtappendtxdatatydataR!R"Rt RIGHT_CLICKRR(teventtpolyRRt preview_poly(RRR(R)R*tpreview_polygon_drawnt temp_listtundo_pos(sAlib/python2.7/site-packages/skimage/future/manual_segmentation.pyt_extend_polygon]s*     tbutton_press_eventtblockc3s.|]$\}}t|ˆjd |ƒVqdS(iN(RR (t.0tiR(timage(sAlib/python2.7/site-packages/skimage/future/manual_segmentation.pys †ststartiiN(ii(tndimt ValueErrorRtsubplotstsubplots_adjusttimshowt set_axis_offtadd_axest matplotlibtwidgetstButtont on_clickedR#t mpl_connecttshowRt enumerateRtstackRtmaximumt broadcast_toR (RBRt return_allR+t undo_buttonR=tlabels(( RRR(RBR)R*R:R;R<sAlib/python2.7/site-packages/skimage/future/manual_segmentation.pytmanual_polygon_segmentations,#   $%  c sWg‰g‰ˆjdkr*tdƒ‚ntjƒ\‰‰ˆjddƒˆjˆddƒˆjƒ‡‡‡fd†}ˆjd d d d gƒ}tj j |d ƒ}|j |ƒ‡‡‡‡fd †}tj j ˆ|ƒ}tj dtƒ‡fd†tˆddƒDƒ}|r-tj|ƒSttj|tjdˆjd ƒƒSdS(s*Return a label image based on freeform selections made with the mouse. Parameters ---------- image : (M, N[, 3]) array Grayscale or RGB image. alpha : float, optional Transparency value for polygons drawn over the image. return_all : bool, optional If True, an array containing each separate polygon drawn is returned. (The polygons may overlap.) If False (default), latter polygons "overwrite" earlier ones where they overlap. Returns ------- labels : array of int, shape ([Q, ]M, N) The segmented regions. If mode is `'separate'`, the leading dimension of the array corresponds to the number of regions that the user drew. Notes ----- Press and hold the left mouse button to draw around each object. Examples -------- >>> from skimage import data, future, io >>> camera = data.camera() >>> mask = future.manual_lasso_segmentation(camera) # doctest: +SKIP >>> io.imshow(mask) # doctest: +SKIP >>> io.show() # doctest: +SKIP iis.Only 2D grayscale or RGB images are supported.Rgš™™™™™É?RR cs:ˆr6ˆjƒˆjƒ}|jƒˆjjƒndS(N(R!R"R#R$(R%R&R'(R(R)R*(sAlib/python2.7/site-packages/skimage/future/manual_segmentation.pyR+»s    g333333ë?gš™™™™™©?g333333³?u⟲csSt|ƒdkrdSˆj|ƒtˆ|dˆƒ}ˆj|ƒtjƒdS(NiR(tlenR3RRR(RR(RRR)R*(sAlib/python2.7/site-packages/skimage/future/manual_segmentation.pyt_on_lasso_selectionÈs   R?c3s.|]$\}}t|ˆjd |ƒVqdS(iN(RR (R@RAR(RB(sAlib/python2.7/site-packages/skimage/future/manual_segmentation.pys ÔsRCiiN(ii(RDRERRFRGRHRIRJRKRLRMRNt LassoSelectorRPRRQRRRRRSRTR ( RBRRUR+R<RVRZtlassoRW((RRR(RBR)R*sAlib/python2.7/site-packages/skimage/future/manual_segmentation.pytmanual_lasso_segmentationŽs("     (t functoolsRtnumpyRRKtmatplotlib.pyplottpyplotRtmatplotlib.patchesRtmatplotlib.collectionsRRRR2R6RRtFalseRXR](((sAlib/python2.7/site-packages/skimage/future/manual_segmentation.pyts   o