barrett/cp.nvim
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix tle verdict

Browse source
This commit is contained in:
Barrett Ruth 2025-10-02 23:28:51 -04:00
parent cddd61f061
commit d480975652
1 changed files with 12 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

15
lua/cp/runner/execute.lua
View file

@ -63,7 +63,7 @@ function M.compile(language_config, substitutions)
return r return r
end end
local function parse_and_strip_time_v(output) local function parse_and_strip_time_v(output, memory_mb)
local lines = vim.split(output or '', '\n', { plain = true }) local lines = vim.split(output or '', '\n', { plain = true })
local timing_idx local timing_idx
@ -73,6 +73,12 @@ local function parse_and_strip_time_v(output)
break break
end end
end end
if not timing_idx then
while #lines > 0 and lines[#lines]:match('^%s*$') do
table.remove(lines, #lines)
end
return table.concat(lines, '\n'), 0, false
end
local start_idx = timing_idx local start_idx = timing_idx
local k = timing_idx - 1 local k = timing_idx - 1
@ -81,11 +87,14 @@ local function parse_and_strip_time_v(output)
k = k - 1 k = k - 1
end end
local peak_mb = 0 local peak_mb, mled = 0, false
for j = timing_idx, #lines do for j = timing_idx, #lines do
local kb = lines[j]:match('Maximum resident set size %(kbytes%):%s*(%d+)') local kb = lines[j]:match('Maximum resident set size %(kbytes%):%s*(%d+)')
if kb then if kb then
peak_mb = tonumber(kb) / 1024.0 peak_mb = tonumber(kb) / 1024.0
if memory_mb and memory_mb > 0 and peak_mb > memory_mb then
mled = true
end
end end
end end
@ -97,7 +106,7 @@ local function parse_and_strip_time_v(output)
table.remove(lines, #lines) table.remove(lines, #lines)
end end
return table.concat(lines, '\n'), peak_mb return table.concat(lines, '\n'), peak_mb, mled
end end
function M.run(cmd, stdin, timeout_ms, memory_mb) function M.run(cmd, stdin, timeout_ms, memory_mb)