我正在使用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”部分。)如何让它返回标记为行中的所有声明?
答案 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())