o
    Ďi*\                  
   @   s   d dl Z d dlZd dl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yE Z zededZ[ww zedd W n ey^ Z zeeedZ[ww d dlZ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j%dkZ&e'ddd du Z(G dd de ej)Z*G dd de"ej+Z,G dd de	ej+Z-e.e*G dd dej/Z0e.e*G dd dej1Z2e.e*G dd dej3Z4e.e*ej5 e.e*ej6 e-Z7G dd de!Z8ej9G dd deZ:dS )    N)_apibackend_toolscbook)ToolContainerBaseMouseButtonKeyEventLocationEvent
MouseEventResizeEvent
CloseEvent)require_versionz#The GTK4 backends require PyGObjectGtkz4.0)GioGLibr   Gdk	GdkPixbuf   )_backend_gtk)_BackendGTK_FigureCanvasGTK_FigureManagerGTK_NavigationToolbar2GTKTimerGTK)   /   r         c                       s   e Zd ZdZdZedd Zd8 fdd	Zdd	 Z	d
d Z
d8d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d d! Zd8d"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d6d7 Z   Z!S )9FigureCanvasGTK4gtk4Fc                 C   s   t S N)FigureManagerGTK4)cls r"   \/home/jeff/fluffinator/venv/lib/python3.10/site-packages/matplotlib/backends/backend_gtk4.py<lambda>(   s    zFigureCanvasGTK4.<lambda>Nc           	         s  t  j|d | d | d d| _d | _| | j | d| j	 t
r.| d| j n| d| j t }|d |d| j |d| j | | t }|d	| j |d
| j | | t }|d| j |d| j |d| j | | tjtjj}|d| j | | | d t  }d}t!dddd u r|"|d n|"|#d | $ }|%|tj& |'d d S )N)figureTr   resizerealizeznotify::scale-factorpressedreleasedzkey-pressedzkey-releasedmotionenterleavescrollz/.matplotlib-canvas { background-color: white; }r   	   r   zutf-8zmatplotlib-canvas)(super__init__set_hexpandset_vexpand_idle_draw_id_rubberband_rectset_draw_func
_draw_funcconnectresize_event_GTK_GE_4_12_realize_event_update_device_pixel_ratior   GestureClick
set_buttonbutton_press_eventbutton_release_eventadd_controllerEventControllerKeykey_press_eventkey_release_eventEventControllerMotionmotion_notify_evententer_notify_eventleave_notify_eventEventControllerScrollnewEventControllerScrollFlagsVERTICALscroll_eventset_focusableCssProvidercheck_versionload_from_dataencodeget_style_contextadd_provider#STYLE_PROVIDER_PRIORITY_APPLICATION	add_class)	selfr%   clickkeyr*   r-   cssstyle	style_ctx	__class__r"   r#   r1   *   sL   







zFigureCanvasGTK4.__init__c                 C   s   t d|   d S )Nclose_event)r   _processrW   r"   r"   r#   destroy\   s   zFigureCanvasGTK4.destroyc                 C   s   |  t| d S r   )set_cursor_from_namer   mpl_to_gtk_cursor_name)rW   cursorr"   r"   r#   
set_cursor_   s   zFigureCanvasGTK4.set_cursorc                 C   sb   |du r|    }||    \}}}}n|\}}|| j }| jjj	|| j  }||fS )a  
        Convert the *xy* position of a GTK event, or of the current cursor
        position if *xy* 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_nativeget_surfaceget_device_positionget_displayget_default_seatget_pointerdevice_pixel_ratior%   bboxheight)rW   xysurfaceis_overxymaskr"   r"   r#   _mpl_coordsc   s   
