" EasyMotion - Vim motions on speed! " " Author: Kim Silkebækken " Maintainer: haya14busa " Source: https://github.com/haya14busa/vim-easymotion " Original: https://github.com/Lokaltog/vim-easymotion "============================================================================= " Saving 'cpoptions' {{{ scriptencoding utf-8 let s:save_cpo = &cpo set cpo&vim " }}} " Init: {{{ function! EasyMotion#init() " Init Migemo Dictionary let s:previous = {} let s:dot_repeat = {} let s:migemo_dicts = {} let s:EasyMotion_is_active = 0 call EasyMotion#reset() " Anywhere regular expression: {{{ let re = '\v' . \ '(<.|^$)' . '|' . \ '(.>|^$)' . '|' . \ '(\l)\zs(\u)' . '|' . \ '(_\zs.)' . '|' . \ '(#\zs.)' " 1. word " 2. end of word " 3. CamelCase " 4. after '_' hoge_foo " 5. after '#' hoge#foo let g:EasyMotion_re_anywhere = get(g:, 'EasyMotion_re_anywhere', re) " Anywhere regular expression within line: let re = '\v' . \ '(<.|^$)' . '|' . \ '(.>|^$)' . '|' . \ '(\l)\zs(\u)' . '|' . \ '(_\zs.)' . '|' . \ '(#\zs.)' let g:EasyMotion_re_line_anywhere = get(g:, 'EasyMotion_re_line_anywhere', re) "}}} return "" endfunction "}}} " Reset: {{{ function! EasyMotion#reset() let s:flag = { \ 'within_line' : 0, \ 'dot_repeat' : 0, \ 'regexp' : 0, \ } let s:current = { \ 'is_operator' : 0, \ 'dot_repeat_target_cnt' : 0, \ 'dot_prompt_user_cnt' : 0, \ 'changedtick' : 0, \ } return "" endfunction "}}} " Motion Functions: {{{ " -- Find Motion ------------------------- function! EasyMotion#S(num_strokes, visualmode, direction) " {{{ let is_exclusive = mode(1) ==# 'no' ? 1 : 0 let re = s:findMotion(a:num_strokes) if s:handleEmpty(re, a:visualmode) | return | endif call s:EasyMotion(re, a:direction, a:visualmode ? visualmode() : '', is_exclusive) endfunction " }}} function! EasyMotion#T(num_strokes, visualmode, direction) " {{{ let is_exclusive = mode(1) ==# 'no' ? 1 : 0 let re = s:findMotion(a:num_strokes) if s:handleEmpty(re, a:visualmode) | return | endif let re = a:direction == 1 ? '\('.re.'\)\zs.' : '.\ze\('.re.'\)' call s:EasyMotion(re, a:direction, a:visualmode ? visualmode() : '', is_exclusive) endfunction " }}} " -- Word Motion ------------------------- function! EasyMotion#WB(visualmode, direction) " {{{ let s:current.is_operator = mode(1) ==# 'no' ? 1: 0 call s:EasyMotion('\(\<.\|^$\)', a:direction, a:visualmode ? visualmode() : '', 0) endfunction " }}} function! EasyMotion#WBW(visualmode, direction) " {{{ let s:current.is_operator = mode(1) ==# 'no' ? 1: 0 call s:EasyMotion('\(\(^\|\s\)\@<=\S\|^$\)', a:direction, a:visualmode ? visualmode() : '', 0) endfunction " }}} function! EasyMotion#E(visualmode, direction) " {{{ let s:current.is_operator = mode(1) ==# 'no' ? 1: 0 let is_exclusive = mode(1) ==# 'no' ? 1 : 0 call s:EasyMotion('\(.\>\|^$\)', a:direction, a:visualmode ? visualmode() : '', is_exclusive) endfunction " }}} function! EasyMotion#EW(visualmode, direction) " {{{ let s:current.is_operator = mode(1) ==# 'no' ? 1: 0 let is_exclusive = mode(1) ==# 'no' ? 1 : 0 call s:EasyMotion('\(\S\(\s\|$\)\|^$\)', a:direction, a:visualmode ? visualmode() : '', is_exclusive) endfunction " }}} " -- JK Motion --------------------------- function! EasyMotion#JK(visualmode, direction) " {{{ let s:current.is_operator = mode(1) ==# 'no' ? 1: 0 "FIXME: support exclusive if g:EasyMotion_startofline call s:EasyMotion('^\(\w\|\s*\zs\|$\)', a:direction, a:visualmode ? visualmode() : '', 0) else let prev_column = getpos('.')[2] - 1 call s:EasyMotion('^.\{,' . prev_column . '}\zs\(.\|$\)', a:direction, a:visualmode ? visualmode() : '', 0) endif endfunction " }}} function! EasyMotion#Sol(visualmode, direction) " {{{ let s:current.is_operator = mode(1) ==# 'no' ? 1: 0 call s:EasyMotion('^\(\w\|\s*\zs\|$\)', a:direction, a:visualmode ? visualmode() : '', '') endfunction " }}} function! EasyMotion#Eol(visualmode, direction) " {{{ let s:current.is_operator = mode(1) ==# 'no' ? 1: 0 call s:EasyMotion('\(\w\|\s*\zs\|.\|^\)$', a:direction, a:visualmode ? visualmode() : '', '') endfunction " }}} " -- Search Motion ----------------------- function! EasyMotion#Search(visualmode, direction) " {{{ let s:current.is_operator = mode(1) ==# 'no' ? 1: 0 call s:EasyMotion(@/, a:direction, a:visualmode ? visualmode() : '', 0) endfunction " }}} " -- JumpToAnywhere Motion --------------- function! EasyMotion#JumpToAnywhere(visualmode, direction) " {{{ let s:current.is_operator = mode(1) ==# 'no' ? 1: 0 call s:EasyMotion( g:EasyMotion_re_anywhere, a:direction, a:visualmode ? visualmode() : '', 0) endfunction " }}} " -- Line Motion ------------------------- function! EasyMotion#SL(num_strokes, visualmode, direction) " {{{ let is_exclusive = mode(1) ==# 'no' ? 1 : 0 let s:flag.within_line = 1 let re = s:findMotion(a:num_strokes) if s:handleEmpty(re, a:visualmode) | return | endif call s:EasyMotion(re, a:direction, a:visualmode ? visualmode() : '', is_exclusive) endfunction " }}} function! EasyMotion#TL(num_strokes, visualmode, direction) " {{{ let is_exclusive = mode(1) ==# 'no' ? 1 : 0 let s:flag.within_line = 1 let re = s:findMotion(a:num_strokes) if s:handleEmpty(re, a:visualmode) | return | endif let re = a:direction == 1 ? '\('.re.'\)\zs.' : '.\ze\('.re.'\)' call s:EasyMotion(re, a:direction, a:visualmode ? visualmode() : '', is_exclusive) endfunction " }}} function! EasyMotion#WBL(visualmode, direction) " {{{ let s:current.is_operator = mode(1) ==# 'no' ? 1: 0 let is_exclusive = mode(1) ==# 'no' ? 1 : 0 let s:flag.within_line = 1 call s:EasyMotion('\(\<.\|^$\)', a:direction, a:visualmode ? visualmode() : '', 0) endfunction " }}} function! EasyMotion#EL(visualmode, direction) " {{{ let s:flag.within_line = 1 let s:current.is_operator = mode(1) ==# 'no' ? 1: 0 let is_exclusive = mode(1) ==# 'no' ? 1 : 0 call s:EasyMotion('\(.\>\|^$\)', a:direction, a:visualmode ? visualmode() : '', is_exclusive) endfunction " }}} function! EasyMotion#LineAnywhere(visualmode, direction) " {{{ let s:flag.within_line = 1 let s:current.is_operator = mode(1) ==# 'no' ? 1: 0 let re = g:EasyMotion_re_line_anywhere call s:EasyMotion(re, a:direction, a:visualmode ? visualmode() : '', 0) endfunction " }}} " -- Special Motion ---------------------- function! EasyMotion#SelectLines() "{{{ let orig_pos = [line('.'), col('.')] call s:EasyMotion('^\(\w\|\s*\zs\|$\)', 2, '', 0, 0, 0, 1) if s:EasyMotion_cancelled keepjumps call cursor(orig_pos[0], orig_pos[1]) return '' else let pos1 = [line('.'), col('.')] keepjumps call cursor(orig_pos[0], orig_pos[1]) call s:EasyMotion('^\(\w\|\s*\zs\|$\)', 2, '', 0, pos1[0], 1, 1) if s:EasyMotion_cancelled keepjumps call cursor(orig_pos[0], orig_pos[1]) return '' else normal! V keepjumps call cursor(pos1[0], pos1[1]) normal! o return 1 endif endif endfunction "}}} function! EasyMotion#SelectLinesYank() "{{{ let orig_pos = [line('.'), col('.')] call EasyMotion#SelectLines() normal! y keepjumps call cursor(orig_pos[0], orig_pos[1]) endfunction "}}} function! EasyMotion#SelectLinesDelete() "{{{ let orig_pos = [line('.'), col('.')] " if cancelled? if EasyMotion#SelectLines() " Prepare the number of lines "{{{ let start_of_line = line("v") silent exec "normal!" "o" let end_of_line = line("v") "}}} normal! d if orig_pos[0] < max([start_of_line,end_of_line]) keepjumps call cursor(orig_pos[0], orig_pos[1]) else " if delete lines above cursor line keepjumps call cursor(orig_pos[0]-abs(end_of_line-start_of_line)-1, orig_pos[1]) endif else keepjumps call cursor(orig_pos[0], orig_pos[1]) endif endfunction "}}} function! EasyMotion#SelectPhrase() "{{{ let chars = s:GetSearchChar2(0) if empty(chars) return endif " Generate regexp {{{ if chars[0] ==# chars[1] let re = s:convertRegep(chars[0]) else " Convert chars {{{ " let g:EasyMotion_smartcase to 0 temporarily let save_smart = g:EasyMotion_smartcase let g:EasyMotion_smartcase = 0 let re1 = s:convertRegep(chars[0]) let re2 = s:convertRegep(chars[1]) let g:EasyMotion_smartcase = save_smart unlet save_smart "}}} let re = re1 . '\|' . re2 if g:EasyMotion_smartcase && chars[0] =~# '\U' || chars[1] =~# '\U' let re = '\c' . re else let re = '\C' . re endif endif "}}} " Store original pos let orig_pos = [line('.'), col('.')] " First call s:EasyMotion(re, 2, '', 0, 0, 0, 0, 0) if s:EasyMotion_cancelled keepjumps call cursor(orig_pos[0], orig_pos[1]) return '' endif " Store first pos let pos1 = [line('.'), col('.')] keepjumps call cursor(orig_pos[0], orig_pos[1]) " Second call s:EasyMotion(re, 2, '', 0, 0, 0, 0, pos1) if s:EasyMotion_cancelled keepjumps call cursor(orig_pos[0], orig_pos[1]) return '' endif " Success normal! v keepjumps call cursor(pos1[0], pos1[1]) normal! o return 1 endfunction "}}} function! EasyMotion#SelectPhraseYank() "{{{ let orig_pos = [line('.'), col('.')] call EasyMotion#SelectPhrase() normal! y keepjumps call cursor(orig_pos[0], orig_pos[1]) endfunction "}}} function! EasyMotion#SelectPhraseDelete() "{{{ let orig_pos = [line('.'), col('.')] " If cancelled? if EasyMotion#SelectPhrase() " Prepare the number of lines "{{{ let start_of_line = line("v") silent exec "normal!" "o" let end_of_line = line("v") "}}} normal! d if orig_pos[0] < max([start_of_line,end_of_line]) keepjumps call cursor(orig_pos[0], orig_pos[1]) else " if you delete phrase above cursor line and phrase is over lines keepjumps call cursor(orig_pos[0]-abs(end_of_line-start_of_line), orig_pos[1]) endif else keepjumps call cursor(orig_pos[0], orig_pos[1]) endif endfunction "}}} " -- User Motion ------------------------- function! EasyMotion#User(pattern, mode, direction) " {{{ let s:current.is_operator = mode(1) ==# 'no' ? 1: 0 let visualmode = match('\v([Vv])|(C-v)', a:mode) > 0 ? visualmode() : '' let re = escape(a:pattern, '|') call s:EasyMotion(re, a:direction, visualmode, 0) endfunction " }}} " -- Repeat Motion ----------------------- function! EasyMotion#Repeat(visualmode) " {{{ " Repeat previous motion with previous targets if s:previous ==# {} call s:Message("Previous targets doesn't exist") return endif let re = s:previous.regexp let direction = s:previous.direction let s:flag.within_line = s:previous.line_flag let s:current.is_operator = mode(1) ==# 'no' ? 1: 0 let is_exclusive = mode(1) ==# 'no' ? 1 : 0 call s:EasyMotion(re, direction, a:visualmode ? visualmode() : '', is_exclusive) endfunction " }}} function! EasyMotion#DotRepeat(visualmode) " {{{ " Repeat previous motion with previous targets if s:dot_repeat ==# {} call s:Message("Previous motion doesn't exist") return endif let re = s:dot_repeat.regexp let direction = s:dot_repeat.direction let is_exclusive = s:dot_repeat.is_exclusive let s:flag.within_line = s:dot_repeat.line_flag let s:current.is_operator = 1 for cnt in range(v:count1) let s:flag.dot_repeat = 1 " s:EasyMotion() always call reset silent call s:EasyMotion(re, direction, 0, is_exclusive) endfor endfunction " }}} function! EasyMotion#NextPrevious(visualmode, direction) " {{{ " Repeat previous motion with previous targets if s:previous ==# {} call s:Message("Previous targets doesn't exist") return endif let re = s:previous.regexp let search_direction = (a:direction >= 1 ? 'b' : '') if g:EasyMotion_move_highlight call EasyMotion#highlight#attach_autocmd() call EasyMotion#highlight#add_highlight(re,'EasyMotionMoveHL') endif if ! empty(a:visualmode) " FIXME: blink highlight silent exec 'normal! gv' endif for i in range(v:count1) call searchpos(re, search_direction) endfor call EasyMotion#reset() " -- Activate EasyMotion ----------------- {{{ let s:EasyMotion_is_active = 1 call EasyMotion#attach_active_autocmd() "}}} endfunction " }}} " }}} " Helper Functions: {{{ " -- Message ----------------------------- function! s:Message(message) " {{{ echo 'EasyMotion: ' . a:message endfunction " }}} function! s:Prompt(message) " {{{ echohl Question echo a:message . ': ' echohl None endfunction " }}} " -- Save & Restore values --------------- function! s:VarReset(var, ...) " {{{ if ! exists('s:var_reset') let s:var_reset = {} endif if a:0 == 0 && has_key(s:var_reset, a:var) " Reset var to original value " setbufbar( or bufname): '' or '%' can be used for the current buffer call setbufvar("", a:var, s:var_reset[a:var]) elseif a:0 == 1 " Save original value and set new var value let new_value = a:0 == 1 ? a:1 : '' " Store original value let s:var_reset[a:var] = getbufvar("", a:var) " Set new var value call setbufvar("", a:var, new_value) endif endfunction " }}} function! s:SaveValue() "{{{ call s:VarReset('&scrolloff', 0) call s:VarReset('&modified', 0) call s:VarReset('&modifiable', 1) call s:VarReset('&readonly', 0) call s:VarReset('&spell', 0) call s:VarReset('&virtualedit', '') call s:VarReset('&foldmethod', 'manual') endfunction "}}} function! s:RestoreValue() "{{{ call s:VarReset('&scrolloff') call s:VarReset('&modified') call s:VarReset('&modifiable') call s:VarReset('&readonly') call s:VarReset('&spell') call s:VarReset('&virtualedit') call s:VarReset('&foldmethod') endfunction "}}} " -- Draw -------------------------------- function! s:SetLines(lines, key) " {{{ for [line_num, line] in a:lines call setline(line_num, line[a:key]) endfor endfunction " }}} " -- Get characters from user input ------ function! s:GetChar() " {{{ let char = getchar() if char == 27 " Escape key pressed redraw call s:Message('Cancelled') return '' endif return nr2char(char) endfunction " }}} function! s:GetSearchChar2(visualmode) " {{{ let chars = [] for i in [1, 2] redraw call s:Prompt('Search for character ' . i) let char = s:GetChar() " Check that we have an input char if empty(char) " Restore selection if ! empty(a:visualmode) silent exec 'normal! gv' endif return '' endif call add(chars, char) endfor return chars endfunction " }}} function! s:GetSearchChar(visualmode) " {{{ call s:Prompt('Search for character') let char = s:GetChar() " Check that we have an input char if empty(char) " Restore selection if ! empty(a:visualmode) silent exec 'normal! gv' endif return '' endif return char endfunction " }}} " -- Find Motion Helper ------------------ function! s:findMotion(num_strokes) "{{{ " Find Motion: S,F,T let s:current.is_operator = mode(1) ==# 'no' ? 1: 0 let s:flag.regexp = a:num_strokes == -1 ? 1 : 0 let s:previous['input'] = get(s:previous, 'input', '') let input = EasyMotion#command_line#GetInput(a:num_strokes, s:previous.input) let s:previous['input'] = input " Check that we have an input char if empty(input) redraw | return '' endif let re = s:convertRegep(input) return re endfunction "}}} function! s:convertRegep(input) "{{{ let re = s:should_use_regexp() ? a:input : escape(a:input, '.$^~\[]') if s:should_use_migemo(a:input) let re = s:convertMigemo(re) endif if s:should_use_smartsign(a:input) let re = s:convertSmartsign(re, a:input) endif let case_flag = s:should_use_smartcase(a:input) ? '\c' : '\C' let re .= case_flag return re endfunction "}}} function! s:convertMigemo(re) "{{{ let re = a:re if len(re) > 1 " System cmigemo return EasyMotion#cmigemo#getMigemoPattern(re) endif " EasyMoton migemo one key dict if ! has_key(s:migemo_dicts, &l:encoding) let s:migemo_dicts[&l:encoding] = EasyMotion#helper#load_migemo_dict() endif if re =~# '^\a$' let re = get(s:migemo_dicts[&l:encoding], re, a:re) endif return re endfunction "}}} function! s:convertSmartsign(re, char) "{{{ let smart_dict = s:load_smart_dict() let upper_sign = escape(get(smart_dict, a:char, ''), '.$^~') if upper_sign ==# '' return a:re else let re = a:re . '\|' . upper_sign return re endif endfunction "}}} function! s:convertSmartcase(re, char) "{{{ let re = a:re if a:char =~# '\U' "nonuppercase return '\c' . re else "uppercase return '\C' . re endif endfunction "}}} function! s:should_use_regexp() "{{{ return g:EasyMotion_use_regexp == 1 && s:flag.regexp == 1 endfunction "}}} function! s:should_use_migemo(char) "{{{ if ! g:EasyMotion_use_migemo || match(a:char, '\A') != -1 return 0 endif " TODO: use direction if s:flag.within_line == 1 let first_line = line('.') let end_line = line('.') else let first_line = line('w0') let end_line = line('w$') endif for line in range(first_line, end_line) if s:is_folded(line) continue endif if EasyMotion#helper#include_multibyte_char(getline(line)) == 1 return 1 endif endfor return 0 endfunction "}}} function! s:should_use_smartsign(char) "{{{ if (exists('g:EasyMotion_use_smartsign_us') || \ exists('g:EasyMotion_use_smartsign_jp')) && \ match(a:char, '\A') != -1 return 1 else return 0 endif endfunction "}}} function! s:should_use_smartcase(input) "{{{ if g:EasyMotion_smartcase == 0 return 0 endif " return 1 if input didn't match upporcase letter return match(a:input, '\u') == -1 endfunction "}}} function! s:handleEmpty(input, visualmode) "{{{ " if empty, reselect and return 1 if empty(a:input) if ! empty(a:visualmode) silent exec 'normal! gv' endif return 1 endif return 0 endfunction "}}} function! s:load_smart_dict() "{{{ if exists('g:EasyMotion_use_smartsign_us') return g:EasyMotion#sticky_table#us elseif exists('g:EasyMotion_use_smartsign_jp') return g:EasyMotion#sticky_table#jp else return '' endif endfunction "}}} " -- Handle Visual Mode ------------------ function! s:GetVisualStartPosition(c_pos, v_start, v_end, search_direction) "{{{ let vmode = mode(1) if match('Vv',vmode) < 0 throw 'Unkown visual mode:'.vmode endif if vmode ==# 'V' "line-wise Visual " Line-wise Visual {{{ if a:v_start[0] == a:v_end[0] if a:search_direction == '' return a:v_start elseif a:search_direction == 'b' return a:v_end else throw 'Unkown search_direction' endif else if a:c_pos[0] == a:v_start[0] return a:v_end elseif a:c_pos[0] == a:v_end[0] return a:v_start endif endif "}}} else " Character-wise or Block-wise Visual"{{{ if a:c_pos == a:v_start return a:v_end elseif a:c_pos == a:v_end return a:v_start endif " virtualedit if a:c_pos[0] == a:v_start[0] return a:v_end elseif a:c_pos[0] == a:v_end[0] return a:v_start else throw 'Unkown a:c_pos' endif "}}} endif endfunction "}}} " -- Others ------------------------------ function! s:is_folded(line) "{{{ " Return false if g:EasyMotion_skipfoldedline == 1 " and line is start of folded lines return foldclosed(a:line) != -1 && \ (g:EasyMotion_skipfoldedline == 1 || \ a:line != foldclosed(a:line)) endfunction "}}} function! s:is_cmdwin() "{{{ return bufname('%') ==# '[Command Line]' endfunction "}}} function! s:should_use_wundo() "{{{ " wundu cannot use in command-line window and " unless undolist is not empty return ! s:is_cmdwin() && undotree().seq_last != 0 endfunction "}}} "}}} " Grouping Algorithms: {{{ let s:grouping_algorithms = { \ 1: 'SCTree' \ , 2: 'Original' \ } " -- Single-key/closest target priority tree {{{ " This algorithm tries to assign one-key jumps to all the targets closest to the cursor. " It works recursively and will work correctly with as few keys as two. function! s:GroupingAlgorithmSCTree(targets, keys) "{{{ " Prepare variables for working let targets_len = len(a:targets) let keys_len = len(a:keys) let groups = {} let keys = reverse(copy(a:keys)) " Semi-recursively count targets {{{ " We need to know exactly how many child nodes (targets) this branch will have " in order to pass the correct amount of targets to the recursive function. " Prepare sorted target count list {{{ " This is horrible, I know. But dicts aren't sorted in vim, so we need to " work around that. That is done by having one sorted list with key counts, " and a dict which connects the key with the keys_count list. let keys_count = [] let keys_count_keys = {} let i = 0 for key in keys call add(keys_count, 0) let keys_count_keys[key] = i let i += 1 endfor " }}} let targets_left = targets_len let level = 0 let i = 0 while targets_left > 0 " Calculate the amount of child nodes based on the current level let childs_len = (level == 0 ? 1 : (keys_len - 1) ) for key in keys " Add child node count to the keys_count array let keys_count[keys_count_keys[key]] += childs_len " Subtract the child node count let targets_left -= childs_len if targets_left <= 0 " Subtract the targets left if we added too many too " many child nodes to the key count let keys_count[keys_count_keys[key]] += targets_left break endif let i += 1 endfor let level += 1 endwhile " }}} " Create group tree {{{ let i = 0 let key = 0 call reverse(keys_count) for key_count in keys_count if key_count > 1 " We need to create a subgroup " Recurse one level deeper let groups[a:keys[key]] = s:GroupingAlgorithmSCTree(a:targets[i : i + key_count - 1], a:keys) elseif key_count == 1 " Assign single target key let groups[a:keys[key]] = a:targets[i] else " No target continue endif let key += 1 let i += key_count endfor " }}} " Finally! return groups endfunction "}}} " }}} " -- Original ---------------------------- {{{ function! s:GroupingAlgorithmOriginal(targets, keys) " Split targets into groups (1 level) let targets_len = len(a:targets) let keys_len = len(a:keys) let groups = {} let i = 0 let root_group = 0 try while root_group < targets_len let groups[a:keys[root_group]] = {} for key in a:keys let groups[a:keys[root_group]][key] = a:targets[i] let i += 1 endfor let root_group += 1 endwhile catch | endtry " Flatten the group array if len(groups) == 1 let groups = groups[a:keys[0]] endif return groups endfunction " }}} " -- Coord/key dictionary creation ------- {{{ function! s:CreateCoordKeyDict(groups, ...) " Dict structure: " 1,2 : a " 2,3 : b let sort_list = [] let coord_keys = {} let group_key = a:0 == 1 ? a:1 : '' for [key, item] in items(a:groups) let key = group_key . key "let key = ( ! empty(group_key) ? group_key : key) if type(item) == 3 " List " Destination coords " The key needs to be zero-padded in order to " sort correctly let dict_key = printf('%05d,%05d', item[0], item[1]) let coord_keys[dict_key] = key " We need a sorting list to loop correctly in " PromptUser, dicts are unsorted call add(sort_list, dict_key) else " Item is a dict (has children) let coord_key_dict = s:CreateCoordKeyDict(item, key) " Make sure to extend both the sort list and the " coord key dict call extend(sort_list, coord_key_dict[0]) call extend(coord_keys, coord_key_dict[1]) endif unlet item endfor return [sort_list, coord_keys] endfunction " }}} " }}} " Core Functions: {{{ function! s:PromptUser(groups, allows_repeat, fixed_column) "{{{ let group_values = values(a:groups) " -- If only one possible match, jump directly to it {{{ if len(group_values) == 1 redraw return group_values[0] endif " }}} " -- Prepare marker lines ---------------- {{{ let lines = {} let hl_coords = [] let hl2_first_coords = [] " Highlight for two characters let hl2_second_coords = [] " Highlight for two characters let coord_key_dict = s:CreateCoordKeyDict(a:groups) for dict_key in sort(coord_key_dict[0]) let target_key = coord_key_dict[1][dict_key] let [line_num, col_num] = split(dict_key, ',') let line_num = str2nr(line_num) let col_num = str2nr(col_num) " Add original line and marker line if ! has_key(lines, line_num) let current_line = getline(line_num) let lines[line_num] = { 'orig': current_line, 'marker': current_line, 'mb_compensation': 0 } endif " Solve multibyte issues by matching the byte column " number instead of the visual column let col_num -= lines[line_num]['mb_compensation'] " Compensate for byte difference between marker " character and target character " " This has to be done in order to match the correct " column; \%c matches the byte column and not display " column. let target_char_len = strdisplaywidth(matchstr(lines[line_num]['marker'], '\%' . col_num . 'c.')) let target_key_len = strdisplaywidth(target_key) let target_line_byte_len = strlen(lines[line_num]['marker']) let target_char_byte_len = strlen(matchstr(lines[line_num]['marker'], '\%' . col_num . 'c.')) if a:fixed_column let firstS = match(lines[line_num]['marker'], '\S') if firstS >= 4 let leftText = strpart(lines[line_num]['marker'], 0, firstS - 3) else let leftText = '' endif if firstS >= 1 let rightText = strpart(lines[line_num]['marker'], firstS - 1) elseif firstS == 0 let rightText = ' ' . lines[line_num]['marker'] else let rightText = '' endif if target_key_len < 2 let text = ' ' . target_key call add(hl_coords, '\%' . line_num . 'l\%2c') else let text = target_key call add(hl2_first_coords, '\%' . line_num . 'l\%1c') call add(hl2_second_coords, '\%' . line_num . 'l\%2c') endif let lines[line_num]['marker'] = text . ' ' . lines[line_num]['marker'] else if strlen(lines[line_num]['marker']) > 0 " Substitute marker character if line length > 0 let c = 0 while c < target_key_len && c < 2 if strlen(lines[line_num]['marker']) >= col_num + c " Substitute marker character if line length > 0 if c == 0 let lines[line_num]['marker'] = substitute( \ lines[line_num]['marker'], \ '\%' . (col_num + c) . 'c.', \ strpart(target_key, c, 1) . repeat(' ', target_char_len - 1), \ '') else let lines[line_num]['marker'] = substitute( \ lines[line_num]['marker'], \ '\%' . (col_num + c) . 'c.', \ strpart(target_key, c, 1), \ '') endif else let lines[line_num]['marker'] = lines[line_num]['marker'] . strpart(target_key, c, 1) endif let c += 1 endwhile else " Set the line to the marker character if the line is empty let lines[line_num]['marker'] = target_key endif endif " Add highlighting coordinates if !a:fixed_column if target_key_len == 1 call add(hl_coords, '\%' . line_num . 'l\%' . col_num . 'c') else call add(hl2_first_coords, '\%' . line_num . 'l\%' . (col_num) . 'c') call add(hl2_second_coords, '\%' . line_num . 'l\%' . (col_num + 1) . 'c') endif endif " Add marker/target length difference for multibyte " compensation let lines[line_num]['mb_compensation'] += (target_line_byte_len - strlen(lines[line_num]['marker']) ) endfor let lines_items = items(lines) " }}} " -- Highlight targets ------------------- {{{ if len(hl_coords) > 0 call EasyMotion#highlight#add_highlight( \ join(hl_coords, '\|'), \ g:EasyMotion_hl_group_target) endif if len(hl2_second_coords) > 0 call EasyMotion#highlight#add_highlight( \ join(hl2_second_coords, '\|'), \ g:EasyMotion_hl2_second_group_target) endif if len(hl2_first_coords) > 0 call EasyMotion#highlight#add_highlight( \ join(hl2_first_coords, '\|'), \ g:EasyMotion_hl2_first_group_target) endif " }}} " -- Put labels on targets & Get User Input & Restore all {{{ " Save undo tree {{{ let s:undo_file = tempname() if s:should_use_wundo() execute "wundo" s:undo_file endif "}}} try " Set lines with markers {{{ call s:SetLines(lines_items, 'marker') redraw "}}} " Get target character {{{ call s:Prompt('Target key') let char = s:GetChar() "}}} " Convert uppercase {{{ if g:EasyMotion_use_upper == 1 && match(g:EasyMotion_keys, '\l') == -1 let char = toupper(char) endif "}}} " Jump first target when Enter key is pressed "{{{ if char ==# ' ' && g:EasyMotion_enter_jump_first == 1 let char = g:EasyMotion_keys[0] endif "}}} " For dot repeat {{{ if mode(1) ==# 'no' " Store previous target when operator pending mode if s:current.dot_prompt_user_cnt == 0 " Store let s:previous['target'] = char else " Append target chars let s:previous['target'] .= char endif endif "}}} finally " Restore original lines call s:SetLines(lines_items, 'orig') " Un-highlight targets {{{ call EasyMotion#highlight#delete_highlight( \ g:EasyMotion_hl_group_target, \ g:EasyMotion_hl2_first_group_target, \ g:EasyMotion_hl2_second_group_target, \ ) " }}} " Restore undo tree {{{ if s:should_use_wundo() && filereadable(s:undo_file) silent execute "rundo" s:undo_file unlet s:undo_file else " Break undo history (undobreak) let old_undolevels = &undolevels set undolevels=-1 call setline('.', getline('.')) let &undolevels = old_undolevels unlet old_undolevels " FIXME: Error occur by GundoToggle for undo number 2 is empty call setline('.', getline('.')) endif "}}} redraw endtry "}}} " -- Check if we have an input char ------ {{{ if empty(char) throw 'Cancelled' endif " }}} " -- Repeat EasyMotion ------------------- {{{ if a:allows_repeat && \ char == '.' && \ exists('s:previous_target_coord') " For SelectLines return s:previous_target_coord endif "}}} " -- Check if the input char is valid ---- {{{ if ! has_key(a:groups, char) throw 'Invalid target' endif " }}} let target = a:groups[char] if type(target) == 3 " Return target coordinates return target else " Prompt for new target character let s:current.dot_prompt_user_cnt += 1 return s:PromptUser(target, a:allows_repeat, a:fixed_column) endif endfunction "}}} function! s:DotPromptUser(groups) "{{{ " Get char from previous target let char = s:previous.target[s:current.dot_repeat_target_cnt] " For dot repeat target chars let s:current.dot_repeat_target_cnt += 1 let target = a:groups[char] if type(target) == 3 " Return target coordinates return target else " Prompt for new target character return s:PromptUser(target, a:allows_repeat, a:fixed_column) endif endfunction "}}} function! s:EasyMotion(regexp, direction, visualmode, is_exclusive, ...) " {{{ " For Special Function {{{ " For SelectLines(), to highlight previous selected line let hlcurrent = a:0 >= 1 ? a:1 : 0 " For SelectLines(), to allows '.' to repeat the previously pressed " character let allows_repeat = a:0 >= 2 ? a:2 : 0 " For SelectLines(), a flag to display character only at the beginning " of the line let fixed_column = a:0 >= 3 ? a:3 : 0 " For SelectPhrase() let hlchar = a:0 >= 4 ? a:4 : 0 "}}} let orig_pos = [line('.'), col('.')] let win_first_line = line('w0') let win_last_line = line('w$') let targets = [] if s:flag.dot_repeat != 1 " Store Regular Expression let s:previous['regexp'] = a:regexp let s:previous['direction'] = a:direction let s:previous['line_flag'] = s:flag.within_line == 1 ? 1 : 0 let s:previous['is_exclusive'] = a:is_exclusive let s:previous['operator'] = v:operator endif " To avoid side effect of overwriting buffer for tpope/repeat " store current b:changedtick let s:current.changedtick = b:changedtick try " -- Reset properties -------------------- {{{ " Save original value and set new value call s:SaveValue() " }}} " Setup searchpos args {{{ let search_direction = (a:direction >= 1 ? 'b' : '') let search_stopline = a:direction >= 1 ? win_first_line : win_last_line let search_at_cursor = fixed_column ? 'c' : '' if s:flag.within_line == 1 let search_stopline = orig_pos[0] endif "}}} " Handle visual mode {{{ if ! empty(a:visualmode) " Decide at where visual mode start {{{ normal! gv let c_pos = orig_pos " current_position let v_start = [line("'<"),col("'<")] " visual_start_position let v_end = [line("'>"),col("'>")] " visual_end_position let v_original_pos = s:GetVisualStartPosition(c_pos, v_start, v_end, search_direction) "}}} " Reselect visual text {{{ keepjumps call cursor(v_original_pos) exec "normal! " . a:visualmode keepjumps call cursor(c_pos) "}}} " Update orig_pos {{{ let orig_pos = v_original_pos " }}} endif " }}} " Construct match dict {{{ while 1 " Note: searchpos() has side effect which call jump cursor position. " You can disable this side effect by add 'n' flags, " but in this case, it's better to allows jump side effect. let pos = searchpos(a:regexp, search_direction . search_at_cursor, search_stopline) let search_at_cursor = '' " Reached end of search range if pos == [0, 0] break endif " Skip folded lines {{{ if s:is_folded(pos[0]) if search_direction ==# 'b' keepjumps call cursor(foldclosed(pos[0]-1), 0) else keepjumps call cursor(foldclosedend(pos[0]+1), 0) endif continue endif "}}} call add(targets, pos) endwhile "}}} " Handle bidirection "{{{ " Reconstruct match dict if a:direction == 2 " Forward if ! empty(a:visualmode) keepjumps call cursor(c_pos[0], c_pos[1]) else keepjumps call cursor(orig_pos[0], orig_pos[1]) endif let targets2 = [] if s:flag.within_line == 0 let search_stopline = win_last_line else let search_stopline = !empty(a:visualmode) ? c_pos[0] : orig_pos[0] endif while 1 let pos = searchpos(a:regexp, '', search_stopline) " Reached end of search range if pos == [0, 0] break endif " Skip folded lines {{{ if s:is_folded(pos[0]) " Always forward keepjumps call cursor(foldclosedend(pos[0]+1), 0) continue endif "}}} call add(targets2, pos) endwhile " Merge match target dict"{{{ let t1 = 0 " backward let t2 = 0 " forward let targets3 = [] while t1 < len(targets) || t2 < len(targets2) " Forward -> Backward -> F -> B -> ... if t2 < len(targets2) call add(targets3, targets2[t2]) let t2 += 1 endif if t1 < len(targets) call add(targets3, targets[t1]) let t1 += 1 endif endwhile let targets = targets3 "}}} endif "}}} " Handle no match"{{{ let targets_len = len(targets) if targets_len == 0 throw 'No matches' endif "}}} let GroupingFn = function('s:GroupingAlgorithm' . s:grouping_algorithms[g:EasyMotion_grouping]) let groups = GroupingFn(targets, split(g:EasyMotion_keys, '\zs')) " -- Shade inactive source --------------- {{{ if g:EasyMotion_do_shade && targets_len != 1 && s:flag.dot_repeat != 1 if !empty(a:visualmode) let shade_hl_pos = '\%' . c_pos[0] . 'l\%'. c_pos[1] .'c' else let shade_hl_pos = '\%' . orig_pos[0] . 'l\%'. orig_pos[1] .'c' endif if a:direction == 1 " Backward let shade_hl_re = '\%'. win_first_line .'l\_.*' . shade_hl_pos elseif a:direction == 0 " Forward let shade_hl_re = shade_hl_pos . '\_.*\%'. win_last_line .'l' elseif a:direction == 2 " Both directions" let shade_hl_re = '.*' endif if !fixed_column call EasyMotion#highlight#add_highlight( \ shade_hl_re, \ g:EasyMotion_hl_group_shade) endif endif if hlcurrent != 0 call EasyMotion#highlight#add_highlight( \ '\%'. hlcurrent .'l.*', \ g:EasyMotion_hl_line_group_shade) endif if !empty(hlchar) call EasyMotion#highlight#add_highlight( \ '\%'. hlchar[0] .'l\%' . hlchar[1] .'c', \ g:EasyMotion_hl_line_group_shade) endif " }}} " -- Jump back before prompt for visual scroll {{{ " Because searchpos() change current cursor position and " if you just use cursor([orig_num, orig_pos]) to jump back, " current line will become center of window if ! empty(a:visualmode) keepjumps call cursor(win_first_line,0) normal! zt endif "}}} " -- Prompt user for target group/character {{{ if s:flag.dot_repeat != 1 let coords = s:PromptUser(groups, allows_repeat, fixed_column) let s:previous_target_coord = coords else let coords = s:DotPromptUser(groups) endif "}}} " -- Update cursor position -------------- {{{ keepjumps call cursor(orig_pos[0], orig_pos[1]) " Consider EasyMotion as jump motion :h jump-motion normal! m` " Update selection {{{ if ! empty(a:visualmode) exec 'normal! ' . a:visualmode endif " }}} if s:flag.dot_repeat == 1 " support dot repeat {{{ " Use visual mode to emulate dot repeat normal! v if s:dot_repeat.is_exclusive == 0 if s:dot_repeat.direction == 0 "Forward let coords[1] -= 1 elseif s:dot_repeat.direction == 1 "Backward " Shift visual selection to left by making cursor one key " left. normal! hoh endif endif keepjumps call cursor(coords[0], coords[1]) let cmd = s:dot_repeat.operator if s:dot_repeat.operator ==# 'c' let cmd .= getreg('.') endif exec 'normal! ' . cmd "}}} else " Handle operator-pending mode {{{ if a:is_exclusive == 1 " Exclusive motion requires that we eat one more " character to the right if we're using " a forward motion normal! v endif " }}} " Adjust screen for visual scroll {{{ if ! empty(a:visualmode) keepjumps call cursor(win_first_line, 0) normal! zt endif "}}} keepjumps call cursor(coords[0], coords[1]) " To avoid side effect of overwriting buffer {{{ " for tpope/vim-repeat if exists('g:repeat_tick') if g:repeat_tick == s:current.changedtick let g:repeat_tick = b:changedtick endif endif "}}} endif " Set tpope/vim-repeat {{{ if s:current.is_operator == 1 && \ !(v:operator ==# 'y' && match(&cpo, 'y') == -1) " Store previous info for dot repeat {{{ let s:dot_repeat.regexp = a:regexp let s:dot_repeat.direction = a:direction let s:dot_repeat.line_flag = s:flag.within_line == 1 ? 1 : 0 let s:dot_repeat.is_exclusive = a:is_exclusive let s:dot_repeat.operator = v:operator "}}} silent! call repeat#set("\(easymotion-dotrepeat)") endif "}}} call s:Message('Jumping to [' . coords[0] . ', ' . coords[1] . ']') let s:EasyMotion_cancelled = 0 "}}} catch redraw " Show exception message call s:Message(v:exception) " -- Restore original cursor position/selection {{{ if ! empty(a:visualmode) silent exec 'normal! gv' keepjumps call cursor(c_pos[0], c_pos[1]) else keepjumps call cursor(orig_pos[0], orig_pos[1]) endif " }}} let s:EasyMotion_cancelled = 1 finally " -- Restore properties ------------------ {{{ call s:RestoreValue() call EasyMotion#reset() " }}} " -- Remove shading ---------------------- {{{ call EasyMotion#highlight#delete_highlight() " }}} " -- Landing Highlight ------------------- {{{ if g:EasyMotion_landing_highlight call EasyMotion#highlight#add_highlight(a:regexp, 'EasyMotionMoveHL') call EasyMotion#highlight#attach_autocmd() endif "}}} " -- Activate EasyMotion ----------------- {{{ let s:EasyMotion_is_active = 1 call EasyMotion#attach_active_autocmd() "}}} endtry endfunction " }}} function! EasyMotion#attach_active_autocmd() "{{{ " Reference: https://github.com/justinmk/vim-sneak augroup plugin-easymotion-active autocmd! autocmd InsertEnter,WinLeave,BufLeave \ let s:EasyMotion_is_active = 0 \ | autocmd! plugin-easymotion-active * autocmd CursorMoved \ autocmd plugin-easymotion-active CursorMoved \ let s:EasyMotion_is_active = 0 \ | autocmd! plugin-easymotion-active * augroup END endfunction "}}} function! EasyMotion#is_active() "{{{ return s:EasyMotion_is_active endfunction "}}} "}}} " Call Init: {{{ call EasyMotion#init() "}}} " Restore 'cpoptions' {{{ let &cpo = s:save_cpo unlet s:save_cpo " }}} " vim: fdm=marker:et:ts=4:sw=4:sts=4