o
    Ďi0W                  
   @   s   d dl Z d dlZd dlZd dlmZ d dlZd dlmZmZm	Z	 d dl
mZmZmZmZmZmZmZ zd dlmZ W n eyK Z zededZ[ww zedd W n eyd Z zeeedZ[ww d d	lmZmZmZmZmZ d
dlm Z  d
dl m!Z!m"Z"m#Z#m$Z$m%Z& e'e(Z)e j*dd Z+G dd de"ej,Z-G dd de$ej.Z/G dd deej0Z1e2e-G dd dej3Z4e2e-G dd dej5Z6e2e-G dd dej7Z8e1Z.e2e-e j9 e2e-e j: G dd de#Z;e!j<G dd de!Z=dS )    N)Path)_apibackend_toolscbook)ToolContainerBaseMouseButton
CloseEventKeyEventLocationEvent
MouseEventResizeEvent)require_versionz#The GTK3 backends require PyGObjectGtkz3.0)GioGLibGObjectr   Gdk   )_backend_gtk)_BackendGTK_FigureCanvasGTK_FigureManagerGTK_NavigationToolbar2GTKTimerGTKc                 C   s   t jt j t| S N)r   Cursornew_from_nameDisplayget_defaultr   mpl_to_gtk_cursor_name)
mpl_cursor r!   \/home/jeff/fluffinator/venv/lib/python3.10/site-packages/matplotlib/backends/backend_gtk3.py_mpl_to_gtk_cursor%   s   r#   c                       s@  e Zd ZdZedd Zejj	ejj
B ejjB ejjB ejjB ejjB ejjB ejjB ejjB Zd6 fdd	Zdd Zd	d
 Zd6ddZdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Ze dd  Z!e dd!d"d#Z"d$d% Z#d&d' Z$d(d) Z%d*d+ Z&d,d- Z'd.d/ Z(d0d1 Z)d2d3 Z*d4d5 Z+  Z,S )7FigureCanvasGTK3gtk3c                 C   s   t S r   )FigureManagerGTK3)clsr!   r!   r"   <lambda>.   s    zFigureCanvasGTK3.<lambda>Nc                    s,  t  j|d d| _d | _| d| j | d| j | d| j | d| j | d| j	 | d| j	 | d	| j
 | d	| j | d
| j | d| j | d| j | d| j | d| j | d| j | | jj | d t }|d |  }||tj |d d S )N)figurer   scroll_eventbutton_press_eventbutton_release_eventconfigure_eventzscreen-changedznotify::scale-factordrawkey_press_eventkey_release_eventmotion_notify_evententer_notify_eventleave_notify_eventsize_allocateTs/   .matplotlib-canvas { background-color: white; }zmatplotlib-canvas)super__init___idle_draw_id_rubberband_rectconnectr*   r+   r,   r-   _update_device_pixel_ratioon_draw_event
_post_drawr/   r0   r1   r2   r3   r4   
set_events	__class__
event_maskset_can_focusr   CssProviderload_from_dataget_style_contextadd_provider#STYLE_PROVIDER_PRIORITY_APPLICATION	add_class)selfr)   css	style_ctxr>   r!   r"   r6   ;   s0   

zFigureCanvasGTK3.__init__c                 C   s   t d|   d S )Nclose_event)r   _processrG   r!   r!   r"   destroyZ   s   zFigureCanvasGTK3.destroyc                 C   s<   |  d}|d ur|t| tj }|d d S d S )NwindowT)get_property
set_cursorr#   r   MainContextdefault	iteration)rG   cursorrO   contextr!   r!   r"   rQ   ]   s   

zFigureCanvasGTK3.set_cursorc                 C   sd   |du r|   }||   \}}}}n|j|j}}|| j }| jj	j
|| j  }||fS )a  
        Convert the position of a GTK event, or of the current cursor position
        if *event* is None, to Matplotlib coordinates.

        GTK use logical pixels, but the figure is scaled to physical pixels for
        rendering.  Transform to physical pixels so that all of the down-stream
        transforms work as expected.

        Also, the origin is different and needs to be corrected.
        N)
