U
eI7 @ s d dl mZ d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl
m Z d dl m
Z zd 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 ZdZd$ddZdd Zd d
Zd%d d Zd
d Zdd Z e
dZ d
dddZd&ddZd'ddZdd Zd(d d Zd d Zd d! Zd)d"d#ZdS )*AnnotationCCodeWriterzComplete cythonized codeNTFc C s t j| |||d || _|d kr`t | _d | _tttt| _ tttt
| _ tttt | _
n(|j| _|j | _ |j | _ |j
| _
|j| _d S )N)copy_formatting)r
__init__show_entire_c_coder annotation_bufferlast_annotated_posr r list annotationsstrcodesetscopes)self create_frombufferr r source_desc r 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.
)r9 watermarkr% Z more_infoz'Raw output: %s
)
formatr
fromtimestampr5 textwrapdedentr< r rQ rC rD rE r, )r rN rJ 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
NrP r> ) rB strip_get_line_coveragedictr r extendrX _save_annotation_bodyrZ r8 ) r r rM rN rJ rK rV
covered_linesannotation_itemsr rW r r r rI s z&AnnotationCCodeWriter._save_annotationc C s d }| dD ]R}|ds q|d|ksBtj|d|krL|} qbq||dr|}q|d krnd S tdd | dD S )Nz.//classr% c s s* | ]"}t |d t |dfV qdS )numberhitsN)r5 rB ).0r* r r r s z;AnnotationCCodeWriter._get_line_coverage..z
lines/line)iterfindrB rC rD abspathendswithr] )r rK rJ Z
coverage_dataentryr r r r\ s
z(AnnotationCCodeWriter._get_line_coveragec C s z,ddl m} ddlm}m} ddlm} W n tk
rH t| Y S X |dkr`|ddd}n |dkrv|ddd}nt|S ||||d d
}|S ) Nr ) highlight) CythonLexerCppLexerr0 cythonF)ZstripnlZstripallc/cppT)Znowrap) Zpygmentsrj Zpygments.lexersrk rl r6 r1 r7 html_escape) r r languagerj rk rl r1 lexerZ html_coder r r
_htmlify_code s" z#AnnotationCCodeWriter._htmlify_codec s dg}d}t dd d D j}| d fdd} | |d }
ttt|
} |s`d }t|
d D ] \} }
z || }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 rP| |krP|| }|d k rP|rLdnd}|dj|||| | |
|d |rj|dj|||d qj|d | jr|d d}||j| jtjd | | j d }|d j|d |d |S )!Nzu
/* … */
c s s | ]}|d fV qdS )r Nr )rd namer r r re 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_nameZcallsr r r. s z=AnnotationCCodeWriter._save_annotation_body..annotaterm r r> 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})r{ r} r zz2
+ {title}
)r titlern z%
{code})r z