U
    Âmœd³F  ã                   @   sn  d Z ddlZddlZddlZddlZzddlZejdk r<e‚W nN ek
rŒ   zddl	ZW n, ek
r† Z
 zedƒe
‚W 5 dZ
[
X Y nX Y nX 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 ddlmZ dd	lmZ dd
lmZ ddd„Zdd„ ZG dd„ deƒZG dd„ deƒZ G dd„ dƒZ!G dd„ deƒZ"e #d¡G dd„ deƒƒZ$ej%G dd„ deƒƒZ&dS )z
A Cairo backend for Matplotlib
==============================
:Author: Steve Chaplin and others

This backend depends on cairocffi or pycairo.
é    N)é   é   r   zEcairo backend requires that pycairo>=1.14.0 or cairocffi is installedé   )Ú_apiÚcbookÚfont_manager)Ú_BackendÚFigureCanvasBaseÚFigureManagerBaseÚGraphicsContextBaseÚRendererBase)ÚttfFontProperty)ÚPath)ÚAffine2Dc           	      C   sÔ   |j |d|dD ]¾\}}|tjkr.| j|Ž  q|tjkrB|  ¡  q|tjkrX| j|Ž  q|tjkrºt	 
|  ¡ ¡}|d d… }|dd … }| j|d |d d  |d d |d  |žŽ  q|tjkr| j|Ž  qd S )NT)Zremove_nansÚclipr   éþÿÿÿé   )Úiter_segmentsr   ZMOVETOÚmove_toZ	CLOSEPOLYZ
close_pathZLINETOZline_toZCURVE3ÚnpÚasarrayZget_current_pointZcurve_toZCURVE4)	ÚctxÚpathÚ	transformr   ÚpointsÚcodeÚcurÚaÚb© r   úZ/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/matplotlib/backends/backend_cairo.pyÚ_append_path%   s$      ÿ




