/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/revision.py

  • Committer: Martin Pool
  • Date: 2005-09-16 09:56:24 UTC
  • Revision ID: mbp@sourcefrog.net-20050916095623-ca0dff452934f21f
- make progress bar more tolerant of out-of-range values

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# (C) 2005 Canonical
 
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 as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
 
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
 
 
13
# You should have received a copy of the GNU General Public License
 
14
# along with this program; if not, write to the Free Software
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
 
 
18
import bzrlib.errors
 
19
 
 
20
 
 
21
class RevisionReference(object):
 
22
    """
 
23
    Reference to a stored revision.
 
24
 
 
25
    Includes the revision_id and revision_sha1.
 
26
    """
 
27
    revision_id = None
 
28
    revision_sha1 = None
 
29
    def __init__(self, revision_id, revision_sha1=None):
 
30
        if revision_id == None \
 
31
           or isinstance(revision_id, basestring):
 
32
            self.revision_id = revision_id
 
33
        else:
 
34
            raise ValueError('bad revision_id %r' % revision_id)
 
35
 
 
36
        if revision_sha1 != None:
 
37
            if isinstance(revision_sha1, basestring) \
 
38
               and len(revision_sha1) == 40:
 
39
                self.revision_sha1 = revision_sha1
 
40
            else:
 
41
                raise ValueError('bad revision_sha1 %r' % revision_sha1)
 
42
                
 
43
 
 
44
 
 
45
class Revision(object):
 
46
    """Single revision on a branch.
 
47
 
 
48
    Revisions may know their revision_hash, but only once they've been
 
49
    written out.  This is not stored because you cannot write the hash
 
50
    into the file it describes.
 
51
 
 
52
    After bzr 0.0.5 revisions are allowed to have multiple parents.
 
53
 
 
54
    parents
 
55
        List of parent revisions, each is a RevisionReference.
 
56
    """
 
57
    inventory_id = None
 
58
    inventory_sha1 = None
 
59
    revision_id = None
 
60
    timestamp = None
 
61
    message = None
 
62
    timezone = None
 
63
    committer = None
 
64
    
 
65
    def __init__(self, **args):
 
66
        self.__dict__.update(args)
 
67
        self.parents = []
 
68
 
 
69
 
 
70
    def __repr__(self):
 
71
        return "<Revision id %s>" % self.revision_id
 
72
 
 
73
    def __eq__(self, other):
 
74
        if not isinstance(other, Revision):
 
75
            return False
 
76
        return (self.inventory_id == other.inventory_id
 
77
                and self.inventory_sha1 == other.inventory_sha1
 
78
                and self.revision_id == other.revision_id
 
79
                and self.timestamp == other.timestamp
 
80
                and self.message == other.message
 
81
                and self.timezone == other.timezone
 
82
                and self.committer == other.committer)
 
83
 
 
84
    def __ne__(self, other):
 
85
        return not self.__eq__(other)
 
86
 
 
87
        
 
88
 
 
89
REVISION_ID_RE = None
 
90
 
 
91
def validate_revision_id(rid):
 
92
    """Check rid is syntactically valid for a revision id."""
 
93
    global REVISION_ID_RE
 
94
    if not REVISION_ID_RE:
 
95
        import re
 
96
        REVISION_ID_RE = re.compile('[\w.-]+@[\w.-]+--?\d+--?[0-9a-f]+\Z')
 
97
 
 
98
    if not REVISION_ID_RE.match(rid):
 
99
        raise ValueError("malformed revision-id %r" % rid)
 
100
 
 
101
 
 
102
def is_ancestor(revision_id, candidate_id, branch):
 
103
    """Return true if candidate_id is an ancestor of revision_id.
 
104
 
 
105
    A false negative will be returned if any intermediate descendent of
 
106
    candidate_id is not present in any of the revision_sources.
 
107
    
 
108
    revisions_source is an object supporting a get_revision operation that
 
109
    behaves like Branch's.
 
110
    """
 
111
    return candidate_id in branch.get_ancestry(revision_id)
 
112
 
 
113
 
 
114
def iter_ancestors(revision_id, revision_source, only_present=False):
 
115
    ancestors = (revision_id,)
 
116
    distance = 0
 
117
    while len(ancestors) > 0:
 
118
        new_ancestors = []
 
119
        for ancestor in ancestors:
 
120
            if not only_present:
 
121
                yield ancestor, distance
 
122
            try:
 
123
                revision = revision_source.get_revision(ancestor)
 
124
            except bzrlib.errors.NoSuchRevision, e:
 
125
                if e.revision == revision_id:
 
126
                    raise 
 
127
                else:
 
128
                    continue
 
129
            if only_present:
 
130
                yield ancestor, distance
 
131
            new_ancestors.extend([p.revision_id for p in revision.parents])
 
132
        ancestors = new_ancestors
 
133
        distance += 1
 
134
 
 
135
 
 
136
def find_present_ancestors(revision_id, revision_source):
 
137
    """Return the ancestors of a revision present in a branch.
 
138
 
 
139
    It's possible that a branch won't have the complete ancestry of
 
140
    one of its revisions.  
 
141
 
 
142
    """
 
143
    found_ancestors = {}
 
