/brz/remove-bazaar

# Copyright (C) 2008 Canonical Ltd

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

"""Import processor that dump stats about the input (and doesn't import)."""

from bzrlib.trace import (

    note,

    warning,

    )

from bzrlib.plugins.fastimport import (

    commands,

    processor,

    )

# Maximum number of parents for a merge commit

_MAX_PARENTS = 16

class InfoProcessor(processor.ImportProcessor):

    """An import processor that dumps statistics about the input.

    No changes to the current repository are made.

    As well as providing useful information about an import

    stream before importing it, this processor is useful for

    benchmarking the speed at which data can be extracted from

    the source.

    """

    def __init__(self, target=None, params=None, verbose=False):

        # Allow creation without a target

        processor.ImportProcessor.__init__(self, target, params, verbose)

    def pre_process(self):

        # Init statistics

        self.cmd_counts = {}

        for cmd in commands.COMMAND_NAMES:

            self.cmd_counts[cmd] = 0

        self.file_cmd_counts = {}

        for fc in commands.FILE_COMMAND_NAMES:

            self.file_cmd_counts[fc] = 0

        self.parent_counts = {}

        for i in xrange(0, _MAX_PARENTS):

            self.parent_counts[i] = 0

        self.committers = set()

        self.separate_authors_found = False

        self.symlinks_found = False

        self.executables_found = False

        self.sha_blob_references = False

        self.lightweight_tags = 0

        self.named_branches = []

        # Blob usage tracking

        self.blobs = {}

        for usage in ['new', 'used', 'multi', 'unknown']:

            self.blobs[usage] = set()

    def post_process(self):

        # Dump statistics

        if self.verbose:

            note('# Configuration file generated by bzr fast-import --info')

        cmd_names = commands.COMMAND_NAMES

        fc_names = commands.FILE_COMMAND_NAMES

        cmd_values = [self.cmd_counts[c] for c in cmd_names]

        fc_values = [self.file_cmd_counts[c] for c in fc_names]

        self._dump_stats_group("Command counts", cmd_names, cmd_values, str)

        self._dump_stats_group("File command counts", fc_names, fc_values, str)

        # Commit stats

        if self.cmd_counts['commit']:

            p_names = []

            p_values = []

            for i in xrange(0, _MAX_PARENTS):

                count = self.parent_counts[i]

                if count > 0:

                    p_names.append("parents-%d" % i)

                    p_values.append(count)

            flags = {

                'separate authors found': self.separate_authors_found,

                'executables': self.executables_found,

                'symlinks': self.symlinks_found,

                'blobs referenced by SHA': self.sha_blob_references,

                }

            self._dump_stats_group("Parent counts", p_names, p_values, str)

            self._dump_stats_group("Commit analysis", flags.keys(),

                flags.values(), _found)

            # note("\t%d\t%s" % (len(self.committers), 'unique committers'))

        # Blob stats

        if self.cmd_counts['blob']:

            # In verbose mode, don't list every blob used

            if self.verbose:

                del self.blobs['used']

            self._dump_stats_group("Blob usage tracking", self.blobs.keys(),

                self.blobs.values(), len)

        # Other stats

        if self.cmd_counts['reset']:

            reset_stats = {

                'lightweight tags': self.lightweight_tags,

                'other resets': self.named_branches,

                }

            self._dump_stats_group("Reset analysis", reset_stats.keys(),

                reset_stats.values())

    def _dump_stats_group(self, title, names, values, formatter):

        """Dump a statistics group.

        In verbose mode, do so as a config file so

        that other processors can load the information if they want to.

        """

        if self.verbose:

            print "[%s]" % (title,)

            for name, value in zip(names, values):

                print "%s = %s" % (name.replace(' ', '-'),value)

            print ""

        else:

            print "%s:" % (title,)

            for name, value in zip(names, values):

                if formatter is not None:

                    value = formatter(value)

                print "\t%s\t%s" % (value,name)

    def progress_handler(self, cmd):

        """Process a ProgressCommand."""

        self.cmd_counts[cmd.name] += 1

    def blob_handler(self, cmd):

        """Process a BlobCommand."""

        self.cmd_counts[cmd.name] += 1

        self.blobs['new'].add(cmd.mark)

    def checkpoint_handler(self, cmd):

        """Process a CheckpointCommand."""

        self.cmd_counts[cmd.name] += 1

    def commit_handler(self, cmd):

        """Process a CommitCommand."""

        self.cmd_counts[cmd.name] += 1

        self.parent_counts[len(cmd.parents)] += 1

        self.committers.add(cmd.committer)

        if cmd.author is not None:

            self.separate_authors_found = True

        for fc in cmd.file_iter():

            self.file_cmd_counts[fc.name] += 1

            if isinstance(fc, commands.FileModifyCommand):

                if fc.is_executable:

                    self.executables_found = True

                if fc.kind == commands.SYMLINK_KIND:

                    self.symlinks_found = True

                if fc.dataref is not None:

                    if fc.dataref[0] == ':':

                        self._track_blob(fc.dataref[1:])

                    else:

                        self.sha_blob_references = True

    def reset_handler(self, cmd):

        """Process a ResetCommand."""

        self.cmd_counts[cmd.name] += 1

        if cmd.ref.startswith('refs/tags/'):

            self.lightweight_tags += 1

        else:

            self.named_branches.append(cmd.ref)

    def tag_handler(self, cmd):

        """Process a TagCommand."""

        self.cmd_counts[cmd.name] += 1

    def _track_blob(self, mark):

        if mark in self.blobs['multi']:

            pass

        elif mark in self.blobs['used']:

            self.blobs['multi'].add(mark)

            self.blobs['used'].remove(mark)

        elif mark in self.blobs['new']:

            self.blobs['used'].add(mark)

            self.blobs['new'].remove(mark)

        else:

            self.blobs['unknown'].add(mark)

def _found(b):

    """Format a found boolean as a string."""

    return ['no', 'found'][b]