← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/merge3.py

  • Committer: John Arbash Meinel
  • Date: 2010-08-02 17:16:12 UTC
  • mto: This revision was merged to the branch mainline in revision 5369.
  • Revision ID: john@arbash-meinel.com-20100802171612-rdh5ods70w2bl3j7
We also have to re-implement it for _simple_set_pyx.pyx

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/merge3.py
1
 
# Copyright (C) 2004, 2005 Canonical Ltd
 
1
# Copyright (C) 2005-2010 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
 
18
18
# mbp: "you know that thing where cvs gives you conflict markers?"
19
19
# s: "i hate that."
20
20
 
21
 
 
22
 
from bzrlib.errors import CantReprocessAndShowBase
23
 
import bzrlib.patiencediff
24
 
from bzrlib.textfile import check_text_lines
 
21
from bzrlib import (
 
22
    errors,
 
23
    patiencediff,
 
24
    textfile,
 
25
    )
25
26
 
26
27
 
27
28
def intersect(ra, rb):
36
37
    >>> intersect((0, 9), (7, 15))
37
38
    (7, 9)
38
39
    """
39
 
    assert ra[0] <= ra[1]
40
 
    assert rb[0] <= rb[1]
41
 
    
 
40
    # preconditions: (ra[0] <= ra[1]) and (rb[0] <= rb[1])
 
41
 
42
42
    sa = max(ra[0], rb[0])
43
43
    sb = min(ra[1], rb[1])
44
44
    if sa < sb:
57
57
            return False
58
58
    else:
59
59
        return True
60
 
        
 
60
 
61
61
 
62
62
 
63
63
 
67
67
    Given BASE, OTHER, THIS, tries to produce a combined text
68
68
    incorporating the changes from both BASE->OTHER and BASE->THIS.
69
69
    All three will typically be sequences of lines."""
70
 
    def __init__(self, base, a, b):
71
 
        check_text_lines(base)
72
 
        check_text_lines(a)
73
 
        check_text_lines(b)
 
70
 
 
71
    def __init__(self, base, a, b, is_cherrypick=False, allow_objects=False):
 
72
        """Constructor.
 
73
 
 
74
        :param base: lines in BASE
 
75
        :param a: lines in A
 
76
        :param b: lines in B
 
77
        :param is_cherrypick: flag indicating if this merge is a cherrypick.
 
78
            When cherrypicking b => a, matches with b and base do not conflict.
 
79
        :param allow_objects: if True, do not require that base, a and b are
 
80
            plain Python strs.  Also prevents BinaryFile from being raised.
 
81
            Lines can be any sequence of comparable and hashable Python
 
82
            objects.
 
83
        """
 
84
        if not allow_objects:
 
85
            textfile.check_text_lines(base)
 
86
            textfile.check_text_lines(a)
 
87
            textfile.check_text_lines(b)
74
88
        self.base = base
75
89
        self.a = a
76
90
        self.b = b
77
 
 
78
 
 
 
91
        self.is_cherrypick = is_cherrypick
79
92
 
