← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/revisiontree.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2009-06-18 20:25:52 UTC
  • mfrom: (4413.5.15 1.16-chk-direct)
  • Revision ID: pqm@pqm.ubuntu.com-20090618202552-xyl6tcvbxtm8bupf
(jam) Improve initial commit performance by creating a CHKMap in bulk,
        rather than via O(tree) map() calls.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/revisiontree.py
1
 
# Copyright (C) 2006-2010 Canonical Ltd
 
1
# Copyright (C) 2005, 2007 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
20
20
 
21
21
from bzrlib import (
22
22
    errors,
 
23
    osutils,
23
24
    revision,
 
25
    symbol_versioning,
24
26
    tree,
25
27
    )
26
28
 
85
87
                      default_revision=revision.CURRENT_REVISION):
86
88
        """See Tree.annotate_iter"""
87
89
        text_key = (file_id, self.inventory[file_id].revision)
88
 
        annotator = self._repository.texts.get_annotator()
89
 
        annotations = annotator.annotate_flat(text_key)
 
90
        annotations = self._repository.texts.annotate(text_key)
90
91
        return [(key[-1], line) for key, line in annotations]
91
92
 
92
93
    def get_file_size(self, file_id):
101
102
 
102
103
    def get_file_mtime(self, file_id, path=None):
103
104
        ie = self._inventory[file_id]
104
 
        try:
105
 
            revision = self._repository.get_revision(ie.revision)
106
 
        except errors.NoSuchRevision:
107
 
            raise errors.FileTimestampUnavailable(self.id2path(file_id))
 
105
        revision = self._repository.get_revision(ie.revision)
108
106
        return revision.timestamp
109
107
 
110
108
    def is_executable(self, file_id, path=None):
174
172
        return set(self._repository.get_ancestry(self._revision_id,
175
173
                                                 topo_sorted=False))
176
174
 
177
 
    def is_locked(self):
178
 
        return self._repository.is_locked()
179
 
 
180
175
    def lock_read(self):
181
176
        self._repository.lock_read()
182
 
        return self
183
177
 
184
178
    def __repr__(self):
185
179
        return '<%s instance at %x, rev_id=%r>' % (
247
241
        lookup_trees = [self.source]
248
242
        if extra_trees:
249
243
             lookup_trees.extend(extra_trees)
250
 
        # The ids of items we need to examine to insure delta consistency.
251
 
        precise_file_ids = set()
252
 
        discarded_changes = {}
253
244
        if specific_files == []:
254
245
            specific_file_ids = []
255
246
        else:
256
247
            specific_file_ids = self.target.paths2ids(specific_files,
257
248
                lookup_trees, require_versioned=require_versioned)
 
249
 
258
250
        # FIXME: It should be possible to delegate include_unchanged handling
259
251
        # to CHKInventory.iter_changes and do a better job there -- vila
260
252
        # 20090304
261
 
        changed_file_ids = set()
 
253
        if include_unchanged:
 
254
            changed_file_ids = []
262
255
        for result in self.target.inventory.iter_changes(self.source.inventory):
263
 
            if specific_file_ids is not None:
264
 
                file_id = result[0]
265
 
                if file_id not in specific_file_ids:
266
 
                    # A change from the whole tree that we don't want to show yet.
267
 
                    # We may find that we need to show it for delta consistency, so
268
 
                    # stash it.
269
 
                    discarded_changes[result[0]] = result
270
 
                    continue
271
 
                new_parent_id = result[4][1]
272
 
                precise_file_ids.add(new_parent_id)
 
256
            if (specific_file_ids is not None
 
257
                and not result[0] in specific_file_ids):
 
258
                # CHKMap.iter_changes is clean and fast. Better filter out
 
259
                # the specific files *after* it did its job.
 
260
                continue
273
261
            yield result
274
 
            changed_file_ids.add(result[0])
275
 
        if specific_file_ids is not None:
276
 
            for result in self._handle_precise_ids(precise_file_ids,
277
 
                changed_file_ids, discarded_changes=discarded_changes):
278
 
                yield result
 
262
            if include_unchanged:
 
263
                # Keep track of yielded results (cheaper than building the
 
264
                # whole inventory).
 
265
                changed_file_ids.append(result[0])
279
266
        if include_unchanged:
280
267
            # CHKMap avoid being O(tree), so we go to O(tree) only if
281
268
            # required to.