From bbf75bd35d388b81c74b7895ab75ffe79c74bd0b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Karl=20Yngve=20Lerv=C3=A5g?= <karl.yngve+git@gmail.com>
Date: Fri, 22 Apr 2016 22:50:36 +0200
Subject: [PATCH] Minor update to folding

---
 autoload/vimtex/fold.vim | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/autoload/vimtex/fold.vim b/autoload/vimtex/fold.vim
index 50e5707..47bacd5 100644
--- a/autoload/vimtex/fold.vim
+++ b/autoload/vimtex/fold.vim
@@ -36,6 +36,21 @@ function! vimtex#fold#init_script() " {{{1
   let s:secs  = '\v^\s*(\\|\% Fake)(' . join(g:vimtex_fold_sections,  '|') . ')>'
   let s:notbslash = '\%(\\\@<!\%(\\\\\)*\)\@<='
   let s:notcomment = '\%(\%(\\\@<!\%(\\\\\)*\)\@<=%.*\)\@<!'
+
+  "
+  " List of identifiers for improving efficiency
+  "
+  let s:folded  = '\v^\s*\%'
+  let s:folded .= '|\\%(' . join([
+        \   'begin',
+        \   'end',
+        \   '%(sub)*%(section|paragraph)',
+        \   'chapter',
+        \   'document',
+        \   '%(front|main|back)matter',
+        \   'appendix',
+        \   'part',
+        \ ], '|') . ')'
 endfunction
 
 " }}}1
@@ -111,10 +126,7 @@ function! vimtex#fold#level(lnum) " {{{1
 
   " Check for normal lines first (optimization)
   let line = getline(a:lnum)
-  if line !~# '\v^\s*\%|\\%(document|begin|end|%(front|main|back)matter'
-          \ . '|appendix|part|chapter|%(sub)*%(section|paragraph))'
-    return '='
-  endif
+  if line !~# s:folded | return '=' | endif
 
   " Fold preamble
   if g:vimtex_fold_preamble