o
    e8                     @   s  d dl mZ d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm	Z	 d dl
mZ d dlmZ zd dlmZ W n eyI   d dlmZ Y nw dd	lmZ dd
lmZ ddlmZ G dd deZeddjZeddejjZG dd deZdS )    )absolute_importN)datetime)partial)defaultdict)escape)StringIO   )Version)CCodeWriter   )Utilsc                   @   s   e Zd ZdZd$ddZdd Zd	d
 Zd%ddZdd Zdd Z	e
dZddddZd&ddZd&ddZdd Zd'ddZdd Zd d! Zd&d"d#ZdS )(AnnotationCCodeWriterzComplete cythonized codeNTFc                 C   s   t j| |||d || _|d u r1t | _d | _tttt| _	tttt
| _tttt| _d S |j| _|j	| _	|j| _|j| _|j| _d S )N)copy_formatting)r
   __init__show_entire_c_coder   annotation_bufferlast_annotated_posr   r   listannotationsstrcodesetscopes)selfcreate_frombufferr   r   source_desc r   HD:\Projects\ConvertPro\env\Lib\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_new1   s   z AnnotationCCodeWriter.create_newc                 C   s   | j | | j| d S r   )r   writer   )r   sr   r   r   _write_to_buffer4   s   z&AnnotationCCodeWriter._write_to_bufferc                 C   s   |d ur#t | || | jr#| jjr#| j|d j |d  | jj | jr=| j\}}}| j|j }||  | j	
 7  < t | _	|| _d S )Nr   r   )r
   mark_posZ	funcstatescoper   filenameaddr   r   r   getvaluer   )r   postracer   line_Zpos_coder   r   r   r$   8   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   annotateE   s   *zAnnotationCCodeWriter.annotatec                 C   s   | j g}tdD ]}dtdd|d    }|d||f  qzddlm} W n	 ty1   Y n
w || d	 d
|S )z2css template will later allow to choose a colormap   zFFFF%02xg     o@g      ?g      $@z).cython.score-%d {background-color: #%s;}r   HtmlFormatterz.cython
)	_css_templaterangeintr-   pygments.formattersr2   ImportErrorZget_style_defsjoin)r   cssicolorr2   r   r   r   _cssH   s   
zAnnotationCCodeWriter._cssa  
        body.cython { font-family: courier; font-size: 12; }

        .cython.tag  {  }
        .cython.line { color: #000000; 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    n1 sw   Y  | j|i }t tj|}tj	|d d }t
j|ddd}	|	| ||||| W d    d S 1 sSw   Y  d S )Nr   z.htmlwzUTF-8)encoding)r   Zopen_source_filereadr   getZdecode_filenameospathbasenamesplitextcodecsopenr!   _save_annotation)
r   source_filenameZtarget_filenamecoverage_xmlfr   generated_codec_fileZhtml_filenameZ
out_bufferr   r   r   save_annotationy   s   
"z%AnnotationCCodeWriter.save_annotationc                 C   sl   d}|rdj tt|d d}tdj |  tj|r#t	j
|nd|dg}|r4|d||f  |S )Nr?   z$ with coverage data from {timestamp}i  )	timestampa              <!DOCTYPE html>
            <!-- Generated by Cython {watermark} -->
            <html>
            <head>
                <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
                <title>Cython: {filename}</title>
                <style type="text/css">
                {css}
                </style>
            </head>
            <body class="cython">
            <p><span style="border-bottom: solid 1px grey;">Generated by Cython {watermark}</span>{more_info}</p>
            <p>
                <span style="background-color: #FFFF00">Yellow lines</span> hint at Python interaction.<br />
                Click on a line that starts with a "<code>+</code>" to see the C code that Cython generated for it.
            </p>
            )r:   	watermarkr&   Z	more_infoz'<p>Raw output: <a href="%s">%s</a></p>
)formatr   fromtimestampr6   textwrapdedentr=   r	   rR   rD   rE   rF   r-   )r   rO   rK   coverage_timestampZcoverage_infooutlistr   r   r   _save_annotation_header   s   z-AnnotationCCodeWriter._save_annotation_headerc                 C   s   dS )N)z</body></html>
r   r   r   r   r   _save_annotation_footer   s   z-AnnotationCCodeWriter._save_annotation_footerc              	   C   s   |dur|r| dd }| ||}nd }}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
        NrQ   r?   )rC   strip_get_line_coveragedictr   r   extendrY   _save_annotation_bodyr[   r9   )r   r   rN   rO   rK   rL   rW   covered_linesannotation_itemsr   rX   r   r   r   rJ      s   
z&AnnotationCCodeWriter._save_annotationc                 C   s   d }| dD ](}|dsq|d|ks!tj|d|kr%|} n||dr/|}q|d u r6d S tdd | dD S )Nz.//classr&   c                 s   s,    | ]}t |d t |dfV  qdS )numberhitsN)r6   rC   ).0r+   r   r   r   	<genexpr>   s
    
