From 6f92838e19eee4d2bc0360864d93970b3d25b36a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marcel=20Fabian=20Kr=C3=BCger?= <tex@2krueger.de>
Date: Fri, 21 May 2021 17:30:23 +0200
Subject: [PATCH] Small improvements/fixes

---
 luamml-tex.lua | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/luamml-tex.lua b/luamml-tex.lua
index 1ccea18..f8d4b03 100644
--- a/luamml-tex.lua
+++ b/luamml-tex.lua
@@ -23,6 +23,14 @@ lua.get_functions_table()[funcid] = function()
   end
 end
 
+local function shallow_copy(t)
+  local tt = {}
+  for k,v in next, t do
+    tt[k] = v
+  end
+  return tt
+end
+
 -- Possible flag values:
 --   0: Normal (This is the only supported one in display mode)
 --   1: Like 0, result is display math
@@ -48,14 +56,16 @@ local function save_result(xml, display)
   mlist_result, mlist_display = xml, display
   token.put_next(filename_token)
   local filename = token.scan_argument()
+  local tracing = tex.count.tracingmathml > 1
+  local xml_root = (filename ~= '' or tracing) and make_root(shallow_copy(xml), display and 0 or 2)
   if filename ~= '' then
     assert(io.open(filename, 'w'))
-      :write(write_xml(make_root({[0] = 'mrow', xml}, display and 0 or 2), true):sub(2) .. '\n')
+      :write(write_xml(xml_root, true):sub(2) .. '\n')
       :close()
   end
-  if tex.count.tracingmathml > 1 then
+  if tracing then
     -- Here xml gets wrapped in an mrow to avoid modifying it.
-    texio.write_nl(write_xml(make_root({[0] = 'mrow', xml}, display and 0 or 2)) .. '\n')
+    texio.write_nl(write_xml(xml_root) .. '\n')
   end
 end
 
@@ -81,7 +91,7 @@ luatexbase.add_to_callback('pre_mlist_to_hlist_filter', function(mlist, style)
     save_result(xml, style == 'display' or flag & 1 == 1)
   end
   if flag & 8 == 8 then
-    write_struct(make_root(mlist_result, mlist_display and 0 or 2))
+    write_struct(make_root(shallow_copy(xml), (style == 'display' or flag & 1 == 1) and 0 or 2))
   end
   if style == 'text' then
     local startmath = tex.nest.top.tail