Pygment lexer多个令牌

时间:2012-09-20 00:45:13

标签: c++ lexer pygments

我正在使用Pygments的词法分析器,一个Python插件。我想为C ++代码获取令牌,特别是在声明新变量时,例如

int a=3,b=5,c=4;

这里a,b,c应该给出“声明变量”类型,它与

不同
a=3,b=5,c=4;

这里a,b,c应该简单地给出“Variables”类型,因为它们之前已经被声明过了。

我想使用词法分析器一次扫描多个标记的能力(See Pygments documentation)我想按照

的方式编写一个正则表达式
(int)(\s)(?:([a-z]+)(=)([0-9]+)(,))*, bygroups(Type,Space,Name,Equal,Number,Comma)

(“?:”只是告诉Pygments这个分组不应该在bygroups中使用。)

但是,它不是在行中匹配任意数量的声明,而只返回行中最后一个声明的标记(在这种情况下,“c = 4”部分。)如何让它返回标记为行中的所有声明?

1 个答案:

答案 0 :(得分:2)

你需要的是一个有状态的词法分析器。你的正则表达式不会的原因 工作是因为这些团体不是连续的。

int a=3,b=5,c=4;

在这里,您希望字符0..2为Type,3..3 Space,4..7 Name,Equal 数字和逗号再次名称,等于,数字和逗号。那不是 好。

解决方案是记住何时看到类型声明, 进入一个新的词法分析器模式,一直持续到下一个分号。看到 在pygments文档中Changing states

以下是使用CFamilyLexer并添加三个新词法分析器的解决方案 状态。所以当它在function中看到这样的一行时 状态:

int m = 3 * a + b, x = /* comments ; everywhere */ a * a;

首先消耗:

int

它符合我添加的新规则,因此它进入vardecl状态:

m

哦变量的名字!由于词法分析器处于vardecl状态, 这是一个新定义的变量。将其作为NameDecl令牌发出。然后 进入varvalue状态。

3

只是一个数字。

*

只是一名操作员。

a

哦变量的名字!但现在我们处于varvalue状态 是不是变量声明,只是一个常规变量引用。

+ b

运算符和另一个变量引用。

,

变量m的值已完全声明。返回vardecl州。

x =

新变量声明。

/* comments ; everywhere */

另一个状态被推到了堆栈上。在评论标记中会这样做 否则会忽略;之类的重要性。

a * a

x变量的值。

;

返回function州。特殊变量声明规则 完成了。

from pygments import highlight
from pygments.formatters import HtmlFormatter, TerminalFormatter
from pygments.formatters.terminal import TERMINAL_COLORS
from pygments.lexer import inherit
from pygments.lexers.compiled import CFamilyLexer
from pygments.token import *

# New token type for variable declarations. Red makes them stand out
# on the console.
NameDecl = Token.NameDecl
STANDARD_TYPES[NameDecl] = 'ndec'
TERMINAL_COLORS[NameDecl] = ('red', 'red')

class CDeclLexer(CFamilyLexer):
    tokens = {
        # Only touch variables declared inside functions.
        'function': [
            # The obvious fault that is hard to get around is that
            # user-defined types won't be cathed by this regexp.
            (r'(?<=\s)(bool|int|long|float|short|double|char|unsigned|signed|void|'
             r'[a-z_][a-z0-9_]*_t)\b',
             Keyword.Type, 'vardecl'),
            inherit
        ],
        'vardecl' : [
            (r'\s+', Text),
            # Comments
            (r'/(\\\n)?[*](.|\n)*?[*](\\\n)?/', Comment.Multiline),
            (r';', Punctuation, '#pop'),
            (r'[~!%^&*+=|?:<>/-]', Operator),
            # After the name of the variable has been tokenized enter
            # a new mode for the value.
            (r'[a-zA-Z_][a-zA-Z0-9_]*', NameDecl, 'varvalue'),
        ],
        'varvalue' : [
            (r'\s+', Text),
            (r',', Punctuation, '#pop'),
            (r';', Punctuation, '#pop:2'),
            # Comments
            (r'/(\\\n)?[*](.|\n)*?[*](\\\n)?/', Comment.Multiline),                
            (r'[~!%^&*+=|?:<>/-\[\]]', Operator),
            (r'\d+[LlUu]*', Number.Integer),            
            # Rules for strings and chars.
            (r'L?"', String, 'string'),
            (r"L?'(\\.|\\[0-7]{1,3}|\\x[a-fA-F0-9]{1,2}|[^\\\'\n])'", String.Char),
            (r'[a-zA-Z_][a-zA-Z0-9_]*', Name),
            # Getting arrays right is tricky.
            (r'{', Punctuation, 'arrvalue'),
        ],
        'arrvalue' : [
            (r'\s+', Text),
            (r'\d+[LlUu]*', Number.Integer),
            (r'}', Punctuation, '#pop'),
            (r'[~!%^&*+=|?:<>/-\[\]]', Operator),
            (r',', Punctuation),
            (r'[a-zA-Z_][a-zA-Z0-9_]*', Name),
            (r'{', Punctuation, '#push'),
        ]
    }

code = '''
#include <stdio.h>

void main(int argc, char *argv[]) 
{
    int vec_a, vec_b;
    int a = 3, /* Mo;yo */ b=5, c=7;
    int m = 3 * a + b, x = /* comments everywhere */ a * a;
    char *myst = "hi;there";
    char semi = ';';
    time_t now = /* Null; */ NULL;
    int arr[10] = {1, 2, 9 / c};
    int foo[][2] = {{1, 2}};

    a = b * 9;
    c = 77;
    d = (int) 99;
}
'''
for formatter in [TerminalFormatter, HtmlFormatter]:
    print highlight(code, CDeclLexer(), formatter())