def make_nodes(self):
# If the current changedtick is unchanged, we don't need to do
# anything:
if not self.is_outdated():
return self.nodes_made
self._check_version_location()
target_f = util.vim().eval('g:mundo_target_f')
ut = util.vim().eval('undotree()')
entries = ut['entries']
seq_last = ut['seq_last']
current_changedtick = util.vim().eval('b:changedtick')
root = Node(0, None, False, 0, 0)
nodes = []
# TODO only compute new values (not all values)
self._make_nodes(entries, nodes, root)
nodes.append(root)
nmap = dict((node.n, node) for node in nodes)
# cache values for later use
self.target_f = target_f
self.seq_last = seq_last
self.nodes_made = (nodes, nmap)
self.changedtick = current_changedtick
return self.nodes_made
def make_nodes(self):
# If the current changedtick is unchanged, we don't need to do
# anything:
if not self.is_outdated():
return self.nodes_made
self._check_version_location()
target_f = util.vim().eval('g:mundo_target_f')
ut = util.vim().eval('undotree()')
entries = ut['entries']
seq_last = ut['seq_last']
current_changedtick = util.vim().eval('b:changedtick')
root = Node(0, None, False, 0, 0)
nodes = []
# TODO only compute new values (not all values)
self._make_nodes(entries, nodes, root)
nodes.append(root)
nmap = dict((node.n, node) for node in nodes)
# cache values for later use
self.target_f = target_f
self.seq_last = seq_last
self.nodes_made = (nodes, nmap)
self.changedtick = current_changedtick
return self.nodes_made
def _get_lines(self, node):
n = 0
if node:
n = node.n
if n not in self.lines:
util._undo_to(n)
self.lines[n] = util.vim().current.buffer[:]
return self.lines[n]
def _get_lines(self,node):
n = 0
if node:
n = node.n
if n not in self.lines:
util._undo_to(n)
self.lines[n] = util.vim().current.buffer[:]
return self.lines[n]
def current(self):
""" Return the number of the current change. """
self._check_version_location()
nodes, nmap = self.make_nodes()
_curhead_l = list(itertools.dropwhile(lambda n: not n.curhead, nodes))
if _curhead_l:
current = _curhead_l[0].parent.n
else:
current = int(util.vim().eval('changenr()'))
return current
def current(self):
""" Return the number of the current change. """
self._check_version_location()
nodes, nmap = self.make_nodes()
_curhead_l = list(itertools.dropwhile(lambda n: not n.curhead, nodes))
if _curhead_l:
current = _curhead_l[0].parent.n
else:
current = int(util.vim().eval('changenr()'))
return current
def _validate_cache(self):
""" Checks if the targeted buffer number has changed, and if so
clears all cached data and stores the new target buffer number.
"""
target_n = int(util.vim().eval('g:mundo_target_n'))
if self.target_n != target_n:
self.clear_cache()
self.clear_oneline_diffs()
self.target_n = target_n
def make_nodes(self):
# Clear cache if it is invalid
self._validate_cache()
# If the current changedtick is unchanged, we don't need to do
# anything:
if not self.is_outdated():
return self.nodes_made
ut = util.vim().eval('undotree()')
# TODO only compute new values (not all values)
nodes = []
root = Node(0, None, False, 0, 0)
self._make_nodes(ut['entries'], nodes, root)
nodes.append(root)
nmap = dict((node.n, node) for node in nodes)
# cache values for later use
self.seq_last = ut['seq_last']
self.nodes_made = (nodes, nmap)
self.changedtick = util.vim().eval('b:changedtick')
return self.nodes_made
def change_preview_diff(self, before, after):
self._check_version_location()
key = "%s-%s-cpd" % (before.n, after.n)
if key in self.diffs:
return self.diffs[key]
util._goto_window_for_buffer(util.vim().eval('g:mundo_target_n'))
before_lines = self._get_lines(before)
after_lines = self._get_lines(after)
before_name = str(before.n or 'Original')
before_time = before.time and self._fmt_time(before.time) or ''
after_name = str(after.n or 'Original')
after_time = after.time and self._fmt_time(after.time) or ''
util._undo_to(self.current())
self.diffs[key] = list(
difflib.unified_diff(before_lines, after_lines, before_name,
after_name, before_time, after_time))
return self.diffs[key]
def change_preview_diff(self,before,after):
self._check_version_location()
key = "%s-%s-cpd"%(before.n,after.n)
if key in self.diffs:
return self.diffs[key]
util._goto_window_for_buffer(util.vim().eval('g:mundo_target_n'))
before_lines = self._get_lines(before)
after_lines = self._get_lines(after)
before_name = str(before.n or 'Original')
before_time = before.time and self._fmt_time(before.time) or ''
after_name = str(after.n or 'Original')
after_time = after.time and self._fmt_time(after.time) or ''
util._undo_to(self.current())
self.diffs[key] = list(difflib.unified_diff(before_lines, after_lines,
before_name, after_name,
before_time, after_time))
return self.diffs[key]
def is_outdated(self):
util._goto_window_for_buffer(util.vim().eval('g:mundo_target_n'))
current_changedtick = util.vim().eval('b:changedtick')
return self.changedtick != current_changedtick
def _check_version_location(self):
util._goto_window_for_buffer(util.vim().eval('g:mundo_target_n'))
target_f = util.vim().eval('g:mundo_target_f')
if target_f != self.target_f:
self._clear_cache()
def preview_diff(self, before, after, unified=True, inline=False):
"""
Generate a diff comparing two versions of a file.
Parameters:
current - ?
before
after
unified - If True, generate a unified diff
inline - Generate a one line summary line.
"""
self._check_version_location()
bn = 0
an = 0
if not after.n: # we're at the original file
pass
elif not before.n: # we're at a pseudo-root state
an = after.n
else:
bn = before.n
an = after.n
key = "%s-%s-pd-%s" % (bn, an, unified)
needs_oneline = inline and key not in self.diff_has_oneline
if key in self.diffs and not needs_oneline:
return self.diffs[key]
if not after.n: # we're at the original file
before_lines = []
after_lines = self._get_lines(None)
before_name = 'n/a'
before_time = ''
after_name = 'Original'
after_time = ''
elif not before.n: # we're at a pseudo-root state
before_lines = self._get_lines(None)
after_lines = self._get_lines(after)
before_name = 'Original'
before_time = ''
after_name = str(after.n)
after_time = self._fmt_time(after.time)
else:
before_lines = self._get_lines(before)
after_lines = self._get_lines(after)
before_name = str(before.n)
before_time = self._fmt_time(before.time)
after_name = str(after.n)
after_time = self._fmt_time(after.time)
if unified:
self.diffs[key] = list(
difflib.unified_diff(before_lines, after_lines, before_name,
after_name, before_time, after_time))
elif inline:
maxwidth = int(util.vim().eval("winwidth(0)"))
self.diffs[key] = diff.one_line_diff_str('\n'.join(before_lines),
'\n'.join(after_lines),
maxwidth)
self.diff_has_oneline[key] = True
else:
self.diffs[key] = ""
return self.diffs[key]
def is_outdated(self):
util._goto_window_for_buffer(util.vim().eval("g:gundo_target_n"))
current_changedtick = util.vim().eval("b:changedtick")
return self.changedtick != current_changedtick
def is_outdated(self):
""" Checks if the target buffer undo tree has changed since the last
update. Note that this moves to the target buffer.
"""
util._goto_window_for_buffer(int(util.vim().eval('g:mundo_target_n')))
return self.changedtick != util.vim().eval('b:changedtick')
def is_outdated(self):
util._goto_window_for_buffer(util.vim().eval('g:mundo_target_n'))
current_changedtick = util.vim().eval('b:changedtick')
return self.changedtick != current_changedtick
def _check_version_location(self):
util._goto_window_for_buffer(util.vim().eval('g:mundo_target_n'))
target_f = util.vim().eval('g:mundo_target_f')
if target_f != self.target_f:
self._clear_cache()
def preview_diff(self, before, after, unified=True, inline=False):
"""
Generate a diff comparing two versions of a file.
Parameters:
current - ?
before
after
unified - If True, generate a unified diff
inline - Generate a one line summary line.
"""
self._check_version_location()
bn = 0
an = 0
if not after.n: # we're at the original file
pass
elif not before.n: # we're at a pseudo-root state
an = after.n
else:
bn = before.n
an = after.n
key = "%s-%s-pd-%s"%(bn,an,unified)
needs_oneline = inline and key not in self.diff_has_oneline
if key in self.diffs and not needs_oneline:
return self.diffs[key]
if not after.n: # we're at the original file
before_lines = []
after_lines = self._get_lines(None)
before_name = 'n/a'
before_time = ''
after_name = 'Original'
after_time = ''
elif not before.n: # we're at a pseudo-root state
before_lines = self._get_lines(None)
after_lines = self._get_lines(after)
before_name = 'Original'
before_time = ''
after_name = str(after.n)
after_time = self._fmt_time(after.time)
else:
before_lines = self._get_lines(before)
after_lines = self._get_lines(after)
before_name = str(before.n)
before_time = self._fmt_time(before.time)
after_name = str(after.n)
after_time = self._fmt_time(after.time)
if unified:
self.diffs[key] = list(difflib.unified_diff(before_lines, after_lines,
before_name, after_name,
before_time, after_time))
elif inline:
maxwidth = int(util.vim().eval("winwidth(0)"))
self.diffs[key] = diff.one_line_diff_str('\n'.join(before_lines),'\n'.join(after_lines),maxwidth)
self.diff_has_oneline[key] = True
else:
self.diffs[key] = ""
return self.diffs[key]
