a s=ic@sddlZddlmZddlZddlZddlZddlZddlZ ddl Zddl Zddl Zddl mZddlZddlmZmZddlZddlmZmZmZmZddlmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'ddl(m)Z)dd l*m+Z+e,e-Z.e j/Z0e!j1d e!j2d e!j3d e!j4d e!j5de!j6de!j7diZ8eddZ9iZ:de;j<Z=dZ>dZ?ddZ@d3ddZAGdddeZBGdddeZCGdddeZDGdddee jEZFGd d!d!ZGeHeCGd"d#d#ejIZJejKd$d%d&Gd'd(d(ejLZMGd)d*d*e e jEZNeHeCGd+d,d,ejOZPeHeCGd-d.d.ejQZReHeCGd/d0d0ejSZTeNZUeFeD_VeNeD_WejXGd1d2d2eZYdS)4N)contextmanager) SimpleDialog)ImageImageTk)_api backend_toolscbook_c_internal_utils) _BackendFigureCanvasBaseFigureManagerBaseNavigationToolbar2 TimerBaseToolContainerBasecursors_Mode CloseEventKeyEvent LocationEvent MouseEvent ResizeEvent)Gcf)_tkaggZfleurZhand2arrowZtcrosswatchZsb_h_double_arrowZsb_v_double_arrowc csBt}zdVWtjdr>t|ntjdrz#TimerTk._on_timer..)rKrUZ_singlerJrTrMrS after_idlerVrPrVr rUs    zTimerTk._on_timer) __name__ __module__ __qualname____doc__rLrWrRrU __classcell__rrrPr rH~s  rHcseZdZdZeddZd3fdd Zd4ddZd d Z d d Z d dZ ddZ ddZ ddZddZd5ddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd6d-d.Zd/d0Zd1d2ZZS)7FigureCanvasTkr%cCstSrI)FigureManagerTk)clsrrr rYrZzFigureCanvasTk.Ncst|d_d_jdd\}}tj|d||ddd_tjj||d_ jj |d|dj dj d j t jd krj d jj d jj d jj djj djj djdD]}j |jqdD]}j |jqdD]}j |jq$dD]}j |jq@j}| djdfdd}| d|djd_dS)NTZphysicalwhiter)master backgroundrBrA borderwidthZhighlightthicknessrfrBrAr/imagez win32zzzz )z z z )zzz)zzz)z z z +cs|jjurtddS)NZ close_event)widget _tkcanvasr_processeventrVrr filter_destroys z/FigureCanvasTk.__init__..filter_destroyz )rKrL _idle_draw_id_event_loop_idget_width_heightr%ZCanvasrq PhotoImage_tkphoto create_imagebindresizesysplatform_update_device_pixel_ratio key_pressmotion_notify_evententer_notify_eventleave_notify_event key_releasebutton_press_eventbutton_dblclick_eventbutton_release_event scroll_eventZwinfo_toplevelscroll_event_windows focus_set_rubberband_rect)rNfigurerfwhnamerootrurPrVr rLsD     zFigureCanvasTk.__init__cCsHt|jjdddd}||rD|jdd\}}|jj||ddS)Nr%scalinggUUUUUU?r/TrdrBrA)roundrqr%r=Z_set_device_pixel_ratiorx configure)rNrtratiorrrrr rs z)FigureCanvasTk._update_device_pixel_ratiocCs|j|j}}|jj}||}||}|jj||dd|j|jtj |jt |t |d|_|jj t |dt |d|jdt d| |dS)NF)forwardrir/rjZ resize_event)rBrArZdpiZset_size_inchesrqdeleterzr%ryintr{rrr draw_idle)rNrtrBrAZdpivalZwinchZhinchrrr r}szFigureCanvasTk.resizecs(jr dSfdd}j|_dS)Ncs zWd_nd_0dSrI)ZdrawrvrFrVrr idle_draws z+FigureCanvasTk.draw_idle..idle_draw)rvrqr[)rNrrrVr rs zFigureCanvasTk.draw_idlecCs|jS)z Return the Tk widget used to implement FigureCanvasTkAgg. Although the initial implementation uses a Tk canvas, this routine is intended to hide that fact. )rqrVrrr get_tk_widgetszFigureCanvasTk.get_tk_widgetcCs&|j|j|jjj|j|jfSrI)rqcanvasxxrr@rAcanvasyyrNrtrrr _event_mpl_coords s z FigureCanvasTk._event_mpl_coordscCs&td|g||Rd|idS)NrguiEvent)rrrrrrrr rs z"FigureCanvasTk.motion_notify_eventcCs&td|g||Rd|idS)NZfigure_enter_eventrrrrrrrrr rs z!FigureCanvasTk.enter_notify_eventcCs&td|g||Rd|idS)NZfigure_leave_eventrrrrrr rs z!FigureCanvasTk.leave_notify_eventFcCs^|jt|dd}tjdkr2ddd||}td|g|||R||ddS)Nnumdarwinr/r/rr)dblclickr) rqrgetattrr~rgetrrrr)rNrtrrrrr r"s   z!FigureCanvasTk.button_press_eventcCs|j|dddS)NT)r)rrrrr r-sz$FigureCanvasTk.button_dblclick_eventcCsRt|dd}tjdkr(ddd||}td|g|||Rd|idS)Nrrrr/rrr)rr~rrrrrr)rNrtrrrr r0s  z#FigureCanvasTk.button_release_eventcCsPt|dd}|dkrdn|dkr$dnd}td|g||R||ddS) Nrrrrstepr)rrrrr)rNrtrrrrr r8s zFigureCanvasTk.scroll_eventcCs||j|j|j}||jkr dS|j|j|}|jjj |j |j| }|j d}t d|||||ddS)zMouseWheel event processorNxrr)rpZwinfo_containingZx_rootZy_rootrqr winfo_rootxrr@rAr winfo_rootydeltarrr)rNrtrrrrrrr r?s  z#FigureCanvasTk.scroll_event_windowscCs|j}t||j}tjdkr(gd}ntjdkr@rP||vrP|dkr||sPd||}qP|S) Nrl)r/Zctrlcontrol)altrrshiftrr)r)rrrr)rrKrK)r)rrrr)rKrKrrz{0}+{1})charrZ_unikey_or_keysym_to_mplkeyZkeysymr~rstateformat)rNrtZunikeykey modifiersZbitmaskprefixZkey_namerrr _get_keyLs     zFigureCanvasTk._get_keycCs.td|||g||Rd|idS)NZkey_press_eventrrrrrrrrrr rsszFigureCanvasTk.key_presscCs.td|||g||Rd|idS)NZkey_release_eventrrrrrr rxszFigureCanvasTk.key_releasecOst|jg|Ri|SrI)rHrq)rNrFrOrrr new_timer}szFigureCanvasTk.new_timercCs|jdSrI)rqupdaterVrrr flush_eventsszFigureCanvasTk.flush_eventsrcCsN|dkr@td|}|dkr0|j||j|_n|j|j|_|jdS)Nri)rrqrSstop_event_looprwr[mainloop)rNtimeoutZ millisecondsrrr start_event_loops zFigureCanvasTk.start_event_loopcCs(|jr|j|jd|_|jdSrI)rwrqrXquitrVrrr rszFigureCanvasTk.stop_event_loopcCs0z|jjt|dWntjy*Yn0dS)N)cursor)rqrcursordtkinterr?rNrrrr set_cursorszFigureCanvasTk.set_cursor)NN)N)F)r)r\r]r^Zrequired_interactive_frameworkrZ classproperty manager_classrLrr}rrrrrrrrrrrrrrrrrrrr`rrrPr ras01     ' racsheZdZdZdZfddZeddZddZd d Z d d Z d dZ ddZ ddZ ddZZS)rbz Attributes ---------- canvas : `FigureCanvas` The FigureCanvas instance num : int or str The Figure number toolbar : tk.Toolbar The tk.Toolbar window : tk.Window The tk.Window Fcs||_t|||j|jjjtjtj ddt | d}tj |dd|d|_ d|_t||jr|j d|j|_d |_dS) Nr)sidefillexpand`Z window_dpi)rfvaluerwriteF)windowrKrLwithdrawcanvasrqpackr%TOPZBOTHrZwm_frameIntVar _window_dpi_window_dpi_cbnamerZenable_dpi_awarenessr&Z trace_add_update_window_dpi_shown)rNrrrZ window_framerPrr rLs zFigureManagerTk.__init__c Cs ttdur$tttjdd}|t t d}t j ||d}t t d}t j ||d}z| d||Wn0ty} ztd| WYd} ~ n d} ~ 00|||d} || ||} tr| | | WdS1s0YdS) N matplotlib)Z classNamezimages/matplotlib.png)filerfzimages/matplotlib_large.pngFz"Could not load matplotlib icon: %srf)r!r"_get_running_interactive_frameworkZ_setup_new_guiappr Z Win32_SetProcessDpiAwareness_maxr%ZTkrr'_get_data_pathrryZ iconphoto Exception_loginforis_interactiveshowr) rcZ canvas_classrrrZ icon_fnameZicon_imgZicon_fname_largeZicon_img_largeexcrmanagerrrr create_with_canvass4  "  z"FigureManagerTk.create_with_canvascGsH|j}|jdd|d|jr:t|jdr:|j|jdS)Nr%rH_rescale) rrrr=Ztoolbarhasattrrrr)rNrFZnewdpirrr rs   z"FigureManagerTk._update_window_dpicCsPd}||ks||kr:tjdkr:td|d|d|d|jjj||ddS)Ni\linuxz/You have requested to resize the Tk window to (z, z), one of which is bigger than z. At larger sizes xorg will either exit with an error on newer versions (~1.20) or cause corruption on older version (~1.19). We do not expect a window over a million pixel wide or tall to be intended behavior.r)r~r ValueErrorrrqr)rNrBrAmax_sizerrr r}szFigureManagerTk.resizecstjs@fdd}jd|jjjn jt j drxjj j ddjj j ddd_Wdn1s0YdS) NcstdSrI)rdestroyrrVrr rsz%FigureManagerTk.show..destroyZWM_DELETE_WINDOWzfigure.raise_windowz-topmostrrT) r!rrprotocolZ deiconifyrrqrrrrr attributes)rNrrrVr rs    zFigureManagerTk.showcsjjrjjjjjjr4jjjjjrJjdjfdd}t dkrxj j j d|nj |dS)Nrcs&jjr"ts"jdSrI)rr_owns_mainlooprZget_num_fig_managersrrrVrr delayed_destroys z0FigureManagerTk.destroy..delayed_destroyr%r)rrvrqrXrwrrZ trace_removerrrr[rSr)rNrFrrrVr r s   zFigureManagerTk.destroycCs |jSrIrZwm_titlerVrrr get_window_title'sz FigureManagerTk.get_window_titlecCs|j|dSrIr)rNtitlerrr set_window_title*sz FigureManagerTk.set_window_titlecCs$t|jd}|jd| dS)Nz -fullscreen)boolrr)rNZ is_fullscreenrrr full_screen_toggle-sz"FigureManagerTk.full_screen_toggle)r\r]r^r_rrL classmethodrrr}rrrrrr`rrrPr rbs   $rbcseZdZejdddeddZd$ddd d Zd d Zd dZ fddZ fddZ ddZ ddZ ddZededdZddZddZddZd d!Zd"d#ZZS)%NavigationToolbar2Tk3.6z self.master alternativecCs|jSrIrrVrrr rY4rZzNavigationToolbar2Tk.NT) pack_toolbarc Cs<|dur|j}tjj||dt|jjjddi|_ |j D]j\}}}}|dur\| q>|j |t td|d|dvt||d|j |<}|dur>t||q>tjj|d d |_tj||jd d } | jtjd tj|d|_tj||j|jtjd|_|jjtjd t|||r8|jtjtjddS)a  Parameters ---------- canvas : `FigureCanvas` The figure canvas on which to operate. window : tk.Window The tk.Window which owns this toolbar. pack_toolbar : bool, default: True If True, add the toolbar to the parent's pack manager's packing list during initialization with ``side="bottom"`` and ``fill="x"``. If you want to use the toolbar with a different layout manager, use ``pack_toolbar=False``. Nr/2)rfrhrBrAzimages/.png)zoompan)togglecommand )rsizeu   )rffonttextrr)rfr textvariablejustifyrr) rrfr%FramerLrrr@rB_buttonsZ toolitems_Spacer_Buttonr'rrrToolTip createToolTiprrFont _label_fontLabelrRIGHT StringVarmessage_message_labelr ZBOTTOMX) rNrrrrZ tooltip_text image_filecallbackbuttonlabelrrr rL6s>      zNavigationToolbar2Tk.__init__cCs|D]t}t|tjtjfr8t|dr|t||q|qt|tjr`|j ddd|j ddqt|tj rnqt d|jq|jj dd d S) a Scale all children of the toolbar to current DPI setting. Before this is called, the Tk scaling setting will have been updated to match the new DPI. Tk widgets do not update for changes to scaling, but all measurements made after the change will match the new scaling. Thus this function re-applies all the same sizes in points, which Tk will scale correctly to pixels. _image_fileZ22pZ1p)rAZpadyZ4p)padxzUnknown child class %srrN)Zwinfo_children isinstancer%Button Checkbuttonrr_set_image_for_buttonrrpack_configurerrwarningZ winfo_classrrNrprrr rms    zNavigationToolbar2Tk._rescalecCsTdtjfdtjffD]:\}}||jvr|j|kr@|j|q|j|qdS)NZZoomPan)rZZOOMZPANrmodeselectdeselect)rNrr4rrr _update_buttons_checkeds   z,NavigationToolbar2Tk._update_buttons_checkedcstj||dSrI)rKr r7rNrFrPrr r s zNavigationToolbar2Tk.pancstj||dSrI)rKr r7r8rPrr r s zNavigationToolbar2Tk.zoomcCs|j|dSrI)r"setrNsrrr set_messagesz NavigationToolbar2Tk.set_messagecCsR|jjr|jj|jj|jjjj}||}||}|jj|||||j_dSrI)rrrqrrr@rAZcreate_rectangle)rNrtx0y0rCrDrArrr draw_rubberbands z$NavigationToolbar2Tk.draw_rubberbandcCs&|jjr"|jj|jjd|j_dSrI)rrrqrrVrrr remove_rubberbandsz&NavigationToolbar2Tk.remove_rubberbandcCs|jjSrI)rrrVrrr rYrZc sjdurdStdj}||jdd}d}fddfdd }d d }t|d krr| rr|n|p}t j | ||f|d }|_ dtd} ||| } t j | ||f|d } | _Wdn1s0Y|dr d| i} nd|i} ttjrddkr|jdkrDd} n>d\}}}d\}}}||d||d||df} || r| | d<n|| d<jfi| ddddS)z Set the image for a button based on its pixel size. The pixel size is determined by the DPI scaling of the window. NZimagesr z _large.png18pcs|SrI) winfo_rgbcget)Z color_name)r'rr _get_colorsz>NavigationToolbar2Tk._set_image_for_button.._get_colorcst|tr|}t|dkS)Ng@)r,r'r5)color)rDrr _is_darks z._is_darkcSsHt|}|dddfdkjdd}|||ddf<tj|ddS)N.rrr)ZaxisZRGBA)r4)r0r1copyallrZ fromarray)rkrEZ image_dataZ black_maskrrr _recolor_iconszANavigationToolbar2Tk._set_image_for_button.._recolor_iconrg o?rrgrk selectcolorrZx11Zactivebackgroundr/Z selectimage)rArB)r)rr with_namerreplaceZ winfo_pixelsropenexistsrryr}Z_ntimager0arrayrBrCZ _ntimage_altr,r%r.Z_windowingsystemr)rNr'Z path_regularZ path_largerrFrIZimrkrZim_altZ image_altZ image_kwargsrKr1g1b1r2g2b2r)rDr'r r/sP      $     "  z*NavigationToolbar2Tk._set_image_for_buttonc Cs|stj|||dddd}n,tj|d}tj|||d|dddd}||_||_|durdt||n|j|j d|j tj d |S) NZflatZgroover)rfrrrelief overreliefrhrF)rfrrZ indicatoronvariableZ offreliefrXrh)rr) r%r-rr.varr)rr/rrrLEFT)rNrr%r rbrZrrr rs$ zNavigationToolbar2Tk._ButtoncCs(tj|dtjdd}|jtjdd|S)NrAZDarkGray)rfrArWbgZ3p)rr*)r%rZRIDGErr[r:rrr rszNavigationToolbar2Tk._Spacerc Gs|j}|j}||}||fgt|}dd|D}d}tj t j d}|j } t jj|jjd|||| d} | dvrdS|dkrtjt| t j d<z|jj| Wn6ty} zt jdt| WYd} ~ n d} ~ 00dS) NcSsg|]\}}|d|fqS)z*.%sr).0extrrrr &rZz4NavigationToolbar2Tk.save_figure..rzsavefig.directoryzSave the figure)rfr filetypesdefaultextension initialdir initialfile)rrzError saving file)rZget_supported_filetypesrGZget_default_filetyper#sorteditemsospath expanduserrrZget_default_filenamerZ filedialogZasksaveasfilenamerrfdirnamer'rZsavefigrZ messageboxZ showerror) rNrFraZdefault_filetypeZdefault_filetype_nameZsorted_filetypesZ tk_filetypesrbrcrdfnamerGrrr save_figures6      z NavigationToolbar2Tk.save_figurecCsntjtjd}|jjdk}|jjt|jjdk}d|jvrN|||jdd<d|jvrj|||jdd<dS)N)TFrrBackrForward)r%NORMALZDISABLEDZ _nav_stack_poslenZ _elementsr)rNZ state_mapZcan_backZ can_forwardrrr set_history_buttonsEs   z(NavigationToolbar2Tk.set_history_buttons)N)r\r]r^r deprecatedpropertyrrLrr7r r r<r?r@lastrectr/rrrlrrr`rrrPr r2s&  7    M(rc@s4eZdZdZeddZddZddZdd Zd S) rzi Tooltip recipe from http://www.voidspace.org.uk/python/weblog/arch_d7_2006_07_01.shtml#e387 cs>t|fdd}fdd}|d||d|dS)NcsdSrI)showtiprsrtoolTiprr enterYsz$ToolTip.createToolTip..entercs dSrI)hidetiprs)rxrr leave[sz$ToolTip.createToolTip..leavermrn)rr|)rprryr{rrwr rVs   zToolTip.createToolTipcCs"||_d|_d|_d|_|_dS)Nr)rp tipwindowr<rrr2rrr rL`szToolTip.__init__cCs||_|js|jsdS|jd\}}}}||j|j}||j}t|j|_}| d| d||fz|j dd|j ddWntj yYn0tj||jtjtjdd }|jdd dS) zDisplay text in tooltip window.Ninsertrz+%d+%dz!::tk::unsupported::MacWindowStylestylehelpZ noActivates)rrrWrh)Zipadx)rr|rpr@rZ winfo_widthrr%ZToplevelZwm_overrideredirectZ wm_geometryr=_wr?rr[ZSOLIDr)rNrrr_twr(rrr rvfs(  zToolTip.showtipcCs|j}d|_|r|dSrI)r|r)rNrrrr rz|szToolTip.hidetipN) r\r]r^r_ staticmethodrrLrvrzrrrr rQs  rc@s2eZdZddZddZededdZdS) RubberbandTkcCst|d||||dSrI)rr?"_make_classic_style_pseudo_toolbar)rNr=r>rCrDrrr r?szRubberbandTk.draw_rubberbandcCst|dSrI)rr@rrVrrr r@szRubberbandTk.remove_rubberbandrcCs |jjjSrI)rrrrVrrr rYrZzRubberbandTk.N) r\r]r^r?r@rrsrtrurrrr rs  rz3.5Z ToolSetCursorrc@seZdZddZdS) SetCursorTkcCst||dSrI)rrrrrrr rszSetCursorTk.set_cursorN)r\r]r^rrrrr rsrc@sVeZdZdddZddZddZdd Zd d Zd d ZddZ ddZ ddZ dS) ToolbarTkNcCst|||dur"|jjj}|jjjjj\}}d||}}t j j||t |t |ddt j jdd|_t j|d|_t j||j|jd|_|jjt jdi|_|jt jt jd i|_dS) Nr r/)rfrBrArhrr+r)rfrrrr)rrL toolmanagerrrrfrr@Z intervalxr%rrrrrrr!_messagerr#rr  _toolitemsrr$_groups)rNrrZxminZxmaxrArBrrr rLs$    zToolbarTk.__init__cCs t|SrI)rrrVrrr rszToolbarTk._rescalec s|}|}|t|ks&|dkr,d} n||} t|||fdd} | j| d|durpt| |j gj | dS)Nrcs SrI) _button_clickrrrNrr rYrZz(ToolbarTk.add_toolitem..)before) _get_groupframeZ pack_slavesrqrrr0rrr setdefaultappend) rNrgrouppositionr% descriptionr frameZbuttonsrr'rrr add_toolitems     zToolbarTk.add_toolitemcCsT||jvrJ|jr|tj|dd}|jtjtjd|j|_||j|<|j|S)Nr)rfrhr)r_add_separatorr%rrr[Yr)rNrrrrr rs  zToolbarTk._get_groupframecCs t|SrI)rrrVrrr rszToolbarTk._add_separatorcCs||dSrI)Z trigger_tool)rNrrrr rszToolbarTk._button_clickcCs8||jvrdS|j|D]}|r*|q|qdSrI)rr5r6)rNrZtoggledtoolitemrrr toggle_toolitems   zToolbarTk.toggle_toolitemcCs$|j|D] }|q |j|=dSrI)rZ pack_forget)rNrrrrr remove_toolitems zToolbarTk.remove_toolitemcCs|j|dSrI)rr9r:rrr r<szToolbarTk.set_message)N) r\r]r^rLrrrrrrrr<rrrr rs   rc@seZdZddZdS) SaveFigureTkcGst|dSrI)rrlrr8rrr triggerszSaveFigureTk.triggerNr\r]r^rrrrr rsrc@seZdZddZdS)ConfigureSubplotsTkcGst|dSrI)rZconfigure_subplotsr8rrr rszConfigureSubplotsTk.triggerNrrrrr rsrc@seZdZddZdS)HelpTkcs*t|jjj|dgfdd_dS)NOKcs jjSrI)rrfr)rZdialogrr rYrZz HelpTk.trigger..)rrrrqZ_get_help_textdoner8rrr rszHelpTk.triggerNrrrrr rsrc@seZdZeZeddZdS) _BackendTkcCsNt}|rJ|d}t|}|jr&dSd|_z|jWd|_nd|_0dS)NrTF)rZget_all_fig_managerstyperrr)ZmanagersZ first_managerrrrr rs z_BackendTk.mainloopN)r\r]r^rbZ FigureManagerrrrrrr rsr)N)Zuuid contextlibrloggingr6os.pathrgr~rr%Ztkinter.filedialogZ tkinter.fontZtkinter.messageboxZtkinter.simpledialogrnumpyr0ZPILrrrrrrrr Zmatplotlib.backend_basesr r r r rrrrrrrrrZmatplotlib._pylab_helpersrrr getLoggerr\rZ TkVersionZbackend_versionZMOVEZHANDPOINTERZ SELECT_REGIONZWAITZRESIZE_HORIZONTALZRESIZE_VERTICALrr!r"uuid4hexr>r:r;r.r$rHrarbrrrZ_register_tool_classZRubberbandBaserrsZ SetCursorBaserrZSaveFigureBaserZConfigureSubplotsBaserZ ToolHelpBaserZToolbarZ_toolbar2_classZ_toolmanager_toolbar_classexportrrrrr sx  <      7&|!2 H