get_windowget_device_positionget_displayget_device_managerget_client_pointerxydevice_pixel_ratior)   bboxheight)rG   eventrO   tr\   r]   stater!   r!   r"   _mpl_coordse   s   
zFigureCanvasGTK3._mpl_coordsc                 C   sH   |j tjjkr	dnd}td| g| |R || |j|d  dS )Nr   r*   )step	modifiersguiEventF)		directionr   ScrollDirectionUPr   rd   _mpl_modifiersrc   rL   )rG   widgetra   rf   r!   r!   r"   r*   {   s   
zFigureCanvasGTK3.scroll_eventc                 C   6   t d| g| ||jR | |j|d  dS )Nr+   rg   rh   Fr   rd   buttonrl   rc   rL   rG   rm   ra   r!   r!   r"   r+         
z#FigureCanvasGTK3.button_press_eventc                 C   rn   )Nr,   ro   Frp   rr   r!   r!   r"   r,      rs   z%FigureCanvasGTK3.button_release_eventc                 C   ,   t d| | |g|  R d|i  dS )Nr/   rh   Tr	   _get_keyrd   rL   rr   r!   r!   r"   r/         z FigureCanvasGTK3.key_press_eventc                 C   rt   )Nr0   rh   Tru   rr   r!   r!   r"   r0      rw   z"FigureCanvasGTK3.key_release_eventc                 C   s:   t d| g| |R | |j| |j|d  dS )Nr1   )buttonsrg   rh   F)r   rd   _mpl_buttonsrc   rl   rL   rr   r!   r!   r"   r1      s   

z$FigureCanvasGTK3.motion_notify_eventc                 C   B   t j|   }td| g| |R | ||d  d S )Nfigure_enter_eventro   	r   Keymapget_for_displayrY   get_modifier_stater
   rd   rl   rL   rG   rm   ra   gtk_modsr!   r!   r"   r2         
z#FigureCanvasGTK3.enter_notify_eventc                 C   rz   )Nfigure_leave_eventro   r|   r   r!   r!   r"   r3      r   z#FigureCanvasGTK3.leave_notify_eventc                 C   sT   | j j}|j| j | }|j| j | }| j j||dd td|   |   d S )NFforwardresize_event)	r)   dpiwidthr^   r`   set_size_inchesr   rL   	draw_idle)rG   rm   
allocationdpivalwinchhinchr!   r!   r"   r4      s   zFigureCanvasGTK3.size_allocatec                    sR   t jtjjft jtjjft jtjjft j	tjj
ft jtjjfg} fdd|D S )Nc                    s   g | ]
\}} |@ r|qS r!   r!   ).0namemaskevent_stater!   r"   
<listcomp>   s    z1FigureCanvasGTK3._mpl_buttons.<locals>.<listcomp>)r   LEFTr   ModifierTypeBUTTON1_MASKMIDDLEBUTTON2_MASKRIGHTBUTTON3_MASKBACKBUTTON4_MASKFORWARDBUTTON5_MASK)r   rg   r!   r   r"   ry      s   zFigureCanvasGTK3._mpl_buttonsexcludec                   sH   dt jjdfdt jjdfdt jjdfdt jjdfg} fdd|D S )Nctrlcontrolaltshiftr5   c                    s&   g | ]\}}}|kr |@ r|qS r!   r!   )r   r   r   keyr   r   r!   r"   r      s    z3FigureCanvasGTK3._mpl_modifiers.<locals>.<listcomp>)r   r   CONTROL_MASK	MOD1_MASK
SHIFT_MASK	MOD4_MASK)r   r   rg   r!   r   r"   rl      s   zFigureCanvasGTK3._mpl_modifiersc                 C   s`   t t|j}t|t|j}| j|j|d}d|v r'|	 r'|
d dg ||S )Nr   r   +)chrr   keyval_to_unicodekeyvalr   _unikey_or_keysym_to_mplkeykeyval_namerl   rc   isprintableremovejoin)rG   ra   unikeyr   modsr!   r!   r"   rv      s   
zFigureCanvasGTK3._get_keyc                 O   s&   |  |  r|   |   d S d S r   )_set_device_pixel_ratioget_scale_factorqueue_resize
queue_drawrG   argskwargsr!   r!   r"   r:      s   z+FigureCanvasGTK3._update_device_pixel_ratioc                 C   sd   | dd u r	d S |j| j }|j| j }|dk s|dk rd S | jj}| jj|| || dd dS )NrO      Fr   )rP   r   r^   r`   r)   r   r   )rG   rm   ra   whr   r!   r!   r"   r-      s   z FigureCanvasGTK3.configure_eventc                 C   s   || _ |   d S r   )r8   r   )rG   rectr!   r!   r"   _draw_rubberband   s   z!FigureCanvasGTK3._draw_rubberbandc           	         s    j d u rd S  fdd j D \}}}}|| }|| }||| ||| ||| ||| ||| ||| ||| ||| |d |d |dd |ddd |  |dd |ddd |  d S )Nc                 3   s    | ]}| j  V  qd S r   )r^   )r   dimrM   r!   r"   	<genexpr>   s    z.FigureCanvasGTK3._post_draw.<locals>.<genexpr>r   )r   r   r   r   )	r8   move_toline_toset_antialiasset_line_widthset_dashset_source_rgbstroke_preservestroke)	rG   rm   ctxx0y0r   r   x1y1r!   rM   r"   r<      s.   