2
r!   c                    s\   ‡ fdd„}|dƒ}t td|dƒ ¡ › ƒ}|dƒ}tj ||¡dk rLtjntj}|||fS )z{
    Convert a `.FontProperties` or a `.FontEntry` to arguments that can be
    passed to `.Context.select_font_face`.
    c                    s8   zt ˆ d| › ƒƒ W S  tk
r2   t ˆ | ƒ Y S X d S )NÚget_)ÚgetattrÚAttributeError)Úfield©Úpropr   r    Úattr<   s    z-_cairo_font_args_from_font_prop.<locals>.attrÚnameZFONT_SLANT_ÚstyleÚweighti&  )r#   ÚcairoÚupperr   Zweight_dictÚgetZFONT_WEIGHT_NORMALZFONT_WEIGHT_BOLD)r'   r(   r)   Zslantr+   r   r&   r    Ú_cairo_font_args_from_font_prop7   s    ÿþr/   c                       s¦   e Zd Z‡ fdd„Zdd„ Zejddddd	„ ƒZe d¡d
d„ ƒZdd„ Z	d"dd„Z
d#dd„Zdd„ Zd$dd„Zdd„ Zdd„ Z‡ fdd„Zdd„ Zd d!„ Z‡  ZS )%ÚRendererCairoc                    sD   || _ t| d| _d | _d | _t t tjdd¡¡| _	t
ƒ  ¡  d S )N)Úrendererr   )ÚdpiÚGraphicsContextCairoÚgcÚwidthÚheightr,   ÚContextÚImageSurfaceÚFORMAT_ARGB32Útext_ctxÚsuperÚ__init__)Úselfr2   ©Ú	__class__r   r    r<   L   s    ÿzRendererCairo.__init__c                 C   s¢   |  ¡ }t|dƒr.t|dƒr.| ¡ | ¡ f}n\t|dƒrN| ¡ }|j|jf}n<| ¡  | ¡  | 	¡ ^}}|rvt
dƒ‚|j|jf}| ¡  || j_|\| _| _d S )NÚ	get_widthÚ
get_heightÚget_extentszCannot infer surface size)Ú
get_targetÚhasattrr@   rA   rB   r5   r6   ÚsaveZ
reset_clipZcopy_clip_rectangle_listÚ	TypeErrorÚrestorer4   r   )r=   r   ÚsurfaceÚsizeÚextÚrectÚrestr   r   r    Úset_contextU   s    
zRendererCairo.set_contextú3.6rM   )Úalternativec                 C   s   t  |¡| j_d S ©N)r,   r7   r4   r   )r=   rH   r   r   r    Úset_ctx_from_surfaceg   s    z"RendererCairo.set_ctx_from_surfacec                 C   s   || _ || _d S rP   ©r5   r6   )r=   r5   r6   r   r   r    Úset_width_heightk   s    zRendererCairo.set_width_heightc                 C   sz   |d k	rn|  ¡  t|ƒdks |r>| |d |d |d |¡ n | |d |d |d |d ¡ | ¡  | ¡  | ¡  d S )Nr   r   r   r   )rE   ÚlenÚset_source_rgbaÚfill_preserverG   Ústroke)r=   r   Zfill_cÚalphaZalpha_overridesr   r   r    Ú_fill_and_strokep   s     zRendererCairo._fill_and_strokeNc                 C   st   |j }|d kr"| ¡ d kr"| ¡ nd }|tƒ  dd¡ d| j¡ }| ¡  t||||ƒ |  	||| 
¡ | ¡ ¡ d S ©Nr   éÿÿÿÿr   )r   Z	get_hatchZclip_extentsr   ÚscaleÚ	translater6   Únew_pathr!   rY   Ú	get_alphaÚget_forced_alpha)r=   r4   r   r   ÚrgbFacer   r   r   r   r    Ú	draw_path{   s$    ÿ
ÿ
þÿ   ÿzRendererCairo.draw_pathc                 C   s<  |j }| ¡  t|||tƒ  dd¡ ƒ | ¡ }| ¡ \}}	}
}|dkrj|	dkrj|
dkrj|dkrjd}d }nd}|tƒ  dd¡ d| j¡ }| ¡  t	|j
|ddƒD ]t\}\}}t|ƒr¤|dd … \}}| ¡  | ||¡ | |¡ | ¡  |s |d dkr¤|  ||| ¡ | ¡ ¡ q¤|s8|  ||| ¡ | ¡ ¡ d S )	Nr   r[   r   FT)Úsimplifyr   iè  )r   r^   r!   r   r\   Zcopy_path_flatZfill_extentsr]   r6   Ú	enumerater   rT   rE   Zappend_pathrG   rY   r_   r`   )r=   r4   Zmarker_pathZmarker_transr   r   ra   r   Úx1Úy1Zx2Úy2ZfilledÚiZverticesÚcodesÚxÚyr   r   r    Údraw_markers‰   sH     ÿÿ
   ÿ   ÿzRendererCairo.draw_markersc                 C   s’   t  |d d d… ¡}tj | ¡ jtj|jd |jd |jd d ¡}|j	}| j
| |jd  }| ¡  | |t|ƒt|ƒ¡ | ¡  | ¡  d S )Nr[   r   r   é   )r   Z._unmultiplied_rgba8888_to_premultiplied_argb32r,   r8   Zcreate_for_dataZravelÚdatar9   Úshaper   r6   rE   Zset_source_surfaceÚfloatZpaintrG   )r=   r4   rj   rk   ZimrH   r   r   r   r    Ú
draw_image¹   s       þzRendererCairo.draw_imageFc	                 C   s¸   |r|   ||||||¡ nš|j}	|	 ¡  |	 ||¡ |	 ¡  |	jt|ƒŽ  |	 |  | 	¡ ¡¡ t
 ¡ }
|
 tjd rzt
jnt
j¡ |	 |
¡ |r¢|	 t | ¡¡ |	 |¡ |	 ¡  d S )Nztext.antialiased)Ú_draw_mathtextr   r^   r   rE   Úselect_font_facer/   Úset_font_sizeÚpoints_to_pixelsÚget_size_in_pointsr,   ZFontOptionsÚset_antialiasÚmplZrcParamsÚANTIALIAS_DEFAULTÚANTIALIAS_NONEZset_font_optionsÚrotater   Údeg2radÚ	show_textrG   )r=   r4   rj   rk   Úsr'   ÚangleÚismathZmtextr   Úoptsr   r   r    Ú	draw_textÆ   s$    þ

zRendererCairo.draw_textc                 C   sø   |j }| jj || j|¡\}}	}
}}| ¡  | ||¡ |rN| t 	| ¡¡ |D ]T\}}}}}| 
¡  | || ¡ |jtt|ƒƒŽ  | |  |¡¡ | t|ƒ¡ qR|D ]>\}}}}| 
¡  | || || ¡ | ddd¡ | ¡  q¬| ¡  d S ©Nr   )r   Ú
_text2pathÚmathtext_parserÚparser2   rE   r]   r{   r   r|   r^   r   rs   r/   r   rt   ru   r}   ÚchrÚ	rectangleÚset_source_rgbrV   rG   )r=   r4   rj   rk   r~   r'   r   r   r5   r6   ÚdescentZglyphsZrectsZfontZfontsizeÚidxÚoxZoyÚwÚhr   r   r    rr   à   s*    ÿ
ÿ
zRendererCairo._draw_mathtextc                 C   s   | j | jfS rP   rR   ©r=   r   r   r    Úget_canvas_width_heightú   s    z%RendererCairo.get_canvas_width_heightc                    s    |dkrt ƒ  |||¡S |rB| jj || j|¡^}}}}|||fS | j}| ¡  |jt	|ƒŽ  | 
|  | ¡ ¡¡ | |¡dd… \}	}
}| ¡  |
|||	 fS )NÚTeXr   rm   )r;   Úget_text_width_height_descentr„   r…   r†   r2   r:   rE   rs   r/   rt   ru   rv   Ztext_extentsrG   )r=   r~   r'   r€   r5   r6   rŠ   Ú_r   Z	y_bearingr   rŽ   r>   r   r    r’   þ   s    ÿ

z+RendererCairo.get_text_width_height_descentc                 C   s"   | j j ¡  d| j _d| j _| j S )Nr   F)r4   r   rE   Ú_alphaZ_forced_alphar   r   r   r    Únew_gc  s    zRendererCairo.new_gcc                 C   s   |d | j  S )NéH   )r2   )r=   r   r   r   r    ru     s    zRendererCairo.points_to_pixels)N)N)FN)Ú__name__Ú
__module__Ú__qualname__r<   rM   r   Ú
deprecatedrQ   rS   rY   rb   rl   rq   r‚   rr   r   r’   r•   ru   Ú__classcell__r   r   r>   r    r0   K   s"   	


 ÿ
0
r0   c                       s¢   e Zd ZejejejdœZejej	ej
dœZ‡ fdd„Zdd„ Z‡ fdd„Zd	d
„ Zdd„ Zdd„ Zdd„ Zdd„ Zd‡ fdd„	Zdd„ Zdd„ Zdd„ Z‡  ZS )r3   )ZbevelZmiterÚround)ZbuttZ
projectingrœ   c                    s   t ƒ  ¡  || _d S rP   )r;   r<   r1   )r=   r1   r>   r   r    r<   /  s    
zGraphicsContextCairo.__init__c                 C   s   | j  ¡  d S rP   )r   rG   r   r   r   r    rG   3  s    zGraphicsContextCairo.restorec                    sh   t ƒ  |¡ |  ¡ }| j}|  ¡ rB| j |d |d |d |¡ n"| j |d |d |d |d ¡ d S )Nr   r   r   r   )r;   Ú	set_alphar_   Ú_rgbr`   r   rU   )r=   rX   r”   Úrgbr>   r   r    r   6  s     zGraphicsContextCairo.set_alphac                 C   s   | j  |rtjntj¡ d S rP   )r   rw   r,   ry   rz   )r=   r   r   r   r    Úset_antialiased?  s    ÿz$GraphicsContextCairo.set_antialiasedc                 C   s"   | j  tj| j|d¡ || _d S )N)Zcapstyle)r   Zset_line_capr   Úcheck_getitemÚ_capdZ	_capstyle)r=   Úcsr   r   r    Úset_capstyleC  s    z!GraphicsContextCairo.set_capstylec                 C   sR   |sd S t  |j¡\}}}}| j}| ¡  | || jj| | ||¡ | ¡  d S rP   )	r   rœ   Zboundsr   r^   rˆ   r1   r6   r   )r=   rˆ   rj   rk   r   rŽ   r   r   r   r    Úset_clip_rectangleG  s    z'GraphicsContextCairo.set_clip_rectanglec                 C   sX   |sd S |  ¡ \}}| j}| ¡  |tƒ  dd¡ d| jj¡ }t|||ƒ | 	¡  d S rZ   )
Zget_transformed_path_and_affiner   r^   r   r\   r]   r1   r6   r!   r   )r=   r   ZtpathZaffiner   r   r   r    Úset_clip_pathP  s    ÿz"GraphicsContextCairo.set_clip_pathc                 C   sF   ||f| _ |d kr"| j g d¡ n | j t| j t |¡¡ƒ|¡ d S rƒ   )Z_dashesr   Zset_dashÚlistr1   ru   r   r   )r=   ÚoffsetÚdashesr   r   r    Ú
set_dashes[  s    
þzGraphicsContextCairo.set_dashesNc                    s>   t ƒ  ||¡ t| jƒdkr,| jj| jŽ  n| jj| jŽ  d S ©Nr   )r;   Úset_foregroundrT   rž   r   r‰   rU   )r=   ZfgZisRGBAr>   r   r    r¬   d  s    z#GraphicsContextCairo.set_foregroundc                 C   s   | j  ¡  ¡ d d… S r«   )r   Ú
get_sourceZget_rgbar   r   r   r    Úget_rgbk  s    zGraphicsContextCairo.get_rgbc                 C   s"   | j  tj| j|d¡ || _d S )N)Z	joinstyle)r   Zset_line_joinr   r¡   Ú_joindZ
_joinstyle)r=   Újsr   r   r    Úset_joinstylen  s    z"GraphicsContextCairo.set_joinstylec                 C   s"   t |ƒ| _| j | j |¡¡ d S rP   )rp   Z
_linewidthr   Zset_line_widthr1   ru   )r=   r   r   r   r    Úset_linewidthr  s    
z"GraphicsContextCairo.set_linewidth)N)r—   r˜   r™   r,   ZLINE_JOIN_BEVELZLINE_JOIN_MITERZLINE_JOIN_ROUNDr¯   ZLINE_CAP_BUTTZLINE_CAP_SQUAREZLINE_CAP_ROUNDr¢   r<   rG   r   r    r¤   r¥   r¦   rª   r¬   r®   r±   r²   r›   r   r   r>   r    r3   "  s(   ýý			r3   c                   @   s   e Zd Zdd„ ZdS )Ú_CairoRegionc                 C   s   || _ || _d S rP   )Ú_slicesÚ_data)r=   Zslicesrn   r   r   r    r<   x  s    z_CairoRegion.__init__N)r—   r˜   r™   r<   r   r   r   r    r³   w  s   r³   c                   @   sŠ   e Zd Zedd„ ƒZdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Z	e	Z
dd„ Zddœdd„Ze ed¡Ze ed¡Ze ed¡Ze ed¡ZdS )ÚFigureCanvasCairoc                 C   s   t | dƒst| jjƒ| _| jS )NÚ_cached_renderer)rD   r0   Úfigurer2   r·   r   r   r   r    Ú	_renderer~  s    
zFigureCanvasCairo._rendererc                 C   s   | j S rP   )r¹   r   r   r   r    Úget_rendererˆ  s    zFigureCanvasCairo.get_rendererc                 C   s  | j jj ¡ }t|tjƒs"tdƒ‚| ¡ }| 	¡ }t
 |j¡}t
 |j¡}t
 ||j ¡}t
 ||j ¡}d|kr¢||kr¢|j|jkr¢d|kr¢||kr¢|j|jksªtdƒ‚t||t|| dƒ ƒt||t|| dƒ ƒf}	t | ¡ tj¡ ||f¡|	  ¡ }
