a ضa+@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||_||_dSNfig eventslist)selfr r rri/Users/vegardjervell/Documents/master/model/venv/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_loopr eventrrron_event(s   zBlockingInput.on_eventcCsdS)z-For baseclass, do nothing but collect events.Nrr rrrr8szBlockingInput.post_eventcCs$|jD]}|jj|qg|_dS)zDisconnect all callbacks.N) callbacksr rZmpl_disconnect)r cbrrrcleanup;s zBlockingInput.cleanupcCs|j|dS)z5For base class, this just appends an event to events.N)rappendrrrrrAszBlockingInput.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_eventcsrtjt|d|_g_jjjr.jfddj D_ zjjj |dW n 0jS)z%Blocking call to retrieve *n* events.)rcsg|]}jj|jqSr)r rZ mpl_connectr).0rrrr Zsz*BlockingInput.__call__..)timeout) rZcheck_isinstancerrrr rmanagershowr rZstart_event_loopr)r rr)rrr__call__Ps   zBlockingInput.__call__N)r)r!)r%r&) __name__ __module__ __qualname____doc__rrrrrr$r"r,rrrrr!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)r r Z mouse_addZ mouse_popZ mouse_stop __class__rrrss zBlockingMouseInput.__init__cCs@t|jdkrtdn"|jdjdkr4|n|dS)zProcess an event.rz No events yetr!r3N)rrrwarningr key_event mouse_eventrrrrr}s   zBlockingMouseInput.post_eventcCsT|jd}|j}||jkr&||n*||jkr<||n||jkrP||dS)zProcess a mouse click event.r!N)rbuttonr6mouse_event_popr7mouse_event_stopr5mouse_event_add)r rr=rrrr<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)escapeZenter)rkeyr>r?r@rrrrr;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"rrrrr@s z"BlockingMouseInput.mouse_event_addcCst||jjdS)z Process an button-2 event (end blocking input). Parameters ---------- event : `~.backend_bases.MouseEvent` N)rr"r rrrrrrr?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"rrrrrr>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_clicksmlinesZLine2DrDZadd_linemarksr rdraw)r rlinerrrrEs  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)rJr"rKrMremover rrNr rr#rrr 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)rRr4r"rQr8rrr"s zBlockingMouseInput.popNcs:|jr,|jD] }|q g|_|jjtdS)z{ Parameters ---------- event : `~.backend_bases.MouseEvent`, optional Not used N)rKrMrPr rrNr4r)r rmarkr8rrrs    zBlockingMouseInput.cleanupr%r&Tcs(||_g|_g|_tj||d|jS)zV Blocking call to retrieve *n* coordinate pairs through mouse clicks. rr))rKrJrMr4r,)r rr)rKr8rrr,s zBlockingMouseInput.__call__)r!)r!)N)r%r&T)r-r.r/r0rLEFTr5RIGHTr6ZMIDDLEr7rrr<r;r@r?r>rErRr"rr, __classcell__rrr8rr1fs&      r1csPeZdZdZfddZddZdddZd 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 )csr4rZaxesfigure)r rYr8rrrszBlockingContourLabeler.__init__cCs||dSr )button1rrrrrEsz BlockingContourLabeler.add_clickr!cCs||dSr )button3rQrrrrRsz 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) rDrYaxZadd_label_nearxyinliner]r rrNrr"rrrrr[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)rarYZ pop_labelr^rZrrNrrrrr\0s  zBlockingContourLabeler.button3cs"||_||_tj||dddS)NF)rr)rK)rar]r4r,)r rar]rr)r8rrr,BszBlockingContourLabeler.__call__)r!)rbr!r!) r-r.r/r0rrErRr[r\r,rWrrr8rrXs  rXcs6eZdZdZfddZddZd fdd ZZS) BlockingKeyMouseInputzQ Callable for retrieving mouse clicks and key presses in a blocking way. cstj|dddSr2)r4r)r r r8rrrMszBlockingKeyMouseInput.__init__cCs(|jr|jdjdk|_n tddS)zDetermine if it is a key event.r!r3zNo events yet.N)rr keyormouserr:rrrrrQsz 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%rT)rdr4r,)r r)r8rrr,XszBlockingKeyMouseInput.__call__)r&)r-r.r/r0rrr,rWrrr8rrcHs rc)r0loggingZnumbersrZ matplotlibrZmatplotlib.backend_basesrZmatplotlib.lineslinesrLZwarn_deprecatedr- getLoggerrrr1rXrcrrrrs     E):