zFigureCanvasGTK3._post_drawc                 C   s   d S r   r!   )rG   rm   r   r!   r!   r"   r;     s   zFigureCanvasGTK3.on_draw_eventc                 C   s   |   r
|   d S d S r   )is_drawabler   rM   r!   r!   r"   r.     s   zFigureCanvasGTK3.drawc                    s*    j dkrd S  fdd}t| _ d S )Nr   c                     s   z
    W d _dS d _w )Nr   F)r.   r7   )r   rM   r!   r"   	idle_draw   s
   
z-FigureCanvasGTK3.draw_idle.<locals>.idle_draw)r7   r   idle_add)rG   r   r!   rM   r"   r     s   
zFigureCanvasGTK3.draw_idlec                 C   s,   t j }| r|d | s	d S d S )NT)r   rR   rS   pendingrT   )rG   rV   r!   r!   r"   flush_events(  s   

zFigureCanvasGTK3.flush_eventsr   )-__name__
__module____qualname__required_interactive_frameworkr   classpropertymanager_classr   	EventMaskBUTTON_PRESS_MASKBUTTON_RELEASE_MASKEXPOSURE_MASKKEY_PRESS_MASKKEY_RELEASE_MASKENTER_NOTIFY_MASKLEAVE_NOTIFY_MASKPOINTER_MOTION_MASKSCROLL_MASKr?   r6   rN   rQ   rd   r*   r+   r,   r/   r0   r1   r2   r3   r4   staticmethodry   rl   rv   r:   r-   r   r<   r;   r.   r   r   __classcell__r!   r!   rJ   r"   r$   ,   s\    



	
r$   c                   @   s   e Zd Zdd Zdd ZdS )NavigationToolbar2GTK3c           
      C   sf  t j |  | tjj i | _| jD ]Y\}}}}|d u r&| t	 d qtj
tjttd| dtjj}|dv rDt nt  | j|< }|| || |dt| ||_|| | |d qt }| |d t }	|	d |d | |	 t }| |d t | _!| j!"tj#j$ | | j! | %  t&| | d S )Nre   images-symbolic.svg)zoompanclickedu   <small> 
 </small>T)'r   r6   	set_styler   ToolbarStyleICONS_gtk_ids	toolitemsinsertSeparatorToolItemImagenew_from_giconr   Iconnew_for_stringstrr   _get_data_pathIconSizeLARGE_TOOLBARToggleToolButton
ToolButton	set_labelset_icon_widgetr9   getattr_signal_handlerset_tooltip_textToolItemLabel
set_markup
set_expandaddmessageset_justifyJustificationr   show_allr   )
rG   canvastexttooltip_text
image_filecallbackimagerq   toolitemlabelr!   r!   r"   r6   0  sP   






zNavigationToolbar2GTK3.__init__c           
   
      s  t jdj t jjt jt jjt j	t jj
fd j  D ]+\}}t  }|| |D ]
}|d|  q. | j |v rJ | qt jd fdd} tjd   j   d   }  }  }j |  d	 }    |t jj
krd S tjd rt!j"#|tjd< zjj$j%||d
 |W S  t&y }	 z!t j'j t(|	t j)j*t j+j
