manasthakur · March 20, 2017 04:31 · Mar 20, 2017 · Mar 20, 2017 · Mar 20, 2017
diff --git a/latex-fold.vim b/latex-fold.vim
@@ -1,4 +1,3 @@
-```vim
 function! MyFoldLevel(lnum)
     let cur_line = getline(a:lnum)
 
@@ -30,5 +29,4 @@ function! MyFoldLevel(lnum)
     endif
 
     return '='
-endfunction
-```
+endfunction
diff --git a/latex-fold.vim b/latex-fold.vim
@@ -2,14 +2,17 @@
 function! MyFoldLevel(lnum)
     let cur_line = getline(a:lnum)
 
+    " Fold sections at level 1
     if cur_line =~ '^\s*\\section'
         return '>1'
     endif
 
+    " Fold subsections at level 2
     if cur_line =~ '^\s*\\subsection'
         return '>2'
     endif
 
+    " Fold subsubsections at level 3
     if cur_line =~ '^\s*\\subsubsection'
         return '>3'
     endif

diff --git a/latex-fold.vim b/latex-fold.vim
@@ -0,0 +1,31 @@
+```vim
+function! MyFoldLevel(lnum)
+    let cur_line = getline(a:lnum)
+
+    if cur_line =~ '^\s*\\section'
+        return '>1'
+    endif
+
+    if cur_line =~ '^\s*\\subsection'
+        return '>2'
+    endif
+
+    if cur_line =~ '^\s*\\subsubsection'
+        return '>3'
+    endif
+
+    " Fold following environments
+    let fold_envs = ['figure', 'algorithm', 'frame']
+    let envs = '\(' . join(fold_envs, '\|') . '\)'
+
+    if cur_line =~ '^\s*\\begin{' . envs
+        return 'a1'
+    endif
+
+    if cur_line =~ '^\s*\\end{' . envs
+        return 's1'
+    endif
+
+    return '='
+endfunction
+```