3 ƽh%,@sdZddlZddlmZddlmZddlmZddlj Z ej 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@sTeZdZdZffddZddZddZdd Zd d Zdd dZ e Z dddZ dS) BlockingInputz1Callable for retrieving events in a blocking way.cCs||_||_dS)N)fig eventslist)selfrrr ?/tmp/pip-build-7iwl8md4/matplotlib/matplotlib/blocking_input.py__init__#szBlockingInput.__init__cCsT|j|tjdt|j|jt|j|jko>dknrP|jjj dS)zY Event handler; will be passed to the current figure to retrieve events. zEvent %irN) add_event_loginfolenevents post_eventnrcanvasstop_event_loop)reventr r r on_event's   zBlockingInput.on_eventcCsdS)z-For baseclass, do nothing but collect events.Nr )rr r r r7szBlockingInput.post_eventcCs(x|jD]}|jjj|qWg|_dS)zDisconnect all callbacks.N) callbacksrrZmpl_disconnect)rcbr r r cleanup:s zBlockingInput.cleanupcCs|jj|dS)z5For base class, this just appends an event to events.N)rappend)rrr r r r @szBlockingInput.add_eventcCs|jj|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_eventDszBlockingInput.pop_eventc sntjt|d|_g_tjjdr2jjfddj D_ zjjj |dWdj XjS)z%Blocking call to retrieve *n* events.)rmanagercsg|]}jjj|jqSr )rrZ mpl_connectr).0name)rr r Ysz*BlockingInput.__call__..)timeoutN) rZ_check_isinstancerrrhasattrrrshowrrZstart_event_loopr)rrr$r )rr __call__Os    zBlockingInput.__call__N)r()rr) __name__ __module__ __qualname____doc__r rrrr rrr'r r r r r s  rc@seZdZdZejZejZej Z ejejej fddZ ddZ ddZ dd Zd d Zd d ZddZddZdddZd!ddZd"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||_||_||_dS)Nbutton_press_eventkey_press_event)rr)r.r/)rr button_add button_pop button_stop)rrZ mouse_addZ mouse_popZ mouse_stopr r r r rs zBlockingMouseInput.__init__cCs@t|jdkrtjdn"|jdjdkr4|jn|jdS)zProcess an event.rz No events yetrr/Nr()rrr warningr" key_event mouse_event)rr r r r}s   zBlockingMouseInput.post_eventcCsT|jd}|j}||jkr&|j|n*||jkr<|j|n||jkrP|j|dS)zProcess a mouse click event.rNr()rbuttonr1mouse_event_popr2mouse_event_stopr0mouse_event_add)rrr6r r r r5s      zBlockingMouseInput.mouse_eventcCsX|jd}|jdkrdS|jj}|dkr6|j|n|d krJ|j|n |j|dS) zV Process a key press event, mapping keys to appropriate mouse clicks. rN backspacedeleteescapeenterr()r:r;)r<r=)rkeylowerr7r8r9)rrr>r r r r4s     zBlockingMouseInput.key_eventcCs |jr|j|n tj|dS)z Process an button-1 event (add a click if inside axes). Parameters ---------- event : `~.backend_bases.MouseEvent` N)inaxes add_clickrr)rrr r r r9s z"BlockingMouseInput.mouse_event_addcCstj||jjjdS)z Process an button-2 event (end blocking input). Parameters ---------- event : `~.backend_bases.MouseEvent` N)rrrrr)rrr r r r8s z#BlockingMouseInput.mouse_event_stopcCstj||jr|j|dS)z Process an button-3 event (remove the last click). Parameters ---------- event : `~.backend_bases.MouseEvent` N)rrr)rrr r r r7s z"BlockingMouseInput.mouse_event_popcCsv|jj|j|jftjdt|j|j|j|jrrtj |jg|jgddd}|j j ||j j||j jjdS)z Add the coordinates of an event to the list of clicks. Parameters ---------- event : `~.backend_bases.MouseEvent` zinput %i: %f, %f+r)markercolorN)clicksrZxdataZydatar rr show_clicksmlinesZLine2Dr@Zadd_linemarksrrdraw)rrliner r r rAs   zBlockingMouseInput.add_clickrcCs2|jj||jr.|jj|j|jjjdS)z Remove a click (by default, the last) from the list of clicks. Parameters ---------- event : `~.backend_bases.MouseEvent` N)rFrrGrIremoverrrJ)rrrr r r pop_clicks zBlockingMouseInput.pop_clickcCs|j||tj||dS)zw Remove a click and the associated event from the list of clicks. Defaults to the last click. N)rMrr)rrrr r r rs zBlockingMouseInput.popNcCs>|jr0x|jD] }|jqWg|_|jjjtj|dS)z{ Parameters ---------- event : `~.backend_bases.MouseEvent`, optional Not used N)rGrIrLrrrJrr)rrmarkr r r rs    zBlockingMouseInput.cleanuprTcCs(||_g|_g|_tj|||d|jS)zV Blocking call to retrieve *n* coordinate pairs through mouse clicks. )rr$)rGrFrIrr')rrr$rGr r r r's zBlockingMouseInput.__call__r()r(r()r()N)rrT)r)r*r+r,rLEFTr0RIGHTr1ZMIDDLEr2r rr5r4r9r8r7rArMrrr'r r r r r-es$    r-c@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)csr-r axfigure)rrRr r r r szBlockingContourLabeler.__init__cCs|j|dS)N)button1)rrr r r rAsz BlockingContourLabeler.add_clickrcCs|j|dS)N)button3)rrrr r r rMsz BlockingContourLabeler.pop_clickcCsH|j|jjkr:|jj|j|j|j|jdd|jj j n t j |dS)z Process an button-1 event (add a label to a contour). Parameters ---------- event : `~.backend_bases.MouseEvent` F)inline_spacingZ transformN) r@rRrSZadd_label_nearxyinlinerWrrrJrr)rrr r r rU s zBlockingContourLabeler.button1cCs&|jrn|jj|jjjjjdS)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)rZrRZ pop_labelrSrTrrJ)rrr r r rV1s  zBlockingContourLabeler.button3cCs"||_||_tj|||dddS)NF)rr$rG)rZrWr-r')rrZrWrr$r r r r'Cs zBlockingContourLabeler.__call__Nr()r(r(r()r[r(r() r)r*r+r,r rArMrUrVr'r r r r rQs rQc@s*eZdZdZddZddZd ddZd S) BlockingKeyMouseInputzQ Callable for retrieving mouse clicks and key presses in a blocking way. cCstj||dddS)Nr.r/)rr)r.r/)rr )rrr r r r OszBlockingKeyMouseInput.__init__cCs(|jr|jdjdk|_n tjddS)zDetermine if it is a key event.rr/zNo events yet.Nr()rr" keyormouser r3)rr r r rSsz BlockingKeyMouseInput.post_eventrcCsd|_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)rr$)r]rr')rr$r r r r'ZszBlockingKeyMouseInput.__call__N)r)r)r*r+r,r rr'r r r r r\Jsr\)r,loggingnumbersrZ matplotlibrZmatplotlib.backend_basesrZmatplotlib.lineslinesrH getLoggerr)r rr-rQr\r r r r s     E+;