← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/merge3.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2006-05-19 06:14:38 UTC
  • mfrom: (1704.2.23 bzr.mbp.integration)
  • Revision ID: pqm@pqm.ubuntu.com-20060519061438-6300caf3926c3cff
(mbp) small fixes

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/merge3.py
1
 
# Copyright (C) 2004, 2005 Canonical Ltd
2
 
#
 
1
# Copyright (C) 2004, 2005 by Canonical Ltd
 
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
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
#
 
7
 
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
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
 
18
18
# mbp: "you know that thing where cvs gives you conflict markers?"
19
19
# s: "i hate that."
20
20
 
21
21
 
 
22
from difflib import SequenceMatcher
 
23
 
22
24
from bzrlib.errors import CantReprocessAndShowBase
23
 
import bzrlib.patiencediff
24
25
from bzrlib.textfile import check_text_lines
25
26
 
26
 
 
27
27
def intersect(ra, rb):
28
28
    """Given two ranges return the range where they intersect or None.
29
29
 
36
36
    >>> intersect((0, 9), (7, 15))
37
37
    (7, 9)
38
38
    """
39
 
    # preconditions: (ra[0] <= ra[1]) and (rb[0] <= rb[1])
40
 
 
 
39
    assert ra[0] <= ra[1]
 
40
    assert rb[0] <= rb[1]
 
41
    
41
42
    sa = max(ra[0], rb[0])
42
43
    sb = min(ra[1], rb[1])
43
44
    if sa < sb:
56
57
            return False
57
58
    else:
58
59
        return True
59
 
 
 
60
        
60
61
 
61
62
 
62
63
 
66
67
    Given BASE, OTHER, THIS, tries to produce a combined text
67
68
    incorporating the changes from both BASE->OTHER and BASE->THIS.
68
69
    All three will typically be sequences of lines."""
69
 
    def __init__(self, base, a, b, is_cherrypick=False):
 
70
    def __init__(self, base, a, b):
70
71
        check_text_lines(base)
71
72
        check_text_lines(a)
72
73
        check_text_lines(b)
73
74
        self.base = base
74
75
        self.a = a
75
76
        self.b = b
76
 
        self.is_cherrypick = is_cherrypick
 
77
 
 
78
 
77
79
 
78
80
    def merge_lines(self,
79
81
                    name_a=None,
86
88
                    reprocess=False):
87
89
        """Return merge in cvs-like form.
88
90
        """
89
 
        newline = '\n'
90
 
        if len(self.a) > 0:
91
 
            if self.a[0].endswith('\r\n'):
92
 
                newline = '\r\n'
93
 
            elif self.a[0].endswith('\r'):
94
 
                newline = '\r'
95
91
        if base_marker and reprocess:
96
92
            raise CantReprocessAndShowBase()
97
93
        if name_a:
115
111
                for i in range(t[1], t[2]):
116
112
                    yield self.b[i]
117
113
            elif what == 'conflict':
118
 
                yield start_marker + newline
 
114
                yield start_marker + '\n'
119
115
                for i in range(t[3], t[4]):
120
116
                    yield self.a[i]
121
117
                if base_marker is not None:
122
 
                    yield base_marker + newline
 
118
                    yield base_marker + '\n'
123
119
                    for i in range(t[1], t[2]):
124
120
                        yield self.base[i]
125
 
                yield mid_marker + newline
 
121
                yield mid_marker + '\n'
126
122
                for i in range(t[5], t[6]):
127
123
                    yield self.b[i]
128
 
                yield end_marker + newline
 
124
                yield end_marker + '\n'
129
125
            else:
130
126
                raise ValueError(what)
 
127
        
 
128
        
 
129
 
 
130
 
131
131
 
132
132
    def merge_annotated(self):
133
133
        """Return merge with conflicts, showing origin of lines.
134
134
 
135
 
        Most useful for debugging merge.
 
135
        Most useful for debugging merge.        
136
136
        """
137
137
        for t in self.merge_regions():
138
138
            what = t[0]
155
155
                yield '>>>>\n'
156
156
            else:
157
157
                raise ValueError(what)
 
158
        
 
159
        
 
160
 
 
161
 
158
162
 
159
163
    def merge_groups(self):
160
164
        """Yield sequence of line groups.  Each one is a tuple:
190
194
            else:
191
195
                raise ValueError(what)
192
196
 
 
197
 
193
198
    def merge_regions(self):
