B ]t\h+@srdZddlZddlmZddlmZeeZ Gddde Z Gddde Z Gdd d e Z Gd d d e ZdS) a This provides several 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 `waitforbuttonpress`. `BlockingMouseInput` Creates a callable object to retrieve mouse clicks in a blocking way for interactive sessions. Used by `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 `clabel`. N)Integralc@sReZdZdZdddZddZddZd d Zd d ZdddZ e Z dddZ dS) BlockingInputz1Callable for retrieving events in a blocking way.cCs||_||_dS)N)fig eventslist)selfrrrr8lib/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_eventnrcanvasstop_event_loop)reventrrron_event%s   zBlockingInput.on_eventcCsdS)z-For baseclass, do nothing but collect events.Nr)rrrrr5szBlockingInput.post_eventcCs(x|jD]}|jj|qWg|_dS)zDisconnect all callbacks.N) callbacksrrZmpl_disconnect)rcbrrrcleanup8s zBlockingInput.cleanupcCs|j|dS)z5For base class, this just appends an event to events.N)rappend)rrrrrr >szBlockingInput.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)rindexrrr pop_eventBszBlockingInput.pop_eventcsrt|tstd|_g_tjjdr6jfddj D_ zjjj |dWd XjS)z%Blocking call to retrieve *n* events.zRequires an integer argumentZmanagercsg|]}jj|jqSr)rrZ mpl_connectr).0name)rrr Xsz*BlockingInput.__call__..)timeoutN) isinstancer ValueErrorrrhasattrrrZshowrrZstart_event_loopr)rrr"r)rr__call__Ms     zBlockingInput.__call__N)r)r)rr) __name__ __module__ __qualname____doc__r rrrr rrr&rrrrrs  rc@seZdZdZdZdZdZd!ddZddZd d Z d d Z d dZ ddZ ddZ ddZd"ddZd#ddZd$ddZd%dd ZdS)&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 like mouse button 3, enter is like mouse button 2 and all others are like mouse button 1. rcCs&tj||dd||_||_||_dS)N)button_press_eventkey_press_event)rr)rr button_add button_pop button_stop)rrZ mouse_addZ mouse_popZ mouse_stoprrrr qs zBlockingMouseInput.__init__cCs@t|jdkrtdn"|jdjdkr4|n|dS)zProcess an event.rz No events yetrr/N)r rr warningr key_event mouse_event)rrrrrys   zBlockingMouseInput.post_eventcCsJ|jd}|j}||jkr&||n ||jkr<||n ||dS)zProcess a mouse click event.rN)rbuttonr1mouse_event_popr2mouse_event_stopmouse_event_add)rrr6rrrr5s     zBlockingMouseInput.mouse_eventcCsX|jd}|jdkrdS|j}|dkr6||n|dkrJ||n ||dS)zV Process a key press event, mapping keys to appropriate mouse clicks. rN)Z backspacedelete)escapeZenter)rkeylowerr7r8r9)rrr<rrrr4s     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)rrrrrr9s z"BlockingMouseInput.mouse_event_addcCst||jjdS)z Process an button-2 event (end blocking input). Parameters ---------- event : `~.backend_bases.MouseEvent` N)rrrrr)rrrrrr8s z#BlockingMouseInput.mouse_event_stopcCst||jr||dS)z Process an button-3 event (remove the last click). Parameters ---------- event : `~.backend_bases.MouseEvent` N)rrr)rrrrrr7s 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)clicksrZxdataZydatar r r show_clicksmlinesZLine2Dr>Zadd_linemarksrrdraw)rrlinerrrr?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)rBrrCrEremoverrrF)rrrrrr pop_clicks zBlockingMouseInput.pop_clickcCs|||t||dS)zx Removes a click and the associated event from the list of clicks. Defaults to the last click. N)rIrr)rrrrrrrs zBlockingMouseInput.popNcCs>|jr0x|jD] }|qWg|_|jjt|dS)z{ Parameters ---------- event : `~.backend_bases.MouseEvent`, optional Not used N)rCrErHrrrFrr)rrZmarkrrrrs    zBlockingMouseInput.cleanuprTcCs(||_g|_g|_tj|||d|jS)zV Blocking call to retrieve *n* coordinate pairs through mouse clicks. )rr")rCrBrErr&)rrr"rCrrrr&s zBlockingMouseInput.__call__)rr,r-)r)r)N)rrT)r'r(r)r*r0r1r2r rr5r4r9r8r7r?rIrrr&rrrrr+ds      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)rrKrrrr szBlockingContourLabeler.__init__cCs||dS)N)button1)rrrrrr?sz BlockingContourLabeler.add_clickrcCs||dS)N)button3)rrrrrrrIsz 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>rKrLZadd_label_nearxyinlinerPrrrFrr)rrrrrrNs 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_labelrLrMrrF)rrrrrrO-s  zBlockingContourLabeler.button3cCs"||_||_tj|||dddS)NF)rr"rC)rSrPr+r&)rrSrPrr"rrrr&?s zBlockingContourLabeler.__call__N)r)rTrr) r'r(r)r*r r?rIrNrOr&rrrrrJ s rJc@s*eZdZdZddZddZd ddZd S) BlockingKeyMouseInputzQ Callable for retrieving mouse clicks and key presses in a blocking way. cCstj||dddS)N)r.r/)rr)rr )rrrrrr KszBlockingKeyMouseInput.__init__cCs(|jr|jdjdk|_n tddS)zDetermine if it is a key event.rr/zNo events yet.N)rr keyormouser r3)rrrrrOsz 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")rVrr&)rr"rrrr&VszBlockingKeyMouseInput.__call__N)r)r'r(r)r*r rr&rrrrrUFsrU)r*ZloggingZnumbersrZmatplotlib.lineslinesrDZ getLoggerr'r objectrr+rJrUrrrrs   F(;