B \@sddlZddlmZddlmZmZddgZGdddeZGdddeZ e d krdd l m Z dd l mZe ZeeZ ej\ZZe e Zeded gd eded gfe_e dS) N)lines)CanvasToolBase ToolHandlesLineTool ThickLineToolcsreZdZdZdfdd 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} | |dk r@|ni| d|_||_d|_d} d} t | | g|_ t j | | fd d d | |_ |j|j t|j| | |d |_|jd |j |jjg|_|dkrd d}||_|j|dS)N)on_moveon_enter on_releaserg?Zbutt)Zcolor linewidthZalphaZsolid_capstyler)rrFT)ZvisibleZanimated)Z marker_propscSs<t|\}}tdtt|dt|ddS)Nzlength = %0.2f)np transposeprintZsqrtZdiff)ptsxyrBlib/python3.7/site-packages/skimage/viewer/canvastools/linetool.pyr >sz#LineTool.__init__..on_enter)superr__init__dictupdatermaxdist _active_ptrr_end_ptsrZLine2D_lineaxZadd_liner_handles set_visibleZartistZartistsZcallback_on_entermanagerZadd_tool) selfr#r r r r line_props handle_propskwargsZpropsrr) __class__rrr$s*    zLineTool.__init__cCs |jtS)N)rZastypeint)r$rrr end_pointsEszLineTool.end_pointscCsTt||_|jt||jt||j|j| d| dS)NT) rZasarrayrrset_datarr!Z set_linewidthrr"redraw)r$rrrrr*Is   cCsR|jdks|j|sdS|j|j|j\}}||jkrD||_dSd|_dSdS)Nr FT) buttonr in_axesr!Zclosestrrrr)r$eventidxZpx_distrrrhit_testTs zLineTool.hit_testcCsD|d|jdkr@d|_|j|j}}t||g||gg|_dS)NTr)r"rxdataydatarZarrayr)r$r/rrrrron_mouse_press_s   zLineTool.on_mouse_presscCs,|jdkrdSd|_||j|dS)Nr )r-rZcallback_on_releasegeometryr,)r$r/rrron_mouse_releasefs   zLineTool.on_mouse_releasecCsH|jdks|jdkrdS|j|s(dS||j|j||jdS)Nr ) r-rr r.rr2r3Zcallback_on_mover5)r$r/rrrr ms  zLineTool.on_movecCs,|dk r ||f|j|jddf<|j|_dS)N)rrr*)r$rrrrrruszLineTool.updatecCs|jS)N)r*)r$rrrr5zszLineTool.geometry)NNNrNN)NN)__name__ __module__ __qualname____doc__rpropertyr*setterr1r4r6r rr5 __classcell__rr)r(rr s    csBeZdZdZdfdd 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|dkr.dd}||_dS)N)r r r rr%r&cWsdS)Nr)argsrrr on_changesz)ThickLineTool.__init__..on_change)rrrcallback_on_change) r$r#r r r r?rr%r&)r(rrrs zThickLineTool.__init__cCs4|js dS|jdkr|n|jdkr0|dS)NZupZdown)Zinaxesr-_thicken_scan_line_shrink_scan_line)r$r/rrr on_scrolls    zThickLineTool.on_scrollcCs*|jdkr|n|jdkr&|dS)N+-)keyrArB)r$r/rrr on_key_presss   zThickLineTool.on_key_presscCs&|jd7_|||jdS)Nr )rrr@r5)r$rrrrAsz ThickLineTool._thicken_scan_linecCs0|jdkr,|jd8_|||jdS)Nr )rrr@r5)r$rrrrBs zThickLineTool._shrink_scan_line)NNNNrNN) r7r8r9r:rrCrGrArBr=rr)r(rrs__main__)data) ImageViewerr)ZnumpyrZ matplotlibrZviewer.canvastools.baserr__all__rrr7rIZviewerrJZcameraZimageshapehwZ line_toolr*Zshowrrrrs vJ   &