t|	|
ƒS )Nz;copy_from_bbox only works when rendering to an ImageSurfacer   zInvalid bbox)r¹   r4   r   rC   Ú
isinstancer,   r8   ÚRuntimeErrorr@   rA   ÚmathÚceilÚx0Úfloorre   rf   Úy0Ú
ValueErrorÚsliceÚmaxr   Ú
frombufferÚget_dataÚuint32ÚreshapeÚcopyr³   )r=   ZbboxrH   ÚswÚshr¿   re   rÁ   rf   Úslsrn   r   r   r    Úcopy_from_bbox‹  s4    ÿÿÿ
ÿ0ÿÿz FigureCanvasCairo.copy_from_bboxc                 C   s’   | j jj ¡ }t|tjƒs"tdƒ‚| ¡  | 	¡ }| 
¡ }|j\}}|jt | ¡ tj¡ ||f¡||f< | |j|j|j|j |j|j ¡ d S )Nz;restore_region only works when rendering to an ImageSurface)r¹   r4   r   rC   r»   r,   r8   r¼   Úflushr@   rA   r´   rµ   r   rÅ   rÆ   rÇ   rÈ   Zmark_dirty_rectangleÚstartÚstop)r=   ÚregionrH   rÊ   rË   ZslyZslxr   r   r    Úrestore_regionž  s*    ÿ
ÿÿÿ  
 
