o i%,@sdZddlZddlmZddlmZddlmZddlm Z e e Z GdddZGdd d eZGd d d eZGd d d eZdS)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)cbook) MouseButtonc@sReZdZdZdddZddZddZd d Zd d ZdddZ e Z dddZ dS) BlockingInputz1Callable for retrieving events in a blocking way.cCs||_||_dSNfig eventslist)selfr r rrT/var/www/edux/Edux_v2/venv/lib/python3.10/site-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 eventrrr on_event's "zBlockingInput.on_eventcCsdS)z-For baseclass, do nothing but collect events.Nrr rrr r7szBlockingInput.post_eventcCs$|jD] }|jj|qg|_dS)zDisconnect all callbacks.N) callbacksr rmpl_disconnect)r cbrrr cleanup:s  zBlockingInput.cleanupcC|j|dS)z5For base class, this just appends an event to events.N)rappendrrrr r@szBlockingInput.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_eventDszBlockingInput.pop_eventcsttjt|d|_g_tjjdrjfddj D_ zjjj |dW jS w)z%Blocking call to retrieve *n* events.)rmanagercsg|] }jj|jqSr)r r mpl_connectr).0namerrr Ysz*BlockingInput.__call__..)timeout) r_check_isinstancerrrhasattrr rshowr rstart_event_loopr)r rr,rrr __call__Os   zBlockingInput.__call__N)rr!)r%r&) __name__ __module__ __qualname____doc__r rrrrr$r"r1rrrr r s   rc@seZdZdZejZejZej Z ejejej fddZ ddZ ddZ dd Zd d Zd d ZddZddZdddZdddZd ddZd!ddZdS)"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. cCs&tj||dd||_||_||_dSN)button_press_eventkey_press_eventr)rr button_add button_pop button_stop)r r mouse_add mouse_pop mouse_stoprrr r rs  zBlockingMouseInput.__init__cCsDt|jdkrtddS|jdjdkr|dS|dS)zProcess an event.rz No events yetr!r:N)rrrwarningr* key_event mouse_eventrrrr r}s   zBlockingMouseInput.post_eventcCs\|jd}|j}||jkr||dS||jkr ||dS||jkr,||dSdS)zProcess a mouse click event.r!N)rbuttonr<mouse_event_popr=mouse_event_stopr;mouse_event_add)r rrDrrr rCs    zBlockingMouseInput.mouse_eventcCs\|jd}|jdur dS|j}|dvr||dS|dvr'||dS||dS)zV Process a key press event, mapping keys to appropriate mouse clicks. r!N) backspacedelete)escapeenter)rkeylowerrErFrG)r rrLrrr rBs   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"rrrr rGsz"BlockingMouseInput.mouse_event_addcCst||jjdS)z Process an button-2 event (end blocking input). Parameters ---------- event : `~.backend_bases.MouseEvent` N)rr"r rrrrrr rFs 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"rrrrr rEs 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)markercolorN)clicksr xdataydatarrr show_clicksmlinesLine2DrNadd_linemarksr rdraw)r rlinerrr rOs  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)rTr"rWr[remover rr\r rr#rrr pop_clicks zBlockingMouseInput.pop_clickcCs|||t||dS)zw Remove a click and the associated event from the list of clicks. Defaults to the last click. N)r`rr"r_rrr r"s zBlockingMouseInput.popNcCs:|jr|jD]}|qg|_|jjt|dS)z{ Parameters ---------- event : `~.backend_bases.MouseEvent`, optional Not used N)rWr[r^r rr\rr)r rmarkrrr rs    zBlockingMouseInput.cleanupr%r&TcCs(||_g|_g|_tj|||d|jS)zV Blocking call to retrieve *n* coordinate pairs through mouse clicks. rr,)rWrTr[rr1)r rr,rWrrr r1s zBlockingMouseInput.__call__r2r)r%r&T)r3r4r5r6rLEFTr;RIGHTr<MIDDLEr=r rrCrBrGrFrErOr`r"rr1rrrr r7es(       r7c@sDeZdZdZddZddZdddZd d Zd d ZdddZ dS)BlockingContourLabelerzt Callable for retrieving mouse clicks and key presses in a blocking way. Used to place contour labels. cCs||_tj||jjddS)N)r )csr7r axfigure)r rgrrr r szBlockingContourLabeler.__init__cC||dSr)button1rrrr rOz BlockingContourLabeler.add_clickr!cCrjr)button3r_rrr r`rlz 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_spacing transformN) rNrgrhadd_label_nearxyinlinernr rr\rr"rrrr rk s 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)rsrg pop_labelrhrirr\rrrr rm1s  zBlockingContourLabeler.button3cCs"||_||_tj|||dddS)NF)rr,rW)rsrnr7r1)r rsrnrr,rrr r1Cs   zBlockingContourLabeler.__call__Nr2)rur!r!) r3r4r5r6r rOr`rkrmr1rrrr rfs rfc@s*eZdZdZddZddZd ddZd S) BlockingKeyMouseInputzQ Callable for retrieving mouse clicks and key presses in a blocking way. cCstj||dddSr8)rr )r r rrr r OszBlockingKeyMouseInput.__init__cCs*|jr|jdjdk|_dStddS)zDetermine if it is a key event.r!r:zNo events yet.N)rr* keyormouserrArrrr rSsz BlockingKeyMouseInput.post_eventr&cCsd|_tj|d|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%rb)rwrr1)r r,rrr r1ZszBlockingKeyMouseInput.__call__N)r&)r3r4r5r6r rr1rrrr rvJs rv)r6loggingnumbersr matplotlibrmatplotlib.backend_basesrmatplotlib.lineslinesrX getLoggerr3rrr7rfrvrrrr s     E+;