d        W Y d }	~	d S d }	~	ww )NzSave the figure)titletransient_foractionrx   z*.znotify::filterc                     sD       }j | d } tt  d|  d S )Nr   .)	
get_filterget_namer  get_supported_filetypes_groupedset_current_namer   r   get_current_namewith_suffix)r   r   fmtdialogrG   r!   r"   on_notify_filterr  s
   z<NavigationToolbar2GTK3.save_figure.<locals>.on_notify_filterzsavefig.directoryTr   )format)r  r  message_typerx   ),r   FileChooserDialogr  get_toplevelFileChooserActionSAVESTOCK_CANCELResponseTypeCANCEL
STOCK_SAVEOKr  items
FileFilterset_nameadd_pattern
add_filterget_default_filetype
set_filter	functoolspartialr9   set_current_foldermplrcParamsr  get_default_filenameset_do_overwrite_confirmationrunget_filenamer  r  rN   ospathdirnamer)   savefig	ExceptionMessageDialogr   MessageTypeERRORButtonsType)
rG   r   r   fmtsffr   r#  responsefnameer!   r!  r"   save_figure`  sX   







z"NavigationToolbar2GTK3.save_figureN)r   r   r   r6   rM  r!   r!   r!   r"   r   /  s    0r   c                   @   sP   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dd Z
dd ZdS )ToolbarGTK3r   c                 C   sl   t | | tj|  | dtjj t | _| j	tj
j | | jddd |   i | _i | _d S )NorientationFr   )r   r6   r   Boxset_propertyOrientation
HORIZONTALr  _messager  r  r   pack_endr  _groups
_toolitems)rG   toolmanagerr!   r!   r"   r6     s   

zToolbarGTK3.__init__c           
      C   s   |rt  }nt  }|| |d ur&t jtj|t j	j
}|| |d u r,d}| ||| |d| j|}	|| |  | j|g  | j| ||	f d S )Nre   r   )r   r   r   r   r   r   r   r   r   r   r   r  _add_buttonr9   
_call_toolr  r  rW  
setdefaultappend)
rG   r   grouppositionr  descriptiontogglerq   r  signalr!   r!   r"   add_toolitem  s$   




zToolbarGTK3.add_toolitemc                 C   sf   || j vr(| j r|   t }|tjj | |ddd |  || j |< | j | 	|| d S )NFr   )
rV  _add_separatorr   Toolbarr   r   r   
pack_startr  r   )rG   rq   r]  r^  toolbarr!   r!   r"   rY    s   

zToolbarGTK3._add_buttonc                 C   s   |  | d S r   )trigger_tool)rG   btnr   r!   r!   r"   rZ    s   zToolbarGTK3._call_toolc                 C   sD   || j vrd S | j | D ]\}}|| || || qd S r   )rW  handler_block
set_activehandler_unblock)rG   r   toggledr  ra  r!   r!   r"   toggle_toolitem  s   


zToolbarGTK3.toggle_toolitemc                 C   sF   | j |g D ]\}}| jD ]}|| j| v r| j| | qqd S r   )rW  poprV  r   )rG   r   r  _signalr]  r!   r!   r"   remove_toolitem  s   
zToolbarGTK3.remove_toolitemc                 C   s4   t  }|dt jj | |ddd |  d S )NrO  FTr   )r   	SeparatorrQ  rR  VERTICALre  r  )rG   sepr!   r!   r"   rc    s   zToolbarGTK3._add_separatorc                 C   s   | j | d S r   )rT  r   )rG   sr!   r!   r"   set_message  s   zToolbarGTK3.set_messageN)r   r   r   _icon_extensionr6   rb  rY  rZ  rm  rp  rc  ru  r!   r!   r!   r"   rN    s    rN  c                   @      e Zd Zdd ZdS )SaveFigureGTK3c                 O   s   t |   d S r   )r   rM  "_make_classic_style_pseudo_toolbarr   r!   r!   r"   trigger  s   zSaveFigureGTK3.triggerNr   r   r   rz  r!   r!   r!   r"   rx        rx  c                   @   s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )HelpGTK3c                 C   s~   ddddd}| d}dd |d	d
 D }|d
 }||v r$|| }nt|dkr/| }n	| r8|dg7 }d|| S )z
        Convert Matplotlib key presses to GTK+ accelerator identifiers.

        Related to `FigureCanvasGTK3._get_key`.
        	BackSpace	Page_DownPage_UpScroll_Lock)	backspacepagedownpageupscroll_lockr   c                 S   s   g | ]}d | d qS )<>r!   )r   modr!   r!   r"   r     s    z0HelpGTK3._normalize_shortcut.<locals>.<listcomp>Nre   r   z<shift> )splitlen