194
199
        """Return sequences of matching and conflicting regions.
195
200
 
219
224
 
220
225
        # section a[0:ia] has been disposed of, etc
221
226
        iz = ia = ib = 0
222
 
 
 
227
        
223
228
        for zmatch, zend, amatch, aend, bmatch, bend in self.find_sync_regions():
 
229
            #print 'match base [%d:%d]' % (zmatch, zend)
 
230
            
224
231
            matchlen = zend - zmatch
225
 
            # invariants:
226
 
            #   matchlen >= 0
227
 
            #   matchlen == (aend - amatch)
228
 
            #   matchlen == (bend - bmatch)
 
232
            assert matchlen >= 0
 
233
            assert matchlen == (aend - amatch)
 
234
            assert matchlen == (bend - bmatch)
 
235
            
229
236
            len_a = amatch - ia
230
237
            len_b = bmatch - ib
231
238
            len_base = zmatch - iz
232
 
            # invariants:
233
 
            # assert len_a >= 0
234
 
            # assert len_b >= 0
235
 
            # assert len_base >= 0
 
239
            assert len_a >= 0
 
240
            assert len_b >= 0
 
241
            assert len_base >= 0
236
242
 
237
243
            #print 'unmatched a=%d, b=%d' % (len_a, len_b)
238
244
 
239
245
            if len_a or len_b:
240
246
                # try to avoid actually slicing the lists
 
247
                equal_a = compare_range(self.a, ia, amatch,
 
248
                                        self.base, iz, zmatch)
 
249
                equal_b = compare_range(self.b, ib, bmatch,
 
250
                                        self.base, iz, zmatch)
241
251
                same = compare_range(self.a, ia, amatch,
242
252
                                     self.b, ib, bmatch)
243
253
 
244
254
                if same:
245
255
                    yield 'same', ia, amatch
 
256
                elif equal_a and not equal_b:
 
257
                    yield 'b', ib, bmatch
 
258
                elif equal_b and not equal_a:
 
259
                    yield 'a', ia, amatch
 
260
                elif not equal_a and not equal_b:
 
261
                    yield 'conflict', iz, zmatch, ia, amatch, ib, bmatch
246
262
                else:
247
 
                    equal_a = compare_range(self.a, ia, amatch,
248
 
                                            self.base, iz, zmatch)
249
 
                    equal_b = compare_range(self.b, ib, bmatch,
250
 
                                            self.base, iz, zmatch)
251
 
                    if equal_a and not equal_b:
252
 
                        yield 'b', ib, bmatch
253
 
                    elif equal_b and not equal_a:
254
 
                        yield 'a', ia, amatch
255
 
                    elif not equal_a and not equal_b:
256
 
                        if self.is_cherrypick:
257
 
                            for node in self._refine_cherrypick_conflict(
258
 
                                                    iz, zmatch, ia, amatch,
259
 
                                                    ib, bmatch):
260
 
                                yield node
261
 
                        else:
262
 
                            yield 'conflict', iz, zmatch, ia, amatch, ib, bmatch
263
 
                    else:
264
 
                        raise AssertionError("can't handle a=b=base but unmatched")
 
263
                    raise AssertionError("can't handle a=b=base but unmatched")
265
264
 
266
265
                ia = amatch
267
266
                ib = bmatch
270
269
            # if the same part of the base was deleted on both sides
271
270
            # that's OK, we can just skip it.
272
271
 
 
272
                
273
273
            if matchlen > 0:
274
 
                # invariants:
275
 
                # assert ia == amatch
276
 
                # assert ib == bmatch
277
 
                # assert iz == zmatch
278
 
 
 
274
                assert ia == amatch
 
275
                assert ib == bmatch
 
276
                assert iz == zmatch
 
277
                
279
278
                yield 'unchanged', zmatch, zend
280
279
                iz = zend
281
280
                ia = aend
282
281
                ib = bend
283
 
 
284
 
    def _refine_cherrypick_conflict(self, zstart, zend, astart, aend, bstart, bend):
285
 
        """When cherrypicking b => a, ignore matches with b and base."""
286
 
        # Do not emit regions which match, only regions which do not match
287
 
        matches = bzrlib.patiencediff.PatienceSequenceMatcher(None,
288
 
            self.base[zstart:zend], self.b[bstart:bend]).get_matching_blocks()
289
 
        last_base_idx = 0
290
 
        last_b_idx = 0
291
 
        last_b_idx = 0
292
 
        yielded_a = False
293
 
        for base_idx, b_idx, match_len in matches:
294
 
            conflict_z_len = base_idx - last_base_idx
295
 
            conflict_b_len = b_idx - last_b_idx
296
 
            if conflict_b_len == 0: # There are no lines in b which conflict,
297
 
                                    # so skip it
298
 
                pass
299
 
            else:
300
 
                if yielded_a:
301
 
                    yield ('conflict',
302
 
                           zstart + last_base_idx, zstart + base_idx,
303
 
                           aend, aend, bstart + last_b_idx, bstart + b_idx)
304
 
                else:
305
 
                    # The first conflict gets the a-range
306
 
                    yielded_a = True
307
 
                    yield ('conflict', zstart + last_base_idx, zstart +
308
 
                    base_idx,
309
 
                           astart, aend, bstart + last_b_idx, bstart + b_idx)
310
 
            last_base_idx = base_idx + match_len
311
 
            last_b_idx = b_idx + match_len
312
 
        if last_base_idx != zend - zstart or last_b_idx != bend - bstart:
313
 
            if yielded_a:
314
 
                yield ('conflict', zstart + last_base_idx, zstart + base_idx,
315
 
                       aend, aend, bstart + last_b_idx, bstart + b_idx)
316
 
            else:
317
 
                # The first conflict gets the a-range
318
 
                yielded_a = True
319
 
                yield ('conflict', zstart + last_base_idx, zstart + base_idx,
320
 
                       astart, aend, bstart + last_b_idx, bstart + b_idx)
321
 
        if not yielded_a:
322
 
            yield ('conflict', zstart, zend, astart, aend, bstart, bend)
 
282
    
323
283
 
324
284
    def reprocess_merge_regions(self, merge_regions):
325
285
        """Where there are conflict regions, remove the agreed lines.