ÿz FigureCanvasCairo.restore_regionc                 C   s   |   ¡  |¡ d S rP   )Ú_get_printed_image_surfaceZwrite_to_png)r=   Úfobjr   r   r    Ú	print_png¬  s    zFigureCanvasCairo.print_pngc              	   C   s>   |   ¡ \}}|  ¡  ¡ }| t t |¡ ||df¡¡¡ d S )Nrm   )	Úget_width_heightrÓ   rÆ   Úwriter   Z._premultiplied_argb32_to_unmultiplied_rgba8888r   r   rÈ   )r=   rÔ   r5   r6   Úbufr   r   r    Ú
print_rgba¯  s
    ÿzFigureCanvasCairo.print_rgbac                 C   sL   | j j| j_|  ¡ \}}t tj||¡}| j t |¡¡ | j  	| j¡ |S rP   )
r¸   r2   r¹   rÖ   r,   r8   r9   rM   r7   Údraw)r=   r5   r6   rH   r   r   r    rÓ   ·  s    z,FigureCanvasCairo._get_printed_image_surfaceZportrait)Úorientationc                C   s~  d}|| j _| j  ¡ \}}|| ||  }}|dkr>|| }}|dkrhttdƒsXtdƒ‚t |||¡}	n’|dkr’ttdƒs‚tdƒ‚t |||¡}	nh|d	krìttd
ƒs¬tdƒ‚|dkrÜt|t	ƒrÌt
 |d¡}nt
