64 lines
1.7 KiB
Python
64 lines
1.7 KiB
Python
"""
|
|
pygments.lexers.numbair
|
|
~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
Lexer for other Numba Intermediate Representation.
|
|
|
|
:copyright: Copyright 2006-2025 by the Pygments team, see AUTHORS.
|
|
:license: BSD, see LICENSE for details.
|
|
"""
|
|
|
|
from pygments.lexer import RegexLexer, include, bygroups, words
|
|
from pygments.token import Whitespace, Name, String, Punctuation, Keyword, \
|
|
Operator, Number
|
|
|
|
__all__ = ["NumbaIRLexer"]
|
|
|
|
class NumbaIRLexer(RegexLexer):
|
|
"""
|
|
Lexer for Numba IR
|
|
"""
|
|
name = 'Numba_IR'
|
|
url = "https://numba.readthedocs.io/en/stable/developer/architecture.html#stage-2-generate-the-numba-ir"
|
|
aliases = ['numba_ir', 'numbair']
|
|
filenames = ['*.numba_ir']
|
|
mimetypes = ['text/x-numba_ir', 'text/x-numbair']
|
|
version_added = '2.19'
|
|
|
|
identifier = r'\$[a-zA-Z0-9._]+'
|
|
fun_or_var = r'([a-zA-Z_]+[a-zA-Z0-9]*)'
|
|
|
|
tokens = {
|
|
'root' : [
|
|
(r'(label)(\ [0-9]+)(:)$',
|
|
bygroups(Keyword, Name.Label, Punctuation)),
|
|
|
|
(r'=', Operator),
|
|
include('whitespace'),
|
|
include('keyword'),
|
|
|
|
(identifier, Name.Variable),
|
|
(fun_or_var + r'(\()',
|
|
bygroups(Name.Function, Punctuation)),
|
|
(fun_or_var + r'(\=)',
|
|
bygroups(Name.Attribute, Punctuation)),
|
|
(fun_or_var, Name.Constant),
|
|
(r'[0-9]+', Number),
|
|
|
|
# <built-in function some>
|
|
(r'<[^>\n]*>', String),
|
|
|
|
(r'[=<>{}\[\]()*.,!\':]|x\b', Punctuation)
|
|
],
|
|
|
|
'keyword':[
|
|
(words((
|
|
'del', 'jump', 'call', 'branch',
|
|
), suffix=' '), Keyword),
|
|
],
|
|
|
|
'whitespace': [
|
|
(r'(\n|\s)+', Whitespace),
|
|
],
|
|
}
|