144
    anc_iter = enumerate(iter_ancestors(revision_id, revision_source,
 
145
                         only_present=True))
 
146
    for anc_order, (anc_id, anc_distance) in anc_iter:
 
147
        if not found_ancestors.has_key(anc_id):
 
148
            found_ancestors[anc_id] = (anc_order, anc_distance)
 
149
    return found_ancestors
 
150
    
 
151
 
 
152
def __get_closest(intersection):
 
153
    intersection.sort()
 
154
    matches = [] 
 
155
    for entry in intersection:
 
156
        if entry[0] == intersection[0][0]:
 
157
            matches.append(entry[2])
 
158
    return matches
 
159
 
 
160
 
 
161
def common_ancestor(revision_a, revision_b, revision_source):
 
162
    """Find the ancestor common to both revisions that is closest to both.
 
163
    """
 
164
    from bzrlib.trace import mutter
 
165
    a_ancestors = find_present_ancestors(revision_a, revision_source)
 
166
    b_ancestors = find_present_ancestors(revision_b, revision_source)
 
167
    a_intersection = []
 
168
    b_intersection = []
 
169
    # a_order is used as a tie-breaker when two equally-good bases are found
 
170
    for revision, (a_order, a_distance) in a_ancestors.iteritems():
 
171
        if b_ancestors.has_key(revision):
 
172
            a_intersection.append((a_distance, a_order, revision))
 
173
            b_intersection.append((b_ancestors[revision][1], a_order, revision))
 
174
    mutter("a intersection: %r" % a_intersection)
 
175
    mutter("b intersection: %r" % b_intersection)
 
176
 
 
177
    a_closest = __get_closest(a_intersection)
 
178
    if len(a_closest) == 0:
 
179
        return None
 
180
    b_closest = __get_closest(b_intersection)
 
181
    assert len(b_closest) != 0
 
182
    mutter ("a_closest %r" % a_closest)
 
183
    mutter ("b_closest %r" % b_closest)
 
184
    if a_closest[0] in b_closest:
 
185
        return a_closest[0]
 
186
    elif b_closest[0] in a_closest:
 
187
        return b_closest[0]
 
188
    else:
 
189
        raise bzrlib.errors.AmbiguousBase((a_closest[0], b_closest[0]))
 
190
    return a_closest[0]
 
191
 
 
192
class MultipleRevisionSources(object):
 
193
    """Proxy that looks in multiple branches for revisions."""
 
194
    def __init__(self, *args):
 
195
        object.__init__(self)
 
196
        assert len(args) != 0
 
197
        self._revision_sources = args
 
198
 
 
199
    def get_revision(self, revision_id):
 
200
        for source in self._revision_sources:
 
201
            try:
 
202
                return source.get_revision(revision_id)
 
203
            except bzrlib.errors.NoSuchRevision, e:
 
204
                pass
 
205
        raise e
 
206
 
 
207
def get_intervening_revisions(ancestor_id, rev_id, rev_source, 
 
208
                              revision_history=None):
 
209
    """Find the longest line of descent from maybe_ancestor to revision.
 
210
    Revision history is followed where possible.
 
211
 
 
212
    If ancestor_id == rev_id, list will be empty.
 
213
    Otherwise, rev_id will be the last entry.  ancestor_id will never appear.
 
214
    If ancestor_id is not an ancestor, NotAncestor will be thrown
 
215
    """
 
216
    [rev_source.get_revision(r) for r in (ancestor_id, rev_id)]
 
217
    if ancestor_id == rev_id:
 
218
        return []
 
219
    def historical_lines(line):
 
220
        """Return a tuple of historical/non_historical lines, for sorting.
 
221
        The non_historical count is negative, since non_historical lines are
 
222
        a bad thing.
 
223
        """
 
224
        good_count = 0
 
225
        bad_count = 0
 
226
        for revision in line:
 
227
            if revision in revision_history:
 
228
                good_count += 1
 
229
            else:
 
230
                bad_count -= 1
 
231
        return good_count, bad_count
 
232
    active = [[rev_id]]
 
233
    successful_lines = []
 
234
    while len(active) > 0:
 
235
        new_active = []
 
236
        for line in active:
 
237
            parent_ids = [p.revision_id for p in 
 
238
                          rev_source.get_revision(line[-1]).parents]
 
239
            for parent in parent_ids:
 
240
                line_copy = line[:]
 
241
                if parent == ancestor_id:
 
242
                    successful_lines.append(line_copy)
 
243
                else:
 
244
                    line_copy.append(parent)
 
245
                    new_active.append(line_copy)
 
246
        active = new_active
 
247
    if len(successful_lines) == 0:
 
248
        raise bzrlib.errors.NotAncestor(rev_id, ancestor_id)
 
249
    for line in successful_lines:
 
250
        line.reverse()
 
251
    if revision_history is not None:
 
252
        by_historical_lines = []
 
253
        for line in successful_lines:
 
254
            count = historical_lines(line)
 
255
            by_historical_lines.append((count, line))
 
256
        by_historical_lines.sort()
 
257
        if by_historical_lines[-1][0][0] > 0:
 
258
            return by_historical_lines[-1][1]
 
259
    assert len(successful_lines)
 
260
    successful_lines.sort(cmp, len)
 
261
    return successful_lines[-1]