leigh/nvim-cmp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Prefer above identifier for locality sort

Browse Source
This commit is contained in:
hrsh7th
2022-03-31 13:48:35 +09:00
parent 3a1649297a
commit c1589830bd
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lua/cmp/config/compare.lua
View File

@@ -101,6 +101,7 @@ compare.order = function(entry1, entry2)
end end
end end
-- locality
compare.locality = setmetatable({ compare.locality = setmetatable({
lines_count = 10, lines_count = 10,
lines_cache = cache.new(), lines_cache = cache.new(),
@@ -122,6 +123,7 @@ compare.locality = setmetatable({
self.locality_map = {} self.locality_map = {}
for i = math.max(0, cursor_row - self.lines_count), math.min(max, cursor_row + self.lines_count) do for i = math.max(0, cursor_row - self.lines_count), math.min(max, cursor_row + self.lines_count) do
local is_above = i < cursor_row
local buffer = vim.api.nvim_buf_get_lines(buf, i, i + 1, false)[1] or '' local buffer = vim.api.nvim_buf_get_lines(buf, i, i + 1, false)[1] or ''
local locality_map = self.lines_cache:ensure({ 'line', buffer }, function() local locality_map = self.lines_cache:ensure({ 'line', buffer }, function()
local locality_map = {} local locality_map = {}
@@ -130,7 +132,8 @@ compare.locality = setmetatable({
local s, e = regexp:match_str(buffer) local s, e = regexp:match_str(buffer)
if s and e then if s and e then
local w = string.sub(buffer, s + 1, e) local w = string.sub(buffer, s + 1, e)
locality_map[w] = math.min(locality_map[w] or math.huge, math.abs(i - cursor_row)) local d = math.abs(i - cursor_row) - (is_above and 0.1 or 0)
locality_map[w] = math.min(locality_map[w] or math.huge, d)
buffer = string.sub(buffer, e + 1) buffer = string.sub(buffer, e + 1)
else else
break break