z;AnnotationCCodeWriter._get_line_coverage.<locals>.<genexpr>z
lines/line)ZiterfindrC   rD   rE   abspathendswithr^   )r   rL   rK   Zcoverage_dataentryr   r   r   r]      s    
z(AnnotationCCodeWriter._get_line_coveragec           	      C   s   zddl m} ddlm}m} ddlm} W n ty#   t| Y S w |dkr/|ddd}n|dkr:|ddd}nt|S ||||d	d
}|S )Nr   )	highlight)CythonLexerCppLexerr1   cythonF)stripnlstripallc/cppT)Znowrap)	pygmentsrj   Zpygments.lexersrk   rl   r7   r2   r8   html_escape)	r   r   languagerj   rk   rl   r2   lexerZ	html_coder   r   r   _htmlify_code   s"   z#AnnotationCCodeWriter._htmlify_codec                    s  dg}d}t dd d D j}| d   fdd}	| |d }
ttt|
}|s0d }t|
d	D ]\}}z|| }W n t	yJ   d
}Y nw t
||}||r]|t|d  }t|}|  t|	|}d d  d d    d   d  }|r| j}d}nd
}d}d
}|d ur||v r|| }|d ur|rdnd}|dj|||||| |d |r|dj|||d q5|d | jr|d d}||j| jtjd | | j d}|dj|d |d  |S )!Nz<div class="cython">u
   /* … */
c                 s   s    | ]}|d fV  qdS )r   Nr   )re   namer   r   r   rf      s    
z>AnnotationCCodeWriter._save_annotation_body.<locals>.<genexpr>zGrefnanny trace py_macro_api py_c_api pyx_macro_api pyx_c_api error_gotoc                    s(   | j } |  d7  < d|| |f S )Nr   z<span class='%s'>%s</span>)	lastgroupgroup)match
group_nameZcallsr   r   r/      s
   
z=AnnotationCCodeWriter._save_annotation_body.<locals>.annotaterm   r   r?      Zpy_c_apir   Z	pyx_c_apiZpy_macro_apiZpyx_macro_api+z&#xA0;runZmisz<pre class="cython line score-{score}"{onclick}>{expandsymbol}<span class="{covered}">{line:0{lineno_width}d}</span>: {code}</pre>
)scoreexpandsymbolcoveredlineno_widthr+   r   onclickz=<pre class='cython code score-{score} {covered}'>{code}</pre>)r   r   r   z</div>z<p><div class="cython">z2<pre class='cython line'{onclick}>+ {title}</pre>
)r   titlerp   z%<pre class='cython code'>{code}</pre>)r   z
</div></p>)r^   splitcopyr$   ru   
splitlineslenr   	enumerateKeyError_replace_pos_comment
startswithrr   _parse_code_onclick_attrr-   rS   rstripr   r   COMPLETE_CODE_TITLEr   r(   )r   Zcython_coderN   rb   r   ra   rX   Zpos_comment_markerZnew_calls_mapr/   linesr   kr+   Zc_coder   r   r   r   rd   Zonclick_titleZcomplete_code_as_htmlr   r{   r   r`      s   






z+AnnotationCCodeWriter._save_annotation_body)NNTFN)Tr   )NNN)__name__
__module____qualname__r   r   r    r#   r$   r/   r=   rU   rV   r4   rS   replacer   rP   rY   r[   rJ   r]   ru   r`   r   r   r   r   r      s(    





 
r   sm  (?P<refnanny>__Pyx_X?(?:GOT|GIVE)REF|__Pyx_RefNanny[A-Za-z]+)|(?P<trace>__Pyx_Trace[A-Za-z]+)|(?:(?P<pyx_macro_api>__Pyx_[A-Z][A-Z_]+)|(?P<pyx_c_api>(?:__Pyx_[A-Z][a-z_][A-Za-z_]*)|__pyx_convert_[A-Za-z_]*)|(?P<py_macro_api>Py[A-Z][a-z]+_[A-Z][A-Z_]+)|(?P<py_c_api>Py[A-Z][a-z]+_[A-Z][a-z][A-Za-z_]*))(?=\()|(?P<error_goto>(?:(?<=;) *if [^;]* +)?__PYX_ERR\([^)]+\))asciis)   ^\s*/\*(?:(?:[^*]|\*[^/])*\n)+\s*\*/\s*\nc                   @   s&   e Zd Zd
ddZdd Zdd Zd	S )AnnotationItemr?   r   c                 C   s   || _ || _|| _|| _d S r   )styletexttagsize)r   r   r   r   r   r   r   r   r   K  s   
zAnnotationItem.__init__c                 C   s   d| j | j| jf S )Nz)<span class='cython tag %s' title='%s'>%s)r   r   r   rZ   r   r   r   startQ  s   zAnnotationItem.startc                 C   s
   | j dfS )Nz</span>)r   rZ   r   r   r   endT  s   
zAnnotationItem.endN)r?   r   )r   r   r   r   r   r   r   r   r   r   r   I  s    
r   ) 
__future__r   rD   Zos.pathrerH   rU   r   	functoolsr   collectionsr   Zxml.sax.saxutilsr   rr   r   r8   ior?   r	   ZCoder
   r   r   compiledecodesubr   Mr   objectr   r   r   r   r   <module>   sD     	
