From 45697796012866ceda4dd0ef1b66e4c8a0e74546 Mon Sep 17 00:00:00 2001
From: Jan Larres <jan@majutsushi.net>
Date: Mon, 27 Aug 2012 01:57:56 +1200
Subject: [PATCH] Only generate prototypes on-demand

---
 autoload/tagbar.vim | 133 ++++++++++++++++++++++----------------------
 1 file changed, 65 insertions(+), 68 deletions(-)

diff --git a/autoload/tagbar.vim b/autoload/tagbar.vim
index 0299251..3667488 100644
--- a/autoload/tagbar.vim
+++ b/autoload/tagbar.vim
@@ -913,7 +913,7 @@ function! s:MapKeys() abort
 
     nnoremap <script> <silent> <buffer> <CR>    :call <SID>JumpToTag(0)<CR>
     nnoremap <script> <silent> <buffer> p       :call <SID>JumpToTag(1)<CR>
-    nnoremap <script> <silent> <buffer> <Space> :call <SID>ShowPrototype(1)<CR>
+    nnoremap <script> <silent> <buffer> <Space> :call <SID>ShowPrototype(0)<CR>
 
     nnoremap <script> <silent> <buffer> +        :call <SID>OpenFold()<CR>
     nnoremap <script> <silent> <buffer> <kPlus>  :call <SID>OpenFold()<CR>
@@ -949,7 +949,7 @@ function! s:CreateAutocommands() abort
     augroup TagbarAutoCmds
         autocmd!
         autocmd BufEnter   __Tagbar__ nested call s:QuitIfOnlyWindow()
-        autocmd CursorHold __Tagbar__ call s:ShowPrototype(0)
+        autocmd CursorHold __Tagbar__ call s:ShowPrototype(1)
 
         autocmd BufWritePost * call
                     \ s:AutoUpdate(fnamemodify(expand('<afile>'), ':p'), 1)
@@ -1141,6 +1141,7 @@ function! s:BaseTag._init(name) abort dict
     let self.fields        = {}
     let self.fields.line   = 0
     let self.fields.column = 1
+    let self.prototype     = ''
     let self.path          = ''
     let self.fullpath      = a:name
     let self.depth         = 0
@@ -1166,8 +1167,8 @@ function! s:BaseTag.isKindheader() abort dict
 endfunction
 
 " s:BaseTag.getPrototype() {{{3
-function! s:BaseTag.getPrototype() abort dict
-    return ''
+function! s:BaseTag.getPrototype(short) abort dict
+    return self.prototype
 endfunction
 
 " s:BaseTag._getPrefix() {{{3
@@ -1330,8 +1331,57 @@ function! s:NormalTag.str(longsig, full) abort dict
 endfunction
 
 " s:NormalTag.getPrototype() {{{3
-function! s:NormalTag.getPrototype() abort dict
-    return self.prototype
+function! s:NormalTag.getPrototype(short) abort dict
+    if self.prototype != ''
+        let prototype = self.prototype
+    else
+        let bufnr = self.fileinfo.bufnr
+
+        let line = getbufline(bufnr, self.fields.line)[0]
+        let list = split(line, '\zs')
+
+        let start = index(list, '(')
+        if start == -1
+            return substitute(line, '^\s\+', '', '')
+        endif
+
+        let opening = count(list, '(', 0, start)
+        let closing = count(list, ')', 0, start)
+        if closing >= opening
+            return substitute(line, '^\s\+', '', '')
+        endif
+
+        let balance = opening - closing
+
+        let prototype = line
+        let curlinenr = self.fields.line + 1
+        while balance > 0
+            let curline = getbufline(bufnr, curlinenr)[0]
+            let curlist = split(curline, '\zs')
+            let balance += count(curlist, '(')
+            let balance -= count(curlist, ')')
+            let prototype .= "\n" . curline
+            let curlinenr += 1
+        endwhile
+
+        let self.prototype = prototype
+    endif
+
+    if a:short
+        " join all lines and remove superfluous spaces
+        let prototype = substitute(prototype, '^\s\+', '', '')
+        let prototype = substitute(prototype, '\_s\+', ' ', 'g')
+        let prototype = substitute(prototype, '(\s\+', '(', 'g')
+        let prototype = substitute(prototype, '\s\+)', ')', 'g')
+        " Avoid hit-enter prompts
+        let maxlen = &columns - 12
+        if len(prototype) > maxlen
+            let prototype = prototype[:maxlen - 1 - 3]
+            let prototype .= '...'
+        endif
+    endif
+
+    return prototype
 endfunction
 
 " Pseudo tag {{{2
