/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/revisiontree.py

  • Committer: Robert Collins
  • Date: 2007-07-15 15:40:37 UTC
  • mto: (2592.3.33 repository)
  • mto: This revision was merged to the branch mainline in revision 2624.
  • Revision ID: robertc@robertcollins.net-20070715154037-3ar8g89decddc9su
Make GraphIndex accept nodes as key, value, references, so that the method
signature is closer to what a simple key->value index delivers. Also
change the behaviour when the reference list count is zero to accept
key, value as nodes, and emit key, value to make it identical in that case
to a simple key->value index. This may not be a good idea, but for now it
seems ok.

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
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
"""RevisionTree - a Tree implementation backed by repository data for a revision."""
18
18
 
19
19
from cStringIO import StringIO
20
20
 
21
21
from bzrlib import (
22
 
    errors,
 
22
    osutils,
23
23
    revision,
24
 
    tree,
25
24
    )
26
 
 
27
 
 
28
 
class RevisionTree(tree.Tree):
 
25
from bzrlib.tree import Tree
 
26
 
 
27
 
 
28
class RevisionTree(Tree):
29
29
    """Tree viewing a previous revision.
30
30
 
31
31
    File text can be retrieved from the text store.
32
32
    """
33
 
 
 
33
    
34
34
    def __init__(self, branch, inv, revision_id):
35
 
        # for compatability the 'branch' parameter has not been renamed to
 
35
        # for compatability the 'branch' parameter has not been renamed to 
36
36
        # repository at this point. However, we should change RevisionTree's
37
 
        # construction to always be via Repository and not via direct
 
37
        # construction to always be via Repository and not via direct 
38
38
        # construction - this will mean that we can change the constructor
39
39
        # with much less chance of breaking client code.
40
40
        self._repository = branch
 
41
        self._weave_store = branch.weave_store
41
42
        self._inventory = inv
42
 
        self._revision_id = revision_id
43
 
        self._rules_searcher = None
 
43
        self._revision_id = osutils.safe_revision_id(revision_id)
44
44
 
45
45
    def supports_tree_reference(self):
46
 
        return getattr(self._repository._format, "supports_tree_reference",
47
 
            False)
 
46
        return True
48
47
 
49
48
    def get_parent_ids(self):
50
49
        """See Tree.get_parent_ids.
57
56
            parent_ids = self._repository.get_revision(
58
57
                self._revision_id).parent_ids
59
58
        return parent_ids
60
 
 
 
59
        
61
60
    def get_revision_id(self):
62
61
        """Return the revision id associated with this tree."""
63
62
        return self._revision_id
64
63
 
65
 
    def get_file_text(self, file_id, path=None):
66
 
        _, content = list(self.iter_files_bytes([(file_id, None)]))[0]
67
 
        return ''.join(content)
68
 
 
69
 
    def get_file(self, file_id, path=None):
 
64
    def get_weave(self, file_id):
 
65
        file_id = osutils.safe_file_id(file_id)
 
66
        return self._weave_store.get_weave(file_id,
 
67
                self._repository.get_transaction())
 
68
 
 
69
    def get_file_lines(self, file_id):
 
70
        file_id = osutils.safe_file_id(file_id)
 
71
        ie = self._inventory[file_id]
 
72
        weave = self.get_weave(file_id)
 
73
        return weave.get_lines(ie.revision)
 
74
 
 
75
    def get_file_text(self, file_id):
 
76
        file_id = osutils.safe_file_id(file_id)
 
77
        return ''.join(self.get_file_lines(file_id))
 
78
 
 
79
    def get_file(self, file_id):
 
80
        file_id = osutils.safe_file_id(file_id)
70
81
        return StringIO(self.get_file_text(file_id))
71
82
 
72
 
    def iter_files_bytes(self, desired_files):
73
 
        """See Tree.iter_files_bytes.
