← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/revisiontree.py

  • Committer: John Arbash Meinel
  • Date: 2009-06-16 13:57:14 UTC
  • mto: This revision was merged to the branch mainline in revision 4449.
  • Revision ID: john@arbash-meinel.com-20090616135714-8o7jdtqqsfuv914z
The new code removes a get_parent_map call.

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):
116
114
    def has_filename(self, filename):
117
115
        return bool(self.inventory.path2id(filename))
118
116
 
119
 
    def list_files(self, include_root=False, from_dir=None, recursive=True):
 
117
    def list_files(self, include_root=False):
120
118
        # The only files returned by this are those from the version
121
 
        inv = self.inventory
122
 
        if from_dir is None:
123
 
            from_dir_id = None
124
 
        else:
125
 
            from_dir_id = inv.path2id(from_dir)
126
 
            if from_dir_id is None:
127
 
                # Directory not versioned
128
 
                return
129
 
        entries = inv.iter_entries(from_dir=from_dir_id, recursive=recursive)
130
 
        if inv.root is not None and not include_root and from_dir is None:
 
119
        entries = self.inventory.iter_entries()
 
120
        # skip the root for compatability with the current apis.
 
121
        if self.inventory.root is not None and not include_root:
131
122
            # skip the root for compatability with the current apis.
132
123
            entries.next()
133
124
        for path, entry in entries:
174
165
        return set(self._repository.get_ancestry(self._revision_id,
175
166
                                                 topo_sorted=False))
176
167
 
177
 
    def is_locked(self):
178
 
        return self._repository.is_locked()
179
 
 
180
168
    def lock_read(self):
181
169
        self._repository.lock_read()
182
 
        return self
183
170
 
184
171
    def __repr__(self):
185
172
        return '<%s instance at %x, rev_id=%r>' % (
247
234
        lookup_trees = [self.source]
248
235
        if extra_trees:
249
236
             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
237
        if specific_files == []:
254
238
            specific_file_ids = []
255
239
        else:
256
240
            specific_file_ids = self.target.paths2ids(specific_files,
257
241
                lookup_trees, require_versioned=require_versioned)
 
242
 
258
243
        # FIXME: It should be possible to delegate include_unchanged handling
259
244
        # to CHKInventory.iter_changes and do a better job there -- vila
260
245
        # 20090304
261
 
        changed_file_ids = set()
 
246
        if include_unchanged:
 
247
            changed_file_ids = []
262
248
        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)
 
249
            if (specific_file_ids is not None
 
250
                and not result[0] in specific_file_ids):
 
251
                # CHKMap.iter_changes is clean and fast. Better filter out
 
252
                # the specific files *after* it did its job.
 
253
                continue
273
254
            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
 
255
            if include_unchanged:
 
256
                # Keep track of yielded results (cheaper than building the
 
257
                # whole inventory).
 
258
                changed_file_ids.append(result[0])
279
259
        if include_unchanged:
280
260
            # CHKMap avoid being O(tree), so we go to O(tree) only if
281
261
            # required to.