o
8ήcD% @ sT d Z ddlmZ ede dd Zdd Zdd Zd
d Zd d
ZG dd dZ dS )zI
This module implements code highlighting of numba function annotations.
)warnzFThe pretty_annotate functionality is experimental and might change APIc C sj zddl m} ddlm} ddlm} W n ty tdw | } |d|dd}|| ||}| S )Nr highlight PythonLexer)
HtmlFormatter%please install the 'pygments' packageT) noclassesstylenowrap)pygmentsr pygments.lexersr pygments.formattersr ImportError
splitlines)coder
r r r pylexhfres r M/tmp/pip-target-vg8gfxp4/lib/python/numba/core/annotations/pretty_annotate.pyhllines
s r c C sf zddl m} ddlm} ddlm} W n ty tdw | } ||d}|| ||}| S )Nr r r )TerminalFormatterr )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 . z ddl m} W | dS ty tdw )Nr Template#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
ansi_templater r r r get_ansi_template' r c C r )Nr r r a4
{% 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_templateB r! r" c C s: | d }d}| d D ]\}}|d|| | d }q
|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.
python_indent python_lines
r )
annotation
ident_dictsnlr r r reform_code s
r- c @ s* e Zd ZdZd ddZdd Zdd ZdS )
Annotatea
Construct syntax highlighted annotation for a given jitted function:
Example:
>>> import numba
>>> from numba.pretty_annotate import Annotate
>>> @numba.jit
... def test(q):
... res = 0
... for i in range(q):
... res += i
... return res
...
>>> test(10)
45
>>> Annotate(test)
The last line will return an HTML and/or ANSI representation that will be
displayed accordingly in Jupyter/IPython.
Function annotations persist across compilation for newly encountered
type signatures and as a result annotations are shown for all signatures
by default.
Annotations for a specific signature can be shown by using the
``signature`` parameter.
>>> @numba.jit
... def add(x, y):
... return x + y
...
>>> add(1, 2)
3
>>> add(1.3, 5.7)
7.0
>>> add.signatures
[(int64, int64), (float64, float64)]
>>> Annotate(add, signature=add.signatures[1]) # annotation for (float64, float64)
Nc
K s| | dd}|js
td|j|d}|| _| D ]!\}}tt||}tt||} dd t |d || D |d< qd S ) Nr
defaultz5function need to be jitted for at least one signature) signaturec S s" g | ]
\\}}}}||||fqS r r ).0abcdr r r
s " z%Annotate.__init__..r% pygments_lines)
get
signatures
ValueErrorget_annotation_infoannitemsr r- r zip)
selffunctionr0 kwargsr
r< kvr restr r r __init__
s zAnnotate.__init__c C t j| jdS N) func_data)r" renderr< r? r r r _repr_html_ zAnnotate._repr_html_c C rF rG )r rI r< rJ r r r __repr__ rL zAnnotate.__repr__)N)__name__
__module__ __qualname____doc__rE rK rM r r r r r. s
(
r. N)
rQ warningsr
FutureWarningr r r r" r- r. r r r r s