@@ -1364,7 +1414,7 @@ function! s:KindheaderTag.isKindheader() abort dict
 endfunction
 
 " s:KindheaderTag.getPrototype() {{{3
-function! s:KindheaderTag.getPrototype() abort dict
+function! s:KindheaderTag.getPrototype(short) abort dict
     return self.name . ': ' .
          \ self.numtags . ' ' . (self.numtags > 1 ? 'tags' : 'tag')
 endfunction
@@ -2008,8 +2058,8 @@ function! s:ParseTagline(part1, part2, typeinfo, fileinfo) abort
     else
         let dollar = ''
     endif
-    let pattern           = strpart(pattern, start, end - start)
-    let taginfo.pattern   = '\V\^\C' . pattern . dollar
+    let pattern         = strpart(pattern, start, end - start)
+    let taginfo.pattern = '\V\^\C' . pattern . dollar
 
     let fields = split(a:part2, '\t')
     let taginfo.fields.kind = remove(fields, 0)
@@ -2027,8 +2077,6 @@ function! s:ParseTagline(part1, part2, typeinfo, fileinfo) abort
         let taginfo.fields.line = taginfo.fields.lineno
     endif
 
-    let taginfo.prototype = s:GetPrototype(taginfo.fields.line)
-
     " Make some information easier accessible
     if has_key(a:typeinfo, 'scope2kind')
         for scope in keys(a:typeinfo.scope2kind)
@@ -2059,41 +2107,6 @@ function! s:ParseTagline(part1, part2, typeinfo, fileinfo) abort
     return taginfo
 endfunction
 
-" s:GetPrototype() {{{2
-" Look for unbalanced opening parentheses and add lines until they are
-" balanced do get the complete prototype.
-function! s:GetPrototype(linenr) abort
-    let line = getline(a:linenr)
-    let list = split(line, '\zs')
-
-    let start = index(list, '(')
-    if start == -1
-        return substitute(line, '^\s\+', '', '')
-    endif
-
-    let opening = count(list, '(', 0, start)
-    let closing = count(list, ')', 0, start)
-    if closing >= opening
-        return substitute(line, '^\s\+', '', '')
-    endif
-
-    let balance = opening - closing
-
-    let prototype = line
-    let curlinenr = a:linenr
-    while balance > 0
-        let curlinenr += 1
-        let curline = getline(curlinenr)
-        let curlist = split(curline, '\zs')
-        let opening += count(curlist, '(')
-        let closing += count(curlist, ')')
-        let balance = opening - closing
-        let prototype .= "\n" . curline
-    endwhile
-
-    return prototype
-endfunction
-
 " s:AddScopedTags() {{{2
 " Recursively process tags. Unfortunately there is a problem: not all tags in
 " a hierarchy are actually there. For example, in C++ a class can be defined
@@ -2798,30 +2811,14 @@ function! s:JumpToTag(stay_in_tagbar) abort
 endfunction
 
 " s:ShowPrototype() {{{2
-function! s:ShowPrototype(long) abort
+function! s:ShowPrototype(short) abort
     let taginfo = s:GetTagInfo(line('.'), 1)
 
     if empty(taginfo)
-        return
+        return ''
     endif
 
-    let prototype = taginfo.getPrototype()
-
-    if !a:long
-        " join all lines
-        let prototype = substitute(prototype, '^\s\+', '', '')
-        let prototype = substitute(prototype, '\_s\+', ' ', 'g')
-        let prototype = substitute(prototype, '(\s\+', '(', 'g')
-        let prototype = substitute(prototype, '\s\+)', ')', 'g')
-        " Avoid hit-enter prompts
-        let maxlen = &columns - 12
-        if len(prototype) > maxlen
-            let prototype = prototype[:maxlen - 1 - 3]
-            let prototype .= '...'
-        endif
-    endif
-
-    echo prototype
+    echo taginfo.getPrototype(a:short)
 endfunction
 
 " s:ToggleHelp() {{{2
@@ -3261,7 +3258,7 @@ function! s:GetTagInfo(linenr, ignorepseudo) abort
     endif
 
     " Don't do anything in empty and comment lines
-    let curline = getline(a:linenr)
+    let curline = getbufline(bufnr('__Tagbar__'), a:linenr)[0]
     if curline =~ '^\s*$' || curline[0] == '"'
         return {}
     endif
@@ -3342,10 +3339,10 @@ function! TagbarBalloonExpr() abort
     let taginfo = s:GetTagInfo(v:beval_lnum, 1)
 
     if empty(taginfo)
-        return
+        return ''
     endif
 
-    return taginfo.getPrototype()
+    return taginfo.getPrototype(0)
 endfunction
 
 " TagbarGenerateStatusline() {{{2