226 lines
10 KiB
Python
226 lines
10 KiB
Python
import color, mode2, tab2
|
|
from lex2 import Grammar, PatternRule, RegionRule
|
|
from mode_python import StringGrammar
|
|
|
|
# this might not be complete...
|
|
# see http://gcc.gnu.org/onlinedocs/gcc-2.95.3/cpp_3.html#SEC44
|
|
class MacroGrammar(Grammar):
|
|
rules = [
|
|
PatternRule('name', r'(?:(?<=#define )) *[a-zA-Z_][a-zA-Z0-9_]*'),
|
|
|
|
PatternRule(r"unop", r"\+=|-=|\*=|/=|//=|%=|&=\|\^=|>>=|<<=|\*\*="),
|
|
PatternRule(r'binop', r"\+|<>|<<|<=|<|-|>>|>=|>|\*\*|&|\*|\||/|\^|==|//|~|!=|%"),
|
|
PatternRule(r"delimiter", r"->|\.|\(|\)|\[|\]|{|}|@|,|:|`|;|=|\?"),
|
|
|
|
PatternRule(r"integer", r"-?(?:0(?![x0-9])|[1-9][0-9]*|0[0-7]+|0[xX][0-9a-fA-F]+)[lL]?"),
|
|
PatternRule(r"float", r"-?(?:[0-9]+\.[0-9]*|\.[0-9]+|(?:[0-9]|[0-9]+\.[0-9]*|\.[0-9]+)[eE][\+-]?[0-9]+)"),
|
|
RegionRule(r'string', '"', StringGrammar, '"'),
|
|
PatternRule(r'char', r"'.'|'\\.'|'\\[0-7]{3}'"),
|
|
|
|
PatternRule(r"continued", r"\\\n$"),
|
|
]
|
|
|
|
class CGrammar(Grammar):
|
|
rules = [
|
|
PatternRule(r'include', r'#include(?!=[a-zA-Z0-9_])'),
|
|
PatternRule(r'header', r'<[-A-Za-z/0-9_\.]+>|"[-A-Za-z/0-9_\.]+"'),
|
|
RegionRule(r'macrocomment', r'#if +(?:0|NULL|FALSE)', Grammar, r'#endif'),
|
|
RegionRule(r'macro', r'#(?:assert|cpu|define|elif|else|endif|error|ident|ifdef|ifndef|if|import|include_next|line|machine|pragma|pragma_once|system|unassert|undef|warning)(?!=[a-zA-Z0-9_])', MacroGrammar, r'\n$'),
|
|
|
|
RegionRule(r'comment', '/\*', Grammar, '\*/'),
|
|
PatternRule(r'comment', r'//.*$'),
|
|
|
|
PatternRule(r'keyword', r"(?:auto|break|case|char|const|continue|default|double|do|else|enum|extern|float|for|goto|if|int|long|register|return|short|signed|sizeof|static|struct|switch|typedef|union|unsigned|void|volatile|while)(?![a-zA-z_])"),
|
|
PatternRule(r'label', r'[a-zA-Z_][a-zA-Z0-9_]*(?=:)'),
|
|
PatternRule(r'structname', r'(?<=struct ) *[a-zA-Z_][a-zA-Z0-9_]*'),
|
|
PatternRule(r'enumname', r'(?<=enum ) *[a-zA-Z_][a-zA-Z0-9_]*'),
|
|
PatternRule(r'function', r'[a-zA-Z_][a-zA-Z0-9_]*(?= *\()'),
|
|
|
|
PatternRule(r'builtin', r"(?:NULL|TRUE|FALSE)"),
|
|
PatternRule(r'identifier', r"[a-zA-Z_][a-zA-Z0-9_]*"),
|
|
PatternRule(r"unop", r"\+=|-=|\*=|/=|//=|%=|&=\|\^=|>>=|<<=|\*\*="),
|
|
PatternRule(r'binop', r"\+|<>|<<|<=|<|-|>>|>=|>|\*\*|&|\*|\||/|\^|==|//|~|!=|%"),
|
|
|
|
# this is sketchy as hell
|
|
PatternRule(r"delimiter", r"->|\.|\(|\)|\[|\]|{|}|@|,|:|`|;|=|\?"),
|
|
|
|
PatternRule(r"integer", r"(?:0(?![x0-9])|[1-9][0-9]*|0[0-7]+|0[xX][0-9a-fA-F]+)[lL]?"),
|
|
PatternRule(r"float", r"[0-9]+\.[0-9]*|\.[0-9]+|(?:[0-9]|[0-9]+\.[0-9]*|\.[0-9]+)[eE][\+-]?[0-9]+"),
|
|
RegionRule(r'string', '"', StringGrammar, '"'),
|
|
PatternRule(r'char', r"'.'|'\\.'|'\\[0-7]{3}'"),
|
|
PatternRule(r"eol", r"\n$"),
|
|
]
|
|
|
|
class CTabber(tab2.StackTabber):
|
|
def is_base(self, y):
|
|
if y == 0:
|
|
return True
|
|
highlighter = self.mode.window.buffer.highlights[self.mode.name()]
|
|
if not highlighter.tokens[y]:
|
|
return False
|
|
|
|
# this assumes that people aren't gonna use these macros inside of
|
|
# blocks, which is probably ok.
|
|
t0 = highlighter.tokens[y][0]
|
|
if t0.name == 'macro.start' and t0.string in ('#define', '#include'):
|
|
return True
|
|
|
|
# detecting function declarations is annoying; this assumes that people
|
|
# won't put a variable type and name on different lines, but that they
|
|
# might do that for function return type and name.
|
|
#
|
|
# unfortunately, valid function return types might include any of the
|
|
# four types of tokens below
|
|
decl = False
|
|
for t in highlighter.tokens[y]:
|
|
if t.name in ('keyword', 'identifier', 'structname', 'enumname'):
|
|
decl = True
|
|
continue
|
|
if decl and t.name == 'function':
|
|
break
|
|
else:
|
|
decl = False
|
|
break
|
|
if decl:
|
|
return True
|
|
|
|
return False
|
|
|
|
def _handle_open_token(self, currlvl, y, i):
|
|
self._opt_pop('cont')
|
|
token = self.get_token(y, i)
|
|
if token.string == '{':
|
|
self._opt_pop('cond')
|
|
currlvl = tab2.StackTabber._handle_open_token(self, currlvl, y, i)
|
|
return currlvl
|
|
def _handle_close_token(self, currlvl, y, i):
|
|
self._opt_pop('cont')
|
|
currlvl = tab2.StackTabber._handle_close_token(self, currlvl, y, i)
|
|
token = self.get_token(y, i)
|
|
if self.is_rightmost_token(y, i):
|
|
if token.string == '}':
|
|
self._opt_pop('cond')
|
|
self._opt_pop('cont')
|
|
elif self._peek_name() == 'cond':
|
|
pass
|
|
else:
|
|
self._opt_append('cont', currlvl + 4)
|
|
return currlvl
|
|
def _handle_other_token(self, currlvl, y, i):
|
|
token = self.get_token(y, i)
|
|
fqname = token.fqname()
|
|
if fqname == 'delimiter' and token.string == ';':
|
|
self._opt_pop('cond')
|
|
self._opt_pop('cont')
|
|
self._opt_pop('cond')
|
|
#currlvl = self.get_curr_level()
|
|
|
|
elif fqname == 'keyword':
|
|
if token.string in ('do', 'else', 'for', 'if', 'while'):
|
|
self._append('cond', currlvl + 4)
|
|
elif token.string == 'break':
|
|
self._opt_pop('case')
|
|
elif token.string == 'case':
|
|
self._opt_append('case', currlvl + 4)
|
|
|
|
elif fqname == 'string.start':
|
|
self._opt_append('string', None)
|
|
elif fqname == 'string.end':
|
|
self._opt_pop('string')
|
|
if self.is_rightmost_token(y, i):
|
|
self._opt_append('cont', currlvl + 4)
|
|
|
|
# TODO: this could be a lot better
|
|
elif fqname == 'macro':
|
|
currlvl = 0
|
|
elif fqname.startswith('macro.start'):
|
|
self._opt_append('macro', None)
|
|
currlvl = 0
|
|
elif fqname.startswith('macro.end'):
|
|
self._opt_pop('macro', None)
|
|
|
|
elif fqname.startswith('macroblock.start'):
|
|
self._opt_append('macroblock', None)
|
|
currlvl = 0
|
|
elif fqname.startswith('macroblock.end'):
|
|
self._opt_pop('macroblock', None)
|
|
|
|
if self.is_rightmost_token(y, i):
|
|
if self._has_markers() and self._peek_name() == 'cond':
|
|
pass
|
|
elif(not fqname.startswith('string') and
|
|
not fqname.startswith('comment') and
|
|
not fqname.startswith('macro') and
|
|
not fqname == 'delimiter' and
|
|
not fqname == 'header' and
|
|
not fqname == 'null' and
|
|
not fqname == 'eol' and
|
|
token.string not in ('}', ';', '(', '{', '[', ',')):
|
|
self._opt_append('cont', currlvl + 4)
|
|
return currlvl
|
|
|
|
class C(mode2.Fundamental):
|
|
tabbercls = CTabber
|
|
grammar = CGrammar
|
|
opentoken = 'delimiter'
|
|
opentags = {'(': ')', '[': ']', '{': '}'}
|
|
closetoken = 'delimiter'
|
|
closetags = {')': '(', ']': '[', '}': '{'}
|
|
def __init__(self, w):
|
|
mode2.Fundamental.__init__(self, w)
|
|
self.add_bindings('close-paren', (')',))
|
|
self.add_bindings('close-brace', ('}',))
|
|
self.add_bindings('close-bracket', (']',))
|
|
self.colors = {
|
|
'macrocomment.start': color.build('red', 'default'),
|
|
'macrocomment.null': color.build('red', 'default'),
|
|
'macrocomment.end': color.build('red', 'default'),
|
|
'comment': color.build('red', 'default'),
|
|
'comment.start': color.build('red', 'default'),
|
|
'comment.end': color.build('red', 'default'),
|
|
'comment.null': color.build('red', 'default'),
|
|
|
|
'include': color.build('blue', 'default'),
|
|
'header': color.build('green', 'default'),
|
|
|
|
'macro': color.build('blue', 'default'),
|
|
'macro.start': color.build('blue', 'default'),
|
|
'macro.name': color.build('yellow', 'default'),
|
|
'macro.null': color.build('magenta', 'default'),
|
|
#'macro.null': color.build('default', 'default'),
|
|
'macro.continued': color.build('red', 'default'),
|
|
'macro.delimiter': color.build('default', 'default'),
|
|
'macro.integer': color.build('green', 'default'),
|
|
'macro.float': color.build('green', 'default'),
|
|
'macro.char': color.build('green', 'default'),
|
|
'macro.string.start': color.build('green', 'default'),
|
|
'macro.string.escaped': color.build('magenta', 'default'),
|
|
'macro.string.octal': color.build('magenta', 'default'),
|
|
#'macro.string.escaped': color.build('default', 'default'),
|
|
#'macro.string.octal': color.build('default', 'default'),
|
|
'macro.string.null': color.build('green', 'default'),
|
|
'macro.string.end': color.build('green', 'default'),
|
|
'macro.end': color.build('magenta', 'default'),
|
|
#'macro.end': color.build('default', 'default'),
|
|
|
|
'label': color.build('magenta', 'default'),
|
|
'keyword': color.build('cyan', 'default'),
|
|
'function': color.build('blue', 'default'),
|
|
'builtin': color.build('magenta', 'default'),
|
|
'structname': color.build('yellow', 'default'),
|
|
'enumname': color.build('yellow', 'default'),
|
|
|
|
'char': color.build('green', 'default'),
|
|
'string.start': color.build('green', 'default'),
|
|
'string.octal': color.build('green', 'default'),
|
|
'string.escaped': color.build('green', 'default'),
|
|
'string.null': color.build('green', 'default'),
|
|
'string.end': color.build('green', 'default'),
|
|
'integer': color.build('green', 'default'),
|
|
'float': color.build('green', 'default'),
|
|
|
|
'bizzaro': color.build('magenta', 'green'),
|
|
}
|
|
def name(self):
|
|
return "C"
|