aboutsummaryrefslogtreecommitdiffhomepage
path: root/lexlua/latex.lua
blob: e28b30480b6df1e6ef965f3dc688bccf90567b32 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
-- Copyright 2006-2020 Mitchell mitchell.att.foicica.com. See License.txt.
-- Latex LPeg lexer.
-- Modified by Brian Schott.
-- Modified by Robert Gieseke.

local lexer = require('lexer')
local token, word_match = lexer.token, lexer.word_match
local P, S = lpeg.P, lpeg.S

local lex = lexer.new('latex')

-- Whitespace.
lex:add_rule('whitespace', token(lexer.WHITESPACE, lexer.space^1))

-- Comments.
local line_comment = lexer.to_eol('%')
local block_comment = lexer.range('\\begin' * P(' ')^0 * '{comment}',
  '\\end' * P(' ')^0 * '{comment}')
lex:add_rule('comment', token(lexer.COMMENT, line_comment + block_comment))

-- Math environments.
local math_word = word_match[[
  align displaymath eqnarray equation gather math multline
]]
local math_begin_end = (P('begin') + P('end')) * P(' ')^0 *
  '{' * math_word * P('*')^-1 * '}'
lex:add_rule('math', token('math', '$' + '\\' * (S('[]()') + math_begin_end)))
lex:add_style('math', lexer.styles['function'])

-- LaTeX environments.
lex:add_rule('environment', token('environment', '\\' *
  (P('begin') + P('end')) * P(' ')^0 * '{' * lexer.word * P('*')^-1 * '}'))
lex:add_style('environment', lexer.styles.keyword)

-- Sections.
lex:add_rule('section', token('section', '\\' * word_match[[
  part chapter section subsection subsubsection paragraph subparagraph
]] * P('*')^-1))
lex:add_style('section', lexer.styles.class)

-- Commands.
lex:add_rule('command', token('command', '\\' *
  (lexer.alpha^1 + S('#$&~_^%{}'))))
lex:add_style('command', lexer.styles.keyword)

-- Operators.
lex:add_rule('operator', token(lexer.OPERATOR, S('&#{}[]')))

-- Fold points.
lex:add_fold_point(lexer.COMMENT, '\\begin', '\\end')
lex:add_fold_point(lexer.COMMENT, lexer.fold_consecutive_lines('%'))
lex:add_fold_point('environment', '\\begin', '\\end')
lex:add_fold_point(lexer.OPERATOR, '{', '}')

return lex