B c+@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||_||_dS)N)fig eventslist)selfr r rr8lib/python3.7/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_loop)r eventrrr on_event(s   zBlockingInput.on_eventcCsdS)z-For baseclass, do nothing but collect events.Nr)r rrr r8szBlockingInput.post_eventcCs(x|jD]}|jj|qWg|_dS)zDisconnect all callbacks.N) callbacksr rZmpl_disconnect)r cbrrr cleanup;s zBlockingInput.cleanupcCs|j|dS)z5For base class, this just appends an event to events.N)rappend)r rrrr rAszBlockingInput.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)r indexrrr pop_eventEszBlockingInput.pop_eventcsjtjt|d|_g_jjjr.jfddj D_ zjjj |dWd XjS)z%Blocking call to retrieve *n* events.)rcsg|]}jj|jqSr)r rZ mpl_connectr).0r)r rr Zsz*BlockingInput.__call__..)timeoutN) rZcheck_isinstancerrrr rZmanagerZshowr rZstart_event_loopr)r rr%r)r r __call__Ps     zBlockingInput.__call__N)r)r)r!r") __name__ __module__ __qualname____doc__r rrrrr rr&rrrr r!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||_||_||_dS)N)button_press_eventkey_press_event)r r )superr button_add button_pop button_stop)r r Z mouse_addZ mouse_popZ mouse_stop) __class__rr r ss zBlockingMouseInput.__init__cCs@t|jdkrtdn"|jdjdkr4|n|dS)zProcess an event.rz No events yetrr-N)rrrwarningr key_event mouse_event)r rrr r}s   zBlockingMouseInput.post_eventcCsT|jd}|j}||jkr&||n*||jkr<||n||jkrP||dS)zProcess a mouse click event.rN)rbuttonr0mouse_event_popr1mouse_event_stopr/mouse_event_add)r rr6rrr r5s      zBlockingMouseInput.mouse_eventcCsR|jd}|jdkrdS|jdkr.||n |jdkrD||n ||dS)zV Process a key press event, mapping keys to appropriate mouse clicks. rN)Z backspacedelete)escapeZenter)rkeyr7r8r9)r rrrr r4s      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)r rrrr r9s z"BlockingMouseInput.mouse_event_addcCst||jjdS)z Process an button-2 event (end blocking input). Parameters ---------- event : `~.backend_bases.MouseEvent` N)rrr rr)r rrrr r8s z#BlockingMouseInput.mouse_event_stopcCst||jr||dS)z Process an button-3 event (remove the last click). Parameters ---------- event : `~.backend_bases.MouseEvent` N)rrr)r rrrr r7s 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)ZmarkerZcolorN)clicksrZxdataZydatarrr show_clicksmlinesZLine2Dr=Zadd_linemarksr rdraw)r rlinerrr r>s   zBlockingMouseInput.add_clickrcCs2|j||jr.|j||jjdS)z Remove a click (by default, the last) from the list of clicks. Parameters ---------- event : `~.backend_bases.MouseEvent` N)rArrBrDremover rrE)r rrrrr 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)rHr.r)r rr)r2rr rs zBlockingMouseInput.popNcs>|jr0x|jD] }|qWg|_|jjtdS)z{ Parameters ---------- event : `~.backend_bases.MouseEvent`, optional Not used N)rBrDrGr rrEr.r)r rZmark)r2rr rs    zBlockingMouseInput.cleanupr!r"Tcs(||_g|_g|_tj||d|jS)zV Blocking call to retrieve *n* coordinate pairs through mouse clicks. )rr%)rBrArDr.r&)r rr%rB)r2rr r&s zBlockingMouseInput.__call__)r)r)N)r!r"T)r'r(r)r*rZLEFTr/ZRIGHTr0ZMIDDLEr1r rr5r4r9r8r7r>rHrrr& __classcell__rr)r2r r+fs$     r+csPeZdZdZfddZddZdddZd 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 )csr.r Zaxesfigure)r rK)r2rr r szBlockingContourLabeler.__init__cCs||dS)N)button1)r rrrr r>sz BlockingContourLabeler.add_clickrcCs||dS)N)button3)r rrrrr rHsz 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) r=rKaxZadd_label_nearxyinlinerOr rrErr)r rrrr rMs 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)rSrKZ pop_labelrPrLrrE)r rrrr rN0s  zBlockingContourLabeler.button3cs"||_||_tj||dddS)NF)rr%rB)rSrOr.r&)r rSrOrr%)r2rr r&BszBlockingContourLabeler.__call__)r)rTrr) r'r(r)r*r r>rHrMrNr&rIrr)r2r rJs  rJcs6eZdZdZfddZddZd fdd ZZS) BlockingKeyMouseInputzQ Callable for retrieving mouse clicks and key presses in a blocking way. cstj|dddS)N)r,r-)r r )r.r )r r )r2rr r MszBlockingKeyMouseInput.__init__cCs(|jr|jdjdk|_n tddS)zDetermine if it is a key event.rr-zNo events yet.N)rr keyormouserr3)r rrr rQsz 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!)rr%)rVr.r&)r r%)r2rr r&XszBlockingKeyMouseInput.__call__)r")r'r(r)r*r rr&rIrr)r2r rUHs rU)r*ZloggingZnumbersrZ matplotlibrZmatplotlib.backend_basesrZmatplotlib.lineslinesrCZwarn_deprecatedr'Z getLoggerrrr+rJrUrrrr s     E):