zFigureCanvasGTK4._mpl_coordsc                 C   s:   t d| g|  R || |tr| nd d  dS )NrM   )step	modifiersguiEventT)r	   rv   _mpl_modifiers_GOBJECT_GE_3_47get_current_eventr`   )rW   
controllerdxdyr"   r"   r#   rM   y   s   zFigureCanvasGTK4.scroll_eventc                 C   sN   t d| g| ||f| R | |tr| nd d  |   d S )Nr?   rx   ry   )r	   rv   get_current_buttonrz   r{   r|   r`   
grab_focusrW   r}   n_pressrs   rt   r"   r"   r#   r?      s   z#FigureCanvasGTK4.button_press_eventc                 C   sF   t d| g| ||f| R | |tr| nd d  d S )Nr@   r   )r	   rv   r   rz   r{   r|   r`   r   r"   r"   r#   r@      s   
z%FigureCanvasGTK4.button_release_eventc                 C   <   t d| | |||g|  R dtr| nd i  dS )NrC   ry   Tr   _get_keyrv   r{   r|   r`   rW   r}   keyvalkeycodestater"   r"   r#   rC         z FigureCanvasGTK4.key_press_eventc                 C   r   )NrD   ry   Tr   r   r"   r"   r#   rD      r   z"FigureCanvasGTK4.key_release_eventc                 C   sF   t d| g| ||fR | || |tr| nd d  d S )NrF   )buttonsrx   ry   )r	   rv   _mpl_buttonsrz   r{   r|   r`   rW   r}   rs   rt   r"   r"   r#   rF      s   
z$FigureCanvasGTK4.motion_notify_eventc                 C   s<   t d| g| ||fR |  tr| nd d  d S )Nfigure_enter_eventr   r   rv   rz   r{   r|   r`   r   r"   r"   r#   rG      s   
z#FigureCanvasGTK4.enter_notify_eventc                 C   s6   t d| g|  R |  tr| nd d  d S )Nfigure_leave_eventr   r   )rW   r}   r"   r"   r#   rH      s   
z#FigureCanvasGTK4.leave_notify_eventc                 C   sX   |    | jj}|| j | }|| j | }| jj||dd td|   |   d S )NF)forwardr9   )r<   r%   dpirm   set_size_inchesr
   r`   	draw_idle)rW   areawidthro   r   winchhinchr"   r"   r#   r9      s   zFigureCanvasGTK4.resize_eventc                    s|   |    }||    \}}} 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   h | ]
\}} |@ r|qS r"   r"   .0nameru   event_stater"   r#   	<setcomp>       z0FigureCanvasGTK4._mpl_buttons.<locals>.<setcomp>)rg   rh   ri   rj   rk   rl   r   LEFTr   ModifierTypeBUTTON1_MASKMIDDLEBUTTON2_MASKRIGHTBUTTON3_MASKBACKBUTTON4_MASKFORWARDBUTTON5_MASKrW   r}   rq   rr   rs   rt   	mod_tabler"   r   r#   r      s   zFigureCanvasGTK4._mpl_buttonsc                    sz   |d u r|    }||    \}}} n|  dtjj	fdtjj
fdtjjfdtjjfg} fdd|D S )Nctrlaltshiftr0   c                    s   g | ]
\}} |@ r|qS r"   r"   r   r   r"   r#   
<listcomp>   r   z3FigureCanvasGTK4._mpl_modifiers.<locals>.<listcomp>)rg   rh   ri   rj   rk   rl   get_current_event_stater   r   CONTROL_MASKALT_MASK
SHIFT_MASK
SUPER_MASKr   r"   r   r#   rz      s   



