/brz/remove-bazaar : contents of equivalence

: (revision 0.18.8)

To get this branch, use:

bzr branch
http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

# Copyright (C) 2008 Canonical Limited.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as published
# by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
#

"""Functions for dealing with a persistent equivalency table."""


SENTINEL = -1


class EquivalenceTable(object):
    """This class tracks equivalencies between lists of hashable objects.

    :ivar _left_lines: The 'static' lines that will be preserved between runs.
    :ival _matching_lines: A dict of {line:[matching offsets]}
    """

    def __init__(self, left_lines):
        self._left_lines = left_lines
        # For each line in 'left' give the offset to the other lines which
        # match it.
        self._generate_matching_left_lines()

    def _generate_matching_left_lines(self):
        matches = {}
        for idx, line in enumerate(self._left_lines):
            matches.setdefault(line, []).append(idx)
        self._matching_lines = matches

    def _update_matching_left_lines(self, new_lines, index):
        matches = self._matching_lines
        start_idx = len(self._left_lines)
        for idx, do_index in enumerate(index):
            if not do_index:
                continue
            matches.setdefault(new_lines[idx], []).append(start_idx + idx)

    def get_left_matches(self, line):
        """Return the lines which match the line in right."""
        try:
            return self._matching_lines[line]
        except KeyError:
            return None

    def extend_left_lines(self, lines, index):
        """Add more lines to the left-lines list.

        :param lines: A list of lines to add
        :param index: A True/False for each node to define if it should be
            indexed.
        """
        self._update_matching_left_lines(lines, index)
        self._left_lines.extend(lines)

0.18.1 by John Arbash Meinel Start working on an EquivalenceTable construct.	1	# Copyright (C) 2008 Canonical Limited.
	2	#
	3	# This program is free software; you can redistribute it and/or modify
	4	# it under the terms of the GNU General Public License version 2 as published
	5	# by the Free Software Foundation.
	6	#
	7	# This program is distributed in the hope that it will be useful,
	8	# but WITHOUT ANY WARRANTY; without even the implied warranty of
	9	# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	10	# GNU General Public License for more details.
	11	#
	12	# You should have received a copy of the GNU General Public License
	13	# along with this program; if not, write to the Free Software
	14	# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
	15	#
	16
	17	"""Functions for dealing with a persistent equivalency table."""
	18
	19
	20	SENTINEL = -1
	21
	22
	23	class EquivalenceTable(object):
	24	"""This class tracks equivalencies between lists of hashable objects.
	25
	26	:ivar _left_lines: The 'static' lines that will be preserved between runs.
0.18.8 by John Arbash Meinel We don't actually care much about the right-line relationships. We only need to track the left-hand side.	27	:ival _matching_lines: A dict of {line:[matching offsets]}
0.18.1 by John Arbash Meinel Start working on an EquivalenceTable construct.	28	"""
	29
	30	def __init__(self, left_lines):
	31	self._left_lines = left_lines
	32	# For each line in 'left' give the offset to the other lines which
	33	# match it.
	34	self._generate_matching_left_lines()
	35
	36	def _generate_matching_left_lines(self):
	37	matches = {}
	38	for idx, line in enumerate(self._left_lines):
0.18.8 by John Arbash Meinel We don't actually care much about the right-line relationships. We only need to track the left-hand side.	39	matches.setdefault(line, []).append(idx)
0.18.1 by John Arbash Meinel Start working on an EquivalenceTable construct.	40	self._matching_lines = matches
	41
0.18.4 by John Arbash Meinel Allow ignoring some of the new lines.	42	def _update_matching_left_lines(self, new_lines, index):
0.18.3 by John Arbash Meinel We can now add more lines to left lines, and continue to track the right info.	43	matches = self._matching_lines
	44	start_idx = len(self._left_lines)
0.18.8 by John Arbash Meinel We don't actually care much about the right-line relationships. We only need to track the left-hand side.	45	for idx, do_index in enumerate(index):
0.18.4 by John Arbash Meinel Allow ignoring some of the new lines.	46	if not do_index:
	47	continue
0.18.8 by John Arbash Meinel We don't actually care much about the right-line relationships. We only need to track the left-hand side.	48	matches.setdefault(new_lines[idx], []).append(start_idx + idx)
	49
	50	def get_left_matches(self, line):
0.18.2 by John Arbash Meinel we can now extract what lines in left match the right	51	"""Return the lines which match the line in right."""
0.18.8 by John Arbash Meinel We don't actually care much about the right-line relationships. We only need to track the left-hand side.	52	try:
	53	return self._matching_lines[line]
	54	except KeyError:
	55	return None
0.18.3 by John Arbash Meinel We can now add more lines to left lines, and continue to track the right info.	56
0.18.4 by John Arbash Meinel Allow ignoring some of the new lines.	57	def extend_left_lines(self, lines, index):
	58	"""Add more lines to the left-lines list.
	59
	60	:param lines: A list of lines to add
	61	:param index: A True/False for each node to define if it should be
	62	indexed.
	63	"""
	64	self._update_matching_left_lines(lines, index)
0.18.3 by John Arbash Meinel We can now add more lines to left lines, and continue to track the right info.	65	self._left_lines.extend(lines)