o :a+@sdZddlZddlmZddlmZddlmZddlm Z ej de dde e ZGd d d ZGd d d eZGd ddeZGdddeZdS)a Classes used for blocking interaction with figure windows: `BlockingInput` Creates a callable object to retrieve events in a blocking way for interactive sessions. Base class of the other classes listed here. `BlockingKeyMouseInput` Creates a callable object to retrieve key or mouse clicks in a blocking way for interactive sessions. Used by `~.Figure.waitforbuttonpress`. `BlockingMouseInput` Creates a callable object to retrieve mouse clicks in a blocking way for interactive sessions. Used by `~.Figure.ginput`. `BlockingContourLabeler` Creates a callable object to retrieve mouse clicks in a blocking way that will then be used to place labels on a `.ContourSet`. Used by `~.Axes.clabel`. N)Integral)_api) MouseButtonz3.5module)nameZobj_typec@sReZdZdZdddZddZddZd d Zd d ZdddZ e Z dddZ dS) BlockingInputz1Callable for retrieving events in a blocking way.cCs||_||_dSNfig eventslist)selfr r rr;/usr/lib/python3/dist-packages/matplotlib/blocking_input.py__init__$s zBlockingInput.__init__cCsZ||tdt|j|t|j|jkr dkr+ndS|jj dSdS)zY Event handler; will be passed to the current figure to retrieve events. zEvent %irN) add_event_loginfolenevents post_eventnr canvasstop_event_loopr eventrrron_event(s "zBlockingInput.on_eventcCsdS)z-For baseclass, do nothing but collect events.Nrr rrrr8szBlockingInput.post_eventcCs$|jD] }|jj|qg|_dS)zDisconnect all callbacks.N) callbacksr rZmpl_disconnect)r cbrrrcleanup;s  zBlockingInput.cleanupcC|j|dS)z5For base class, this just appends an event to events.N)rappendrrrrrAszBlockingInput.add_eventcCr )z Remove an event from the event list -- by default, the last. Note that this does not check that there are events, much like the normal pop method. If no events exist, this will throw an exception. N)rpop)r indexrrr pop_eventEszBlockingInput.pop_eventcsptjt|d|_g_jjjrjfddj D_ zjjj |dW jS w)z%Blocking call to retrieve *n* events.)rcsg|] }jj|jqSr)r rZ mpl_connectr).0rrrr Zsz*BlockingInput.__call__..)timeout) rZcheck_isinstancerrrr rZmanagerZshowr rZstart_event_loopr)r rr*rrr__call__Ps    zBlockingInput.__call__N)rr")r&r') __name__ __module__ __qualname____doc__rrrrrr%r#r+rrrrr!s   rcseZdZdZejZejZej Z ejejej ffdd Z ddZ ddZ dd Zd d Zd d ZddZddZdddZdfdd Zd fdd Zd!fdd ZZS)"BlockingMouseInputa Callable for retrieving mouse clicks in a blocking way. This class will also retrieve keypresses and map them to mouse clicks: delete and backspace are a right click, enter is like a middle click, and all others are like a left click. cs&tj|dd||_||_||_dSN)Zbutton_press_eventkey_press_eventr )superr button_add button_pop button_stop)r r Z mouse_addZ mouse_popZ mouse_stop __class__rrrss  zBlockingMouseInput.__init__cCsDt|jdkrtddS|jdjdkr|dS|dS)zProcess an event.rz No events yetr"r3N)rrrwarningr key_event mouse_eventrrrrr}s   zBlockingMouseInput.post_eventcCs\|jd}|j}||jkr||dS||jkr ||dS||jkr,||dSdS)zProcess a mouse click event.r"N)rbuttonr6mouse_event_popr7mouse_event_stopr5mouse_event_add)r rr=rrrr<s    zBlockingMouseInput.mouse_eventcCsV|jd}|jdur dS|jdvr||dS|jdvr$||dS||dS)zV Process a key press event, mapping keys to appropriate mouse clicks. r"N)Z backspacedelete)escapeZenter)rkeyr>r?r@rrrrr;s    zBlockingMouseInput.key_eventcCs"|jr ||dSt|dS)z Process an button-1 event (add a click if inside axes). Parameters ---------- event : `~.backend_bases.MouseEvent` N)inaxes add_clickrr#rrrrr@sz"BlockingMouseInput.mouse_event_addcCst||jjdS)z Process an button-2 event (end blocking input). Parameters ---------- event : `~.backend_bases.MouseEvent` N)rr#r rrrrrrr?s z#BlockingMouseInput.mouse_event_stopcCs"t||jr||dSdS)z Process an button-3 event (remove the last click). Parameters ---------- event : `~.backend_bases.MouseEvent` N)rr#rrrrrr>s z"BlockingMouseInput.mouse_event_popcCsz|j|j|jftdt|j|j|j|jr;tj |jg|jgddd}|j ||j ||j jdSdS)z Add the coordinates of an event to the list of clicks. Parameters ---------- event : `~.backend_bases.MouseEvent` zinput %i: %f, %f+r)ZmarkerZcolorN)clicksr!ZxdataZydatarrr show_clicksmlinesZLine2DrDZadd_linemarksr rdraw)r rlinerrrrEs  zBlockingMouseInput.add_clickr"cCs6|j||jr|j||jjdSdS)z Remove a click (by default, the last) from the list of clicks. Parameters ---------- event : `~.backend_bases.MouseEvent` N)rHr#rIrKremover rrLr rr$rrr pop_clicks zBlockingMouseInput.pop_clickcs|||t|dS)zw Remove a click and the associated event from the list of clicks. Defaults to the last click. N)rPr4r#rOr8rrr#s zBlockingMouseInput.popNcs:|jr|jD]}|qg|_|jjtdS)z{ Parameters ---------- event : `~.backend_bases.MouseEvent`, optional Not used N)rIrKrNr rrLr4r)r rZmarkr8rrrs    zBlockingMouseInput.cleanupr&r'Tcs(||_g|_g|_tj||d|jS)zV Blocking call to retrieve *n* coordinate pairs through mouse clicks. rr*)rIrHrKr4r+)r rr*rIr8rrr+s zBlockingMouseInput.__call__r,r )r&r'T)r-r.r/r0rZLEFTr5ZRIGHTr6ZMIDDLEr7rrr<r;r@r?r>rErPr#rr+ __classcell__rrr8rr1fs(       r1csPeZdZdZfddZddZdddZd d Zd d Zdfdd Z Z S)BlockingContourLabelerzt Callable for retrieving mouse clicks and key presses in a blocking way. Used to place contour labels. cs||_tj|jjddS)N)r )csr4rZaxesfigure)r rTr8rrrszBlockingContourLabeler.__init__cC||dSr )button1rrrrrEz BlockingContourLabeler.add_clickr"cCrVr )button3rOrrrrPrXz BlockingContourLabeler.pop_clickcCsJ|j|jjkr|jj|j|j|j|jdd|jj dSt |dS)z Process an button-1 event (add a label to a contour). Parameters ---------- event : `~.backend_bases.MouseEvent` F)inline_spacingZ transformN) rDrTaxZadd_label_nearxyinlinerZr rrLrr#rrrrrWs zBlockingContourLabeler.button1cCs(|jrdS|j|jjjjdS)a Process an button-3 event (remove a label if not in inline mode). Unfortunately, if one is doing inline labels, then there is currently no way to fix the broken contour - once humpty-dumpty is broken, he can't be put back together. In inline mode, this does nothing. Parameters ---------- event : `~.backend_bases.MouseEvent` N)r^rTZ pop_labelr[rUrrLrrrrrY0s  zBlockingContourLabeler.button3cs"||_||_tj||dddS)NF)rr*rI)r^rZr4r+)r r^rZrr*r8rrr+BszBlockingContourLabeler.__call__r,)r_r"r") r-r.r/r0rrErPrWrYr+rRrrr8rrSs  rScs6eZdZdZfddZddZd fdd ZZS) BlockingKeyMouseInputzQ Callable for retrieving mouse clicks and key presses in a blocking way. cstj|dddSr2)r4r)r r r8rrrMs zBlockingKeyMouseInput.__init__cCs*|jr|jdjdk|_dStddS)zDetermine if it is a key event.r"r3zNo events yet.N)rr keyormouserr:rrrrrQsz BlockingKeyMouseInput.post_eventr'csd|_tjd|d|jS)z Blocking call to retrieve a single mouse click or key press. Returns ``True`` if key press, ``False`` if mouse click, or ``None`` if timed out. Nr&rQ)rar4r+)r r*r8rrr+XszBlockingKeyMouseInput.__call__)r')r-r.r/r0rrr+rRrrr8rr`Hs  r`)r0ZloggingZnumbersrZ matplotlibrZmatplotlib.backend_basesrZmatplotlib.lineslinesrJZwarn_deprecatedr-Z getLoggerrrr1rSr`rrrrs     E):