80
93
    def merge_lines(self,
81
94
                    name_a=None,
95
108
            elif self.a[0].endswith('\r'):
96
109
                newline = '\r'
97
110
        if base_marker and reprocess:
98
 
            raise CantReprocessAndShowBase()
 
111
            raise errors.CantReprocessAndShowBase()
99
112
        if name_a:
100
113
            start_marker = start_marker + ' ' + name_a
101
114
        if name_b:
130
143
                yield end_marker + newline
131
144
            else:
132
145
                raise ValueError(what)
133
 
        
134
 
        
135
 
 
136
 
 
137
146
 
138
147
    def merge_annotated(self):
139
148
        """Return merge with conflicts, showing origin of lines.
140
149
 
141
 
        Most useful for debugging merge.        
 
150
        Most useful for debugging merge.
142
151
        """
143
152
        for t in self.merge_regions():
144
153
            what = t[0]
161
170
                yield '>>>>\n'
162
171
            else:
163
172
                raise ValueError(what)
164
 
        
165
 
        
166
 
 
167
 
 
168
173
 
169
174
    def merge_groups(self):
170
175
        """Yield sequence of line groups.  Each one is a tuple:
200
205
            else:
201
206
                raise ValueError(what)
202
207
 
203
 
 
204
208
    def merge_regions(self):
205
209
        """Return sequences of matching and conflicting regions.
206
210
 
230
234
 
231
235
        # section a[0:ia] has been disposed of, etc
232
236
        iz = ia = ib = 0
233
 
        
 
237
 
234
238
        for zmatch, zend, amatch, aend, bmatch, bend in self.find_sync_regions():
235
 
            #print 'match base [%d:%d]' % (zmatch, zend)
236
 
            
237
239
            matchlen = zend - zmatch
238
 
            assert matchlen >= 0
239
 
            assert matchlen == (aend - amatch)
240
 
            assert matchlen == (bend - bmatch)
241
 
            
 
240
            # invariants:
 
241
            #   matchlen >= 0
 
242
            #   matchlen == (aend - amatch)
 
243
            #   matchlen == (bend - bmatch)
242
244
            len_a = amatch - ia
243
245
            len_b = bmatch - ib
244
246
            len_base = zmatch - iz
245
 
            assert len_a >= 0
246
 
            assert len_b >= 0
247
 
            assert len_base >= 0
 
247
            # invariants:
 
248
            # assert len_a >= 0
 
249
            # assert len_b >= 0
 
250
            # assert len_base >= 0
248
251
 
249
252
            #print 'unmatched a=%d, b=%d' % (len_a, len_b)
250
253
 
251
254
            if len_a or len_b:
252
255
                # try to avoid actually slicing the lists
253
 
                equal_a = compare_range(self.a, ia, amatch,
254
 
                                        self.base, iz, zmatch)
255
 
                equal_b = compare_range(self.b, ib, bmatch,
256
 
                                        self.base, iz, zmatch)
257
256
                same = compare_range(self.a, ia, amatch,
258
257
                                     self.b, ib, bmatch)
259
258
 
260
259
                if same:
261
260
                    yield 'same', ia, amatch
262
 
                elif equal_a and not equal_b:
263
 
                    yield 'b', ib, bmatch
264
 
                elif equal_b and not equal_a:
265
 
                    yield 'a', ia, amatch
266
 
                elif not equal_a and not equal_b:
267
 
                    yield 'conflict', iz, zmatch, ia, amatch, ib, bmatch
268
261
                else:
269
 
                    raise AssertionError("can't handle a=b=base but unmatched")
 
262
                    equal_a = compare_range(self.a, ia, amatch,
 
263
                                            self.base, iz, zmatch)
 
264
                    equal_b = compare_range(self.b, ib, bmatch,
 
265
                                            self.base, iz, zmatch)
 
266
                    if equal_a and not equal_b:
 
267
                        yield 'b', ib, bmatch
 
268
                    elif equal_b and not equal_a:
 
269
                        yield 'a', ia, amatch
 
270
                    elif not equal_a and not equal_b:
 
271
                        if self.is_cherrypick:
 
272
                            for node in self._refine_cherrypick_conflict(
 
273
                                                    iz, zmatch, ia, amatch,
 
274
                                                    ib, bmatch):
 
275
                                yield node
 
276
                        else:
 
277
                            yield 'conflict', iz, zmatch, ia, amatch, ib, bmatch
 
278
                    else:
 
279
                        raise AssertionError("can't handle a=b=base but unmatched")
270
280
 
271
281
                ia = amatch
272
282
                ib = bmatch
275
285
            # if the same part of the base was deleted on both sides
276
286
            # that's OK, we can just skip it.
277
287
 
278
 
                
279
288
            if matchlen > 0:
280
 
                assert ia == amatch
281
 
                assert ib == bmatch
282
 
                assert iz == zmatch
283
 
                
 
289
                # invariants:
 
290
                # assert ia == amatch
 
291
                # assert ib == bmatch
 
292
                # assert iz == zmatch
 
293
 
284
294
                yield 'unchanged', zmatch, zend
285
295
                iz = zend
286
296
                ia = aend
287
297
                ib = bend
288
 
    
 
298
 
 
299
    def _refine_cherrypick_conflict(self, zstart, zend, astart, aend, bstart, bend):
 
300
        """When cherrypicking b => a, ignore matches with b and base."""
 
301
        # Do not emit regions which match, only regions which do not match
 
302
        matches = patiencediff.PatienceSequenceMatcher(None,
 
303
            self.base[zstart:zend], self.b[bstart:bend]).get_matching_blocks()
 
304
        last_base_idx = 0
 
305
        last_b_idx = 0
 
306
        last_b_idx = 0
 
307
        yielded_a = False
 
308
        for base_idx, b_idx, match_len in matches:
 
309
            conflict_z_len = base_idx - last_base_idx
 
310
            conflict_b_len = b_idx - last_b_idx
 
311
            if conflict_b_len == 0: # There are no lines in b which conflict,
 
312
                                    # so skip it
 
313
                pass
 
314
            else:
 
315
                if yielded_a:
 
316
                    yield ('conflict',
 
317
                           zstart + last_base_idx, zstart + base_idx,
 
318
                           aend, aend, bstart + last_b_idx, bstart + b_idx)
 
319
                else:
 
320
                    # The first conflict gets the a-range
 
321
                    yielded_a = True
 
322
                    yield ('conflict', zstart + last_base_idx, zstart +
 
323
                    base_idx,
 
324
                           astart, aend, bstart + last_b_idx, bstart + b_idx)
 
325
            last_base_idx = base_idx + match_len
 
326
            last_b_idx = b_idx + match_len
 
327
        if last_base_idx != zend - zstart or last_b_idx != bend - bstart:
 
328
            if yielded_a:
 
329
                yield ('conflict', zstart + last_base_idx, zstart + base_idx,
 
330
                       aend, aend, bstart + last_b_idx, bstart + b_idx)
 
331
            else:
 
332
                # The first conflict gets the a-range
 
333
                yielded_a = True
 
334
                yield ('conflict', zstart + last_base_idx, zstart + base_idx,
 
335
                       astart, aend, bstart + last_b_idx, bstart + b_idx)
 
336
        if not yielded_a:
 
337
            yield ('conflict', zstart, zend, astart, aend, bstart, bend)
289
338
 
290
339
    def reprocess_merge_regions(self, merge_regions):
291
340
        """Where there are conflict regions, remove the agreed lines.
292
341
 
293
 
        Lines where both A and B have made the same changes are 
 
342
        Lines where both A and B have made the same changes are
294
343
        eliminated.
295
344
        """
296
345
        for region in merge_regions:
300
349
            type, iz, zmatch, ia, amatch, ib, bmatch = region
301
350
            a_region = self.a[ia:amatch]
302
351
            b_region = self.b[ib:bmatch]
303
 
            matches = bzrlib.patiencediff.PatienceSequenceMatcher(
 
352
            matches = patiencediff.PatienceSequenceMatcher(
304
353
                    None, a_region, b_region).get_matching_blocks()
305
354
            next_a = ia
306
355
            next_b = ib
318
367
            if reg is not None:
319
368
                yield reg
320
369
 
321
 
 
322
370
    @staticmethod
323
371
    def mismatch_region(next_a, region_ia,  next_b, region_ib):
324
372
        if next_a < region_ia or next_b < region_ib:
325
373
            return 'conflict', None, None, next_a, region_ia, next_b, region_ib
326
 
            
327
374
 
328
375
    def find_sync_regions(self):
329
376
        """Return a list of sync regions, where both descendents match the base.
333
380
        """
334
381
 
335
382
        ia = ib = 0
336
 
        amatches = bzrlib.patiencediff.PatienceSequenceMatcher(
 
383
        amatches = patiencediff.PatienceSequenceMatcher(
337
384
                None, self.base, self.a).get_matching_blocks()
338
 
        bmatches = bzrlib.patiencediff.PatienceSequenceMatcher(
 
385
        bmatches = patiencediff.PatienceSequenceMatcher(
339
386
                None, self.base, self.b).get_matching_blocks()
340
387
        len_a = len(amatches)
341
388
        len_b = len(bmatches)
356
403
 
357
404
                # found a match of base[i[0], i[1]]; this may be less than
358
405
                # the region that matches in either one
359
 
                assert intlen <= alen
360
 
                assert intlen <= blen
361
 
                assert abase <= intbase
362
 
                assert bbase <= intbase
 
406
                # assert intlen <= alen
 
407
                # assert intlen <= blen
 
408
                # assert abase <= intbase
 
409
                # assert bbase <= intbase
363
410
 
364
411
                asub = amatch + (intbase - abase)
365
412
                bsub = bmatch + (intbase - bbase)
366
413
                aend = asub + intlen
367
414
                bend = bsub + intlen
368
415
 
369
 
                assert self.base[intbase:intend] == self.a[asub:aend], \
370
 
                       (self.base[intbase:intend], self.a[asub:aend])
371
 
 
372
 
                assert self.base[intbase:intend] == self.b[bsub:bend]
 
416
                # assert self.base[intbase:intend] == self.a[asub:aend], \
 
417
                #       (self.base[intbase:intend], self.a[asub:aend])
 
418
                # assert self.base[intbase:intend] == self.b[bsub:bend]
373
419
 
374
420
                sl.append((intbase, intend,
375
421
                           asub, aend,
376
422
                           bsub, bend))
377
 
 
378
423
            # advance whichever one ends first in the base text
379
424
            if (abase + alen) < (bbase + blen):
380
425
                ia += 1
381
426
            else:
382
427
                ib += 1
383
 
            
 
428
 
384
429
        intbase = len(self.base)
385
430
        abase = len(self.a)
386
431
        bbase = len(self.b)
388
433
 
389
434
        return sl
390
435
 
391
 
 
392
 
 
393
436
    def find_unconflicted(self):
394
437
        """Return a list of ranges in base that are not conflicted."""
395
 
        am = bzrlib.patiencediff.PatienceSequenceMatcher(
 
438
        am = patiencediff.PatienceSequenceMatcher(
396
439
                None, self.base, self.a).get_matching_blocks()
397
 
        bm = bzrlib.patiencediff.PatienceSequenceMatcher(
 
440
        bm = patiencediff.PatienceSequenceMatcher(
398
441
                None, self.base, self.b).get_matching_blocks()
399
442
 
400
443
        unc = []
414
457
                del am[0]
415
458
            else:
416
459
                del bm[0]
417
 
                
 
460
 
418
461
        return unc
419
462
 
420
463