" Vim indent file
" Language:		Python
" Maintainer:		Bram Moolenaar <Bram@vim.org>
" Original Author:	David Bustos <bustos@caltech.edu>
" Last Change:		2008 Mar 14
" Changed my Jason Casden to try to prettify the line continuations

" Only load this indent file when no other was loaded.
if exists("b:did_indent")
  finish
endif
let b:did_indent = 1
let g:linecont = 0

" Some preliminary settings
setlocal nolisp		" Make sure lisp indenting doesn't supersede us
setlocal autoindent	" indentexpr isn't much help otherwise

setlocal indentexpr=GetPythonIndent(v:lnum)
setlocal indentkeys+=<:>,=elif,=except

" Only define the function once.
if exists("*GetPythonIndent")
  finish
endif

" Come here when loading the script the first time.

let s:maxoff = 50	" maximum number of lines to look backwards for ()


function GetPythonParenContinue(lnum3)
  call cursor(a:lnum3,1)

  
  "JMC
  let pnum2 = searchpair('(\|{\|\[', '', ')\|}\|\]', 'rnbW',
        \ "line('.') < " . (a:lnum3 - s:maxoff) . " ? dummy :"
        \ . " synIDattr(synID(line('.'), col('.'), 1), 'name')"
        \ . " =~ '\\(Comment\\|String\\)$'")

  if pnum2 > 0
    let g:linecont = 1
    return pnum2
  else
    return a:lnum3
  endif

endfunction


function GetPythonExplicitContinue(lnum2)

  let i = 1
  let ret_lnum = a:lnum2

  while (getline(a:lnum2 - i) =~ '\\\s*$')
    let g:linecont = 1
    let ret_lnum = a:lnum2 - i
    let i = i + 1
  endwhile

  return ret_lnum

endfunction

function GetPythonIndent(lnum)

"  " If this line is explicitly joined: If the previous line was also joined,
  "" line it up with that one, otherwise add two 'shiftwidth'
  "if getline(a:lnum - 1) =~ '\\$'
    "if a:lnum > 1 && getline(a:lnum - 2) =~ '\\\s*$'
      "return indent(a:lnum - 1)
    "endif
    "return indent(a:lnum - 1) + (exists("g:pyindent_continue") ? eval(g:pyindent_continue) : (&sw * 2))
  "endif

  " If the start of the line is in a string don't change the indent.
  if has('syntax_items')
	\ && synIDattr(synID(a:lnum, 1, 1), "name") =~ "String$"
    return -1
  endif

  " Search backwards for the previous non-empty line.
  let plnum = prevnonblank(v:lnum - 1)

  if plnum == 0
    " This is the first non-empty line, use zero indent.
    return 0
  endif

  " If the previous line is inside parenthesis, use the indent of the starting
  " line.
  " Trick: use the non-existing "dummy" variable to break out of the loop when
  " going too far back.
  call cursor(plnum, 1)

  "JMC
"  let parlnum = searchpair('(\|{\|\[', '', ')\|}\|\]', 'rnbW',
        "\ "line('.') < " . (plnum - s:maxoff) . " ? dummy :"
        "\ . " synIDattr(synID(line('.'), col('.'), 1), 'name')"
        "\ . " =~ '\\(Comment\\|String\\)$'")
        let parlnumcomp = 0
        let parlnum = plnum
        while (parlnumcomp != parlnum)
          let parlnumcomp = parlnum
          let parlnum = GetPythonParenContinue(parlnum)
          let parlnum = GetPythonExplicitContinue(parlnum)
        endwhile
  " If this line is explicitly joined: If the previous line was also joined,
  " line it up with that one, otherwise add two 'shiftwidth'
  if getline(plnum) =~ '\\$'
    return indent(parlnum) + (&sw * 2)
  else
    let plindent = indent(parlnum)
  endif

    let plnumstart = parlnum

  " When inside parenthesis: If at the first line below the parenthesis add
  " two 'shiftwidth', otherwise same as previous line.
  " i = (a
  "       + b
  "       + c)
  call cursor(a:lnum, 1)
  " JMC, changed to searchpairpos
  let [p,parcol] = searchpairpos('(\|{\|\[', '', ')\|}\|\]', 'bW',
	  \ "line('.') < " . (a:lnum - s:maxoff) . " ? dummy :"
	  \ . " synIDattr(synID(line('.'), col('.'), 1), 'name')"
	  \ . " =~ '\\(Comment\\|String\\)$'")
  if p > 0
    "if p == plnum
      " JMC
"      let [pp,parcol2] = searchpairpos('(\|{\|\[', '', ')\|}\|\]', 'bW',
	  "\ "line('.') < " . (a:lnum - s:maxoff) . " ? dummy :"
	  "\ . " synIDattr(synID(line('.'), col('.'), 1), 'name')"
	  "\ . " =~ '\\(Comment\\|String\\)$'")
      "if pp > 0
		"return indent(plnum) + (exists("g:pyindent_nested_paren") ? eval(g:pyindent_nested_paren) : parcol2)
      "endif
      
      " JMC, changed to return column position, unless specifically asked
      " otherwise
      "return indent(plnum) + (exists("g:pyindent_open_paren") ? eval(g:pyindent_open_paren) : (&sw * 2))
     return (exists("g:pyindent_open_paren") ? indent(p) + eval(g:pyindent_open_paren) : parcol)
    "endif
    "if plnumstart == p
      "return parcol
    "endif
    "return plindent
  endif


  " Get the line and remove a trailing comment.
  " Use syntax highlighting attributes when possible.
  let pline = getline(plnum)
  let pline_len = strlen(pline)
  if has('syntax_items')
    " If the last character in the line is a comment, do a binary search for
    " the start of the comment.  synID() is slow, a linear search would take
    " too long on a long line.
    if synIDattr(synID(plnum, pline_len, 1), "name") =~ "Comment$"
      let min = 1
      let max = pline_len
      while min < max
	let col = (min + max) / 2
	if synIDattr(synID(plnum, col, 1), "name") =~ "Comment$"
	  let max = col
	else
	  let min = col + 1
	endif
      endwhile
      let pline = strpart(pline, 0, min - 1)
    endif
  else
    let col = 0
    while col < pline_len
      if pline[col] == '#'
	let pline = strpart(pline, 0, col)
	break
      endif
      let col = col + 1
    endwhile
  endif

  " If the previous line ended with a colon, indent this line
  if pline =~ ':\s*$'
    " commented because it might not be a good idea:
    " only for new lines
    " JMC
    "if (getline(a:lnum) =~ '^\s*$')
      return plindent + &sw
    "endif
  endif

  " If the previous line was a stop-execution statement...
  if getline(plnum) =~ '^\s*\(break\|continue\|raise\|return\|pass\)\>'
    " See if the user has already dedented
    if indent(a:lnum) > indent(plnum) - &sw
      " If not, recommend one dedent
      return indent(plnum) - &sw
    endif
    " Otherwise, trust the user
    return -1
  endif

  " If the current line begins with a keyword that lines up with "try"
  if getline(a:lnum) =~ '^\s*\(except\|finally\)\>'
    let lnum = a:lnum - 1
    while lnum >= 1
      if getline(lnum) =~ '^\s*\(try\|except\)\>'
	let ind = indent(lnum)
	if ind >= indent(a:lnum)
	  return -1	" indent is already less than this
	endif
	return ind	" line up with previous try or except
      endif
      let lnum = lnum - 1
    endwhile
    return -1		" no matching "try"!
  endif

  " If the current line begins with a header keyword, dedent
  if getline(a:lnum) =~ '^\s*\(elif\|else\)\>'

    " Unless the previous line was a one-liner
    if getline(plnumstart) =~ '^\s*\(for\|if\|try\)\>'
      return plindent
    endif

    " Or the user has already dedented
    if indent(a:lnum) <= plindent - &sw
      return -1
    endif

    return plindent - &sw
  endif

  " JMC
  " If the previous line is a continuation
  " make the next line line up, but only for new lines
"let curline = getline(a:lnum)
  if (g:linecont == 1) 
    if (getline(a:lnum) =~ '^\s*$')
      return plindent
  endif
endif

  " JMC commented
  " When after a () construct we probably want to go back to the start line.
  " a = (b
  "       + c)
  " here
  "if parlnum > 0
    "return plindent
  "endif

  return -1

endfunction

" vim:sw=2