62 lines
2.7 KiB
Lua
62 lines
2.7 KiB
Lua
local l = lpeg or require'lpeg'
|
|
|
|
local line = (1-l.P'\n')^0 * '\n'
|
|
|
|
local id = l.R'09'^1/tonumber
|
|
local non_final_list_block = (l.C((1-l.P'\n')^1) * '\n' - '### ' + '\n')^0
|
|
local math_lists_block = l.Ct('### ' * l.Cg(l.C'display' * ' ', 'display')^-1 * 'math mode entered at line ' * l.Cg(l.R'09'^1 / tonumber, 'line') * '\n'
|
|
* non_final_list_block)^1
|
|
local generic_list_block = '### ' * (line - 'current page:') * non_final_list_block
|
|
local luamml_block = l.Cg('LUAMML_FORMULA_BEGIN:' * id * ':' * l.Ct(
|
|
l.Cg(id, 'flag') * ':' * l.Cg((1-l.S':\n')^0, 'tag') * ':' * l.Cg((1-l.P'\n')^1, 'label')^-1 * l.P'\n'^1
|
|
|
|
* (math_lists_block + generic_list_block/0)^0
|
|
* (line - 'LUAMML_FORMULA_END\n')^0
|
|
* 'LUAMML_FORMULA_END\n') * l.Cc'groups')
|
|
|
|
local luamml_mark = l.Cg('LUAMML_MARK:' * id * ':' * l.Cs((1 - l.P'\n' + l.Cg('\n' * l.Cc'' - '\nLUAMML_MARK_END\n'))^0) * '\nLUAMML_MARK_END\n' * l.Cc'marks')
|
|
|
|
local function add(a, b) return a + b end
|
|
local count_block = '### ' * line * l.Cf(l.Cc(0) * (('\\' * l.Cc(1))^-1 * line - '### ')^0, add)
|
|
local luamml_precount = l.Cg('LUAMML_COUNT:' * id * l.P'\n'^1
|
|
* count_block * l.Cc'precount')
|
|
|
|
local luamml_postcount = l.Cg('LUAMML_COUNT_END:' * id * l.P'\n'^1
|
|
* count_block * l.Cc'postcount')
|
|
|
|
local luamml_instruction = l.Cg('LUAMML_INSTRUCTION:' * l.Cc(nil) * l.C((1 - l.P'\n')^0) * '\n' * l.Cc'instructions')
|
|
|
|
local function multi_table_set(t, key, value, table)
|
|
table = t[table]
|
|
table[key or #table + 1] = value
|
|
return t
|
|
end
|
|
local log_file = l.Cf(l.Ct(l.Cg(l.Ct'', 'groups')
|
|
* l.Cg(l.Ct'', 'precount')
|
|
* l.Cg(l.Ct'', 'postcount')
|
|
* l.Cg(l.Ct'', 'marks')
|
|
* l.Cg(l.Ct'', 'instructions'))
|
|
* (luamml_block + luamml_mark + luamml_instruction + luamml_precount + luamml_postcount + line)^0,
|
|
multi_table_set)
|
|
|
|
return function(filename)
|
|
local f
|
|
if filename and filename ~= '-' then
|
|
local msg f, msg = assert(io.open(filename, 'r'))
|
|
if not f then return f, msg end
|
|
end
|
|
local content = (f or io.stdin):read'a'
|
|
if f then f:close() end
|
|
-- The following does *not* end with * -1 since we want to allow the last line to not end with \n.
|
|
-- In that case we ignore the last line, but that's safe since the last line never contains our markers.
|
|
local parsed = assert(log_file:match(content))
|
|
local precount, postcount, count = parsed.precount, parsed.postcount, {}
|
|
for id, pre in next, precount do
|
|
local post = assert(postcount[id], 'Unbalanced count')
|
|
count[id], postcount[id] = post-pre, nil
|
|
end
|
|
assert(not next(postcount), 'Unbalanced count')
|
|
parsed.precount, parsed.postcount, parsed.count = nil, nil, count
|
|
return parsed
|
|
end
|