Make `Doc` keep track of max line length

This commit is contained in:
Guldoman 2021-08-23 05:37:14 +02:00 committed by Francesco
parent 3eb6f1dbd4
commit e52362e55f
2 changed files with 81 additions and 0 deletions

View File

@ -393,4 +393,12 @@ function common.rm(path, recursively)
return true return true
end end
function common.get_table_size(t)
local count = 0
for _ in pairs(t) do
count = count + 1
end
return count
end
return common return common

View File

@ -32,6 +32,7 @@ end
function Doc:reset() function Doc:reset()
self.lines = { "\n" } self.lines = { "\n" }
self.long_lines = { line_numbers = { [1] = true }, length = 0 }
self.selections = { 1, 1, 1, 1 } self.selections = { 1, 1, 1, 1 }
self.cursor_clipboard = {} self.cursor_clipboard = {}
self.undo_stack = { idx = 1 } self.undo_stack = { idx = 1 }
@ -60,19 +61,32 @@ end
function Doc:load(filename) function Doc:load(filename)
local fp = assert( io.open(filename, "rb") ) local fp = assert( io.open(filename, "rb") )
local max_length = 0
local line_numbers = { [1] = true }
self:reset() self:reset()
self.lines = {} self.lines = {}
local i = 1
for line in fp:lines() do for line in fp:lines() do
if line:byte(-1) == 13 then if line:byte(-1) == 13 then
line = line:sub(1, -2) line = line:sub(1, -2)
self.crlf = true self.crlf = true
end end
table.insert(self.lines, line .. "\n") table.insert(self.lines, line .. "\n")
local line_len = string.len(line)
if line_len > max_length then
max_length = line_len
line_numbers = { [i] = true } -- new longest line
elseif line_len == max_length then
line_numbers[i] = true -- add to longest lines
end
i = i + 1
end end
if #self.lines == 0 then if #self.lines == 0 then
table.insert(self.lines, "\n") table.insert(self.lines, "\n")
end end
fp:close() fp:close()
self.long_lines.line_numbers = line_numbers
self.long_lines.length = max_length
self:reset_syntax() self:reset_syntax()
end end
@ -347,6 +361,20 @@ function Doc:raw_insert(line, col, text, undo_stack, time)
push_undo(undo_stack, time, "selection", unpack(self.selections)) push_undo(undo_stack, time, "selection", unpack(self.selections))
push_undo(undo_stack, time, "remove", line, col, line2, col2) push_undo(undo_stack, time, "remove", line, col, line2, col2)
if #lines > 1 then
-- Need to shift all the subsequent long lines
local line_numbers = {}
for n in pairs(self.long_lines.line_numbers) do
if n > line then
line_numbers[n + #lines - 1] = true
else
line_numbers[n] = true
end
end
self.long_lines.line_numbers = line_numbers
end
self:update_max_line_len_range(line, line2)
-- update highlighter and assure selection is in bounds -- update highlighter and assure selection is in bounds
self.highlighter:invalidate(line) self.highlighter:invalidate(line)
self:sanitize_selection() self:sanitize_selection()
@ -374,6 +402,23 @@ function Doc:raw_remove(line1, col1, line2, col2, undo_stack, time)
self:set_selections(idx, cline1 - line_removal, ccol1 - column_removal, cline2 - line_removal, ccol2 - column_removal) self:set_selections(idx, cline1 - line_removal, ccol1 - column_removal, cline2 - line_removal, ccol2 - column_removal)
end end
local nlines = line2 - line1 + 1
if nlines > 1 then
-- Need to shift all the subsequent long lines
local line_numbers = {}
for n in pairs(self.long_lines.line_numbers) do
if n > line2 then
line_numbers[n - nlines + 1] = true
elseif n > line1 then
line_numbers[n] = nil -- invalidate any line that has been deleted
else
line_numbers[n] = true
end
end
self.long_lines.line_numbers = line_numbers
end
self:update_max_line_len_range(line1, line2)
-- update highlighter and assure selection is in bounds -- update highlighter and assure selection is in bounds
self.highlighter:invalidate(line1) self.highlighter:invalidate(line1)
self:sanitize_selection() self:sanitize_selection()
@ -529,6 +574,34 @@ function Doc:indent_text(unindent, line1, col1, line2, col2)
return line1, col1 + #text, line1, col1 + #text return line1, col1 + #text, line1, col1 + #text
end end
function Doc:update_max_line_len_range(start_line, end_line)
local line_numbers = self.long_lines.line_numbers
local max_length = self.long_lines.length
end_line = math.min(end_line, #self.lines)
for line=start_line,end_line do
local line_len = string.len(self.lines[line])
if line_len > max_length then
max_length = line_len
line_numbers = { [line] = true }
elseif line_len == max_length then
line_numbers[line] = true
else
line_numbers[line] = nil
end
end
if common.get_table_size(line_numbers) == 0 then
-- Recalc needed
self.long_lines.length = 0
self.long_lines.line_numbers = { [1] = true }
return self:update_max_line_len_range(1, #self.lines)
end
self.long_lines.line_numbers = line_numbers
self.long_lines.length = max_length
end
-- For plugins to add custom actions of document change -- For plugins to add custom actions of document change
function Doc:on_text_change(type) function Doc:on_text_change(type)
end end