" When . repeats g@, repeat the last macro.
fun! AtAtAtRepeat(_)
" If no count is supplied use the one saved in s:atcount.
" Otherwise save the new count in s:atcount, so it will be
" applied to repeats.
let s:atcount = v:count ? v:count : s:atcount
" feedkeys() rather than :normal allows finishing in Insert
" mode, should the macro do that. @@ is remapped, so opfunc will'opfunc'
" will be correct, even if the macro changes it.
call feedkeys(s:atcount.'@@')
endfun
fun! AtNextAtSetRepeat(_)
set opfunc=AtAtopfunc=AtRepeat
endfun
" Called by g@ being invoked directly for the first time. Sets
" 'opfunc' ready for repeats with . by calling AtNextAtSetRepeat().
fun! AtInit()
" Make sure setting 'opfunc' happens here, after initial playback
" of the macro recording, in case 'opfunc' is set there.
set opfunc=AtNextopfunc=AtSetRepeat
return 'g@l'
endfun
" Enable calling a function within the mapping for @
nno <expr> <plug>@init AtInit()
" A macro could, albeit unusually, end in Insert mode.
ino <expr> <plug>@init "\<c-o>".AtInit()
fun! AtReg()
let s:atcount = v:count1
let c = nr2char(getchar())
return '@'.c."\<plug>@init"
endfun
nmap <expr> @ AtReg()
fun! QrepeatQRepeat(_)
call feedkeys('@'.s:qreg)
endfun
fun! QsetrepeatQSetRepeat(_)
set opfunc=Qrepeatopfunc=QRepeat
endfun
fun! QstopQStop()
set opfunc=Qsetrepeatopfunc=QSetRepeat
return 'g@l'
endfun
nno <expr> <plug>qstop QstopQStop()
ino <expr> <plug>qstop "\<c-o>".QstopQStop()
let s:qrec = 0
fun! QstartQStart()
if s:qrec == 1
let s:qrec = 0
set opfunc=Qstop
return "q\<plug>qstop"
endif
let s:qrec = 1
let s:qreg = nr2char(getchar())
return 'q'.s:qreg
endfun
nmap <expr> q QstartQStart()