a r=ic+@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)nameobj_typec@sReZdZdZdddZddZddZd d Zd d ZdddZ e Z dddZ dS) BlockingInputz1Callable for retrieving events in a blocking way.cCs||_||_dSNfig eventslist)selfr r r r j/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/matplotlib/blocking_input.py__init__$szBlockingInput.__init__cCsT||tdt|j|t|j|jkr@dkrPnn |jj dS)zY Event handler; will be passed to the current figure to retrieve events. zEvent %irN) add_event_loginfolenevents post_eventnr canvasstop_event_loopreventr r ron_event(s   zBlockingInput.on_eventcCsdS)z-For baseclass, do nothing but collect events.Nr rr r rr8szBlockingInput.post_eventcCs$|jD]}|jj|qg|_dS)zDisconnect all callbacks.N) callbacksr rZmpl_disconnect)rcbr r rcleanup;s zBlockingInput.cleanupcCs|j|dS)z5For base class, this just appends an event to events.N)rappendrr r rrAszBlockingInput.add_eventcCs|j|dS)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)rindexr r r pop_eventEszBlockingInput.pop_eventcsrtjt|d|_g_jjjr.jfddj D_ zjjj |dW n 0jS)z%Blocking call to retrieve *n* events.)rcsg|]}jj|jqSr )r rZ mpl_connectr).0rrr r Zsz*BlockingInput.__call__..)timeout) rZcheck_isinstancerrrr rmanagershowr rZstart_event_loopr )rrr*r rr__call__Ps   zBlockingInput.__call__N)r )r")r&r') __name__ __module__ __qualname____doc__rrrr rr%r#r-r r r rr!s  rcseZdZdZejZejZej Z ejejej ffdd Z ddZ ddZ dd Zd d Zd d ZddZddZdddZd fdd 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)rr Z mouse_addZ mouse_popZ mouse_stop __class__r rrss zBlockingMouseInput.__init__cCs@t|jdkrtdn"|jdjdkr4|n|dS)zProcess an event.rz No events yetr"r4N)rrrwarningr key_event mouse_eventrr r rr}s   zBlockingMouseInput.post_eventcCsT|jd}|j}||jkr&||n*||jkr<||n||jkrP||dS)zProcess a mouse click event.r"N)rbuttonr7mouse_event_popr8mouse_event_stopr6mouse_event_add)rrr>r r rr=s      zBlockingMouseInput.mouse_eventcCsR|jd}|jdurdS|jdvr.||n |jdvrD||n ||dS)zV Process a key press event, mapping keys to appropriate mouse clicks. r"N)Z backspacedelete)escapeenter)rkeyr?r@rArr r rr<s      zBlockingMouseInput.key_eventcCs |jr||n t|dS)z Process an button-1 event (add a click if inside axes). Parameters ---------- event : `~.backend_bases.MouseEvent` N)inaxes add_clickrr#rr r rrAs z"BlockingMouseInput.mouse_event_addcCst||jjdS)z Process an button-2 event (end blocking input). Parameters ---------- event : `~.backend_bases.MouseEvent` N)rr#r rrrr r rr@s z#BlockingMouseInput.mouse_event_stopcCst||jr||dS)z Process an button-3 event (remove the last click). Parameters ---------- event : `~.backend_bases.MouseEvent` N)rr#rrr r rr?s z"BlockingMouseInput.mouse_event_popcCsv|j|j|jftdt|j|j|j|jrrtj |jg|jgddd}|j ||j ||j jdS)z Add the coordinates of an event to the list of clicks. Parameters ---------- event : `~.backend_bases.MouseEvent` zinput %i: %f, %f+r)markercolorN)clicksr!ZxdataZydatarrr show_clicksmlinesZLine2DrFadd_linemarksr rdraw)rrliner r rrGs  zBlockingMouseInput.add_clickr"cCs2|j||jr.|j||jjdS)z Remove a click (by default, the last) from the list of clicks. Parameters ---------- event : `~.backend_bases.MouseEvent` N)rLr#rMrPremover rrQrrr$r r r 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)rUr5r#rTr9r rr#s zBlockingMouseInput.popNcs:|jr,|jD] }|q g|_|jjtdS)z{ Parameters ---------- event : `~.backend_bases.MouseEvent`, optional Not used N)rMrPrSr rrQr5r )rrmarkr9r rr s    zBlockingMouseInput.cleanupr&r'Tcs(||_g|_g|_tj||d|jS)zV Blocking call to retrieve *n* coordinate pairs through mouse clicks. rr*)rMrLrPr5r-)rrr*rMr9r rr-s zBlockingMouseInput.__call__)r")r")N)r&r'T)r.r/r0r1rLEFTr6RIGHTr7ZMIDDLEr8rrr=r<rAr@r?rGrUr#r r- __classcell__r r r9rr2fs&      r2csPeZdZdZfddZddZdddZd 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 )csr5rZaxesfigure)rr\r9r rrszBlockingContourLabeler.__init__cCs||dSr )button1rr r rrGsz BlockingContourLabeler.add_clickr"cCs||dSr )button3rTr r rrUsz BlockingContourLabeler.pop_clickcCsH|j|jjkr:|jj|j|j|j|jdd|jj n t |dS)z Process an button-1 event (add a label to a contour). Parameters ---------- event : `~.backend_bases.MouseEvent` F)inline_spacingZ transformN) rFr\axZadd_label_nearxyinliner`r rrQrr#rr r rr^s zBlockingContourLabeler.button1cCs&|jrn|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)rdr\Z pop_labelrar]rrQrr r rr_0s  zBlockingContourLabeler.button3cs"||_||_tj||dddS)NF)rr*rM)rdr`r5r-)rrdr`rr*r9r rr-BszBlockingContourLabeler.__call__)r")rer"r") r.r/r0r1rrGrUr^r_r-rZr r r9rr[s  r[cs6eZdZdZfddZddZd fdd ZZS) BlockingKeyMouseInputzQ Callable for retrieving mouse clicks and key presses in a blocking way. cstj|dddSr3)r5r)rr r9r rrMszBlockingKeyMouseInput.__init__cCs(|jr|jdjdk|_n tddS)zDetermine if it is a key event.r"r4zNo events yet.N)rr keyormouserr;rr r rrQsz 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&rW)rgr5r-)rr*r9r rr-XszBlockingKeyMouseInput.__call__)r')r.r/r0r1rrr-rZr r r9rrfHs rf)r1loggingnumbersrZ matplotlibrZmatplotlib.backend_basesrZmatplotlib.lineslinesrNZwarn_deprecatedr. getLoggerrrr2r[rfr r r rs     E):