74
 
 
75
 
        This version is implemented on top of Repository.extract_files_bytes"""
76
 
        repo_desired_files = [(f, self.inventory[f].revision, i)
77
 
                              for f, i in desired_files]
78
 
        try:
79
 
            for result in self._repository.iter_files_bytes(repo_desired_files):
80
 
                yield result
81
 
        except errors.RevisionNotPresent, e:
82
 
            raise errors.NoSuchFile(e.revision_id)
83
 
 
84
 
    def annotate_iter(self, file_id,
85
 
                      default_revision=revision.CURRENT_REVISION):
 
83
    def annotate_iter(self, file_id):
86
84
        """See Tree.annotate_iter"""
87
 
        text_key = (file_id, self.inventory[file_id].revision)
88
 
        annotator = self._repository.texts.get_annotator()
89
 
        annotations = annotator.annotate_flat(text_key)
90
 
        return [(key[-1], line) for key, line in annotations]
 
85
        file_id = osutils.safe_file_id(file_id)
 
86
        w = self.get_weave(file_id)
 
87
        return w.annotate_iter(self.inventory[file_id].revision)
91
88
 
92
89
    def get_file_size(self, file_id):
93
 
        """See Tree.get_file_size"""
 
90
        file_id = osutils.safe_file_id(file_id)
94
91
        return self._inventory[file_id].text_size
95
92
 
96
93
    def get_file_sha1(self, file_id, path=None, stat_value=None):
 
94
        file_id = osutils.safe_file_id(file_id)
97
95
        ie = self._inventory[file_id]
98
96
        if ie.kind == "file":
99
97
            return ie.text_sha1
100
98
        return None
101
99
 
102
100
    def get_file_mtime(self, file_id, path=None):
 
101
        file_id = osutils.safe_file_id(file_id)
103
102
        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))
 
103
        revision = self._repository.get_revision(ie.revision)
108
104
        return revision.timestamp
109
105
 
110
106
    def is_executable(self, file_id, path=None):
 
107
        file_id = osutils.safe_file_id(file_id)
111
108
        ie = self._inventory[file_id]
112
109
        if ie.kind != "file":
113
110
            return None
116
113
    def has_filename(self, filename):
117
114
        return bool(self.inventory.path2id(filename))
118
115
 
119
 
    def list_files(self, include_root=False, from_dir=None, recursive=True):
 
116
    def list_files(self, include_root=False):
120
117
        # 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:
 
118
        entries = self.inventory.iter_entries()
 
119
        # skip the root for compatability with the current apis.
 
120
        if self.inventory.root is not None and not include_root:
131
121
            # skip the root for compatability with the current apis.
132
122
            entries.next()
133
123
        for path, entry in entries:
134
124
            yield path, 'V', entry.kind, entry.file_id, entry
135
125
 
136
126
    def get_symlink_target(self, file_id):
 
127
        file_id = osutils.safe_file_id(file_id)
137
128
        ie = self._inventory[file_id]
138
 
        # Inventories store symlink targets in unicode
139
 
        return ie.symlink_target
 
129
        return ie.symlink_target;
140
130
 
141
131
    def get_reference_revision(self, file_id, path=None):
142
132
        return self.inventory[file_id].reference_revision
146
136
            return self.inventory.root.file_id
147
137
 
148
138
    def kind(self, file_id):
 
139
        file_id = osutils.safe_file_id(file_id)
149
140
        return self._inventory[file_id].kind
150
141
 
151
 
    def path_content_summary(self, path):
152
 
        """See Tree.path_content_summary."""
153
 
        id = self.inventory.path2id(path)
154
 
        if id is None:
155
 
            return ('missing', None, None, None)
156
 
        entry = self._inventory[id]
157
 
        kind = entry.kind
158
 
        if kind == 'file':
159
 
            return (kind, entry.text_size, entry.executable, entry.text_sha1)
160
 
        elif kind == 'symlink':
161
 
            return (kind, None, None, entry.symlink_target)
162
 
        else:
163
 
            return (kind, None, None, None)
164
 
 
165
142
    def _comparison_data(self, entry, path):
166
143
        if entry is None:
167
144
            return None, False, None
168
145
        return entry.kind, entry.executable, None
169
146
 
170
147
    def _file_size(self, entry, stat_value):
 
148
        assert entry.text_size is not None
171
149
        return entry.text_size
172
150
 
173
 
    def _get_ancestors(self, default_revision):
174
 
        return set(self._repository.get_ancestry(self._revision_id,
175
 
                                                 topo_sorted=False))
176
 
 
177
 
    def is_locked(self):
178
 
        return self._repository.is_locked()
179
 
 
180
151
    def lock_read(self):
181
152
        self._repository.lock_read()
182
 
        return self
183
153
 
184
154
    def __repr__(self):
185
155
        return '<%s instance at %x, rev_id=%r>' % (
216
186
            for dir in reversed(dirblock):
217
187
                if dir[2] == _directory:
218
188
                    pending.append(dir)
219
 
 
220
 
    def _get_rules_searcher(self, default_searcher):
221
 
        """See Tree._get_rules_searcher."""
222
 
        if self._rules_searcher is None:
223
 
            self._rules_searcher = super(RevisionTree,
224
 
                self)._get_rules_searcher(default_searcher)
225
 
        return self._rules_searcher
226
 
 
227
 
 
228
 
class InterCHKRevisionTree(tree.InterTree):
229
 
    """Fast path optimiser for RevisionTrees with CHK inventories."""
230
 
 
231
 
    @staticmethod
232
 
    def is_compatible(source, target):
233
 
        if (isinstance(source, RevisionTree)
234
 
            and isinstance(target, RevisionTree)):
235
 
            try:
236
 
                # Only CHK inventories have id_to_entry attribute
237
 
                source.inventory.id_to_entry
238
 
                target.inventory.id_to_entry
239
 
                return True
240
 
            except AttributeError:
241
 
                pass
242
 
        return False
243
 
 
244
 
    def iter_changes(self, include_unchanged=False,
245
 
                     specific_files=None, pb=None, extra_trees=[],
246
 
                     require_versioned=True, want_unversioned=False):
247
 
        lookup_trees = [self.source]
248
 
        if extra_trees:
249
 
             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
 
        if specific_files == []:
254
 
            specific_file_ids = []
255
 
        else:
256
 
            specific_file_ids = self.target.paths2ids(specific_files,
257
 
                lookup_trees, require_versioned=require_versioned)
258
 
        # FIXME: It should be possible to delegate include_unchanged handling
259
 
        # to CHKInventory.iter_changes and do a better job there -- vila
260
 
        # 20090304
261
 
        changed_file_ids = set()
262
 
        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)
273
 
            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
279
 
        if include_unchanged:
280
 
            # CHKMap avoid being O(tree), so we go to O(tree) only if
281
 
            # required to.
282
 
            # Now walk the whole inventory, excluding the already yielded
283
 
            # file ids
284
 
            changed_file_ids = set(changed_file_ids)
285
 
            for relpath, entry in self.target.inventory.iter_entries():
286
 
                if (specific_file_ids is not None
287
 
                    and not entry.file_id in specific_file_ids):
288
 
                    continue
289
 
                if not entry.file_id in changed_file_ids:
290
 
                    yield (entry.file_id,
291
 
                           (relpath, relpath), # Not renamed
292
 
                           False, # Not modified
293
 
                           (True, True), # Still  versioned
294
 
                           (entry.parent_id, entry.parent_id),
295
 
                           (entry.name, entry.name),
296
 
                           (entry.kind, entry.kind),
297
 
                           (entry.executable, entry.executable))
298
 
 
299
 
 
300
 
tree.InterTree.register_optimiser(InterCHKRevisionTree)