zFigureCanvasGTK4._mpl_modifiersc                    s|   t t|tt| 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 }dg | S )	Nr   controlr   r   r0   c                    s6   g | ]\}}}| kr|@ r|d kr  s|qS )r   )isprintable)r   modru   mod_keyrY   r   unikeyr"   r#   r      s    z-FigureCanvasGTK4._get_key.<locals>.<listcomp>+)chrr   keyval_to_unicoder   _unikey_or_keysym_to_mplkeykeyval_namer   r   r   r   r   join)rW   r   r   r   rx   modsr"   r   r#   r      s   zFigureCanvasGTK4._get_keyc                 C   s&   |    }|d| j |   d S )Nznotify::scale)rg   rh   r8   r<   )rW   objrq   r"   r"   r#   r;      s   zFigureCanvasGTK4._realize_eventc                 O   sD   t r|    }n|  }|d usJ | |r |   d S d S r   )r:   rg   rh   	get_scaleget_scale_factor_set_device_pixel_ratiodraw)rW   argskwargsscaler"   r"   r#   r<      s   
z+FigureCanvasGTK4._update_device_pixel_ratioc                 C   s   || _ |   d S r   )r5   
queue_draw)rW   rectr"   r"   r#   _draw_rubberband  s   z!FigureCanvasGTK4._draw_rubberbandc                 C   s   |  | | | | | d S r   )on_draw_event
_post_draw)rW   drawing_areactxr   ro   r"   r"   r#   r7   
  s   zFigureCanvasGTK4._draw_funcc                    s   j d u rd S d}d} fdd j D \}}}}|| }	|| }
||| |||
 ||| ||	| |||
 ||	|
 ||	| ||	|
 |d || |||fd |ddd |  |||f| |ddd |  d S )Nr   r   c                 3   s    | ]}| j  V  qd S r   )rm   )r   dimra   r"   r#   	<genexpr>  s    z.FigureCanvasGTK4._post_draw.<locals>.<genexpr>r   )	r5   move_toline_toset_antialiasset_line_widthset_dashset_source_rgbstroke_preservestroke)rW   widgetr   lwdashx0y0whx1y1r"   ra   r#   r     s2   



zFigureCanvasGTK4._post_drawc                 C   s   d S r   r"   )rW   r   r   r"   r"   r#   r   .  s   zFigureCanvasGTK4.on_draw_eventc                 C   s   |   r
|   d S d S r   )is_drawabler   ra   r"   r"   r#   r   2  s   zFigureCanvasGTK4.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   r4   )r   ra   r"   r#   	idle_draw;  s
   
z-FigureCanvasGTK4.draw_idle.<locals>.idle_draw)r4   r   idle_add)rW   r   r"   ra   r#   r   7  s   
zFigureCanvasGTK4.draw_idlec                 C   s,   t j }| r|d | s	d S d S )NT)r   MainContextdefaultpending	iteration)rW   contextr"   r"   r#   flush_eventsC  s   

zFigureCanvasGTK4.flush_eventsr   )"__name__
__module____qualname__required_interactive_frameworksupports_blitr   classpropertymanager_classr1   rb   rf   rv   rM   r?   r@   rC   rD   rF   rG   rH   r9   r   rz   r   r;   r<   r   r7   r   r   r   r   r   __classcell__r"   r"   r]   r#   r   %   s:    2
		
 r   c                   @   s   e Zd Zdd Zdd ZdS )NavigationToolbar2GTK4c           	      C   s2  t j|  | d i | _| jD ]Y\}}}}|d u r#| t   qt j	t
jttd| d}|dv r>t  nt   | j|< }|| |d |d |dt| ||_|| | | qt  }|d |d	 | | t  | _| jt jj | | j t| | d S )
Ntoolbarimages-symbolic.svg)zoompanflatimage-buttonclicked   <small> 
 </small>T)r   Boxr1   add_css_class_gtk_ids	toolitemsappend	SeparatorImagenew_from_giconr   Iconnew_for_stringstrr   _get_data_pathToggleButtonButton	set_childr8   getattr_signal_handlerset_tooltip_textLabel
set_markupr2   messageset_justifyJustificationr   r   )	rW   canvastexttooltip_text
image_filecallbackimagebuttonlabelr"   r"   r#   r1   K  sF   








zNavigationToolbar2GTK4.__init__c                    sj  t jd j t jjdd}| _t  }|d |	d |
| || g }d }t j  D ]/\}\}}t  }|| |D ]
}	|	d|	  qG|
| ||  j |v re|}q6|| g|d | ||d d  }|dd	|| |d|d
  |tjtjtjd  | j  t|j d fdd}
