B
\$ @ sT d Z ddlmZ ede dd Zdd Zdd Zd
d Zd d
ZG dd dZ dS )a
This module implements code highlighting of numba function annotations.
Example:
>>> import numba
... from numba_annotate import Annotate
... @numba.jit
... def test(q):
... res = 0
... for i in range(q):
... res += i
... return res
...
... test(10)
... Annotate(test)
The last line will return an HTML and/or ANSI representation that will be
displayed accordingly in IPython/Jupyter.
)warnzFThe pretty_annotate functionality is experimental and might change APIc C sn y(ddl m} ddlm} ddlm} W n tk
rD tdY nX | }|d|dd}|| ||}| S )Nr ) highlight) PythonLexer)
HtmlFormatterz%please install the 'pygments' packageT)Z noclassesstyleZnowrap)pygmentsr pygments.lexersr pygments.formattersr ImportError
splitlines)coder r r r pylexhfres r 4lib/python3.7/site-packages/numba/pretty_annotate.pyhllines s r c C sj y(ddl m} ddlm} ddlm} W n tk
rD tdY nX | }||d}|| ||}| S )Nr )r )r )TerminalFormatterz%please install the 'pygments' package)r )r r r r r r r
r )r r r r r r
r r r r r htlines* s
r c C s6 yddl m} W n tk
r, tdY nX | dS )Nr )Templatez#please install the 'jinja2' packagea
{%- for func_key in func_data.keys() -%}
Function name: [34m{{func_data[func_key]['funcname']}}[39;49;00m
{%- if func_data[func_key]['filename'] -%}
{{'
'}}In file: [34m{{func_data[func_key]['filename'] -}}[39;49;00m
{%- endif -%}
{{'
'}}With signature: [34m{{func_key[1]}}[39;49;00m
{{- "
" -}}
{%- for num, line, hl, hc in func_data[func_key]['pygments_lines'] -%}
{{-'
'}}{{ num}}: {{hc-}}
{%- if func_data[func_key]['ir_lines'][num] -%}
{%- for ir_line, ir_line_type in func_data[func_key]['ir_lines'][num] %}
{{-'
'}}--{{- ' '*func_data[func_key]['python_indent'][num]}}
{{- ' '*(func_data[func_key]['ir_indent'][num][loop.index0]+4)
}}{{ir_line }}[41m{{ir_line_type-}}[39;49;00m
{%- endfor -%}
{%- endif -%}
{%- endfor -%}
{%- endfor -%}
)jinja2r r
Z
ansi_template)r r r r get_ansi_template9 s r c C s6 yddl m} W n tk
r, tdY nX | dS )Nr )r z#please install the 'jinja2' packagea
{% for func_key in func_data.keys() %}
Function name: {{func_data[func_key]['funcname']}}
{% if func_data[func_key]['filename'] %}
in file: {{func_data[func_key]['filename']|escape}}
{% endif %}
with signature: {{func_key[1]|e}}
{%- for num, line, hl, hc in func_data[func_key]['pygments_lines'] -%}
{%- if func_data[func_key]['ir_lines'][num] %}
{{num}}:
{{' '*func_data[func_key]['python_indent'][num]}}{{hl}}
{%- for ir_line, ir_line_type in func_data[func_key]['ir_lines'][num] %}
{{- ' '*func_data[func_key]['python_indent'][num]}}
{{ ' '*func_data[func_key]['ir_indent'][num][loop.index0]}}{{ir_line|e -}}
{{ir_line_type}}
|
{%- endfor -%}
|
{% else -%}
{{num}}:
{{' '*func_data[func_key]['python_indent'][num]}}{{hl}}
|
{%- endif -%}
{%- endfor -%}
{% endfor %}
)r r r
)r r r r get_html_templateT s r c C s> | d }d}x,| d D ] \}}|d|| | d }qW |S )z
Extract the code from the Numba annotation datastructure.
Pygments can only highlight full multi-line strings, the Numba
annotation is list of single lines, with indentation removed.
Z
python_indent python_lines
r )Z
annotationZ
ident_dictsnlr r r reform_code s
r c @ s( e Zd ZdZdd Zdd Zdd ZdS ) Annotatea
Construct syntax highlighted annotation for a given jitted function:
Example:
>>> import numba
... from numba_annotate import Annotate
... @numba.jit
... def test(q):
... res = 0
... for i in range(q):
... res += i
... return res
...
... test(10)
... Annotate(test)
Function annotations persist across compilation for newly encountered
type signatures and as a result annotations are shown for all signatures.
c
K s | dd}|jstdx|jD ]}||}q"W || _xN| D ]B\}}tt||}tt||} dd t |d || D |d< qDW d S )Nr defaultz5function need to be jitted for at least one signaturec S s" g | ]\\}}}}||||fqS r r ).0abcdr r r
s z%Annotate.__init__..r Zpygments_lines)
getZ
signatures
ValueErrorZget_annotation_infoannitemsr r r zip)
selfZfunctionkwargsr Zsigr+ kvr restr r r __init__ s zAnnotate.__init__c C s t j| jdS )N) func_data)r renderr+ )r. r r r _repr_html_ s zAnnotate._repr_html_c C s t j| jdS )N)r4 )r r5 r+ )r. r r r __repr__ s zAnnotate.__repr__N)__name__
__module__ __qualname____doc__r3 r6 r7 r r r r r! s r! N)
r; warningsr
FutureWarningr r r r r r! r r r r s