63 lines
1.9 KiB
Lua
63 lines
1.9 KiB
Lua
vim.opt.runtimepath:prepend("scripts/benchmark.nvim")
|
|
vim.opt.runtimepath:prepend(".")
|
|
|
|
local bm = require("benchmark")
|
|
bm.sandbox()
|
|
|
|
---@module 'oil'
|
|
---@type oil.SetupOpts
|
|
local setup_opts = {
|
|
-- columns = { "icon", "permissions", "size", "mtime" },
|
|
}
|
|
|
|
local DIR_SIZE = tonumber(vim.env.DIR_SIZE) or 100000
|
|
local ITERATIONS = tonumber(vim.env.ITERATIONS) or 10
|
|
local WARM_UP = tonumber(vim.env.WARM_UP) or 1
|
|
local OUTLIERS = tonumber(vim.env.OUTLIERS) or math.floor(ITERATIONS / 10)
|
|
local TEST_DIR = "perf/tmp/test_" .. DIR_SIZE
|
|
|
|
vim.fn.mkdir(TEST_DIR, "p")
|
|
require("benchmark.files").create_files(TEST_DIR, "file %d.txt", DIR_SIZE)
|
|
|
|
function _G.jit_profile()
|
|
require("oil").setup(setup_opts)
|
|
local finish = bm.jit_profile({ filename = TEST_DIR .. "/profile.txt" })
|
|
bm.wait_for_user_event("OilEnter", function()
|
|
finish()
|
|
end)
|
|
require("oil").open(TEST_DIR)
|
|
end
|
|
|
|
function _G.flame_profile()
|
|
local start, stop = bm.flame_profile({
|
|
pattern = "oil*",
|
|
filename = "profile.json",
|
|
})
|
|
require("oil").setup(setup_opts)
|
|
start()
|
|
bm.wait_for_user_event("OilEnter", function()
|
|
stop(function()
|
|
vim.cmd.qall({ mods = { silent = true } })
|
|
end)
|
|
end)
|
|
require("oil").open(TEST_DIR)
|
|
end
|
|
|
|
function _G.benchmark()
|
|
require("oil").setup(setup_opts)
|
|
bm.run({ title = "oil.nvim", iterations = ITERATIONS, warm_up = WARM_UP }, function(callback)
|
|
bm.wait_for_user_event("OilEnter", callback)
|
|
require("oil").open(TEST_DIR)
|
|
end, function(times)
|
|
local avg = bm.avg(times, { trim_outliers = OUTLIERS })
|
|
local std_dev = bm.std_dev(times, { trim_outliers = OUTLIERS })
|
|
local lines = {
|
|
table.concat(vim.tbl_map(bm.format_time, times), " "),
|
|
string.format("Average: %s", bm.format_time(avg)),
|
|
string.format("Std deviation: %s", bm.format_time(std_dev)),
|
|
}
|
|
|
|
vim.fn.writefile(lines, "perf/tmp/benchmark.txt")
|
|
vim.cmd.qall({ mods = { silent = true } })
|
|
end)
|
|
end
|