fix formatting;
This commit is contained in:
parent
0db6fa96b4
commit
03807e46e0
4 changed files with 15 additions and 14 deletions
|
|
@ -21,7 +21,6 @@ local function get_plugin_path()
|
||||||
return vim.fn.fnamemodify(plugin_path, ":h:h:h")
|
return vim.fn.fnamemodify(plugin_path, ":h:h:h")
|
||||||
end
|
end
|
||||||
|
|
||||||
|
|
||||||
local competition_types = { "atcoder", "codeforces", "cses" }
|
local competition_types = { "atcoder", "codeforces", "cses" }
|
||||||
|
|
||||||
local function setup_contest(contest_type)
|
local function setup_contest(contest_type)
|
||||||
|
|
@ -58,7 +57,7 @@ local function setup_problem(problem_id, problem_letter)
|
||||||
vim.g.cp_diff_mode = false
|
vim.g.cp_diff_mode = false
|
||||||
end
|
end
|
||||||
|
|
||||||
vim.cmd('silent only')
|
vim.cmd("silent only")
|
||||||
|
|
||||||
local scrape_result = scrape.scrape_problem(vim.g.cp_contest, problem_id, problem_letter)
|
local scrape_result = scrape.scrape_problem(vim.g.cp_contest, problem_id, problem_letter)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -90,20 +90,20 @@ function M.scrape_problem(contest, problem_id, problem_letter)
|
||||||
if #data.test_cases > 0 then
|
if #data.test_cases > 0 then
|
||||||
local all_inputs = {}
|
local all_inputs = {}
|
||||||
local all_outputs = {}
|
local all_outputs = {}
|
||||||
|
|
||||||
for i, test_case in ipairs(data.test_cases) do
|
for i, test_case in ipairs(data.test_cases) do
|
||||||
local input_lines = vim.split(test_case.input:gsub("\r", ""):gsub("\n+$", ""), "\n")
|
local input_lines = vim.split(test_case.input:gsub("\r", ""):gsub("\n+$", ""), "\n")
|
||||||
local output_lines = vim.split(test_case.output:gsub("\r", ""):gsub("\n+$", ""), "\n")
|
local output_lines = vim.split(test_case.output:gsub("\r", ""):gsub("\n+$", ""), "\n")
|
||||||
|
|
||||||
for _, line in ipairs(input_lines) do
|
for _, line in ipairs(input_lines) do
|
||||||
table.insert(all_inputs, line)
|
table.insert(all_inputs, line)
|
||||||
end
|
end
|
||||||
|
|
||||||
for _, line in ipairs(output_lines) do
|
for _, line in ipairs(output_lines) do
|
||||||
table.insert(all_outputs, line)
|
table.insert(all_outputs, line)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
vim.fn.writefile(all_inputs, input_file)
|
vim.fn.writefile(all_inputs, input_file)
|
||||||
vim.fn.writefile(all_outputs, expected_file)
|
vim.fn.writefile(all_outputs, expected_file)
|
||||||
end
|
end
|
||||||
|
|
|
||||||
|
|
@ -35,7 +35,7 @@ function M.restore_layout(state)
|
||||||
end
|
end
|
||||||
|
|
||||||
vim.cmd.diffoff()
|
vim.cmd.diffoff()
|
||||||
|
|
||||||
local problem_id = vim.fn.expand("%:t:r")
|
local problem_id = vim.fn.expand("%:t:r")
|
||||||
if problem_id == "" then
|
if problem_id == "" then
|
||||||
for win, win_state in pairs(state.windows) do
|
for win, win_state in pairs(state.windows) do
|
||||||
|
|
@ -50,13 +50,13 @@ function M.restore_layout(state)
|
||||||
end
|
end
|
||||||
|
|
||||||
if problem_id ~= "" then
|
if problem_id ~= "" then
|
||||||
vim.cmd('silent only')
|
vim.cmd("silent only")
|
||||||
|
|
||||||
local base_fp = vim.fn.getcwd()
|
local base_fp = vim.fn.getcwd()
|
||||||
local input = ("%s/io/%s.in"):format(base_fp, problem_id)
|
local input = ("%s/io/%s.in"):format(base_fp, problem_id)
|
||||||
local output = ("%s/io/%s.out"):format(base_fp, problem_id)
|
local output = ("%s/io/%s.out"):format(base_fp, problem_id)
|
||||||
local source = problem_id .. ".cc"
|
local source = problem_id .. ".cc"
|
||||||
|
|
||||||
vim.cmd.edit(source)
|
vim.cmd.edit(source)
|
||||||
vim.cmd.vsplit(output)
|
vim.cmd.vsplit(output)
|
||||||
vim.bo.filetype = "cp"
|
vim.bo.filetype = "cp"
|
||||||
|
|
@ -68,7 +68,7 @@ function M.restore_layout(state)
|
||||||
vim.cmd.wincmd("h")
|
vim.cmd.wincmd("h")
|
||||||
else
|
else
|
||||||
vim.cmd(state.layout)
|
vim.cmd(state.layout)
|
||||||
|
|
||||||
for win, win_state in pairs(state.windows) do
|
for win, win_state in pairs(state.windows) do
|
||||||
if vim.api.nvim_win_is_valid(win) then
|
if vim.api.nvim_win_is_valid(win) then
|
||||||
vim.api.nvim_set_current_win(win)
|
vim.api.nvim_set_current_win(win)
|
||||||
|
|
@ -86,7 +86,7 @@ end
|
||||||
|
|
||||||
function M.setup_diff_layout(actual_output, expected_output, input_file)
|
function M.setup_diff_layout(actual_output, expected_output, input_file)
|
||||||
vim.cmd.diffoff()
|
vim.cmd.diffoff()
|
||||||
vim.cmd('silent only')
|
vim.cmd("silent only")
|
||||||
|
|
||||||
local output_lines = vim.split(actual_output, "\n")
|
local output_lines = vim.split(actual_output, "\n")
|
||||||
local output_buf = vim.api.nvim_create_buf(false, true)
|
local output_buf = vim.api.nvim_create_buf(false, true)
|
||||||
|
|
|
||||||
|
|
@ -88,9 +88,11 @@ def main():
|
||||||
test_cases = []
|
test_cases = []
|
||||||
for input_data, output_data in tests:
|
for input_data, output_data in tests:
|
||||||
test_cases.append({"input": input_data, "output": output_data})
|
test_cases.append({"input": input_data, "output": output_data})
|
||||||
|
|
||||||
if test_cases:
|
if test_cases:
|
||||||
combined_input = str(len(test_cases)) + "\n" + "\n".join(tc["input"] for tc in test_cases)
|
combined_input = (
|
||||||
|
str(len(test_cases)) + "\n" + "\n".join(tc["input"] for tc in test_cases)
|
||||||
|
)
|
||||||
combined_output = "\n".join(tc["output"] for tc in test_cases)
|
combined_output = "\n".join(tc["output"] for tc in test_cases)
|
||||||
test_cases = [{"input": combined_input, "output": combined_output}]
|
test_cases = [{"input": combined_input, "output": combined_output}]
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue