if !has('python')

finish

endif

if !exists("g:vimux_pyutils_use_tslime")

let g:vimux_pyutils_use_tslime=0

endif

python << endpython

import re

# http://stackoverflow.com/questions/2695443/can-you-access-registers-from-python-functions-in-vim

def set_register(reg, value):

vim.command("let @%s='%s'" % (reg, value.replace("'","''")))

def send_to_tmux(lines):

# Global variable can be used to switch between vimux and tslime

if vim.eval("g:vimux_pyutils_use_tslime") == "1":

vim.command(':call Send_to_Tmux("\%cpaste\n")')

# Send by chunk of 20 lines

lines_step = 20

for i in xrange(0, len(lines), lines_step):

lines_chunk = lines[i:i+lines_step]

l = "\n".join(lines_chunk)

l = l.replace('\\', '\\\\')

l = l.replace('"', '\\"')

vim.command("echo 'sending to tslime length : %i'"%len(l))

vim.command(':call Send_to_Tmux("%s\\n")' % l)

# TODO: This is needed on some systems (OSX) to avoid getting scrambled

# output. But this is a bit of an ugly fix

# Sleep long enough to let the cpaste finish

vim.command(':sleep 200m')

vim.command(':call Send_to_Tmux("\n--\n")')

else:

l = "\n".join(lines)

set_register('+', l)

vim.command(':call VimuxRunCommand("%paste\n", 0)')

def run_tmux_python_chunk():

"""

Will copy/paste the currently selected block into the tmux split.

The code is unindented so the first selected line has 0 indentation

So you can select a statement from inside a function and it will run

without python complaining about indentation.

"""

r = vim.current.range

#vim.command("echo 'Range : %i %i'" % (r.start, r.end))

# Count indentation on first selected line

firstline = vim.current.buffer[r.start]

nindent = 0

for i in xrange(0, len(firstline)):

if firstline[i] == ' ':

nindent += 1

else:

break

# vim.command("echo '%i'" % nindent)

# Shift the whole text by nindent spaces (so the first line has 0 indent)

lines = vim.current.buffer[r.start:r.end+1]

if nindent > 0:

pat = '\s'*nindent

lines = [re.sub('^%s'%pat, '', l) for l in lines]

# Add empty newline at the end

lines.append('\n\n')

send_to_tmux(lines)

# Now, there are multiple solutions to copy that to tmux

# 1. With cpaste

#vim.command(':call VimuxRunCommand("%cpaste\n", 0)')

#vim.command(':call VimuxRunCommand("%s", 0)' % lines)

#vim.command(':call VimuxRunCommand("\n--\n", 0)')

# 2. With cpaste (better, only one command, but rely on system clipboard)

# Move cursor to the end of the selection

vim.current.window.cursor=(r.end+1, 0)

def run_tmux_python_cell(restore_cursor=False):

"""

This is to emulate MATLAB's cell mode

Cells are delimited by ##. Note that there should be a ## at the end of the

file

The :?##?;/##/ part creates a range with the following

?##? search backwards for ##

Then ';' starts the range from the result of the previous search (##)

/##/ End the range at the next ##

See the doce on 'ex ranges' here :

http://tnerual.eriogerg.free.fr/vimqrc.html

Then, we simply call run_tmux_python_chunk that will run the range

of the current buffer

"""

if restore_cursor:

# Save cursor position

(row, col) = vim.current.window.cursor

# Run chunk on cell range

vim.command(':?##?;/##/ :python run_tmux_python_chunk()')

if restore_cursor:

# Restore cursor position

vim.current.window.cursor = (row, col)

endpython

vmap <silent> <C-c> :python run_tmux_python_chunk()<CR>

noremap <silent> <C-b> :python run_tmux_python_cell(False)<CR>

noremap <silent> <C-g> :python run_tmux_python_cell(True)<CR>