326
286
 
327
 
        Lines where both A and B have made the same changes are
 
287
        Lines where both A and B have made the same changes are 
328
288
        eliminated.
329
289
        """
330
290
        for region in merge_regions:
334
294
            type, iz, zmatch, ia, amatch, ib, bmatch = region
335
295
            a_region = self.a[ia:amatch]
336
296
            b_region = self.b[ib:bmatch]
337
 
            matches = bzrlib.patiencediff.PatienceSequenceMatcher(
338
 
                    None, a_region, b_region).get_matching_blocks()
 
297
            matches = SequenceMatcher(None, a_region, 
 
298
                                      b_region).get_matching_blocks()
339
299
            next_a = ia
340
300
            next_b = ib
341
301
            for region_ia, region_ib, region_len in matches[:-1]:
352
312
            if reg is not None:
353
313
                yield reg
354
314
 
 
315
 
355
316
    @staticmethod
356
317
    def mismatch_region(next_a, region_ia,  next_b, region_ib):
357
318
        if next_a < region_ia or next_b < region_ib:
358
319
            return 'conflict', None, None, next_a, region_ia, next_b, region_ib
 
320
            
359
321
 
360
322
    def find_sync_regions(self):
361
323
        """Return a list of sync regions, where both descendents match the base.
365
327
        """
366
328
 
367
329
        ia = ib = 0
368
 
        amatches = bzrlib.patiencediff.PatienceSequenceMatcher(
369
 
                None, self.base, self.a).get_matching_blocks()
370
 
        bmatches = bzrlib.patiencediff.PatienceSequenceMatcher(
371
 
                None, self.base, self.b).get_matching_blocks()
 
330
        amatches = SequenceMatcher(None, self.base, self.a).get_matching_blocks()
 
331
        bmatches = SequenceMatcher(None, self.base, self.b).get_matching_blocks()
372
332
        len_a = len(amatches)
373
333
        len_b = len(bmatches)
374
334
 
388
348
 
389
349
                # found a match of base[i[0], i[1]]; this may be less than
390
350
                # the region that matches in either one
391
 
                # assert intlen <= alen
392
 
                # assert intlen <= blen
393
 
                # assert abase <= intbase
394
 
                # assert bbase <= intbase
 
351
                assert intlen <= alen
 
352
                assert intlen <= blen
 
353
                assert abase <= intbase
 
354
                assert bbase <= intbase
395
355
 
396
356
                asub = amatch + (intbase - abase)
397
357
                bsub = bmatch + (intbase - bbase)
398
358
                aend = asub + intlen
399
359
                bend = bsub + intlen
400
360
 
401
 
                # assert self.base[intbase:intend] == self.a[asub:aend], \
402
 
                #       (self.base[intbase:intend], self.a[asub:aend])
403
 
                # assert self.base[intbase:intend] == self.b[bsub:bend]
 
361
                assert self.base[intbase:intend] == self.a[asub:aend], \
 
362
                       (self.base[intbase:intend], self.a[asub:aend])
 
363
 
 
364
                assert self.base[intbase:intend] == self.b[bsub:bend]
404
365
 
405
366
                sl.append((intbase, intend,
406
367
                           asub, aend,
407
368
                           bsub, bend))
 
369
 
408
370
            # advance whichever one ends first in the base text
409
371
            if (abase + alen) < (bbase + blen):
410
372
                ia += 1
411
373
            else:
412
374
                ib += 1
413
 
 
 
375
            
414
376
        intbase = len(self.base)
415
377
        abase = len(self.a)
416
378
        bbase = len(self.b)
418
380
 
419
381
        return sl
420
382
 
 
383
 
 
384
 
421
385
    def find_unconflicted(self):
422
386
        """Return a list of ranges in base that are not conflicted."""
423
 
        am = bzrlib.patiencediff.PatienceSequenceMatcher(
424
 
                None, self.base, self.a).get_matching_blocks()
425
 
        bm = bzrlib.patiencediff.PatienceSequenceMatcher(
426
 
                None, self.base, self.b).get_matching_blocks()
 
387
 
 
388
        import re
 
389
 
 
390
        # don't sync-up on lines containing only blanks or pounds
 
391
        junk_re = re.compile(r'^[ \t#]*$')
 
392
        
 
393
        am = SequenceMatcher(junk_re.match, self.base, self.a).get_matching_blocks()
 
394
        bm = SequenceMatcher(junk_re.match, self.base, self.b).get_matching_blocks()
427
395
 
428
396
        unc = []
429
397
 
442
410
                del am[0]
443
411
            else:
444
412
                del bm[0]
445
 
 
 
413
                
446
414
        return unc
447
415
 
448
416