capitalizeisupperr   )rG   r   specialpartsr   r!   r!   r"   _normalize_shortcut  s   



zHelpGTK3._normalize_shortcutc                 C   s   d|vo	| d S )z
        Check for a valid shortcut to be displayed.

        - GTK will never send 'cmd+' (see `FigureCanvasGTK3._get_key`).
        - The shortcut window only shows keyboard shortcuts, not mouse buttons.
        zcmd+zMouseButton.)
startswith)rG   r   r!   r!   r"   _is_valid_shortcut  s   zHelpGTK3._is_valid_shortcutc                    s   t  }t jj D ]6\}}|jsqt  }|| |	dd d  t j
d fdd j|D |j|jd}|| qt jdd jj d	}|  || |  d S )
Nc                 S   s
   |  dS )NF)set_visible)rm   datar!   r!   r"   r(     s   
 z1HelpGTK3._show_shortcuts_window.<locals>.<lambda> c                 3   s$    | ]}  |r |V  qd S r   )r  r  )r   r   rM   r!   r"   r     s    
z2HelpGTK3._show_shortcuts_window.<locals>.<genexpr>)acceleratorr  subtitleHelpT)r  modalr  )r   ShortcutsSectionsortedrX  toolsr/  r_  ShortcutsGroupr	  forallShortcutsShortcutr   get_tool_keymapr   ShortcutsWindow_figurer  r'  showr  )rG   sectionr   toolr]  shortcutrO   r!   rM   r"   _show_shortcuts_window
  s.   



zHelpGTK3._show_shortcuts_windowc                 C   s>   t j| jj dt jjt jj| 	 dd}|
  |  d S )Nr   r  )r  )r   rD  r  r  r'  rE  INFOrG  r.  _get_help_textr=  rN   )rG   r"  r!   r!   r"   _show_shortcuts_dialog,  s   
zHelpGTK3._show_shortcuts_dialogc                 G   s*   t dddd u r|   d S |   d S )Nr      r   )r   check_versionr  r  )rG   r   r!   r!   r"   rz  4  s   zHelpGTK3.triggerN)r   r   r   r  r  r  r  rz  r!   r!   r!   r"   r}    s    	"r}  c                   @   rw  )ToolCopyToClipboardGTK3c           
      O   sH   t jtj}| j }| \}}}}t|||||}	|	|	 d S r   )
r   	Clipboardgetr   SELECTION_CLIPBOARDr  rW   get_geometrypixbuf_get_from_window	set_image)
rG   r   r   	clipboardrO   r\   r]   r   r`   pbr!   r!   r"   rz  =  s
   
zToolCopyToClipboardGTK3.triggerNr{  r!   r!   r!   r"   r  ;  r|  r  c                   @      e Zd ZeZeZdS )r&   N)r   r   r   r   _toolbar2_classrN  _toolmanager_toolbar_classr!   r!   r!   r"   r&   L  s    r&   c                   @   r  )_BackendGTK3N)r   r   r   r$   FigureCanvasr&   FigureManagerr!   r!   r!   r"   r  Q  s    r  )>r6  loggingr?  pathlibr   
matplotlibr9  r   r   r   matplotlib.backend_basesr   r   r   r	   r
   r   r   gir   gi_require_versionImportErrorerr
ValueErrorrL  gi.repositoryr   r   r   r   r   r  r   r   r   r   r   r   	TimerGTK3	getLoggerr   _logcacher#   DrawingArear$   rd  r   rP  rN  _register_tool_classSaveFigureBaserx  ToolHelpBaser}  ToolCopyToClipboardBaser  ConfigureSubplotsGTKRubberbandGTKr&   exportr  r!   r!   r!   r"   <module>   s\    $



  cLU	