From a89e7d1db91a044e3184668d8c2730a2359e6b40 Mon Sep 17 00:00:00 2001
From: UncleBill <billbill290@gmail.com>
Date: Fri, 5 Sep 2014 23:34:31 +0800
Subject: [PATCH] Optimize

---
 autoload/colorizer.vim | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/autoload/colorizer.vim b/autoload/colorizer.vim
index 0c2ebe3..eaff882 100644
--- a/autoload/colorizer.vim
+++ b/autoload/colorizer.vim
@@ -280,13 +280,17 @@ function! colorizer#ColorHighlight(update, ...) "{{{1
   if g:colorizer_fgcontrast != s:saved_fgcontrast || (exists("a:1") && a:1 == '!')
     let s:force_group_update = 1
   endif
-  for i in range(line("w0"), line("w$"))
+
+  let cur_line = line("w0")
+  let end_line = line("w$")
+  while cur_line < end_line
     " skip folded lines
-    if foldclosed(i) > 0
-      continue
+    if foldclosed(cur_line) > 0
+      let cur_line = foldclosedend(cur_line) + 1
     endif
-    call s:PreviewColorInLine(i)
-  endfor
+    call s:PreviewColorInLine(cur_line)
+    let cur_line += 1
+  endwhile
 
   " A hack for colorizing after opening a folding
   if ( winheight('.') - line('w$') + line('w0') ) > 1