vimtex/indent/tex.vim

164 lines
4.3 KiB
VimL
Raw Normal View History

2015-03-07 17:02:15 -05:00
" vimtex - LaTeX plugin for Vim
2014-07-22 18:08:57 -04:00
"
" Maintainer: Karl Yngve Lervåg
" Email: karl.yngve@gmail.com
"
2015-03-19 10:12:12 -04:00
if exists('b:did_indent')
2013-10-10 08:47:08 -04:00
finish
endif
let b:did_indent = 1
2016-02-23 15:03:53 -05:00
let b:did_vimtex_indent = 1
2015-01-26 09:41:40 -05:00
2015-03-07 17:02:15 -05:00
call vimtex#util#set_default('g:vimtex_indent_enabled', 1)
if !g:vimtex_indent_enabled | finish | endif
2015-01-26 09:41:40 -05:00
2013-10-10 08:47:08 -04:00
let s:cpo_save = &cpo
set cpo&vim
setlocal autoindent
setlocal indentexpr=VimtexIndent()
2013-10-10 08:47:08 -04:00
setlocal indentkeys&
2015-10-26 17:43:36 -04:00
setlocal indentkeys+=[,(,{,),},],\&,=item
2013-10-10 08:47:08 -04:00
function! VimtexIndent() " {{{1
2016-03-27 18:22:10 -04:00
let l:nprev = s:get_prev_line(prevnonblank(v:lnum - 1))
if l:nprev == 0 | return 0 | endif
2013-10-10 08:47:08 -04:00
2016-03-24 07:52:45 -04:00
" Get current and previous line and remove comments
let l:cur = substitute(getline(v:lnum), '\\\@<!%.*', '', '')
let l:prev = substitute(getline(l:nprev), '\\\@<!%.*', '', '')
2013-10-10 08:47:08 -04:00
" Check for verbatim modes
2016-03-27 18:22:10 -04:00
if s:is_verbatim(l:cur, v:lnum)
2016-03-24 07:52:45 -04:00
return empty(l:cur) ? indent(l:nprev) : indent(v:lnum)
2013-10-10 08:47:08 -04:00
endif
" Align on ampersands
2016-03-24 07:52:45 -04:00
if l:cur =~# '^\s*&' && l:prev =~# '\\\@<!&.*'
return indent(v:lnum) + match(l:prev, '\\\@<!&') - stridx(l:cur, '&')
endif
" Use previous indentation for comments
if l:cur =~# '^\s*%'
return indent(v:lnum)
2013-10-10 08:47:08 -04:00
endif
2016-03-27 18:22:10 -04:00
let l:nprev = s:get_prev_line(l:nprev, 'ignore-ampersands')
if l:nprev == 0 | return 0 | endif
let l:prev = getline(l:nprev)
2013-10-10 08:47:08 -04:00
2016-03-24 07:52:45 -04:00
let l:ind = indent(l:nprev)
let l:ind += s:indent_envs(l:cur, l:prev)
let l:ind += s:indent_delims(l:cur, l:prev)
2016-03-24 18:50:36 -04:00
let l:ind += s:indent_tikz(l:nprev, l:prev)
return l:ind
endfunction
"}}}
2013-10-10 08:47:08 -04:00
2016-03-27 18:22:10 -04:00
function! s:get_prev_line(lnum, ...) " {{{1
let l:ignore_amps = a:0 > 0
let l:lnum = a:lnum
let l:prev = getline(l:lnum)
while l:lnum != 0
\ && (l:prev =~# '^\s*%'
\ || s:is_verbatim(l:prev, l:lnum)
\ || match(l:prev, '\\\@<!&') >= 0)
let l:lnum = prevnonblank(l:lnum - 1)
let l:prev = getline(l:lnum)
endwhile
return l:lnum
endfunction
" }}}1
function! s:is_verbatim(line, lnum) " {{{1
let l:env = a:line !~# '\v\\%(begin|end)\{%(verbatim|lstlisting|minted)'
let l:syn = synIDattr(synID(a:lnum, 1, 1), 'name') ==# 'texZone'
return l:env && l:syn
endfunction
" }}}1
function! s:indent_envs(cur, prev) " {{{1
let l:ind = 0
2013-10-10 08:47:08 -04:00
" First for general environments
2016-03-27 18:22:10 -04:00
let l:ind += &sw*((a:prev =~# '\\begin{.*}') && (a:prev !~# 'document'))
let l:ind -= &sw*((a:cur =~# '\\end{.*}') && (a:cur !~# 'document'))
2013-10-10 08:47:08 -04:00
" Indentation for prolonged items in lists
let l:ind += &sw*((a:prev =~# s:envs_item) && (a:cur !~# s:envs_enditem))
let l:ind -= &sw*((a:cur =~# s:envs_item) && (a:prev !~# s:envs_begitem))
let l:ind -= &sw*((a:cur =~# s:envs_endlist) && (a:prev !~# s:envs_begitem))
2013-10-10 08:47:08 -04:00
return l:ind
endfunction
let s:envs_lists = 'itemize\|description\|enumerate\|thebibliography'
let s:envs_item = '^\s*\\item'
let s:envs_beglist = '\\begin{\%(' . s:envs_lists . '\)'
let s:envs_endlist = '\\end{\%(' . s:envs_lists . '\)'
let s:envs_begitem = s:envs_item . '\|' . s:envs_beglist
let s:envs_enditem = s:envs_item . '\|' . s:envs_endlist
" }}}1
function! s:indent_delims(cur, prev) " {{{1
let [l:open, l:close] = vimtex#delim#get_valid_regexps(v:lnum, col('.'))
return &sw*( max([s:count(a:prev, l:open) - s:count(a:prev, l:close), 0])
\ - max([s:count(a:cur, l:close) - s:count(a:cur, l:open), 0]))
endfunction
2013-10-10 08:47:08 -04:00
" }}}1
2016-03-24 18:50:36 -04:00
function! s:indent_tikz(lnum, prev) " {{{1
if vimtex#env#is_inside('tikzpicture')
let l:prev_starts = a:prev =~# s:tikz_commands
let l:prev_stops = a:prev =~# ';\s*$'
" Increase indent on tikz command start
if l:prev_starts && ! l:prev_stops
return &sw
endif
" Decrease indent on tikz command end, i.e. on semicolon
if ! l:prev_starts && l:prev_stops
let l:context = join(getline(max([1,a:lnum-4]), a:lnum-1), '')
return -&sw*(l:context =~# s:tikz_commands)
endif
endif
2016-03-24 18:50:36 -04:00
return 0
2013-10-10 08:47:08 -04:00
endfunction
let s:tikz_commands = '\v\\%(' . join([
\ 'draw',
\ 'fill',
\ 'path',
\ 'node',
\ 'coordinate',
\ 'add%(legendentry|plot)',
\ ], '|') . ')'
" }}}1
function! s:count(line, pattern) " {{{1
2013-10-10 08:47:08 -04:00
let sum = 0
let indx = match(a:line, a:pattern)
while indx >= 0
let sum += 1
let match = matchstr(a:line, a:pattern, indx)
let indx += len(match)
let indx = match(a:line, a:pattern, indx)
endwhile
return sum
endfunction
2015-03-07 17:02:15 -05:00
" }}}1
2013-10-10 08:47:08 -04:00
let &cpo = s:cpo_save
unlet s:cpo_save
2014-12-08 14:44:17 -05:00
" vim: fdm=marker sw=2