jd d|d}t |||¡}	ntd |¡ƒ‚| j j| j_| j t |	¡¡ | jjj}
|dkrJ|
 tjd ¡ |
 d| ¡ | j  | j¡ |
 ¡  |	 ¡  |dkrz| ¡  d S )Nr–   Z	landscapeÚpsÚ	PSSurfacez3cairo has not been compiled with PS support enabledÚpdfÚ
PDFSurfacez4cairo has not been compiled with PDF support enabled)ÚsvgÚsvgzÚ
SVGSurfacez4cairo has not been compiled with SVG support enabledrá   Úwb)ÚfileobjzUnknown format: {!r}r   r   )r¸   r2   Zget_size_inchesrD   r,   r¼   rÝ   rß   r»   ÚstrÚgzipÚGzipFilerâ   rÂ   Úformatr¹   rM   r7   r4   r   r{   r   Úpir]   rÚ   Z	show_pageÚfinishÚclose)r=   ÚfmtrÔ   rÛ   r2   Zw_inZh_inZwidth_in_pointsZheight_in_pointsrH   r   r   r   r    Ú_save¿  sH     ÿ






zFigureCanvasCairo._saverÞ   rÜ   rà   rá   N)r—   r˜   r™   Úpropertyr¹   rº   rÍ   rÒ   rÕ   rÙ   Z	print_rawrÓ   rí   Ú	functoolsÚpartialmethodZ	print_pdfZprint_psZ	print_svgZ
print_svgzr   r   r   r    r¶   }  s   
	3r¶   rN   c                   @   s   e Zd Zdd„ ZdS )Ú_RendererGTKCairoc                 C   sJ   t jdkr>t|t jƒs>t jjt j dt|ƒtj	 ¡d dd}|| j
_d S )NÚ	cairocffiz
cairo_t **r   T)Zincref)r,   r—   r»   r7   Z_from_pointerÚffiÚcastÚidÚobjectÚ__basicsize__r4   r   )r=   r   r   r   r    rM   ú  s    

ÿþþüz_RendererGTKCairo.set_contextN)r—   r˜   r™   rM   r   r   r   r    rñ   ø  s   rñ   c                   @   s   e Zd ZejZeZeZ	dS )Ú_BackendCairoN)
r—   r˜   r™   r,   ÚversionZbackend_versionr¶   ZFigureCanvasr
   ZFigureManagerr   r   r   r    rø     s   rø   )N)'Ú__doc__rï   ræ   r½   Únumpyr   r,   Úversion_infoÚImportErrorrò   ÚerrZ
matplotlibrx   Ú r   r   r   Zmatplotlib.backend_basesr   r	   r
   r   r   Zmatplotlib.font_managerr   Zmatplotlib.pathr   Zmatplotlib.transformsr   r!   r/   r0   r3   r³   r¶   rš   rñ   Zexportrø   r   r   r   r    Ú<module>   sD   
ÿþ
 XU{