|!   j"S )NzSave the figureT)titletransient_foractionmodalz	All files*z*.r   formatzFile formatr   savefig.directoryresponsec              
      s   |   }| d} j | d }|   d  _|tjjkr!d S t	j
d r1| }| t	j
d< z jjj| |d W d S  tyi } ztj j tjjtjjdt|d}|  W Y d }~d S d }~ww )Nr&  r   r'  )r&  T)r"  message_typer   r$  r  )get_file
get_choicer  get_supported_filetypes_groupedrb   _save_dialogr   ResponseTypeACCEPTmplrcParams
get_parentget_pathr%   savefig	ExceptionMessageDialogget_rootMessageTypeERRORButtonsTypeOKr  show)dialogr(  filefmtparentemsgra   r"   r#   on_response  s,   

z7NavigationToolbar2GTK4.save_figure.<locals>.on_response)#r   FileChooserNativer  r7  FileChooserActionSAVEr-  
FileFilterset_nameadd_pattern
add_filter
set_filter	enumerater,  itemsr  get_default_filetype
add_choice
set_choiceset_current_folderr   Filenew_for_pathospath
expanduserr0  r1  set_current_nameget_default_filename	functoolspartialr8   r<  UNKNOWN_SAVED_STATUS)rW   r   r=  ffformatsdefault_formatir   fmtsr?  rC  r"   ra   r#   save_figureu  sN   







z"NavigationToolbar2GTK4.save_figureN)r   r   r   r1   ra  r"   r"   r"   r#   r   J  s    *r   c                   @   sX   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d ZdS )ToolbarGTK4r   c                 C   s   t | | tj|  | dtjj t | _| | j i | _	i | _
t }|d |d | | t | _| jtjj | | j d S )Norientationr  T)r   r1   r   r  set_propertyOrientation
HORIZONTAL	_tool_boxr  _groups
_toolitemsr  r  r2   _messager  r  r   )rW   toolmanagerr   r"   r"   r#   r1     s    



zToolbarGTK4.__init__c           
      C   s   |rt  }nt  }|| |d |d ur-t jtj	|}|
| |d |d u r3d}| ||| |d| j|}	|| | j|g  | j| ||	f d S )Nr   r   r/   r   )r   r  r  	set_labelr  r  r	  r   r
  r  r  _add_buttonr8   
_call_toolr  ri  
setdefaultr  )
rW   r   grouppositionr  descriptiontoggler  r  signalr"   r"   r#   add_toolitem  s$   






zToolbarGTK4.add_toolitemc                 C   s>   d g}| j |  }|d ur|| | }|d us|| S r   )rh  get_first_childr  get_next_sibling)rW   rp  rq  childrenchildr"   r"   r#   _find_child_at_position  s   
z#ToolbarGTK4._find_child_at_positionc                 C   sT   || j vr| j r|   t }| j| || j |< | j | || || d S r   )rh  _add_separatorr   r  rg  r  insert_child_afterrz  )rW   r  rp  rq  	group_boxr"   r"   r#   rm    s   


zToolbarGTK4._add_buttonc                 C   s   |  | d S r   )trigger_tool)rW   btnr   r"   r"   r#   rn    s   zToolbarGTK4._call_toolc                 C   sD   || j vrd S | j | D ]\}}|| || || qd S r   )ri  handler_block
set_activehandler_unblock)rW   r   toggledtoolitemrt  r"   r"   r#   toggle_toolitem  s   


zToolbarGTK4.toggle_toolitemc                 C   sF   | j |g D ]\}}| jD ]}|| j| v r| j| | qqd S r   )ri  poprh  remove)rW   r   r  _signalrp  r"   r"   r#   remove_toolitem  s   
zToolbarGTK4.remove_toolitemc                 C   s(   t  }|dt jj | j| d S )Nrc  )r   r  rd  re  rL   rg  r  )rW   sepr"   r"   r#   r{    s   zToolbarGTK4._add_separatorc                 C   s   | j | d S r   )rj  rl  )rW   sr"   r"   r#   set_message  s   zToolbarGTK4.set_messageN)r   r   r   _icon_extensionr1   ru  rz  rm  rn  r  r  r{  r  r"   r"   r"   r#   rb    s    
rb  c                   @      e Zd Zdd ZdS )SaveFigureGTK4c                 O   s   t |   d S r   )r   ra  "_make_classic_style_pseudo_toolbar)rW   r   r   r"   r"   r#   trigger  s   zSaveFigureGTK4.triggerNr   r   r   r  r"   r"   r"   r#   r        r  c                   @   s$   e Zd Zdd Zdd Zdd ZdS )HelpGTK4c                 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 `FigureCanvasGTK4._get_key`.
        	BackSpace	Page_DownPage_UpScroll_Lock)	backspacepagedownpageupscroll_lockr   c                 S   s   g | ]}d | d qS )<>r"   )r   r   r"   r"   r#   r   ,  s    z0HelpGTK4._normalize_shortcut.<locals>.<listcomp>Nr/   r   z<shift> )splitlen
capitalizeisupperr   )rW   rY   specialpartsr   r"   r"   r#   _normalize_shortcut  s   



zHelpGTK4._normalize_shortcutc                 C   s   d|vo	| d S )z
        Check for a valid shortcut to be displayed.

        - GTK will never send 'cmd+' (see `FigureCanvasGTK4._get_key`).
        - The shortcut window only shows keyboard shortcuts, not mouse buttons.
        zcmd+zMouseButton.)
startswith)rW   rY   r"   r"   r#   _is_valid_shortcut8  s   zHelpGTK4._is_valid_shortcutc           	         s   t  }t jj D ]C\}}|jsqt  }|| |	 }|d ur2|
d | }|d us%t jd fdd j|D |j|jd}|| qt jdd jj d}|| |  d S )	NF c                 3   s$    | ]}  |r |V  qd S r   )r  r  )r   rY   ra   r"   r#   r   U  s    
z#HelpGTK4.trigger.<locals>.<genexpr>)acceleratorr!  subtitleHelpT)r!  r$  r"  )r   ShortcutsSectionsortedrk  toolsrM  rr  ShortcutsGroupr  rv  set_visiblerw  ShortcutsShortcutr   get_tool_keymapr   ShortcutsWindow_figurer  r7  r  r<  )	rW   r   sectionr   toolrp  ry  shortcutwindowr"   ra   r#   r  A  s4   




zHelpGTK4.triggerN)r   r   r   r  r  r  r"   r"   r"   r#   r    s    	r  c                   @   r  )ToolCopyToClipboardGTK4c                 O   s~   t  '}| j| | j \}}tj| tj	j
dd|||d }W d    n1 s.w   Y  | j }|| d S )NT   r   )ioBytesIOr  
print_rgbaget_width_heightr   Pixbufnew_from_data	getbuffer
ColorspaceRGBget_clipboardset)rW   r   r   fr   r   pb	clipboardr"   r"   r#   r  h  s   

zToolCopyToClipboardGTK4.triggerNr  r"   r"   r"   r#   r  f  r  r  c                   @      e Zd ZeZeZdS )r    N)r   r   r   r   _toolbar2_classrb  _toolmanager_toolbar_classr"   r"   r"   r#   r    z  s    r    c                   @   r  )_BackendGTK4N)r   r   r   r   FigureCanvasr    FigureManagerr"   r"   r"   r#   r    s    r  );rY  r  rT  
matplotlibr0  r   r   r   matplotlib.backend_basesr   r   r   r   r	   r
   r   gir   gi_require_versionImportErrorerr
ValueErrorrA  gi.repositoryr   r   r   r   r   r  r   r   r   r   r   r   	TimerGTK4version_infor{   rP   r:   DrawingArear   r  r   rb  _register_tool_classSaveFigureBaser  ToolHelpBaser  ToolCopyToClipboardBaser  ConfigureSubplotsGTKRubberbandGTKToolbarr    exportr  r"   r"   r"   r#   <module>   sZ    $


  'k`I