vimtex/indent/tex.vim
2016-02-23 21:03:53 +01:00

153 lines
3.5 KiB
VimL

" vimtex - LaTeX plugin for Vim
"
" Maintainer: Karl Yngve Lervåg
" Email: karl.yngve@gmail.com
"
if exists('b:did_indent')
finish
endif
let b:did_indent = 1
let b:did_vimtex_indent = 1
call vimtex#util#set_default('g:vimtex_indent_enabled', 1)
if !g:vimtex_indent_enabled | finish | endif
let s:cpo_save = &cpo
set cpo&vim
setlocal autoindent
setlocal indentexpr=VimtexIndent()
setlocal indentkeys&
setlocal indentkeys+=[,(,{,),},],\&,=item
function! VimtexIndent() " {{{1
" Find a non-blank non-comment line above the current line
let lnum = prevnonblank(v:lnum - 1)
while lnum != 0 && getline(lnum) =~# '^\s*%'
let lnum = prevnonblank(lnum - 1)
endwhile
" Zero indent for top of file
if lnum == 0
return 0
endif
" Get current and previous line, remove comments
let cline = substitute(getline(v:lnum), '\\\@<!%.*', '', '')
let pline = substitute(getline(lnum), '\\\@<!%.*', '', '')
" Check for verbatim modes
if synIDattr(synID(v:lnum, indent(v:lnum), 1), 'name') ==# 'texZone'
if empty(cline)
return indent(lnum)
else
return indent(v:lnum)
end
endif
" Align on ampersands
if cline =~# '^\s*&' && pline =~# '\\\@<!&.*'
return indent(v:lnum) + match(pline, '\\\@<!&') - stridx(cline, '&')
endif
" Find previous non-empty non-comment non-ampersand line
while lnum != 0 && (match(pline, '\\\@<!&') != -1 || pline =~# '^\s*%')
let lnum = prevnonblank(lnum - 1)
let pline = getline(lnum)
endwhile
" Zero indent for top of file
if lnum == 0
return 0
endif
" Use previous indentation for comments
if cline =~# '^\s*%'
return indent(v:lnum)
endif
let ind = indent(lnum)
" Add indent on begin environment
if pline =~# '\\begin{.*}' && pline !~ s:envs_noindent
let ind = ind + &sw
" Add extra indent for list environments
if pline =~ s:envs_lists
let ind = ind + &sw
endif
endif
" Subtract indent on end environment
if cline =~# '\\end{.*}' && cline !~ s:envs_noindent
let ind = ind - &sw
" Subtract extra indent for list environments
if cline =~ s:envs_lists
let ind = ind - &sw
endif
endif
" Indent opening and closing delimiters
let [l:re_open, l:re_close] = vimtex#delim#get_valid_regexps(v:lnum, col('.'))
let ind += &sw*(
\ max([s:count(pline, l:re_open) - s:count(pline, l:re_close), 0])
\ - max([s:count(cline, l:re_close) - s:count(cline, l:re_open), 0]))
" Indent list items
if pline =~# '^\s*\\\(bib\)\?item'
let ind += &sw
endif
if cline =~# '^\s*\\\(bib\)\?item'
let ind -= &sw
endif
" Indent tikz elements
if pline =~ s:tikz_commands
let ind += &sw
let s:tikz_indented += 1
endif
if s:tikz_indented > 0 && pline =~# ';'
let ind -= &sw
let s:tikz_indented -= 1
endif
return ind
endfunction
"}}}
function! s:count(line, pattern) " {{{1
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
" }}}1
" {{{1 Script variables
let s:tikz_indented = 0
" Define some common patterns
let s:envs_lists = 'itemize\|description\|enumerate\|thebibliography'
let s:envs_noindent = 'document\|verbatim\|lstlisting'
let s:tikz_commands = '\\\(' . join([
\ 'draw',
\ 'fill',
\ 'path',
\ 'node',
\ 'add\(legendentry\|plot\)',
\ ], '\|') . '\)'
" }}}1
let &cpo = s:cpo_save
unlet s:cpo_save
" vim: fdm=marker sw=2