o :ήc@sddlZddlmZddlmZmZddgZGdddeZGdddeZ e d kr`dd l m Z dd l mZe ZeeZ ej\ZZe e Zeded gd eded gfe_e dSdS) N)lines)CanvasToolBase ToolHandlesLineTool ThickLineToolcsveZdZdZ  dfdd ZeddZejddZd d Zd d Z d dZ ddZ dddZ eddZ ZS)ravWidget for line selection in a plot. Parameters ---------- manager : Viewer or PlotPlugin. Skimage viewer or plot plugin object. 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. maxdist : float Maximum pixel distance allowed when selecting control handle. line_props : dict Properties for :class:`matplotlib.lines.Line2D`. handle_props : dict Marker properties for the handles (also see :class:`matplotlib.lines.Line2D`). Attributes ---------- end_points : 2D array End points of line ((x1, y1), (x2, y2)). N c  stt|j|f|||d|tddddd} | |dur |ni| d|_||_d|_d} d} t | | g|_ t j | | fd d d | |_ |j|j t|j| | |d |_|jd |j |jjg|_|durod d}||_|j|dS)N)on_moveon_enter on_releaserg?butt)color linewidthalphasolid_capstyler)rrFT)visibleanimated) marker_propscSs<t|\}}tdtt|dt|ddS)Nzlength = %0.2f)np transposeprintsqrtdiff)ptsxyrJ/tmp/pip-target-vg8gfxp4/lib/python/skimage/viewer/canvastools/linetool.pyr >s  z#LineTool.__init__..on_enter)superr__init__dictupdatermaxdist _active_ptrr_end_ptsrLine2D_lineaxadd_liner_handles set_visibleartistartistscallback_on_entermanageradd_tool) selfr1r r r r% line_props handle_propskwargspropsrr __class__rr r"$s2   zLineTool.__init__cCs |jtSN)r'astypeintr3rrr end_pointsEs zLineTool.end_pointscCsTt||_|jt||jt||j|j| d| dS)NT) rasarrayr'r)set_datarr, set_linewidthrr-redraw)r3rrrr r>Is   cCsN|jdks |j|s dS|j|j|j\}}||jkr"||_dSd|_dS)Nr FT) buttonr*in_axesr,closestrrr%r&)r3eventidxpx_distrrr hit_testTs zLineTool.hit_testcCsH|d|jdur"d|_|j|j}}t||g||gg|_dSdS)NTr)r-r&xdataydatararrayr')r3rFrrrrr on_mouse_press_s  zLineTool.on_mouse_presscCs,|jdkrdSd|_||j|dSNr )rCr&callback_on_releasegeometryrBr3rFrrr on_mouse_releasefs   zLineTool.on_mouse_releasecCsH|jdks |jdur dS|j|sdS||j|j||jdSrN) rCr&r*rDr$rJrKcallback_on_moverPrQrrr r ms  zLineTool.on_movecCs,|dur||f|j|jddf<|j|_dSr:)r'r&r>)r3rrrrr r$us zLineTool.updatecCs|jSr:)r>r=rrr rPzszLineTool.geometry)NNNrNN)NN)__name__ __module__ __qualname____doc__r"propertyr>setterrIrMrRr r$rP __classcell__rrr8r r s !     csFeZdZdZ  dfdd ZddZdd Zd d Zd d ZZ S)ra<Widget for line selection in a plot. The thickness of the line can be varied using the mouse scroll wheel, or with the '+' and '-' keys. Parameters ---------- manager : Viewer or PlotPlugin. Skimage viewer or plot plugin object. 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. on_change : function Function called whenever the line thickness is changed. maxdist : float Maximum pixel distance allowed when selecting control handle. line_props : dict Properties for :class:`matplotlib.lines.Line2D`. handle_props : dict Marker properties for the handles (also see :class:`matplotlib.lines.Line2D`). Attributes ---------- end_points : 2D array End points of line ((x1, y1), (x2, y2)). Nrc s8tt|j|||||||d|durdd}||_dS)N)r r r r%r4r5cWsdSr:r)argsrrr on_changesz)ThickLineTool.__init__..on_change)r!rr"callback_on_change) r3r1r r r r\r%r4r5r8rr r"s  zThickLineTool.__init__cCs:|jsdS|jdkr|dS|jdkr|dSdS)Nupdown)inaxesrC_thicken_scan_line_shrink_scan_linerQrrr on_scrolls    zThickLineTool.on_scrollcCs0|jdkr |dS|jdkr|dSdS)N+-)keyrarbrQrrr on_key_presss    zThickLineTool.on_key_presscCs&|jd7_|||jdSrNrr$r]rPr=rrr rasz ThickLineTool._thicken_scan_linecCs4|jdkr|jd8_|||jdSdSrNrhr=rrr rbs zThickLineTool._shrink_scan_line)NNNNrNN) rTrUrVrWr"rcrgrarbrZrrr8r rs __main__)data) ImageViewerr)numpyr matplotlibrviewer.canvastools.baserr__all__rrrTrjviewerrkcameraimageshapehw line_toolr>showrrrr s  vJ   &