B
\2 @ s d dl mZ d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl
m Z d dl m
Z yd dlmZ W n ek
r d dlmZ Y nX dd lmZ dd
lmZ d d lmZ G d
d deZeddjZeddej jZ G dd de Z dS ) )absolute_importN)datetime)partial) defaultdict)escape)StringIO )Version) CCodeWriter )Utilsc @ s e Zd Zd"ddZdd Zdd Zd#d d
Zd d Zd
d Ze
dZ d d
ddZd$ddZd%ddZdd Zd&ddZd d Zd d Zd'd d!ZdS )(AnnotationCCodeWriterNTc C s t j| |||d |d krZt | _d | _tttt| _tttt | _
tttt | _ n(|j| _|j| _|j
| _
|j | _ |j| _d S )N)copy_formatting)
r
__init__r annotation_bufferlast_annotated_posr r list annotationsstrcodesetscopes)self create_frombufferr r 7lib/python3.7/site-packages/Cython/Compiler/Annotate.pyr s z AnnotationCCodeWriter.__init__c C s t |||S )N)r
)r r r r r r r
create_new- s z AnnotationCCodeWriter.create_newc C s t | | | j| d S )N)r
writer )r sr r r r 0 s zAnnotationCCodeWriter.writec C s |d k rFt | || | jrF| jjrF| j|d j |d | jj | jrz| j\}}}| j|j }|| | j
7 < t | _ || _d S )Nr r ) r
mark_posZ funcstateZscoper filenameaddr r r getvaluer )r posZtraceZ source_descline_Zpos_coder r r r 4 s " z AnnotationCCodeWriter.mark_posc C s* | j |d j |d |d |f d S )Nr r r )r r! append)r r$ itemr r r annotateA s z AnnotationCCodeWriter.annotatec C s | j g}x:tdD ].}dtdd|d }|d||f qW yddlm} W n tk
rh Y nX || d d |S )
z2css template will later allow to choose a colormap zFFFF%02xr g $@z).cython.score-%d {background-color: #%s;}r )
HtmlFormatterz.cython
)
_css_templaterangeintr' pygments.formattersr+ ImportErrorZget_style_defsjoin)r cssiZcolorr+ r r r _cssD s zAnnotationCCodeWriter._cssa
body.cython { font-family: courier; font-size: 12; }
.cython.tag { }
.cython.line { margin: 0em }
.cython.code { font-size: 9; color: #444444; display: none; margin: 0px 0px 0px 8px; border-left: 8px none; }
.cython.line .run { background-color: #B0FFB0; }
.cython.line .mis { background-color: #FFB0B0; }
.cython.code.run { border-left: 8px solid #B0FFB0; }
.cython.code.mis { border-left: 8px solid #FFB0B0; }
.cython.code .py_c_api { color: red; }
.cython.code .py_macro_api { color: #FF7000; }
.cython.code .pyx_c_api { color: #FF3000; }
.cython.code .pyx_macro_api { color: #FF7000; }
.cython.code .refnanny { color: #FFA000; }
.cython.code .trace { color: #FFA000; }
.cython.code .error_goto { color: #FFA000; }
.cython.code .coerce { color: #008000; border: 1px dotted #008000 }
.cython.code .py_attr { color: #FF0000; font-weight: bold; }
.cython.code .c_attr { color: #0000FF; }
.cython.code .py_call { color: #FF0000; font-weight: bold; }
.cython.code .c_call { color: #0000FF; }
z onclick="{0}"zg(function(s){ s.display = s.display === 'block' ? 'none' : 'block'})(this.nextElementSibling.style) c
C s t |}| }W d Q R X | j|i }t tj|}tj |d d }t
j |ddd } | |
||||| W d Q R X d S )Nr z.htmlwzUTF-8)encoding)r Zopen_source_filereadr getZdecode_filenameospathbasenamesplitextcodecsopenr _save_annotation)
r source_filenameZtarget_filename coverage_xmlfr generated_codec_fileZ
html_filenameZ
out_bufferr r r save_annotationu s z%AnnotationCCodeWriter.save_annotationc C sl d}|r"dj tt|d d}tdj | tj|rFt j
|nd|dg}|rh| d||f |S )Nr7 z$ with coverage data from {timestamp}i ) timestampa
Cython: {filename}
Generated by Cython {watermark}{more_info}
Yellow lines hint at Python interaction.
Click on a line that starts with a "+" to see the C code that Cython generated for it.
)r3 watermarkr! Z more_infoz'Raw output: %s
)
formatr Z
fromtimestampr/ textwrapdedentr5 r rJ r< r= r> r' )r rG rC coverage_timestampZ
coverage_infooutlistr r r _save_annotation_header s
z-AnnotationCCodeWriter._save_annotation_headerc C s dS )N)z
r )r r r r _save_annotation_footer s z-AnnotationCCodeWriter._save_annotation_footerc C s |dk r*|r*| dd }| ||}nd }}t| j| }t| j| } g }
|
| ||| |
| |||| | |
| d
|
S )a#
lines : original cython source code split by lines
generated_code : generated c code keyed by line number in original file
target filename : name of the file in which to store the generated html
c_file : filename in which the c_code has been written
NrI r7 ) r; strip_get_line_coveragedictr r extendrP _save_annotation_bodyrQ r2 ) r r rF rG rC rD rN
covered_linesannotation_itemsr rO r r r rB s z&AnnotationCCodeWriter._save_annotationc C s d }x^| dD ]P}|ds q|d|ksDtj|d|krL|}P q||dr|}qW |d krpd S tdd | dD S )Nz.//classr! c s s* | ]"}t |d t |dfV qdS )ZnumberhitsN)r/ r; ).0r% r r r s z;AnnotationCCodeWriter._get_line_coverage..z
lines/line)Ziterfindr; r< r= abspathendswithrT )r rD rC Z
coverage_dataentryr r r rS s
z(AnnotationCCodeWriter._get_line_coveragec C s` y(ddl m} ddlm} ddlm} W n tk
r@ t|S X |||ddd|dd}|S ) Nr ) highlight) CythonLexer)r+ F)ZstripnlZstripallT)Znowrap)Zpygmentsr_ Zpygments.lexersr` r0 r+ r1 html_escape)r r r_ r` r+ Z html_coder r r
_htmlify_code s
z#AnnotationCCodeWriter._htmlify_codec s dg}d}t dd d D j}| d fdd} | | }
ttt|
} |s^d }x0t|
dD ] \} }
y || }W n t k
r d }Y n.X t
||}| |r|t|d }t |}| t
| |}d
d d d
d d }|r| j}d}nd }d}d }|d k rR| |krR|| }|d k rR|rNdnd}|dj|||| | |
|d |rl|dj|||d qlW |d |S )Nzu
/* … */
c s s | ]}|d fV qdS )r Nr )rZ namer r r r[ s z>AnnotationCCodeWriter._save_annotation_body.
.zGrefnanny trace py_macro_api py_c_api pyx_macro_api pyx_c_api error_gotoc s( | j } | d7 < d|| |f S )Nr z%s) lastgroupgroup)matchZ
group_name)callsr r r) s z=AnnotationCCodeWriter._save_annotation_body..annotater r7 Zpy_c_apir Z pyx_c_apiZ py_macro_apiZ
pyx_macro_api+z runZmisz{expandsymbol}{line:0{lineno_width}d}: {code}
)score expandsymbolcovered lineno_widthr% r onclickz={code})rk rm r z