U
    hâËdÙ  ã                   @   sÂ   zd dl mZ W n  ek
r0   dZeeƒ‚Y nX d dlZd dlmZ d dlm	Z	 d dl
mZ d dlmZmZmZmZ d dlmZmZmZmZmZmZmZ d d	lmZ G d
d„ deƒZdd„ ZdS )é    )ÚDefaultStylez0Please install pygments to see highlighted dumpsN)Ú
ManniStyle)ÚMonokaiStyle)ÚNativeStyle)Ú
RegexLexerÚincludeÚbygroupsÚwords)ÚTextÚNameÚStringÚPunctuationÚKeywordÚOperatorÚNumber)ÚStylec                   @   s°   e Zd ZdZdZdgZdgZdZdZde	e
ejeƒfdefed	ƒed
ƒeejfed e	ejeƒfed e	ejeƒfeejfdefdefdefgeddd�e
fgdefgdœZdS )ÚNumbaIRLexerzL
    Pygments style lexer for Numba IR (for use with highlighting etc).
    ZNumba_IRZnumba_irz
*.numba_irz\$[a-zA-Z0-9._]+z([a-zA-Z_]+[a-zA-Z0-9]*)z(label)(\ [0-9]+)(:)$z = Ú
whitespaceÚkeywordz(\()z(\=)z[0-9]+z	<[^>\n]*>z[=<>{}\[\]()*.,!\':]|x\b)ÚdelZjumpÚcallÚbranchú )Úsuffixz(\n|\s))Úrootr   r   N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚnameÚaliasesÚ	filenamesÚ
identifierZ
fun_or_varr   r   r   ÚLabelr   r   r   ÚVariableÚFunctionÚ	AttributeÚConstantr   r   r	   r
   Útokens© r)   r)   úN/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numba/misc/dump_style.pyr      sB   ÿ
ÿ
ÿíþþÿÿãr   c                  C   sJ   t j ¡ } |  tjdi¡ tdtfd| iƒ}|tt	t
t dœ}|tjjj S )z\
    Get appropriate style for highlighting according to
    NUMBA_COLOR_SCHEME setting
    z#888888ZCustomDefaultStyleÚstyles)Úno_colorZdark_bgZlight_bgZblue_bgZ
jupyter_nb)r   r+   ÚcopyÚupdater   r$   Útyper   r   r   r   ÚnumbaÚcoreÚconfigZCOLOR_SCHEME)r+   Zcustom_defaultÚ	style_mapr)   r)   r*   Úby_colorschemeA   s    
 ÿûr4   )Zpygments.styles.defaultr   ÚImportErrorÚmsgZnumba.core.configr0   Zpygments.styles.mannir   Zpygments.styles.monokair   Zpygments.styles.nativer   Zpygments.lexerr   r   r   r	   Zpygments.tokenr
   r   r   r   r   r   r   Zpygments.styler   r   r4   r)   r)   r)   r*   Ú<module>   s   $.