6916312961

Committed 18 Nov 2023 09:12PM UTC coverage: 62.199% (-6.6%) from 68.751%

Build # 6916312961

Build Type

push

github

Committed by

web-flow

Commit Message

Merge a29a0997b into f64e235fa

Run Details

9647 of 15510 relevant lines covered (62.2%)

6520.8 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

78.91

/packages/math/texlike.lua

local syms = require("packages.math.unicode-symbols")
local bits = require("core.parserbits")

local epnf = require("epnf")
local lpeg = require("lpeg")

local atomType = syms.atomType
local symbolDefaults = syms.symbolDefaults
local symbols = syms.symbols

-- Grammar to parse TeX-like math
-- luacheck: push ignore
---@diagnostic disable: undefined-global, unused-local, lowercase-global
local mathGrammar = function (_ENV)
  local _ = WS^0
  local eol = S"\r\n"
  local digit = R("09")
  local natural = digit^1 / tostring
  local pos_natural = R("19") * digit^0 / tonumber
  local ctrl_word = R("AZ", "az")^1
  local ctrl_symbol = P(1) - S"{}\\"
  local ctrl_sequence_name = C(ctrl_word + ctrl_symbol) / 1
  local comment = (
      P"%" *
      P(1-eol)^0 *
      eol^-1
    )
  local utf8cont = R("\128\191")
  local utf8code = lpeg.R("\0\127")
    + lpeg.R("\194\223") * utf8cont
    + lpeg.R("\224\239") * utf8cont * utf8cont
    + lpeg.R("\240\244") * utf8cont * utf8cont * utf8cont
  -- Identifiers inside \mo and \mi tags
  local sileID = C(bits.identifier + P(1)) / 1
  local mathMLID = (utf8code - S"\\{}%")^1 / function (...)
      local ret = ""
      local t = {...}
      for _,b in ipairs(t) do
        ret = ret .. b
      end
      return ret
    end
  local group = P"{" * V"mathlist" * (P"}" + E("`}` expected"))
  local element_no_infix =
    V"def" +
    V"command" +
    group +
    V"argument" +
    V"atom"
  local element =
    V"supsub" +
    V"subsup" +
    V"sup" +
    V"sub" +
    element_no_infix
  local sep = S",;" * _
  local quotedString = (P'"' * C((1-P'"')^1) * P'"')
  local value = ( quotedString + (1-S",;]")^1 )
  local pair = Cg(sileID * _ * "=" * _ * C(value)) * sep^-1 / function (...)
      local t = {...}; return t[1], t[#t]
    end
  local list = Cf(Ct"" * pair^0, rawset)
  local parameters = (
      P"[" *
      list *
      P"]"
    )^-1 / function (a)
        return type(a)=="table" and a or {}
      end

  local dim2_arg_inner = Ct(V"mathlist" * (P"&" * V"mathlist")^0) /
    function (t)
      t.id = "mathlist"
      return t
    end
  local dim2_arg =
    Cg(P"{" *
       dim2_arg_inner *
       (P"\\\\" * dim2_arg_inner)^1 *
       (P"}" + E("`}` expected"))
      ) / function (...)
        local t = {...}
        -- Remove the last mathlist if empty. This way,
        -- `inner1 \\ inner2 \\` is the same as `inner1 \\ inner2`.
        if not t[#t][1] or not t[#t][1][1] then table.remove(t) end
        return table.unpack(t)
      end

  START "texlike_math"
  texlike_math = V"mathlist" * EOF"Unexpected character at end of math code"
  mathlist = (comment + (WS * _) + element)^0
  supsub = element_no_infix * _ * P"^" * _ * element_no_infix * _ *
    P"_" * _ * element_no_infix
  subsup = element_no_infix * _ * P"_" * _ * element_no_infix * _ *
    P"^" * _ * element_no_infix
  sup = element_no_infix * _ * P"^" * _ * element_no_infix
  sub = element_no_infix * _ * P"_" * _ * element_no_infix
  atom = natural + C(utf8code - S"\\{}%^_&") +
    (P"\\{" + P"\\}") / function (s) return string.sub(s, -1) end
  command = (
      P"\\" *
      Cg(ctrl_sequence_name, "command") *
      Cg(parameters, "options") *
      (dim2_arg + group^0)
    )
  def = P"\\def" * _ * P"{" *
    Cg(ctrl_sequence_name, "command-name") * P"}" * _ *
    --P"[" * Cg(digit^1, "arity") * P"]" * _ *
    P"{" * V"mathlist" * P"}"
  argument = P"#" * Cg(pos_natural, "index")
end
-- luacheck: pop
---@diagnostic enable: undefined-global, unused-local, lowercase-global

local mathParser = epnf.define(mathGrammar)

local commands = {}

-- A command type is a type for each argument it takes: either string or MathML
-- tree. If a command has no type, it is assumed to take only trees.
-- Tags like <mi>, <mo>, <mn> take a string, and this needs to be propagated in
-- commands that use them.

local objType = {
  tree = 1,
  str = 2
}

local function inferArgTypes_aux (accumulator, typeRequired, body)
  if type(body) == "table" then
    if body.id == "argument" then
      local ret = accumulator
      table.insert(ret, body.index, typeRequired)
      return ret
    elseif body.id == "command" then
      if commands[body.command] then
        local cmdArgTypes = commands[body.command][1]
        if #cmdArgTypes ~= #body then
          SU.error("Wrong number of arguments (" .. #body ..
            ") for command " .. body.command .. " (should be " ..
            #cmdArgTypes .. ")")
        else
          for i = 1, #cmdArgTypes do
            accumulator = inferArgTypes_aux(accumulator, cmdArgTypes[i], body[i])
          end
        end
        return accumulator
      elseif body.command == "mi" or body.command == "mo" or body.command == "mn" then
        if #body ~= 1 then
          SU.error("Wrong number of arguments ("..#body..") for command "..
            body.command.." (should be 1)")
        end
        accumulator = inferArgTypes_aux(accumulator, objType.str, body[1])
        return accumulator
      else
        -- Not a macro, recurse on children assuming tree type for all
        -- arguments
        for _, child in ipairs(body) do
          accumulator = inferArgTypes_aux(accumulator, objType.tree, child)
        end
        return accumulator
      end
    elseif body.id == "atom" then
      return accumulator
    else
      -- Simply recurse on children
      for _, child in ipairs(body) do
        accumulator = inferArgTypes_aux(accumulator, typeRequired, child)
      end
      return accumulator
    end
  else SU.error("invalid argument to inferArgTypes_aux") end
end

local inferArgTypes = function (body)
  return inferArgTypes_aux({}, objType.tree, body)
end

local function registerCommand (name, argTypes, func)
  commands[name] = { argTypes, func }
end

-- Computes func(func(... func(init, k1, v1), k2, v2)..., k_n, v_n), i.e. applies
-- func on every key-value pair in the table. Keys with numeric indices are
-- processed in order. This is an important property for MathML compilation below.
local function fold_pairs (func, table)
  local accumulator = {}
  for k, v in pl.utils.kpairs(table) do
    accumulator = func(v, k, accumulator)
  end
  for i, v in ipairs(table) do
    accumulator = func(v, i, accumulator)
  end
  return accumulator
end

local function forall (pred, list)
  for _,x in ipairs(list) do
    if not pred(x) then return false end
  end
  return true
end

local compileToStr = function (argEnv, mathlist)
  if #mathlist == 1 and mathlist.id == "atom" then
    -- List is a single atom
    return mathlist[1]
  elseif #mathlist == 1 and mathlist[1].id == "argument" then
    return argEnv[mathlist[1].index]
  elseif mathlist.id == "argument" then
    return argEnv[mathlist.index]
  else
    local ret = ""
    for _,elt in ipairs(mathlist) do
      if elt.id == "atom" then
        ret = ret .. elt[1]
      elseif elt.id == "command" and symbols[elt.command] then
        ret = ret .. symbols[elt.command]
      else
        SU.error("Encountered non-character token in command that takes a string")
      end
    end
    return ret
  end
end

local function compileToMathML_aux (_, arg_env, tree)
  if type(tree) == "string" then return tree end
  local function compile_and_insert (child, key, accumulator)
    if type(key) ~= "number" then
      accumulator[key] = child
      return accumulator
    -- Compile all children, except if this node is a macro definition (no
    -- evaluation "under lambda") or the application of a registered macro
    -- (since evaluating the nodes depends on the macro's signature, it is more
    -- complex and done below)..
    elseif tree.id == "def" or (tree.id == "command" and commands[tree.command]) then
      -- Conserve unevaluated child
      table.insert(accumulator, child)
    else
      -- Compile next child
      local comp = compileToMathML_aux(nil, arg_env, child)
      if comp then
        if comp.id == "wrapper" then
          -- Insert all children of the wrapper node
          for _, inner_child in ipairs(comp) do
            table.insert(accumulator, inner_child)
          end
        else
          table.insert(accumulator, comp)
        end
      end
    end
    return accumulator
  end
  tree = fold_pairs(compile_and_insert, tree)
  if tree.id == "texlike_math" then
    tree.command = "math"
    -- If the outermost `mrow` contains only other `mrow`s, remove it
    -- (allowing vertical stacking).
    if forall(function (c) return c.command == "mrow" end, tree[1]) then
      tree[1].command = "math"
      return tree[1]
    end
  elseif tree.id == "mathlist" then
    -- Turn mathlist into `mrow` except if it has exactly one `mtr` or `mtd`
    -- child.
    -- Note that `def`s have already been compiled away at this point.
    if #tree == 1 and (tree[1].command == "mtr" or tree[1].command == "mtd") then
      return tree[1]
    else tree.command = "mrow" end
    tree.command = "mrow"
  elseif tree.id == "atom" then
    local codepoints = {}
    for _, cp in luautf8.codes(tree[1]) do
      table.insert(codepoints, cp)
    end
    local cp = codepoints[1]
    if #codepoints == 1 and ( -- If length of UTF-8 string is 1
       cp >= SU.codepoint("A") and cp <= SU.codepoint("Z") or
       cp >= SU.codepoint("a") and cp <= SU.codepoint("z") or
       cp >= SU.codepoint("Α") and cp <= SU.codepoint("Ω") or
       cp >= SU.codepoint("α") and cp <= SU.codepoint("ω")
    ) then
        tree.command = "mi"
    elseif lpeg.match(lpeg.R("09")^1, tree[1]) then
      tree.command = "mn"
    else
      tree.command = "mo"
    end
    tree.options = {}
  -- Translate TeX-like sub/superscripts to `munderover` or `msubsup`,
  -- depending on whether the base is a big operator
  elseif tree.id == "sup" and tree[1].command == "mo"
      and tree[1].atom == atomType.bigOperator then
    tree.command = "mover"
  elseif tree.id == "sub" and tree[1].command == "mo"
      and symbolDefaults[tree[1][1]].atom == atomType.bigOperator then
      tree.command = "munder"
  elseif tree.id == "subsup" and tree[1].command == "mo"
      and symbolDefaults[tree[1][1]].atom == atomType.bigOperator then
    tree.command = "munderover"
  elseif tree.id == "supsub" and tree[1].command == "mo"
      and symbolDefaults[tree[1][1]].atom == atomType.bigOperator then
    tree.command = "munderover"
    local tmp = tree[2]
    tree[2] = tree[3]
    tree[3] = tmp
  elseif tree.id == "sup" then
    tree.command = "msup"
  elseif tree.id == "sub" then
      tree.command = "msub"
  elseif tree.id == "subsup" then
    tree.command = "msubsup"
  elseif tree.id == "supsub" then
    tree.command = "msubsup"
    local tmp = tree[2]
    tree[2] = tree[3]
    tree[3] = tmp
  elseif tree.id == "def" then
    local commandName = tree["command-name"]
    local argTypes = inferArgTypes(tree[1])
    registerCommand(commandName, argTypes, function (compiledArgs)
      return compileToMathML_aux(nil, compiledArgs, tree[1])
    end)
    return nil
  elseif tree.id == "command" and commands[tree.command] then
    local argTypes = commands[tree.command][1]
    local cmdFun = commands[tree.command][2]
    local applicationTree = tree
    local cmdName = tree.command
    if #applicationTree ~= #argTypes then
      SU.error("Wrong number of arguments (" .. #applicationTree ..
        ") for command " .. cmdName .. " (should be " ..
        #argTypes .. ")")
    end
    -- Compile every argument
    local compiledArgs = {}
    for i,arg in pairs(applicationTree) do
      if type(i) == "number" then
        if argTypes[i] == objType.tree then
          table.insert(compiledArgs, compileToMathML_aux(nil, arg_env, arg))
        else
          local x = compileToStr(arg_env, arg)
          table.insert(compiledArgs, x)
        end
      else
        -- Not an argument but an attribute. Add it to the compiled
        -- argument tree as-is
        compiledArgs[i] = applicationTree[i]
      end
    end
    local res = cmdFun(compiledArgs)
    if res.command == "mrow" then
      -- Mark the outer mrow to be unwrapped in the parent
      res.id = "wrapper"
    end
    return res
  elseif tree.id == "command" and symbols[tree.command] then
    local atom = {id = "atom", [1] = symbols[tree.command]}
    tree = compileToMathML_aux(nil, arg_env, atom)
  elseif tree.id == "argument" then
    if arg_env[tree.index] then
      return arg_env[tree.index]
    else
      SU.error("Argument #"..tree.index.." has escaped its scope (probably not fully applied command).")
    end
  end
  tree.id = nil
  return tree
end

local function printMathML (tree)
  if type(tree) == "string" then
    return tree
  end
  local result = "\\" .. tree.command
  if tree.options then
    local options = {}
    for k,v in pairs(tree.options) do
      table.insert(options, k .. "=" .. v)
    end
    if #options > 0 then
      result = result .. "[" .. table.concat(options, ", ") .. "]"
    end
  end
  if #tree > 0 then
    result = result .. "{"
    for _, child in ipairs(tree) do
      result = result .. printMathML(child)
    end
    result = result .. "}"
  end
  return result
end

local function compileToMathML (_, arg_env, tree)
  local result = compileToMathML_aux(_, arg_env, tree)
  SU.debug("texmath", function ()
    return "Resulting MathML: " .. printMathML(result)
  end)
  return result
end

local function convertTexlike (_, content)
  local ret = epnf.parsestring(mathParser, content[1])
  SU.debug("texmath", function ()
    return "Parsed TeX math: " .. pl.pretty.write(ret)
  end)
  return ret
end

registerCommand("%", {}, function ()
  return { "%", command = "mo", options = {} }
end)
registerCommand("mi", { [1] = objType.str }, function (x) return x end)
registerCommand("mo", { [1] = objType.str }, function (x) return x end)
registerCommand("mn", { [1] = objType.str }, function (x) return x end)

compileToMathML(nil, {}, convertTexlike(nil, {[==[
  \def{frac}{\mfrac{#1}{#2}}
  \def{bi}{\mi[mathvariant=bold-italic]{#1}}
  \def{dsi}{\mi[mathvariant=double-struck]{#1}}

  % Standard spaces gleaned from plain TeX
  \def{thinspace}{\mspace[width=thin]}
  \def{negthinspace}{\mspace[width=-thin]}
  \def{,}{\thinspace}
  \def{!}{\negthinspace}
  \def{medspace}{\mspace[width=med]}
  \def{negmedspace}{\mspace[width=-med]}
  \def{>}{\medspace}
  \def{thickspace}{\mspace[width=thick]}
  \def{negthickspace}{\mspace[width=-thick]}
  \def{;}{\thickspace}
  \def{enspace}{\mspace[width=1en]}
  \def{enskip}{\enspace}
  \def{quad}{\mspace[width=1em]}
  \def{qquad}{\mspace[width=2em]}

  % Modulus operator forms
  \def{bmod}{\mo{mod}}
  \def{pmod}{\quad(\mo{mod} #1)}
]==]}))

return { convertTexlike, compileToMathML }

1	local syms = require("packages.math.unicode-symbols")	9✔
2	local bits = require("core.parserbits")	9✔
3
4	local epnf = require("epnf")	9✔
5	local lpeg = require("lpeg")	9✔
6
7	local atomType = syms.atomType	9✔
8	local symbolDefaults = syms.symbolDefaults	9✔
9	local symbols = syms.symbols	9✔
10
11	-- Grammar to parse TeX-like math
12	-- luacheck: push ignore
13	---@diagnostic disable: undefined-global, unused-local, lowercase-global
14	local mathGrammar = function (_ENV)
15	local _ = WS^0	9✔
16	local eol = S"\r\n"	9✔
17	local digit = R("09")	9✔
18	local natural = digit^1 / tostring	9✔
19	local pos_natural = R("19") * digit^0 / tonumber	9✔
20	local ctrl_word = R("AZ", "az")^1	9✔
21	local ctrl_symbol = P(1) - S"{}\\"	9✔
22	local ctrl_sequence_name = C(ctrl_word + ctrl_symbol) / 1	9✔
23	local comment = (
24	P"%" *	9✔
25	P(1-eol)^0 *	9✔
26	eol^-1	9✔
27	)
28	local utf8cont = R("\128\191")	9✔
29	local utf8code = lpeg.R("\0\127")	9✔
30	+ lpeg.R("\194\223") * utf8cont	9✔
31	+ lpeg.R("\224\239") * utf8cont * utf8cont	9✔
32	+ lpeg.R("\240\244") * utf8cont * utf8cont * utf8cont	9✔
33	-- Identifiers inside \mo and \mi tags
34	local sileID = C(bits.identifier + P(1)) / 1	9✔
35	local mathMLID = (utf8code - S"\\{}%")^1 / function (...)	9✔
36	local ret = ""	×
37	local t = {...}	×
38	for _,b in ipairs(t) do	×
39	ret = ret .. b	×
40	end
41	return ret	×
42	end
43	local group = P"{" * V"mathlist" * (P"}" + E("`}` expected"))	18✔
44	local element_no_infix =
45	V"def" +	9✔
46	V"command" +	9✔
47	group +	9✔
48	V"argument" +	9✔
49	V"atom"	9✔
50	local element =
51	V"supsub" +	9✔
52	V"subsup" +	9✔
53	V"sup" +	9✔
54	V"sub" +	9✔
55	element_no_infix	9✔
56	local sep = S",;" * _	9✔
57	local quotedString = (P'"' * C((1-P'"')^1) * P'"')	9✔
58	local value = ( quotedString + (1-S",;]")^1 )	9✔
59	local pair = Cg(sileID * _ * "=" * _ * C(value)) * sep^-1 / function (...)	9✔
60	local t = {...}; return t[1], t[#t]	125✔
61	end
62	local list = Cf(Ct"" * pair^0, rawset)	9✔
63	local parameters = (
64	P"[" *	9✔
65	list *	9✔
66	P"]"	9✔
67	)^-1 / function (a)	9✔
68	return type(a)=="table" and a or {}	404✔
69	end
70
71	local dim2_arg_inner = Ct(V"mathlist" * (P"&" * V"mathlist")^0) /	9✔
72	function (t)
73	t.id = "mathlist"	16✔
74	return t	16✔
75	end
76	local dim2_arg =
77	Cg(P"{" *	18✔
78	dim2_arg_inner *	9✔
79	(P"\\\\" * dim2_arg_inner)^1 *	9✔
80	(P"}" + E("`}` expected"))	18✔
81	) / function (...)	×
82	local t = {...}	4✔
83	-- Remove the last mathlist if empty. This way,
84	-- `inner1 \\ inner2 \\` is the same as `inner1 \\ inner2`.
85	if not t[#t][1] or not t[#t][1][1] then table.remove(t) end	4✔
86	return table.unpack(t)	4✔
87	end
88
89	START "texlike_math"	9✔
90	texlike_math = V"mathlist" * EOF"Unexpected character at end of math code"	18✔
91	mathlist = (comment + (WS * _) + element)^0	9✔
92	supsub = element_no_infix * _ * P"^" * _ * element_no_infix * _ *	9✔
93	P"_" * _ * element_no_infix	9✔
94	subsup = element_no_infix * _ * P"_" * _ * element_no_infix * _ *	9✔
95	P"^" * _ * element_no_infix	9✔
96	sup = element_no_infix * _ * P"^" * _ * element_no_infix	9✔
97	sub = element_no_infix * _ * P"_" * _ * element_no_infix	9✔
98	atom = natural + C(utf8code - S"\\{}%^_&") +	9✔
99	(P"\\{" + P"\\}") / function (s) return string.sub(s, -1) end	9✔
100	command = (	×
101	P"\\" *	9✔
102	Cg(ctrl_sequence_name, "command") *	9✔
103	Cg(parameters, "options") *	9✔
104	(dim2_arg + group^0)	9✔
105	)	9✔
106	def = P"\\def" * _ * P"{" *	9✔
107	Cg(ctrl_sequence_name, "command-name") * P"}" * _ *	9✔
108	--P"[" * Cg(digit^1, "arity") * P"]" * _ *
109	P"{" * V"mathlist" * P"}"	9✔
110	argument = P"#" * Cg(pos_natural, "index")	9✔
111	end
112	-- luacheck: pop
113	---@diagnostic enable: undefined-global, unused-local, lowercase-global
114
115	local mathParser = epnf.define(mathGrammar)	9✔
116
117	local commands = {}	9✔
118
119	-- A command type is a type for each argument it takes: either string or MathML
120	-- tree. If a command has no type, it is assumed to take only trees.
121	-- Tags like <mi>, <mo>, <mn> take a string, and this needs to be propagated in
122	-- commands that use them.
123
124	local objType = {	9✔
125	tree = 1,
126	str = 2	×
127	}
128
129	local function inferArgTypes_aux (accumulator, typeRequired, body)
130	if type(body) == "table" then	535✔
131	if body.id == "argument" then	535✔
132	local ret = accumulator	45✔
133	table.insert(ret, body.index, typeRequired)	45✔
134	return ret	45✔
135	elseif body.id == "command" then	490✔
136	if commands[body.command] then	183✔
137	local cmdArgTypes = commands[body.command][1]	93✔
138	if #cmdArgTypes ~= #body then	93✔
139	SU.error("Wrong number of arguments (" .. #body ..	×
140	") for command " .. body.command .. " (should be " ..	×
141	#cmdArgTypes .. ")")	×
142	else
143	for i = 1, #cmdArgTypes do	132✔
144	accumulator = inferArgTypes_aux(accumulator, cmdArgTypes[i], body[i])	78✔
145	end
146	end
147	return accumulator	93✔
148	elseif body.command == "mi" or body.command == "mo" or body.command == "mn" then	90✔
149	if #body ~= 1 then	×
150	SU.error("Wrong number of arguments ("..#body..") for command "..	×
151	body.command.." (should be 1)")	×
152	end
153	accumulator = inferArgTypes_aux(accumulator, objType.str, body[1])	×
154	return accumulator	×
155	else
156	-- Not a macro, recurse on children assuming tree type for all
157	-- arguments
158	for _, child in ipairs(body) do	108✔
159	accumulator = inferArgTypes_aux(accumulator, objType.tree, child)	36✔
160	end
161	return accumulator	90✔
162	end
163	elseif body.id == "atom" then	307✔
164	return accumulator	76✔
165	else
166	-- Simply recurse on children
167	for _, child in ipairs(body) do	535✔
168	accumulator = inferArgTypes_aux(accumulator, typeRequired, child)	608✔
169	end
170	return accumulator	231✔
171	end
172	else SU.error("invalid argument to inferArgTypes_aux") end	×
173	end
174
175	local inferArgTypes = function (body)
176	return inferArgTypes_aux({}, objType.tree, body)	174✔
177	end
178
179	local function registerCommand (name, argTypes, func)
180	commands[name] = { argTypes, func }	210✔
181	end
182
183	-- Computes func(func(... func(init, k1, v1), k2, v2)..., k_n, v_n), i.e. applies
184	-- func on every key-value pair in the table. Keys with numeric indices are
185	-- processed in order. This is an important property for MathML compilation below.
186	local function fold_pairs (func, table)
187	local accumulator = {}	1,233✔
188	for k, v in pl.utils.kpairs(table) do	10,153✔
189	accumulator = func(v, k, accumulator)	6,454✔
190	end
191	for i, v in ipairs(table) do	2,912✔
192	accumulator = func(v, i, accumulator)	3,358✔
193	end
194	return accumulator	1,233✔
195	end
196
197	local function forall (pred, list)
198	for _,x in ipairs(list) do	46✔
199	if not pred(x) then return false end	74✔
200	end
201	return true	9✔
202	end
203
204	local compileToStr = function (argEnv, mathlist)
205	if #mathlist == 1 and mathlist.id == "atom" then	100✔
206	-- List is a single atom
207	return mathlist[1]	×
208	elseif #mathlist == 1 and mathlist[1].id == "argument" then	100✔
209	return argEnv[mathlist[1].index]	×
210	elseif mathlist.id == "argument" then	100✔
211	return argEnv[mathlist.index]	×
212	else
213	local ret = ""	100✔
214	for _,elt in ipairs(mathlist) do	410✔
215	if elt.id == "atom" then	310✔
216	ret = ret .. elt[1]	310✔
217	elseif elt.id == "command" and symbols[elt.command] then	×
218	ret = ret .. symbols[elt.command]	×
219	else
220	SU.error("Encountered non-character token in command that takes a string")	×
221	end
222	end
223	return ret	100✔
224	end
225	end
226
227	local function compileToMathML_aux (_, arg_env, tree)
228	if type(tree) == "string" then return tree end	1,578✔
229	local function compile_and_insert (child, key, accumulator)
230	if type(key) ~= "number" then	4,906✔
231	accumulator[key] = child	3,227✔
232	return accumulator	3,227✔
233	-- Compile all children, except if this node is a macro definition (no
234	-- evaluation "under lambda") or the application of a registered macro
235	-- (since evaluating the nodes depends on the macro's signature, it is more
236	-- complex and done below)..
237	elseif tree.id == "def" or (tree.id == "command" and commands[tree.command]) then	1,679✔
238	-- Conserve unevaluated child
239	table.insert(accumulator, child)	307✔
240	else
241	-- Compile next child
242	local comp = compileToMathML_aux(nil, arg_env, child)	1,372✔
243	if comp then	1,372✔
244	if comp.id == "wrapper" then	1,198✔
245	-- Insert all children of the wrapper node
246	for _, inner_child in ipairs(comp) do	170✔
247	table.insert(accumulator, inner_child)	87✔
248	end
249	else
250	table.insert(accumulator, comp)	1,115✔
251	end
252	end
253	end
254	return accumulator	1,679✔
255	end
256	tree = fold_pairs(compile_and_insert, tree)	2,466✔
257	if tree.id == "texlike_math" then	1,233✔
258	tree.command = "math"	46✔
259	-- If the outermost `mrow` contains only other `mrow`s, remove it
260	-- (allowing vertical stacking).
261	if forall(function (c) return c.command == "mrow" end, tree[1]) then	129✔
262	tree[1].command = "math"	9✔
263	return tree[1]	9✔
264	end
265	elseif tree.id == "mathlist" then	1,187✔
266	-- Turn mathlist into `mrow` except if it has exactly one `mtr` or `mtd`
267	-- child.
268	-- Note that `def`s have already been compiled away at this point.
269	if #tree == 1 and (tree[1].command == "mtr" or tree[1].command == "mtd") then	278✔
270	return tree[1]	×
271	else tree.command = "mrow" end	278✔
272	tree.command = "mrow"	278✔
273	elseif tree.id == "atom" then	909✔
274	local codepoints = {}	345✔
275	for _, cp in luautf8.codes(tree[1]) do	702✔
276	table.insert(codepoints, cp)	357✔
277	end
278	local cp = codepoints[1]	345✔
279	if #codepoints == 1 and ( -- If length of UTF-8 string is 1	345✔
280	cp >= SU.codepoint("A") and cp <= SU.codepoint("Z") or	811✔
281	cp >= SU.codepoint("a") and cp <= SU.codepoint("z") or	793✔
282	cp >= SU.codepoint("Α") and cp <= SU.codepoint("Ω") or	523✔
283	cp >= SU.codepoint("α") and cp <= SU.codepoint("ω")	517✔
284	) then	×
285	tree.command = "mi"	108✔
286	elseif lpeg.match(lpeg.R("09")^1, tree[1]) then	237✔
287	tree.command = "mn"	100✔
288	else
289	tree.command = "mo"	137✔
290	end
291	tree.options = {}	345✔
292	-- Translate TeX-like sub/superscripts to `munderover` or `msubsup`,
293	-- depending on whether the base is a big operator
294	elseif tree.id == "sup" and tree[1].command == "mo"	564✔
295	and tree[1].atom == atomType.bigOperator then	×
296	tree.command = "mover"	×
297	elseif tree.id == "sub" and tree[1].command == "mo"	564✔
298	and symbolDefaults[tree[1][1]].atom == atomType.bigOperator then	4✔
299	tree.command = "munder"	×
300	elseif tree.id == "subsup" and tree[1].command == "mo"	564✔
301	and symbolDefaults[tree[1][1]].atom == atomType.bigOperator then	10✔
302	tree.command = "munderover"	8✔
303	elseif tree.id == "supsub" and tree[1].command == "mo"	556✔
304	and symbolDefaults[tree[1][1]].atom == atomType.bigOperator then	×
305	tree.command = "munderover"	×
306	local tmp = tree[2]	×
307	tree[2] = tree[3]	×
308	tree[3] = tmp	×
309	elseif tree.id == "sup" then	556✔
310	tree.command = "msup"	10✔
311	elseif tree.id == "sub" then	546✔
312	tree.command = "msub"	28✔
313	elseif tree.id == "subsup" then	518✔
314	tree.command = "msubsup"	6✔
315	elseif tree.id == "supsub" then	512✔
316	tree.command = "msubsup"	×
317	local tmp = tree[2]	×
318	tree[2] = tree[3]	×
319	tree[3] = tmp	×
320	elseif tree.id == "def" then	512✔
321	local commandName = tree["command-name"]	174✔
322	local argTypes = inferArgTypes(tree[1])	174✔
323	registerCommand(commandName, argTypes, function (compiledArgs)	348✔
324	return compileToMathML_aux(nil, compiledArgs, tree[1])	83✔
325	end)
326	return nil	174✔
327	elseif tree.id == "command" and commands[tree.command] then	338✔
328	local argTypes = commands[tree.command][1]	184✔
329	local cmdFun = commands[tree.command][2]	184✔
330	local applicationTree = tree	184✔
331	local cmdName = tree.command	184✔
332	if #applicationTree ~= #argTypes then	184✔
333	SU.error("Wrong number of arguments (" .. #applicationTree ..	×
334	") for command " .. cmdName .. " (should be " ..	×
335	#argTypes .. ")")	×
336	end
337	-- Compile every argument
338	local compiledArgs = {}	184✔
339	for i,arg in pairs(applicationTree) do	1,053✔
340	if type(i) == "number" then	869✔
341	if argTypes[i] == objType.tree then	133✔
342	table.insert(compiledArgs, compileToMathML_aux(nil, arg_env, arg))	66✔
343	else
344	local x = compileToStr(arg_env, arg)	100✔
345	table.insert(compiledArgs, x)	100✔
346	end
347	else
348	-- Not an argument but an attribute. Add it to the compiled
349	-- argument tree as-is
350	compiledArgs[i] = applicationTree[i]	736✔
351	end
352	end
353	local res = cmdFun(compiledArgs)	184✔
354	if res.command == "mrow" then	184✔
355	-- Mark the outer mrow to be unwrapped in the parent
356	res.id = "wrapper"	83✔
357	end
358	return res	184✔
359	elseif tree.id == "command" and symbols[tree.command] then	154✔
360	local atom = {id = "atom", [1] = symbols[tree.command]}	44✔
361	tree = compileToMathML_aux(nil, arg_env, atom)	88✔
362	elseif tree.id == "argument" then	110✔
363	if arg_env[tree.index] then	33✔
364	return arg_env[tree.index]	33✔
365	else
366	SU.error("Argument #"..tree.index.." has escaped its scope (probably not fully applied command).")	×
367	end
368	end
369	tree.id = nil	833✔
370	return tree	833✔
371	end
372
373	local function printMathML (tree)
374	if type(tree) == "string" then	×
375	return tree	×
376	end
377	local result = "\\" .. tree.command	×
378	if tree.options then	×
379	local options = {}	×
380	for k,v in pairs(tree.options) do	×
381	table.insert(options, k .. "=" .. v)	×
382	end
383	if #options > 0 then	×
384	result = result .. "[" .. table.concat(options, ", ") .. "]"	×
385	end
386	end
387	if #tree > 0 then	×
388	result = result .. "{"	×
389	for _, child in ipairs(tree) do	×
390	result = result .. printMathML(child)	×
391	end
392	result = result .. "}"	×
393	end
394	return result	×
395	end
396
397	local function compileToMathML (_, arg_env, tree)
398	local result = compileToMathML_aux(_, arg_env, tree)	46✔
399	SU.debug("texmath", function ()	92✔
400	return "Resulting MathML: " .. printMathML(result)	×
401	end)
402	return result	46✔
403	end
404
405	local function convertTexlike (_, content)
406	local ret = epnf.parsestring(mathParser, content[1])	46✔
407	SU.debug("texmath", function ()	92✔
408	return "Parsed TeX math: " .. pl.pretty.write(ret)	×
409	end)
410	return ret	46✔
411	end
412
413	registerCommand("%", {}, function ()	18✔
414	return { "%", command = "mo", options = {} }	1✔
415	end)
416	registerCommand("mi", { [1] = objType.str }, function (x) return x end)	94✔
417	registerCommand("mo", { [1] = objType.str }, function (x) return x end)	24✔
418	registerCommand("mn", { [1] = objType.str }, function (x) return x end)	9✔
419
420	compileToMathML(nil, {}, convertTexlike(nil, {[==[	18✔
421	\def{frac}{\mfrac{#1}{#2}}
422	\def{bi}{\mi[mathvariant=bold-italic]{#1}}
423	\def{dsi}{\mi[mathvariant=double-struck]{#1}}
424
425	% Standard spaces gleaned from plain TeX
426	\def{thinspace}{\mspace[width=thin]}
427	\def{negthinspace}{\mspace[width=-thin]}
428	\def{,}{\thinspace}
429	\def{!}{\negthinspace}
430	\def{medspace}{\mspace[width=med]}
431	\def{negmedspace}{\mspace[width=-med]}
432	\def{>}{\medspace}
433	\def{thickspace}{\mspace[width=thick]}
434	\def{negthickspace}{\mspace[width=-thick]}
435	\def{;}{\thickspace}
436	\def{enspace}{\mspace[width=1en]}
437	\def{enskip}{\enspace}
438	\def{quad}{\mspace[width=1em]}
439	\def{qquad}{\mspace[width=2em]}
440
441	% Modulus operator forms
442	\def{bmod}{\mo{mod}}
443	\def{pmod}{\quad(\mo{mod} #1)}
444	]==]}))
445
446	return { convertTexlike, compileToMathML }	9✔

sile-typesetter / sile / 6916312961

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous